"""
	This file has some functions to read or write files.
"""

import random
from preprocessOfApneaECG.mit2Segments import Mit2Segment, SEGMENTS_BASE_PATH, SEGMENTS_NUMBER_TRAIN


def get_database(database_name, rdf, numRead=-1, is_debug=False):
	"""
	Return the database you want to obtain.
	:param list database_name: name of database
	:param int rdf: 0 means read raw ecg, 1 means denoised ecg.
	:param int numRead: the number you want read from database, -1 means read all data.
	:param bool is_debug: whether is debug mode.
	:return Mit2Segment list: database
	"""
	
	if database_name == ["apnea-ecg", "train"]:
		base_floder_path = SEGMENTS_BASE_PATH + "train/"
	elif database_name == ["apnea-ecg", "test"]:
		base_floder_path = SEGMENTS_BASE_PATH + "test/"
	else:
		raise Exception("Error database name.")
	
	# get the segment amount
	read_file_path = base_floder_path + "/extra_info.txt"
	with open(read_file_path) as f:
		_ = f.readline()
		attrs_value = f.readline().replace("\n", "").split(" ")
		segment_amount = int(attrs_value[0])
	# read ecg segment
	read_num = 0
	database = []
	for segment_number in range(segment_amount):
		if is_debug is True:
			print("now read file: " + str(segment_number))
		if numRead != -1 and read_num >= numRead:
			break
		eds = Mit2Segment()
		eds.global_id = segment_number
		eds.read_ecg_segment(rdf, database_name)
		database.append(eds)
		read_num += 1
	if is_debug is True:
		print("length of database: %s" % len(database))
	return database


def my_random_func(length, min_value, max_value):
	"""
	Given a range and the list length, return a no duplicates list.
	:param int length: list length.
	:param int min_value: min of range.
	:param int max_value: max of range.
	:return list: random list.
	"""
	random_list = []
	count = 0
	while count < length:
		random_value = random.randint(min_value, max_value)
		if random_value not in random_list:
			random_list.append(random_value)
			count = count + 1
	
	return random_list


def write_txt_file(list_info, write_file_path):
	"""
	Write list object to TXT file.
	:param list list_info: List object you want to write.
	:param string write_file_path: TXT file path.
	:return: None
	"""
	with open(write_file_path, "w") as f:
		for info in list_info:
			f.write(str(info) + "\n")


def read_txt_file(read_file_path):
	"""
	Read TXT file to list object.
	:param string read_file_path: TXT file path.
	:return list: list object
	"""
	with open(read_file_path) as f:
		list_t = []
		lines = f.readlines()
		for line in lines:
			float_list = []
			str_t = line.replace("[", "").replace("]", "").replace("\n", "").split(",")
			for str_s in str_t:
				if str_s != "":
					float_list.append(float(str_s))
				else:
					print(read_file_path)
			list_t.append(float_list)
	
	return list_t




def get_noise_dataset():
	"""
	读取ECG噪音片段.
	:return:
	"""
	
	id_path = "G:/Apnea-ecg/ecg_segments/data/denosing ecg data/" + "train/train_noise_id_matlab.txt"
	
	id_set = []
	with open(id_path) as f:
		lines = f.readlines()
		for line in lines:
			line = line.replace("\n", "")
			id_set.append(int(float(line)))
	
	# read ecg segment
	database = []
	for segment_number in id_set:
		print("now read file: " + str(segment_number))
		
		eds = ECGDataSegment()
		eds.global_id = segment_number
		# read_file_path = base_floder_path + str(segment_number) + "/" + "ecg segment data.txt"
		eds.read_denosing_ecg_segment("train")
		database.append(eds)
	
	print("length of database: %s" % len(database))
	
	return database


def get_some_ecg_segments(number):
	"""
	Return some ecg samples for debug.
	:param int number: the number of segments you want to sample from all ecg segments.
	:return list: ecg segments.
	"""
	
	random_list = my_random_func(number, 0, SEGMENTS_NUMBER_TRAIN)
	
	base_floder_path = SEGMENTS_BASE_PATH + "raw ecg data/train/"
	
	sample_segments = []
	for value in random_list:
		print("read %s" % value)
		eds = ECGDataSegment()
		eds.global_id = value
		read_file_path = base_floder_path + str(value) + "/" + "ecg segment data.txt"
		eds.read_raw_ecg_segment(read_file_path)
		sample_segments.append(eds)
	return sample_segments


def get_A_N_number(database):
	"""
	求出数据库database中A和N的片段数量.
	:param database:
	:return:
	"""
	
	A_number = 0
	N_number = 0
	for segment in database:
		if segment.label == 0:
			N_number += 1
		else:
			A_number += 1
	print("A number...")
	print(A_number)
	print("N number...")
	print(N_number)


if __name__ == '__main__':
	"""
	Test sentences:
	"""
	# Read trainset of apnea-ecg
	# trainset = get_database(["apnea-ecg", "train"], 0, is_debug=True)
	# Read testset of apnea-ecg
	testset = get_database(["apnea-ecg", "test"], 0, is_debug=True)