Add some classifier code

2025-04-23 10:39:04 +08:00
parent 70e14c2ea3
commit 48d5647ebf
14 changed files with 139 additions and 47 deletions
@@ -18,10 +18,25 @@ class ChanBI():
        self.start_time = klc.start_time
        self.macd_hist = 0
        self.macd_div = 0
-    def set_macd_hist(self, macd_hist):
+    def set_macdhist(self, macd_hist):
        self.macd_hist = macd_hist
    def set_macd_div(self, macd_div):
        self.macd_div = macd_div
+    def cal_macd_div(self):
+        self.macd_div = 0.0
+        if self.pre and self.pre.pre:
+            if self.pre.pre.macd_hist == 0:
+                self.macd_div = 0.0
+            else:
+                self.macd_div = self.macd_hist / self.pre.pre.macd_hist
+    def cal_macdhist(self):
+        self.macd_hist = 0
+        for klc in self.klc_list:
+            for klu in klc.klus:
+                if self.dir == Chan_BI_DIR.UP and klu.macdhist > 0: 
+                    self.macd_hist += klu.macdhist
+                if self.dir == Chan_BI_DIR.DOWN and klu.macdhist < 0:
+                    self.macd_hist -= klu.macdhist
    def check_overlap(self):
        if self.next and self.next.next:
            if self.dir == Chan_BI_DIR.UP:
@@ -61,6 +76,8 @@ class ChanBI():
                    break
        if not added:
            self.klc_list.append(klc)
+        self.cal_macdhist()
+        self.cal_macd_div()
    def append_klc_list(self, klc_list):
        self.klc_list.append(klc_list)
    def update_bi(self, klc):