mirror of
https://github.com/dptech-corp/Uni-Lab-OS.git
synced 2025-12-18 21:41:16 +00:00
bb3ca645a4
fix(reaction_station): 为步骤参数添加Value字段传个BY后端 fix(bioyond/warehouses): 修正仓库尺寸和物品排列参数 调整仓库的x轴和z轴物品数量以及物品尺寸参数，使其符合4x1x4的规格要求 fix warehouse serialize/deserialize fix bioyond converter fix itemized_carrier.unassign_child_resource allow not-loaded MSG in registry add layout serializer & converter warehouseuse A1-D4; add warehouse layout fix(graphio): 修正bioyond到plr资源转换中的坐标计算错误 Fix resource assignment and type mapping issues Corrects resource assignment in ItemizedCarrier by using the correct spot key from _ordering. Updates graphio to use 'typeName' instead of 'name' for type mapping in resource_bioyond_to_plr. Renames DummyWorkstation to BioyondWorkstation in workstation_http_service for clarity.
112 lines
3.7 KiB
Python
112 lines
3.7 KiB
Python
from typing import Dict, Optional, List, Union
|
||
from pylabrobot.resources import Coordinate
|
||
from pylabrobot.resources.carrier import ResourceHolder, create_homogeneous_resources
|
||
|
||
from unilabos.resources.itemized_carrier import ItemizedCarrier, ResourcePLR
|
||
|
||
|
||
LETTERS = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
||
|
||
|
||
def warehouse_factory(
|
||
name: str,
|
||
num_items_x: int = 1,
|
||
num_items_y: int = 4,
|
||
num_items_z: int = 4,
|
||
dx: float = 137.0,
|
||
dy: float = 96.0,
|
||
dz: float = 120.0,
|
||
item_dx: float = 10.0,
|
||
item_dy: float = 10.0,
|
||
item_dz: float = 10.0,
|
||
removed_positions: Optional[List[int]] = None,
|
||
empty: bool = False,
|
||
category: str = "warehouse",
|
||
model: Optional[str] = None,
|
||
):
|
||
# 创建16个板架位 (4层 x 4位置)
|
||
locations = []
|
||
for layer in range(num_items_z): # 4层
|
||
for row in range(num_items_y): # 4行
|
||
for col in range(num_items_x): # 1列 (每层4x1=4个位置)
|
||
# 计算位置
|
||
x = dx + col * item_dx
|
||
y = dy + (num_items_y - row - 1) * item_dy
|
||
z = dz + (num_items_z - layer - 1) * item_dz
|
||
locations.append(Coordinate(x, y, z))
|
||
if removed_positions:
|
||
locations = [loc for i, loc in enumerate(locations) if i not in removed_positions]
|
||
_sites = create_homogeneous_resources(
|
||
klass=ResourceHolder,
|
||
locations=locations,
|
||
resource_size_x=127.0,
|
||
resource_size_y=86.0,
|
||
name_prefix=name,
|
||
)
|
||
len_x, len_y = (num_items_x, num_items_y) if num_items_z == 1 else (num_items_y, num_items_z) if num_items_x == 1 else (num_items_x, num_items_z)
|
||
keys = [f"{LETTERS[j]}{i + 1}" for i in range(len_x) for j in range(len_y)]
|
||
sites = {i: site for i, site in zip(keys, _sites.values())}
|
||
|
||
return WareHouse(
|
||
name=name,
|
||
size_x=dx + item_dx * num_items_x,
|
||
size_y=dy + item_dy * num_items_y,
|
||
size_z=dz + item_dz * num_items_z,
|
||
num_items_x = num_items_x,
|
||
num_items_y = num_items_y,
|
||
num_items_z = num_items_z,
|
||
# ordered_items=ordered_items,
|
||
# ordering=ordering,
|
||
sites=sites,
|
||
category=category,
|
||
model=model,
|
||
)
|
||
|
||
|
||
class WareHouse(ItemizedCarrier):
|
||
"""堆栈载体类 - 可容纳16个板位的载体（4层x4行x1列）"""
|
||
def __init__(
|
||
self,
|
||
name: str,
|
||
size_x: float,
|
||
size_y: float,
|
||
size_z: float,
|
||
num_items_x: int,
|
||
num_items_y: int,
|
||
num_items_z: int,
|
||
layout: str = "x-y",
|
||
sites: Optional[Dict[Union[int, str], Optional[ResourcePLR]]] = None,
|
||
category: str = "warehouse",
|
||
model: Optional[str] = None,
|
||
):
|
||
|
||
super().__init__(
|
||
name=name,
|
||
size_x=size_x,
|
||
size_y=size_y,
|
||
size_z=size_z,
|
||
# ordered_items=ordered_items,
|
||
# ordering=ordering,
|
||
num_items_x=num_items_x,
|
||
num_items_y=num_items_y,
|
||
num_items_z=num_items_z,
|
||
layout=layout,
|
||
sites=sites,
|
||
category=category,
|
||
model=model,
|
||
)
|
||
|
||
def get_site_by_layer_position(self, row: int, col: int, layer: int) -> ResourceHolder:
|
||
if not (0 <= layer < 4 and 0 <= row < 4 and 0 <= col < 1):
|
||
raise ValueError("无效的位置: layer={}, row={}, col={}".format(layer, row, col))
|
||
|
||
site_index = layer * 4 + row * 1 + col
|
||
return self.sites[site_index]
|
||
|
||
def add_rack_to_position(self, row: int, col: int, layer: int, rack) -> None:
|
||
site = self.get_site_by_layer_position(row, col, layer)
|
||
site.assign_child_resource(rack)
|
||
|
||
def get_rack_at_position(self, row: int, col: int, layer: int):
|
||
site = self.get_site_by_layer_position(row, col, layer)
|
||
return site.resource |