0529-2
Former-commit-id: 09404bbedb1ff5b9ff9ad0753867012826cf84dc
This commit is contained in:
88
utils.py
88
utils.py
@ -1,6 +1,94 @@
|
||||
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"):
|
||||
|
Reference in New Issue
Block a user