mirror of
https://github.com/dptech-corp/Uni-Lab-OS.git
synced 2026-02-06 06:25:06 +00:00
Compare commits
4 Commits
feat/sampl
...
e30c01d54e
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
e30c01d54e | ||
|
|
c4a3be1498 | ||
|
|
e11070315d | ||
|
|
50ebcad9d7 |
@@ -46,7 +46,7 @@ requirements:
|
|||||||
- jinja2
|
- jinja2
|
||||||
- requests
|
- requests
|
||||||
- uvicorn
|
- uvicorn
|
||||||
- opcua # [not osx]
|
- opcua
|
||||||
- pyserial
|
- pyserial
|
||||||
- pandas
|
- pandas
|
||||||
- pymodbus
|
- pymodbus
|
||||||
|
|||||||
@@ -452,9 +452,8 @@ unilab --ak your_ak --sk your_sk -g test/experiments/mock_devices/mock_all.json
|
|||||||
**操作步骤:**
|
**操作步骤:**
|
||||||
|
|
||||||
1. 将两个 `container` 拖拽到 `workstation` 中
|
1. 将两个 `container` 拖拽到 `workstation` 中
|
||||||
2. 将 `virtual_multiway_valve` 拖拽到 `workstation` 中
|
2. 将 `virtual_transfer_pump` 拖拽到 `workstation` 中
|
||||||
3. 将 `virtual_transfer_pump` 拖拽到 `workstation` 中
|
3. 在画布上连接它们(建立父子关系)
|
||||||
4. 在画布上连接它们(建立父子关系)
|
|
||||||
|
|
||||||

|

|
||||||
|
|
||||||
|
|||||||
Binary file not shown.
|
Before Width: | Height: | Size: 415 KiB After Width: | Height: | Size: 275 KiB |
@@ -54,7 +54,6 @@ class JobAddReq(BaseModel):
|
|||||||
action_type: str = Field(
|
action_type: str = Field(
|
||||||
examples=["unilabos_msgs.action._str_single_input.StrSingleInput"], description="action type", default=""
|
examples=["unilabos_msgs.action._str_single_input.StrSingleInput"], description="action type", default=""
|
||||||
)
|
)
|
||||||
sample_material: dict = Field(examples=[{"string": "string"}], description="sample uuid to material uuid")
|
|
||||||
action_args: dict = Field(examples=[{"string": "string"}], description="action arguments", default_factory=dict)
|
action_args: dict = Field(examples=[{"string": "string"}], description="action arguments", default_factory=dict)
|
||||||
task_id: str = Field(examples=["task_id"], description="task uuid (auto-generated if empty)", default="")
|
task_id: str = Field(examples=["task_id"], description="task uuid (auto-generated if empty)", default="")
|
||||||
job_id: str = Field(examples=["job_id"], description="goal uuid (auto-generated if empty)", default="")
|
job_id: str = Field(examples=["job_id"], description="goal uuid (auto-generated if empty)", default="")
|
||||||
|
|||||||
@@ -327,7 +327,6 @@ def job_add(req: JobAddReq) -> JobData:
|
|||||||
queue_item,
|
queue_item,
|
||||||
action_type=action_type,
|
action_type=action_type,
|
||||||
action_kwargs=action_args,
|
action_kwargs=action_args,
|
||||||
sample_material=req.sample_material,
|
|
||||||
server_info=server_info,
|
server_info=server_info,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
@@ -545,7 +545,7 @@ class MessageProcessor:
|
|||||||
try:
|
try:
|
||||||
message_str = json.dumps(msg, ensure_ascii=False)
|
message_str = json.dumps(msg, ensure_ascii=False)
|
||||||
await self.websocket.send(message_str)
|
await self.websocket.send(message_str)
|
||||||
# logger.trace(f"[MessageProcessor] Message sent: {msg.get('action', 'unknown')}") # type: ignore # noqa: E501
|
logger.trace(f"[MessageProcessor] Message sent: {msg.get('action', 'unknown')}") # type: ignore # noqa: E501
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(f"[MessageProcessor] Failed to send message: {str(e)}")
|
logger.error(f"[MessageProcessor] Failed to send message: {str(e)}")
|
||||||
logger.error(traceback.format_exc())
|
logger.error(traceback.format_exc())
|
||||||
@@ -657,8 +657,6 @@ class MessageProcessor:
|
|||||||
async def _handle_job_start(self, data: Dict[str, Any]):
|
async def _handle_job_start(self, data: Dict[str, Any]):
|
||||||
"""处理job_start消息"""
|
"""处理job_start消息"""
|
||||||
try:
|
try:
|
||||||
if not data.get("sample_material"):
|
|
||||||
data["sample_material"] = {}
|
|
||||||
req = JobAddReq(**data)
|
req = JobAddReq(**data)
|
||||||
|
|
||||||
job_log = format_job_log(req.job_id, req.task_id, req.device_id, req.action)
|
job_log = format_job_log(req.job_id, req.task_id, req.device_id, req.action)
|
||||||
@@ -690,7 +688,6 @@ class MessageProcessor:
|
|||||||
queue_item,
|
queue_item,
|
||||||
action_type=req.action_type,
|
action_type=req.action_type,
|
||||||
action_kwargs=req.action_args,
|
action_kwargs=req.action_args,
|
||||||
sample_material=req.sample_material,
|
|
||||||
server_info=req.server_info,
|
server_info=req.server_info,
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -1304,7 +1301,7 @@ class WebSocketClient(BaseCommunicationClient):
|
|||||||
},
|
},
|
||||||
}
|
}
|
||||||
self.message_processor.send_message(message)
|
self.message_processor.send_message(message)
|
||||||
# logger.trace(f"[WebSocketClient] Device status published: {device_id}.{property_name}")
|
logger.trace(f"[WebSocketClient] Device status published: {device_id}.{property_name}")
|
||||||
|
|
||||||
def publish_job_status(
|
def publish_job_status(
|
||||||
self, feedback_data: dict, item: QueueItem, status: str, return_info: Optional[dict] = None
|
self, feedback_data: dict, item: QueueItem, status: str, return_info: Optional[dict] = None
|
||||||
|
|||||||
@@ -95,29 +95,8 @@ def get_vessel_liquid_volume(G: nx.DiGraph, vessel: str) -> float:
|
|||||||
return total_volume
|
return total_volume
|
||||||
|
|
||||||
|
|
||||||
def is_integrated_pump(node_class: str, node_name: str = "") -> bool:
|
def is_integrated_pump(node_name):
|
||||||
"""
|
return "pump" in node_name and "valve" in node_name
|
||||||
判断是否为泵阀一体设备
|
|
||||||
"""
|
|
||||||
class_lower = (node_class or "").lower()
|
|
||||||
name_lower = (node_name or "").lower()
|
|
||||||
|
|
||||||
if "pump" not in class_lower and "pump" not in name_lower:
|
|
||||||
return False
|
|
||||||
|
|
||||||
integrated_markers = [
|
|
||||||
"valve",
|
|
||||||
"pump_valve",
|
|
||||||
"pumpvalve",
|
|
||||||
"integrated",
|
|
||||||
"transfer_pump",
|
|
||||||
]
|
|
||||||
|
|
||||||
for marker in integrated_markers:
|
|
||||||
if marker in class_lower or marker in name_lower:
|
|
||||||
return True
|
|
||||||
|
|
||||||
return False
|
|
||||||
|
|
||||||
|
|
||||||
def find_connected_pump(G, valve_node):
|
def find_connected_pump(G, valve_node):
|
||||||
@@ -207,9 +186,7 @@ def build_pump_valve_maps(G, pump_backbone):
|
|||||||
debug_print(f"🔧 过滤后的骨架: {filtered_backbone}")
|
debug_print(f"🔧 过滤后的骨架: {filtered_backbone}")
|
||||||
|
|
||||||
for node in filtered_backbone:
|
for node in filtered_backbone:
|
||||||
node_data = G.nodes.get(node, {})
|
if is_integrated_pump(G.nodes[node]["class"]):
|
||||||
node_class = node_data.get("class", "") or ""
|
|
||||||
if is_integrated_pump(node_class, node):
|
|
||||||
pumps_from_node[node] = node
|
pumps_from_node[node] = node
|
||||||
valve_from_node[node] = node
|
valve_from_node[node] = node
|
||||||
debug_print(f" - 集成泵-阀: {node}")
|
debug_print(f" - 集成泵-阀: {node}")
|
||||||
|
|||||||
@@ -1,11 +1,15 @@
|
|||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import asyncio
|
||||||
import time
|
import time
|
||||||
import traceback
|
import traceback
|
||||||
from collections import Counter
|
from collections import Counter
|
||||||
from typing import List, Sequence, Optional, Literal, Union, Iterator, Dict, Any, Callable, Set, cast
|
from typing import List, Sequence, Optional, Literal, Union, Iterator, Dict, Any, Callable, Set, cast
|
||||||
|
|
||||||
|
from typing_extensions import TypedDict
|
||||||
from pylabrobot.liquid_handling import LiquidHandler, LiquidHandlerBackend, LiquidHandlerChatterboxBackend, Strictness
|
from pylabrobot.liquid_handling import LiquidHandler, LiquidHandlerBackend, LiquidHandlerChatterboxBackend, Strictness
|
||||||
|
from unilabos.devices.liquid_handling.rviz_backend import UniLiquidHandlerRvizBackend
|
||||||
|
from unilabos.devices.liquid_handling.laiyu.backend.laiyu_v_backend import UniLiquidHandlerLaiyuBackend
|
||||||
from pylabrobot.liquid_handling.liquid_handler import TipPresenceProbingMethod
|
from pylabrobot.liquid_handling.liquid_handler import TipPresenceProbingMethod
|
||||||
from pylabrobot.liquid_handling.standard import GripDirection
|
from pylabrobot.liquid_handling.standard import GripDirection
|
||||||
from pylabrobot.resources import (
|
from pylabrobot.resources import (
|
||||||
@@ -23,38 +27,26 @@ from pylabrobot.resources import (
|
|||||||
Trash,
|
Trash,
|
||||||
Tip,
|
Tip,
|
||||||
)
|
)
|
||||||
from typing_extensions import TypedDict
|
|
||||||
|
|
||||||
from unilabos.devices.liquid_handling.rviz_backend import UniLiquidHandlerRvizBackend
|
|
||||||
from unilabos.registry.placeholder_type import ResourceSlot
|
from unilabos.registry.placeholder_type import ResourceSlot
|
||||||
from unilabos.resources.resource_tracker import (
|
from unilabos.ros.nodes.base_device_node import BaseROS2DeviceNode
|
||||||
ResourceTreeSet,
|
from unilabos.resources.resource_tracker import ResourceTreeSet
|
||||||
ResourceDict,
|
|
||||||
EXTRA_SAMPLE_UUID,
|
|
||||||
EXTRA_UNILABOS_SAMPLE_UUID,
|
|
||||||
)
|
|
||||||
from unilabos.ros.nodes.base_device_node import BaseROS2DeviceNode, ROS2DeviceNode
|
|
||||||
|
|
||||||
|
|
||||||
class SimpleReturn(TypedDict):
|
class SimpleReturn(TypedDict):
|
||||||
samples: List[List[ResourceDict]]
|
samples: list
|
||||||
volumes: List[float]
|
volumes: list
|
||||||
|
|
||||||
|
|
||||||
class SetLiquidReturn(TypedDict):
|
class SetLiquidReturn(TypedDict):
|
||||||
wells: List[List[ResourceDict]]
|
wells: list
|
||||||
volumes: List[float]
|
volumes: list
|
||||||
|
|
||||||
|
|
||||||
class SetLiquidFromPlateReturn(TypedDict):
|
class SetLiquidFromPlateReturn(TypedDict):
|
||||||
plate: List[List[ResourceDict]]
|
plate: list
|
||||||
wells: List[List[ResourceDict]]
|
wells: list
|
||||||
volumes: List[float]
|
volumes: list
|
||||||
|
|
||||||
|
|
||||||
class TransferLiquidReturn(TypedDict):
|
|
||||||
sources: List[List[ResourceDict]]
|
|
||||||
targets: List[List[ResourceDict]]
|
|
||||||
|
|
||||||
|
|
||||||
class LiquidHandlerMiddleware(LiquidHandler):
|
class LiquidHandlerMiddleware(LiquidHandler):
|
||||||
@@ -236,11 +228,12 @@ class LiquidHandlerMiddleware(LiquidHandler):
|
|||||||
res_samples = []
|
res_samples = []
|
||||||
res_volumes = []
|
res_volumes = []
|
||||||
for resource, volume, channel in zip(resources, vols, use_channels):
|
for resource, volume, channel in zip(resources, vols, use_channels):
|
||||||
sample_uuid_value = resource.unilabos_extra.get(EXTRA_SAMPLE_UUID, None)
|
res_samples.append(
|
||||||
res_samples.append({"name": resource.name, EXTRA_SAMPLE_UUID: sample_uuid_value})
|
{"name": resource.name, "sample_uuid": resource.unilabos_extra.get("sample_uuid", None)}
|
||||||
|
)
|
||||||
res_volumes.append(volume)
|
res_volumes.append(volume)
|
||||||
self.pending_liquids_dict[channel] = {
|
self.pending_liquids_dict[channel] = {
|
||||||
EXTRA_SAMPLE_UUID: sample_uuid_value,
|
"sample_uuid": resource.unilabos_extra.get("sample_uuid", None),
|
||||||
"volume": volume,
|
"volume": volume,
|
||||||
}
|
}
|
||||||
return SimpleReturn(samples=res_samples, volumes=res_volumes)
|
return SimpleReturn(samples=res_samples, volumes=res_volumes)
|
||||||
@@ -282,10 +275,10 @@ class LiquidHandlerMiddleware(LiquidHandler):
|
|||||||
res_samples = []
|
res_samples = []
|
||||||
res_volumes = []
|
res_volumes = []
|
||||||
for resource, volume, channel in zip(resources, vols, use_channels):
|
for resource, volume, channel in zip(resources, vols, use_channels):
|
||||||
res_uuid = self.pending_liquids_dict[channel][EXTRA_SAMPLE_UUID]
|
res_uuid = self.pending_liquids_dict[channel]["sample_uuid"]
|
||||||
self.pending_liquids_dict[channel]["volume"] -= volume
|
self.pending_liquids_dict[channel]["volume"] -= volume
|
||||||
resource.unilabos_extra[EXTRA_SAMPLE_UUID] = res_uuid
|
resource.unilabos_extra["sample_uuid"] = res_uuid
|
||||||
res_samples.append({"name": resource.name, EXTRA_SAMPLE_UUID: res_uuid})
|
res_samples.append({"name": resource.name, "sample_uuid": res_uuid})
|
||||||
res_volumes.append(volume)
|
res_volumes.append(volume)
|
||||||
|
|
||||||
return SimpleReturn(samples=res_samples, volumes=res_volumes)
|
return SimpleReturn(samples=res_samples, volumes=res_volumes)
|
||||||
@@ -689,15 +682,14 @@ class LiquidHandlerAbstract(LiquidHandlerMiddleware):
|
|||||||
wells=ResourceTreeSet.from_plr_resources(wells, known_newly_created=False).dump(), volumes=res_volumes # type: ignore
|
wells=ResourceTreeSet.from_plr_resources(wells, known_newly_created=False).dump(), volumes=res_volumes # type: ignore
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
def set_liquid_from_plate(
|
def set_liquid_from_plate(
|
||||||
self, plate: ResourceSlot, well_names: list[str], liquid_names: list[str], volumes: list[float]
|
cls, plate: ResourceSlot, well_names: list[str], liquid_names: list[str], volumes: list[float]
|
||||||
) -> SetLiquidFromPlateReturn:
|
) -> SetLiquidFromPlateReturn:
|
||||||
"""Set the liquid in wells of a plate by well names (e.g., A1, A2, B3).
|
"""Set the liquid in wells of a plate by well names (e.g., A1, A2, B3).
|
||||||
|
|
||||||
如果 liquid_names 和 volumes 为空,但 plate 和 well_names 不为空,直接返回 plate 和 wells。
|
如果 liquid_names 和 volumes 为空,但 plate 和 well_names 不为空,直接返回 plate 和 wells。
|
||||||
"""
|
"""
|
||||||
assert issubclass(plate.__class__, Plate), "plate must be a Plate"
|
|
||||||
plate: Plate = cast(Plate, cast(Resource, plate))
|
|
||||||
# 根据 well_names 获取对应的 Well 对象
|
# 根据 well_names 获取对应的 Well 对象
|
||||||
wells = [plate.get_well(name) for name in well_names]
|
wells = [plate.get_well(name) for name in well_names]
|
||||||
res_volumes = []
|
res_volumes = []
|
||||||
@@ -714,14 +706,6 @@ class LiquidHandlerAbstract(LiquidHandlerMiddleware):
|
|||||||
well.set_liquids([(liquid_name, volume)]) # type: ignore
|
well.set_liquids([(liquid_name, volume)]) # type: ignore
|
||||||
res_volumes.append(volume)
|
res_volumes.append(volume)
|
||||||
|
|
||||||
task = ROS2DeviceNode.run_async_func(self._ros_node.update_resource, True, **{"resources": wells})
|
|
||||||
submit_time = time.time()
|
|
||||||
while not task.done():
|
|
||||||
if time.time() - submit_time > 10:
|
|
||||||
self._ros_node.lab_logger().info(f"set_liquid_from_plate {plate} 超时")
|
|
||||||
break
|
|
||||||
time.sleep(0.01)
|
|
||||||
|
|
||||||
return SetLiquidFromPlateReturn(
|
return SetLiquidFromPlateReturn(
|
||||||
plate=ResourceTreeSet.from_plr_resources([plate], known_newly_created=False).dump(), # type: ignore
|
plate=ResourceTreeSet.from_plr_resources([plate], known_newly_created=False).dump(), # type: ignore
|
||||||
wells=ResourceTreeSet.from_plr_resources(wells, known_newly_created=False).dump(), # type: ignore
|
wells=ResourceTreeSet.from_plr_resources(wells, known_newly_created=False).dump(), # type: ignore
|
||||||
@@ -1127,7 +1111,7 @@ class LiquidHandlerAbstract(LiquidHandlerMiddleware):
|
|||||||
mix_liquid_height: Optional[float] = None,
|
mix_liquid_height: Optional[float] = None,
|
||||||
delays: Optional[List[int]] = None,
|
delays: Optional[List[int]] = None,
|
||||||
none_keys: List[str] = [],
|
none_keys: List[str] = [],
|
||||||
) -> TransferLiquidReturn:
|
):
|
||||||
"""Transfer liquid with automatic mode detection.
|
"""Transfer liquid with automatic mode detection.
|
||||||
|
|
||||||
Supports three transfer modes:
|
Supports three transfer modes:
|
||||||
@@ -1267,11 +1251,6 @@ class LiquidHandlerAbstract(LiquidHandlerMiddleware):
|
|||||||
"Supported modes: 1->N, N->1, or N->N."
|
"Supported modes: 1->N, N->1, or N->N."
|
||||||
)
|
)
|
||||||
|
|
||||||
return TransferLiquidReturn(
|
|
||||||
sources=ResourceTreeSet.from_plr_resources(list(sources), known_newly_created=False).dump(), # type: ignore
|
|
||||||
targets=ResourceTreeSet.from_plr_resources(list(targets), known_newly_created=False).dump(), # type: ignore
|
|
||||||
)
|
|
||||||
|
|
||||||
async def _transfer_one_to_one(
|
async def _transfer_one_to_one(
|
||||||
self,
|
self,
|
||||||
sources: Sequence[Container],
|
sources: Sequence[Container],
|
||||||
|
|||||||
@@ -52,7 +52,6 @@ from unilabos.devices.liquid_handling.liquid_handler_abstract import (
|
|||||||
SimpleReturn,
|
SimpleReturn,
|
||||||
SetLiquidReturn,
|
SetLiquidReturn,
|
||||||
SetLiquidFromPlateReturn,
|
SetLiquidFromPlateReturn,
|
||||||
TransferLiquidReturn,
|
|
||||||
)
|
)
|
||||||
from unilabos.registry.placeholder_type import ResourceSlot
|
from unilabos.registry.placeholder_type import ResourceSlot
|
||||||
from unilabos.ros.nodes.base_device_node import BaseROS2DeviceNode
|
from unilabos.ros.nodes.base_device_node import BaseROS2DeviceNode
|
||||||
@@ -155,29 +154,25 @@ class PRCXI9300Plate(Plate):
|
|||||||
**kwargs,
|
**kwargs,
|
||||||
):
|
):
|
||||||
# 如果 ordered_items 不为 None,直接使用
|
# 如果 ordered_items 不为 None,直接使用
|
||||||
items = None
|
|
||||||
ordering_param = None
|
|
||||||
if ordered_items is not None:
|
if ordered_items is not None:
|
||||||
items = ordered_items
|
items = ordered_items
|
||||||
elif ordering is not None:
|
elif ordering is not None:
|
||||||
# 检查 ordering 中的值是否是字符串(从 JSON 反序列化时的情况)
|
# 检查 ordering 中的值是否是字符串(从 JSON 反序列化时的情况)
|
||||||
# 如果是字符串,说明这是位置名称,需要让 Plate 自己创建 Well 对象
|
# 如果是字符串,说明这是位置名称,需要让 Plate 自己创建 Well 对象
|
||||||
# 我们只传递位置信息(键),不传递值,使用 ordering 参数
|
# 我们只传递位置信息(键),不传递值,使用 ordering 参数
|
||||||
if ordering:
|
if ordering and isinstance(next(iter(ordering.values()), None), str):
|
||||||
values = list(ordering.values())
|
# ordering 的值是字符串,只使用键(位置信息)创建新的 OrderedDict
|
||||||
value = values[0]
|
# 传递 ordering 参数而不是 ordered_items,让 Plate 自己创建 Well 对象
|
||||||
if isinstance(value, str):
|
items = None
|
||||||
# ordering 的值是字符串,只使用键(位置信息)创建新的 OrderedDict
|
# 使用 ordering 参数,只包含位置信息(键)
|
||||||
# 传递 ordering 参数而不是 ordered_items,让 Plate 自己创建 Well 对象
|
ordering_param = collections.OrderedDict((k, None) for k in ordering.keys())
|
||||||
items = None
|
|
||||||
# 使用 ordering 参数,只包含位置信息(键)
|
|
||||||
ordering_param = collections.OrderedDict((k, None) for k in ordering.keys())
|
|
||||||
elif value is None:
|
|
||||||
ordering_param = ordering
|
|
||||||
else:
|
else:
|
||||||
# ordering 的值已经是对象,可以直接使用
|
# ordering 的值已经是对象,可以直接使用
|
||||||
items = ordering
|
items = ordering
|
||||||
ordering_param = None
|
ordering_param = None
|
||||||
|
else:
|
||||||
|
items = None
|
||||||
|
ordering_param = None
|
||||||
|
|
||||||
# 根据情况传递不同的参数
|
# 根据情况传递不同的参数
|
||||||
if items is not None:
|
if items is not None:
|
||||||
@@ -718,7 +713,7 @@ class PRCXI9300Handler(LiquidHandlerAbstract):
|
|||||||
mix_liquid_height: Optional[float] = None,
|
mix_liquid_height: Optional[float] = None,
|
||||||
delays: Optional[List[int]] = None,
|
delays: Optional[List[int]] = None,
|
||||||
none_keys: List[str] = [],
|
none_keys: List[str] = [],
|
||||||
) -> TransferLiquidReturn:
|
):
|
||||||
return await super().transfer_liquid(
|
return await super().transfer_liquid(
|
||||||
sources,
|
sources,
|
||||||
targets,
|
targets,
|
||||||
|
|||||||
376
unilabos/devices/motor/ZDT_X42.py
Normal file
376
unilabos/devices/motor/ZDT_X42.py
Normal file
@@ -0,0 +1,376 @@
|
|||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
"""
|
||||||
|
ZDT X42 Closed-Loop Stepper Motor Driver
|
||||||
|
RS485 Serial Communication via USB-Serial Converter
|
||||||
|
|
||||||
|
- Baudrate: 115200
|
||||||
|
"""
|
||||||
|
|
||||||
|
import serial
|
||||||
|
import time
|
||||||
|
import threading
|
||||||
|
import struct
|
||||||
|
import logging
|
||||||
|
from typing import Optional, Any
|
||||||
|
|
||||||
|
try:
|
||||||
|
from unilabos.device_comms.universal_driver import UniversalDriver
|
||||||
|
except ImportError:
|
||||||
|
class UniversalDriver:
|
||||||
|
def __init__(self, *args, **kwargs):
|
||||||
|
self.logger = logging.getLogger(self.__class__.__name__)
|
||||||
|
def execute_command_from_outer(self, command: Any): pass
|
||||||
|
|
||||||
|
from serial.rs485 import RS485Settings
|
||||||
|
|
||||||
|
|
||||||
|
class ZDTX42Driver(UniversalDriver):
|
||||||
|
"""
|
||||||
|
ZDT X42 闭环步进电机驱动器
|
||||||
|
|
||||||
|
支持功能:
|
||||||
|
- 速度模式运行
|
||||||
|
- 位置模式运行 (相对/绝对)
|
||||||
|
- 位置读取和清零
|
||||||
|
- 使能/禁用控制
|
||||||
|
|
||||||
|
通信协议:
|
||||||
|
- 帧格式: [设备ID] [功能码] [数据...] [校验位=0x6B]
|
||||||
|
- 响应长度根据功能码决定
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
port: str,
|
||||||
|
baudrate: int = 115200,
|
||||||
|
device_id: int = 1,
|
||||||
|
timeout: float = 0.5,
|
||||||
|
debug: bool = False
|
||||||
|
):
|
||||||
|
"""
|
||||||
|
初始化 ZDT X42 电机驱动
|
||||||
|
|
||||||
|
Args:
|
||||||
|
port: 串口设备路径
|
||||||
|
baudrate: 波特率 (默认 115200)
|
||||||
|
device_id: 设备地址 (1-255)
|
||||||
|
timeout: 通信超时时间(秒)
|
||||||
|
debug: 是否启用调试输出
|
||||||
|
"""
|
||||||
|
super().__init__()
|
||||||
|
self.id = device_id
|
||||||
|
self.debug = debug
|
||||||
|
self.lock = threading.RLock()
|
||||||
|
self.status = "idle" # 对应注册表中的 status (str)
|
||||||
|
self.position = 0 # 对应注册表中的 position (int)
|
||||||
|
|
||||||
|
try:
|
||||||
|
self.ser = serial.Serial(
|
||||||
|
port=port,
|
||||||
|
baudrate=baudrate,
|
||||||
|
timeout=timeout,
|
||||||
|
bytesize=serial.EIGHTBITS,
|
||||||
|
parity=serial.PARITY_NONE,
|
||||||
|
stopbits=serial.STOPBITS_ONE
|
||||||
|
)
|
||||||
|
|
||||||
|
# 启用 RS485 模式
|
||||||
|
try:
|
||||||
|
self.ser.rs485_mode = RS485Settings(
|
||||||
|
rts_level_for_tx=True,
|
||||||
|
rts_level_for_rx=False
|
||||||
|
)
|
||||||
|
except Exception:
|
||||||
|
pass # RS485 模式是可选的
|
||||||
|
|
||||||
|
self.logger.info(
|
||||||
|
f"ZDT X42 Motor connected: {port} "
|
||||||
|
f"(Baud: {baudrate}, ID: {device_id})"
|
||||||
|
)
|
||||||
|
# 自动使能电机,确保初始状态可运动
|
||||||
|
self.enable(True)
|
||||||
|
|
||||||
|
# 启动背景轮询线程,确保 position 实时刷新
|
||||||
|
self._stop_event = threading.Event()
|
||||||
|
self._polling_thread = threading.Thread(
|
||||||
|
target=self._update_loop,
|
||||||
|
name=f"ZDTPolling_{port}",
|
||||||
|
daemon=True
|
||||||
|
)
|
||||||
|
self._polling_thread.start()
|
||||||
|
except Exception as e:
|
||||||
|
self.logger.error(f"Failed to open serial port {port}: {e}")
|
||||||
|
self.ser = None
|
||||||
|
|
||||||
|
def _update_loop(self):
|
||||||
|
"""背景循环读取电机位置"""
|
||||||
|
while not self._stop_event.is_set():
|
||||||
|
try:
|
||||||
|
self.get_position()
|
||||||
|
except Exception as e:
|
||||||
|
if self.debug:
|
||||||
|
self.logger.error(f"Polling error: {e}")
|
||||||
|
time.sleep(1.0) # 每1秒刷新一次位置数据
|
||||||
|
|
||||||
|
def _send(self, func_code: int, payload: list) -> bytes:
|
||||||
|
"""
|
||||||
|
发送指令并接收响应
|
||||||
|
|
||||||
|
Args:
|
||||||
|
func_code: 功能码
|
||||||
|
payload: 数据负载 (list of bytes)
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
响应数据 (bytes)
|
||||||
|
"""
|
||||||
|
if not self.ser:
|
||||||
|
self.logger.error("Serial port not available")
|
||||||
|
return b""
|
||||||
|
|
||||||
|
with self.lock:
|
||||||
|
# 清空输入缓冲区
|
||||||
|
self.ser.reset_input_buffer()
|
||||||
|
|
||||||
|
# 构建消息: [ID] [功能码] [数据...] [校验位=0x6B]
|
||||||
|
message = bytes([self.id, func_code] + payload + [0x6B])
|
||||||
|
|
||||||
|
# 发送
|
||||||
|
self.ser.write(message)
|
||||||
|
|
||||||
|
# 根据功能码决定响应长度
|
||||||
|
# 查询类指令返回 10 字节,控制类指令返回 4 字节
|
||||||
|
read_len = 10 if func_code in [0x31, 0x32, 0x35, 0x24, 0x27] else 4
|
||||||
|
response = self.ser.read(read_len)
|
||||||
|
|
||||||
|
# 调试输出
|
||||||
|
if self.debug:
|
||||||
|
sent_hex = message.hex().upper()
|
||||||
|
recv_hex = response.hex().upper() if response else 'TIMEOUT'
|
||||||
|
print(f"[ID {self.id}] TX: {sent_hex} → RX: {recv_hex}")
|
||||||
|
|
||||||
|
return response
|
||||||
|
|
||||||
|
def enable(self, on: bool = True) -> bool:
|
||||||
|
"""
|
||||||
|
使能/禁用电机
|
||||||
|
|
||||||
|
Args:
|
||||||
|
on: True=使能(锁轴), False=禁用(松轴)
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
是否成功
|
||||||
|
"""
|
||||||
|
state = 1 if on else 0
|
||||||
|
resp = self._send(0xF3, [0xAB, state, 0])
|
||||||
|
return len(resp) >= 4
|
||||||
|
|
||||||
|
def move_speed(
|
||||||
|
self,
|
||||||
|
speed_rpm: int,
|
||||||
|
direction: str = "CW",
|
||||||
|
acceleration: int = 10
|
||||||
|
) -> bool:
|
||||||
|
"""
|
||||||
|
速度模式运行
|
||||||
|
|
||||||
|
Args:
|
||||||
|
speed_rpm: 转速 (RPM)
|
||||||
|
direction: 方向 ("CW"=顺时针, "CCW"=逆时针)
|
||||||
|
acceleration: 加速度 (0-255)
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
是否成功
|
||||||
|
"""
|
||||||
|
dir_val = 0 if direction.upper() in ["CW", "顺时针"] else 1
|
||||||
|
speed_bytes = struct.pack('>H', int(speed_rpm))
|
||||||
|
self.status = f"moving@{speed_rpm}rpm"
|
||||||
|
resp = self._send(0xF6, [dir_val, speed_bytes[0], speed_bytes[1], acceleration, 0])
|
||||||
|
return len(resp) >= 4
|
||||||
|
|
||||||
|
def move_position(
|
||||||
|
self,
|
||||||
|
pulses: int,
|
||||||
|
speed_rpm: int,
|
||||||
|
direction: str = "CW",
|
||||||
|
acceleration: int = 10,
|
||||||
|
absolute: bool = False
|
||||||
|
) -> bool:
|
||||||
|
"""
|
||||||
|
位置模式运行
|
||||||
|
|
||||||
|
Args:
|
||||||
|
pulses: 脉冲数
|
||||||
|
speed_rpm: 转速 (RPM)
|
||||||
|
direction: 方向 ("CW"=顺时针, "CCW"=逆时针)
|
||||||
|
acceleration: 加速度 (0-255)
|
||||||
|
absolute: True=绝对位置, False=相对位置
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
是否成功
|
||||||
|
"""
|
||||||
|
dir_val = 0 if direction.upper() in ["CW", "顺时针"] else 1
|
||||||
|
speed_bytes = struct.pack('>H', int(speed_rpm))
|
||||||
|
self.status = f"moving_to_{pulses}"
|
||||||
|
pulse_bytes = struct.pack('>I', int(pulses))
|
||||||
|
abs_flag = 1 if absolute else 0
|
||||||
|
|
||||||
|
payload = [
|
||||||
|
dir_val,
|
||||||
|
speed_bytes[0], speed_bytes[1],
|
||||||
|
acceleration,
|
||||||
|
pulse_bytes[0], pulse_bytes[1], pulse_bytes[2], pulse_bytes[3],
|
||||||
|
abs_flag,
|
||||||
|
0
|
||||||
|
]
|
||||||
|
|
||||||
|
resp = self._send(0xFD, payload)
|
||||||
|
return len(resp) >= 4
|
||||||
|
|
||||||
|
def stop(self) -> bool:
|
||||||
|
"""
|
||||||
|
停止电机
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
是否成功
|
||||||
|
"""
|
||||||
|
self.status = "idle"
|
||||||
|
resp = self._send(0xFE, [0x98, 0])
|
||||||
|
return len(resp) >= 4
|
||||||
|
|
||||||
|
def rotate_quarter(self, speed_rpm: int = 60, direction: str = "CW") -> bool:
|
||||||
|
"""
|
||||||
|
电机旋转 1/4 圈 (阻塞式)
|
||||||
|
假设电机细分为 3200 脉冲/圈,1/4 圈 = 800 脉冲
|
||||||
|
"""
|
||||||
|
pulses = 800
|
||||||
|
success = self.move_position(pulses=pulses, speed_rpm=speed_rpm, direction=direction, absolute=False)
|
||||||
|
|
||||||
|
if success:
|
||||||
|
# 计算预估旋转时间并进行阻塞等待 (Time = revolutions / (RPM/60))
|
||||||
|
# 1/4 rev / (RPM/60) = 15.0 / RPM
|
||||||
|
estimated_time = 15.0 / max(1, speed_rpm)
|
||||||
|
time.sleep(estimated_time + 0.5) # 额外给 0.5 秒缓冲
|
||||||
|
self.status = "idle"
|
||||||
|
|
||||||
|
return success
|
||||||
|
|
||||||
|
def wait_time(self, duration_s: float) -> bool:
|
||||||
|
"""
|
||||||
|
等待指定时间 (秒)
|
||||||
|
"""
|
||||||
|
self.logger.info(f"Waiting for {duration_s} seconds...")
|
||||||
|
time.sleep(duration_s)
|
||||||
|
return True
|
||||||
|
|
||||||
|
def set_zero(self) -> bool:
|
||||||
|
"""
|
||||||
|
清零当前位置
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
是否成功
|
||||||
|
"""
|
||||||
|
resp = self._send(0x0A, [])
|
||||||
|
return len(resp) >= 4
|
||||||
|
|
||||||
|
def get_position(self) -> Optional[int]:
|
||||||
|
"""
|
||||||
|
读取当前位置 (脉冲数)
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
当前位置脉冲数,失败返回 None
|
||||||
|
"""
|
||||||
|
resp = self._send(0x32, [])
|
||||||
|
|
||||||
|
if len(resp) >= 8:
|
||||||
|
# 响应格式: [ID] [Func] [符号位] [数值4字节] [校验]
|
||||||
|
sign = resp[2] # 0=正, 1=负
|
||||||
|
value = struct.unpack('>I', resp[3:7])[0]
|
||||||
|
self.position = -value if sign == 1 else value
|
||||||
|
|
||||||
|
if self.debug:
|
||||||
|
print(f"[Position] Raw: {resp.hex().upper()}, Parsed: {self.position}")
|
||||||
|
|
||||||
|
return self.position
|
||||||
|
|
||||||
|
self.logger.warning("Failed to read position")
|
||||||
|
return None
|
||||||
|
|
||||||
|
def close(self):
|
||||||
|
"""关闭串口连接并停止线程"""
|
||||||
|
if hasattr(self, '_stop_event'):
|
||||||
|
self._stop_event.set()
|
||||||
|
|
||||||
|
if self.ser and self.ser.is_open:
|
||||||
|
self.ser.close()
|
||||||
|
self.logger.info("Serial port closed")
|
||||||
|
|
||||||
|
|
||||||
|
# ============================================================
|
||||||
|
# 测试和调试代码
|
||||||
|
# ============================================================
|
||||||
|
|
||||||
|
def test_motor():
|
||||||
|
"""基础功能测试"""
|
||||||
|
logging.basicConfig(level=logging.INFO)
|
||||||
|
|
||||||
|
print("="*60)
|
||||||
|
print("ZDT X42 电机驱动测试")
|
||||||
|
print("="*60)
|
||||||
|
|
||||||
|
driver = ZDTX42Driver(
|
||||||
|
port="/dev/tty.usbserial-3110",
|
||||||
|
baudrate=115200,
|
||||||
|
device_id=2,
|
||||||
|
debug=True
|
||||||
|
)
|
||||||
|
|
||||||
|
if not driver.ser:
|
||||||
|
print("❌ 串口打开失败")
|
||||||
|
return
|
||||||
|
|
||||||
|
try:
|
||||||
|
# 测试 1: 读取位置
|
||||||
|
print("\n[1] 读取当前位置")
|
||||||
|
pos = driver.get_position()
|
||||||
|
print(f"✓ 当前位置: {pos} 脉冲")
|
||||||
|
|
||||||
|
# 测试 2: 使能
|
||||||
|
print("\n[2] 使能电机")
|
||||||
|
driver.enable(True)
|
||||||
|
time.sleep(0.3)
|
||||||
|
print("✓ 电机已锁定")
|
||||||
|
|
||||||
|
# 测试 3: 相对位置运动
|
||||||
|
print("\n[3] 相对位置运动 (1000脉冲)")
|
||||||
|
driver.move_position(pulses=1000, speed_rpm=60, direction="CW")
|
||||||
|
time.sleep(2)
|
||||||
|
pos = driver.get_position()
|
||||||
|
print(f"✓ 新位置: {pos}")
|
||||||
|
|
||||||
|
# 测试 4: 速度运动
|
||||||
|
print("\n[4] 速度模式 (30RPM, 3秒)")
|
||||||
|
driver.move_speed(speed_rpm=30, direction="CW")
|
||||||
|
time.sleep(3)
|
||||||
|
driver.stop()
|
||||||
|
pos = driver.get_position()
|
||||||
|
print(f"✓ 停止后位置: {pos}")
|
||||||
|
|
||||||
|
# 测试 5: 禁用
|
||||||
|
print("\n[5] 禁用电机")
|
||||||
|
driver.enable(False)
|
||||||
|
print("✓ 电机已松开")
|
||||||
|
|
||||||
|
print("\n" + "="*60)
|
||||||
|
print("✅ 测试完成")
|
||||||
|
print("="*60)
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
print(f"\n❌ 测试失败: {e}")
|
||||||
|
import traceback
|
||||||
|
traceback.print_exc()
|
||||||
|
finally:
|
||||||
|
driver.close()
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
test_motor()
|
||||||
@@ -623,6 +623,119 @@ class ChinweDevice(UniversalDriver):
|
|||||||
time.sleep(duration)
|
time.sleep(duration)
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
def separation_step(self, motor_id: int = 5, speed: int = 60, pulses: int = 700,
|
||||||
|
max_cycles: int = 0, timeout: int = 300) -> bool:
|
||||||
|
"""
|
||||||
|
分液步骤 - 液位传感器与电机联动
|
||||||
|
当液位传感器检测到"有液"时,电机顺时针旋转指定脉冲数
|
||||||
|
当液位传感器检测到"无液"时,电机逆时针旋转指定脉冲数
|
||||||
|
|
||||||
|
:param motor_id: 电机ID (必须在初始化时配置的motor_ids中)
|
||||||
|
:param speed: 电机转速 (RPM)
|
||||||
|
:param pulses: 每次旋转的脉冲数 (默认700约为1/4圈,假设3200脉冲/圈)
|
||||||
|
:param max_cycles: 最大执行循环次数 (0=无限制,默认0)
|
||||||
|
:param timeout: 整体超时时间 (秒)
|
||||||
|
:return: 成功返回True,超时或失败返回False
|
||||||
|
"""
|
||||||
|
motor_id = int(motor_id)
|
||||||
|
speed = int(speed)
|
||||||
|
pulses = int(pulses)
|
||||||
|
max_cycles = int(max_cycles)
|
||||||
|
timeout = int(timeout)
|
||||||
|
|
||||||
|
# 检查电机是否存在
|
||||||
|
if motor_id not in self.motors:
|
||||||
|
self.logger.error(f"Motor {motor_id} not found in configured motors: {list(self.motors.keys())}")
|
||||||
|
return False
|
||||||
|
|
||||||
|
# 检查传感器是否可用
|
||||||
|
if not self.sensor:
|
||||||
|
self.logger.error("Sensor not initialized")
|
||||||
|
return False
|
||||||
|
|
||||||
|
motor = self.motors[motor_id]
|
||||||
|
|
||||||
|
# 停止轮询线程,避免与 separation_step 同时读取传感器造成串口冲突
|
||||||
|
self.logger.info("Stopping polling thread for separation_step...")
|
||||||
|
self._stop_event.set()
|
||||||
|
if self._poll_thread and self._poll_thread.is_alive():
|
||||||
|
self._poll_thread.join(timeout=2.0)
|
||||||
|
|
||||||
|
# 使能电机
|
||||||
|
self.logger.info(f"Enabling motor {motor_id}...")
|
||||||
|
motor.enable(True)
|
||||||
|
time.sleep(0.2)
|
||||||
|
|
||||||
|
self.logger.info(f"Starting separation step: motor_id={motor_id}, speed={speed} RPM, "
|
||||||
|
f"pulses={pulses}, max_cycles={max_cycles}, timeout={timeout}s")
|
||||||
|
|
||||||
|
# 记录上一次的液位状态
|
||||||
|
last_level = None
|
||||||
|
cycle_count = 0
|
||||||
|
start_time = time.time()
|
||||||
|
error_count = 0
|
||||||
|
|
||||||
|
try:
|
||||||
|
while True:
|
||||||
|
# 检查超时
|
||||||
|
if time.time() - start_time > timeout:
|
||||||
|
self.logger.warning(f"Separation step timeout after {timeout} seconds")
|
||||||
|
return False
|
||||||
|
|
||||||
|
# 检查循环次数限制
|
||||||
|
if max_cycles > 0 and cycle_count >= max_cycles:
|
||||||
|
self.logger.info(f"Separation step completed: reached max_cycles={max_cycles}")
|
||||||
|
return True
|
||||||
|
|
||||||
|
# 读取传感器数据
|
||||||
|
data = self.sensor.read_level()
|
||||||
|
|
||||||
|
if data is None:
|
||||||
|
error_count += 1
|
||||||
|
if error_count > 5:
|
||||||
|
self.logger.warning("Sensor read failed multiple times, retrying...")
|
||||||
|
error_count = 0
|
||||||
|
time.sleep(0.5)
|
||||||
|
continue
|
||||||
|
|
||||||
|
error_count = 0
|
||||||
|
current_level = data['level']
|
||||||
|
rssi = data['rssi']
|
||||||
|
|
||||||
|
# 检测状态变化 (包括首次检测)
|
||||||
|
if current_level != last_level:
|
||||||
|
cycle_count += 1
|
||||||
|
|
||||||
|
if current_level:
|
||||||
|
# 有液 -> 电机顺时针旋转
|
||||||
|
self.logger.info(f"[Cycle {cycle_count}] Liquid detected (RSSI={rssi}), "
|
||||||
|
f"rotating motor {motor_id} clockwise {pulses} pulses")
|
||||||
|
motor.run_position(pulses=pulses, speed_rpm=speed, direction=0, absolute=False)
|
||||||
|
|
||||||
|
# 等待电机完成 (预估时间)
|
||||||
|
estimated_time = 15.0 / max(1, speed)
|
||||||
|
time.sleep(estimated_time + 0.5)
|
||||||
|
|
||||||
|
else:
|
||||||
|
# 无液 -> 电机逆时针旋转
|
||||||
|
self.logger.info(f"[Cycle {cycle_count}] No liquid detected (RSSI={rssi}), "
|
||||||
|
f"rotating motor {motor_id} counter-clockwise {pulses} pulses")
|
||||||
|
motor.run_position(pulses=pulses, speed_rpm=speed, direction=1, absolute=False)
|
||||||
|
|
||||||
|
# 等待电机完成 (预估时间)
|
||||||
|
estimated_time = 15.0 / max(1, speed)
|
||||||
|
time.sleep(estimated_time + 0.5)
|
||||||
|
|
||||||
|
# 更新状态
|
||||||
|
last_level = current_level
|
||||||
|
|
||||||
|
# 轮询间隔
|
||||||
|
time.sleep(0.1)
|
||||||
|
finally:
|
||||||
|
# 恢复轮询线程
|
||||||
|
self.logger.info("Restarting polling thread...")
|
||||||
|
self._start_polling()
|
||||||
|
|
||||||
def execute_command_from_outer(self, command_dict: Dict[str, Any]) -> bool:
|
def execute_command_from_outer(self, command_dict: Dict[str, Any]) -> bool:
|
||||||
"""支持标准 JSON 指令调用"""
|
"""支持标准 JSON 指令调用"""
|
||||||
return super().execute_command_from_outer(command_dict)
|
return super().execute_command_from_outer(command_dict)
|
||||||
|
|||||||
379
unilabos/devices/separator/xkc_sensor.py
Normal file
379
unilabos/devices/separator/xkc_sensor.py
Normal file
@@ -0,0 +1,379 @@
|
|||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
"""
|
||||||
|
XKC RS485 液位传感器 (Modbus RTU)
|
||||||
|
|
||||||
|
说明:
|
||||||
|
1. 遵循 Modbus-RTU 协议。
|
||||||
|
2. 数据寄存器: 0x0001 (液位状态, 1=有液, 0=无液), 0x0002 (RSSI 信号强度)。
|
||||||
|
3. 地址寄存器: 0x0004 (可读写, 范围 1-254)。
|
||||||
|
4. 波特率寄存器: 0x0005 (可写, 代码表见 change_baudrate 方法)。
|
||||||
|
"""
|
||||||
|
|
||||||
|
import struct
|
||||||
|
import threading
|
||||||
|
import time
|
||||||
|
import logging
|
||||||
|
import serial
|
||||||
|
from typing import Optional, Dict, Any, List
|
||||||
|
|
||||||
|
from unilabos.device_comms.universal_driver import UniversalDriver
|
||||||
|
|
||||||
|
class TransportManager:
|
||||||
|
"""
|
||||||
|
统一通信管理类。
|
||||||
|
仅支持 串口 (Serial/有线) 连接。
|
||||||
|
"""
|
||||||
|
def __init__(self, port: str, baudrate: int = 9600, timeout: float = 3.0, logger=None):
|
||||||
|
self.port = port
|
||||||
|
self.baudrate = baudrate
|
||||||
|
self.timeout = timeout
|
||||||
|
self.logger = logger
|
||||||
|
self.lock = threading.RLock() # 线程锁,确保多设备共用一个连接时不冲突
|
||||||
|
|
||||||
|
self.serial = None
|
||||||
|
self._connect_serial()
|
||||||
|
|
||||||
|
def _connect_serial(self):
|
||||||
|
try:
|
||||||
|
self.serial = serial.Serial(
|
||||||
|
port=self.port,
|
||||||
|
baudrate=self.baudrate,
|
||||||
|
timeout=self.timeout
|
||||||
|
)
|
||||||
|
except Exception as e:
|
||||||
|
raise ConnectionError(f"Serial open failed: {e}")
|
||||||
|
|
||||||
|
def close(self):
|
||||||
|
"""关闭连接"""
|
||||||
|
if self.serial and self.serial.is_open:
|
||||||
|
self.serial.close()
|
||||||
|
|
||||||
|
def clear_buffer(self):
|
||||||
|
"""清空缓冲区 (Thread-safe)"""
|
||||||
|
with self.lock:
|
||||||
|
if self.serial:
|
||||||
|
self.serial.reset_input_buffer()
|
||||||
|
|
||||||
|
def write(self, data: bytes):
|
||||||
|
"""发送原始字节"""
|
||||||
|
with self.lock:
|
||||||
|
if self.serial:
|
||||||
|
self.serial.write(data)
|
||||||
|
|
||||||
|
def read(self, size: int) -> bytes:
|
||||||
|
"""读取指定长度字节"""
|
||||||
|
if self.serial:
|
||||||
|
return self.serial.read(size)
|
||||||
|
return b''
|
||||||
|
|
||||||
|
class XKCSensorDriver(UniversalDriver):
|
||||||
|
"""XKC RS485 液位传感器 (Modbus RTU)"""
|
||||||
|
|
||||||
|
def __init__(self, port: str, baudrate: int = 9600, device_id: int = 6,
|
||||||
|
threshold: int = 300, timeout: float = 3.0, debug: bool = False):
|
||||||
|
super().__init__()
|
||||||
|
self.port = port
|
||||||
|
self.baudrate = baudrate
|
||||||
|
self.device_id = device_id
|
||||||
|
self.threshold = threshold
|
||||||
|
self.timeout = timeout
|
||||||
|
self.debug = debug
|
||||||
|
self.level = False
|
||||||
|
self.rssi = 0
|
||||||
|
self.status = {"level": self.level, "rssi": self.rssi}
|
||||||
|
|
||||||
|
try:
|
||||||
|
self.transport = TransportManager(port, baudrate, timeout, logger=self.logger)
|
||||||
|
self.logger.info(f"XKCSensorDriver connected to {port} (ID: {device_id})")
|
||||||
|
except Exception as e:
|
||||||
|
self.logger.error(f"Failed to connect XKCSensorDriver: {e}")
|
||||||
|
self.transport = None
|
||||||
|
|
||||||
|
# 启动背景轮询线程,确保 status 实时刷新
|
||||||
|
self._stop_event = threading.Event()
|
||||||
|
self._polling_thread = threading.Thread(
|
||||||
|
target=self._update_loop,
|
||||||
|
name=f"XKCPolling_{port}",
|
||||||
|
daemon=True
|
||||||
|
)
|
||||||
|
if self.transport:
|
||||||
|
self._polling_thread.start()
|
||||||
|
|
||||||
|
def _update_loop(self):
|
||||||
|
"""背景循环读取传感器数据"""
|
||||||
|
while not self._stop_event.is_set():
|
||||||
|
try:
|
||||||
|
self.read_level()
|
||||||
|
except Exception as e:
|
||||||
|
if self.debug:
|
||||||
|
self.logger.error(f"Polling error: {e}")
|
||||||
|
time.sleep(2.0) # 每2秒刷新一次数据
|
||||||
|
|
||||||
|
def _crc(self, data: bytes) -> bytes:
|
||||||
|
crc = 0xFFFF
|
||||||
|
for byte in data:
|
||||||
|
crc ^= byte
|
||||||
|
for _ in range(8):
|
||||||
|
if crc & 0x0001: crc = (crc >> 1) ^ 0xA001
|
||||||
|
else: crc >>= 1
|
||||||
|
return struct.pack('<H', crc)
|
||||||
|
|
||||||
|
def read_level(self) -> Optional[Dict[str, Any]]:
|
||||||
|
"""
|
||||||
|
读取液位。
|
||||||
|
返回: {'level': bool, 'rssi': int}
|
||||||
|
"""
|
||||||
|
if not self.transport:
|
||||||
|
return None
|
||||||
|
|
||||||
|
with self.transport.lock:
|
||||||
|
self.transport.clear_buffer()
|
||||||
|
# Modbus Read Registers: 01 03 00 01 00 02 CRC
|
||||||
|
payload = struct.pack('>HH', 0x0001, 0x0002)
|
||||||
|
msg = struct.pack('BB', self.device_id, 0x03) + payload
|
||||||
|
msg += self._crc(msg)
|
||||||
|
|
||||||
|
if self.debug:
|
||||||
|
self.logger.info(f"TX (ID {self.device_id}): {msg.hex().upper()}")
|
||||||
|
|
||||||
|
self.transport.write(msg)
|
||||||
|
|
||||||
|
# Read header
|
||||||
|
h = self.transport.read(3) # Addr, Func, Len
|
||||||
|
if self.debug:
|
||||||
|
self.logger.info(f"RX Header: {h.hex().upper()}")
|
||||||
|
|
||||||
|
if len(h) < 3: return None
|
||||||
|
length = h[2]
|
||||||
|
|
||||||
|
# Read body + CRC
|
||||||
|
body = self.transport.read(length + 2)
|
||||||
|
if self.debug:
|
||||||
|
self.logger.info(f"RX Body+CRC: {body.hex().upper()}")
|
||||||
|
if len(body) < length + 2:
|
||||||
|
# Firmware bug fix specific to some modules
|
||||||
|
if len(body) == 4 and length == 4:
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
return None
|
||||||
|
|
||||||
|
data = body[:-2]
|
||||||
|
# 根据手册说明:
|
||||||
|
# 寄存器 0x0001 (data[0:2]): 液位状态 (00 01 为有液, 00 00 为无液)
|
||||||
|
# 寄存器 0x0002 (data[2:4]): 信号强度 RSSI
|
||||||
|
|
||||||
|
hw_level = False
|
||||||
|
rssi = 0
|
||||||
|
|
||||||
|
if len(data) >= 4:
|
||||||
|
hw_level = ((data[0] << 8) | data[1]) == 1
|
||||||
|
rssi = (data[2] << 8) | data[3]
|
||||||
|
elif len(data) == 2:
|
||||||
|
# 兼容模式: 某些老固件可能只返回 1 个寄存器
|
||||||
|
rssi = (data[0] << 8) | data[1]
|
||||||
|
hw_level = rssi > self.threshold
|
||||||
|
else:
|
||||||
|
return None
|
||||||
|
|
||||||
|
# 最终判定: 优先使用硬件层级的 level 判定,但 RSSI 阈值逻辑作为补充/校验
|
||||||
|
# 注意: 如果用户显式设置了 THRESHOLD,我们可以在逻辑中做权衡
|
||||||
|
self.level = hw_level or (rssi > self.threshold)
|
||||||
|
self.rssi = rssi
|
||||||
|
result = {
|
||||||
|
'level': self.level,
|
||||||
|
'rssi': self.rssi
|
||||||
|
}
|
||||||
|
self.status = result
|
||||||
|
return result
|
||||||
|
|
||||||
|
def wait_level(self, target_state: bool, timeout: float = 60.0) -> bool:
|
||||||
|
"""
|
||||||
|
等待液位达到目标状态 (阻塞式)
|
||||||
|
"""
|
||||||
|
self.logger.info(f"Waiting for level: {target_state}")
|
||||||
|
start_time = time.time()
|
||||||
|
while (time.time() - start_time) < timeout:
|
||||||
|
res = self.read_level()
|
||||||
|
if res and res.get('level') == target_state:
|
||||||
|
return True
|
||||||
|
time.sleep(0.5)
|
||||||
|
self.logger.warning(f"Wait level timeout ({timeout}s)")
|
||||||
|
return False
|
||||||
|
|
||||||
|
def wait_for_liquid(self, target_state: bool, timeout: float = 120.0) -> bool:
|
||||||
|
"""
|
||||||
|
实时检测电导率(RSSI)并等待用户指定的“有液”或“无液”状态。
|
||||||
|
一旦检测到符合目标状态,立即返回。
|
||||||
|
|
||||||
|
Args:
|
||||||
|
target_state: True 为“有液”, False 为“无液”
|
||||||
|
timeout: 最大等待时间(秒)
|
||||||
|
"""
|
||||||
|
state_str = "有液" if target_state else "无液"
|
||||||
|
self.logger.info(f"开始实时检测电导率,等待状态: {state_str} (超时: {timeout}s)")
|
||||||
|
|
||||||
|
start_time = time.time()
|
||||||
|
while (time.time() - start_time) < timeout:
|
||||||
|
res = self.read_level() # 内部已更新 self.level 和 self.rssi
|
||||||
|
if res:
|
||||||
|
current_level = res.get('level')
|
||||||
|
current_rssi = res.get('rssi')
|
||||||
|
if current_level == target_state:
|
||||||
|
self.logger.info(f"✅ 检测到目标状态: {state_str} (当前电导率/RSSI: {current_rssi})")
|
||||||
|
return True
|
||||||
|
|
||||||
|
if self.debug:
|
||||||
|
self.logger.debug(f"当前状态: {'有液' if current_level else '无液'}, RSSI: {current_rssi}")
|
||||||
|
|
||||||
|
time.sleep(0.2) # 高频采样
|
||||||
|
|
||||||
|
self.logger.warning(f"❌ 等待 {state_str} 状态超时 ({timeout}s)")
|
||||||
|
return False
|
||||||
|
|
||||||
|
def set_threshold(self, threshold: int):
|
||||||
|
"""设置液位判定阈值"""
|
||||||
|
self.threshold = int(threshold)
|
||||||
|
self.logger.info(f"Threshold updated to: {self.threshold}")
|
||||||
|
|
||||||
|
def change_device_id(self, new_id: int) -> bool:
|
||||||
|
"""
|
||||||
|
修改设备的 Modbus 从站地址。
|
||||||
|
寄存器: 0x0004, 功能码: 0x06
|
||||||
|
"""
|
||||||
|
if not (1 <= new_id <= 254):
|
||||||
|
self.logger.error(f"Invalid device ID: {new_id}. Must be 1-254.")
|
||||||
|
return False
|
||||||
|
|
||||||
|
self.logger.info(f"Changing device ID from {self.device_id} to {new_id}")
|
||||||
|
success = self._write_single_register(0x0004, new_id)
|
||||||
|
if success:
|
||||||
|
self.device_id = new_id # 更新内存中的地址
|
||||||
|
self.logger.info(f"Device ID update command sent successfully (target {new_id}).")
|
||||||
|
return success
|
||||||
|
|
||||||
|
def change_baudrate(self, baud_code: int) -> bool:
|
||||||
|
"""
|
||||||
|
更改通讯波特率 (寄存器: 0x0005)。
|
||||||
|
设置成功后传感器 LED 会闪烁,通常无数据返回。
|
||||||
|
|
||||||
|
波特率代码对照表 (16进制):
|
||||||
|
05: 2400
|
||||||
|
06: 4800
|
||||||
|
07: 9600 (默认)
|
||||||
|
08: 14400
|
||||||
|
09: 19200
|
||||||
|
0A: 28800
|
||||||
|
0C: 57600
|
||||||
|
0D: 115200
|
||||||
|
0E: 128000
|
||||||
|
0F: 256000
|
||||||
|
"""
|
||||||
|
self.logger.info(f"Sending baudrate change command (Code: {baud_code:02X})")
|
||||||
|
# 写入寄存器 0x0005
|
||||||
|
self._write_single_register(0x0005, baud_code)
|
||||||
|
self.logger.info("Baudrate change command executed. Device LED should flash. Please update connection settings.")
|
||||||
|
return True
|
||||||
|
|
||||||
|
def factory_reset(self) -> bool:
|
||||||
|
"""
|
||||||
|
恢复出厂设置 (通过广播地址 FF)。
|
||||||
|
设置地址为 01,逻辑为向 0x0004 写入 0x0002
|
||||||
|
"""
|
||||||
|
self.logger.info("Sending factory reset command via broadcast address FF...")
|
||||||
|
# 广播指令通常无回显
|
||||||
|
self._write_single_register(0x0004, 0x0002, slave_id=0xFF)
|
||||||
|
self.logger.info("Factory reset command sent. Device address should be 01 now.")
|
||||||
|
return True
|
||||||
|
|
||||||
|
def _write_single_register(self, reg_addr: int, value: int, slave_id: Optional[int] = None) -> bool:
|
||||||
|
"""内部辅助函数: Modbus 功能码 06 写单个寄存器"""
|
||||||
|
if not self.transport: return False
|
||||||
|
|
||||||
|
target_id = slave_id if slave_id is not None else self.device_id
|
||||||
|
msg = struct.pack('BBHH', target_id, 0x06, reg_addr, value)
|
||||||
|
msg += self._crc(msg)
|
||||||
|
|
||||||
|
with self.transport.lock:
|
||||||
|
self.transport.clear_buffer()
|
||||||
|
if self.debug:
|
||||||
|
self.logger.info(f"TX Write (Reg {reg_addr:#06x}): {msg.hex().upper()}")
|
||||||
|
|
||||||
|
self.transport.write(msg)
|
||||||
|
|
||||||
|
# 广播地址、波特率修改或厂家特定指令可能无回显
|
||||||
|
if target_id == 0xFF or reg_addr == 0x0005:
|
||||||
|
time.sleep(0.5)
|
||||||
|
return True
|
||||||
|
|
||||||
|
# 等待返回 (正常应返回相同报文)
|
||||||
|
resp = self.transport.read(len(msg))
|
||||||
|
if self.debug:
|
||||||
|
self.logger.info(f"RX Write Response: {resp.hex().upper()}")
|
||||||
|
|
||||||
|
return resp == msg
|
||||||
|
|
||||||
|
def close(self):
|
||||||
|
if self.transport:
|
||||||
|
self.transport.close()
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
# 快速实例化测试
|
||||||
|
import logging
|
||||||
|
# 减少冗余日志,仅显示重要信息
|
||||||
|
logging.basicConfig(level=logging.INFO, format='%(levelname)s: %(message)s')
|
||||||
|
|
||||||
|
# 硬件配置 (根据实际情况修改)
|
||||||
|
TEST_PORT = "/dev/tty.usbserial-3110"
|
||||||
|
SLAVE_ID = 1
|
||||||
|
THRESHOLD = 300
|
||||||
|
|
||||||
|
print("\n" + "="*50)
|
||||||
|
print(f" XKC RS485 传感器独立测试程序")
|
||||||
|
print(f" 端口: {TEST_PORT} | 地址: {SLAVE_ID} | 阈值: {THRESHOLD}")
|
||||||
|
print("="*50)
|
||||||
|
|
||||||
|
sensor = XKCSensorDriver(port=TEST_PORT, device_id=SLAVE_ID, threshold=THRESHOLD, debug=False)
|
||||||
|
|
||||||
|
try:
|
||||||
|
if sensor.transport:
|
||||||
|
print(f"\n开始实时连续采样测试 (持续 15 秒)...")
|
||||||
|
print(f"按 Ctrl+C 可提前停止\n")
|
||||||
|
|
||||||
|
start_time = time.time()
|
||||||
|
duration = 15
|
||||||
|
count = 0
|
||||||
|
|
||||||
|
while time.time() - start_time < duration:
|
||||||
|
count += 1
|
||||||
|
res = sensor.read_level()
|
||||||
|
if res:
|
||||||
|
rssi = res['rssi']
|
||||||
|
level = res['level']
|
||||||
|
status_str = "【有液】" if level else "【无液】"
|
||||||
|
# 使用 \r 实现单行刷新显示 (或者不刷,直接打印历史)
|
||||||
|
# 为了方便查看变化,我们直接打印
|
||||||
|
elapsed = time.time() - start_time
|
||||||
|
print(f" [{elapsed:4.1f}s] 采样 {count:<3}: 电导率/RSSI = {rssi:<5} | 判定结果: {status_str}")
|
||||||
|
else:
|
||||||
|
print(f" [{time.time()-start_time:4.1f}s] 采样 {count:<3}: 通信失败 (无响应)")
|
||||||
|
|
||||||
|
time.sleep(0.5) # 每秒采样 2 次
|
||||||
|
|
||||||
|
print(f"\n--- 15 秒采样测试完成 (总计 {count} 次) ---")
|
||||||
|
|
||||||
|
# [3] 测试动态修改阈值
|
||||||
|
print(f"\n[3] 动态修改阈值演示...")
|
||||||
|
new_threshold = 400
|
||||||
|
sensor.set_threshold(new_threshold)
|
||||||
|
res = sensor.read_level()
|
||||||
|
if res:
|
||||||
|
print(f" 采样 (当前阈值={new_threshold}): 电导率/RSSI = {res['rssi']:<5} | 判定结果: {'【有液】' if res['level'] else '【无液】'}")
|
||||||
|
sensor.set_threshold(THRESHOLD) # 还原
|
||||||
|
|
||||||
|
except KeyboardInterrupt:
|
||||||
|
print("\n[!] 用户中断测试")
|
||||||
|
except Exception as e:
|
||||||
|
print(f"\n[!] 测试运行出错: {e}")
|
||||||
|
finally:
|
||||||
|
sensor.close()
|
||||||
|
print("\n--- 测试程序已退出 ---\n")
|
||||||
@@ -15,35 +15,35 @@ class VirtualPumpMode(Enum):
|
|||||||
|
|
||||||
class VirtualTransferPump:
|
class VirtualTransferPump:
|
||||||
"""虚拟转移泵类 - 模拟泵的基本功能,无需实际硬件 🚰"""
|
"""虚拟转移泵类 - 模拟泵的基本功能,无需实际硬件 🚰"""
|
||||||
|
|
||||||
_ros_node: BaseROS2DeviceNode
|
_ros_node: BaseROS2DeviceNode
|
||||||
|
|
||||||
def __init__(self, device_id: str = None, config: dict = None, **kwargs):
|
def __init__(self, device_id: str = None, config: dict = None, **kwargs):
|
||||||
"""
|
"""
|
||||||
初始化虚拟转移泵
|
初始化虚拟转移泵
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
device_id: 设备ID
|
device_id: 设备ID
|
||||||
config: 配置字典,包含max_volume, port等参数
|
config: 配置字典,包含max_volume, port等参数
|
||||||
**kwargs: 其他参数,确保兼容性
|
**kwargs: 其他参数,确保兼容性
|
||||||
"""
|
"""
|
||||||
self.device_id = device_id or "virtual_transfer_pump"
|
self.device_id = device_id or "virtual_transfer_pump"
|
||||||
|
|
||||||
# 从config或kwargs中获取参数,确保类型正确
|
# 从config或kwargs中获取参数,确保类型正确
|
||||||
if config:
|
if config:
|
||||||
self.max_volume = float(config.get("max_volume", 25.0))
|
self.max_volume = float(config.get('max_volume', 25.0))
|
||||||
self.port = config.get("port", "VIRTUAL")
|
self.port = config.get('port', 'VIRTUAL')
|
||||||
else:
|
else:
|
||||||
self.max_volume = float(kwargs.get("max_volume", 25.0))
|
self.max_volume = float(kwargs.get('max_volume', 25.0))
|
||||||
self.port = kwargs.get("port", "VIRTUAL")
|
self.port = kwargs.get('port', 'VIRTUAL')
|
||||||
|
|
||||||
self._transfer_rate = float(kwargs.get("transfer_rate", 0))
|
self._transfer_rate = float(kwargs.get('transfer_rate', 0))
|
||||||
self.mode = kwargs.get("mode", VirtualPumpMode.Normal)
|
self.mode = kwargs.get('mode', VirtualPumpMode.Normal)
|
||||||
|
|
||||||
# 状态变量 - 确保都是正确类型
|
# 状态变量 - 确保都是正确类型
|
||||||
self._status = "Idle"
|
self._status = "Idle"
|
||||||
self._position = 0.0 # float
|
self._position = 0.0 # float
|
||||||
self._max_velocity = 5.0 # float
|
self._max_velocity = 5.0 # float
|
||||||
self._current_volume = 0.0 # float
|
self._current_volume = 0.0 # float
|
||||||
|
|
||||||
# 🚀 新增:快速模式设置 - 大幅缩短执行时间
|
# 🚀 新增:快速模式设置 - 大幅缩短执行时间
|
||||||
@@ -52,16 +52,14 @@ class VirtualTransferPump:
|
|||||||
self._fast_dispense_time = 1.0 # 快速喷射时间(秒)
|
self._fast_dispense_time = 1.0 # 快速喷射时间(秒)
|
||||||
|
|
||||||
self.logger = logging.getLogger(f"VirtualTransferPump.{self.device_id}")
|
self.logger = logging.getLogger(f"VirtualTransferPump.{self.device_id}")
|
||||||
|
|
||||||
print(f"🚰 === 虚拟转移泵 {self.device_id} 已创建 === ✨")
|
print(f"🚰 === 虚拟转移泵 {self.device_id} 已创建 === ✨")
|
||||||
print(
|
print(f"💨 快速模式: {'启用' if self._fast_mode else '禁用'} | 移动时间: {self._fast_move_time}s | 喷射时间: {self._fast_dispense_time}s")
|
||||||
f"💨 快速模式: {'启用' if self._fast_mode else '禁用'} | 移动时间: {self._fast_move_time}s | 喷射时间: {self._fast_dispense_time}s"
|
|
||||||
)
|
|
||||||
print(f"📊 最大容量: {self.max_volume}mL | 端口: {self.port}")
|
print(f"📊 最大容量: {self.max_volume}mL | 端口: {self.port}")
|
||||||
|
|
||||||
def post_init(self, ros_node: BaseROS2DeviceNode):
|
def post_init(self, ros_node: BaseROS2DeviceNode):
|
||||||
self._ros_node = ros_node
|
self._ros_node = ros_node
|
||||||
|
|
||||||
async def initialize(self) -> bool:
|
async def initialize(self) -> bool:
|
||||||
"""初始化虚拟泵 🚀"""
|
"""初始化虚拟泵 🚀"""
|
||||||
self.logger.info(f"🔧 初始化虚拟转移泵 {self.device_id} ✨")
|
self.logger.info(f"🔧 初始化虚拟转移泵 {self.device_id} ✨")
|
||||||
@@ -70,33 +68,33 @@ class VirtualTransferPump:
|
|||||||
self._current_volume = 0.0
|
self._current_volume = 0.0
|
||||||
self.logger.info(f"✅ 转移泵 {self.device_id} 初始化完成 🚰")
|
self.logger.info(f"✅ 转移泵 {self.device_id} 初始化完成 🚰")
|
||||||
return True
|
return True
|
||||||
|
|
||||||
async def cleanup(self) -> bool:
|
async def cleanup(self) -> bool:
|
||||||
"""清理虚拟泵 🧹"""
|
"""清理虚拟泵 🧹"""
|
||||||
self.logger.info(f"🧹 清理虚拟转移泵 {self.device_id} 🔚")
|
self.logger.info(f"🧹 清理虚拟转移泵 {self.device_id} 🔚")
|
||||||
self._status = "Idle"
|
self._status = "Idle"
|
||||||
self.logger.info(f"✅ 转移泵 {self.device_id} 清理完成 💤")
|
self.logger.info(f"✅ 转移泵 {self.device_id} 清理完成 💤")
|
||||||
return True
|
return True
|
||||||
|
|
||||||
# 基本属性
|
# 基本属性
|
||||||
@property
|
@property
|
||||||
def status(self) -> str:
|
def status(self) -> str:
|
||||||
return self._status
|
return self._status
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def position(self) -> float:
|
def position(self) -> float:
|
||||||
"""当前柱塞位置 (ml) 📍"""
|
"""当前柱塞位置 (ml) 📍"""
|
||||||
return self._position
|
return self._position
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def current_volume(self) -> float:
|
def current_volume(self) -> float:
|
||||||
"""当前注射器中的体积 (ml) 💧"""
|
"""当前注射器中的体积 (ml) 💧"""
|
||||||
return self._current_volume
|
return self._current_volume
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def max_velocity(self) -> float:
|
def max_velocity(self) -> float:
|
||||||
return self._max_velocity
|
return self._max_velocity
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def transfer_rate(self) -> float:
|
def transfer_rate(self) -> float:
|
||||||
return self._transfer_rate
|
return self._transfer_rate
|
||||||
@@ -105,17 +103,17 @@ class VirtualTransferPump:
|
|||||||
"""设置最大速度 (ml/s) 🌊"""
|
"""设置最大速度 (ml/s) 🌊"""
|
||||||
self._max_velocity = max(0.1, min(50.0, velocity)) # 限制在合理范围内
|
self._max_velocity = max(0.1, min(50.0, velocity)) # 限制在合理范围内
|
||||||
self.logger.info(f"🌊 设置最大速度为 {self._max_velocity} mL/s")
|
self.logger.info(f"🌊 设置最大速度为 {self._max_velocity} mL/s")
|
||||||
|
|
||||||
def get_status(self) -> str:
|
def get_status(self) -> str:
|
||||||
"""获取泵状态 📋"""
|
"""获取泵状态 📋"""
|
||||||
return self._status
|
return self._status
|
||||||
|
|
||||||
async def _simulate_operation(self, duration: float):
|
async def _simulate_operation(self, duration: float):
|
||||||
"""模拟操作延时 ⏱️"""
|
"""模拟操作延时 ⏱️"""
|
||||||
self._status = "Busy"
|
self._status = "Busy"
|
||||||
await self._ros_node.sleep(duration)
|
await self._ros_node.sleep(duration)
|
||||||
self._status = "Idle"
|
self._status = "Idle"
|
||||||
|
|
||||||
def _calculate_duration(self, volume: float, velocity: float = None) -> float:
|
def _calculate_duration(self, volume: float, velocity: float = None) -> float:
|
||||||
"""
|
"""
|
||||||
计算操作持续时间 ⏰
|
计算操作持续时间 ⏰
|
||||||
@@ -123,10 +121,10 @@ class VirtualTransferPump:
|
|||||||
"""
|
"""
|
||||||
if velocity is None:
|
if velocity is None:
|
||||||
velocity = self._max_velocity
|
velocity = self._max_velocity
|
||||||
|
|
||||||
# 📊 计算理论时间(用于日志显示)
|
# 📊 计算理论时间(用于日志显示)
|
||||||
theoretical_duration = abs(volume) / velocity
|
theoretical_duration = abs(volume) / velocity
|
||||||
|
|
||||||
# 🚀 如果启用快速模式,使用固定的快速时间
|
# 🚀 如果启用快速模式,使用固定的快速时间
|
||||||
if self._fast_mode:
|
if self._fast_mode:
|
||||||
# 根据操作类型选择快速时间
|
# 根据操作类型选择快速时间
|
||||||
@@ -134,13 +132,13 @@ class VirtualTransferPump:
|
|||||||
actual_duration = self._fast_move_time
|
actual_duration = self._fast_move_time
|
||||||
else: # 很小的操作
|
else: # 很小的操作
|
||||||
actual_duration = 0.5
|
actual_duration = 0.5
|
||||||
|
|
||||||
self.logger.debug(f"⚡ 快速模式: 理论时间 {theoretical_duration:.2f}s → 实际时间 {actual_duration:.2f}s")
|
self.logger.debug(f"⚡ 快速模式: 理论时间 {theoretical_duration:.2f}s → 实际时间 {actual_duration:.2f}s")
|
||||||
return actual_duration
|
return actual_duration
|
||||||
else:
|
else:
|
||||||
# 正常模式使用理论时间
|
# 正常模式使用理论时间
|
||||||
return theoretical_duration
|
return theoretical_duration
|
||||||
|
|
||||||
def _calculate_display_duration(self, volume: float, velocity: float = None) -> float:
|
def _calculate_display_duration(self, volume: float, velocity: float = None) -> float:
|
||||||
"""
|
"""
|
||||||
计算显示用的持续时间(用于日志) 📊
|
计算显示用的持续时间(用于日志) 📊
|
||||||
@@ -149,16 +147,16 @@ class VirtualTransferPump:
|
|||||||
if velocity is None:
|
if velocity is None:
|
||||||
velocity = self._max_velocity
|
velocity = self._max_velocity
|
||||||
return abs(volume) / velocity
|
return abs(volume) / velocity
|
||||||
|
|
||||||
# 新的set_position方法 - 专门用于SetPumpPosition动作
|
# 新的set_position方法 - 专门用于SetPumpPosition动作
|
||||||
async def set_position(self, position: float, max_velocity: float = None):
|
async def set_position(self, position: float, max_velocity: float = None):
|
||||||
"""
|
"""
|
||||||
移动到绝对位置 - 专门用于SetPumpPosition动作 🎯
|
移动到绝对位置 - 专门用于SetPumpPosition动作 🎯
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
position (float): 目标位置 (ml)
|
position (float): 目标位置 (ml)
|
||||||
max_velocity (float): 移动速度 (ml/s)
|
max_velocity (float): 移动速度 (ml/s)
|
||||||
|
|
||||||
Returns:
|
Returns:
|
||||||
dict: 符合SetPumpPosition.action定义的结果
|
dict: 符合SetPumpPosition.action定义的结果
|
||||||
"""
|
"""
|
||||||
@@ -166,19 +164,19 @@ class VirtualTransferPump:
|
|||||||
# 验证并转换参数
|
# 验证并转换参数
|
||||||
target_position = float(position)
|
target_position = float(position)
|
||||||
velocity = float(max_velocity) if max_velocity is not None else self._max_velocity
|
velocity = float(max_velocity) if max_velocity is not None else self._max_velocity
|
||||||
|
|
||||||
# 限制位置在有效范围内
|
# 限制位置在有效范围内
|
||||||
target_position = max(0.0, min(float(self.max_volume), target_position))
|
target_position = max(0.0, min(float(self.max_volume), target_position))
|
||||||
|
|
||||||
# 计算移动距离
|
# 计算移动距离
|
||||||
volume_to_move = abs(target_position - self._position)
|
volume_to_move = abs(target_position - self._position)
|
||||||
|
|
||||||
# 📊 计算显示用的时间(用于日志)
|
# 📊 计算显示用的时间(用于日志)
|
||||||
display_duration = self._calculate_display_duration(volume_to_move, velocity)
|
display_duration = self._calculate_display_duration(volume_to_move, velocity)
|
||||||
|
|
||||||
# ⚡ 计算实际执行时间(快速模式)
|
# ⚡ 计算实际执行时间(快速模式)
|
||||||
actual_duration = self._calculate_duration(volume_to_move, velocity)
|
actual_duration = self._calculate_duration(volume_to_move, velocity)
|
||||||
|
|
||||||
# 🎯 确定操作类型和emoji
|
# 🎯 确定操作类型和emoji
|
||||||
if target_position > self._position:
|
if target_position > self._position:
|
||||||
operation_type = "吸液"
|
operation_type = "吸液"
|
||||||
@@ -189,34 +187,28 @@ class VirtualTransferPump:
|
|||||||
else:
|
else:
|
||||||
operation_type = "保持"
|
operation_type = "保持"
|
||||||
operation_emoji = "📍"
|
operation_emoji = "📍"
|
||||||
|
|
||||||
self.logger.info(f"🎯 SET_POSITION: {operation_type} {operation_emoji}")
|
self.logger.info(f"🎯 SET_POSITION: {operation_type} {operation_emoji}")
|
||||||
self.logger.info(
|
self.logger.info(f" 📍 位置: {self._position:.2f}mL → {target_position:.2f}mL (移动 {volume_to_move:.2f}mL)")
|
||||||
f" 📍 位置: {self._position:.2f}mL → {target_position:.2f}mL (移动 {volume_to_move:.2f}mL)"
|
|
||||||
)
|
|
||||||
self.logger.info(f" 🌊 速度: {velocity:.2f} mL/s")
|
self.logger.info(f" 🌊 速度: {velocity:.2f} mL/s")
|
||||||
self.logger.info(f" ⏰ 预计时间: {display_duration:.2f}s")
|
self.logger.info(f" ⏰ 预计时间: {display_duration:.2f}s")
|
||||||
|
|
||||||
if self._fast_mode:
|
if self._fast_mode:
|
||||||
self.logger.info(f" ⚡ 快速模式: 实际用时 {actual_duration:.2f}s")
|
self.logger.info(f" ⚡ 快速模式: 实际用时 {actual_duration:.2f}s")
|
||||||
|
|
||||||
# 🚀 模拟移动过程
|
# 🚀 模拟移动过程
|
||||||
if volume_to_move > 0.01: # 只有当移动距离足够大时才显示进度
|
if volume_to_move > 0.01: # 只有当移动距离足够大时才显示进度
|
||||||
start_position = self._position
|
start_position = self._position
|
||||||
steps = 5 if actual_duration > 0.5 else 2 # 根据实际时间调整步数
|
steps = 5 if actual_duration > 0.5 else 2 # 根据实际时间调整步数
|
||||||
step_duration = actual_duration / steps
|
step_duration = actual_duration / steps
|
||||||
|
|
||||||
self.logger.info(f"🚀 开始{operation_type}... {operation_emoji}")
|
self.logger.info(f"🚀 开始{operation_type}... {operation_emoji}")
|
||||||
|
|
||||||
for i in range(steps + 1):
|
for i in range(steps + 1):
|
||||||
# 计算当前位置和进度
|
# 计算当前位置和进度
|
||||||
progress = (i / steps) * 100 if steps > 0 else 100
|
progress = (i / steps) * 100 if steps > 0 else 100
|
||||||
current_pos = (
|
current_pos = start_position + (target_position - start_position) * (i / steps) if steps > 0 else target_position
|
||||||
start_position + (target_position - start_position) * (i / steps)
|
|
||||||
if steps > 0
|
|
||||||
else target_position
|
|
||||||
)
|
|
||||||
|
|
||||||
# 更新状态
|
# 更新状态
|
||||||
if i < steps:
|
if i < steps:
|
||||||
self._status = f"{operation_type}中"
|
self._status = f"{operation_type}中"
|
||||||
@@ -224,10 +216,10 @@ class VirtualTransferPump:
|
|||||||
else:
|
else:
|
||||||
self._status = "Idle"
|
self._status = "Idle"
|
||||||
status_emoji = "✅"
|
status_emoji = "✅"
|
||||||
|
|
||||||
self._position = current_pos
|
self._position = current_pos
|
||||||
self._current_volume = current_pos
|
self._current_volume = current_pos
|
||||||
|
|
||||||
# 显示进度(每25%或最后一步)
|
# 显示进度(每25%或最后一步)
|
||||||
if i == 0:
|
if i == 0:
|
||||||
self.logger.debug(f" 🔄 {operation_type}开始: {progress:.0f}%")
|
self.logger.debug(f" 🔄 {operation_type}开始: {progress:.0f}%")
|
||||||
@@ -235,7 +227,7 @@ class VirtualTransferPump:
|
|||||||
self.logger.debug(f" 🔄 {operation_type}进度: {progress:.0f}%")
|
self.logger.debug(f" 🔄 {operation_type}进度: {progress:.0f}%")
|
||||||
elif i == steps:
|
elif i == steps:
|
||||||
self.logger.info(f" ✅ {operation_type}完成: {progress:.0f}% | 当前位置: {current_pos:.2f}mL")
|
self.logger.info(f" ✅ {operation_type}完成: {progress:.0f}% | 当前位置: {current_pos:.2f}mL")
|
||||||
|
|
||||||
# 等待一小步时间
|
# 等待一小步时间
|
||||||
if i < steps and step_duration > 0:
|
if i < steps and step_duration > 0:
|
||||||
await self._ros_node.sleep(step_duration)
|
await self._ros_node.sleep(step_duration)
|
||||||
@@ -244,27 +236,25 @@ class VirtualTransferPump:
|
|||||||
self._position = target_position
|
self._position = target_position
|
||||||
self._current_volume = target_position
|
self._current_volume = target_position
|
||||||
self.logger.info(f" 📍 微调完成: {target_position:.2f}mL")
|
self.logger.info(f" 📍 微调完成: {target_position:.2f}mL")
|
||||||
|
|
||||||
# 确保最终位置准确
|
# 确保最终位置准确
|
||||||
self._position = target_position
|
self._position = target_position
|
||||||
self._current_volume = target_position
|
self._current_volume = target_position
|
||||||
self._status = "Idle"
|
self._status = "Idle"
|
||||||
|
|
||||||
# 📊 最终状态日志
|
# 📊 最终状态日志
|
||||||
if volume_to_move > 0.01:
|
if volume_to_move > 0.01:
|
||||||
self.logger.info(
|
self.logger.info(f"🎉 SET_POSITION 完成! 📍 最终位置: {self._position:.2f}mL | 💧 当前体积: {self._current_volume:.2f}mL")
|
||||||
f"🎉 SET_POSITION 完成! 📍 最终位置: {self._position:.2f}mL | 💧 当前体积: {self._current_volume:.2f}mL"
|
|
||||||
)
|
|
||||||
|
|
||||||
# 返回符合action定义的结果
|
# 返回符合action定义的结果
|
||||||
return {
|
return {
|
||||||
"success": True,
|
"success": True,
|
||||||
"message": f"✅ 成功移动到位置 {self._position:.2f}mL ({operation_type})",
|
"message": f"✅ 成功移动到位置 {self._position:.2f}mL ({operation_type})",
|
||||||
"final_position": self._position,
|
"final_position": self._position,
|
||||||
"final_volume": self._current_volume,
|
"final_volume": self._current_volume,
|
||||||
"operation_type": operation_type,
|
"operation_type": operation_type
|
||||||
}
|
}
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
error_msg = f"❌ 设置位置失败: {str(e)}"
|
error_msg = f"❌ 设置位置失败: {str(e)}"
|
||||||
self.logger.error(error_msg)
|
self.logger.error(error_msg)
|
||||||
@@ -272,136 +262,134 @@ class VirtualTransferPump:
|
|||||||
"success": False,
|
"success": False,
|
||||||
"message": error_msg,
|
"message": error_msg,
|
||||||
"final_position": self._position,
|
"final_position": self._position,
|
||||||
"final_volume": self._current_volume,
|
"final_volume": self._current_volume
|
||||||
}
|
}
|
||||||
|
|
||||||
# 其他泵操作方法
|
# 其他泵操作方法
|
||||||
async def pull_plunger(self, volume: float, velocity: float = None):
|
async def pull_plunger(self, volume: float, velocity: float = None):
|
||||||
"""
|
"""
|
||||||
拉取柱塞(吸液) 📥
|
拉取柱塞(吸液) 📥
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
volume (float): 要拉取的体积 (ml)
|
volume (float): 要拉取的体积 (ml)
|
||||||
velocity (float): 拉取速度 (ml/s)
|
velocity (float): 拉取速度 (ml/s)
|
||||||
"""
|
"""
|
||||||
new_position = min(self.max_volume, self._position + volume)
|
new_position = min(self.max_volume, self._position + volume)
|
||||||
actual_volume = new_position - self._position
|
actual_volume = new_position - self._position
|
||||||
|
|
||||||
if actual_volume <= 0:
|
if actual_volume <= 0:
|
||||||
self.logger.warning("⚠️ 无法吸液 - 已达到最大容量")
|
self.logger.warning("⚠️ 无法吸液 - 已达到最大容量")
|
||||||
return
|
return
|
||||||
|
|
||||||
display_duration = self._calculate_display_duration(actual_volume, velocity)
|
display_duration = self._calculate_display_duration(actual_volume, velocity)
|
||||||
actual_duration = self._calculate_duration(actual_volume, velocity)
|
actual_duration = self._calculate_duration(actual_volume, velocity)
|
||||||
|
|
||||||
self.logger.info(f"📥 开始吸液: {actual_volume:.2f}mL")
|
self.logger.info(f"📥 开始吸液: {actual_volume:.2f}mL")
|
||||||
self.logger.info(f" 📍 位置: {self._position:.2f}mL → {new_position:.2f}mL")
|
self.logger.info(f" 📍 位置: {self._position:.2f}mL → {new_position:.2f}mL")
|
||||||
self.logger.info(f" ⏰ 预计时间: {display_duration:.2f}s")
|
self.logger.info(f" ⏰ 预计时间: {display_duration:.2f}s")
|
||||||
|
|
||||||
if self._fast_mode:
|
if self._fast_mode:
|
||||||
self.logger.info(f" ⚡ 快速模式: 实际用时 {actual_duration:.2f}s")
|
self.logger.info(f" ⚡ 快速模式: 实际用时 {actual_duration:.2f}s")
|
||||||
|
|
||||||
await self._simulate_operation(actual_duration)
|
await self._simulate_operation(actual_duration)
|
||||||
|
|
||||||
self._position = new_position
|
self._position = new_position
|
||||||
self._current_volume = new_position
|
self._current_volume = new_position
|
||||||
|
|
||||||
self.logger.info(f"✅ 吸液完成: {actual_volume:.2f}mL | 💧 当前体积: {self._current_volume:.2f}mL")
|
self.logger.info(f"✅ 吸液完成: {actual_volume:.2f}mL | 💧 当前体积: {self._current_volume:.2f}mL")
|
||||||
|
|
||||||
async def push_plunger(self, volume: float, velocity: float = None):
|
async def push_plunger(self, volume: float, velocity: float = None):
|
||||||
"""
|
"""
|
||||||
推出柱塞(排液) 📤
|
推出柱塞(排液) 📤
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
volume (float): 要推出的体积 (ml)
|
volume (float): 要推出的体积 (ml)
|
||||||
velocity (float): 推出速度 (ml/s)
|
velocity (float): 推出速度 (ml/s)
|
||||||
"""
|
"""
|
||||||
new_position = max(0, self._position - volume)
|
new_position = max(0, self._position - volume)
|
||||||
actual_volume = self._position - new_position
|
actual_volume = self._position - new_position
|
||||||
|
|
||||||
if actual_volume <= 0:
|
if actual_volume <= 0:
|
||||||
self.logger.warning("⚠️ 无法排液 - 已达到最小容量")
|
self.logger.warning("⚠️ 无法排液 - 已达到最小容量")
|
||||||
return
|
return
|
||||||
|
|
||||||
display_duration = self._calculate_display_duration(actual_volume, velocity)
|
display_duration = self._calculate_display_duration(actual_volume, velocity)
|
||||||
actual_duration = self._calculate_duration(actual_volume, velocity)
|
actual_duration = self._calculate_duration(actual_volume, velocity)
|
||||||
|
|
||||||
self.logger.info(f"📤 开始排液: {actual_volume:.2f}mL")
|
self.logger.info(f"📤 开始排液: {actual_volume:.2f}mL")
|
||||||
self.logger.info(f" 📍 位置: {self._position:.2f}mL → {new_position:.2f}mL")
|
self.logger.info(f" 📍 位置: {self._position:.2f}mL → {new_position:.2f}mL")
|
||||||
self.logger.info(f" ⏰ 预计时间: {display_duration:.2f}s")
|
self.logger.info(f" ⏰ 预计时间: {display_duration:.2f}s")
|
||||||
|
|
||||||
if self._fast_mode:
|
if self._fast_mode:
|
||||||
self.logger.info(f" ⚡ 快速模式: 实际用时 {actual_duration:.2f}s")
|
self.logger.info(f" ⚡ 快速模式: 实际用时 {actual_duration:.2f}s")
|
||||||
|
|
||||||
await self._simulate_operation(actual_duration)
|
await self._simulate_operation(actual_duration)
|
||||||
|
|
||||||
self._position = new_position
|
self._position = new_position
|
||||||
self._current_volume = new_position
|
self._current_volume = new_position
|
||||||
|
|
||||||
self.logger.info(f"✅ 排液完成: {actual_volume:.2f}mL | 💧 当前体积: {self._current_volume:.2f}mL")
|
self.logger.info(f"✅ 排液完成: {actual_volume:.2f}mL | 💧 当前体积: {self._current_volume:.2f}mL")
|
||||||
|
|
||||||
# 便捷操作方法
|
# 便捷操作方法
|
||||||
async def aspirate(self, volume: float, velocity: float = None):
|
async def aspirate(self, volume: float, velocity: float = None):
|
||||||
"""吸液操作 📥"""
|
"""吸液操作 📥"""
|
||||||
await self.pull_plunger(volume, velocity)
|
await self.pull_plunger(volume, velocity)
|
||||||
|
|
||||||
async def dispense(self, volume: float, velocity: float = None):
|
async def dispense(self, volume: float, velocity: float = None):
|
||||||
"""排液操作 📤"""
|
"""排液操作 📤"""
|
||||||
await self.push_plunger(volume, velocity)
|
await self.push_plunger(volume, velocity)
|
||||||
|
|
||||||
async def transfer(self, volume: float, aspirate_velocity: float = None, dispense_velocity: float = None):
|
async def transfer(self, volume: float, aspirate_velocity: float = None, dispense_velocity: float = None):
|
||||||
"""转移操作(先吸后排) 🔄"""
|
"""转移操作(先吸后排) 🔄"""
|
||||||
self.logger.info(f"🔄 开始转移操作: {volume:.2f}mL")
|
self.logger.info(f"🔄 开始转移操作: {volume:.2f}mL")
|
||||||
|
|
||||||
# 吸液
|
# 吸液
|
||||||
await self.aspirate(volume, aspirate_velocity)
|
await self.aspirate(volume, aspirate_velocity)
|
||||||
|
|
||||||
# 短暂停顿
|
# 短暂停顿
|
||||||
self.logger.debug("⏸️ 短暂停顿...")
|
self.logger.debug("⏸️ 短暂停顿...")
|
||||||
await self._ros_node.sleep(0.1)
|
await self._ros_node.sleep(0.1)
|
||||||
|
|
||||||
# 排液
|
# 排液
|
||||||
await self.dispense(volume, dispense_velocity)
|
await self.dispense(volume, dispense_velocity)
|
||||||
|
|
||||||
async def empty_syringe(self, velocity: float = None):
|
async def empty_syringe(self, velocity: float = None):
|
||||||
"""清空注射器"""
|
"""清空注射器"""
|
||||||
await self.set_position(0, velocity)
|
await self.set_position(0, velocity)
|
||||||
|
|
||||||
async def fill_syringe(self, velocity: float = None):
|
async def fill_syringe(self, velocity: float = None):
|
||||||
"""充满注射器"""
|
"""充满注射器"""
|
||||||
await self.set_position(self.max_volume, velocity)
|
await self.set_position(self.max_volume, velocity)
|
||||||
|
|
||||||
async def stop_operation(self):
|
async def stop_operation(self):
|
||||||
"""停止当前操作"""
|
"""停止当前操作"""
|
||||||
self._status = "Idle"
|
self._status = "Idle"
|
||||||
self.logger.info("Operation stopped")
|
self.logger.info("Operation stopped")
|
||||||
|
|
||||||
# 状态查询方法
|
# 状态查询方法
|
||||||
def get_position(self) -> float:
|
def get_position(self) -> float:
|
||||||
"""获取当前位置"""
|
"""获取当前位置"""
|
||||||
return self._position
|
return self._position
|
||||||
|
|
||||||
def get_current_volume(self) -> float:
|
def get_current_volume(self) -> float:
|
||||||
"""获取当前体积"""
|
"""获取当前体积"""
|
||||||
return self._current_volume
|
return self._current_volume
|
||||||
|
|
||||||
def get_remaining_capacity(self) -> float:
|
def get_remaining_capacity(self) -> float:
|
||||||
"""获取剩余容量"""
|
"""获取剩余容量"""
|
||||||
return self.max_volume - self._current_volume
|
return self.max_volume - self._current_volume
|
||||||
|
|
||||||
def is_empty(self) -> bool:
|
def is_empty(self) -> bool:
|
||||||
"""检查是否为空"""
|
"""检查是否为空"""
|
||||||
return self._current_volume <= 0.01 # 允许小量误差
|
return self._current_volume <= 0.01 # 允许小量误差
|
||||||
|
|
||||||
def is_full(self) -> bool:
|
def is_full(self) -> bool:
|
||||||
"""检查是否已满"""
|
"""检查是否已满"""
|
||||||
return self._current_volume >= (self.max_volume - 0.01) # 允许小量误差
|
return self._current_volume >= (self.max_volume - 0.01) # 允许小量误差
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
return (
|
return f"VirtualTransferPump({self.device_id}: {self._current_volume:.2f}/{self.max_volume} ml, {self._status})"
|
||||||
f"VirtualTransferPump({self.device_id}: {self._current_volume:.2f}/{self.max_volume} ml, {self._status})"
|
|
||||||
)
|
|
||||||
|
|
||||||
def __repr__(self):
|
def __repr__(self):
|
||||||
return self.__str__()
|
return self.__str__()
|
||||||
|
|
||||||
@@ -410,20 +398,20 @@ class VirtualTransferPump:
|
|||||||
async def demo():
|
async def demo():
|
||||||
"""虚拟泵使用示例"""
|
"""虚拟泵使用示例"""
|
||||||
pump = VirtualTransferPump("demo_pump", {"max_volume": 50.0})
|
pump = VirtualTransferPump("demo_pump", {"max_volume": 50.0})
|
||||||
|
|
||||||
await pump.initialize()
|
await pump.initialize()
|
||||||
|
|
||||||
print(f"Initial state: {pump}")
|
print(f"Initial state: {pump}")
|
||||||
|
|
||||||
# 测试set_position方法
|
# 测试set_position方法
|
||||||
result = await pump.set_position(10.0, max_velocity=2.0)
|
result = await pump.set_position(10.0, max_velocity=2.0)
|
||||||
print(f"Set position result: {result}")
|
print(f"Set position result: {result}")
|
||||||
print(f"After setting position to 10ml: {pump}")
|
print(f"After setting position to 10ml: {pump}")
|
||||||
|
|
||||||
# 吸液测试
|
# 吸液测试
|
||||||
await pump.aspirate(5.0, velocity=2.0)
|
await pump.aspirate(5.0, velocity=2.0)
|
||||||
print(f"After aspirating 5ml: {pump}")
|
print(f"After aspirating 5ml: {pump}")
|
||||||
|
|
||||||
# 清空测试
|
# 清空测试
|
||||||
result = await pump.set_position(0.0)
|
result = await pump.set_position(0.0)
|
||||||
print(f"Empty result: {result}")
|
print(f"Empty result: {result}")
|
||||||
|
|||||||
@@ -11,10 +11,9 @@ Virtual Workbench Device - 模拟工作台设备
|
|||||||
|
|
||||||
注意:调用来自线程池,使用 threading.Lock 进行同步
|
注意:调用来自线程池,使用 threading.Lock 进行同步
|
||||||
"""
|
"""
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
import time
|
import time
|
||||||
from typing import Dict, Any, Optional, List
|
from typing import Dict, Any, Optional
|
||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
from enum import Enum
|
from enum import Enum
|
||||||
from threading import Lock, RLock
|
from threading import Lock, RLock
|
||||||
@@ -23,46 +22,37 @@ from typing_extensions import TypedDict
|
|||||||
|
|
||||||
from unilabos.ros.nodes.base_device_node import BaseROS2DeviceNode
|
from unilabos.ros.nodes.base_device_node import BaseROS2DeviceNode
|
||||||
from unilabos.utils.decorator import not_action
|
from unilabos.utils.decorator import not_action
|
||||||
from unilabos.resources.resource_tracker import SampleUUIDsType, LabSample, RETURN_UNILABOS_SAMPLES
|
|
||||||
|
|
||||||
|
|
||||||
# ============ TypedDict 返回类型定义 ============
|
# ============ TypedDict 返回类型定义 ============
|
||||||
|
|
||||||
|
|
||||||
class MoveToHeatingStationResult(TypedDict):
|
class MoveToHeatingStationResult(TypedDict):
|
||||||
"""move_to_heating_station 返回类型"""
|
"""move_to_heating_station 返回类型"""
|
||||||
|
|
||||||
success: bool
|
success: bool
|
||||||
station_id: int
|
station_id: int
|
||||||
material_id: str
|
material_id: str
|
||||||
material_number: int
|
material_number: int
|
||||||
message: str
|
message: str
|
||||||
unilabos_samples: List[LabSample]
|
|
||||||
|
|
||||||
|
|
||||||
class StartHeatingResult(TypedDict):
|
class StartHeatingResult(TypedDict):
|
||||||
"""start_heating 返回类型"""
|
"""start_heating 返回类型"""
|
||||||
|
|
||||||
success: bool
|
success: bool
|
||||||
station_id: int
|
station_id: int
|
||||||
material_id: str
|
material_id: str
|
||||||
material_number: int
|
material_number: int
|
||||||
message: str
|
message: str
|
||||||
unilabos_samples: List[LabSample]
|
|
||||||
|
|
||||||
|
|
||||||
class MoveToOutputResult(TypedDict):
|
class MoveToOutputResult(TypedDict):
|
||||||
"""move_to_output 返回类型"""
|
"""move_to_output 返回类型"""
|
||||||
|
|
||||||
success: bool
|
success: bool
|
||||||
station_id: int
|
station_id: int
|
||||||
material_id: str
|
material_id: str
|
||||||
unilabos_samples: List[LabSample]
|
|
||||||
|
|
||||||
|
|
||||||
class PrepareMaterialsResult(TypedDict):
|
class PrepareMaterialsResult(TypedDict):
|
||||||
"""prepare_materials 返回类型 - 批量准备物料"""
|
"""prepare_materials 返回类型 - 批量准备物料"""
|
||||||
|
|
||||||
success: bool
|
success: bool
|
||||||
count: int
|
count: int
|
||||||
material_1: int # 物料编号1
|
material_1: int # 物料编号1
|
||||||
@@ -71,15 +61,12 @@ class PrepareMaterialsResult(TypedDict):
|
|||||||
material_4: int # 物料编号4
|
material_4: int # 物料编号4
|
||||||
material_5: int # 物料编号5
|
material_5: int # 物料编号5
|
||||||
message: str
|
message: str
|
||||||
unilabos_samples: List[LabSample]
|
|
||||||
|
|
||||||
|
|
||||||
# ============ 状态枚举 ============
|
# ============ 状态枚举 ============
|
||||||
|
|
||||||
|
|
||||||
class HeatingStationState(Enum):
|
class HeatingStationState(Enum):
|
||||||
"""加热台状态枚举"""
|
"""加热台状态枚举"""
|
||||||
|
|
||||||
IDLE = "idle" # 空闲
|
IDLE = "idle" # 空闲
|
||||||
OCCUPIED = "occupied" # 已放置物料,等待加热
|
OCCUPIED = "occupied" # 已放置物料,等待加热
|
||||||
HEATING = "heating" # 加热中
|
HEATING = "heating" # 加热中
|
||||||
@@ -88,7 +75,6 @@ class HeatingStationState(Enum):
|
|||||||
|
|
||||||
class ArmState(Enum):
|
class ArmState(Enum):
|
||||||
"""机械臂状态枚举"""
|
"""机械臂状态枚举"""
|
||||||
|
|
||||||
IDLE = "idle" # 空闲
|
IDLE = "idle" # 空闲
|
||||||
BUSY = "busy" # 工作中
|
BUSY = "busy" # 工作中
|
||||||
|
|
||||||
@@ -96,7 +82,6 @@ class ArmState(Enum):
|
|||||||
@dataclass
|
@dataclass
|
||||||
class HeatingStation:
|
class HeatingStation:
|
||||||
"""加热台数据结构"""
|
"""加热台数据结构"""
|
||||||
|
|
||||||
station_id: int
|
station_id: int
|
||||||
state: HeatingStationState = HeatingStationState.IDLE
|
state: HeatingStationState = HeatingStationState.IDLE
|
||||||
current_material: Optional[str] = None # 当前物料 (如 "A1", "A2")
|
current_material: Optional[str] = None # 当前物料 (如 "A1", "A2")
|
||||||
@@ -152,7 +137,8 @@ class VirtualWorkbench:
|
|||||||
|
|
||||||
# 加热台状态 (station_id -> HeatingStation) - 立即初始化,不依赖initialize()
|
# 加热台状态 (station_id -> HeatingStation) - 立即初始化,不依赖initialize()
|
||||||
self._heating_stations: Dict[int, HeatingStation] = {
|
self._heating_stations: Dict[int, HeatingStation] = {
|
||||||
i: HeatingStation(station_id=i) for i in range(1, self.NUM_HEATING_STATIONS + 1)
|
i: HeatingStation(station_id=i)
|
||||||
|
for i in range(1, self.NUM_HEATING_STATIONS + 1)
|
||||||
}
|
}
|
||||||
self._stations_lock = RLock() # 可重入锁,保护加热台状态
|
self._stations_lock = RLock() # 可重入锁,保护加热台状态
|
||||||
|
|
||||||
@@ -192,16 +178,14 @@ class VirtualWorkbench:
|
|||||||
station.heating_progress = 0.0
|
station.heating_progress = 0.0
|
||||||
|
|
||||||
# 初始化状态
|
# 初始化状态
|
||||||
self.data.update(
|
self.data.update({
|
||||||
{
|
"status": "Ready",
|
||||||
"status": "Ready",
|
"arm_state": ArmState.IDLE.value,
|
||||||
"arm_state": ArmState.IDLE.value,
|
"arm_current_task": None,
|
||||||
"arm_current_task": None,
|
"heating_stations": self._get_stations_status(),
|
||||||
"heating_stations": self._get_stations_status(),
|
"active_tasks_count": 0,
|
||||||
"active_tasks_count": 0,
|
"message": "工作台就绪",
|
||||||
"message": "工作台就绪",
|
})
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
self.logger.info(f"工作台初始化完成: {self.NUM_HEATING_STATIONS}个加热台就绪")
|
self.logger.info(f"工作台初始化完成: {self.NUM_HEATING_STATIONS}个加热台就绪")
|
||||||
return True
|
return True
|
||||||
@@ -220,14 +204,12 @@ class VirtualWorkbench:
|
|||||||
with self._tasks_lock:
|
with self._tasks_lock:
|
||||||
self._active_tasks.clear()
|
self._active_tasks.clear()
|
||||||
|
|
||||||
self.data.update(
|
self.data.update({
|
||||||
{
|
"status": "Offline",
|
||||||
"status": "Offline",
|
"arm_state": ArmState.IDLE.value,
|
||||||
"arm_state": ArmState.IDLE.value,
|
"heating_stations": {},
|
||||||
"heating_stations": {},
|
"message": "工作台已关闭",
|
||||||
"message": "工作台已关闭",
|
})
|
||||||
}
|
|
||||||
)
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
def _get_stations_status(self) -> Dict[int, Dict[str, Any]]:
|
def _get_stations_status(self) -> Dict[int, Dict[str, Any]]:
|
||||||
@@ -245,14 +227,12 @@ class VirtualWorkbench:
|
|||||||
|
|
||||||
def _update_data_status(self, message: Optional[str] = None):
|
def _update_data_status(self, message: Optional[str] = None):
|
||||||
"""更新状态数据"""
|
"""更新状态数据"""
|
||||||
self.data.update(
|
self.data.update({
|
||||||
{
|
"arm_state": self._arm_state.value,
|
||||||
"arm_state": self._arm_state.value,
|
"arm_current_task": self._arm_current_task,
|
||||||
"arm_current_task": self._arm_current_task,
|
"heating_stations": self._get_stations_status(),
|
||||||
"heating_stations": self._get_stations_status(),
|
"active_tasks_count": len(self._active_tasks),
|
||||||
"active_tasks_count": len(self._active_tasks),
|
})
|
||||||
}
|
|
||||||
)
|
|
||||||
if message:
|
if message:
|
||||||
self.data["message"] = message
|
self.data["message"] = message
|
||||||
|
|
||||||
@@ -300,7 +280,6 @@ class VirtualWorkbench:
|
|||||||
|
|
||||||
def prepare_materials(
|
def prepare_materials(
|
||||||
self,
|
self,
|
||||||
sample_uuids: SampleUUIDsType,
|
|
||||||
count: int = 5,
|
count: int = 5,
|
||||||
) -> PrepareMaterialsResult:
|
) -> PrepareMaterialsResult:
|
||||||
"""
|
"""
|
||||||
@@ -318,7 +297,10 @@ class VirtualWorkbench:
|
|||||||
# 生成物料列表 A1 - A{count}
|
# 生成物料列表 A1 - A{count}
|
||||||
materials = [i for i in range(1, count + 1)]
|
materials = [i for i in range(1, count + 1)]
|
||||||
|
|
||||||
self.logger.info(f"[准备物料] 生成 {count} 个物料: " f"A1-A{count} -> material_1~material_{count}")
|
self.logger.info(
|
||||||
|
f"[准备物料] 生成 {count} 个物料: "
|
||||||
|
f"A1-A{count} -> material_1~material_{count}"
|
||||||
|
)
|
||||||
|
|
||||||
return {
|
return {
|
||||||
"success": True,
|
"success": True,
|
||||||
@@ -329,12 +311,10 @@ class VirtualWorkbench:
|
|||||||
"material_4": materials[3] if len(materials) > 3 else 0,
|
"material_4": materials[3] if len(materials) > 3 else 0,
|
||||||
"material_5": materials[4] if len(materials) > 4 else 0,
|
"material_5": materials[4] if len(materials) > 4 else 0,
|
||||||
"message": f"已准备 {count} 个物料: A1-A{count}",
|
"message": f"已准备 {count} 个物料: A1-A{count}",
|
||||||
"unilabos_samples": [LabSample(sample_uuid=sample_uuid, oss_path="", extra={"material_uuid": content} if isinstance(content, str) else content.serialize()) for sample_uuid, content in sample_uuids.items()]
|
|
||||||
}
|
}
|
||||||
|
|
||||||
def move_to_heating_station(
|
def move_to_heating_station(
|
||||||
self,
|
self,
|
||||||
sample_uuids: SampleUUIDsType,
|
|
||||||
material_number: int,
|
material_number: int,
|
||||||
) -> MoveToHeatingStationResult:
|
) -> MoveToHeatingStationResult:
|
||||||
"""
|
"""
|
||||||
@@ -411,9 +391,6 @@ class VirtualWorkbench:
|
|||||||
"material_id": material_id,
|
"material_id": material_id,
|
||||||
"material_number": material_number,
|
"material_number": material_number,
|
||||||
"message": f"{material_id}已成功移动到加热台{station_id}",
|
"message": f"{material_id}已成功移动到加热台{station_id}",
|
||||||
"unilabos_samples": [
|
|
||||||
LabSample(sample_uuid=sample_uuid, oss_path="", extra={"material_uuid": content} if isinstance(content, str) else content.serialize()) for
|
|
||||||
sample_uuid, content in sample_uuids.items()]
|
|
||||||
}
|
}
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
@@ -426,14 +403,10 @@ class VirtualWorkbench:
|
|||||||
"material_id": material_id,
|
"material_id": material_id,
|
||||||
"material_number": material_number,
|
"material_number": material_number,
|
||||||
"message": f"移动失败: {str(e)}",
|
"message": f"移动失败: {str(e)}",
|
||||||
"unilabos_samples": [
|
|
||||||
LabSample(sample_uuid=sample_uuid, oss_path="", extra={"material_uuid": content} if isinstance(content, str) else content.serialize()) for
|
|
||||||
sample_uuid, content in sample_uuids.items()]
|
|
||||||
}
|
}
|
||||||
|
|
||||||
def start_heating(
|
def start_heating(
|
||||||
self,
|
self,
|
||||||
sample_uuids: SampleUUIDsType,
|
|
||||||
station_id: int,
|
station_id: int,
|
||||||
material_number: int,
|
material_number: int,
|
||||||
) -> StartHeatingResult:
|
) -> StartHeatingResult:
|
||||||
@@ -456,9 +429,6 @@ class VirtualWorkbench:
|
|||||||
"material_id": "",
|
"material_id": "",
|
||||||
"material_number": material_number,
|
"material_number": material_number,
|
||||||
"message": f"无效的加热台ID: {station_id}",
|
"message": f"无效的加热台ID: {station_id}",
|
||||||
"unilabos_samples": [
|
|
||||||
LabSample(sample_uuid=sample_uuid, oss_path="", extra={"material_uuid": content} if isinstance(content, str) else content.serialize()) for
|
|
||||||
sample_uuid, content in sample_uuids.items()]
|
|
||||||
}
|
}
|
||||||
|
|
||||||
with self._stations_lock:
|
with self._stations_lock:
|
||||||
@@ -471,9 +441,6 @@ class VirtualWorkbench:
|
|||||||
"material_id": "",
|
"material_id": "",
|
||||||
"material_number": material_number,
|
"material_number": material_number,
|
||||||
"message": f"加热台{station_id}上没有物料",
|
"message": f"加热台{station_id}上没有物料",
|
||||||
"unilabos_samples": [
|
|
||||||
LabSample(sample_uuid=sample_uuid, oss_path="", extra={"material_uuid": content} if isinstance(content, str) else content.serialize()) for
|
|
||||||
sample_uuid, content in sample_uuids.items()]
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if station.state == HeatingStationState.HEATING:
|
if station.state == HeatingStationState.HEATING:
|
||||||
@@ -483,9 +450,6 @@ class VirtualWorkbench:
|
|||||||
"material_id": station.current_material,
|
"material_id": station.current_material,
|
||||||
"material_number": material_number,
|
"material_number": material_number,
|
||||||
"message": f"加热台{station_id}已经在加热中",
|
"message": f"加热台{station_id}已经在加热中",
|
||||||
"unilabos_samples": [
|
|
||||||
LabSample(sample_uuid=sample_uuid, oss_path="", extra={"material_uuid": content} if isinstance(content, str) else content.serialize()) for
|
|
||||||
sample_uuid, content in sample_uuids.items()]
|
|
||||||
}
|
}
|
||||||
|
|
||||||
material_id = station.current_material
|
material_id = station.current_material
|
||||||
@@ -535,14 +499,10 @@ class VirtualWorkbench:
|
|||||||
"material_id": material_id,
|
"material_id": material_id,
|
||||||
"material_number": material_number,
|
"material_number": material_number,
|
||||||
"message": f"加热台{station_id}加热完成",
|
"message": f"加热台{station_id}加热完成",
|
||||||
"unilabos_samples": [
|
|
||||||
LabSample(sample_uuid=sample_uuid, oss_path="", extra={"material_uuid": content} if isinstance(content, str) else content.serialize()) for
|
|
||||||
sample_uuid, content in sample_uuids.items()]
|
|
||||||
}
|
}
|
||||||
|
|
||||||
def move_to_output(
|
def move_to_output(
|
||||||
self,
|
self,
|
||||||
sample_uuids: SampleUUIDsType,
|
|
||||||
station_id: int,
|
station_id: int,
|
||||||
material_number: int,
|
material_number: int,
|
||||||
) -> MoveToOutputResult:
|
) -> MoveToOutputResult:
|
||||||
@@ -565,9 +525,6 @@ class VirtualWorkbench:
|
|||||||
"material_id": "",
|
"material_id": "",
|
||||||
"output_position": f"C{output_number}",
|
"output_position": f"C{output_number}",
|
||||||
"message": f"无效的加热台ID: {station_id}",
|
"message": f"无效的加热台ID: {station_id}",
|
||||||
"unilabos_samples": [
|
|
||||||
LabSample(sample_uuid=sample_uuid, oss_path="", extra={"material_uuid": content} if isinstance(content, str) else content.serialize()) for
|
|
||||||
sample_uuid, content in sample_uuids.items()]
|
|
||||||
}
|
}
|
||||||
|
|
||||||
with self._stations_lock:
|
with self._stations_lock:
|
||||||
@@ -581,9 +538,6 @@ class VirtualWorkbench:
|
|||||||
"material_id": "",
|
"material_id": "",
|
||||||
"output_position": f"C{output_number}",
|
"output_position": f"C{output_number}",
|
||||||
"message": f"加热台{station_id}上没有物料",
|
"message": f"加热台{station_id}上没有物料",
|
||||||
"unilabos_samples": [
|
|
||||||
LabSample(sample_uuid=sample_uuid, oss_path="", extra={"material_uuid": content} if isinstance(content, str) else content.serialize()) for
|
|
||||||
sample_uuid, content in sample_uuids.items()]
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if station.state != HeatingStationState.COMPLETED:
|
if station.state != HeatingStationState.COMPLETED:
|
||||||
@@ -593,9 +547,6 @@ class VirtualWorkbench:
|
|||||||
"material_id": material_id,
|
"material_id": material_id,
|
||||||
"output_position": f"C{output_number}",
|
"output_position": f"C{output_number}",
|
||||||
"message": f"加热台{station_id}尚未完成加热 (当前状态: {station.state.value})",
|
"message": f"加热台{station_id}尚未完成加热 (当前状态: {station.state.value})",
|
||||||
"unilabos_samples": [
|
|
||||||
LabSample(sample_uuid=sample_uuid, oss_path="", extra={"material_uuid": content} if isinstance(content, str) else content.serialize()) for
|
|
||||||
sample_uuid, content in sample_uuids.items()]
|
|
||||||
}
|
}
|
||||||
|
|
||||||
output_position = f"C{output_number}"
|
output_position = f"C{output_number}"
|
||||||
@@ -644,9 +595,6 @@ class VirtualWorkbench:
|
|||||||
"material_id": material_id,
|
"material_id": material_id,
|
||||||
"output_position": output_position,
|
"output_position": output_position,
|
||||||
"message": f"{material_id}已成功移动到{output_position}",
|
"message": f"{material_id}已成功移动到{output_position}",
|
||||||
"unilabos_samples": [
|
|
||||||
LabSample(sample_uuid=sample_uuid, oss_path="", extra={"material_uuid": content} if isinstance(content, str) else content.serialize()) for
|
|
||||||
sample_uuid, content in sample_uuids.items()]
|
|
||||||
}
|
}
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
@@ -659,9 +607,6 @@ class VirtualWorkbench:
|
|||||||
"material_id": "",
|
"material_id": "",
|
||||||
"output_position": output_position,
|
"output_position": output_position,
|
||||||
"message": f"移动失败: {str(e)}",
|
"message": f"移动失败: {str(e)}",
|
||||||
"unilabos_samples": [
|
|
||||||
LabSample(sample_uuid=sample_uuid, oss_path="", extra={"material_uuid": content} if isinstance(content, str) else content.serialize()) for
|
|
||||||
sample_uuid, content in sample_uuids.items()]
|
|
||||||
}
|
}
|
||||||
|
|
||||||
# ============ 状态属性 ============
|
# ============ 状态属性 ============
|
||||||
|
|||||||
@@ -317,6 +317,47 @@ separator.chinwe:
|
|||||||
- port
|
- port
|
||||||
type: object
|
type: object
|
||||||
type: UniLabJsonCommand
|
type: UniLabJsonCommand
|
||||||
|
separation_step:
|
||||||
|
goal:
|
||||||
|
max_cycles: 0
|
||||||
|
motor_id: 5
|
||||||
|
pulses: 700
|
||||||
|
speed: 60
|
||||||
|
timeout: 300
|
||||||
|
handles: {}
|
||||||
|
schema:
|
||||||
|
description: 分液步骤 - 液位传感器与电机联动 (有液→顺时针, 无液→逆时针)
|
||||||
|
properties:
|
||||||
|
goal:
|
||||||
|
properties:
|
||||||
|
max_cycles:
|
||||||
|
default: 0
|
||||||
|
description: 最大循环次数 (0=无限制)
|
||||||
|
type: integer
|
||||||
|
motor_id:
|
||||||
|
default: '5'
|
||||||
|
description: 选择电机
|
||||||
|
enum:
|
||||||
|
- '4'
|
||||||
|
- '5'
|
||||||
|
title: '注: 4=搅拌, 5=旋钮'
|
||||||
|
type: string
|
||||||
|
pulses:
|
||||||
|
default: 700
|
||||||
|
description: 每次旋转脉冲数 (约1/4圈)
|
||||||
|
type: integer
|
||||||
|
speed:
|
||||||
|
default: 60
|
||||||
|
description: 电机转速 (RPM)
|
||||||
|
type: integer
|
||||||
|
timeout:
|
||||||
|
default: 300
|
||||||
|
description: 超时时间 (秒)
|
||||||
|
type: integer
|
||||||
|
required:
|
||||||
|
- motor_id
|
||||||
|
type: object
|
||||||
|
type: UniLabJsonCommand
|
||||||
wait_sensor_level:
|
wait_sensor_level:
|
||||||
goal:
|
goal:
|
||||||
target_state: 有液
|
target_state: 有液
|
||||||
|
|||||||
@@ -638,7 +638,7 @@ liquid_handler:
|
|||||||
placeholder_keys: {}
|
placeholder_keys: {}
|
||||||
result: {}
|
result: {}
|
||||||
schema:
|
schema:
|
||||||
description: 吸头迭代函数。用于自动管理和切换枪头盒中的吸头,实现批量实验中的吸头自动分配和追踪。该函数监控吸头使用状态,自动切换到下一个可用吸头位置,确保实验流程的连续性。适用于高通量实验、批量处理、自动化流水线等需要大量吸头管理的应用场景。
|
description: 吸头迭代函数。用于自动管理和切换吸头架中的吸头,实现批量实验中的吸头自动分配和追踪。该函数监控吸头使用状态,自动切换到下一个可用吸头位置,确保实验流程的连续性。适用于高通量实验、批量处理、自动化流水线等需要大量吸头管理的应用场景。
|
||||||
properties:
|
properties:
|
||||||
feedback: {}
|
feedback: {}
|
||||||
goal:
|
goal:
|
||||||
@@ -712,43 +712,6 @@ liquid_handler:
|
|||||||
title: set_group参数
|
title: set_group参数
|
||||||
type: object
|
type: object
|
||||||
type: UniLabJsonCommand
|
type: UniLabJsonCommand
|
||||||
auto-set_liquid_from_plate:
|
|
||||||
feedback: {}
|
|
||||||
goal: {}
|
|
||||||
goal_default:
|
|
||||||
liquid_names: null
|
|
||||||
plate: null
|
|
||||||
volumes: null
|
|
||||||
well_names: null
|
|
||||||
handles: {}
|
|
||||||
placeholder_keys: {}
|
|
||||||
result: {}
|
|
||||||
schema:
|
|
||||||
description: ''
|
|
||||||
properties:
|
|
||||||
feedback: {}
|
|
||||||
goal:
|
|
||||||
properties:
|
|
||||||
liquid_names:
|
|
||||||
type: string
|
|
||||||
plate:
|
|
||||||
type: string
|
|
||||||
volumes:
|
|
||||||
type: string
|
|
||||||
well_names:
|
|
||||||
type: string
|
|
||||||
required:
|
|
||||||
- plate
|
|
||||||
- well_names
|
|
||||||
- liquid_names
|
|
||||||
- volumes
|
|
||||||
type: object
|
|
||||||
result: {}
|
|
||||||
required:
|
|
||||||
- goal
|
|
||||||
title: set_liquid_from_plate参数
|
|
||||||
type: object
|
|
||||||
type: UniLabJsonCommand
|
|
||||||
auto-set_tiprack:
|
auto-set_tiprack:
|
||||||
feedback: {}
|
feedback: {}
|
||||||
goal: {}
|
goal: {}
|
||||||
@@ -758,7 +721,7 @@ liquid_handler:
|
|||||||
placeholder_keys: {}
|
placeholder_keys: {}
|
||||||
result: {}
|
result: {}
|
||||||
schema:
|
schema:
|
||||||
description: 枪头盒设置函数。用于配置和初始化液体处理系统的枪头盒信息,包括枪头盒位置、类型、容量等参数。该函数建立吸头资源管理系统,为后续的吸头选择和使用提供基础配置。适用于系统初始化、枪头盒更换、实验配置等需要吸头资源管理的操作场景。
|
description: 吸头架设置函数。用于配置和初始化液体处理系统的吸头架信息,包括吸头架位置、类型、容量等参数。该函数建立吸头资源管理系统,为后续的吸头选择和使用提供基础配置。适用于系统初始化、吸头架更换、实验配置等需要吸头资源管理的操作场景。
|
||||||
properties:
|
properties:
|
||||||
feedback: {}
|
feedback: {}
|
||||||
goal:
|
goal:
|
||||||
@@ -4130,32 +4093,32 @@ liquid_handler:
|
|||||||
- 0
|
- 0
|
||||||
handles:
|
handles:
|
||||||
input:
|
input:
|
||||||
- data_key: sources
|
- data_key: liquid
|
||||||
data_source: handle
|
data_source: handle
|
||||||
data_type: resource
|
data_type: resource
|
||||||
handler_key: sources
|
handler_key: sources
|
||||||
label: 待移动液体
|
label: sources
|
||||||
- data_key: targets
|
- data_key: liquid
|
||||||
data_source: handle
|
|
||||||
data_type: resource
|
|
||||||
handler_key: targets
|
|
||||||
label: 转移目标
|
|
||||||
- data_key: tip_racks
|
|
||||||
data_source: handle
|
|
||||||
data_type: resource
|
|
||||||
handler_key: tip_rack
|
|
||||||
label: 枪头盒
|
|
||||||
output:
|
|
||||||
- data_key: sources.@flatten
|
|
||||||
data_source: executor
|
data_source: executor
|
||||||
data_type: resource
|
data_type: resource
|
||||||
|
handler_key: targets
|
||||||
|
label: targets
|
||||||
|
- data_key: liquid
|
||||||
|
data_source: executor
|
||||||
|
data_type: resource
|
||||||
|
handler_key: tip_rack
|
||||||
|
label: tip_rack
|
||||||
|
output:
|
||||||
|
- data_key: liquid
|
||||||
|
data_source: handle
|
||||||
|
data_type: resource
|
||||||
handler_key: sources_out
|
handler_key: sources_out
|
||||||
label: 移液后源孔
|
label: sources
|
||||||
- data_key: targets.@flatten
|
- data_key: liquid
|
||||||
data_source: executor
|
data_source: executor
|
||||||
data_type: resource
|
data_type: resource
|
||||||
handler_key: targets_out
|
handler_key: targets_out
|
||||||
label: 移液后目标孔
|
label: targets
|
||||||
placeholder_keys:
|
placeholder_keys:
|
||||||
sources: unilabos_resources
|
sources: unilabos_resources
|
||||||
targets: unilabos_resources
|
targets: unilabos_resources
|
||||||
@@ -5151,34 +5114,19 @@ liquid_handler.biomek:
|
|||||||
- 0
|
- 0
|
||||||
handles:
|
handles:
|
||||||
input:
|
input:
|
||||||
- data_key: sources
|
- data_key: liquid
|
||||||
data_source: handle
|
data_source: handle
|
||||||
data_type: resource
|
data_type: resource
|
||||||
handler_key: sources
|
handler_key: liquid-input
|
||||||
io_type: target
|
io_type: target
|
||||||
label: 待移动液体
|
label: Liquid Input
|
||||||
- data_key: targets
|
|
||||||
data_source: handle
|
|
||||||
data_type: resource
|
|
||||||
handler_key: targets
|
|
||||||
label: 转移目标
|
|
||||||
- data_key: tip_racks
|
|
||||||
data_source: handle
|
|
||||||
data_type: resource
|
|
||||||
handler_key: tip_rack
|
|
||||||
label: 枪头盒
|
|
||||||
output:
|
output:
|
||||||
- data_key: sources.@flatten
|
- data_key: liquid
|
||||||
data_source: executor
|
data_source: executor
|
||||||
data_type: resource
|
data_type: resource
|
||||||
handler_key: sources_out
|
handler_key: liquid-output
|
||||||
io_type: source
|
io_type: source
|
||||||
label: 移液后源孔
|
label: Liquid Output
|
||||||
- data_key: targets.@flatten
|
|
||||||
data_source: executor
|
|
||||||
data_type: resource
|
|
||||||
handler_key: targets_out
|
|
||||||
label: 移液后目标孔
|
|
||||||
placeholder_keys:
|
placeholder_keys:
|
||||||
sources: unilabos_resources
|
sources: unilabos_resources
|
||||||
targets: unilabos_resources
|
targets: unilabos_resources
|
||||||
@@ -9468,7 +9416,7 @@ liquid_handler.prcxi:
|
|||||||
well_names: null
|
well_names: null
|
||||||
handles:
|
handles:
|
||||||
input:
|
input:
|
||||||
- data_key: '@this.0@@@plate'
|
- data_key: plate
|
||||||
data_source: handle
|
data_source: handle
|
||||||
data_type: resource
|
data_type: resource
|
||||||
handler_key: input_plate
|
handler_key: input_plate
|
||||||
@@ -9590,195 +9538,9 @@ liquid_handler.prcxi:
|
|||||||
- volumes
|
- volumes
|
||||||
type: object
|
type: object
|
||||||
result:
|
result:
|
||||||
$defs:
|
|
||||||
ResourceDict:
|
|
||||||
properties:
|
|
||||||
class:
|
|
||||||
description: Resource class name
|
|
||||||
title: Class
|
|
||||||
type: string
|
|
||||||
config:
|
|
||||||
additionalProperties: true
|
|
||||||
description: Resource configuration
|
|
||||||
title: Config
|
|
||||||
type: object
|
|
||||||
data:
|
|
||||||
additionalProperties: true
|
|
||||||
description: 'Resource data, eg: container liquid data'
|
|
||||||
title: Data
|
|
||||||
type: object
|
|
||||||
description:
|
|
||||||
default: ''
|
|
||||||
description: Resource description
|
|
||||||
title: Description
|
|
||||||
type: string
|
|
||||||
extra:
|
|
||||||
additionalProperties: true
|
|
||||||
description: 'Extra data, eg: slot index'
|
|
||||||
title: Extra
|
|
||||||
type: object
|
|
||||||
icon:
|
|
||||||
default: ''
|
|
||||||
description: Resource icon
|
|
||||||
title: Icon
|
|
||||||
type: string
|
|
||||||
id:
|
|
||||||
description: Resource ID
|
|
||||||
title: Id
|
|
||||||
type: string
|
|
||||||
model:
|
|
||||||
additionalProperties: true
|
|
||||||
description: Resource model
|
|
||||||
title: Model
|
|
||||||
type: object
|
|
||||||
name:
|
|
||||||
description: Resource name
|
|
||||||
title: Name
|
|
||||||
type: string
|
|
||||||
parent:
|
|
||||||
anyOf:
|
|
||||||
- $ref: '#/$defs/ResourceDict'
|
|
||||||
- type: 'null'
|
|
||||||
default: null
|
|
||||||
description: Parent resource object
|
|
||||||
parent_uuid:
|
|
||||||
anyOf:
|
|
||||||
- type: string
|
|
||||||
- type: 'null'
|
|
||||||
default: null
|
|
||||||
description: Parent resource uuid
|
|
||||||
title: Parent Uuid
|
|
||||||
pose:
|
|
||||||
$ref: '#/$defs/ResourceDictPosition'
|
|
||||||
description: Resource position
|
|
||||||
schema:
|
|
||||||
additionalProperties: true
|
|
||||||
description: Resource schema
|
|
||||||
title: Schema
|
|
||||||
type: object
|
|
||||||
type:
|
|
||||||
anyOf:
|
|
||||||
- const: device
|
|
||||||
type: string
|
|
||||||
- type: string
|
|
||||||
description: Resource type
|
|
||||||
title: Type
|
|
||||||
uuid:
|
|
||||||
description: Resource UUID
|
|
||||||
title: Uuid
|
|
||||||
type: string
|
|
||||||
required:
|
|
||||||
- id
|
|
||||||
- uuid
|
|
||||||
- name
|
|
||||||
- type
|
|
||||||
- class
|
|
||||||
- config
|
|
||||||
- data
|
|
||||||
- extra
|
|
||||||
title: ResourceDict
|
|
||||||
type: object
|
|
||||||
ResourceDictPosition:
|
|
||||||
properties:
|
|
||||||
cross_section_type:
|
|
||||||
default: rectangle
|
|
||||||
description: Cross section type
|
|
||||||
enum:
|
|
||||||
- rectangle
|
|
||||||
- circle
|
|
||||||
- rounded_rectangle
|
|
||||||
title: Cross Section Type
|
|
||||||
type: string
|
|
||||||
layout:
|
|
||||||
default: x-y
|
|
||||||
description: Resource layout
|
|
||||||
enum:
|
|
||||||
- 2d
|
|
||||||
- x-y
|
|
||||||
- z-y
|
|
||||||
- x-z
|
|
||||||
title: Layout
|
|
||||||
type: string
|
|
||||||
position:
|
|
||||||
$ref: '#/$defs/ResourceDictPositionObject'
|
|
||||||
description: Resource position
|
|
||||||
position3d:
|
|
||||||
$ref: '#/$defs/ResourceDictPositionObject'
|
|
||||||
description: Resource position in 3D space
|
|
||||||
rotation:
|
|
||||||
$ref: '#/$defs/ResourceDictPositionObject'
|
|
||||||
description: Resource rotation
|
|
||||||
scale:
|
|
||||||
$ref: '#/$defs/ResourceDictPositionScale'
|
|
||||||
description: Resource scale
|
|
||||||
size:
|
|
||||||
$ref: '#/$defs/ResourceDictPositionSize'
|
|
||||||
description: Resource size
|
|
||||||
title: ResourceDictPosition
|
|
||||||
type: object
|
|
||||||
ResourceDictPositionObject:
|
|
||||||
properties:
|
|
||||||
x:
|
|
||||||
default: 0.0
|
|
||||||
description: X coordinate
|
|
||||||
title: X
|
|
||||||
type: number
|
|
||||||
y:
|
|
||||||
default: 0.0
|
|
||||||
description: Y coordinate
|
|
||||||
title: Y
|
|
||||||
type: number
|
|
||||||
z:
|
|
||||||
default: 0.0
|
|
||||||
description: Z coordinate
|
|
||||||
title: Z
|
|
||||||
type: number
|
|
||||||
title: ResourceDictPositionObject
|
|
||||||
type: object
|
|
||||||
ResourceDictPositionScale:
|
|
||||||
properties:
|
|
||||||
x:
|
|
||||||
default: 0.0
|
|
||||||
description: x scale
|
|
||||||
title: X
|
|
||||||
type: number
|
|
||||||
y:
|
|
||||||
default: 0.0
|
|
||||||
description: y scale
|
|
||||||
title: Y
|
|
||||||
type: number
|
|
||||||
z:
|
|
||||||
default: 0.0
|
|
||||||
description: z scale
|
|
||||||
title: Z
|
|
||||||
type: number
|
|
||||||
title: ResourceDictPositionScale
|
|
||||||
type: object
|
|
||||||
ResourceDictPositionSize:
|
|
||||||
properties:
|
|
||||||
depth:
|
|
||||||
default: 0.0
|
|
||||||
description: Depth
|
|
||||||
title: Depth
|
|
||||||
type: number
|
|
||||||
height:
|
|
||||||
default: 0.0
|
|
||||||
description: Height
|
|
||||||
title: Height
|
|
||||||
type: number
|
|
||||||
width:
|
|
||||||
default: 0.0
|
|
||||||
description: Width
|
|
||||||
title: Width
|
|
||||||
type: number
|
|
||||||
title: ResourceDictPositionSize
|
|
||||||
type: object
|
|
||||||
properties:
|
properties:
|
||||||
plate:
|
plate:
|
||||||
items:
|
items: {}
|
||||||
items:
|
|
||||||
$ref: '#/$defs/ResourceDict'
|
|
||||||
type: array
|
|
||||||
title: Plate
|
title: Plate
|
||||||
type: array
|
type: array
|
||||||
volumes:
|
volumes:
|
||||||
@@ -9787,10 +9549,7 @@ liquid_handler.prcxi:
|
|||||||
title: Volumes
|
title: Volumes
|
||||||
type: array
|
type: array
|
||||||
wells:
|
wells:
|
||||||
items:
|
items: {}
|
||||||
items:
|
|
||||||
$ref: '#/$defs/ResourceDict'
|
|
||||||
type: array
|
|
||||||
title: Wells
|
title: Wells
|
||||||
type: array
|
type: array
|
||||||
required:
|
required:
|
||||||
@@ -10163,18 +9922,18 @@ liquid_handler.prcxi:
|
|||||||
data_source: handle
|
data_source: handle
|
||||||
data_type: resource
|
data_type: resource
|
||||||
handler_key: tip_rack_identifier
|
handler_key: tip_rack_identifier
|
||||||
label: 枪头盒
|
label: 墙头盒
|
||||||
output:
|
output:
|
||||||
- data_key: sources.@flatten
|
- data_key: liquid
|
||||||
data_source: executor
|
data_source: handle
|
||||||
data_type: resource
|
data_type: resource
|
||||||
handler_key: sources_out
|
handler_key: sources_out
|
||||||
label: 移液后源孔
|
label: sources
|
||||||
- data_key: targets.@flatten
|
- data_key: liquid
|
||||||
data_source: executor
|
data_source: executor
|
||||||
data_type: resource
|
data_type: resource
|
||||||
handler_key: targets_out
|
handler_key: targets_out
|
||||||
label: 移液后目标孔
|
label: targets
|
||||||
placeholder_keys:
|
placeholder_keys:
|
||||||
sources: unilabos_resources
|
sources: unilabos_resources
|
||||||
targets: unilabos_resources
|
targets: unilabos_resources
|
||||||
|
|||||||
286
unilabos/registry/devices/motor.yaml
Normal file
286
unilabos/registry/devices/motor.yaml
Normal file
@@ -0,0 +1,286 @@
|
|||||||
|
motor.zdt_x42:
|
||||||
|
category:
|
||||||
|
- motor
|
||||||
|
class:
|
||||||
|
action_value_mappings:
|
||||||
|
auto-enable:
|
||||||
|
feedback: {}
|
||||||
|
goal: {}
|
||||||
|
goal_default:
|
||||||
|
'on': true
|
||||||
|
handles: {}
|
||||||
|
placeholder_keys: {}
|
||||||
|
result: {}
|
||||||
|
schema:
|
||||||
|
description: 使能或禁用电机。使能后电机进入锁轴状态,可接收运动指令;禁用后电机进入松轴状态。
|
||||||
|
properties:
|
||||||
|
feedback: {}
|
||||||
|
goal:
|
||||||
|
properties:
|
||||||
|
'on':
|
||||||
|
default: true
|
||||||
|
type: boolean
|
||||||
|
required: []
|
||||||
|
type: object
|
||||||
|
result: {}
|
||||||
|
required:
|
||||||
|
- goal
|
||||||
|
title: enable参数
|
||||||
|
type: object
|
||||||
|
type: UniLabJsonCommand
|
||||||
|
auto-get_position:
|
||||||
|
feedback: {}
|
||||||
|
goal: {}
|
||||||
|
goal_default: {}
|
||||||
|
handles: {}
|
||||||
|
placeholder_keys: {}
|
||||||
|
result: {}
|
||||||
|
schema:
|
||||||
|
description: 获取当前电机脉冲位置。
|
||||||
|
properties:
|
||||||
|
feedback: {}
|
||||||
|
goal:
|
||||||
|
properties: {}
|
||||||
|
required: []
|
||||||
|
type: object
|
||||||
|
result:
|
||||||
|
properties:
|
||||||
|
position:
|
||||||
|
type: integer
|
||||||
|
type: object
|
||||||
|
required:
|
||||||
|
- goal
|
||||||
|
title: get_position参数
|
||||||
|
type: object
|
||||||
|
type: UniLabJsonCommand
|
||||||
|
auto-move_position:
|
||||||
|
feedback: {}
|
||||||
|
goal: {}
|
||||||
|
goal_default:
|
||||||
|
absolute: false
|
||||||
|
acceleration: 10
|
||||||
|
direction: CW
|
||||||
|
pulses: 1000
|
||||||
|
speed_rpm: 60
|
||||||
|
handles: {}
|
||||||
|
placeholder_keys: {}
|
||||||
|
result: {}
|
||||||
|
schema:
|
||||||
|
description: 位置模式运行。控制电机移动到指定脉冲位置或相对于当前位置移动指定脉冲数。
|
||||||
|
properties:
|
||||||
|
feedback: {}
|
||||||
|
goal:
|
||||||
|
properties:
|
||||||
|
absolute:
|
||||||
|
default: false
|
||||||
|
type: boolean
|
||||||
|
acceleration:
|
||||||
|
default: 10
|
||||||
|
maximum: 255
|
||||||
|
minimum: 0
|
||||||
|
type: integer
|
||||||
|
direction:
|
||||||
|
default: CW
|
||||||
|
enum:
|
||||||
|
- CW
|
||||||
|
- CCW
|
||||||
|
type: string
|
||||||
|
pulses:
|
||||||
|
default: 1000
|
||||||
|
type: integer
|
||||||
|
speed_rpm:
|
||||||
|
default: 60
|
||||||
|
minimum: 0
|
||||||
|
type: integer
|
||||||
|
required:
|
||||||
|
- pulses
|
||||||
|
- speed_rpm
|
||||||
|
type: object
|
||||||
|
result: {}
|
||||||
|
required:
|
||||||
|
- goal
|
||||||
|
title: move_position参数
|
||||||
|
type: object
|
||||||
|
type: UniLabJsonCommand
|
||||||
|
auto-move_speed:
|
||||||
|
feedback: {}
|
||||||
|
goal: {}
|
||||||
|
goal_default:
|
||||||
|
acceleration: 10
|
||||||
|
direction: CW
|
||||||
|
speed_rpm: 60
|
||||||
|
handles: {}
|
||||||
|
placeholder_keys: {}
|
||||||
|
result: {}
|
||||||
|
schema:
|
||||||
|
description: 速度模式运行。控制电机以指定转速和方向持续转动。
|
||||||
|
properties:
|
||||||
|
feedback: {}
|
||||||
|
goal:
|
||||||
|
properties:
|
||||||
|
acceleration:
|
||||||
|
default: 10
|
||||||
|
maximum: 255
|
||||||
|
minimum: 0
|
||||||
|
type: integer
|
||||||
|
direction:
|
||||||
|
default: CW
|
||||||
|
enum:
|
||||||
|
- CW
|
||||||
|
- CCW
|
||||||
|
type: string
|
||||||
|
speed_rpm:
|
||||||
|
default: 60
|
||||||
|
minimum: 0
|
||||||
|
type: integer
|
||||||
|
required:
|
||||||
|
- speed_rpm
|
||||||
|
type: object
|
||||||
|
result: {}
|
||||||
|
required:
|
||||||
|
- goal
|
||||||
|
title: move_speed参数
|
||||||
|
type: object
|
||||||
|
type: UniLabJsonCommand
|
||||||
|
auto-rotate_quarter:
|
||||||
|
feedback: {}
|
||||||
|
goal: {}
|
||||||
|
goal_default:
|
||||||
|
direction: CW
|
||||||
|
speed_rpm: 60
|
||||||
|
handles: {}
|
||||||
|
placeholder_keys: {}
|
||||||
|
result: {}
|
||||||
|
schema:
|
||||||
|
description: 电机旋转 1/4 圈 (阻塞式)。
|
||||||
|
properties:
|
||||||
|
feedback: {}
|
||||||
|
goal:
|
||||||
|
properties:
|
||||||
|
direction:
|
||||||
|
default: CW
|
||||||
|
enum:
|
||||||
|
- CW
|
||||||
|
- CCW
|
||||||
|
type: string
|
||||||
|
speed_rpm:
|
||||||
|
default: 60
|
||||||
|
minimum: 1
|
||||||
|
type: integer
|
||||||
|
required: []
|
||||||
|
type: object
|
||||||
|
result: {}
|
||||||
|
required:
|
||||||
|
- goal
|
||||||
|
title: rotate_quarter参数
|
||||||
|
type: object
|
||||||
|
type: UniLabJsonCommand
|
||||||
|
auto-set_zero:
|
||||||
|
feedback: {}
|
||||||
|
goal: {}
|
||||||
|
goal_default: {}
|
||||||
|
handles: {}
|
||||||
|
placeholder_keys: {}
|
||||||
|
result: {}
|
||||||
|
schema:
|
||||||
|
description: 将当前电机位置设为零点。
|
||||||
|
properties:
|
||||||
|
feedback: {}
|
||||||
|
goal:
|
||||||
|
properties: {}
|
||||||
|
required: []
|
||||||
|
type: object
|
||||||
|
result: {}
|
||||||
|
required:
|
||||||
|
- goal
|
||||||
|
title: set_zero参数
|
||||||
|
type: object
|
||||||
|
type: UniLabJsonCommand
|
||||||
|
auto-stop:
|
||||||
|
feedback: {}
|
||||||
|
goal: {}
|
||||||
|
goal_default: {}
|
||||||
|
handles: {}
|
||||||
|
placeholder_keys: {}
|
||||||
|
result: {}
|
||||||
|
schema:
|
||||||
|
description: 立即停止电机运动。
|
||||||
|
properties:
|
||||||
|
feedback: {}
|
||||||
|
goal:
|
||||||
|
properties: {}
|
||||||
|
required: []
|
||||||
|
type: object
|
||||||
|
result: {}
|
||||||
|
required:
|
||||||
|
- goal
|
||||||
|
title: stop参数
|
||||||
|
type: object
|
||||||
|
type: UniLabJsonCommand
|
||||||
|
auto-wait_time:
|
||||||
|
feedback: {}
|
||||||
|
goal: {}
|
||||||
|
goal_default:
|
||||||
|
duration_s: 1.0
|
||||||
|
handles: {}
|
||||||
|
placeholder_keys: {}
|
||||||
|
result: {}
|
||||||
|
schema:
|
||||||
|
description: 等待指定时间 (秒)。
|
||||||
|
properties:
|
||||||
|
feedback: {}
|
||||||
|
goal:
|
||||||
|
properties:
|
||||||
|
duration_s:
|
||||||
|
default: 1.0
|
||||||
|
minimum: 0
|
||||||
|
type: number
|
||||||
|
required:
|
||||||
|
- duration_s
|
||||||
|
type: object
|
||||||
|
result: {}
|
||||||
|
required:
|
||||||
|
- goal
|
||||||
|
title: wait_time参数
|
||||||
|
type: object
|
||||||
|
type: UniLabJsonCommand
|
||||||
|
module: unilabos.devices.motor.ZDT_X42:ZDTX42Driver
|
||||||
|
status_types:
|
||||||
|
position: int
|
||||||
|
status: str
|
||||||
|
type: python
|
||||||
|
config_info: []
|
||||||
|
description: ZDT X42 闭环步进电机驱动。支持速度运行、精确位置控制、位置查询和清零功能。适用于各种需要精确运动控制的实验室自动化场景。
|
||||||
|
handles: []
|
||||||
|
icon: ''
|
||||||
|
init_param_schema:
|
||||||
|
config:
|
||||||
|
properties:
|
||||||
|
baudrate:
|
||||||
|
default: 115200
|
||||||
|
type: integer
|
||||||
|
debug:
|
||||||
|
default: false
|
||||||
|
type: boolean
|
||||||
|
device_id:
|
||||||
|
default: 1
|
||||||
|
type: integer
|
||||||
|
port:
|
||||||
|
type: string
|
||||||
|
timeout:
|
||||||
|
default: 0.5
|
||||||
|
type: number
|
||||||
|
required:
|
||||||
|
- port
|
||||||
|
type: object
|
||||||
|
data:
|
||||||
|
properties:
|
||||||
|
position:
|
||||||
|
type: integer
|
||||||
|
status:
|
||||||
|
type: string
|
||||||
|
required:
|
||||||
|
- status
|
||||||
|
- position
|
||||||
|
type: object
|
||||||
|
version: 1.0.0
|
||||||
148
unilabos/registry/devices/sensor.yaml
Normal file
148
unilabos/registry/devices/sensor.yaml
Normal file
@@ -0,0 +1,148 @@
|
|||||||
|
sensor.xkc_rs485:
|
||||||
|
category:
|
||||||
|
- sensor
|
||||||
|
- separator
|
||||||
|
class:
|
||||||
|
action_value_mappings:
|
||||||
|
auto-change_baudrate:
|
||||||
|
goal:
|
||||||
|
baud_code: 7
|
||||||
|
handles: {}
|
||||||
|
schema:
|
||||||
|
description: '更改通讯波特率 (设置成功后无返回,且需手动切换波特率重连)。代码表 (16进制): 05=2400, 06=4800,
|
||||||
|
07=9600, 08=14400, 09=19200, 0A=28800, 0C=57600, 0D=115200, 0E=128000,
|
||||||
|
0F=256000'
|
||||||
|
properties:
|
||||||
|
goal:
|
||||||
|
properties:
|
||||||
|
baud_code:
|
||||||
|
description: '波特率代码 (例如: 7 为 9600, 13 即 0x0D 为 115200)'
|
||||||
|
type: integer
|
||||||
|
required:
|
||||||
|
- baud_code
|
||||||
|
type: object
|
||||||
|
type: UniLabJsonCommand
|
||||||
|
auto-change_device_id:
|
||||||
|
goal:
|
||||||
|
new_id: 1
|
||||||
|
handles: {}
|
||||||
|
schema:
|
||||||
|
description: 修改传感器的 Modbus 从站地址
|
||||||
|
properties:
|
||||||
|
goal:
|
||||||
|
properties:
|
||||||
|
new_id:
|
||||||
|
description: 新的从站地址 (1-254)
|
||||||
|
maximum: 254
|
||||||
|
minimum: 1
|
||||||
|
type: integer
|
||||||
|
required:
|
||||||
|
- new_id
|
||||||
|
type: object
|
||||||
|
type: UniLabJsonCommand
|
||||||
|
auto-factory_reset:
|
||||||
|
goal: {}
|
||||||
|
handles: {}
|
||||||
|
schema:
|
||||||
|
description: 恢复出厂设置 (地址重置为 01)
|
||||||
|
properties:
|
||||||
|
goal:
|
||||||
|
type: object
|
||||||
|
type: UniLabJsonCommand
|
||||||
|
auto-read_level:
|
||||||
|
goal: {}
|
||||||
|
handles: {}
|
||||||
|
schema:
|
||||||
|
description: 直接读取当前液位及信号强度
|
||||||
|
properties:
|
||||||
|
goal:
|
||||||
|
type: object
|
||||||
|
type: object
|
||||||
|
type: UniLabJsonCommand
|
||||||
|
auto-set_threshold:
|
||||||
|
goal:
|
||||||
|
threshold: 300
|
||||||
|
handles: {}
|
||||||
|
schema:
|
||||||
|
description: 设置液位判定阈值
|
||||||
|
properties:
|
||||||
|
goal:
|
||||||
|
properties:
|
||||||
|
threshold:
|
||||||
|
type: integer
|
||||||
|
required:
|
||||||
|
- threshold
|
||||||
|
type: object
|
||||||
|
type: UniLabJsonCommand
|
||||||
|
auto-wait_for_liquid:
|
||||||
|
goal:
|
||||||
|
target_state: true
|
||||||
|
timeout: 120
|
||||||
|
handles: {}
|
||||||
|
schema:
|
||||||
|
description: 实时检测电导率(RSSI)并等待用户指定的状态
|
||||||
|
properties:
|
||||||
|
goal:
|
||||||
|
properties:
|
||||||
|
target_state:
|
||||||
|
default: true
|
||||||
|
description: 目标状态 (True=有液, False=无液)
|
||||||
|
type: boolean
|
||||||
|
timeout:
|
||||||
|
default: 120
|
||||||
|
description: 超时时间 (秒)
|
||||||
|
required:
|
||||||
|
- target_state
|
||||||
|
type: object
|
||||||
|
type: UniLabJsonCommand
|
||||||
|
auto-wait_level:
|
||||||
|
goal:
|
||||||
|
level: true
|
||||||
|
timeout: 10
|
||||||
|
handles: {}
|
||||||
|
schema:
|
||||||
|
description: 等待液位达到目标状态
|
||||||
|
properties:
|
||||||
|
goal:
|
||||||
|
properties:
|
||||||
|
level:
|
||||||
|
type: boolean
|
||||||
|
timeout:
|
||||||
|
type: number
|
||||||
|
required:
|
||||||
|
- level
|
||||||
|
type: object
|
||||||
|
type: UniLabJsonCommand
|
||||||
|
module: unilabos.devices.separator.xkc_sensor:XKCSensorDriver
|
||||||
|
status_types:
|
||||||
|
level: bool
|
||||||
|
rssi: int
|
||||||
|
type: python
|
||||||
|
config_info: []
|
||||||
|
description: XKC RS485 非接触式液位传感器 (Modbus RTU)
|
||||||
|
handles: []
|
||||||
|
icon: ''
|
||||||
|
init_param_schema:
|
||||||
|
config:
|
||||||
|
properties:
|
||||||
|
baudrate:
|
||||||
|
default: 9600
|
||||||
|
type: integer
|
||||||
|
debug:
|
||||||
|
default: false
|
||||||
|
type: boolean
|
||||||
|
device_id:
|
||||||
|
default: 1
|
||||||
|
type: integer
|
||||||
|
port:
|
||||||
|
type: string
|
||||||
|
threshold:
|
||||||
|
default: 300
|
||||||
|
type: integer
|
||||||
|
timeout:
|
||||||
|
default: 3.0
|
||||||
|
type: number
|
||||||
|
required:
|
||||||
|
- port
|
||||||
|
type: object
|
||||||
|
version: 1.0.0
|
||||||
@@ -5835,25 +5835,6 @@ virtual_workbench:
|
|||||||
- material_number
|
- material_number
|
||||||
type: object
|
type: object
|
||||||
result:
|
result:
|
||||||
$defs:
|
|
||||||
LabSample:
|
|
||||||
properties:
|
|
||||||
extra:
|
|
||||||
additionalProperties: true
|
|
||||||
title: Extra
|
|
||||||
type: object
|
|
||||||
oss_path:
|
|
||||||
title: Oss Path
|
|
||||||
type: string
|
|
||||||
sample_uuid:
|
|
||||||
title: Sample Uuid
|
|
||||||
type: string
|
|
||||||
required:
|
|
||||||
- sample_uuid
|
|
||||||
- oss_path
|
|
||||||
- extra
|
|
||||||
title: LabSample
|
|
||||||
type: object
|
|
||||||
description: move_to_heating_station 返回类型
|
description: move_to_heating_station 返回类型
|
||||||
properties:
|
properties:
|
||||||
material_id:
|
material_id:
|
||||||
@@ -5872,18 +5853,12 @@ virtual_workbench:
|
|||||||
success:
|
success:
|
||||||
title: Success
|
title: Success
|
||||||
type: boolean
|
type: boolean
|
||||||
unilabos_samples:
|
|
||||||
items:
|
|
||||||
$ref: '#/$defs/LabSample'
|
|
||||||
title: Unilabos Samples
|
|
||||||
type: array
|
|
||||||
required:
|
required:
|
||||||
- success
|
- success
|
||||||
- station_id
|
- station_id
|
||||||
- material_id
|
- material_id
|
||||||
- material_number
|
- material_number
|
||||||
- message
|
- message
|
||||||
- unilabos_samples
|
|
||||||
title: MoveToHeatingStationResult
|
title: MoveToHeatingStationResult
|
||||||
type: object
|
type: object
|
||||||
required:
|
required:
|
||||||
@@ -5928,25 +5903,6 @@ virtual_workbench:
|
|||||||
- material_number
|
- material_number
|
||||||
type: object
|
type: object
|
||||||
result:
|
result:
|
||||||
$defs:
|
|
||||||
LabSample:
|
|
||||||
properties:
|
|
||||||
extra:
|
|
||||||
additionalProperties: true
|
|
||||||
title: Extra
|
|
||||||
type: object
|
|
||||||
oss_path:
|
|
||||||
title: Oss Path
|
|
||||||
type: string
|
|
||||||
sample_uuid:
|
|
||||||
title: Sample Uuid
|
|
||||||
type: string
|
|
||||||
required:
|
|
||||||
- sample_uuid
|
|
||||||
- oss_path
|
|
||||||
- extra
|
|
||||||
title: LabSample
|
|
||||||
type: object
|
|
||||||
description: move_to_output 返回类型
|
description: move_to_output 返回类型
|
||||||
properties:
|
properties:
|
||||||
material_id:
|
material_id:
|
||||||
@@ -5958,16 +5914,10 @@ virtual_workbench:
|
|||||||
success:
|
success:
|
||||||
title: Success
|
title: Success
|
||||||
type: boolean
|
type: boolean
|
||||||
unilabos_samples:
|
|
||||||
items:
|
|
||||||
$ref: '#/$defs/LabSample'
|
|
||||||
title: Unilabos Samples
|
|
||||||
type: array
|
|
||||||
required:
|
required:
|
||||||
- success
|
- success
|
||||||
- station_id
|
- station_id
|
||||||
- material_id
|
- material_id
|
||||||
- unilabos_samples
|
|
||||||
title: MoveToOutputResult
|
title: MoveToOutputResult
|
||||||
type: object
|
type: object
|
||||||
required:
|
required:
|
||||||
@@ -6022,25 +5972,6 @@ virtual_workbench:
|
|||||||
required: []
|
required: []
|
||||||
type: object
|
type: object
|
||||||
result:
|
result:
|
||||||
$defs:
|
|
||||||
LabSample:
|
|
||||||
properties:
|
|
||||||
extra:
|
|
||||||
additionalProperties: true
|
|
||||||
title: Extra
|
|
||||||
type: object
|
|
||||||
oss_path:
|
|
||||||
title: Oss Path
|
|
||||||
type: string
|
|
||||||
sample_uuid:
|
|
||||||
title: Sample Uuid
|
|
||||||
type: string
|
|
||||||
required:
|
|
||||||
- sample_uuid
|
|
||||||
- oss_path
|
|
||||||
- extra
|
|
||||||
title: LabSample
|
|
||||||
type: object
|
|
||||||
description: prepare_materials 返回类型 - 批量准备物料
|
description: prepare_materials 返回类型 - 批量准备物料
|
||||||
properties:
|
properties:
|
||||||
count:
|
count:
|
||||||
@@ -6067,11 +5998,6 @@ virtual_workbench:
|
|||||||
success:
|
success:
|
||||||
title: Success
|
title: Success
|
||||||
type: boolean
|
type: boolean
|
||||||
unilabos_samples:
|
|
||||||
items:
|
|
||||||
$ref: '#/$defs/LabSample'
|
|
||||||
title: Unilabos Samples
|
|
||||||
type: array
|
|
||||||
required:
|
required:
|
||||||
- success
|
- success
|
||||||
- count
|
- count
|
||||||
@@ -6081,7 +6007,6 @@ virtual_workbench:
|
|||||||
- material_4
|
- material_4
|
||||||
- material_5
|
- material_5
|
||||||
- message
|
- message
|
||||||
- unilabos_samples
|
|
||||||
title: PrepareMaterialsResult
|
title: PrepareMaterialsResult
|
||||||
type: object
|
type: object
|
||||||
required:
|
required:
|
||||||
@@ -6137,25 +6062,6 @@ virtual_workbench:
|
|||||||
- material_number
|
- material_number
|
||||||
type: object
|
type: object
|
||||||
result:
|
result:
|
||||||
$defs:
|
|
||||||
LabSample:
|
|
||||||
properties:
|
|
||||||
extra:
|
|
||||||
additionalProperties: true
|
|
||||||
title: Extra
|
|
||||||
type: object
|
|
||||||
oss_path:
|
|
||||||
title: Oss Path
|
|
||||||
type: string
|
|
||||||
sample_uuid:
|
|
||||||
title: Sample Uuid
|
|
||||||
type: string
|
|
||||||
required:
|
|
||||||
- sample_uuid
|
|
||||||
- oss_path
|
|
||||||
- extra
|
|
||||||
title: LabSample
|
|
||||||
type: object
|
|
||||||
description: start_heating 返回类型
|
description: start_heating 返回类型
|
||||||
properties:
|
properties:
|
||||||
material_id:
|
material_id:
|
||||||
@@ -6173,18 +6079,12 @@ virtual_workbench:
|
|||||||
success:
|
success:
|
||||||
title: Success
|
title: Success
|
||||||
type: boolean
|
type: boolean
|
||||||
unilabos_samples:
|
|
||||||
items:
|
|
||||||
$ref: '#/$defs/LabSample'
|
|
||||||
title: Unilabos Samples
|
|
||||||
type: array
|
|
||||||
required:
|
required:
|
||||||
- success
|
- success
|
||||||
- station_id
|
- station_id
|
||||||
- material_id
|
- material_id
|
||||||
- material_number
|
- material_number
|
||||||
- message
|
- message
|
||||||
- unilabos_samples
|
|
||||||
title: StartHeatingResult
|
title: StartHeatingResult
|
||||||
type: object
|
type: object
|
||||||
required:
|
required:
|
||||||
|
|||||||
@@ -151,40 +151,12 @@ def canonicalize_links_ports(links: List[Dict[str, Any]], resource_tree_set: Res
|
|||||||
"""
|
"""
|
||||||
# 构建 id 到 uuid 的映射
|
# 构建 id 到 uuid 的映射
|
||||||
id_to_uuid: Dict[str, str] = {}
|
id_to_uuid: Dict[str, str] = {}
|
||||||
uuid_to_id: Dict[str, str] = {}
|
|
||||||
for node in resource_tree_set.all_nodes:
|
for node in resource_tree_set.all_nodes:
|
||||||
id_to_uuid[node.res_content.id] = node.res_content.uuid
|
id_to_uuid[node.res_content.id] = node.res_content.uuid
|
||||||
uuid_to_id[node.res_content.uuid] = node.res_content.id
|
|
||||||
|
|
||||||
# 第三遍处理:为每个 link 添加 source_uuid 和 target_uuid
|
|
||||||
for link in links:
|
|
||||||
source_id = link.get("source")
|
|
||||||
target_id = link.get("target")
|
|
||||||
|
|
||||||
# 添加 source_uuid
|
|
||||||
if source_id and source_id in id_to_uuid:
|
|
||||||
link["source_uuid"] = id_to_uuid[source_id]
|
|
||||||
|
|
||||||
# 添加 target_uuid
|
|
||||||
if target_id and target_id in id_to_uuid:
|
|
||||||
link["target_uuid"] = id_to_uuid[target_id]
|
|
||||||
|
|
||||||
source_uuid = link.get("source_uuid")
|
|
||||||
target_uuid = link.get("target_uuid")
|
|
||||||
|
|
||||||
# 添加 source_uuid
|
|
||||||
if source_uuid and source_uuid in uuid_to_id:
|
|
||||||
link["source"] = uuid_to_id[source_uuid]
|
|
||||||
|
|
||||||
# 添加 target_uuid
|
|
||||||
if target_uuid and target_uuid in uuid_to_id:
|
|
||||||
link["target"] = uuid_to_id[target_uuid]
|
|
||||||
|
|
||||||
# 第一遍处理:将字符串类型的port转换为字典格式
|
# 第一遍处理:将字符串类型的port转换为字典格式
|
||||||
for link in links:
|
for link in links:
|
||||||
port = link.get("port")
|
port = link.get("port")
|
||||||
if port is None:
|
|
||||||
continue
|
|
||||||
if link.get("type", "physical") == "physical":
|
if link.get("type", "physical") == "physical":
|
||||||
link["type"] = "fluid"
|
link["type"] = "fluid"
|
||||||
if isinstance(port, int):
|
if isinstance(port, int):
|
||||||
@@ -207,15 +179,13 @@ def canonicalize_links_ports(links: List[Dict[str, Any]], resource_tree_set: Res
|
|||||||
link["port"] = {link["source"]: None, link["target"]: None}
|
link["port"] = {link["source"]: None, link["target"]: None}
|
||||||
|
|
||||||
# 构建边字典,键为(source节点, target节点),值为对应的port信息
|
# 构建边字典,键为(source节点, target节点),值为对应的port信息
|
||||||
edges = {(link["source"], link["target"]): link["port"] for link in links if link.get("port")}
|
edges = {(link["source"], link["target"]): link["port"] for link in links}
|
||||||
|
|
||||||
# 第二遍处理:填充反向边的dest信息
|
# 第二遍处理:填充反向边的dest信息
|
||||||
delete_reverses = []
|
delete_reverses = []
|
||||||
for i, link in enumerate(links):
|
for i, link in enumerate(links):
|
||||||
s, t = link["source"], link["target"]
|
s, t = link["source"], link["target"]
|
||||||
current_port = link.get("port")
|
current_port = link["port"]
|
||||||
if current_port is None:
|
|
||||||
continue
|
|
||||||
if current_port.get(t) is None:
|
if current_port.get(t) is None:
|
||||||
reverse_key = (t, s)
|
reverse_key = (t, s)
|
||||||
reverse_port = edges.get(reverse_key)
|
reverse_port = edges.get(reverse_key)
|
||||||
@@ -230,6 +200,20 @@ def canonicalize_links_ports(links: List[Dict[str, Any]], resource_tree_set: Res
|
|||||||
current_port[t] = current_port[s]
|
current_port[t] = current_port[s]
|
||||||
# 删除已被使用反向端口信息的反向边
|
# 删除已被使用反向端口信息的反向边
|
||||||
standardized_links = [link for i, link in enumerate(links) if i not in delete_reverses]
|
standardized_links = [link for i, link in enumerate(links) if i not in delete_reverses]
|
||||||
|
|
||||||
|
# 第三遍处理:为每个 link 添加 source_uuid 和 target_uuid
|
||||||
|
for link in standardized_links:
|
||||||
|
source_id = link.get("source")
|
||||||
|
target_id = link.get("target")
|
||||||
|
|
||||||
|
# 添加 source_uuid
|
||||||
|
if source_id and source_id in id_to_uuid:
|
||||||
|
link["source_uuid"] = id_to_uuid[source_id]
|
||||||
|
|
||||||
|
# 添加 target_uuid
|
||||||
|
if target_id and target_id in id_to_uuid:
|
||||||
|
link["target_uuid"] = id_to_uuid[target_id]
|
||||||
|
|
||||||
return standardized_links
|
return standardized_links
|
||||||
|
|
||||||
|
|
||||||
@@ -300,8 +284,6 @@ def modify_to_backend_format(data: list[dict[str, Any]]) -> list[dict[str, Any]]
|
|||||||
edge["sourceHandle"] = port[source]
|
edge["sourceHandle"] = port[source]
|
||||||
elif "source_port" in edge:
|
elif "source_port" in edge:
|
||||||
edge["sourceHandle"] = edge.pop("source_port")
|
edge["sourceHandle"] = edge.pop("source_port")
|
||||||
elif "source_handle" in edge:
|
|
||||||
edge["sourceHandle"] = edge.pop("source_handle")
|
|
||||||
else:
|
else:
|
||||||
typ = edge.get("type")
|
typ = edge.get("type")
|
||||||
if typ == "communication":
|
if typ == "communication":
|
||||||
@@ -310,8 +292,6 @@ def modify_to_backend_format(data: list[dict[str, Any]]) -> list[dict[str, Any]]
|
|||||||
edge["targetHandle"] = port[target]
|
edge["targetHandle"] = port[target]
|
||||||
elif "target_port" in edge:
|
elif "target_port" in edge:
|
||||||
edge["targetHandle"] = edge.pop("target_port")
|
edge["targetHandle"] = edge.pop("target_port")
|
||||||
elif "target_handle" in edge:
|
|
||||||
edge["targetHandle"] = edge.pop("target_handle")
|
|
||||||
else:
|
else:
|
||||||
typ = edge.get("type")
|
typ = edge.get("type")
|
||||||
if typ == "communication":
|
if typ == "communication":
|
||||||
|
|||||||
@@ -5,8 +5,6 @@ from pydantic import BaseModel, field_serializer, field_validator, ValidationErr
|
|||||||
from pydantic import Field
|
from pydantic import Field
|
||||||
from typing import List, Tuple, Any, Dict, Literal, Optional, cast, TYPE_CHECKING, Union
|
from typing import List, Tuple, Any, Dict, Literal, Optional, cast, TYPE_CHECKING, Union
|
||||||
|
|
||||||
from typing_extensions import TypedDict
|
|
||||||
|
|
||||||
from unilabos.resources.plr_additional_res_reg import register
|
from unilabos.resources.plr_additional_res_reg import register
|
||||||
from unilabos.utils.log import logger
|
from unilabos.utils.log import logger
|
||||||
|
|
||||||
@@ -16,26 +14,6 @@ if TYPE_CHECKING:
|
|||||||
|
|
||||||
|
|
||||||
EXTRA_CLASS = "unilabos_resource_class"
|
EXTRA_CLASS = "unilabos_resource_class"
|
||||||
EXTRA_SAMPLE_UUID = "sample_uuid"
|
|
||||||
EXTRA_UNILABOS_SAMPLE_UUID = "unilabos_sample_uuid"
|
|
||||||
|
|
||||||
# 函数参数名常量 - 用于自动注入 sample_uuids 列表
|
|
||||||
PARAM_SAMPLE_UUIDS = "sample_uuids"
|
|
||||||
|
|
||||||
# JSON Command 中的系统参数字段名
|
|
||||||
JSON_UNILABOS_PARAM = "unilabos_param"
|
|
||||||
|
|
||||||
# 返回值中的 samples 字段名
|
|
||||||
RETURN_UNILABOS_SAMPLES = "unilabos_samples"
|
|
||||||
|
|
||||||
# sample_uuids 参数类型 (用于 virtual bench 等设备添加 sample_uuids 参数)
|
|
||||||
SampleUUIDsType = Dict[str, Optional["PLRResource"]]
|
|
||||||
|
|
||||||
|
|
||||||
class LabSample(TypedDict):
|
|
||||||
sample_uuid: str
|
|
||||||
oss_path: str
|
|
||||||
extra: Dict[str, Any]
|
|
||||||
|
|
||||||
|
|
||||||
class ResourceDictPositionSize(BaseModel):
|
class ResourceDictPositionSize(BaseModel):
|
||||||
@@ -551,7 +529,6 @@ class ResourceTreeSet(object):
|
|||||||
plr_resource = sub_cls.deserialize(plr_dict, allow_marshal=True)
|
plr_resource = sub_cls.deserialize(plr_dict, allow_marshal=True)
|
||||||
from pylabrobot.resources import Coordinate
|
from pylabrobot.resources import Coordinate
|
||||||
from pylabrobot.serializer import deserialize
|
from pylabrobot.serializer import deserialize
|
||||||
|
|
||||||
location = cast(Coordinate, deserialize(plr_dict["location"]))
|
location = cast(Coordinate, deserialize(plr_dict["location"]))
|
||||||
plr_resource.location = location
|
plr_resource.location = location
|
||||||
plr_resource.load_all_state(all_states)
|
plr_resource.load_all_state(all_states)
|
||||||
|
|||||||
@@ -4,20 +4,8 @@ import json
|
|||||||
import threading
|
import threading
|
||||||
import time
|
import time
|
||||||
import traceback
|
import traceback
|
||||||
from typing import (
|
from typing import get_type_hints, TypeVar, Generic, Dict, Any, Type, TypedDict, Optional, List, TYPE_CHECKING, Union, \
|
||||||
get_type_hints,
|
Tuple
|
||||||
TypeVar,
|
|
||||||
Generic,
|
|
||||||
Dict,
|
|
||||||
Any,
|
|
||||||
Type,
|
|
||||||
TypedDict,
|
|
||||||
Optional,
|
|
||||||
List,
|
|
||||||
TYPE_CHECKING,
|
|
||||||
Union,
|
|
||||||
Tuple,
|
|
||||||
)
|
|
||||||
|
|
||||||
from concurrent.futures import ThreadPoolExecutor
|
from concurrent.futures import ThreadPoolExecutor
|
||||||
import asyncio
|
import asyncio
|
||||||
@@ -60,9 +48,6 @@ from unilabos.resources.resource_tracker import (
|
|||||||
ResourceTreeSet,
|
ResourceTreeSet,
|
||||||
ResourceTreeInstance,
|
ResourceTreeInstance,
|
||||||
ResourceDictInstance,
|
ResourceDictInstance,
|
||||||
EXTRA_SAMPLE_UUID,
|
|
||||||
PARAM_SAMPLE_UUIDS,
|
|
||||||
JSON_UNILABOS_PARAM,
|
|
||||||
)
|
)
|
||||||
from unilabos.ros.utils.driver_creator import WorkstationNodeCreator, PyLabRobotCreator, DeviceClassCreator
|
from unilabos.ros.utils.driver_creator import WorkstationNodeCreator, PyLabRobotCreator, DeviceClassCreator
|
||||||
from rclpy.task import Task, Future
|
from rclpy.task import Task, Future
|
||||||
@@ -231,15 +216,14 @@ class PropertyPublisher:
|
|||||||
|
|
||||||
def publish_property(self):
|
def publish_property(self):
|
||||||
try:
|
try:
|
||||||
# self.node.lab_logger().trace(f"【.publish_property】开始发布属性: {self.name}")
|
self.node.lab_logger().trace(f"【.publish_property】开始发布属性: {self.name}")
|
||||||
value = self.get_property()
|
value = self.get_property()
|
||||||
if self.print_publish:
|
if self.print_publish:
|
||||||
pass
|
self.node.lab_logger().trace(f"【.publish_property】发布 {self.msg_type}: {value}")
|
||||||
# self.node.lab_logger().trace(f"【.publish_property】发布 {self.msg_type}: {value}")
|
|
||||||
if value is not None:
|
if value is not None:
|
||||||
msg = convert_to_ros_msg(self.msg_type, value)
|
msg = convert_to_ros_msg(self.msg_type, value)
|
||||||
self.publisher_.publish(msg)
|
self.publisher_.publish(msg)
|
||||||
# self.node.lab_logger().trace(f"【.publish_property】属性 {self.name} 发布成功")
|
self.node.lab_logger().trace(f"【.publish_property】属性 {self.name} 发布成功")
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
self.node.lab_logger().error(
|
self.node.lab_logger().error(
|
||||||
f"【.publish_property】发布属性 {self.publisher_.topic} 出错: {str(e)}\n{traceback.format_exc()}"
|
f"【.publish_property】发布属性 {self.publisher_.topic} 出错: {str(e)}\n{traceback.format_exc()}"
|
||||||
@@ -377,7 +361,6 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
from pylabrobot.resources.deck import Deck
|
from pylabrobot.resources.deck import Deck
|
||||||
from pylabrobot.resources import Coordinate
|
from pylabrobot.resources import Coordinate
|
||||||
from pylabrobot.resources import Plate
|
from pylabrobot.resources import Plate
|
||||||
|
|
||||||
# 物料传输到对应的node节点
|
# 物料传输到对应的node节点
|
||||||
client = self._resource_clients["c2s_update_resource_tree"]
|
client = self._resource_clients["c2s_update_resource_tree"]
|
||||||
request = SerialCommand.Request()
|
request = SerialCommand.Request()
|
||||||
@@ -405,27 +388,33 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
rts: ResourceTreeSet = ResourceTreeSet.from_raw_dict_list(input_resources)
|
rts: ResourceTreeSet = ResourceTreeSet.from_raw_dict_list(input_resources)
|
||||||
parent_resource = None
|
parent_resource = None
|
||||||
if bind_parent_id != self.node_name:
|
if bind_parent_id != self.node_name:
|
||||||
parent_resource = self.resource_tracker.figure_resource({"name": bind_parent_id})
|
parent_resource = self.resource_tracker.figure_resource(
|
||||||
|
{"name": bind_parent_id}
|
||||||
|
)
|
||||||
for r in rts.root_nodes:
|
for r in rts.root_nodes:
|
||||||
# noinspection PyUnresolvedReferences
|
# noinspection PyUnresolvedReferences
|
||||||
r.res_content.parent_uuid = parent_resource.unilabos_uuid
|
r.res_content.parent_uuid = parent_resource.unilabos_uuid
|
||||||
else:
|
else:
|
||||||
for r in rts.root_nodes:
|
for r in rts.root_nodes:
|
||||||
r.res_content.parent_uuid = self.uuid
|
r.res_content.parent_uuid = self.uuid
|
||||||
rts_plr_instances = rts.to_plr_resources()
|
|
||||||
if len(rts.root_nodes) == 1 and isinstance(rts_plr_instances[0], RegularContainer):
|
if len(LIQUID_INPUT_SLOT) and LIQUID_INPUT_SLOT[0] == -1 and len(rts.root_nodes) == 1 and isinstance(rts.root_nodes[0], RegularContainer):
|
||||||
# noinspection PyTypeChecker
|
# noinspection PyTypeChecker
|
||||||
container_instance: RegularContainer = rts_plr_instances[0]
|
container_instance: RegularContainer = rts.root_nodes[0]
|
||||||
found_resources = self.resource_tracker.figure_resource({"name": container_instance.name}, try_mode=True)
|
found_resources = self.resource_tracker.figure_resource(
|
||||||
|
{"id": container_instance.name}, try_mode=True
|
||||||
|
)
|
||||||
if not len(found_resources):
|
if not len(found_resources):
|
||||||
self.resource_tracker.add_resource(container_instance)
|
self.resource_tracker.add_resource(container_instance)
|
||||||
logger.info(f"添加物料{container_instance.name}到资源跟踪器")
|
logger.info(f"添加物料{container_instance.name}到资源跟踪器")
|
||||||
else:
|
else:
|
||||||
assert len(found_resources) == 1, f"找到多个同名物料: {container_instance.name}, 请检查物料系统"
|
assert (
|
||||||
|
len(found_resources) == 1
|
||||||
|
), f"找到多个同名物料: {container_instance.name}, 请检查物料系统"
|
||||||
found_resource = found_resources[0]
|
found_resource = found_resources[0]
|
||||||
if isinstance(found_resource, RegularContainer):
|
if isinstance(found_resource, RegularContainer):
|
||||||
logger.info(f"更新物料{container_instance.name}的数据{found_resource.state}")
|
logger.info(f"更新物料{container_instance.name}的数据{found_resource.state}")
|
||||||
found_resource.state.update(container_instance.state)
|
found_resource.state.update(json.loads(container_instance.state))
|
||||||
elif isinstance(found_resource, dict):
|
elif isinstance(found_resource, dict):
|
||||||
raise ValueError("已不支持 字典 版本的RegularContainer")
|
raise ValueError("已不支持 字典 版本的RegularContainer")
|
||||||
else:
|
else:
|
||||||
@@ -433,16 +422,14 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
f"更新物料{container_instance.name}出现不支持的数据类型{type(found_resource)} {found_resource}"
|
f"更新物料{container_instance.name}出现不支持的数据类型{type(found_resource)} {found_resource}"
|
||||||
)
|
)
|
||||||
# noinspection PyUnresolvedReferences
|
# noinspection PyUnresolvedReferences
|
||||||
request.command = json.dumps(
|
request.command = json.dumps({
|
||||||
{
|
"action": "add",
|
||||||
"action": "add",
|
"data": {
|
||||||
"data": {
|
"data": rts.dump(),
|
||||||
"data": rts.dump(),
|
"mount_uuid": parent_resource.unilabos_uuid if parent_resource is not None else "",
|
||||||
"mount_uuid": parent_resource.unilabos_uuid if parent_resource is not None else self.uuid,
|
"first_add": False,
|
||||||
"first_add": False,
|
},
|
||||||
},
|
})
|
||||||
}
|
|
||||||
)
|
|
||||||
tree_response: SerialCommand.Response = await client.call_async(request)
|
tree_response: SerialCommand.Response = await client.call_async(request)
|
||||||
uuid_maps = json.loads(tree_response.response)
|
uuid_maps = json.loads(tree_response.response)
|
||||||
plr_instances = rts.to_plr_resources()
|
plr_instances = rts.to_plr_resources()
|
||||||
@@ -484,9 +471,7 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
if len(ADD_LIQUID_TYPE) == 1 and len(LIQUID_VOLUME) == 1 and len(LIQUID_INPUT_SLOT) > 1:
|
if len(ADD_LIQUID_TYPE) == 1 and len(LIQUID_VOLUME) == 1 and len(LIQUID_INPUT_SLOT) > 1:
|
||||||
ADD_LIQUID_TYPE = ADD_LIQUID_TYPE * len(LIQUID_INPUT_SLOT)
|
ADD_LIQUID_TYPE = ADD_LIQUID_TYPE * len(LIQUID_INPUT_SLOT)
|
||||||
LIQUID_VOLUME = LIQUID_VOLUME * len(LIQUID_INPUT_SLOT)
|
LIQUID_VOLUME = LIQUID_VOLUME * len(LIQUID_INPUT_SLOT)
|
||||||
self.lab_logger().warning(
|
self.lab_logger().warning(f"增加液体资源时,数量为1,自动补全为 {len(LIQUID_INPUT_SLOT)} 个")
|
||||||
f"增加液体资源时,数量为1,自动补全为 {len(LIQUID_INPUT_SLOT)} 个"
|
|
||||||
)
|
|
||||||
for liquid_type, liquid_volume, liquid_input_slot in zip(
|
for liquid_type, liquid_volume, liquid_input_slot in zip(
|
||||||
ADD_LIQUID_TYPE, LIQUID_VOLUME, LIQUID_INPUT_SLOT
|
ADD_LIQUID_TYPE, LIQUID_VOLUME, LIQUID_INPUT_SLOT
|
||||||
):
|
):
|
||||||
@@ -505,15 +490,9 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
input_wells = []
|
input_wells = []
|
||||||
for r in LIQUID_INPUT_SLOT:
|
for r in LIQUID_INPUT_SLOT:
|
||||||
input_wells.append(plr_instance.children[r])
|
input_wells.append(plr_instance.children[r])
|
||||||
final_response["liquid_input_resource_tree"] = ResourceTreeSet.from_plr_resources(
|
final_response["liquid_input_resource_tree"] = ResourceTreeSet.from_plr_resources(input_wells).dump()
|
||||||
input_wells
|
|
||||||
).dump()
|
|
||||||
res.response = json.dumps(final_response)
|
res.response = json.dumps(final_response)
|
||||||
if (
|
if issubclass(parent_resource.__class__, Deck) and hasattr(parent_resource, "assign_child_at_slot") and "slot" in other_calling_param:
|
||||||
issubclass(parent_resource.__class__, Deck)
|
|
||||||
and hasattr(parent_resource, "assign_child_at_slot")
|
|
||||||
and "slot" in other_calling_param
|
|
||||||
):
|
|
||||||
other_calling_param["slot"] = int(other_calling_param["slot"])
|
other_calling_param["slot"] = int(other_calling_param["slot"])
|
||||||
parent_resource.assign_child_at_slot(plr_instance, **other_calling_param)
|
parent_resource.assign_child_at_slot(plr_instance, **other_calling_param)
|
||||||
else:
|
else:
|
||||||
@@ -528,16 +507,14 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
rts_with_parent = ResourceTreeSet.from_plr_resources([parent_resource])
|
rts_with_parent = ResourceTreeSet.from_plr_resources([parent_resource])
|
||||||
if rts_with_parent.root_nodes[0].res_content.uuid_parent is None:
|
if rts_with_parent.root_nodes[0].res_content.uuid_parent is None:
|
||||||
rts_with_parent.root_nodes[0].res_content.parent_uuid = self.uuid
|
rts_with_parent.root_nodes[0].res_content.parent_uuid = self.uuid
|
||||||
request.command = json.dumps(
|
request.command = json.dumps({
|
||||||
{
|
"action": "add",
|
||||||
"action": "add",
|
"data": {
|
||||||
"data": {
|
"data": rts_with_parent.dump(),
|
||||||
"data": rts_with_parent.dump(),
|
"mount_uuid": rts_with_parent.root_nodes[0].res_content.uuid_parent,
|
||||||
"mount_uuid": rts_with_parent.root_nodes[0].res_content.uuid_parent,
|
"first_add": False,
|
||||||
"first_add": False,
|
},
|
||||||
},
|
})
|
||||||
}
|
|
||||||
)
|
|
||||||
tree_response: SerialCommand.Response = await client.call_async(request)
|
tree_response: SerialCommand.Response = await client.call_async(request)
|
||||||
uuid_maps = json.loads(tree_response.response)
|
uuid_maps = json.loads(tree_response.response)
|
||||||
self.resource_tracker.loop_update_uuid(input_resources, uuid_maps)
|
self.resource_tracker.loop_update_uuid(input_resources, uuid_maps)
|
||||||
@@ -834,9 +811,7 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
}
|
}
|
||||||
|
|
||||||
def _handle_update(
|
def _handle_update(
|
||||||
plr_resources: List[Union[ResourcePLR, ResourceDictInstance]],
|
plr_resources: List[Union[ResourcePLR, ResourceDictInstance]], tree_set: ResourceTreeSet, additional_add_params: Dict[str, Any]
|
||||||
tree_set: ResourceTreeSet,
|
|
||||||
additional_add_params: Dict[str, Any],
|
|
||||||
) -> Tuple[Dict[str, Any], List[ResourcePLR]]:
|
) -> Tuple[Dict[str, Any], List[ResourcePLR]]:
|
||||||
"""
|
"""
|
||||||
处理资源更新操作的内部函数
|
处理资源更新操作的内部函数
|
||||||
@@ -861,10 +836,7 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
original_parent_resource = original_instance.parent
|
original_parent_resource = original_instance.parent
|
||||||
original_parent_resource_uuid = getattr(original_parent_resource, "unilabos_uuid", None)
|
original_parent_resource_uuid = getattr(original_parent_resource, "unilabos_uuid", None)
|
||||||
target_parent_resource_uuid = tree.root_node.res_content.uuid_parent
|
target_parent_resource_uuid = tree.root_node.res_content.uuid_parent
|
||||||
not_same_parent = (
|
not_same_parent = original_parent_resource_uuid != target_parent_resource_uuid and original_parent_resource is not None
|
||||||
original_parent_resource_uuid != target_parent_resource_uuid
|
|
||||||
and original_parent_resource is not None
|
|
||||||
)
|
|
||||||
old_name = original_instance.name
|
old_name = original_instance.name
|
||||||
new_name = plr_resource.name
|
new_name = plr_resource.name
|
||||||
parent_appended = False
|
parent_appended = False
|
||||||
@@ -900,16 +872,8 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
else:
|
else:
|
||||||
# 判断是否变更了resource_site,重新登记
|
# 判断是否变更了resource_site,重新登记
|
||||||
target_site = original_instance.unilabos_extra.get("update_resource_site")
|
target_site = original_instance.unilabos_extra.get("update_resource_site")
|
||||||
sites = (
|
sites = original_instance.parent.sites if original_instance.parent is not None and hasattr(original_instance.parent, "sites") else None
|
||||||
original_instance.parent.sites
|
site_names = list(original_instance.parent._ordering.keys()) if original_instance.parent is not None and hasattr(original_instance.parent, "sites") else []
|
||||||
if original_instance.parent is not None and hasattr(original_instance.parent, "sites")
|
|
||||||
else None
|
|
||||||
)
|
|
||||||
site_names = (
|
|
||||||
list(original_instance.parent._ordering.keys())
|
|
||||||
if original_instance.parent is not None and hasattr(original_instance.parent, "sites")
|
|
||||||
else []
|
|
||||||
)
|
|
||||||
if target_site is not None and sites is not None and site_names is not None:
|
if target_site is not None and sites is not None and site_names is not None:
|
||||||
site_index = sites.index(original_instance)
|
site_index = sites.index(original_instance)
|
||||||
site_name = site_names[site_index]
|
site_name = site_names[site_index]
|
||||||
@@ -946,7 +910,9 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
action = i.get("action") # remove, add, update
|
action = i.get("action") # remove, add, update
|
||||||
resources_uuid: List[str] = i.get("data") # 资源数据
|
resources_uuid: List[str] = i.get("data") # 资源数据
|
||||||
additional_add_params = i.get("additional_add_params", {}) # 额外参数
|
additional_add_params = i.get("additional_add_params", {}) # 额外参数
|
||||||
self.lab_logger().trace(f"[资源同步] 处理 {action}, " f"resources count: {len(resources_uuid)}")
|
self.lab_logger().trace(
|
||||||
|
f"[资源同步] 处理 {action}, " f"resources count: {len(resources_uuid)}"
|
||||||
|
)
|
||||||
tree_set = None
|
tree_set = None
|
||||||
if action in ["add", "update"]:
|
if action in ["add", "update"]:
|
||||||
tree_set = await self.get_resource(
|
tree_set = await self.get_resource(
|
||||||
@@ -973,13 +939,9 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
tree.root_node.res_content.parent_uuid = self.uuid
|
tree.root_node.res_content.parent_uuid = self.uuid
|
||||||
r = SerialCommand.Request()
|
r = SerialCommand.Request()
|
||||||
r.command = json.dumps(
|
r.command = json.dumps(
|
||||||
{"data": {"data": new_tree_set.dump()}, "action": "update"}
|
{"data": {"data": new_tree_set.dump()}, "action": "update"}) # 和Update Resource一致
|
||||||
) # 和Update Resource一致
|
|
||||||
response: SerialCommand_Response = await self._resource_clients[
|
response: SerialCommand_Response = await self._resource_clients[
|
||||||
"c2s_update_resource_tree"
|
"c2s_update_resource_tree"].call_async(r) # type: ignore
|
||||||
].call_async(
|
|
||||||
r
|
|
||||||
) # type: ignore
|
|
||||||
self.lab_logger().info(f"确认资源云端 Add 结果: {response.response}")
|
self.lab_logger().info(f"确认资源云端 Add 结果: {response.response}")
|
||||||
results.append(result)
|
results.append(result)
|
||||||
elif action == "update":
|
elif action == "update":
|
||||||
@@ -999,13 +961,9 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
tree.root_node.res_content.parent_uuid = self.uuid
|
tree.root_node.res_content.parent_uuid = self.uuid
|
||||||
r = SerialCommand.Request()
|
r = SerialCommand.Request()
|
||||||
r.command = json.dumps(
|
r.command = json.dumps(
|
||||||
{"data": {"data": new_tree_set.dump()}, "action": "update"}
|
{"data": {"data": new_tree_set.dump()}, "action": "update"}) # 和Update Resource一致
|
||||||
) # 和Update Resource一致
|
|
||||||
response: SerialCommand_Response = await self._resource_clients[
|
response: SerialCommand_Response = await self._resource_clients[
|
||||||
"c2s_update_resource_tree"
|
"c2s_update_resource_tree"].call_async(r) # type: ignore
|
||||||
].call_async(
|
|
||||||
r
|
|
||||||
) # type: ignore
|
|
||||||
self.lab_logger().info(f"确认资源云端 Update 结果: {response.response}")
|
self.lab_logger().info(f"确认资源云端 Update 结果: {response.response}")
|
||||||
results.append(result)
|
results.append(result)
|
||||||
elif action == "remove":
|
elif action == "remove":
|
||||||
@@ -1375,7 +1333,7 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
resource_id=resource_data["id"], with_children=True
|
resource_id=resource_data["id"], with_children=True
|
||||||
)
|
)
|
||||||
if "sample_id" in resource_data:
|
if "sample_id" in resource_data:
|
||||||
plr_resource.unilabos_extra[EXTRA_SAMPLE_UUID] = resource_data["sample_id"]
|
plr_resource.unilabos_extra["sample_uuid"] = resource_data["sample_id"]
|
||||||
queried_resources[idx] = plr_resource
|
queried_resources[idx] = plr_resource
|
||||||
else:
|
else:
|
||||||
uuid_indices.append((idx, unilabos_uuid, resource_data))
|
uuid_indices.append((idx, unilabos_uuid, resource_data))
|
||||||
@@ -1388,7 +1346,7 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
for i, (idx, _, resource_data) in enumerate(uuid_indices):
|
for i, (idx, _, resource_data) in enumerate(uuid_indices):
|
||||||
plr_resource = plr_resources[i]
|
plr_resource = plr_resources[i]
|
||||||
if "sample_id" in resource_data:
|
if "sample_id" in resource_data:
|
||||||
plr_resource.unilabos_extra[EXTRA_SAMPLE_UUID] = resource_data["sample_id"]
|
plr_resource.unilabos_extra["sample_uuid"] = resource_data["sample_id"]
|
||||||
queried_resources[idx] = plr_resource
|
queried_resources[idx] = plr_resource
|
||||||
|
|
||||||
self.lab_logger().debug(f"资源查询结果: 共 {len(queried_resources)} 个资源")
|
self.lab_logger().debug(f"资源查询结果: 共 {len(queried_resources)} 个资源")
|
||||||
@@ -1396,9 +1354,7 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
# 通过资源跟踪器获取本地实例
|
# 通过资源跟踪器获取本地实例
|
||||||
final_resources = queried_resources if is_sequence else queried_resources[0]
|
final_resources = queried_resources if is_sequence else queried_resources[0]
|
||||||
if not is_sequence:
|
if not is_sequence:
|
||||||
plr = self.resource_tracker.figure_resource(
|
plr = self.resource_tracker.figure_resource({"name": final_resources.name}, try_mode=False)
|
||||||
{"name": final_resources.name}, try_mode=False
|
|
||||||
)
|
|
||||||
# 保留unilabos_extra
|
# 保留unilabos_extra
|
||||||
if hasattr(final_resources, "unilabos_extra") and hasattr(plr, "unilabos_extra"):
|
if hasattr(final_resources, "unilabos_extra") and hasattr(plr, "unilabos_extra"):
|
||||||
plr.unilabos_extra = getattr(final_resources, "unilabos_extra", {}).copy()
|
plr.unilabos_extra = getattr(final_resources, "unilabos_extra", {}).copy()
|
||||||
@@ -1437,12 +1393,8 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
execution_success = True
|
execution_success = True
|
||||||
except Exception as _:
|
except Exception as _:
|
||||||
execution_error = traceback.format_exc()
|
execution_error = traceback.format_exc()
|
||||||
error(
|
error(f"异步任务 {ACTION.__name__} 报错了\n{traceback.format_exc()}\n原始输入:{str(action_kwargs)[:1000]}")
|
||||||
f"异步任务 {ACTION.__name__} 报错了\n{traceback.format_exc()}\n原始输入:{str(action_kwargs)[:1000]}"
|
trace(f"异步任务 {ACTION.__name__} 报错了\n{traceback.format_exc()}\n原始输入:{action_kwargs}")
|
||||||
)
|
|
||||||
trace(
|
|
||||||
f"异步任务 {ACTION.__name__} 报错了\n{traceback.format_exc()}\n原始输入:{action_kwargs}"
|
|
||||||
)
|
|
||||||
|
|
||||||
future = ROS2DeviceNode.run_async_func(ACTION, trace_error=False, **action_kwargs)
|
future = ROS2DeviceNode.run_async_func(ACTION, trace_error=False, **action_kwargs)
|
||||||
future.add_done_callback(_handle_future_exception)
|
future.add_done_callback(_handle_future_exception)
|
||||||
@@ -1462,11 +1414,9 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
except Exception as _:
|
except Exception as _:
|
||||||
execution_error = traceback.format_exc()
|
execution_error = traceback.format_exc()
|
||||||
error(
|
error(
|
||||||
f"同步任务 {ACTION.__name__} 报错了\n{traceback.format_exc()}\n原始输入:{str(action_kwargs)[:1000]}"
|
f"同步任务 {ACTION.__name__} 报错了\n{traceback.format_exc()}\n原始输入:{str(action_kwargs)[:1000]}")
|
||||||
)
|
|
||||||
trace(
|
trace(
|
||||||
f"同步任务 {ACTION.__name__} 报错了\n{traceback.format_exc()}\n原始输入:{action_kwargs}"
|
f"同步任务 {ACTION.__name__} 报错了\n{traceback.format_exc()}\n原始输入:{action_kwargs}")
|
||||||
)
|
|
||||||
|
|
||||||
future.add_done_callback(_handle_future_exception)
|
future.add_done_callback(_handle_future_exception)
|
||||||
|
|
||||||
@@ -1533,18 +1483,11 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
if isinstance(rs, list):
|
if isinstance(rs, list):
|
||||||
for r in rs:
|
for r in rs:
|
||||||
res = self.resource_tracker.parent_resource(r) # 获取 resource 对象
|
res = self.resource_tracker.parent_resource(r) # 获取 resource 对象
|
||||||
if res is None:
|
|
||||||
res = rs
|
|
||||||
if id(res) not in seen:
|
|
||||||
seen.add(id(res))
|
|
||||||
unique_resources.append(res)
|
|
||||||
else:
|
else:
|
||||||
res = self.resource_tracker.parent_resource(rs)
|
res = self.resource_tracker.parent_resource(rs)
|
||||||
if res is None:
|
if id(res) not in seen:
|
||||||
res = rs
|
seen.add(id(res))
|
||||||
if id(res) not in seen:
|
unique_resources.append(res)
|
||||||
seen.add(id(res))
|
|
||||||
unique_resources.append(res)
|
|
||||||
|
|
||||||
# 使用新的资源树接口
|
# 使用新的资源树接口
|
||||||
if unique_resources:
|
if unique_resources:
|
||||||
@@ -1596,39 +1539,20 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
try:
|
try:
|
||||||
function_name = target["function_name"]
|
function_name = target["function_name"]
|
||||||
function_args = target["function_args"]
|
function_args = target["function_args"]
|
||||||
# 获取 unilabos 系统参数
|
|
||||||
unilabos_param: Dict[str, Any] = target[JSON_UNILABOS_PARAM]
|
|
||||||
|
|
||||||
assert isinstance(function_args, dict), "执行动作时JSON必须为dict类型\n原JSON: {string}"
|
assert isinstance(function_args, dict), "执行动作时JSON必须为dict类型\n原JSON: {string}"
|
||||||
function = getattr(self.driver_instance, function_name)
|
function = getattr(self.driver_instance, function_name)
|
||||||
assert callable(
|
assert callable(
|
||||||
function
|
function
|
||||||
), f"执行动作时JSON中的function_name对应的函数不可调用: {function_name}\n原JSON: {string}"
|
), f"执行动作时JSON中的function_name对应的函数不可调用: {function_name}\n原JSON: {string}"
|
||||||
|
|
||||||
# 处理参数(包含 unilabos 系统参数如 sample_uuids)
|
# 处理 ResourceSlot 类型参数
|
||||||
args_list = default_manager._analyze_method_signature(function, skip_unilabos_params=False)["args"]
|
args_list = default_manager._analyze_method_signature(function)["args"]
|
||||||
for arg in args_list:
|
for arg in args_list:
|
||||||
arg_name = arg["name"]
|
arg_name = arg["name"]
|
||||||
arg_type = arg["type"]
|
arg_type = arg["type"]
|
||||||
|
|
||||||
# 跳过不在 function_args 中的参数
|
# 跳过不在 function_args 中的参数
|
||||||
if arg_name not in function_args:
|
if arg_name not in function_args:
|
||||||
# 处理 sample_uuids 参数注入
|
|
||||||
if arg_name == PARAM_SAMPLE_UUIDS:
|
|
||||||
raw_sample_uuids = unilabos_param.get(PARAM_SAMPLE_UUIDS, {})
|
|
||||||
# 将 material uuid 转换为 resource 实例
|
|
||||||
# key: sample_uuid, value: material_uuid -> resource 实例
|
|
||||||
resolved_sample_uuids: Dict[str, Any] = {}
|
|
||||||
for sample_uuid, material_uuid in raw_sample_uuids.items():
|
|
||||||
if material_uuid and self.resource_tracker:
|
|
||||||
resource = self.resource_tracker.uuid_to_resources.get(material_uuid)
|
|
||||||
resolved_sample_uuids[sample_uuid] = resource if resource else material_uuid
|
|
||||||
else:
|
|
||||||
resolved_sample_uuids[sample_uuid] = material_uuid
|
|
||||||
function_args[PARAM_SAMPLE_UUIDS] = resolved_sample_uuids
|
|
||||||
self.lab_logger().debug(
|
|
||||||
f"[JsonCommand] 注入 {PARAM_SAMPLE_UUIDS}: {resolved_sample_uuids}"
|
|
||||||
)
|
|
||||||
continue
|
continue
|
||||||
|
|
||||||
# 处理单个 ResourceSlot
|
# 处理单个 ResourceSlot
|
||||||
@@ -1658,7 +1582,6 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
)
|
)
|
||||||
raise JsonCommandInitError(f"ResourceSlot列表参数转换失败: {arg_name}")
|
raise JsonCommandInitError(f"ResourceSlot列表参数转换失败: {arg_name}")
|
||||||
|
|
||||||
# todo: 默认反报送
|
|
||||||
return function(**function_args)
|
return function(**function_args)
|
||||||
except KeyError as ex:
|
except KeyError as ex:
|
||||||
raise JsonCommandInitError(
|
raise JsonCommandInitError(
|
||||||
@@ -1678,23 +1601,21 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
raise ValueError("至少需要提供一个 UUID")
|
raise ValueError("至少需要提供一个 UUID")
|
||||||
|
|
||||||
uuids_list = list(uuids)
|
uuids_list = list(uuids)
|
||||||
future = self._resource_clients["c2s_update_resource_tree"].call_async(
|
future = self._resource_clients["c2s_update_resource_tree"].call_async(SerialCommand.Request(
|
||||||
SerialCommand.Request(
|
command=json.dumps(
|
||||||
command=json.dumps(
|
{
|
||||||
{
|
"data": {"data": uuids_list, "with_children": True},
|
||||||
"data": {"data": uuids_list, "with_children": True},
|
"action": "get",
|
||||||
"action": "get",
|
}
|
||||||
}
|
|
||||||
)
|
|
||||||
)
|
)
|
||||||
)
|
))
|
||||||
|
|
||||||
# 等待结果(使用while循环,每次sleep 0.05秒,最多等待30秒)
|
# 等待结果(使用while循环,每次sleep 0.05秒,最多等待30秒)
|
||||||
timeout = 30.0
|
timeout = 30.0
|
||||||
elapsed = 0.0
|
elapsed = 0.0
|
||||||
while not future.done() and elapsed < timeout:
|
while not future.done() and elapsed < timeout:
|
||||||
time.sleep(0.02)
|
time.sleep(0.05)
|
||||||
elapsed += 0.02
|
elapsed += 0.05
|
||||||
|
|
||||||
if not future.done():
|
if not future.done():
|
||||||
raise Exception(f"资源查询超时: {uuids_list}")
|
raise Exception(f"资源查询超时: {uuids_list}")
|
||||||
@@ -1745,9 +1666,6 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
try:
|
try:
|
||||||
function_name = target["function_name"]
|
function_name = target["function_name"]
|
||||||
function_args = target["function_args"]
|
function_args = target["function_args"]
|
||||||
# 获取 unilabos 系统参数
|
|
||||||
unilabos_param: Dict[str, Any] = target.get(JSON_UNILABOS_PARAM, {})
|
|
||||||
|
|
||||||
assert isinstance(function_args, dict), "执行动作时JSON必须为dict类型\n原JSON: {string}"
|
assert isinstance(function_args, dict), "执行动作时JSON必须为dict类型\n原JSON: {string}"
|
||||||
function = getattr(self.driver_instance, function_name)
|
function = getattr(self.driver_instance, function_name)
|
||||||
assert callable(
|
assert callable(
|
||||||
@@ -1757,30 +1675,14 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
function
|
function
|
||||||
), f"执行动作时JSON中的function并非异步: {function_name}\n原JSON: {string}"
|
), f"执行动作时JSON中的function并非异步: {function_name}\n原JSON: {string}"
|
||||||
|
|
||||||
# 处理参数(包含 unilabos 系统参数如 sample_uuids)
|
# 处理 ResourceSlot 类型参数
|
||||||
args_list = default_manager._analyze_method_signature(function, skip_unilabos_params=False)["args"]
|
args_list = default_manager._analyze_method_signature(function)["args"]
|
||||||
for arg in args_list:
|
for arg in args_list:
|
||||||
arg_name = arg["name"]
|
arg_name = arg["name"]
|
||||||
arg_type = arg["type"]
|
arg_type = arg["type"]
|
||||||
|
|
||||||
# 跳过不在 function_args 中的参数
|
# 跳过不在 function_args 中的参数
|
||||||
if arg_name not in function_args:
|
if arg_name not in function_args:
|
||||||
# 处理 sample_uuids 参数注入
|
|
||||||
if arg_name == PARAM_SAMPLE_UUIDS:
|
|
||||||
raw_sample_uuids = unilabos_param.get(PARAM_SAMPLE_UUIDS, {})
|
|
||||||
# 将 material uuid 转换为 resource 实例
|
|
||||||
# key: sample_uuid, value: material_uuid -> resource 实例
|
|
||||||
resolved_sample_uuids: Dict[str, Any] = {}
|
|
||||||
for sample_uuid, material_uuid in raw_sample_uuids.items():
|
|
||||||
if material_uuid and self.resource_tracker:
|
|
||||||
resource = self.resource_tracker.uuid_to_resources.get(material_uuid)
|
|
||||||
resolved_sample_uuids[sample_uuid] = resource if resource else material_uuid
|
|
||||||
else:
|
|
||||||
resolved_sample_uuids[sample_uuid] = material_uuid
|
|
||||||
function_args[PARAM_SAMPLE_UUIDS] = resolved_sample_uuids
|
|
||||||
self.lab_logger().debug(
|
|
||||||
f"[JsonCommandAsync] 注入 {PARAM_SAMPLE_UUIDS}: {resolved_sample_uuids}"
|
|
||||||
)
|
|
||||||
continue
|
continue
|
||||||
|
|
||||||
# 处理单个 ResourceSlot
|
# 处理单个 ResourceSlot
|
||||||
@@ -2058,9 +1960,7 @@ class ROS2DeviceNode:
|
|||||||
asyncio.set_event_loop(loop)
|
asyncio.set_event_loop(loop)
|
||||||
loop.run_forever()
|
loop.run_forever()
|
||||||
|
|
||||||
ROS2DeviceNode._asyncio_loop_thread = threading.Thread(
|
ROS2DeviceNode._asyncio_loop_thread = threading.Thread(target=run_event_loop, daemon=True, name="ROS2DeviceNode")
|
||||||
target=run_event_loop, daemon=True, name="ROS2DeviceNode"
|
|
||||||
)
|
|
||||||
ROS2DeviceNode._asyncio_loop_thread.start()
|
ROS2DeviceNode._asyncio_loop_thread.start()
|
||||||
logger.info(f"循环线程已启动")
|
logger.info(f"循环线程已启动")
|
||||||
|
|
||||||
|
|||||||
@@ -1,17 +1,17 @@
|
|||||||
import collections
|
import collections
|
||||||
|
from dataclasses import dataclass, field
|
||||||
import json
|
import json
|
||||||
import threading
|
import threading
|
||||||
import time
|
import time
|
||||||
import traceback
|
import traceback
|
||||||
import uuid
|
import uuid
|
||||||
from dataclasses import dataclass, field
|
|
||||||
from typing import TYPE_CHECKING, Optional, Dict, Any, List, ClassVar, Set, Union
|
from typing import TYPE_CHECKING, Optional, Dict, Any, List, ClassVar, Set, Union
|
||||||
|
from typing_extensions import TypedDict
|
||||||
|
|
||||||
from action_msgs.msg import GoalStatus
|
from action_msgs.msg import GoalStatus
|
||||||
from geometry_msgs.msg import Point
|
from geometry_msgs.msg import Point
|
||||||
from rclpy.action import ActionClient, get_action_server_names_and_types_by_node
|
from rclpy.action import ActionClient, get_action_server_names_and_types_by_node
|
||||||
from rclpy.service import Service
|
from rclpy.service import Service
|
||||||
from typing_extensions import TypedDict
|
|
||||||
from unilabos_msgs.msg import Resource # type: ignore
|
from unilabos_msgs.msg import Resource # type: ignore
|
||||||
from unilabos_msgs.srv import (
|
from unilabos_msgs.srv import (
|
||||||
ResourceAdd,
|
ResourceAdd,
|
||||||
@@ -23,20 +23,10 @@ from unilabos_msgs.srv import (
|
|||||||
from unilabos_msgs.srv._serial_command import SerialCommand_Request, SerialCommand_Response
|
from unilabos_msgs.srv._serial_command import SerialCommand_Request, SerialCommand_Response
|
||||||
from unique_identifier_msgs.msg import UUID
|
from unique_identifier_msgs.msg import UUID
|
||||||
|
|
||||||
from unilabos.registry.placeholder_type import ResourceSlot, DeviceSlot
|
|
||||||
from unilabos.registry.registry import lab_registry
|
from unilabos.registry.registry import lab_registry
|
||||||
from unilabos.resources.container import RegularContainer
|
from unilabos.resources.container import RegularContainer
|
||||||
from unilabos.resources.graphio import initialize_resource
|
from unilabos.resources.graphio import initialize_resource
|
||||||
from unilabos.resources.registry import add_schema
|
from unilabos.resources.registry import add_schema
|
||||||
from unilabos.resources.resource_tracker import (
|
|
||||||
ResourceDict,
|
|
||||||
ResourceDictInstance,
|
|
||||||
ResourceTreeSet,
|
|
||||||
ResourceTreeInstance,
|
|
||||||
RETURN_UNILABOS_SAMPLES,
|
|
||||||
JSON_UNILABOS_PARAM,
|
|
||||||
PARAM_SAMPLE_UUIDS,
|
|
||||||
)
|
|
||||||
from unilabos.ros.initialize_device import initialize_device_from_dict
|
from unilabos.ros.initialize_device import initialize_device_from_dict
|
||||||
from unilabos.ros.msgs.message_converter import (
|
from unilabos.ros.msgs.message_converter import (
|
||||||
get_msg_type,
|
get_msg_type,
|
||||||
@@ -47,10 +37,17 @@ from unilabos.ros.msgs.message_converter import (
|
|||||||
)
|
)
|
||||||
from unilabos.ros.nodes.base_device_node import BaseROS2DeviceNode, ROS2DeviceNode, DeviceNodeResourceTracker
|
from unilabos.ros.nodes.base_device_node import BaseROS2DeviceNode, ROS2DeviceNode, DeviceNodeResourceTracker
|
||||||
from unilabos.ros.nodes.presets.controller_node import ControllerNode
|
from unilabos.ros.nodes.presets.controller_node import ControllerNode
|
||||||
|
from unilabos.resources.resource_tracker import (
|
||||||
|
ResourceDict,
|
||||||
|
ResourceDictInstance,
|
||||||
|
ResourceTreeSet,
|
||||||
|
ResourceTreeInstance,
|
||||||
|
)
|
||||||
from unilabos.utils import logger
|
from unilabos.utils import logger
|
||||||
from unilabos.utils.exception import DeviceClassInvalid
|
from unilabos.utils.exception import DeviceClassInvalid
|
||||||
from unilabos.utils.log import warning
|
from unilabos.utils.log import warning
|
||||||
from unilabos.utils.type_check import serialize_result_info
|
from unilabos.utils.type_check import serialize_result_info
|
||||||
|
from unilabos.registry.placeholder_type import ResourceSlot, DeviceSlot
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from unilabos.app.ws_client import QueueItem
|
from unilabos.app.ws_client import QueueItem
|
||||||
@@ -758,7 +755,6 @@ class HostNode(BaseROS2DeviceNode):
|
|||||||
item: "QueueItem",
|
item: "QueueItem",
|
||||||
action_type: str,
|
action_type: str,
|
||||||
action_kwargs: Dict[str, Any],
|
action_kwargs: Dict[str, Any],
|
||||||
sample_material: Dict[str, str],
|
|
||||||
server_info: Optional[Dict[str, Any]] = None,
|
server_info: Optional[Dict[str, Any]] = None,
|
||||||
) -> None:
|
) -> None:
|
||||||
"""
|
"""
|
||||||
@@ -776,14 +772,14 @@ class HostNode(BaseROS2DeviceNode):
|
|||||||
if action_name.startswith("auto-"):
|
if action_name.startswith("auto-"):
|
||||||
action_name = action_name[5:]
|
action_name = action_name[5:]
|
||||||
action_id = f"/devices/{device_id}/_execute_driver_command"
|
action_id = f"/devices/{device_id}/_execute_driver_command"
|
||||||
json_command: Dict[str, Any] = {
|
action_kwargs = {
|
||||||
"function_name": action_name,
|
"string": json.dumps(
|
||||||
"function_args": action_kwargs,
|
{
|
||||||
JSON_UNILABOS_PARAM: {
|
"function_name": action_name,
|
||||||
PARAM_SAMPLE_UUIDS: sample_material,
|
"function_args": action_kwargs,
|
||||||
},
|
}
|
||||||
|
)
|
||||||
}
|
}
|
||||||
action_kwargs = {"string": json.dumps(json_command)}
|
|
||||||
if action_type.startswith("UniLabJsonCommandAsync"):
|
if action_type.startswith("UniLabJsonCommandAsync"):
|
||||||
action_id = f"/devices/{device_id}/_execute_driver_command_async"
|
action_id = f"/devices/{device_id}/_execute_driver_command_async"
|
||||||
else:
|
else:
|
||||||
@@ -794,9 +790,24 @@ class HostNode(BaseROS2DeviceNode):
|
|||||||
raise ValueError(f"ActionClient {action_id} not found.")
|
raise ValueError(f"ActionClient {action_id} not found.")
|
||||||
|
|
||||||
action_client: ActionClient = self._action_clients[action_id]
|
action_client: ActionClient = self._action_clients[action_id]
|
||||||
|
|
||||||
|
# 遍历action_kwargs下的所有子dict,将"sample_uuid"的值赋给"sample_id"
|
||||||
|
def assign_sample_id(obj):
|
||||||
|
if isinstance(obj, dict):
|
||||||
|
if "sample_uuid" in obj:
|
||||||
|
obj["sample_id"] = obj["sample_uuid"]
|
||||||
|
obj.pop("sample_uuid")
|
||||||
|
for k, v in obj.items():
|
||||||
|
if k != "unilabos_extra":
|
||||||
|
assign_sample_id(v)
|
||||||
|
elif isinstance(obj, list):
|
||||||
|
for item in obj:
|
||||||
|
assign_sample_id(item)
|
||||||
|
|
||||||
|
assign_sample_id(action_kwargs)
|
||||||
goal_msg = convert_to_ros_msg(action_client._action_type.Goal(), action_kwargs)
|
goal_msg = convert_to_ros_msg(action_client._action_type.Goal(), action_kwargs)
|
||||||
|
|
||||||
# self.lab_logger().trace(f"[Host Node] Sending goal for {action_id}: {str(goal_msg)[:1000]}")
|
self.lab_logger().info(f"[Host Node] Sending goal for {action_id}: {str(goal_msg)[:1000]}")
|
||||||
self.lab_logger().trace(f"[Host Node] Sending goal for {action_id}: {action_kwargs}")
|
self.lab_logger().trace(f"[Host Node] Sending goal for {action_id}: {action_kwargs}")
|
||||||
self.lab_logger().trace(f"[Host Node] Sending goal for {action_id}: {goal_msg}")
|
self.lab_logger().trace(f"[Host Node] Sending goal for {action_id}: {goal_msg}")
|
||||||
action_client.wait_for_server()
|
action_client.wait_for_server()
|
||||||
@@ -856,14 +867,14 @@ class HostNode(BaseROS2DeviceNode):
|
|||||||
# 适配后端的一些额外处理
|
# 适配后端的一些额外处理
|
||||||
return_value = return_info.get("return_value")
|
return_value = return_info.get("return_value")
|
||||||
if isinstance(return_value, dict):
|
if isinstance(return_value, dict):
|
||||||
unilabos_samples = return_value.pop(RETURN_UNILABOS_SAMPLES, None)
|
unilabos_samples = return_value.pop("unilabos_samples", None)
|
||||||
if isinstance(unilabos_samples, list) and unilabos_samples:
|
if isinstance(unilabos_samples, list) and unilabos_samples:
|
||||||
self.lab_logger().info(
|
self.lab_logger().info(
|
||||||
f"[Host Node] Job {job_id[:8]} returned {len(unilabos_samples)} sample(s): "
|
f"[Host Node] Job {job_id[:8]} returned {len(unilabos_samples)} sample(s): "
|
||||||
f"{[s.get('name', s.get('id', 'unknown')) if isinstance(s, dict) else str(s)[:20] for s in unilabos_samples[:5]]}"
|
f"{[s.get('name', s.get('id', 'unknown')) if isinstance(s, dict) else str(s)[:20] for s in unilabos_samples[:5]]}"
|
||||||
f"{'...' if len(unilabos_samples) > 5 else ''}"
|
f"{'...' if len(unilabos_samples) > 5 else ''}"
|
||||||
)
|
)
|
||||||
return_info["samples"] = unilabos_samples
|
return_info["unilabos_samples"] = unilabos_samples
|
||||||
suc = return_info.get("suc", False)
|
suc = return_info.get("suc", False)
|
||||||
if not suc:
|
if not suc:
|
||||||
status = "failed"
|
status = "failed"
|
||||||
@@ -1169,7 +1180,7 @@ class HostNode(BaseROS2DeviceNode):
|
|||||||
"""
|
"""
|
||||||
更新节点信息回调
|
更新节点信息回调
|
||||||
"""
|
"""
|
||||||
self.lab_logger().trace(f"[Host Node] Node info update request received: {request}")
|
# self.lab_logger().info(f"[Host Node] Node info update request received: {request}")
|
||||||
try:
|
try:
|
||||||
from unilabos.app.communication import get_communication_client
|
from unilabos.app.communication import get_communication_client
|
||||||
from unilabos.app.web.client import HTTPClient, http_client
|
from unilabos.app.web.client import HTTPClient, http_client
|
||||||
|
|||||||
@@ -340,8 +340,6 @@ class ROS2WorkstationNode(BaseROS2DeviceNode):
|
|||||||
plr = self.resource_tracker.figure_resource({"name": res_name}, try_mode=False)
|
plr = self.resource_tracker.figure_resource({"name": res_name}, try_mode=False)
|
||||||
# 获取父资源
|
# 获取父资源
|
||||||
res = self.resource_tracker.parent_resource(plr)
|
res = self.resource_tracker.parent_resource(plr)
|
||||||
if res is None:
|
|
||||||
res = plr
|
|
||||||
if id(res) not in seen:
|
if id(res) not in seen:
|
||||||
seen.add(id(res))
|
seen.add(id(res))
|
||||||
unique_resources.append(res)
|
unique_resources.append(res)
|
||||||
|
|||||||
@@ -52,8 +52,7 @@ class DeviceClassCreator(Generic[T]):
|
|||||||
if self.device_instance is not None:
|
if self.device_instance is not None:
|
||||||
for c in self.children:
|
for c in self.children:
|
||||||
if c.res_content.type != "device":
|
if c.res_content.type != "device":
|
||||||
res = ResourceTreeSet([ResourceTreeInstance(c)]).to_plr_resources()[0]
|
self.resource_tracker.add_resource(c.get_plr_nested_dict())
|
||||||
self.resource_tracker.add_resource(res)
|
|
||||||
|
|
||||||
def create_instance(self, data: Dict[str, Any]) -> T:
|
def create_instance(self, data: Dict[str, Any]) -> T:
|
||||||
"""
|
"""
|
||||||
@@ -120,7 +119,7 @@ class PyLabRobotCreator(DeviceClassCreator[T]):
|
|||||||
# return resource, source_type
|
# return resource, source_type
|
||||||
|
|
||||||
def _process_resource_references(
|
def _process_resource_references(
|
||||||
self, data: Any, processed_child_names: Optional[Dict[str, Any]], to_dict=False, states=None, prefix_path="", name_to_uuid=None
|
self, data: Any, to_dict=False, states=None, prefix_path="", name_to_uuid=None
|
||||||
) -> Any:
|
) -> Any:
|
||||||
"""
|
"""
|
||||||
递归处理资源引用,替换_resource_child_name对应的资源
|
递归处理资源引用,替换_resource_child_name对应的资源
|
||||||
@@ -165,7 +164,6 @@ class PyLabRobotCreator(DeviceClassCreator[T]):
|
|||||||
states[prefix_path] = resource_instance.serialize_all_state()
|
states[prefix_path] = resource_instance.serialize_all_state()
|
||||||
return serialized
|
return serialized
|
||||||
else:
|
else:
|
||||||
processed_child_names[child_name] = resource_instance
|
|
||||||
self.resource_tracker.add_resource(resource_instance)
|
self.resource_tracker.add_resource(resource_instance)
|
||||||
# 立即设置UUID,state已经在resource_ulab_to_plr中处理过了
|
# 立即设置UUID,state已经在resource_ulab_to_plr中处理过了
|
||||||
if name_to_uuid:
|
if name_to_uuid:
|
||||||
@@ -184,12 +182,12 @@ class PyLabRobotCreator(DeviceClassCreator[T]):
|
|||||||
result = {}
|
result = {}
|
||||||
for key, value in data.items():
|
for key, value in data.items():
|
||||||
new_prefix = f"{prefix_path}.{key}" if prefix_path else key
|
new_prefix = f"{prefix_path}.{key}" if prefix_path else key
|
||||||
result[key] = self._process_resource_references(value, processed_child_names, to_dict, states, new_prefix, name_to_uuid)
|
result[key] = self._process_resource_references(value, to_dict, states, new_prefix, name_to_uuid)
|
||||||
return result
|
return result
|
||||||
|
|
||||||
elif isinstance(data, list):
|
elif isinstance(data, list):
|
||||||
return [
|
return [
|
||||||
self._process_resource_references(item, processed_child_names, to_dict, states, f"{prefix_path}[{i}]", name_to_uuid)
|
self._process_resource_references(item, to_dict, states, f"{prefix_path}[{i}]", name_to_uuid)
|
||||||
for i, item in enumerate(data)
|
for i, item in enumerate(data)
|
||||||
]
|
]
|
||||||
|
|
||||||
@@ -236,7 +234,7 @@ class PyLabRobotCreator(DeviceClassCreator[T]):
|
|||||||
# 首先处理资源引用
|
# 首先处理资源引用
|
||||||
states = {}
|
states = {}
|
||||||
processed_data = self._process_resource_references(
|
processed_data = self._process_resource_references(
|
||||||
data, {}, to_dict=True, states=states, name_to_uuid=name_to_uuid
|
data, to_dict=True, states=states, name_to_uuid=name_to_uuid
|
||||||
)
|
)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
@@ -272,12 +270,7 @@ class PyLabRobotCreator(DeviceClassCreator[T]):
|
|||||||
arg_value = spec_args[param_name].annotation
|
arg_value = spec_args[param_name].annotation
|
||||||
data[param_name]["_resource_type"] = self.device_cls.__module__ + ":" + arg_value
|
data[param_name]["_resource_type"] = self.device_cls.__module__ + ":" + arg_value
|
||||||
logger.debug(f"自动补充 _resource_type: {data[param_name]['_resource_type']}")
|
logger.debug(f"自动补充 _resource_type: {data[param_name]['_resource_type']}")
|
||||||
processed_child_names = {}
|
processed_data = self._process_resource_references(data, to_dict=False, name_to_uuid=name_to_uuid)
|
||||||
processed_data = self._process_resource_references(data, processed_child_names, to_dict=False, name_to_uuid=name_to_uuid)
|
|
||||||
for child_name, resource_instance in processed_data.items():
|
|
||||||
for ind, name in enumerate([child.res_content.name for child in self.children]):
|
|
||||||
if name == child_name:
|
|
||||||
self.children.pop(ind)
|
|
||||||
self.device_instance = super(PyLabRobotCreator, self).create_instance(processed_data) # 补全变量后直接调用,调用的自身的attach_resource
|
self.device_instance = super(PyLabRobotCreator, self).create_instance(processed_data) # 补全变量后直接调用,调用的自身的attach_resource
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(f"PyLabRobot创建实例失败: {e}")
|
logger.error(f"PyLabRobot创建实例失败: {e}")
|
||||||
@@ -349,10 +342,9 @@ class WorkstationNodeCreator(DeviceClassCreator[T]):
|
|||||||
try:
|
try:
|
||||||
# 创建实例,额外补充一个给protocol node的字段,后面考虑取消
|
# 创建实例,额外补充一个给protocol node的字段,后面考虑取消
|
||||||
data["children"] = self.children
|
data["children"] = self.children
|
||||||
# super(WorkstationNodeCreator, self).create_instance(data)的时候会attach
|
for child in self.children:
|
||||||
# for child in self.children:
|
if child.res_content.type != "device":
|
||||||
# if child.res_content.type != "device":
|
self.resource_tracker.add_resource(child.get_plr_nested_dict())
|
||||||
# self.resource_tracker.add_resource(child.get_plr_nested_dict())
|
|
||||||
deck_dict = data.get("deck")
|
deck_dict = data.get("deck")
|
||||||
if deck_dict:
|
if deck_dict:
|
||||||
from pylabrobot.resources import Deck, Resource
|
from pylabrobot.resources import Deck, Resource
|
||||||
|
|||||||
@@ -1,795 +0,0 @@
|
|||||||
{
|
|
||||||
"nodes": [
|
|
||||||
{
|
|
||||||
"id": "PRCXI",
|
|
||||||
"name": "PRCXI",
|
|
||||||
"type": "device",
|
|
||||||
"class": "liquid_handler.prcxi",
|
|
||||||
"parent": "",
|
|
||||||
"pose": {
|
|
||||||
"size": {
|
|
||||||
"width": 562,
|
|
||||||
"height": 394,
|
|
||||||
"depth": 0
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"config": {
|
|
||||||
"axis": "Left",
|
|
||||||
"deck": {
|
|
||||||
"_resource_type": "unilabos.devices.liquid_handling.prcxi.prcxi:PRCXI9300Deck",
|
|
||||||
"_resource_child_name": "PRCXI_Deck"
|
|
||||||
},
|
|
||||||
"host": "10.20.30.184",
|
|
||||||
"port": 9999,
|
|
||||||
"debug": true,
|
|
||||||
"setup": true,
|
|
||||||
"is_9320": true,
|
|
||||||
"timeout": 10,
|
|
||||||
"matrix_id": "5de524d0-3f95-406c-86dd-f83626ebc7cb",
|
|
||||||
"simulator": true,
|
|
||||||
"channel_num": 2
|
|
||||||
},
|
|
||||||
"data": {
|
|
||||||
"reset_ok": true
|
|
||||||
},
|
|
||||||
"schema": {},
|
|
||||||
"description": "",
|
|
||||||
"model": null,
|
|
||||||
"position": {
|
|
||||||
"x": 0,
|
|
||||||
"y": 240,
|
|
||||||
"z": 0
|
|
||||||
}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": "PRCXI_Deck",
|
|
||||||
"name": "PRCXI_Deck",
|
|
||||||
|
|
||||||
"children": [],
|
|
||||||
"parent": "PRCXI",
|
|
||||||
"type": "deck",
|
|
||||||
"class": "",
|
|
||||||
"position": {
|
|
||||||
"x": 10,
|
|
||||||
"y": 10,
|
|
||||||
"z": 0
|
|
||||||
},
|
|
||||||
"config": {
|
|
||||||
"type": "PRCXI9300Deck",
|
|
||||||
"size_x": 542,
|
|
||||||
"size_y": 374,
|
|
||||||
"size_z": 0,
|
|
||||||
"rotation": {
|
|
||||||
"x": 0,
|
|
||||||
"y": 0,
|
|
||||||
"z": 0,
|
|
||||||
"type": "Rotation"
|
|
||||||
},
|
|
||||||
"category": "deck",
|
|
||||||
"barcode": null
|
|
||||||
},
|
|
||||||
"data": {}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": "T1",
|
|
||||||
"name": "T1",
|
|
||||||
"children": [],
|
|
||||||
"parent": "PRCXI_Deck",
|
|
||||||
"type": "plate",
|
|
||||||
"class": "",
|
|
||||||
"position": {
|
|
||||||
"x": 0,
|
|
||||||
"y": 288,
|
|
||||||
"z": 0
|
|
||||||
},
|
|
||||||
"config": {
|
|
||||||
"type": "PRCXI9300Container",
|
|
||||||
"size_x": 127,
|
|
||||||
"size_y": 85.5,
|
|
||||||
"size_z": 10,
|
|
||||||
"rotation": {
|
|
||||||
"x": 0,
|
|
||||||
"y": 0,
|
|
||||||
"z": 0,
|
|
||||||
"type": "Rotation"
|
|
||||||
},
|
|
||||||
"category": "plate",
|
|
||||||
"model": null,
|
|
||||||
"barcode": null,
|
|
||||||
"ordering": {},
|
|
||||||
"sites": [
|
|
||||||
{
|
|
||||||
"label": "T1",
|
|
||||||
"visible": true,
|
|
||||||
"position": { "x": 0, "y": 0, "z": 0 },
|
|
||||||
"size": { "width": 128.0, "height": 86, "depth": 0 },
|
|
||||||
"content_type": [
|
|
||||||
"plate",
|
|
||||||
"tip_rack",
|
|
||||||
"plates",
|
|
||||||
"tip_racks",
|
|
||||||
"tube_rack"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
"data": {}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": "T2",
|
|
||||||
"name": "T2",
|
|
||||||
"children": [],
|
|
||||||
"parent": "PRCXI_Deck",
|
|
||||||
"type": "plate",
|
|
||||||
"class": "",
|
|
||||||
"position": {
|
|
||||||
"x": 138,
|
|
||||||
"y": 288,
|
|
||||||
"z": 0
|
|
||||||
},
|
|
||||||
"config": {
|
|
||||||
"type": "PRCXI9300Container",
|
|
||||||
"size_x": 127,
|
|
||||||
"size_y": 85.5,
|
|
||||||
"size_z": 10,
|
|
||||||
"rotation": {
|
|
||||||
"x": 0,
|
|
||||||
"y": 0,
|
|
||||||
"z": 0,
|
|
||||||
"type": "Rotation"
|
|
||||||
},
|
|
||||||
"category": "plate",
|
|
||||||
"model": null,
|
|
||||||
"barcode": null,
|
|
||||||
"ordering": {},
|
|
||||||
"sites": [
|
|
||||||
{
|
|
||||||
"label": "T2",
|
|
||||||
"visible": true,
|
|
||||||
"position": { "x": 0, "y": 0, "z": 0 },
|
|
||||||
"size": { "width": 128.0, "height": 86, "depth": 0 },
|
|
||||||
"content_type": [
|
|
||||||
"plate",
|
|
||||||
"tip_rack",
|
|
||||||
"plates",
|
|
||||||
"tip_racks",
|
|
||||||
"tube_rack"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
"data": {}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": "T3",
|
|
||||||
"name": "T3",
|
|
||||||
"children": [],
|
|
||||||
"parent": "PRCXI_Deck",
|
|
||||||
"type": "plate",
|
|
||||||
"class": "",
|
|
||||||
"position": {
|
|
||||||
"x": 276,
|
|
||||||
"y": 288,
|
|
||||||
"z": 0
|
|
||||||
},
|
|
||||||
"config": {
|
|
||||||
"type": "PRCXI9300Container",
|
|
||||||
"size_x": 127,
|
|
||||||
"size_y": 85.5,
|
|
||||||
"size_z": 10,
|
|
||||||
"rotation": {
|
|
||||||
"x": 0,
|
|
||||||
"y": 0,
|
|
||||||
"z": 0,
|
|
||||||
"type": "Rotation"
|
|
||||||
},
|
|
||||||
"category": "plate",
|
|
||||||
"model": null,
|
|
||||||
"barcode": null,
|
|
||||||
"ordering": {},
|
|
||||||
"sites": [
|
|
||||||
{
|
|
||||||
"label": "T3",
|
|
||||||
"visible": true,
|
|
||||||
"position": { "x": 0, "y": 0, "z": 0 },
|
|
||||||
"size": { "width": 128.0, "height": 86, "depth": 0 },
|
|
||||||
"content_type": [
|
|
||||||
"plate",
|
|
||||||
"tip_rack",
|
|
||||||
"plates",
|
|
||||||
"tip_racks",
|
|
||||||
"tube_rack"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
"data": {}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": "T4",
|
|
||||||
"name": "T4",
|
|
||||||
"children": [],
|
|
||||||
"parent": "PRCXI_Deck",
|
|
||||||
"type": "plate",
|
|
||||||
"class": "",
|
|
||||||
"position": {
|
|
||||||
"x": 414,
|
|
||||||
"y": 288,
|
|
||||||
"z": 0
|
|
||||||
},
|
|
||||||
"config": {
|
|
||||||
"type": "PRCXI9300Container",
|
|
||||||
"size_x": 127,
|
|
||||||
"size_y": 85.5,
|
|
||||||
"size_z": 10,
|
|
||||||
"rotation": {
|
|
||||||
"x": 0,
|
|
||||||
"y": 0,
|
|
||||||
"z": 0,
|
|
||||||
"type": "Rotation"
|
|
||||||
},
|
|
||||||
"category": "plate",
|
|
||||||
"model": null,
|
|
||||||
"barcode": null,
|
|
||||||
"ordering": {},
|
|
||||||
"sites": [
|
|
||||||
{
|
|
||||||
"label": "T4",
|
|
||||||
"visible": true,
|
|
||||||
"position": { "x": 0, "y": 0, "z": 0 },
|
|
||||||
"size": { "width": 128.0, "height": 86, "depth": 0 },
|
|
||||||
"content_type": [
|
|
||||||
"plate",
|
|
||||||
"tip_rack",
|
|
||||||
"plates",
|
|
||||||
"tip_racks",
|
|
||||||
"tube_rack"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
"data": {}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": "T5",
|
|
||||||
"name": "T5",
|
|
||||||
"children": [],
|
|
||||||
"parent": "PRCXI_Deck",
|
|
||||||
"type": "plate",
|
|
||||||
"class": "",
|
|
||||||
"position": {
|
|
||||||
"x": 0,
|
|
||||||
"y": 192,
|
|
||||||
"z": 0
|
|
||||||
},
|
|
||||||
"config": {
|
|
||||||
"type": "PRCXI9300Container",
|
|
||||||
"size_x": 127,
|
|
||||||
"size_y": 85.5,
|
|
||||||
"size_z": 10,
|
|
||||||
"rotation": {
|
|
||||||
"x": 0,
|
|
||||||
"y": 0,
|
|
||||||
"z": 0,
|
|
||||||
"type": "Rotation"
|
|
||||||
},
|
|
||||||
"category": "plate",
|
|
||||||
"model": null,
|
|
||||||
"barcode": null,
|
|
||||||
"ordering": {},
|
|
||||||
"sites": [
|
|
||||||
{
|
|
||||||
"label": "T5",
|
|
||||||
"visible": true,
|
|
||||||
"position": { "x": 0, "y": 0, "z": 0 },
|
|
||||||
"size": { "width": 128.0, "height": 86, "depth": 0 },
|
|
||||||
"content_type": [
|
|
||||||
"plate",
|
|
||||||
"tip_rack",
|
|
||||||
"plates",
|
|
||||||
"tip_racks",
|
|
||||||
"tube_rack"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
"data": {}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": "T6",
|
|
||||||
"name": "T6",
|
|
||||||
"children": [],
|
|
||||||
"parent": "PRCXI_Deck",
|
|
||||||
"type": "plate",
|
|
||||||
"class": "",
|
|
||||||
"position": {
|
|
||||||
"x": 138,
|
|
||||||
"y": 192,
|
|
||||||
"z": 0
|
|
||||||
},
|
|
||||||
"config": {
|
|
||||||
"type": "PRCXI9300Container",
|
|
||||||
"size_x": 127,
|
|
||||||
"size_y": 85.5,
|
|
||||||
"size_z": 10,
|
|
||||||
"rotation": {
|
|
||||||
"x": 0,
|
|
||||||
"y": 0,
|
|
||||||
"z": 0,
|
|
||||||
"type": "Rotation"
|
|
||||||
},
|
|
||||||
"category": "plate",
|
|
||||||
"model": null,
|
|
||||||
"barcode": null,
|
|
||||||
"ordering": {},
|
|
||||||
"sites": [
|
|
||||||
{
|
|
||||||
"label": "T6",
|
|
||||||
"visible": true,
|
|
||||||
"position": { "x": 0, "y": 0, "z": 0 },
|
|
||||||
"size": { "width": 128.0, "height": 86, "depth": 0 },
|
|
||||||
"content_type": [
|
|
||||||
"plate",
|
|
||||||
"tip_rack",
|
|
||||||
"plates",
|
|
||||||
"tip_racks",
|
|
||||||
"tube_rack"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
"data": {}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": "T7",
|
|
||||||
"name": "T7",
|
|
||||||
"children": [],
|
|
||||||
"parent": "PRCXI_Deck",
|
|
||||||
"type": "plate",
|
|
||||||
"class": "",
|
|
||||||
"position": {
|
|
||||||
"x": 276,
|
|
||||||
"y": 192,
|
|
||||||
"z": 0
|
|
||||||
},
|
|
||||||
"config": {
|
|
||||||
"type": "PRCXI9300Container",
|
|
||||||
"size_x": 127,
|
|
||||||
"size_y": 85.5,
|
|
||||||
"size_z": 10,
|
|
||||||
"rotation": {
|
|
||||||
"x": 0,
|
|
||||||
"y": 0,
|
|
||||||
"z": 0,
|
|
||||||
"type": "Rotation"
|
|
||||||
},
|
|
||||||
"category": "plate",
|
|
||||||
"model": null,
|
|
||||||
"barcode": null,
|
|
||||||
"ordering": {},
|
|
||||||
"sites": [
|
|
||||||
{
|
|
||||||
"label": "T7",
|
|
||||||
"visible": true,
|
|
||||||
"position": { "x": 0, "y": 0, "z": 0 },
|
|
||||||
"size": { "width": 128.0, "height": 86, "depth": 0 },
|
|
||||||
"content_type": [
|
|
||||||
"plate",
|
|
||||||
"tip_rack",
|
|
||||||
"plates",
|
|
||||||
"tip_racks",
|
|
||||||
"tube_rack"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
"data": {}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": "T8",
|
|
||||||
"name": "T8",
|
|
||||||
"children": [],
|
|
||||||
"parent": "PRCXI_Deck",
|
|
||||||
"type": "plate",
|
|
||||||
"class": "",
|
|
||||||
"position": {
|
|
||||||
"x": 414,
|
|
||||||
"y": 192,
|
|
||||||
"z": 0
|
|
||||||
},
|
|
||||||
"config": {
|
|
||||||
"type": "PRCXI9300Container",
|
|
||||||
"size_x": 127,
|
|
||||||
"size_y": 85.5,
|
|
||||||
"size_z": 10,
|
|
||||||
"rotation": {
|
|
||||||
"x": 0,
|
|
||||||
"y": 0,
|
|
||||||
"z": 0,
|
|
||||||
"type": "Rotation"
|
|
||||||
},
|
|
||||||
"category": "plate",
|
|
||||||
"model": null,
|
|
||||||
"barcode": null,
|
|
||||||
"ordering": {},
|
|
||||||
"sites": [
|
|
||||||
{
|
|
||||||
"label": "T8",
|
|
||||||
"visible": true,
|
|
||||||
"position": { "x": 0, "y": 0, "z": 0 },
|
|
||||||
"size": { "width": 128.0, "height": 86, "depth": 0 },
|
|
||||||
"content_type": [
|
|
||||||
"plate",
|
|
||||||
"tip_rack",
|
|
||||||
"plates",
|
|
||||||
"tip_racks",
|
|
||||||
"tube_rack"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
"data": {}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": "T9",
|
|
||||||
"name": "T9",
|
|
||||||
"children": [],
|
|
||||||
"parent": "PRCXI_Deck",
|
|
||||||
"type": "plate",
|
|
||||||
"class": "",
|
|
||||||
"position": {
|
|
||||||
"x": 0,
|
|
||||||
"y": 96,
|
|
||||||
"z": 0
|
|
||||||
},
|
|
||||||
"config": {
|
|
||||||
"type": "PRCXI9300Container",
|
|
||||||
"size_x": 127,
|
|
||||||
"size_y": 85.5,
|
|
||||||
"size_z": 10,
|
|
||||||
"rotation": {
|
|
||||||
"x": 0,
|
|
||||||
"y": 0,
|
|
||||||
"z": 0,
|
|
||||||
"type": "Rotation"
|
|
||||||
},
|
|
||||||
"category": "plate",
|
|
||||||
"model": null,
|
|
||||||
"barcode": null,
|
|
||||||
"ordering": {},
|
|
||||||
"sites": [
|
|
||||||
{
|
|
||||||
"label": "T9",
|
|
||||||
"visible": true,
|
|
||||||
"position": { "x": 0, "y": 0, "z": 0 },
|
|
||||||
"size": { "width": 128.0, "height": 86, "depth": 0 },
|
|
||||||
"content_type": [
|
|
||||||
"plate",
|
|
||||||
"tip_rack",
|
|
||||||
"plates",
|
|
||||||
"tip_racks",
|
|
||||||
"tube_rack"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
"data": {}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": "T10",
|
|
||||||
"name": "T10",
|
|
||||||
"children": [],
|
|
||||||
"parent": "PRCXI_Deck",
|
|
||||||
"type": "plate",
|
|
||||||
"class": "",
|
|
||||||
"position": {
|
|
||||||
"x": 138,
|
|
||||||
"y": 96,
|
|
||||||
"z": 0
|
|
||||||
},
|
|
||||||
"config": {
|
|
||||||
"type": "PRCXI9300Container",
|
|
||||||
"size_x": 127,
|
|
||||||
"size_y": 85.5,
|
|
||||||
"size_z": 10,
|
|
||||||
"rotation": {
|
|
||||||
"x": 0,
|
|
||||||
"y": 0,
|
|
||||||
"z": 0,
|
|
||||||
"type": "Rotation"
|
|
||||||
},
|
|
||||||
"category": "plate",
|
|
||||||
"model": null,
|
|
||||||
"barcode": null,
|
|
||||||
"ordering": {},
|
|
||||||
"sites": [
|
|
||||||
{
|
|
||||||
"label": "T10",
|
|
||||||
"visible": true,
|
|
||||||
"position": { "x": 0, "y": 0, "z": 0 },
|
|
||||||
"size": { "width": 128.0, "height": 86, "depth": 0 },
|
|
||||||
"content_type": [
|
|
||||||
"plate",
|
|
||||||
"tip_rack",
|
|
||||||
"plates",
|
|
||||||
"tip_racks",
|
|
||||||
"tube_rack"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
"data": {}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": "T11",
|
|
||||||
"name": "T11",
|
|
||||||
"children": [],
|
|
||||||
"parent": "PRCXI_Deck",
|
|
||||||
"type": "plate",
|
|
||||||
"class": "",
|
|
||||||
"position": {
|
|
||||||
"x": 276,
|
|
||||||
"y": 96,
|
|
||||||
"z": 0
|
|
||||||
},
|
|
||||||
"config": {
|
|
||||||
"type": "PRCXI9300Container",
|
|
||||||
"size_x": 127,
|
|
||||||
"size_y": 85.5,
|
|
||||||
"size_z": 10,
|
|
||||||
"rotation": {
|
|
||||||
"x": 0,
|
|
||||||
"y": 0,
|
|
||||||
"z": 0,
|
|
||||||
"type": "Rotation"
|
|
||||||
},
|
|
||||||
"category": "plate",
|
|
||||||
"model": null,
|
|
||||||
"barcode": null,
|
|
||||||
"ordering": {},
|
|
||||||
"sites": [
|
|
||||||
{
|
|
||||||
"label": "T11",
|
|
||||||
"visible": true,
|
|
||||||
"position": { "x": 0, "y": 0, "z": 0 },
|
|
||||||
"size": { "width": 128.0, "height": 86, "depth": 0 },
|
|
||||||
"content_type": [
|
|
||||||
"plate",
|
|
||||||
"tip_rack",
|
|
||||||
"plates",
|
|
||||||
"tip_racks",
|
|
||||||
"tube_rack"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
"data": {}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": "T12",
|
|
||||||
"name": "T12",
|
|
||||||
"children": [],
|
|
||||||
"parent": "PRCXI_Deck",
|
|
||||||
"type": "plate",
|
|
||||||
"class": "",
|
|
||||||
"position": {
|
|
||||||
"x": 414,
|
|
||||||
"y": 96,
|
|
||||||
"z": 0
|
|
||||||
},
|
|
||||||
"config": {
|
|
||||||
"type": "PRCXI9300Container",
|
|
||||||
"size_x": 127,
|
|
||||||
"size_y": 85.5,
|
|
||||||
"size_z": 10,
|
|
||||||
"rotation": {
|
|
||||||
"x": 0,
|
|
||||||
"y": 0,
|
|
||||||
"z": 0,
|
|
||||||
"type": "Rotation"
|
|
||||||
},
|
|
||||||
"category": "plate",
|
|
||||||
"model": null,
|
|
||||||
"barcode": null,
|
|
||||||
"ordering": {},
|
|
||||||
"sites": [
|
|
||||||
{
|
|
||||||
"label": "T12",
|
|
||||||
"visible": true,
|
|
||||||
"position": { "x": 0, "y": 0, "z": 0 },
|
|
||||||
"size": { "width": 128.0, "height": 86, "depth": 0 },
|
|
||||||
"content_type": [
|
|
||||||
"plate",
|
|
||||||
"tip_rack",
|
|
||||||
"plates",
|
|
||||||
"tip_racks",
|
|
||||||
"tube_rack"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
"data": {}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": "T13",
|
|
||||||
"name": "T13",
|
|
||||||
"children": [],
|
|
||||||
"parent": "PRCXI_Deck",
|
|
||||||
"type": "plate",
|
|
||||||
"class": "",
|
|
||||||
"position": {
|
|
||||||
"x": 0,
|
|
||||||
"y": 0,
|
|
||||||
"z": 0
|
|
||||||
},
|
|
||||||
"config": {
|
|
||||||
"type": "PRCXI9300Container",
|
|
||||||
"size_x": 127,
|
|
||||||
"size_y": 85.5,
|
|
||||||
"size_z": 10,
|
|
||||||
"rotation": {
|
|
||||||
"x": 0,
|
|
||||||
"y": 0,
|
|
||||||
"z": 0,
|
|
||||||
"type": "Rotation"
|
|
||||||
},
|
|
||||||
"category": "plate",
|
|
||||||
"model": null,
|
|
||||||
"barcode": null,
|
|
||||||
"ordering": {},
|
|
||||||
"sites": [
|
|
||||||
{
|
|
||||||
"label": "T13",
|
|
||||||
"visible": true,
|
|
||||||
"position": { "x": 0, "y": 0, "z": 0 },
|
|
||||||
"size": { "width": 128.0, "height": 86, "depth": 0 },
|
|
||||||
"content_type": [
|
|
||||||
"plate",
|
|
||||||
"tip_rack",
|
|
||||||
"plates",
|
|
||||||
"tip_racks",
|
|
||||||
"tube_rack"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
"data": {}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": "T14",
|
|
||||||
"name": "T14",
|
|
||||||
"children": [],
|
|
||||||
"parent": "PRCXI_Deck",
|
|
||||||
"type": "plate",
|
|
||||||
"class": "",
|
|
||||||
"position": {
|
|
||||||
"x": 138,
|
|
||||||
"y": 0,
|
|
||||||
"z": 0
|
|
||||||
},
|
|
||||||
"config": {
|
|
||||||
"type": "PRCXI9300Container",
|
|
||||||
"size_x": 127,
|
|
||||||
"size_y": 85.5,
|
|
||||||
"size_z": 10,
|
|
||||||
"rotation": {
|
|
||||||
"x": 0,
|
|
||||||
"y": 0,
|
|
||||||
"z": 0,
|
|
||||||
"type": "Rotation"
|
|
||||||
},
|
|
||||||
"category": "plate",
|
|
||||||
"model": null,
|
|
||||||
"barcode": null,
|
|
||||||
"ordering": {},
|
|
||||||
"sites": [
|
|
||||||
{
|
|
||||||
"label": "T14",
|
|
||||||
"visible": true,
|
|
||||||
"position": { "x": 0, "y": 0, "z": 0 },
|
|
||||||
"size": { "width": 128.0, "height": 86, "depth": 0 },
|
|
||||||
"content_type": [
|
|
||||||
"plate",
|
|
||||||
"tip_rack",
|
|
||||||
"plates",
|
|
||||||
"tip_racks",
|
|
||||||
"tube_rack"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
"data": {}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": "T15",
|
|
||||||
"name": "T15",
|
|
||||||
"children": [],
|
|
||||||
"parent": "PRCXI_Deck",
|
|
||||||
"type": "plate",
|
|
||||||
"class": "",
|
|
||||||
"position": {
|
|
||||||
"x": 276,
|
|
||||||
"y": 0,
|
|
||||||
"z": 0
|
|
||||||
},
|
|
||||||
"config": {
|
|
||||||
"type": "PRCXI9300Container",
|
|
||||||
"size_x": 127,
|
|
||||||
"size_y": 85.5,
|
|
||||||
"size_z": 10,
|
|
||||||
"rotation": {
|
|
||||||
"x": 0,
|
|
||||||
"y": 0,
|
|
||||||
"z": 0,
|
|
||||||
"type": "Rotation"
|
|
||||||
},
|
|
||||||
"category": "plate",
|
|
||||||
"model": null,
|
|
||||||
"barcode": null,
|
|
||||||
"ordering": {},
|
|
||||||
"sites": [
|
|
||||||
{
|
|
||||||
"label": "T15",
|
|
||||||
"visible": true,
|
|
||||||
"position": { "x": 0, "y": 0, "z": 0 },
|
|
||||||
"size": { "width": 128.0, "height": 86, "depth": 0 },
|
|
||||||
"content_type": [
|
|
||||||
"plate",
|
|
||||||
"tip_rack",
|
|
||||||
"plates",
|
|
||||||
"tip_racks",
|
|
||||||
"tube_rack"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
"data": {}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": "T16",
|
|
||||||
"name": "T16",
|
|
||||||
"children": [],
|
|
||||||
"parent": "PRCXI_Deck",
|
|
||||||
"type": "plate",
|
|
||||||
"class": "",
|
|
||||||
"position": {
|
|
||||||
"x": 414,
|
|
||||||
"y": 0,
|
|
||||||
"z": 0
|
|
||||||
},
|
|
||||||
"config": {
|
|
||||||
"type": "PRCXI9300Container",
|
|
||||||
"size_x": 127,
|
|
||||||
"size_y": 85.5,
|
|
||||||
"size_z": 10,
|
|
||||||
"rotation": {
|
|
||||||
"x": 0,
|
|
||||||
"y": 0,
|
|
||||||
"z": 0,
|
|
||||||
"type": "Rotation"
|
|
||||||
},
|
|
||||||
"category": "plate",
|
|
||||||
"model": null,
|
|
||||||
"barcode": null,
|
|
||||||
"ordering": {},
|
|
||||||
"sites": [
|
|
||||||
{
|
|
||||||
"label": "T16",
|
|
||||||
"visible": true,
|
|
||||||
"position": { "x": 0, "y": 0, "z": 0 },
|
|
||||||
"size": { "width": 128.0, "height": 86, "depth": 0 },
|
|
||||||
"content_type": [
|
|
||||||
"plate",
|
|
||||||
"tip_rack",
|
|
||||||
"plates",
|
|
||||||
"tip_racks",
|
|
||||||
"tube_rack"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
"data": {}
|
|
||||||
}
|
|
||||||
],
|
|
||||||
"edges": []
|
|
||||||
}
|
|
||||||
29
unilabos/test/experiments/xkc_sensor_test.json
Normal file
29
unilabos/test/experiments/xkc_sensor_test.json
Normal file
@@ -0,0 +1,29 @@
|
|||||||
|
{
|
||||||
|
"nodes": [
|
||||||
|
{
|
||||||
|
"id": "Liquid_Sensor_1",
|
||||||
|
"name": "XKC Sensor",
|
||||||
|
"children": [],
|
||||||
|
"parent": null,
|
||||||
|
"type": "device",
|
||||||
|
"class": "sensor.xkc_rs485",
|
||||||
|
"position": {
|
||||||
|
"x": 0,
|
||||||
|
"y": 0,
|
||||||
|
"z": 0
|
||||||
|
},
|
||||||
|
"config": {
|
||||||
|
"port": "/dev/tty.usbserial-3110",
|
||||||
|
"baudrate": 9600,
|
||||||
|
"device_id": 1,
|
||||||
|
"threshold": 300,
|
||||||
|
"timeout": 3.0
|
||||||
|
},
|
||||||
|
"data": {
|
||||||
|
"level": false,
|
||||||
|
"rssi": 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"links": []
|
||||||
|
}
|
||||||
28
unilabos/test/experiments/zdt_motor_test.json
Normal file
28
unilabos/test/experiments/zdt_motor_test.json
Normal file
@@ -0,0 +1,28 @@
|
|||||||
|
{
|
||||||
|
"nodes": [
|
||||||
|
{
|
||||||
|
"id": "ZDT_Motor",
|
||||||
|
"name": "ZDT Motor",
|
||||||
|
"children": [],
|
||||||
|
"parent": null,
|
||||||
|
"type": "device",
|
||||||
|
"class": "motor.zdt_x42",
|
||||||
|
"position": {
|
||||||
|
"x": 0,
|
||||||
|
"y": 0,
|
||||||
|
"z": 0
|
||||||
|
},
|
||||||
|
"config": {
|
||||||
|
"port": "/dev/tty.usbserial-3110",
|
||||||
|
"baudrate": 115200,
|
||||||
|
"device_id": 1,
|
||||||
|
"debug": true
|
||||||
|
},
|
||||||
|
"data": {
|
||||||
|
"position": 0,
|
||||||
|
"status": "idle"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"links": []
|
||||||
|
}
|
||||||
@@ -27,7 +27,6 @@ __all__ = [
|
|||||||
|
|
||||||
from ast import Constant
|
from ast import Constant
|
||||||
|
|
||||||
from unilabos.resources.resource_tracker import PARAM_SAMPLE_UUIDS
|
|
||||||
from unilabos.utils import logger
|
from unilabos.utils import logger
|
||||||
from unilabos.utils.decorator import is_not_action
|
from unilabos.utils.decorator import is_not_action
|
||||||
|
|
||||||
@@ -342,18 +341,13 @@ class ImportManager:
|
|||||||
result["action_methods"][method_name] = method_info
|
result["action_methods"][method_name] = method_info
|
||||||
return result
|
return result
|
||||||
|
|
||||||
def _analyze_method_signature(self, method, skip_unilabos_params: bool = True) -> Dict[str, Any]:
|
def _analyze_method_signature(self, method) -> Dict[str, Any]:
|
||||||
"""
|
"""
|
||||||
分析方法签名,提取具体的命名参数信息
|
分析方法签名,提取具体的命名参数信息
|
||||||
|
|
||||||
注意:此方法会跳过*args和**kwargs,只提取具体的命名参数
|
注意:此方法会跳过*args和**kwargs,只提取具体的命名参数
|
||||||
这样可以确保通过**dict方式传参时的准确性
|
这样可以确保通过**dict方式传参时的准确性
|
||||||
|
|
||||||
Args:
|
|
||||||
method: 要分析的方法
|
|
||||||
skip_unilabos_params: 是否跳过 unilabos 系统参数(如 sample_uuids),
|
|
||||||
registry 补全时为 True,JsonCommand 执行时为 False
|
|
||||||
|
|
||||||
示例用法:
|
示例用法:
|
||||||
method_info = self._analyze_method_signature(some_method)
|
method_info = self._analyze_method_signature(some_method)
|
||||||
params = {"param1": "value1", "param2": "value2"}
|
params = {"param1": "value1", "param2": "value2"}
|
||||||
@@ -374,10 +368,6 @@ class ImportManager:
|
|||||||
if param.kind == param.VAR_KEYWORD: # **kwargs
|
if param.kind == param.VAR_KEYWORD: # **kwargs
|
||||||
continue
|
continue
|
||||||
|
|
||||||
# 跳过 sample_uuids 参数(由系统自动注入,registry 补全时跳过)
|
|
||||||
if skip_unilabos_params and param_name == PARAM_SAMPLE_UUIDS:
|
|
||||||
continue
|
|
||||||
|
|
||||||
is_required = param.default == inspect.Parameter.empty
|
is_required = param.default == inspect.Parameter.empty
|
||||||
if is_required:
|
if is_required:
|
||||||
num_required += 1
|
num_required += 1
|
||||||
@@ -573,9 +563,6 @@ class ImportManager:
|
|||||||
for i, arg in enumerate(node.args.args):
|
for i, arg in enumerate(node.args.args):
|
||||||
if arg.arg == "self":
|
if arg.arg == "self":
|
||||||
continue
|
continue
|
||||||
# 跳过 sample_uuids 参数(由系统自动注入)
|
|
||||||
if arg.arg == PARAM_SAMPLE_UUIDS:
|
|
||||||
continue
|
|
||||||
arg_info = {
|
arg_info = {
|
||||||
"name": arg.arg,
|
"name": arg.arg,
|
||||||
"type": None,
|
"type": None,
|
||||||
|
|||||||
@@ -19,9 +19,7 @@
|
|||||||
|
|
||||||
第一步: 按 slot 去重创建 create_resource 节点(创建板子)
|
第一步: 按 slot 去重创建 create_resource 节点(创建板子)
|
||||||
--------------------------------------------------------------------------------
|
--------------------------------------------------------------------------------
|
||||||
- 首先创建一个 Group 节点(type="Group", minimized=true),用于包含所有 create_resource 节点
|
|
||||||
- 遍历所有 reagent,按 slot 去重,为每个唯一的 slot 创建一个板子
|
- 遍历所有 reagent,按 slot 去重,为每个唯一的 slot 创建一个板子
|
||||||
- 所有 create_resource 节点的 parent_uuid 指向 Group 节点,minimized=true
|
|
||||||
- 生成参数:
|
- 生成参数:
|
||||||
res_id: plate_slot_{slot}
|
res_id: plate_slot_{slot}
|
||||||
device_id: /PRCXI
|
device_id: /PRCXI
|
||||||
@@ -31,13 +29,11 @@
|
|||||||
- 输出端口: labware(用于连接 set_liquid_from_plate)
|
- 输出端口: labware(用于连接 set_liquid_from_plate)
|
||||||
- 控制流: create_resource 之间通过 ready 端口串联
|
- 控制流: create_resource 之间通过 ready 端口串联
|
||||||
|
|
||||||
示例: slot=1, slot=4 -> 创建 1 个 Group + 2 个 create_resource 节点
|
示例: slot=1, slot=4 -> 创建 2 个 create_resource 节点
|
||||||
|
|
||||||
第二步: 为每个 reagent 创建 set_liquid_from_plate 节点(设置液体)
|
第二步: 为每个 reagent 创建 set_liquid_from_plate 节点(设置液体)
|
||||||
--------------------------------------------------------------------------------
|
--------------------------------------------------------------------------------
|
||||||
- 首先创建一个 Group 节点(type="Group", minimized=true),用于包含所有 set_liquid_from_plate 节点
|
|
||||||
- 遍历所有 reagent,为每个试剂创建 set_liquid_from_plate 节点
|
- 遍历所有 reagent,为每个试剂创建 set_liquid_from_plate 节点
|
||||||
- 所有 set_liquid_from_plate 节点的 parent_uuid 指向 Group 节点,minimized=true
|
|
||||||
- 生成参数:
|
- 生成参数:
|
||||||
plate: [](通过连接传递,来自 create_resource 的 labware)
|
plate: [](通过连接传递,来自 create_resource 的 labware)
|
||||||
well_names: ["A1", "A3", "A5"](来自 reagent 的 well 数组)
|
well_names: ["A1", "A3", "A5"](来自 reagent 的 well 数组)
|
||||||
@@ -60,11 +56,7 @@
|
|||||||
==================== 连接关系图 ====================
|
==================== 连接关系图 ====================
|
||||||
|
|
||||||
控制流 (ready 端口串联):
|
控制流 (ready 端口串联):
|
||||||
- create_resource 之间: 无 ready 连接
|
create_resource_1 -> create_resource_2 -> ... -> set_liquid_1 -> set_liquid_2 -> ... -> transfer_liquid_1 -> transfer_liquid_2 -> ...
|
||||||
- set_liquid_from_plate 之间: 无 ready 连接
|
|
||||||
- create_resource 与 set_liquid_from_plate 之间: 无 ready 连接
|
|
||||||
- transfer_liquid 之间: 通过 ready 端口串联
|
|
||||||
transfer_liquid_1 -> transfer_liquid_2 -> transfer_liquid_3 -> ...
|
|
||||||
|
|
||||||
物料流:
|
物料流:
|
||||||
[create_resource] --labware--> [set_liquid_from_plate] --output_wells--> [transfer_liquid] --sources_out/targets_out--> [下一个 transfer_liquid]
|
[create_resource] --labware--> [set_liquid_from_plate] --output_wells--> [transfer_liquid] --sources_out/targets_out--> [下一个 transfer_liquid]
|
||||||
@@ -84,13 +76,6 @@ transfer_liquid:
|
|||||||
输入: sources -> sources_identifier, targets -> targets_identifier
|
输入: sources -> sources_identifier, targets -> targets_identifier
|
||||||
输出: sources -> sources_out, targets -> targets_out
|
输出: sources -> sources_out, targets -> targets_out
|
||||||
|
|
||||||
==================== 设备名配置 (device_name) ====================
|
|
||||||
|
|
||||||
每个节点都有 device_name 字段,指定在哪个设备上执行:
|
|
||||||
- create_resource: device_name = "host_node"(固定)
|
|
||||||
- set_liquid_from_plate: device_name = "PRCXI"(可配置,见 DEVICE_NAME_DEFAULT)
|
|
||||||
- transfer_liquid 等动作: device_name = "PRCXI"(可配置,见 DEVICE_NAME_DEFAULT)
|
|
||||||
|
|
||||||
==================== 校验规则 ====================
|
==================== 校验规则 ====================
|
||||||
|
|
||||||
- 检查 sources/targets 是否在 reagent 中定义
|
- 检查 sources/targets 是否在 reagent 中定义
|
||||||
@@ -112,13 +97,6 @@ Json = Dict[str, Any]
|
|||||||
|
|
||||||
# ==================== 默认配置 ====================
|
# ==================== 默认配置 ====================
|
||||||
|
|
||||||
# 设备名配置
|
|
||||||
DEVICE_NAME_HOST = "host_node" # create_resource 固定在 host_node 上执行
|
|
||||||
DEVICE_NAME_DEFAULT = "PRCXI" # transfer_liquid, set_liquid_from_plate 等动作的默认设备名
|
|
||||||
|
|
||||||
# 节点类型
|
|
||||||
NODE_TYPE_DEFAULT = "ILab" # 所有节点的默认类型
|
|
||||||
|
|
||||||
# create_resource 节点默认参数
|
# create_resource 节点默认参数
|
||||||
CREATE_RESOURCE_DEFAULTS = {
|
CREATE_RESOURCE_DEFAULTS = {
|
||||||
"device_id": "/PRCXI",
|
"device_id": "/PRCXI",
|
||||||
@@ -389,23 +367,9 @@ def build_protocol_graph(
|
|||||||
"res_id": res_id,
|
"res_id": res_id,
|
||||||
}
|
}
|
||||||
|
|
||||||
# 创建 Group 节点,包含所有 create_resource 节点
|
|
||||||
group_node_id = str(uuid.uuid4())
|
|
||||||
G.add_node(
|
|
||||||
group_node_id,
|
|
||||||
name="Resources Group",
|
|
||||||
type="Group",
|
|
||||||
parent_uuid="",
|
|
||||||
lab_node_type="Device",
|
|
||||||
template_name="",
|
|
||||||
resource_name="",
|
|
||||||
footer="",
|
|
||||||
minimized=True,
|
|
||||||
param=None,
|
|
||||||
)
|
|
||||||
|
|
||||||
# 为每个唯一的 slot 创建 create_resource 节点
|
# 为每个唯一的 slot 创建 create_resource 节点
|
||||||
res_index = 0
|
res_index = 0
|
||||||
|
last_create_resource_id = None
|
||||||
for slot, info in slots_info.items():
|
for slot, info in slots_info.items():
|
||||||
node_id = str(uuid.uuid4())
|
node_id = str(uuid.uuid4())
|
||||||
res_id = info["res_id"]
|
res_id = info["res_id"]
|
||||||
@@ -419,10 +383,6 @@ def build_protocol_graph(
|
|||||||
description=f"Create plate on slot {slot}",
|
description=f"Create plate on slot {slot}",
|
||||||
lab_node_type="Labware",
|
lab_node_type="Labware",
|
||||||
footer="create_resource-host_node",
|
footer="create_resource-host_node",
|
||||||
device_name=DEVICE_NAME_HOST,
|
|
||||||
type=NODE_TYPE_DEFAULT,
|
|
||||||
parent_uuid=group_node_id, # 指向 Group 节点
|
|
||||||
minimized=True, # 折叠显示
|
|
||||||
param={
|
param={
|
||||||
"res_id": res_id,
|
"res_id": res_id,
|
||||||
"device_id": CREATE_RESOURCE_DEFAULTS["device_id"],
|
"device_id": CREATE_RESOURCE_DEFAULTS["device_id"],
|
||||||
@@ -434,25 +394,14 @@ def build_protocol_graph(
|
|||||||
)
|
)
|
||||||
slot_to_create_resource[slot] = node_id
|
slot_to_create_resource[slot] = node_id
|
||||||
|
|
||||||
# create_resource 之间不需要 ready 连接
|
# create_resource 之间通过 ready 串联
|
||||||
|
if last_create_resource_id is not None:
|
||||||
|
G.add_edge(last_create_resource_id, node_id, source_port="ready", target_port="ready")
|
||||||
|
last_create_resource_id = node_id
|
||||||
|
|
||||||
# ==================== 第二步:为每个 reagent 创建 set_liquid_from_plate 节点 ====================
|
# ==================== 第二步:为每个 reagent 创建 set_liquid_from_plate 节点 ====================
|
||||||
# 创建 Group 节点,包含所有 set_liquid_from_plate 节点
|
|
||||||
set_liquid_group_id = str(uuid.uuid4())
|
|
||||||
G.add_node(
|
|
||||||
set_liquid_group_id,
|
|
||||||
name="SetLiquid Group",
|
|
||||||
type="Group",
|
|
||||||
parent_uuid="",
|
|
||||||
lab_node_type="Device",
|
|
||||||
template_name="",
|
|
||||||
resource_name="",
|
|
||||||
footer="",
|
|
||||||
minimized=True,
|
|
||||||
param=None,
|
|
||||||
)
|
|
||||||
|
|
||||||
set_liquid_index = 0
|
set_liquid_index = 0
|
||||||
|
last_set_liquid_id = last_create_resource_id # set_liquid_from_plate 连接在 create_resource 之后
|
||||||
|
|
||||||
for labware_id, item in labware_info.items():
|
for labware_id, item in labware_info.items():
|
||||||
# 跳过 Tip/Rack 类型
|
# 跳过 Tip/Rack 类型
|
||||||
@@ -481,10 +430,6 @@ def build_protocol_graph(
|
|||||||
description=f"Set liquid: {labware_id}",
|
description=f"Set liquid: {labware_id}",
|
||||||
lab_node_type="Reagent",
|
lab_node_type="Reagent",
|
||||||
footer="set_liquid_from_plate-liquid_handler.prcxi",
|
footer="set_liquid_from_plate-liquid_handler.prcxi",
|
||||||
device_name=DEVICE_NAME_DEFAULT,
|
|
||||||
type=NODE_TYPE_DEFAULT,
|
|
||||||
parent_uuid=set_liquid_group_id, # 指向 Group 节点
|
|
||||||
minimized=True, # 折叠显示
|
|
||||||
param={
|
param={
|
||||||
"plate": [], # 通过连接传递
|
"plate": [], # 通过连接传递
|
||||||
"well_names": wells, # 孔位名数组,如 ["A1", "A3", "A5"]
|
"well_names": wells, # 孔位名数组,如 ["A1", "A3", "A5"]
|
||||||
@@ -493,7 +438,10 @@ def build_protocol_graph(
|
|||||||
},
|
},
|
||||||
)
|
)
|
||||||
|
|
||||||
# set_liquid_from_plate 之间不需要 ready 连接
|
# ready 连接:上一个节点 -> set_liquid_from_plate
|
||||||
|
if last_set_liquid_id is not None:
|
||||||
|
G.add_edge(last_set_liquid_id, node_id, source_port="ready", target_port="ready")
|
||||||
|
last_set_liquid_id = node_id
|
||||||
|
|
||||||
# 物料流:create_resource 的 labware -> set_liquid_from_plate 的 input_plate
|
# 物料流:create_resource 的 labware -> set_liquid_from_plate 的 input_plate
|
||||||
create_res_node_id = slot_to_create_resource.get(slot)
|
create_res_node_id = slot_to_create_resource.get(slot)
|
||||||
@@ -503,8 +451,7 @@ def build_protocol_graph(
|
|||||||
# set_liquid_from_plate 的输出 output_wells 用于连接 transfer_liquid
|
# set_liquid_from_plate 的输出 output_wells 用于连接 transfer_liquid
|
||||||
resource_last_writer[labware_id] = f"{node_id}:output_wells"
|
resource_last_writer[labware_id] = f"{node_id}:output_wells"
|
||||||
|
|
||||||
# transfer_liquid 之间通过 ready 串联,从 None 开始
|
last_control_node_id = last_set_liquid_id
|
||||||
last_control_node_id = None
|
|
||||||
|
|
||||||
# 端口名称映射:JSON 字段名 -> 实际 handle key
|
# 端口名称映射:JSON 字段名 -> 实际 handle key
|
||||||
INPUT_PORT_MAPPING = {
|
INPUT_PORT_MAPPING = {
|
||||||
@@ -597,11 +544,9 @@ def build_protocol_graph(
|
|||||||
if param_key in params:
|
if param_key in params:
|
||||||
params[param_key] = []
|
params[param_key] = []
|
||||||
|
|
||||||
# 更新 step 的 param、footer、device_name 和 type
|
# 更新 step 的 param 和 footer
|
||||||
step_copy = step.copy()
|
step_copy = step.copy()
|
||||||
step_copy["param"] = params
|
step_copy["param"] = params
|
||||||
step_copy["device_name"] = DEVICE_NAME_DEFAULT # 动作节点使用默认设备名
|
|
||||||
step_copy["type"] = NODE_TYPE_DEFAULT # 节点类型
|
|
||||||
|
|
||||||
# 如果有警告,修改 footer 添加警告标记(警告放前面)
|
# 如果有警告,修改 footer 添加警告标记(警告放前面)
|
||||||
if warnings:
|
if warnings:
|
||||||
|
|||||||
Reference in New Issue
Block a user