Files
Chan/ChanMACDSeg.py
T

39 lines
1.2 KiB
Python

from ChanEnum import Chan_MACDSEG_DIR
class ChanMACDSeg():
def __init__(self, start_time, start_klu, pre_seg, seg_dir):
self.start_time = start_time
self.end_time = None
self.start_klu = start_klu
self.end_klu = None
self.klu_list = []
self.klu_list.append(start_klu)
self.unittf_list = []
self.hist_set = []
self.seg_dir = seg_dir
self.pre = pre_seg
self.next = None
self.peak_klu = start_klu
def set_next(self, next_seg):
self.next = next_seg
def add_klu(self, klu):
if klu:
self.klu_list.append(klu)
klu.set_seg(self)
if self.seg_dir == Chan_MACDSEG_DIR.ABOVE:
if klu.high > self.peak_klu.high:
self.peak_klu = klu
else:
if klu.low < self.peak_klu.low:
self.peak_klu = klu
def add_unittf(self, unittf):
self.unittf_list.append(unittf)
unittf.set_next(self)
def add_histset(self, histset):
self.hist_set.append(histset)
histset.set_next(self)
def set_end_klu(self, end_klu):
self.add_klu(end_klu)
self.end_klu = end_klu
self.end_time = end_klu.time