Add seg calculation

2025-06-23 09:37:11 +08:00
parent 08ca03fc3d
commit 69a0a46a60
2 changed files with 80 additions and 12 deletions
@@ -43,6 +43,7 @@ class ChanSEG():
            self.end_time = bi.end_klc.end_time
        if sure_bi.is_sure:
            self.sure_time = sure_bi.sure_time
+        self.format_bi_list()
    def pre_set_end_bi(self, bi: ChanBI):
        self.end_bi = bi
        if bi:
@@ -51,6 +52,7 @@ class ChanSEG():
            else:
                self.low = bi.low
            self.end_time = bi.end_klc.end_time
+        self.format_bi_list()
    def set_pre(self, seg):
        self.pre = seg
    def set_next(self, seg):
@@ -59,6 +61,16 @@ class ChanSEG():
        if sure_bi.is_sure:
            self.sure_time = sure_bi.sure_time
        self.is_sure = True
+        self.format_bi_list()
+    def format_bi_list(self):
+        bi_list = []
+        bi_list.append(self.start_bi)
+        if self.end_bi:
+            next_bi = self.start_bi.next
+            for i in range(self.start_bi.index, self.end_bi.index):
+                if next_bi:
+                    self.bi_list.append(next_bi)
+                    next_bi = next_bi.next
    def add_bi(self, bi: ChanBI):
-        if len(self.bi_list) > 1:
+        if len(self.bi_list) > 0:
            self.bi_list.append(bi)