use logging instead print

Former-commit-id: bff176e0e99d1c50f48a63a3fd30adcee67777a8

main.py

@@ -4,36 +4,41 @@ from datetime import datetime
 import numpy as np
 import ch340
 import atexit
+import utils
 
 class MAT:
     def __init__(self, videoSourceIndex=0, bounce_time=1):
-        print('Initializing MAT...')
+        # 初始化logging
+        utils.setup_logging()
+        self.system_logger = utils.get_system_logger()
+        self.control_logger = utils.get_control_logger()
+        self.endpoint_logger = utils.get_endpoint_logger()
+        self.volume_logger = utils.get_volume_logger()
+        
+        self.system_logger.info('正在初始化MAT系统...')
         self.videoSourceIndex = videoSourceIndex
         self.cap = cv2.VideoCapture(videoSourceIndex, cv2.CAP_DSHOW)
         self.ch340 = ch340.CH340()
         self.bounce_time = bounce_time
         self.total_volume = 0
-        self.start_time = time.time()
-        self.middle_time = time.time()
-        # 将开始时间转化为年月日时分秒的格式，后续文件命名都已此命名
-        self.formatted_time = datetime.fromtimestamp(self.start_time).strftime('%Y%m%d_%H%M%S')
-
+        self.start_time = 0
+        self.middle_time = 0
+        
         atexit.register(self.ch340.stop)
-        print("实验开始于", self.formatted_time)
         self.history = []  # 滑动窗口历史记录
         self.colored_volume = None  # 首次colored体积
         self.colored_time = None  # 首次colored时间
 
     def ch340_pull(self):
-        print("开始抽取12ml")
+        self.control_logger.info("开始抽取12ml")
         self.ch340.max_speed()
         self.ch340.pull(vol=12)
-        print('完成抽取')
+        self.control_logger.info('完成抽取')
 
     def ch340_init(self):
         self.ch340.push(speed=1,t=1)
         self.ch340.pull(speed=1.2,vol=3)
-        print('CH340 INITED')
+        self.control_logger.info('已初始化')
 
     def ch340_push(self, speed=0.1):
         self.ch340.push_async(speed=speed, t=1)
@@ -51,12 +56,11 @@ class MAT:
         """预测当前图像状态，返回'transport'、'middle'或'colored'"""
         suc, im = self.cap.read()
         if not suc:
-            print("Failed to capture frame from camera.")
+            self.control_logger.error("无法从摄像头捕获帧")
             return None
         
         ret,rate = self.predictor(im)
         if ret is None:
-            print("Fallback")
             return None
         
         now = time.time()
@@ -69,7 +73,7 @@ class MAT:
             self.history.pop(0)
         
         if not self.history:
-            print("Warning: No history available.")
+            self.control_logger.error("未预期的没有可用的历史记录")
             return ret
         
         # 处理检测结果
@@ -79,25 +83,24 @@ class MAT:
                 self.process_left(now)
                 self.typ = 1
                 self.middle_time = now
-                print(f"检测到middle，切换到慢速模式，当前体积: {val}")
+                self.control_logger.info(f"检测到middle，切换到慢速模式，当前体积: {val:.2f} ml")
         
         elif ret == "colored":
             # 检测到colored时记录当前体积
             if self.colored_volume is None:
                 self.colored_volume = val
                 self.colored_time = now
-                print(f"检测到colored，记录体积: {self.colored_volume}")
-            
-            # 检查是否要停止
-            if now - self.colored_time > self.bounce_time:
+                self.endpoint_logger.info(f"检测到colored，记录体积: {self.colored_volume:.2f} ml")
+              # 检查是否要停止
+            if self.colored_time is not None and now - self.colored_time > self.bounce_time:
                 colored_count = sum(1 for _, state, _ in self.history if state == "colored")
                 if colored_count / len(self.history) > 0.9:
-                    print(f"确认终点，最终体积: {self.colored_volume:0.2f} ml")
+                    self.endpoint_logger.info(f"确认终点，最终体积: {self.colored_volume:.2f} ml")
                     self.running = False
                     self.ch340.stop()
                     return "colored"
                 else:
-                    print(f"colored比例小于90%，当前体积: {val}, {colored_count / len(self.history)}")
+                    self.endpoint_logger.warning(f"colored比例小于90%，当前体积: {val:.2f} ml, 比例: {colored_count / len(self.history):.2%}")
                     self.colored_volume = val
                     self.colored_time = now
         else:  # ret == "transport"
@@ -111,13 +114,13 @@ class MAT:
                 self.process_left(now)
                 # TODO: 滑动到第一个middle状态
                 self.middle_time = 0
-                print(f"非middle比例超过90%，切回快速模式，当前体积: {val}")
+                self.control_logger.info(f"非middle比例超过90%，切回快速模式，当前体积: {val:.2f} ml")
         
-        # 如果已记录colored但在bounce_time内colored比例小于90%，重置
-        if self.colored_volume is not None and now - self.colored_time > self.bounce_time:
+          # 如果已记录colored但在bounce_time内colored比例小于90%，重置
+        if self.colored_volume is not None and self.colored_time is not None and now - self.colored_time > self.bounce_time:
             colored_count = sum(1 for _, state, _ in self.history if state == "colored")
             if colored_count / len(self.history) < 0.9:
-                print(f"colored比例小于90%，重置colored记录")
+                self.endpoint_logger.warning(f"colored比例小于90%，重置colored记录")
                 self.colored_volume = None
                 flag = False
                 for t, state,vol in self.history:
@@ -125,13 +128,14 @@ class MAT:
                         if not flag:
                             flag = True
                             continue
-                        print(f"滑动窗口到: {vol} at {t}")
+                        self.endpoint_logger.debug(f"滑动窗口到: {vol:.2f} ml at {t}")
                         self.colored_volume = vol
                         self.colored_time = t
                         break
             else:
-                print("Warning: Unusual condition, colored for over 90%% but last is transport")
-                print(f"疑似滴定终点: {self.colored_volume}")
+                self.endpoint_logger.warning("异常情况，colored超过90%但最后状态是transport")
+                self.endpoint_logger.info(f"疑似滴定终点: {self.colored_volume:.2f} ml")
+        
         cv2.putText(im, 
                     f"State: {ret}, rate: {round(rate,2)}, Vol: {val:.2f} ml, typ: {'slow' if self.typ else 'fast'}", 
                     (10, 30), cv2.FONT_HERSHEY_SIMPLEX, 0.7, (10,215, 255) if self.typ else (255,255,255), 2)
@@ -156,38 +160,52 @@ class MAT:
     def __del__(self):
         self.cap.release()
         cv2.destroyAllWindows()
-        print("Experiment finished.")
 
     def run(self,quick_speed = 0.2,slow_speed = 0.05):
         self.running = True
         self.typ = 0
         self.last = 0
         self.speeds = (quick_speed, slow_speed)
+        
+        # 记录实验开始
+        experiment_params = {
+            "视频源索引  ": self.videoSourceIndex,
+            "防抖时间    ": self.bounce_time,
+            "快速模式速度": f"{quick_speed} ml/次",
+            "慢速模式速度": f"{slow_speed} ml/次"
+        }
+        self.system_logger.info("实验参数:")
+        for key, value in experiment_params.items():
+            self.system_logger.info(f"  {key}: {value}")
+        
+        speed = self.speeds[self.typ]
+        
         while self.running:
             if self.total_volume % 12 == 0:
                 self.ch340_pull()  # 抽取12ml
                 time.sleep(0.01)
 
             if not self.ch340.running:
-                print(f"Current Total Volume: {self.total_volume:0.2f} ml")
-                if self.typ == 0:  # 每次加0.2ml
-                    speed = quick_speed
-                    self.ch340_push(speed)
-                    self.total_volume += speed
-                else:
-                    speed = slow_speed
-                    self.ch340_push(speed)  # 每次加0.05ml
-                    self.total_volume += speed
+                self.volume_logger.info(f"当前体积: {self.total_volume:.2f} ml, 加入速度: {speed:.2f} ml/次")
+                
+                speed = self.speeds[self.typ]
+                self.ch340_push(speed)
+                self.total_volume += speed
 
             if self._pred() is None:
-                print("Prediction failed, skipping frame.")
+                self.control_logger.error("预测失败，跳过当前帧")
                 continue
             
-            
-        print('----->>Visual Endpoint<<-----')
-        print(f"Total Volume: {self.total_volume} ml")
-        print(f"End point volume: {self.colored_volume} ml")
-        print(f"Theorical fixed volume: {self.colored_volume} ml")
+        # 实验结束，记录结果
+        experiment_results = {
+            "总体积  ": f"{self.total_volume:.2f} ml",
+            "终点体积": f"{self.colored_volume:.2f} ml",
+            "理论体积": f"{self.colored_volume:.2f} ml",
+            "实验时长": f"{time.time() - self.start_time:.2f} 秒"
+        }
+        self.system_logger.info("实验结果:")
+        for key, value in experiment_results.items():
+            self.system_logger.info(f"  {key}: {value}")
 
 
 if __name__ == "__main__":
@@ -199,7 +217,7 @@ if __name__ == "__main__":
     
     mat.run(
         slow_speed = 0.05,
-        quick_speed = 0.15,     
+        quick_speed = 0.15,  
     )