mirror of
https://github.com/deepmodeling/Uni-Lab-OS
synced 2026-03-24 09:17:39 +00:00
new workflow & prcxi slot removal
This commit is contained in:
@@ -55,6 +55,7 @@ from unilabos.devices.liquid_handling.liquid_handler_abstract import (
|
|||||||
TransferLiquidReturn,
|
TransferLiquidReturn,
|
||||||
)
|
)
|
||||||
from unilabos.registry.placeholder_type import ResourceSlot
|
from unilabos.registry.placeholder_type import ResourceSlot
|
||||||
|
from unilabos.resources.resource_tracker import ResourceTreeSet
|
||||||
from unilabos.ros.nodes.base_device_node import BaseROS2DeviceNode
|
from unilabos.ros.nodes.base_device_node import BaseROS2DeviceNode
|
||||||
|
|
||||||
|
|
||||||
@@ -90,20 +91,103 @@ class PRCXI9300Deck(Deck):
|
|||||||
该类定义了 PRCXI 9300 的工作台布局和槽位信息。
|
该类定义了 PRCXI 9300 的工作台布局和槽位信息。
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def __init__(self, name: str, size_x: float, size_y: float, size_z: float, **kwargs):
|
# T1-T16 默认位置 (4列×4行)
|
||||||
|
_DEFAULT_SITE_POSITIONS = [
|
||||||
|
(0, 288, 0), (138, 288, 0), (276, 288, 0), (414, 288, 0), # T1-T4
|
||||||
|
(0, 192, 0), (138, 192, 0), (276, 192, 0), (414, 192, 0), # T5-T8
|
||||||
|
(0, 96, 0), (138, 96, 0), (276, 96, 0), (414, 96, 0), # T9-T12
|
||||||
|
(0, 0, 0), (138, 0, 0), (276, 0, 0), (414, 0, 0), # T13-T16
|
||||||
|
]
|
||||||
|
_DEFAULT_SITE_SIZE = {"width": 128.0, "height": 86, "depth": 0}
|
||||||
|
_DEFAULT_CONTENT_TYPE = ["plate", "tip_rack", "plates", "tip_racks", "tube_rack"]
|
||||||
|
|
||||||
|
def __init__(self, name: str, size_x: float, size_y: float, size_z: float,
|
||||||
|
sites: Optional[List[Dict[str, Any]]] = None, **kwargs):
|
||||||
super().__init__(size_x, size_y, size_z, name)
|
super().__init__(size_x, size_y, size_z, name)
|
||||||
self.slots = [None] * 16 # PRCXI 9300/9320 最大有 16 个槽位
|
if sites is not None:
|
||||||
self.slot_locations = [Coordinate(0, 0, 0)] * 16
|
self.sites: List[Dict[str, Any]] = [dict(s) for s in sites]
|
||||||
|
else:
|
||||||
|
self.sites = []
|
||||||
|
for i, (x, y, z) in enumerate(self._DEFAULT_SITE_POSITIONS):
|
||||||
|
self.sites.append({
|
||||||
|
"label": f"T{i + 1}",
|
||||||
|
"visible": True,
|
||||||
|
"position": {"x": x, "y": y, "z": z},
|
||||||
|
"size": dict(self._DEFAULT_SITE_SIZE),
|
||||||
|
"content_type": list(self._DEFAULT_CONTENT_TYPE),
|
||||||
|
})
|
||||||
|
# _ordering: label -> None, 用于外部通过 list(keys()).index(site) 将 Tn 转换为 spot index
|
||||||
|
self._ordering = collections.OrderedDict(
|
||||||
|
(site["label"], None) for site in self.sites
|
||||||
|
)
|
||||||
|
|
||||||
|
def _get_site_location(self, idx: int) -> Coordinate:
|
||||||
|
pos = self.sites[idx]["position"]
|
||||||
|
return Coordinate(pos["x"], pos["y"], pos["z"])
|
||||||
|
|
||||||
|
def _get_site_resource(self, idx: int) -> Optional[Resource]:
|
||||||
|
site_loc = self._get_site_location(idx)
|
||||||
|
for child in self.children:
|
||||||
|
if child.location == site_loc:
|
||||||
|
return child
|
||||||
|
return None
|
||||||
|
|
||||||
|
def assign_child_resource(
|
||||||
|
self,
|
||||||
|
resource: Resource,
|
||||||
|
location: Optional[Coordinate] = None,
|
||||||
|
reassign: bool = True,
|
||||||
|
spot: Optional[int] = None,
|
||||||
|
):
|
||||||
|
idx = spot
|
||||||
|
if spot is not None:
|
||||||
|
idx = spot
|
||||||
|
else:
|
||||||
|
for i, site in enumerate(self.sites):
|
||||||
|
site_loc = self._get_site_location(i)
|
||||||
|
if site.get("label") == resource.name:
|
||||||
|
idx = i
|
||||||
|
break
|
||||||
|
if location is not None and site_loc == location:
|
||||||
|
idx = i
|
||||||
|
break
|
||||||
|
|
||||||
|
if idx is None:
|
||||||
|
for i in range(len(self.sites)):
|
||||||
|
if self._get_site_resource(i) is None:
|
||||||
|
idx = i
|
||||||
|
break
|
||||||
|
|
||||||
|
if idx is None:
|
||||||
|
raise ValueError(f"No available site on deck '{self.name}' for resource '{resource.name}'")
|
||||||
|
|
||||||
|
if not reassign and self._get_site_resource(idx) is not None:
|
||||||
|
raise ValueError(f"Site {idx} ('{self.sites[idx]['label']}') is already occupied")
|
||||||
|
|
||||||
|
loc = self._get_site_location(idx)
|
||||||
|
super().assign_child_resource(resource, location=loc, reassign=reassign)
|
||||||
|
|
||||||
def assign_child_at_slot(self, resource: Resource, slot: int, reassign: bool = False) -> None:
|
def assign_child_at_slot(self, resource: Resource, slot: int, reassign: bool = False) -> None:
|
||||||
if self.slots[slot - 1] is not None and not reassign:
|
self.assign_child_resource(resource, spot=slot - 1, reassign=reassign)
|
||||||
raise ValueError(f"Spot {slot} is already occupied")
|
|
||||||
|
|
||||||
self.slots[slot - 1] = resource
|
def serialize(self) -> dict:
|
||||||
super().assign_child_resource(resource, location=self.slot_locations[slot - 1])
|
data = super().serialize()
|
||||||
|
sites_out = []
|
||||||
|
for i, site in enumerate(self.sites):
|
||||||
|
occupied = self._get_site_resource(i)
|
||||||
|
sites_out.append({
|
||||||
|
"label": site["label"],
|
||||||
|
"visible": site.get("visible", True),
|
||||||
|
"occupied_by": occupied.name if occupied is not None else None,
|
||||||
|
"position": site["position"],
|
||||||
|
"size": site["size"],
|
||||||
|
"content_type": site["content_type"],
|
||||||
|
})
|
||||||
|
data["sites"] = sites_out
|
||||||
|
return data
|
||||||
|
|
||||||
|
|
||||||
class PRCXI9300Container(Plate):
|
class PRCXI9300Container(Container):
|
||||||
"""PRCXI 9300 的专用 Container 类,继承自 Plate,用于槽位定位和未知模块。
|
"""PRCXI 9300 的专用 Container 类,继承自 Plate,用于槽位定位和未知模块。
|
||||||
|
|
||||||
该类定义了 PRCXI 9300 的工作台布局和槽位信息。
|
该类定义了 PRCXI 9300 的工作台布局和槽位信息。
|
||||||
@@ -116,11 +200,10 @@ class PRCXI9300Container(Plate):
|
|||||||
size_y: float,
|
size_y: float,
|
||||||
size_z: float,
|
size_z: float,
|
||||||
category: str,
|
category: str,
|
||||||
ordering: collections.OrderedDict,
|
|
||||||
model: Optional[str] = None,
|
model: Optional[str] = None,
|
||||||
**kwargs,
|
**kwargs,
|
||||||
):
|
):
|
||||||
super().__init__(name, size_x, size_y, size_z, category=category, ordering=ordering, model=model)
|
super().__init__(name, size_x, size_y, size_z, category=category, model=model)
|
||||||
self._unilabos_state = {}
|
self._unilabos_state = {}
|
||||||
|
|
||||||
def load_state(self, state: Dict[str, Any]) -> None:
|
def load_state(self, state: Dict[str, Any]) -> None:
|
||||||
@@ -567,14 +650,14 @@ class PRCXI9300Handler(LiquidHandlerAbstract):
|
|||||||
tablets_info = []
|
tablets_info = []
|
||||||
count = 0
|
count = 0
|
||||||
for child in deck.children:
|
for child in deck.children:
|
||||||
if child.children:
|
# 如果放其他类型的物料,是不可以的
|
||||||
if "Material" in child.children[0]._unilabos_state:
|
if hasattr(child, "_unilabos_state") and "Material" in child._unilabos_state:
|
||||||
number = int(child.name.replace("T", ""))
|
number = int(child.name.replace("T", ""))
|
||||||
tablets_info.append(
|
tablets_info.append(
|
||||||
WorkTablets(
|
WorkTablets(
|
||||||
Number=number, Code=f"T{number}", Material=child.children[0]._unilabos_state["Material"]
|
Number=number, Code=f"T{number}", Material=child._unilabos_state["Material"]
|
||||||
)
|
|
||||||
)
|
)
|
||||||
|
)
|
||||||
if is_9320:
|
if is_9320:
|
||||||
print("当前设备是9320")
|
print("当前设备是9320")
|
||||||
# 始终初始化 step_mode 属性
|
# 始终初始化 step_mode 属性
|
||||||
|
|||||||
@@ -76,7 +76,7 @@ def canonicalize_nodes_data(
|
|||||||
if sample_id:
|
if sample_id:
|
||||||
logger.error(f"{node}的sample_id参数已弃用,sample_id: {sample_id}")
|
logger.error(f"{node}的sample_id参数已弃用,sample_id: {sample_id}")
|
||||||
for k in list(node.keys()):
|
for k in list(node.keys()):
|
||||||
if k not in ["id", "uuid", "name", "description", "schema", "model", "icon", "parent_uuid", "parent", "type", "class", "position", "config", "data", "children", "pose"]:
|
if k not in ["id", "uuid", "name", "description", "schema", "model", "icon", "parent_uuid", "parent", "type", "class", "position", "config", "data", "children", "pose", "extra"]:
|
||||||
v = node.pop(k)
|
v = node.pop(k)
|
||||||
node["config"][k] = v
|
node["config"][k] = v
|
||||||
if outer_host_node_id is not None:
|
if outer_host_node_id is not None:
|
||||||
|
|||||||
@@ -616,7 +616,7 @@ class ResourceTreeSet(object):
|
|||||||
plr_resources.append(plr_resource)
|
plr_resources.append(plr_resource)
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(f"转换 PLR 资源失败: {e}")
|
logger.error(f"转换 PLR 资源失败: {e} {str(plr_dict)[:1000]}")
|
||||||
import traceback
|
import traceback
|
||||||
|
|
||||||
logger.error(f"堆栈: {traceback.format_exc()}")
|
logger.error(f"堆栈: {traceback.format_exc()}")
|
||||||
|
|||||||
@@ -915,8 +915,24 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
else []
|
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 = None
|
||||||
site_name = site_names[site_index]
|
try:
|
||||||
|
# sites 可能是 Resource 列表或 dict 列表 (如 PRCXI9300Deck)
|
||||||
|
# 只有itemized_carrier在使用,准备弃用
|
||||||
|
site_index = sites.index(original_instance)
|
||||||
|
except ValueError:
|
||||||
|
# dict 类型的 sites: 通过name匹配
|
||||||
|
for idx, site in enumerate(sites):
|
||||||
|
if original_instance.name == site["occupied_by"]:
|
||||||
|
site_index = idx
|
||||||
|
break
|
||||||
|
elif (original_instance.location.x == site["position"]["x"] and original_instance.location.y == site["position"]["y"] and original_instance.location.z == site["position"]["z"]):
|
||||||
|
site_index = idx
|
||||||
|
break
|
||||||
|
if site_index is None:
|
||||||
|
site_name = None
|
||||||
|
else:
|
||||||
|
site_name = site_names[site_index]
|
||||||
if site_name != target_site:
|
if site_name != target_site:
|
||||||
parent = self.transfer_to_new_resource(original_instance, tree, additional_add_params)
|
parent = self.transfer_to_new_resource(original_instance, tree, additional_add_params)
|
||||||
if parent is not None:
|
if parent is not None:
|
||||||
@@ -1018,7 +1034,7 @@ class BaseROS2DeviceNode(Node, Generic[T]):
|
|||||||
].call_async(
|
].call_async(
|
||||||
r
|
r
|
||||||
) # type: ignore
|
) # type: ignore
|
||||||
self.lab_logger().info(f"确认资源云端 Update 结果: {response.response}")
|
self.lab_logger().trace(f"确认资源云端 Update 结果: {response.response}")
|
||||||
results.append(result)
|
results.append(result)
|
||||||
elif action == "remove":
|
elif action == "remove":
|
||||||
result = _handle_remove(resources_uuid)
|
result = _handle_remove(resources_uuid)
|
||||||
|
|||||||
@@ -1195,7 +1195,7 @@ class HostNode(BaseROS2DeviceNode):
|
|||||||
self.lab_logger().info(f"[Host Node-Resource] UUID映射: {len(uuid_mapping)} 个节点")
|
self.lab_logger().info(f"[Host Node-Resource] UUID映射: {len(uuid_mapping)} 个节点")
|
||||||
# 还需要加入到资源图中,暂不实现,考虑资源图新的获取方式
|
# 还需要加入到资源图中,暂不实现,考虑资源图新的获取方式
|
||||||
response.response = json.dumps(uuid_mapping)
|
response.response = json.dumps(uuid_mapping)
|
||||||
self.lab_logger().info(f"[Host Node-Resource] Resource tree add completed, success: {success}")
|
self.lab_logger().info(f"[Host Node-Resource] Resource tree update completed, success: {success}")
|
||||||
|
|
||||||
async def _resource_tree_update_callback(self, request: SerialCommand_Request, response: SerialCommand_Response):
|
async def _resource_tree_update_callback(self, request: SerialCommand_Request, response: SerialCommand_Response):
|
||||||
"""
|
"""
|
||||||
|
|||||||
File diff suppressed because it is too large
Load Diff
@@ -26,7 +26,7 @@
|
|||||||
res_id: plate_slot_{slot}
|
res_id: plate_slot_{slot}
|
||||||
device_id: /PRCXI
|
device_id: /PRCXI
|
||||||
class_name: PRCXI_BioER_96_wellplate
|
class_name: PRCXI_BioER_96_wellplate
|
||||||
parent: /PRCXI/PRCXI_Deck/T{slot}
|
parent: /PRCXI/PRCXI_Deck
|
||||||
slot_on_deck: "{slot}"
|
slot_on_deck: "{slot}"
|
||||||
- 输出端口: labware(用于连接 set_liquid_from_plate)
|
- 输出端口: labware(用于连接 set_liquid_from_plate)
|
||||||
- 控制流: create_resource 之间通过 ready 端口串联
|
- 控制流: create_resource 之间通过 ready 端口串联
|
||||||
@@ -122,7 +122,7 @@ NODE_TYPE_DEFAULT = "ILab" # 所有节点的默认类型
|
|||||||
# create_resource 节点默认参数
|
# create_resource 节点默认参数
|
||||||
CREATE_RESOURCE_DEFAULTS = {
|
CREATE_RESOURCE_DEFAULTS = {
|
||||||
"device_id": "/PRCXI",
|
"device_id": "/PRCXI",
|
||||||
"parent_template": "/PRCXI/PRCXI_Deck/T{slot}", # {slot} 会被替换为实际的 slot 值
|
"parent_template": "/PRCXI/PRCXI_Deck",
|
||||||
"class_name": "PRCXI_BioER_96_wellplate",
|
"class_name": "PRCXI_BioER_96_wellplate",
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -424,7 +424,7 @@ def build_protocol_graph(
|
|||||||
"res_id": res_id,
|
"res_id": res_id,
|
||||||
"device_id": CREATE_RESOURCE_DEFAULTS["device_id"],
|
"device_id": CREATE_RESOURCE_DEFAULTS["device_id"],
|
||||||
"class_name": lw_type,
|
"class_name": lw_type,
|
||||||
"parent": CREATE_RESOURCE_DEFAULTS["parent_template"].format(slot=slot),
|
"parent": CREATE_RESOURCE_DEFAULTS["parent_template"],
|
||||||
"bind_locations": {"x": 0.0, "y": 0.0, "z": 0.0},
|
"bind_locations": {"x": 0.0, "y": 0.0, "z": 0.0},
|
||||||
"slot_on_deck": slot,
|
"slot_on_deck": slot,
|
||||||
},
|
},
|
||||||
|
|||||||
Reference in New Issue
Block a user