open_pub/Uni-Lab-OS
1
0
Fork 0
mirror of https://github.com/dptech-corp/Uni-Lab-OS.git synced 2025-12-17 21:11:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

uncompleted refactor

Browse Source
This commit is contained in:
Junhan Chang
2025-08-27 15:19:58 +08:00
parent 332b33c6f4
commit 82d9ef6bf7
3 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
unilabos/registry/devices/work_station.yaml
View File

@@ -6112,7 +6112,7 @@ workstation:
title: initialize_device参数
type: object
type: UniLabJsonCommand
module: unilabos.ros.nodes.presets.protocol_node:ROS2WorkstationNode
module: unilabos.ros.nodes.presets.workstation_node:ROS2WorkstationNode
status_types: {}
type: ros2
config_info: []

14
unilabos/ros/nodes/base_device_node.py
View File

@@ -958,19 +958,19 @@ class ROS2DeviceNode:
driver_class, children=children, resource_tracker=self.resource_tracker
)
else:
from unilabos.ros.nodes.presets.protocol_node import ROS2WorkstationNode
from unilabos.devices.work_station.workstation_base import WorkstationBase
from unilabos.ros.nodes.presets.workstation_node import ROS2WorkstationNode
from unilabos.devices.workstation.workstation_base import WorkstationBase
# 检查是否是WorkstationBase的子类且包含设备子节点
if issubclass(self._driver_class, WorkstationBase) and has_device_children:
# WorkstationBase + 设备子节点 -> 使用WorkstationNode作为ros_instance
self._use_protocol_node_ros = True
self._use_workstation_node_ros = True
self._driver_creator = DeviceClassCreator(driver_class, children=children, resource_tracker=self.resource_tracker)
elif issubclass(self._driver_class, ROS2WorkstationNode): # 是WorkstationNode的子节点，就要调用WorkstationNodeCreator
self._use_protocol_node_ros = False
self._use_workstation_node_ros = False
self._driver_creator = WorkstationNodeCreator(driver_class, children=children, resource_tracker=self.resource_tracker)
else:
self._use_protocol_node_ros = False
self._use_workstation_node_ros = False
self._driver_creator = DeviceClassCreator(driver_class, children=children, resource_tracker=self.resource_tracker)
if driver_is_ros:
@@ -984,9 +984,9 @@ class ROS2DeviceNode:
# 创建ROS2节点
if driver_is_ros:
self._ros_node = self._driver_instance # type: ignore
elif hasattr(self, '_use_protocol_node_ros') and self._use_protocol_node_ros:
elif hasattr(self, '_use_workstation_node_ros') and self._use_workstation_node_ros:
# WorkstationBase + 设备子节点 -> 创建ROS2WorkstationNode作为ros_instance
from unilabos.ros.nodes.presets.protocol_node import ROS2WorkstationNode
from unilabos.ros.nodes.presets.workstation_node import ROS2WorkstationNode
# 从children提取设备协议类型
protocol_types = set()

2
unilabos/ros/nodes/presets/protocol_node.py → unilabos/ros/nodes/presets/workstation_node.py
View File

@@ -100,7 +100,7 @@ class ROS2WorkstationNode(BaseROS2DeviceNode):
def _setup_workstation_integration(self):
"""设置工作站集成 - 统一设备处理模式"""
# 1. 建立协议节点引用
self.workstation_instance.set_protocol_node(self)
self.workstation_instance.set_workstation_node(self)
self.lab_logger().info(f"ROS2WorkstationNode {self.device_id} 与工作站实例 {type(self.workstation_instance).__name__} 集成完成")