import numpy as np
from preprocessOfApneaECG.mit2Segments import Mit2Segment, SEGMENTS_BASE_PATH, SEGMENTS_NUMBER_TRAIN


class ECGSegment(Mit2Segment):
	"""
		Every ECGSegment object includes some base informations and feature vectors.
	"""
	def __init__(self):
		# base attributions
		super(ECGSegment, self).__init__()
		# features
		self.RR_intervals = []
		self.R_peaks_amplitude = []
		self.EDR = []
	
	def read_rri_ramp_edr(self):
		self.RR_intervals = np.load(self.base_file_path + "/RRI.npy")  # RR intervals extracted from ECG.
		self.R_peaks_amplitude = np.load(self.base_file_path + "/RAMP.npy")  # R peaks amplitude extracted from ECG.
		self.EDR = np.load(self.base_file_path + "/EDR.npy")  # ECG-Derived Respiration signal