from ChanEnum import Chan_MACDHISTSET_DIR

class ChanMACDHistSet():
    def __init__(self, start_time, start_klu, pre_histset, dir):
        self.start_time = start_time
        self.end_time = None
        self.klu_list = []
        self.klu_list.append(start_klu)
        self.ref_klu = None
        self.histset_dir = dir
        self.next = None
        self.pre = pre_histset
        self.high_klu = start_klu
        self.low_klu = start_klu
    def set_next(self, next_histset):
        self.next = next_histset
    def add_klu(self, klu):
        self.klu_list.append(klu)
        klu.set_histset(self)
        if self.histset_dir == Chan_MACDHISTSET_DIR.ABOVE:
            if klu.macdhist > self.high_klu.macdhist:
                self.high_klu = klu
            else:
                if klu.macdhist < self.low_klu.macdhist:
                    self.low_klu = klu
        else:
            if klu.macdhist < self.high_klu.macdhist:
                self.high_klu = klu
            else:
                if klu.macdhist > self.low_klu.macdhist:
                    self.low_klu = klu
    def set_end_klu(self, end_klu):
        self.add_klu(end_klu)
        self.end_klu = end_klu
        self.end_time = end_klu.time