146 lines
4.5 KiB
Python
146 lines
4.5 KiB
Python
import logging
|
||
import os
|
||
from datetime import datetime
|
||
from enum import Enum
|
||
|
||
class State:
|
||
"""滴定状态管理类"""
|
||
class Mode(Enum):
|
||
FAST = 0 # 快速模式
|
||
SLOW = 1 # 慢速模式 (middle)
|
||
ABOUT = 2 # 接近终点模式
|
||
# END = 3 # 终点模式
|
||
|
||
def __init__(self, bounce_time=1, end_bounce_time=5):
|
||
self.mode = self.Mode.FAST
|
||
self.bounce_time = bounce_time
|
||
self.end_bounce_time = end_bounce_time
|
||
|
||
# 状态检查标志
|
||
self.in_middle_check = False
|
||
self.in_end_check = False
|
||
|
||
# 时间记录
|
||
self.middle_detected_time = None
|
||
self.end_detected_time = None
|
||
|
||
def is_fast_mode(self):
|
||
return self.mode == self.Mode.FAST
|
||
|
||
def is_slow_mode(self):
|
||
return self.mode == self.Mode.SLOW
|
||
|
||
def is_about_mode(self):
|
||
return self.mode == self.Mode.ABOUT
|
||
|
||
# def is_end_mode(self):
|
||
# return self.mode == self.Mode.END
|
||
|
||
def enter_middle_state(self, current_time):
|
||
"""进入middle状态 - 立即切换到slow模式并开始检查"""
|
||
self.mode = self.Mode.SLOW
|
||
self.in_middle_check = True
|
||
self.middle_detected_time = current_time
|
||
|
||
def enter_about_state(self, current_time):
|
||
"""从middle状态进入about状态"""
|
||
if self.mode == self.Mode.SLOW:
|
||
self.mode = self.Mode.ABOUT
|
||
|
||
def enter_end_check(self, current_time):
|
||
"""进入end检查状态"""
|
||
self.in_end_check = True
|
||
self.end_detected_time = current_time
|
||
self.mode = self.Mode.ABOUT
|
||
|
||
def exit_middle_check(self):
|
||
"""退出middle检查状态，返回fast模式"""
|
||
self.in_middle_check = False
|
||
self.middle_detected_time = None
|
||
self.mode = self.Mode.FAST
|
||
|
||
def exit_about_with_middle(self):
|
||
"""about状态随middle一起退出"""
|
||
if self.mode == self.Mode.ABOUT:
|
||
self.mode = self.Mode.FAST
|
||
|
||
def should_check_middle_exit(self, current_time):
|
||
"""检查是否应该进行middle退出检查"""
|
||
return (self.in_middle_check and
|
||
self.middle_detected_time is not None and
|
||
current_time - self.middle_detected_time > self.bounce_time)
|
||
|
||
def should_check_end_result(self, current_time):
|
||
"""检查是否应该进行end结果检查"""
|
||
return (self.in_end_check and
|
||
self.end_detected_time is not None and
|
||
current_time - self.end_detected_time > self.end_bounce_time)
|
||
|
||
def reset_end_check(self):
|
||
"""重置end检查状态"""
|
||
self.in_end_check = False
|
||
self.end_detected_time = None
|
||
|
||
def get_status_text(self):
|
||
"""获取状态显示文本"""
|
||
status = []
|
||
if self.in_middle_check:
|
||
status.append("MIDCHK")
|
||
if self.in_end_check:
|
||
status.append("ENDCHK")
|
||
return ", " + ", ".join(status) if status else ""
|
||
|
||
|
||
def setup_logging(log_level=logging.INFO, log_dir="logs"):
|
||
"""
|
||
设置logging配置，创建不同模块的logger
|
||
|
||
Args:
|
||
log_level: 日志级别，默认INFO
|
||
log_dir: 日志文件存储目录，默认"logs"
|
||
"""
|
||
# 创建日志目录
|
||
if not os.path.exists(log_dir):
|
||
os.makedirs(log_dir)
|
||
|
||
# 获取当前时间作为日志文件名
|
||
timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
|
||
log_file = os.path.join(log_dir, f"titration_{timestamp}.log")
|
||
|
||
# 配置根logger
|
||
logging.basicConfig(
|
||
level=log_level,
|
||
format='%(asctime)s - %(name)8s - %(levelname)7s - %(message)s',
|
||
handlers=[
|
||
logging.FileHandler(log_file, encoding='utf-8'),
|
||
logging.StreamHandler() # 同时输出到控制台
|
||
]
|
||
)
|
||
|
||
return log_file
|
||
|
||
|
||
def get_system_logger():
|
||
"""系统初始化和控制相关的logger"""
|
||
return logging.getLogger("System")
|
||
|
||
def get_hardware_logger():
|
||
"""硬件控制相关的logger（CH340等）"""
|
||
return logging.getLogger("Hardware")
|
||
|
||
def get_vision_logger():
|
||
"""图像处理和预测相关的logger"""
|
||
return logging.getLogger("Vision")
|
||
|
||
def get_control_logger():
|
||
"""控制逻辑相关的logger"""
|
||
return logging.getLogger("Control")
|
||
|
||
def get_endpoint_logger():
|
||
"""终点检测相关的logger"""
|
||
return logging.getLogger("Endpoint")
|
||
|
||
def get_volume_logger():
|
||
"""体积计算相关的logger"""
|
||
return logging.getLogger("Volume")
|