Initial commit

ChanBI.py

@@ -0,0 +1,67 @@
+import ChanKLC
+from ChanEnum import Chan_BI_DIR
+class ChanBI():
+    def __init__(self, klc: ChanKLC, index, ddir=Chan_BI_DIR.UP):
+        self.start_klc = klc
+        self.end_klc = None
+        self.next = None
+        self.pre = None
+        self.dir = ddir
+        self.index = index
+        self.is_sure = False
+        self.high = klc.high
+        self.low = klc.low
+        self.sure_time = None
+        self.klc_list = []
+        self.klc_list.append(klc)
+        self.end_time = None
+        self.start_time = klc.start_time
+        self.macd_hist = 0
+        self.macd_div = 0
+    def set_macd_hist(self, macd_hist):
+        self.macd_hist = macd_hist
+    def set_macd_div(self, macd_div):
+        self.macd_div = macd_div
+    def check_overlap(self):
+        if self.next and self.next.next:
+            if self.dir == Chan_BI_DIR.UP:
+                return self.high > self.next.low and self.high < self.next.next.high
+            else:
+                return self.high > self.next.high and self.low > self.next.next.low
+        else:
+            return False
+    def set_end_klc(self, klc, sure_klc):
+        if self.dir == Chan_BI_DIR.UP and klc.high > self.high:
+            self.high = klc.high
+        if self.dir == Chan_BI_DIR.DOWN and klc.low < self.low:
+            self.low = klc.low
+        self.end_klc = klc
+        self.set_is_sure(True, sure_klc.end_time)
+        self.end_time = klc.end_time
+    def set_is_sure(self, is_sure, time):
+        self.is_sure = is_sure
+        self.sure_time = time
+    def set_start_klc(self, klc, ddir):
+        self.start_klc = klc
+        self.klc_list = []
+        self.klc_list.append(klc)
+        self.high = klc.high
+        self.low = klc.low
+        self.dir = ddir
+    def set_pre(self, bi):
+        self.pre = bi
+    def set_next(self, bi):
+        self.next = bi
+    def add_klc(self, klc):
+        self.klc_list.append(klc)
+    def append_klc_list(self, klc_list):
+        self.klc_list.append(klc_list)
+    def update_bi(self, klc):
+        self.end_klc = None
+        if self.dir == Chan_BI_DIR.UP and klc.high > self.high:
+            self.high = klc.high
+        if self.dir == Chan_BI_DIR.DOWN and klc.low < self.low:
+            self.low = klc.low
+        self.is_sure = False
+        self.sure_time = None
+        #print(self.start_klc.start_time, klc.start_time, klc.fx, "This bi is extended")