Device registry port (#49)

* Update README and MQTTClient for installation instructions and code improvements * feat: 支持local_config启动 add: 增加对crt path的说明，为传入config.py的相对路径 move: web component * add: registry description * add 3d visualization * 完成在main中启动设备可视化完成在main中启动设备可视化，并输出物料ID：mesh的对应关系resource_model 添加物料模型管理类，遍历物料与resource_model，完成TF数据收集 * 完成TF发布 * 修改模型方向，在yaml中添加变换属性 * 添加物料tf变化时，发送topic到前端另外修改了物料初始化的方法，防止在tf还未发布时提前建立物料模型与发布话题 * 添加关节发布节点与物料可视化节点进入unilab * 使用json启动plr与3D模型仿真 * feat: node_info_update srv fix: OTDeck cant create * close #12 feat: slave node registry * feat: show machine name fix: host node registry not uploaded * feat: add hplc registry * feat: add hplc registry * fix: hplc status typo * fix: devices/ * 完成启动OT并联动rviz * add 3d visualization * 完成在main中启动设备可视化完成在main中启动设备可视化，并输出物料ID：mesh的对应关系resource_model 添加物料模型管理类，遍历物料与resource_model，完成TF数据收集 * 完成TF发布 * 修改模型方向，在yaml中添加变换属性 * 添加物料tf变化时，发送topic到前端另外修改了物料初始化的方法，防止在tf还未发布时提前建立物料模型与发布话题 * 添加关节发布节点与物料可视化节点进入unilab * 使用json启动plr与3D模型仿真 * 完成启动OT并联动rviz * fix: device.class possible null * fix: HPLC additions with online service * fix: slave mode spin not working * fix: slave mode spin not working * 修复rviz位置问题, 修复rviz位置问题, 在无tf变动时减缓发送频率在backend中添加物料跟随方法 * feat: 多ProtocolNode 允许子设备ID相同 feat: 上报发现的ActionClient feat: Host重启动，通过discover机制要求slaveNode重新注册，实现信息及时上报 * feat: 支持env设置config * fix: running logic * fix: running logic * fix: missing ot * 在main中直接初始化republisher和物料的mesh节点 * 将joint_republisher和resource_mesh_manager添加进 main_slave_run.py中 * Device visualization (#14) * add 3d visualization * 完成在main中启动设备可视化完成在main中启动设备可视化，并输出物料ID：mesh的对应关系resource_model 添加物料模型管理类，遍历物料与resource_model，完成TF数据收集 * 完成TF发布 * 修改模型方向，在yaml中添加变换属性 * 添加物料tf变化时，发送topic到前端另外修改了物料初始化的方法，防止在tf还未发布时提前建立物料模型与发布话题 * 添加关节发布节点与物料可视化节点进入unilab * 使用json启动plr与3D模型仿真 * 完成启动OT并联动rviz * add 3d visualization * 完成在main中启动设备可视化完成在main中启动设备可视化，并输出物料ID：mesh的对应关系resource_model 添加物料模型管理类，遍历物料与resource_model，完成TF数据收集 * 完成TF发布 * 修改模型方向，在yaml中添加变换属性 * 添加物料tf变化时，发送topic到前端另外修改了物料初始化的方法，防止在tf还未发布时提前建立物料模型与发布话题 * 添加关节发布节点与物料可视化节点进入unilab * 使用json启动plr与3D模型仿真 * 完成启动OT并联动rviz * 修复rviz位置问题, 修复rviz位置问题, 在无tf变动时减缓发送频率在backend中添加物料跟随方法 * fix: running logic * fix: running logic * fix: missing ot * 在main中直接初始化republisher和物料的mesh节点 * 将joint_republisher和resource_mesh_manager添加进 main_slave_run.py中 --------- Co-authored-by: zhangshixiang <@zhangshixiang> Co-authored-by: wznln <18435084+Xuwznln@users.noreply.github.com> * fix: missing hostname in devices_names fix: upload_file for model file * fix: missing paho-mqtt package bump version to 0.9.0 * fix startup add ResourceCreateFromOuter.action * fix type hint * update actions * update actions * host node add_resource_from_outer fix cmake list * pass device config to device class * add: bind_parent_ids to resource create action fix: message convert string * fix: host node should not be re_discovered * feat: resource tracker support dict * feat: add more necessary params * feat: fix boolean null in registry action data * feat: add outer resource * 编写mesh添加action * feat: append resource * add action * feat: vis 2d for plr * fix * fix: browser on rviz * fix: cloud bridge error fallback to local * fix: salve auto run rviz * 初始化两个plate * Device visualization (#22) * add 3d visualization * 完成在main中启动设备可视化完成在main中启动设备可视化，并输出物料ID：mesh的对应关系resource_model 添加物料模型管理类，遍历物料与resource_model，完成TF数据收集 * 完成TF发布 * 修改模型方向，在yaml中添加变换属性 * 添加物料tf变化时，发送topic到前端另外修改了物料初始化的方法，防止在tf还未发布时提前建立物料模型与发布话题 * 添加关节发布节点与物料可视化节点进入unilab * 使用json启动plr与3D模型仿真 * 完成启动OT并联动rviz * add 3d visualization * 完成在main中启动设备可视化完成在main中启动设备可视化，并输出物料ID：mesh的对应关系resource_model 添加物料模型管理类，遍历物料与resource_model，完成TF数据收集 * 完成TF发布 * 修改模型方向，在yaml中添加变换属性 * 添加物料tf变化时，发送topic到前端另外修改了物料初始化的方法，防止在tf还未发布时提前建立物料模型与发布话题 * 添加关节发布节点与物料可视化节点进入unilab * 使用json启动plr与3D模型仿真 * 完成启动OT并联动rviz * 修复rviz位置问题, 修复rviz位置问题, 在无tf变动时减缓发送频率在backend中添加物料跟随方法 * fix: running logic * fix: running logic * fix: missing ot * 在main中直接初始化republisher和物料的mesh节点 * 将joint_republisher和resource_mesh_manager添加进 main_slave_run.py中 * 编写mesh添加action * add action * fix * fix: browser on rviz * fix: cloud bridge error fallback to local * fix: salve auto run rviz * 初始化两个plate --------- Co-authored-by: zhangshixiang <@zhangshixiang> Co-authored-by: wznln <18435084+Xuwznln@users.noreply.github.com> * fix: multi channel * fix: aspirate * fix: aspirate * fix: aspirate * fix: aspirate * 提交 * fix: jobadd * fix: jobadd * fix: msg converter * tijiao * add resource creat easy action * identify debug msg * mq client id * unify liquid_handler definition * Update virtual_device.yaml * 更正了stir和heater的连接方式 * 区分了虚拟仪器中的八通阀和电磁阀，添加了两个阀门的驱动 * 修改了add protocol * 修复了阀门更新版的bug * 修复了添加protocol前缀导致的不能启动的bug * Fix handles * bump version to 0.9.6 * add resource edge upload * update container registry and handles * add virtual_separator virtual_rotavap fix transfer_pump * fix container value add parent_name to edge device id * 大图的问题都修复好了，添加了gassource和vacuum pump的驱动以及注册表 * default resource upload mode is false * 添加了icon的文件名在注册表里面 * 修改了json图中link的格式 * fix resource and edge upload * fix device ports * Fix edge id * 移除device的父节点关联 * separate registry sync and resource_add * 默认不进行注册表报送，通过命令unilabos-register或者增加启动参数 * 完善tip * protocol node不再嵌套显示 * bump version to 0.9.7 新增一个测试PumpTransferProtocol的teststation，亲测可以运行，将八通阀们和转移泵与pump_protocol适配 * protocol node 执行action不应携带自身device id * 添加了一套简易双八通阀工作站JSON，亲测能跑 * 修复了很多protocol，亲测能跑 * 添加了run column和filter through的protocol，亲测能跑 * fix mock_reactor * 修改了大图和小图的json，但是在前端上没看到改变 --------- Co-authored-by: Harvey Que <Q-Query@outlook.com> Co-authored-by: wznln <18435084+Xuwznln@users.noreply.github.com> Co-authored-by: zhangshixiang <@zhangshixiang> Co-authored-by: q434343 <73513873+q434343@users.noreply.github.com> Co-authored-by: Junhan Chang <changjh@pku.edu.cn>
2026-02-05 22:15:04 +00:00 · 2025-06-22 12:59:59 +08:00
parent f7db8d17c5
commit 46cec82a51
85 changed files with 12934 additions and 1573 deletions
--- a/unilabos/compile/heatchill_protocol.py
+++ b/unilabos/compile/heatchill_protocol.py
@@ -1,33 +1,61 @@
-from typing import List, Dict, Any
+from typing import List, Dict, Any, Optional
 import networkx as nx

+
+def find_connected_heatchill(G: nx.DiGraph, vessel: str) -> str:
+    """
+    查找与指定容器相连的加热/冷却设备
+    """
+    # 查找所有加热/冷却设备节点
+    heatchill_nodes = [node for node in G.nodes() 
+                      if (G.nodes[node].get('class') or '') == 'virtual_heatchill']
+    
+    # 检查哪个加热/冷却设备与目标容器相连（机械连接）
+    for heatchill in heatchill_nodes:
+        if G.has_edge(heatchill, vessel) or G.has_edge(vessel, heatchill):
+            return heatchill
+    
+    # 如果没有直接连接，返回第一个可用的加热/冷却设备
+    if heatchill_nodes:
+        return heatchill_nodes[0]
+    
+    raise ValueError("系统中未找到可用的加热/冷却设备")
+
+
 def generate_heat_chill_protocol(
    G: nx.DiGraph,
    vessel: str,
    temp: float,
    time: float,
-    stir: bool,
-    stir_speed: float,
-    purpose: str
+    stir: bool = False,
+    stir_speed: float = 300.0,
+    purpose: str = "加热/冷却操作"
 ) -> List[Dict[str, Any]]:
    """
-    生成加热/冷却操作的协议序列 - 严格按照 HeatChill.action
+    生成加热/冷却操作的协议序列 - 带时间限制的完整操作
    """
    action_sequence = []
    
-    # 查找加热/冷却设备
-    heatchill_nodes = [node for node in G.nodes() 
-                      if G.nodes[node].get('class') == 'virtual_heatchill']
-    
-    if not heatchill_nodes:
-        raise ValueError("没有找到可用的加热/冷却设备")
-    
-    heatchill_id = heatchill_nodes[0]
+    print(f"HEATCHILL: 开始生成加热/冷却协议")
+    print(f"  - 容器: {vessel}")
+    print(f"  - 目标温度: {temp}°C")
+    print(f"  - 持续时间: {time}秒")
+    print(f"  - 使用内置搅拌: {stir}, 速度: {stir_speed} RPM")
+    print(f"  - 目的: {purpose}")
    
+    # 1. 验证容器存在
    if vessel not in G.nodes():
-        raise ValueError(f"容器 {vessel} 不存在于图中")
+        raise ValueError(f"容器 '{vessel}' 不存在于系统中")
    
-    action_sequence.append({
+    # 2. 查找加热/冷却设备
+    try:
+        heatchill_id = find_connected_heatchill(G, vessel)
+        print(f"HEATCHILL: 找到加热/冷却设备: {heatchill_id}")
+    except ValueError as e:
+        raise ValueError(f"无法找到加热/冷却设备: {str(e)}")
+    
+    # 3. 执行加热/冷却操作
+    heatchill_action = {
        "device_id": heatchill_id,
        "action_name": "heat_chill",
        "action_kwargs": {
@@ -36,10 +64,13 @@ def generate_heat_chill_protocol(
            "time": time,
            "stir": stir,
            "stir_speed": stir_speed,
-            "purpose": purpose
+            "status": "start"
        }
-    })
+    }
    
+    action_sequence.append(heatchill_action)
+    
+    print(f"HEATCHILL: 生成了 {len(action_sequence)} 个动作")
    return action_sequence


@@ -47,25 +78,31 @@ def generate_heat_chill_start_protocol(
    G: nx.DiGraph,
    vessel: str,
    temp: float,
-    purpose: str
+    purpose: str = "开始加热/冷却"
 ) -> List[Dict[str, Any]]:
    """
-    生成开始加热/冷却操作的协议序列 - 严格按照 HeatChillStart.action
+    生成开始加热/冷却操作的协议序列
    """
    action_sequence = []
    
-    heatchill_nodes = [node for node in G.nodes() 
-                      if G.nodes[node].get('class') == 'virtual_heatchill']
-    
-    if not heatchill_nodes:
-        raise ValueError("没有找到可用的加热/冷却设备")
-    
-    heatchill_id = heatchill_nodes[0]
+    print(f"HEATCHILL_START: 开始生成加热/冷却启动协议")
+    print(f"  - 容器: {vessel}")
+    print(f"  - 目标温度: {temp}°C")
+    print(f"  - 目的: {purpose}")
    
+    # 1. 验证容器存在
    if vessel not in G.nodes():
-        raise ValueError(f"容器 {vessel} 不存在于图中")
+        raise ValueError(f"容器 '{vessel}' 不存在于系统中")
    
-    action_sequence.append({
+    # 2. 查找加热/冷却设备
+    try:
+        heatchill_id = find_connected_heatchill(G, vessel)
+        print(f"HEATCHILL_START: 找到加热/冷却设备: {heatchill_id}")
+    except ValueError as e:
+        raise ValueError(f"无法找到加热/冷却设备: {str(e)}")
+    
+    # 3. 执行开始加热/冷却操作
+    heatchill_start_action = {
        "device_id": heatchill_id,
        "action_name": "heat_chill_start",
        "action_kwargs": {
@@ -73,8 +110,11 @@ def generate_heat_chill_start_protocol(
            "temp": temp,
            "purpose": purpose
        }
-    })
+    }
    
+    action_sequence.append(heatchill_start_action)
+    
+    print(f"HEATCHILL_START: 生成了 {len(action_sequence)} 个动作")
    return action_sequence


@@ -84,34 +124,250 @@ def generate_heat_chill_stop_protocol(
 ) -> List[Dict[str, Any]]:
    """
    生成停止加热/冷却操作的协议序列
-    
-    Args:
-        G: 有向图，节点为设备和容器
-        vessel: 容器名称
-    
-    Returns:
-        List[Dict[str, Any]]: 停止加热/冷却操作的动作序列
    """
    action_sequence = []
    
-    # 查找加热/冷却设备
-    heatchill_nodes = [node for node in G.nodes() 
-                      if G.nodes[node].get('class') == 'virtual_heatchill']
-    
-    if not heatchill_nodes:
-        raise ValueError("没有找到可用的加热/冷却设备")
-    
-    heatchill_id = heatchill_nodes[0]
+    print(f"HEATCHILL_STOP: 开始生成加热/冷却停止协议")
+    print(f"  - 容器: {vessel}")
    
+    # 1. 验证容器存在
    if vessel not in G.nodes():
-        raise ValueError(f"容器 {vessel} 不存在于图中")
+        raise ValueError(f"容器 '{vessel}' 不存在于系统中")
    
-    action_sequence.append({
+    # 2. 查找加热/冷却设备
+    try:
+        heatchill_id = find_connected_heatchill(G, vessel)
+        print(f"HEATCHILL_STOP: 找到加热/冷却设备: {heatchill_id}")
+    except ValueError as e:
+        raise ValueError(f"无法找到加热/冷却设备: {str(e)}")
+    
+    # 3. 执行停止加热/冷却操作
+    heatchill_stop_action = {
        "device_id": heatchill_id,
        "action_name": "heat_chill_stop",
        "action_kwargs": {
            "vessel": vessel
        }
-    })
+    }
    
-    return action_sequence
+    action_sequence.append(heatchill_stop_action)
+    
+    print(f"HEATCHILL_STOP: 生成了 {len(action_sequence)} 个动作")
+    return action_sequence
+
+
+def generate_heat_chill_to_temp_protocol(
+    G: nx.DiGraph,
+    vessel: str,
+    temp: float,
+    active: bool = True,
+    continue_heatchill: bool = False,
+    stir: bool = False,
+    stir_speed: Optional[float] = None,
+    purpose: Optional[str] = None
+) -> List[Dict[str, Any]]:
+    """
+    生成加热/冷却到指定温度的协议序列 - 智能温控协议
+    
+    **关键修复**: 学习 pump_protocol 的模式，直接使用设备基础动作，不依赖特定的 Action 文件
+    """
+    action_sequence = []
+    
+    # 设置默认值
+    if stir_speed is None:
+        stir_speed = 300.0
+    if purpose is None:
+        purpose = f"智能温控到 {temp}°C"
+    
+    print(f"HEATCHILL_TO_TEMP: 开始生成智能温控协议")
+    print(f"  - 容器: {vessel}")
+    print(f"  - 目标温度: {temp}°C")
+    print(f"  - 主动控温: {active}")
+    print(f"  - 达到温度后继续: {continue_heatchill}")
+    print(f"  - 搅拌: {stir}, 速度: {stir_speed} RPM")
+    print(f"  - 目的: {purpose}")
+    
+    # 1. 验证容器存在
+    if vessel not in G.nodes():
+        raise ValueError(f"容器 '{vessel}' 不存在于系统中")
+    
+    # 2. 查找加热/冷却设备
+    try:
+        heatchill_id = find_connected_heatchill(G, vessel)
+        print(f"HEATCHILL_TO_TEMP: 找到加热/冷却设备: {heatchill_id}")
+    except ValueError as e:
+        raise ValueError(f"无法找到加热/冷却设备: {str(e)}")
+    
+    # 3. 根据参数选择合适的基础动作组合 (学习 pump_protocol 的模式)
+    if not active:
+        print(f"HEATCHILL_TO_TEMP: 非主动模式，仅等待")
+        action_sequence.append({
+            "action_name": "wait",
+            "action_kwargs": {
+                "time": 10.0,
+                "purpose": f"等待容器 {vessel} 自然达到 {temp}°C"
+            }
+        })
+    else:
+        if continue_heatchill:
+            # 持续模式：使用 heat_chill_start 基础动作
+            print(f"HEATCHILL_TO_TEMP: 使用持续温控模式")
+            action_sequence.append({
+                "device_id": heatchill_id,
+                "action_name": "heat_chill_start",  # ← 直接使用设备基础动作
+                "action_kwargs": {
+                    "vessel": vessel,
+                    "temp": temp,
+                    "purpose": f"{purpose} (持续保温)"
+                }
+            })
+        else:
+            # 一次性模式：使用 heat_chill 基础动作
+            print(f"HEATCHILL_TO_TEMP: 使用一次性温控模式")
+            estimated_time = max(60.0, min(900.0, abs(temp - 25.0) * 30.0))
+            print(f"HEATCHILL_TO_TEMP: 估算所需时间: {estimated_time}秒")
+            
+            action_sequence.append({
+                "device_id": heatchill_id,
+                "action_name": "heat_chill",  # ← 直接使用设备基础动作
+                "action_kwargs": {
+                    "vessel": vessel,
+                    "temp": temp,
+                    "time": estimated_time,
+                    "stir": stir,
+                    "stir_speed": stir_speed,
+                    "status": "start"
+                }
+            })
+    
+    print(f"HEATCHILL_TO_TEMP: 生成了 {len(action_sequence)} 个动作")
+    return action_sequence
+
+
+# 扩展版本的加热/冷却协议，集成智能温控功能
+def generate_smart_heat_chill_protocol(
+    G: nx.DiGraph,
+    vessel: str,
+    temp: float,
+    time: float = 0.0,  # 0表示自动估算
+    active: bool = True,
+    continue_heatchill: bool = False,
+    stir: bool = False,
+    stir_speed: float = 300.0,
+    purpose: str = "智能加热/冷却"
+) -> List[Dict[str, Any]]:
+    """
+    这个函数集成了 generate_heat_chill_to_temp_protocol 的智能逻辑，
+    但使用现有的 Action 类型
+    """
+    # 如果时间为0，自动估算
+    if time == 0.0:
+        estimated_time = max(60.0, min(900.0, abs(temp - 25.0) * 30.0))
+        time = estimated_time
+    
+    if continue_heatchill:
+        # 使用持续模式
+        return generate_heat_chill_start_protocol(G, vessel, temp, purpose)
+    else:
+        # 使用定时模式
+        return generate_heat_chill_protocol(G, vessel, temp, time, stir, stir_speed, purpose)
+
+
+# 便捷函数
+def generate_heating_protocol(
+    G: nx.DiGraph,
+    vessel: str,
+    temp: float,
+    time: float = 300.0,
+    stir: bool = True,
+    stir_speed: float = 300.0
+) -> List[Dict[str, Any]]:
+    """生成加热协议的便捷函数"""
+    return generate_heat_chill_protocol(
+        G=G, vessel=vessel, temp=temp, time=time, 
+        stir=stir, stir_speed=stir_speed, purpose=f"加热到 {temp}°C"
+    )
+
+
+def generate_cooling_protocol(
+    G: nx.DiGraph,
+    vessel: str,
+    temp: float,
+    time: float = 600.0,
+    stir: bool = True,
+    stir_speed: float = 200.0
+) -> List[Dict[str, Any]]:
+    """生成冷却协议的便捷函数"""
+    return generate_heat_chill_protocol(
+        G=G, vessel=vessel, temp=temp, time=time,
+        stir=stir, stir_speed=stir_speed, purpose=f"冷却到 {temp}°C"
+    )
+
+
+# # 温度预设快捷函数
+# def generate_room_temp_protocol(
+#     G: nx.DiGraph,
+#     vessel: str,
+#     stir: bool = False
+# ) -> List[Dict[str, Any]]:
+#     """返回室温的快捷函数"""
+#     return generate_heat_chill_to_temp_protocol(
+#         G=G,
+#         vessel=vessel,
+#         temp=25.0,
+#         active=True,
+#         continue_heatchill=False,
+#         stir=stir,
+#         purpose="冷却到室温"
+#     )
+
+
+# def generate_reflux_heating_protocol(
+#     G: nx.DiGraph,
+#     vessel: str,
+#     temp: float,
+#     time: float = 3600.0  # 1小时回流
+# ) -> List[Dict[str, Any]]:
+#     """回流加热的快捷函数"""
+#     return generate_heat_chill_protocol(
+#         G=G,
+#         vessel=vessel,
+#         temp=temp,
+#         time=time,
+#         stir=True,
+#         stir_speed=400.0,  # 回流时较快搅拌
+#         purpose=f"回流加热到 {temp}°C"
+#     )
+
+
+# def generate_ice_bath_protocol(
+#     G: nx.DiGraph,
+#     vessel: str,
+#     time: float = 600.0  # 10分钟冰浴
+# ) -> List[Dict[str, Any]]:
+#     """冰浴冷却的快捷函数"""
+#     return generate_heat_chill_protocol(
+#         G=G,
+#         vessel=vessel,
+#         temp=0.0,
+#         time=time,
+#         stir=True,
+#         stir_speed=150.0,  # 冰浴时缓慢搅拌
+#         purpose="冰浴冷却到 0°C"
+#     )
+
+
+# 测试函数
+def test_heatchill_protocol():
+    """测试加热/冷却协议的示例"""
+    print("=== HEAT CHILL PROTOCOL 测试 ===")
+    print("完整的四个协议函数：")
+    print("1. generate_heat_chill_protocol - 带时间限制的完整操作")
+    print("2. generate_heat_chill_start_protocol - 持续加热/冷却")
+    print("3. generate_heat_chill_stop_protocol - 停止加热/冷却")
+    print("4. generate_heat_chill_to_temp_protocol - 智能温控 (您的 HeatChillToTemp)")
+    print("测试完成")
+
+
+if __name__ == "__main__":
+    test_heatchill_protocol()