mirror of
https://github.com/dptech-corp/Uni-Lab-OS.git
synced 2026-02-04 05:15:10 +00:00
Compare commits
451 Commits
main
...
feature/or
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
64c748d921 | ||
|
|
15ff0e9d30 | ||
|
|
f8a52860ad | ||
|
|
e30c01d54e | ||
|
|
37ec49f318 | ||
|
|
6bf57f18c1 | ||
|
|
400bb073d4 | ||
|
|
3f63c36505 | ||
|
|
0ae94f7f3c | ||
|
|
7eacae6442 | ||
|
|
f7d2cb4b9e | ||
|
|
bf980d7248 | ||
|
|
27c0544bfc | ||
|
|
d48e77c9ae | ||
|
|
e70a5bea66 | ||
|
|
467d75dc03 | ||
|
|
9feeb0c430 | ||
|
|
b2f26ffb28 | ||
|
|
4b0d1553e9 | ||
|
|
67ddee2ab2 | ||
|
|
1bcdad9448 | ||
|
|
039c96fe01 | ||
|
|
e1555d10a0 | ||
|
|
f2a96b2041 | ||
|
|
329349639e | ||
|
|
e4cc111523 | ||
|
|
d245ceef1b | ||
|
|
6db7fbd721 | ||
|
|
ab05b858e1 | ||
|
|
43e4c71a8e | ||
|
|
c4a3be1498 | ||
|
|
e11070315d | ||
|
|
2cf58ca452 | ||
|
|
fd73bb7dcb | ||
|
|
a02cecfd18 | ||
|
|
d6accc3f1c | ||
|
|
39dc443399 | ||
|
|
37b1fca962 | ||
|
|
216f19fb62 | ||
|
|
50ebcad9d7 | ||
|
|
ec7ca6a1fe | ||
|
|
4c8022ee95 | ||
|
|
ad21644db0 | ||
|
|
9dfd58e9af | ||
|
|
31c9f9a172 | ||
|
|
02cd8de4c5 | ||
|
|
a66603ec1c | ||
|
|
ec015e16cd | ||
|
|
965bf36e8d | ||
|
|
aacf3497e0 | ||
|
|
657f952e7a | ||
|
|
0165590290 | ||
|
|
daea1ab54d | ||
|
|
93cb307396 | ||
|
|
1c312772ae | ||
|
|
bad1db5094 | ||
|
|
f26eb69eca | ||
|
|
12c0770c92 | ||
|
|
3d2d428a96 | ||
|
|
78bf57f590 | ||
|
|
e227cddab3 | ||
|
|
f2b993643f | ||
|
|
2e14bf197c | ||
|
|
66c18c080a | ||
|
|
a1c34f138e | ||
|
|
75bb5ec553 | ||
|
|
bb95c89829 | ||
|
|
394c140830 | ||
|
|
e6d8d41183 | ||
|
|
847a300af3 | ||
|
|
a201d7c307 | ||
|
|
3433766bc5 | ||
|
|
7e9e93b29c | ||
|
|
9e1e6da505 | ||
|
|
8a0f000bab | ||
|
|
2ffeb49acb | ||
|
|
5fec753fb9 | ||
|
|
acbaff7bb7 | ||
|
|
706323dc3e | ||
|
|
b0804d939c | ||
|
|
97788b4e07 | ||
|
|
39cc280c91 | ||
|
|
d0ac452405 | ||
|
|
152d3a7563 | ||
|
|
ef14737839 | ||
|
|
5d5569121c | ||
|
|
d23e85ade4 | ||
|
|
02afafd423 | ||
|
|
6ac510dcd2 | ||
|
|
ed56c1eba2 | ||
|
|
16ee3de086 | ||
|
|
ced961050d | ||
|
|
11b2c99836 | ||
|
|
04024bc8a3 | ||
|
|
154048107d | ||
|
|
0b896870ba | ||
|
|
ee609e4aa2 | ||
|
|
5551fbf360 | ||
|
|
e13b250632 | ||
|
|
b8278c5026 | ||
|
|
53e767a054 | ||
|
|
cf7032fa81 | ||
|
|
97681ba433 | ||
|
|
3fa81ab4f6 | ||
|
|
9f4a69ddf5 | ||
|
|
05ae4e72df | ||
|
|
2870c04086 | ||
|
|
343e87df0d | ||
|
|
5d0807cba6 | ||
|
|
4875977d5f | ||
|
|
956b1c905b | ||
|
|
944911c52a | ||
|
|
a13b790926 | ||
|
|
9feadd68c6 | ||
|
|
c68d5246d0 | ||
|
|
49073f2c77 | ||
|
|
b2afc29f15 | ||
|
|
4061280f6b | ||
|
|
6a681e1d73 | ||
|
|
653e6e1ac3 | ||
|
|
2c774bcd1d | ||
|
|
2ba395b681 | ||
|
|
b6b3d59083 | ||
|
|
f40e3f521c | ||
|
|
7cc2fe036f | ||
|
|
f81d20bb1d | ||
|
|
db1b5a869f | ||
|
|
0136630700 | ||
|
|
3c31811f9e | ||
|
|
64f02ff129 | ||
|
|
7d097b8222 | ||
|
|
d266d21104 | ||
|
|
b6d0bbcb17 | ||
|
|
31ebff8e37 | ||
|
|
2132895ba2 | ||
|
|
850eeae55a | ||
|
|
d869c14233 | ||
|
|
24101b3cec | ||
|
|
3bf8aad4d5 | ||
|
|
a599eb70e5 | ||
|
|
0bf6994f95 | ||
|
|
c36f53791c | ||
|
|
eb4d2d96c5 | ||
|
|
8233c41b1d | ||
|
|
0dfd4ce8a8 | ||
|
|
7953b3820e | ||
|
|
eed233fa76 | ||
|
|
0c55147ee4 | ||
|
|
ce6267b8e0 | ||
|
|
975e51cd96 | ||
|
|
c5056b381c | ||
|
|
c35da65b15 | ||
|
|
659cf05be6 | ||
|
|
3b8deb4d1d | ||
|
|
c796615f9f | ||
|
|
a5bad6074f | ||
|
|
1d3a07a736 | ||
|
|
cc2cd57cdf | ||
|
|
39bb7dc627 | ||
|
|
0fda155f55 | ||
|
|
6e3eacd2f0 | ||
|
|
062f1a2153 | ||
|
|
61e8d67800 | ||
|
|
d0884cdbd8 | ||
|
|
545ea45024 | ||
|
|
b9ddee8f2c | ||
|
|
a0c5095304 | ||
|
|
e504505137 | ||
|
|
4d9d5701e9 | ||
|
|
6016c4b588 | ||
|
|
be02bef9c4 | ||
|
|
e62f0c2585 | ||
|
|
b6de0623e2 | ||
|
|
9d081e9fcd | ||
|
|
85a58e3464 | ||
|
|
85590672d8 | ||
|
|
1d4018196d | ||
|
|
5d34f742af | ||
|
|
5bef19e6d6 | ||
|
|
f816799753 | ||
|
|
a45d841769 | ||
|
|
7f0b33b3e3 | ||
|
|
2006406a24 | ||
|
|
f94985632b | ||
|
|
12ba110569 | ||
|
|
97212be8b7 | ||
|
|
9bdd42f12f | ||
|
|
627140da03 | ||
|
|
5ceedb0565 | ||
|
|
8c77a20c43 | ||
|
|
3ff894feee | ||
|
|
fa5896ffdb | ||
|
|
eb504803ac | ||
|
|
8b0c845661 | ||
|
|
693873bfa9 | ||
|
|
57da2d8da2 | ||
|
|
8d1fd01259 | ||
|
|
388259e64b | ||
|
|
2c130e7f37 | ||
|
|
9f7c3f02f9 | ||
|
|
19dd80dcdb | ||
|
|
9d5ed627a2 | ||
|
|
2d0ff87bc8 | ||
|
|
d78475de9a | ||
|
|
88ae56806c | ||
|
|
95dd8beb81 | ||
|
|
4ab3fadbec | ||
|
|
229888f834 | ||
|
|
b443b39ebf | ||
|
|
0434bbc15b | ||
|
|
5791b81954 | ||
|
|
bd51c74fab | ||
|
|
ba81cbddf8 | ||
|
|
4e92a26057 | ||
|
|
c2895bb197 | ||
|
|
0423f4f452 | ||
|
|
41390fbef9 | ||
|
|
98bdb4e7e4 | ||
|
|
30037a077a | ||
|
|
6972680099 | ||
|
|
9d2c93807d | ||
|
|
e728007bc5 | ||
|
|
9c5ecda7cc | ||
|
|
2d26c3fac6 | ||
|
|
f5753afb7c | ||
|
|
398b2dde3f | ||
|
|
62c4135938 | ||
|
|
027b4269c4 | ||
|
|
3757bd9c58 | ||
|
|
c75b7d5aae | ||
|
|
dfc635189c | ||
|
|
d8f3ebac15 | ||
|
|
4a1e703a3a | ||
|
|
55d22a7c29 | ||
|
|
03a4e4ecba | ||
|
|
2316c34cb5 | ||
|
|
a8887161d3 | ||
|
|
25834f5ba0 | ||
|
|
a1e9332b51 | ||
|
|
357fc038ef | ||
|
|
fd58ef07f3 | ||
|
|
93dee2c1dc | ||
|
|
70fbf19009 | ||
|
|
9149155232 | ||
|
|
1ca1792e3c | ||
|
|
485e7e8dd2 | ||
|
|
4ddabdcb65 | ||
|
|
a5b0325301 | ||
|
|
50b44938c7 | ||
|
|
df0d2235b0 | ||
|
|
4e434eeb97 | ||
|
|
ca027bf0eb | ||
|
|
635a332b4e | ||
|
|
edf7a117ca | ||
|
|
70b2715996 | ||
|
|
7e8dfc2dc5 | ||
|
|
9b626489a8 | ||
|
|
03fe208743 | ||
|
|
e913e540a3 | ||
|
|
aed39b648d | ||
|
|
8c8359fab3 | ||
|
|
5d20be0762 | ||
|
|
09f745d300 | ||
|
|
bbcbcde9a4 | ||
|
|
42b437cdea | ||
|
|
ffd0f2d26a | ||
|
|
32422c0b3d | ||
|
|
c44e597dc0 | ||
|
|
4eef012a8e | ||
|
|
ac69452f3c | ||
|
|
57b30f627b | ||
|
|
2d2a4ca067 | ||
|
|
a2613aad4c | ||
|
|
54f75183ff | ||
|
|
735be067dc | ||
|
|
0fe62d64f0 | ||
|
|
2d4ecec1e1 | ||
|
|
0f976a1874 | ||
|
|
b263a7e679 | ||
|
|
7c7f1b31c5 | ||
|
|
00e668e140 | ||
|
|
4989f65a0b | ||
|
|
9fa3688196 | ||
|
|
40fb1ea49c | ||
|
|
18b0bb397e | ||
|
|
65abc5dbf7 | ||
|
|
2455ca15ba | ||
|
|
05a3ff607a | ||
|
|
ec882df36d | ||
|
|
43b992e3eb | ||
|
|
6422fa5a9a | ||
|
|
434b9e98e0 | ||
|
|
040073f430 | ||
|
|
3d95c9896a | ||
|
|
9aa97ed01e | ||
|
|
0b8bdf5e0a | ||
|
|
299f010754 | ||
|
|
15ce0d6883 | ||
|
|
dec474e1a7 | ||
|
|
5f187899fc | ||
|
|
c8d16c7024 | ||
|
|
25d46dc9d5 | ||
|
|
88c4d1a9d1 | ||
|
|
81fd8291c5 | ||
|
|
3a11eb90d4 | ||
|
|
387866b9c9 | ||
|
|
7f40f141f6 | ||
|
|
6fc7ed1b88 | ||
|
|
93f0e08d75 | ||
|
|
4b43734b55 | ||
|
|
174b1914d4 | ||
|
|
704e13f030 | ||
|
|
0c42d60cf2 | ||
|
|
df33e1a214 | ||
|
|
1f49924966 | ||
|
|
609b6006e8 | ||
|
|
67c01271b7 | ||
|
|
a1783f489e | ||
|
|
a8f6527de9 | ||
|
|
54cfaf15f3 | ||
|
|
5610c28b67 | ||
|
|
cfc1ee6e79 | ||
|
|
1c9d2ee98a | ||
|
|
3fe8f4ca44 | ||
|
|
2476821dcc | ||
|
|
7b426ed5ae | ||
|
|
9bbae96447 | ||
|
|
10aabb7592 | ||
|
|
709eb0d91c | ||
|
|
14b7d52825 | ||
|
|
a5397ffe12 | ||
|
|
c6c2da69ba | ||
|
|
622e579063 | ||
|
|
196e0f7e2b | ||
|
|
a632fd495e | ||
|
|
a8cc02a126 | ||
|
|
ad2e1432c6 | ||
|
|
c3b9583eac | ||
|
|
5c47cd0c8a | ||
|
|
63ab1af45d | ||
|
|
a8419dc0c3 | ||
|
|
34f05f2e25 | ||
|
|
0dc2488f02 | ||
|
|
f13156e792 | ||
|
|
13fd1ac572 | ||
|
|
f8ef6e0686 | ||
|
|
94a7b8aaca | ||
|
|
301bea639e | ||
|
|
4b5a83efa4 | ||
|
|
2889e9be2c | ||
|
|
304aebbba7 | ||
|
|
091c9fa247 | ||
|
|
67ca45a240 | ||
|
|
7aab2ea493 | ||
|
|
62f3a6d696 | ||
|
|
eb70ad0e18 | ||
|
|
768f43880e | ||
|
|
762c3c737c | ||
|
|
ace98a4472 | ||
|
|
41eaa88c6f | ||
|
|
a1a55a2c0a | ||
|
|
2eaa0ca729 | ||
|
|
6f8f070f40 | ||
|
|
da4bd927e0 | ||
|
|
01f8816597 | ||
|
|
e5006285df | ||
|
|
573c724a5c | ||
|
|
09549d2839 | ||
|
|
50c7777cea | ||
|
|
4888f02c09 | ||
|
|
779c9693d9 | ||
|
|
ffa841a41a | ||
|
|
fc669f09f8 | ||
|
|
2ca0311de6 | ||
|
|
94cdcbf24e | ||
|
|
1cd07915e7 | ||
|
|
b600fc666d | ||
|
|
9e214c56c1 | ||
|
|
bdf27a7e82 | ||
|
|
2493fb9f94 | ||
|
|
c7a0ff67a9 | ||
|
|
711a7c65fa | ||
|
|
cde7956896 | ||
|
|
95b6fd0451 | ||
|
|
513e848d89 | ||
|
|
58d1cc4720 | ||
|
|
5676dd6589 | ||
|
|
1ae274a833 | ||
|
|
22b88c8441 | ||
|
|
81bcc1907d | ||
|
|
8cffd3dc21 | ||
|
|
a722636938 | ||
|
|
f68340d932 | ||
|
|
361eae2f6d | ||
|
|
c25283ae04 | ||
|
|
961752fb0d | ||
|
|
55165024dd | ||
|
|
6ddceb8393 | ||
|
|
4e52c7d2f4 | ||
|
|
0b56efc89d | ||
|
|
a27b93396a | ||
|
|
2a60a6c27e | ||
|
|
5dda94044d | ||
|
|
0cfc6f45e3 | ||
|
|
831f4549f9 | ||
|
|
f4d4eb06d3 | ||
|
|
e3b8164f6b | ||
|
|
78c04acc2e | ||
|
|
cd0428ea78 | ||
|
|
bdddbd57ba | ||
|
|
a312de08a5 | ||
|
|
68513b5745 | ||
|
|
19027350fb | ||
|
|
bbbdb06bbc | ||
|
|
cd84e26126 | ||
|
|
ce5bab3af1 | ||
|
|
82d9ef6bf7 | ||
|
|
332b33c6f4 | ||
|
|
1ec642ee3a | ||
|
|
7d8e6d029b | ||
|
|
5ec8a57a1f | ||
|
|
ae3c1100ae | ||
|
|
14bc2e6cda | ||
|
|
9f823a4198 | ||
|
|
02c79363c1 | ||
|
|
227ff1284a | ||
|
|
4b7bde6be5 | ||
|
|
8a669ac35a | ||
|
|
a1538da39e | ||
|
|
0063df4cf3 | ||
|
|
e570ba4976 | ||
|
|
e8c1f76dbb | ||
|
|
f791c1a342 | ||
|
|
ea60cbe891 | ||
|
|
eac9b8ab3d | ||
|
|
573bcf1a6c | ||
|
|
50e93cb1af | ||
|
|
fe1a029a9b | ||
|
|
662c063f50 | ||
|
|
01cbbba0b3 | ||
|
|
e6c556cf19 | ||
|
|
0605f305ed | ||
|
|
37d8108ec4 | ||
|
|
6081dac561 | ||
|
|
5b2d066127 | ||
|
|
06e66765e7 | ||
|
|
98ce360088 | ||
|
|
5cd0f72fbd | ||
|
|
343f394203 | ||
|
|
46aa7a7bd2 | ||
|
|
a66369e2c3 |
328
.cursor/rules/device-drivers.mdc
Normal file
328
.cursor/rules/device-drivers.mdc
Normal file
@@ -0,0 +1,328 @@
|
|||||||
|
---
|
||||||
|
description: 设备驱动开发规范
|
||||||
|
globs: ["unilabos/devices/**/*.py"]
|
||||||
|
---
|
||||||
|
|
||||||
|
# 设备驱动开发规范
|
||||||
|
|
||||||
|
## 目录结构
|
||||||
|
|
||||||
|
```
|
||||||
|
unilabos/devices/
|
||||||
|
├── virtual/ # 虚拟设备(用于测试)
|
||||||
|
│ ├── virtual_stirrer.py
|
||||||
|
│ └── virtual_centrifuge.py
|
||||||
|
├── liquid_handling/ # 液体处理设备
|
||||||
|
├── balance/ # 天平设备
|
||||||
|
├── hplc/ # HPLC设备
|
||||||
|
├── pump_and_valve/ # 泵和阀门
|
||||||
|
├── temperature/ # 温度控制设备
|
||||||
|
├── workstation/ # 工作站(组合设备)
|
||||||
|
└── ...
|
||||||
|
```
|
||||||
|
|
||||||
|
## 设备类完整模板
|
||||||
|
|
||||||
|
```python
|
||||||
|
import asyncio
|
||||||
|
import logging
|
||||||
|
import time as time_module
|
||||||
|
from typing import Dict, Any, Optional
|
||||||
|
|
||||||
|
from unilabos.ros.nodes.base_device_node import BaseROS2DeviceNode
|
||||||
|
|
||||||
|
|
||||||
|
class MyDevice:
|
||||||
|
"""
|
||||||
|
设备类描述
|
||||||
|
|
||||||
|
Attributes:
|
||||||
|
device_id: 设备唯一标识
|
||||||
|
config: 设备配置字典
|
||||||
|
data: 设备状态数据
|
||||||
|
"""
|
||||||
|
|
||||||
|
_ros_node: BaseROS2DeviceNode
|
||||||
|
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
device_id: str = None,
|
||||||
|
config: Dict[str, Any] = None,
|
||||||
|
**kwargs
|
||||||
|
):
|
||||||
|
"""
|
||||||
|
初始化设备
|
||||||
|
|
||||||
|
Args:
|
||||||
|
device_id: 设备ID
|
||||||
|
config: 配置字典
|
||||||
|
**kwargs: 其他参数
|
||||||
|
"""
|
||||||
|
# 兼容不同调用方式
|
||||||
|
if device_id is None and 'id' in kwargs:
|
||||||
|
device_id = kwargs.pop('id')
|
||||||
|
if config is None and 'config' in kwargs:
|
||||||
|
config = kwargs.pop('config')
|
||||||
|
|
||||||
|
self.device_id = device_id or "unknown_device"
|
||||||
|
self.config = config or {}
|
||||||
|
self.data = {}
|
||||||
|
|
||||||
|
# 从config读取参数
|
||||||
|
self.port = self.config.get('port') or kwargs.get('port', 'COM1')
|
||||||
|
self._max_value = self.config.get('max_value', 1000.0)
|
||||||
|
|
||||||
|
# 初始化日志
|
||||||
|
self.logger = logging.getLogger(f"MyDevice.{self.device_id}")
|
||||||
|
|
||||||
|
self.logger.info(f"设备 {self.device_id} 已创建")
|
||||||
|
|
||||||
|
def post_init(self, ros_node: BaseROS2DeviceNode):
|
||||||
|
"""
|
||||||
|
ROS节点注入 - 在ROS节点创建后调用
|
||||||
|
|
||||||
|
Args:
|
||||||
|
ros_node: ROS2设备节点实例
|
||||||
|
"""
|
||||||
|
self._ros_node = ros_node
|
||||||
|
|
||||||
|
async def initialize(self) -> bool:
|
||||||
|
"""
|
||||||
|
初始化设备 - 连接硬件、设置初始状态
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
bool: 初始化是否成功
|
||||||
|
"""
|
||||||
|
self.logger.info(f"初始化设备 {self.device_id}")
|
||||||
|
|
||||||
|
try:
|
||||||
|
# 执行硬件初始化
|
||||||
|
# await self._connect_hardware()
|
||||||
|
|
||||||
|
# 设置初始状态
|
||||||
|
self.data.update({
|
||||||
|
"status": "待机",
|
||||||
|
"is_running": False,
|
||||||
|
"current_value": 0.0,
|
||||||
|
})
|
||||||
|
|
||||||
|
self.logger.info(f"设备 {self.device_id} 初始化完成")
|
||||||
|
return True
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
self.logger.error(f"初始化失败: {e}")
|
||||||
|
self.data["status"] = f"错误: {e}"
|
||||||
|
return False
|
||||||
|
|
||||||
|
async def cleanup(self) -> bool:
|
||||||
|
"""
|
||||||
|
清理设备 - 断开连接、释放资源
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
bool: 清理是否成功
|
||||||
|
"""
|
||||||
|
self.logger.info(f"清理设备 {self.device_id}")
|
||||||
|
|
||||||
|
self.data.update({
|
||||||
|
"status": "离线",
|
||||||
|
"is_running": False,
|
||||||
|
})
|
||||||
|
|
||||||
|
return True
|
||||||
|
|
||||||
|
# ==================== 设备动作 ====================
|
||||||
|
|
||||||
|
async def execute_action(
|
||||||
|
self,
|
||||||
|
param1: float,
|
||||||
|
param2: str = "",
|
||||||
|
**kwargs
|
||||||
|
) -> bool:
|
||||||
|
"""
|
||||||
|
执行设备动作
|
||||||
|
|
||||||
|
Args:
|
||||||
|
param1: 参数1
|
||||||
|
param2: 参数2(可选)
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
bool: 动作是否成功
|
||||||
|
"""
|
||||||
|
# 类型转换和验证
|
||||||
|
try:
|
||||||
|
param1 = float(param1)
|
||||||
|
except (ValueError, TypeError) as e:
|
||||||
|
self.logger.error(f"参数类型错误: {e}")
|
||||||
|
return False
|
||||||
|
|
||||||
|
# 参数验证
|
||||||
|
if param1 > self._max_value:
|
||||||
|
self.logger.error(f"参数超出范围: {param1} > {self._max_value}")
|
||||||
|
return False
|
||||||
|
|
||||||
|
self.logger.info(f"执行动作: param1={param1}, param2={param2}")
|
||||||
|
|
||||||
|
# 更新状态
|
||||||
|
self.data.update({
|
||||||
|
"status": "运行中",
|
||||||
|
"is_running": True,
|
||||||
|
})
|
||||||
|
|
||||||
|
# 执行动作(带进度反馈)
|
||||||
|
duration = 10.0 # 秒
|
||||||
|
start_time = time_module.time()
|
||||||
|
|
||||||
|
while True:
|
||||||
|
elapsed = time_module.time() - start_time
|
||||||
|
remaining = max(0, duration - elapsed)
|
||||||
|
progress = min(100, (elapsed / duration) * 100)
|
||||||
|
|
||||||
|
self.data.update({
|
||||||
|
"status": f"运行中: {progress:.0f}%",
|
||||||
|
"remaining_time": remaining,
|
||||||
|
})
|
||||||
|
|
||||||
|
if remaining <= 0:
|
||||||
|
break
|
||||||
|
|
||||||
|
await self._ros_node.sleep(1.0)
|
||||||
|
|
||||||
|
# 完成
|
||||||
|
self.data.update({
|
||||||
|
"status": "完成",
|
||||||
|
"is_running": False,
|
||||||
|
})
|
||||||
|
|
||||||
|
self.logger.info("动作执行完成")
|
||||||
|
return True
|
||||||
|
|
||||||
|
# ==================== 状态属性 ====================
|
||||||
|
|
||||||
|
@property
|
||||||
|
def status(self) -> str:
|
||||||
|
"""设备状态 - 自动发布为ROS Topic"""
|
||||||
|
return self.data.get("status", "未知")
|
||||||
|
|
||||||
|
@property
|
||||||
|
def is_running(self) -> bool:
|
||||||
|
"""是否正在运行"""
|
||||||
|
return self.data.get("is_running", False)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def current_value(self) -> float:
|
||||||
|
"""当前值"""
|
||||||
|
return self.data.get("current_value", 0.0)
|
||||||
|
|
||||||
|
# ==================== 辅助方法 ====================
|
||||||
|
|
||||||
|
def get_device_info(self) -> Dict[str, Any]:
|
||||||
|
"""获取设备信息"""
|
||||||
|
return {
|
||||||
|
"device_id": self.device_id,
|
||||||
|
"status": self.status,
|
||||||
|
"is_running": self.is_running,
|
||||||
|
"current_value": self.current_value,
|
||||||
|
}
|
||||||
|
|
||||||
|
def __str__(self) -> str:
|
||||||
|
return f"MyDevice({self.device_id}: {self.status})"
|
||||||
|
```
|
||||||
|
|
||||||
|
## 关键规则
|
||||||
|
|
||||||
|
### 1. 参数处理
|
||||||
|
|
||||||
|
所有动作方法的参数都可能以字符串形式传入,必须进行类型转换:
|
||||||
|
|
||||||
|
```python
|
||||||
|
async def my_action(self, value: float, **kwargs) -> bool:
|
||||||
|
# 始终进行类型转换
|
||||||
|
try:
|
||||||
|
value = float(value)
|
||||||
|
except (ValueError, TypeError) as e:
|
||||||
|
self.logger.error(f"参数类型错误: {e}")
|
||||||
|
return False
|
||||||
|
```
|
||||||
|
|
||||||
|
### 2. vessel 参数处理
|
||||||
|
|
||||||
|
vessel 参数可能是字符串ID或字典:
|
||||||
|
|
||||||
|
```python
|
||||||
|
def extract_vessel_id(vessel: Union[str, dict]) -> str:
|
||||||
|
if isinstance(vessel, dict):
|
||||||
|
return vessel.get("id", "")
|
||||||
|
return str(vessel) if vessel else ""
|
||||||
|
```
|
||||||
|
|
||||||
|
### 3. 状态更新
|
||||||
|
|
||||||
|
使用 `self.data` 字典存储状态,属性读取状态:
|
||||||
|
|
||||||
|
```python
|
||||||
|
# 更新状态
|
||||||
|
self.data["status"] = "运行中"
|
||||||
|
self.data["current_speed"] = 300.0
|
||||||
|
|
||||||
|
# 读取状态(通过属性)
|
||||||
|
@property
|
||||||
|
def status(self) -> str:
|
||||||
|
return self.data.get("status", "待机")
|
||||||
|
```
|
||||||
|
|
||||||
|
### 4. 异步等待
|
||||||
|
|
||||||
|
使用 ROS 节点的 sleep 方法:
|
||||||
|
|
||||||
|
```python
|
||||||
|
# 正确
|
||||||
|
await self._ros_node.sleep(1.0)
|
||||||
|
|
||||||
|
# 避免(除非在纯 Python 测试环境)
|
||||||
|
await asyncio.sleep(1.0)
|
||||||
|
```
|
||||||
|
|
||||||
|
### 5. 进度反馈
|
||||||
|
|
||||||
|
长时间运行的操作需要提供进度反馈:
|
||||||
|
|
||||||
|
```python
|
||||||
|
while remaining > 0:
|
||||||
|
progress = (elapsed / total_time) * 100
|
||||||
|
self.data["status"] = f"运行中: {progress:.0f}%"
|
||||||
|
self.data["remaining_time"] = remaining
|
||||||
|
|
||||||
|
await self._ros_node.sleep(1.0)
|
||||||
|
```
|
||||||
|
|
||||||
|
## 虚拟设备
|
||||||
|
|
||||||
|
虚拟设备用于测试和演示,放在 `unilabos/devices/virtual/` 目录:
|
||||||
|
|
||||||
|
- 类名以 `Virtual` 开头
|
||||||
|
- 文件名以 `virtual_` 开头
|
||||||
|
- 模拟真实设备的行为和时序
|
||||||
|
- 使用表情符号增强日志可读性(可选)
|
||||||
|
|
||||||
|
## 工作站设备
|
||||||
|
|
||||||
|
工作站是组合多个设备的复杂设备:
|
||||||
|
|
||||||
|
```python
|
||||||
|
from unilabos.devices.workstation.workstation_base import WorkstationBase
|
||||||
|
|
||||||
|
class MyWorkstation(WorkstationBase):
|
||||||
|
"""组合工作站"""
|
||||||
|
|
||||||
|
async def execute_workflow(self, workflow: Dict[str, Any]) -> bool:
|
||||||
|
"""执行工作流"""
|
||||||
|
pass
|
||||||
|
```
|
||||||
|
|
||||||
|
## 设备注册
|
||||||
|
|
||||||
|
设备类开发完成后,需要在注册表中注册:
|
||||||
|
|
||||||
|
1. 创建/编辑 `unilabos/registry/devices/my_category.yaml`
|
||||||
|
2. 添加设备配置(参考 `virtual_device.yaml`)
|
||||||
|
3. 运行 `--complete_registry` 自动生成 schema
|
||||||
240
.cursor/rules/protocol-development.mdc
Normal file
240
.cursor/rules/protocol-development.mdc
Normal file
@@ -0,0 +1,240 @@
|
|||||||
|
---
|
||||||
|
description: 协议编译器开发规范
|
||||||
|
globs: ["unilabos/compile/**/*.py"]
|
||||||
|
---
|
||||||
|
|
||||||
|
# 协议编译器开发规范
|
||||||
|
|
||||||
|
## 概述
|
||||||
|
|
||||||
|
协议编译器负责将高级实验操作(如 Stir、Add、Filter)编译为设备可执行的动作序列。
|
||||||
|
|
||||||
|
## 文件命名
|
||||||
|
|
||||||
|
- 位置: `unilabos/compile/`
|
||||||
|
- 命名: `{operation}_protocol.py`
|
||||||
|
- 示例: `stir_protocol.py`, `add_protocol.py`, `filter_protocol.py`
|
||||||
|
|
||||||
|
## 协议函数模板
|
||||||
|
|
||||||
|
```python
|
||||||
|
from typing import List, Dict, Any, Union
|
||||||
|
import networkx as nx
|
||||||
|
import logging
|
||||||
|
|
||||||
|
from .utils.unit_parser import parse_time_input
|
||||||
|
from .utils.vessel_parser import extract_vessel_id
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
def generate_{operation}_protocol(
|
||||||
|
G: nx.DiGraph,
|
||||||
|
vessel: Union[str, dict],
|
||||||
|
param1: Union[str, float] = "0",
|
||||||
|
param2: float = 0.0,
|
||||||
|
**kwargs
|
||||||
|
) -> List[Dict[str, Any]]:
|
||||||
|
"""
|
||||||
|
生成{操作}协议序列
|
||||||
|
|
||||||
|
Args:
|
||||||
|
G: 物理拓扑图 (NetworkX DiGraph)
|
||||||
|
vessel: 容器ID或Resource字典
|
||||||
|
param1: 参数1(支持字符串单位,如 "5 min")
|
||||||
|
param2: 参数2
|
||||||
|
**kwargs: 其他参数
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
List[Dict]: 动作序列
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
ValueError: 参数无效时
|
||||||
|
"""
|
||||||
|
# 1. 提取 vessel_id
|
||||||
|
vessel_id = extract_vessel_id(vessel)
|
||||||
|
|
||||||
|
# 2. 验证参数
|
||||||
|
if not vessel_id:
|
||||||
|
raise ValueError("vessel 参数不能为空")
|
||||||
|
|
||||||
|
if vessel_id not in G.nodes():
|
||||||
|
raise ValueError(f"容器 '{vessel_id}' 不存在于系统中")
|
||||||
|
|
||||||
|
# 3. 解析参数(支持单位)
|
||||||
|
parsed_param1 = parse_time_input(param1) # "5 min" -> 300.0
|
||||||
|
|
||||||
|
# 4. 查找设备
|
||||||
|
device_id = find_connected_device(G, vessel_id, device_type="my_device")
|
||||||
|
|
||||||
|
# 5. 生成动作序列
|
||||||
|
action_sequence = []
|
||||||
|
|
||||||
|
action = {
|
||||||
|
"device_id": device_id,
|
||||||
|
"action_name": "my_action",
|
||||||
|
"action_kwargs": {
|
||||||
|
"vessel": {"id": vessel_id}, # 始终使用字典格式
|
||||||
|
"param1": float(parsed_param1),
|
||||||
|
"param2": float(param2),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
action_sequence.append(action)
|
||||||
|
|
||||||
|
logger.info(f"生成协议: {len(action_sequence)} 个动作")
|
||||||
|
return action_sequence
|
||||||
|
|
||||||
|
|
||||||
|
def find_connected_device(
|
||||||
|
G: nx.DiGraph,
|
||||||
|
vessel_id: str,
|
||||||
|
device_type: str = ""
|
||||||
|
) -> str:
|
||||||
|
"""
|
||||||
|
查找与容器相连的设备
|
||||||
|
|
||||||
|
Args:
|
||||||
|
G: 拓扑图
|
||||||
|
vessel_id: 容器ID
|
||||||
|
device_type: 设备类型关键字
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
str: 设备ID
|
||||||
|
"""
|
||||||
|
# 查找所有匹配类型的设备
|
||||||
|
device_nodes = []
|
||||||
|
for node in G.nodes():
|
||||||
|
node_class = G.nodes[node].get('class', '') or ''
|
||||||
|
if device_type.lower() in node_class.lower():
|
||||||
|
device_nodes.append(node)
|
||||||
|
|
||||||
|
# 检查连接
|
||||||
|
if vessel_id and device_nodes:
|
||||||
|
for device in device_nodes:
|
||||||
|
if G.has_edge(device, vessel_id) or G.has_edge(vessel_id, device):
|
||||||
|
return device
|
||||||
|
|
||||||
|
# 返回第一个可用设备
|
||||||
|
if device_nodes:
|
||||||
|
return device_nodes[0]
|
||||||
|
|
||||||
|
# 默认设备
|
||||||
|
return f"{device_type}_1"
|
||||||
|
```
|
||||||
|
|
||||||
|
## 关键规则
|
||||||
|
|
||||||
|
### 1. vessel 参数处理
|
||||||
|
|
||||||
|
vessel 参数可能是字符串或字典,需要统一处理:
|
||||||
|
|
||||||
|
```python
|
||||||
|
def extract_vessel_id(vessel: Union[str, dict]) -> str:
|
||||||
|
"""提取vessel_id"""
|
||||||
|
if isinstance(vessel, dict):
|
||||||
|
# 可能是 {"id": "xxx"} 或完整 Resource 对象
|
||||||
|
return vessel.get("id", list(vessel.values())[0].get("id", ""))
|
||||||
|
return str(vessel) if vessel else ""
|
||||||
|
```
|
||||||
|
|
||||||
|
### 2. action_kwargs 中的 vessel
|
||||||
|
|
||||||
|
始终使用 `{"id": vessel_id}` 格式传递 vessel:
|
||||||
|
|
||||||
|
```python
|
||||||
|
# 正确
|
||||||
|
"action_kwargs": {
|
||||||
|
"vessel": {"id": vessel_id}, # 字符串ID包装为字典
|
||||||
|
}
|
||||||
|
|
||||||
|
# 避免
|
||||||
|
"action_kwargs": {
|
||||||
|
"vessel": vessel_resource, # 不要传递完整 Resource 对象
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### 3. 单位解析
|
||||||
|
|
||||||
|
使用 `parse_time_input` 解析时间参数:
|
||||||
|
|
||||||
|
```python
|
||||||
|
from .utils.unit_parser import parse_time_input
|
||||||
|
|
||||||
|
# 支持格式: "5 min", "1 h", "300", "1.5 hours"
|
||||||
|
time_seconds = parse_time_input("5 min") # -> 300.0
|
||||||
|
time_seconds = parse_time_input(120) # -> 120.0
|
||||||
|
time_seconds = parse_time_input("1 h") # -> 3600.0
|
||||||
|
```
|
||||||
|
|
||||||
|
### 4. 参数验证
|
||||||
|
|
||||||
|
所有参数必须进行验证和类型转换:
|
||||||
|
|
||||||
|
```python
|
||||||
|
# 验证范围
|
||||||
|
if speed < 10.0 or speed > 1500.0:
|
||||||
|
logger.warning(f"速度 {speed} 超出范围,修正为 300")
|
||||||
|
speed = 300.0
|
||||||
|
|
||||||
|
# 类型转换
|
||||||
|
param = float(param) if not isinstance(param, (int, float)) else param
|
||||||
|
```
|
||||||
|
|
||||||
|
### 5. 日志记录
|
||||||
|
|
||||||
|
使用项目日志记录器:
|
||||||
|
|
||||||
|
```python
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
def generate_protocol(...):
|
||||||
|
logger.info(f"开始生成协议...")
|
||||||
|
logger.debug(f"参数: vessel={vessel_id}, time={time}")
|
||||||
|
logger.warning(f"参数修正: {old_value} -> {new_value}")
|
||||||
|
```
|
||||||
|
|
||||||
|
## 便捷函数
|
||||||
|
|
||||||
|
为常用操作提供便捷函数:
|
||||||
|
|
||||||
|
```python
|
||||||
|
def stir_briefly(G: nx.DiGraph, vessel: Union[str, dict],
|
||||||
|
speed: float = 300.0) -> List[Dict[str, Any]]:
|
||||||
|
"""短时间搅拌(30秒)"""
|
||||||
|
return generate_stir_protocol(G, vessel, time="30", stir_speed=speed)
|
||||||
|
|
||||||
|
def stir_vigorously(G: nx.DiGraph, vessel: Union[str, dict],
|
||||||
|
time: str = "5 min") -> List[Dict[str, Any]]:
|
||||||
|
"""剧烈搅拌"""
|
||||||
|
return generate_stir_protocol(G, vessel, time=time, stir_speed=800.0)
|
||||||
|
```
|
||||||
|
|
||||||
|
## 测试函数
|
||||||
|
|
||||||
|
每个协议文件应包含测试函数:
|
||||||
|
|
||||||
|
```python
|
||||||
|
def test_{operation}_protocol():
|
||||||
|
"""测试协议生成"""
|
||||||
|
# 测试参数处理
|
||||||
|
vessel_dict = {"id": "flask_1", "name": "反应瓶1"}
|
||||||
|
vessel_id = extract_vessel_id(vessel_dict)
|
||||||
|
assert vessel_id == "flask_1"
|
||||||
|
|
||||||
|
# 测试单位解析
|
||||||
|
time_s = parse_time_input("5 min")
|
||||||
|
assert time_s == 300.0
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
test_{operation}_protocol()
|
||||||
|
```
|
||||||
|
|
||||||
|
## 现有协议参考
|
||||||
|
|
||||||
|
- `stir_protocol.py` - 搅拌操作
|
||||||
|
- `add_protocol.py` - 添加物料
|
||||||
|
- `filter_protocol.py` - 过滤操作
|
||||||
|
- `heatchill_protocol.py` - 加热/冷却
|
||||||
|
- `separate_protocol.py` - 分离操作
|
||||||
|
- `evaporate_protocol.py` - 蒸发操作
|
||||||
319
.cursor/rules/registry-config.mdc
Normal file
319
.cursor/rules/registry-config.mdc
Normal file
@@ -0,0 +1,319 @@
|
|||||||
|
---
|
||||||
|
description: 注册表配置规范 (YAML)
|
||||||
|
globs: ["unilabos/registry/**/*.yaml"]
|
||||||
|
---
|
||||||
|
|
||||||
|
# 注册表配置规范
|
||||||
|
|
||||||
|
## 概述
|
||||||
|
|
||||||
|
注册表使用 YAML 格式定义设备和资源类型,是 Uni-Lab-OS 的核心配置系统。
|
||||||
|
|
||||||
|
## 目录结构
|
||||||
|
|
||||||
|
```
|
||||||
|
unilabos/registry/
|
||||||
|
├── devices/ # 设备类型注册
|
||||||
|
│ ├── virtual_device.yaml
|
||||||
|
│ ├── liquid_handler.yaml
|
||||||
|
│ └── ...
|
||||||
|
├── device_comms/ # 通信设备配置
|
||||||
|
│ ├── communication_devices.yaml
|
||||||
|
│ └── modbus_ioboard.yaml
|
||||||
|
└── resources/ # 资源类型注册
|
||||||
|
├── bioyond/
|
||||||
|
├── organic/
|
||||||
|
├── opentrons/
|
||||||
|
└── ...
|
||||||
|
```
|
||||||
|
|
||||||
|
## 设备注册表格式
|
||||||
|
|
||||||
|
### 基本结构
|
||||||
|
|
||||||
|
```yaml
|
||||||
|
device_type_id:
|
||||||
|
# 基本信息
|
||||||
|
description: "设备描述"
|
||||||
|
version: "1.0.0"
|
||||||
|
category:
|
||||||
|
- category_name
|
||||||
|
icon: "icon_device.webp"
|
||||||
|
|
||||||
|
# 类配置
|
||||||
|
class:
|
||||||
|
module: "unilabos.devices.my_module:MyClass"
|
||||||
|
type: python
|
||||||
|
|
||||||
|
# 状态类型(属性 -> ROS消息类型)
|
||||||
|
status_types:
|
||||||
|
status: String
|
||||||
|
temperature: Float64
|
||||||
|
is_running: Bool
|
||||||
|
|
||||||
|
# 动作映射
|
||||||
|
action_value_mappings:
|
||||||
|
action_name:
|
||||||
|
type: UniLabJsonCommand # 或 UniLabJsonCommandAsync
|
||||||
|
goal: {}
|
||||||
|
feedback: {}
|
||||||
|
result: {}
|
||||||
|
schema: {...}
|
||||||
|
handles: {}
|
||||||
|
```
|
||||||
|
|
||||||
|
### action_value_mappings 详细格式
|
||||||
|
|
||||||
|
```yaml
|
||||||
|
action_value_mappings:
|
||||||
|
# 同步动作
|
||||||
|
my_sync_action:
|
||||||
|
type: UniLabJsonCommand
|
||||||
|
goal:
|
||||||
|
param1: param1
|
||||||
|
param2: param2
|
||||||
|
feedback: {}
|
||||||
|
result:
|
||||||
|
success: success
|
||||||
|
message: message
|
||||||
|
goal_default:
|
||||||
|
param1: 0.0
|
||||||
|
param2: ""
|
||||||
|
handles: {}
|
||||||
|
placeholder_keys:
|
||||||
|
device_param: unilabos_devices # 设备选择器
|
||||||
|
resource_param: unilabos_resources # 资源选择器
|
||||||
|
schema:
|
||||||
|
title: "动作名称参数"
|
||||||
|
description: "动作描述"
|
||||||
|
type: object
|
||||||
|
properties:
|
||||||
|
goal:
|
||||||
|
type: object
|
||||||
|
properties:
|
||||||
|
param1:
|
||||||
|
type: number
|
||||||
|
param2:
|
||||||
|
type: string
|
||||||
|
required:
|
||||||
|
- param1
|
||||||
|
feedback: {}
|
||||||
|
result:
|
||||||
|
type: object
|
||||||
|
properties:
|
||||||
|
success:
|
||||||
|
type: boolean
|
||||||
|
message:
|
||||||
|
type: string
|
||||||
|
required:
|
||||||
|
- goal
|
||||||
|
|
||||||
|
# 异步动作
|
||||||
|
my_async_action:
|
||||||
|
type: UniLabJsonCommandAsync
|
||||||
|
goal: {}
|
||||||
|
feedback:
|
||||||
|
progress: progress
|
||||||
|
current_status: status
|
||||||
|
result:
|
||||||
|
success: success
|
||||||
|
schema: {...}
|
||||||
|
```
|
||||||
|
|
||||||
|
### 自动生成的动作
|
||||||
|
|
||||||
|
以 `auto-` 开头的动作由系统自动生成:
|
||||||
|
|
||||||
|
```yaml
|
||||||
|
action_value_mappings:
|
||||||
|
auto-initialize:
|
||||||
|
type: UniLabJsonCommandAsync
|
||||||
|
goal: {}
|
||||||
|
feedback: {}
|
||||||
|
result: {}
|
||||||
|
schema: {...}
|
||||||
|
|
||||||
|
auto-cleanup:
|
||||||
|
type: UniLabJsonCommandAsync
|
||||||
|
goal: {}
|
||||||
|
feedback: {}
|
||||||
|
result: {}
|
||||||
|
schema: {...}
|
||||||
|
```
|
||||||
|
|
||||||
|
### handles 配置
|
||||||
|
|
||||||
|
用于工作流编辑器中的数据流连接:
|
||||||
|
|
||||||
|
```yaml
|
||||||
|
handles:
|
||||||
|
input:
|
||||||
|
- handler_key: "input_resource"
|
||||||
|
data_type: "resource"
|
||||||
|
label: "输入资源"
|
||||||
|
data_source: "handle"
|
||||||
|
data_key: "resources"
|
||||||
|
output:
|
||||||
|
- handler_key: "output_labware"
|
||||||
|
data_type: "resource"
|
||||||
|
label: "输出器皿"
|
||||||
|
data_source: "executor"
|
||||||
|
data_key: "created_resource.@flatten"
|
||||||
|
```
|
||||||
|
|
||||||
|
## 资源注册表格式
|
||||||
|
|
||||||
|
```yaml
|
||||||
|
resource_type_id:
|
||||||
|
description: "资源描述"
|
||||||
|
version: "1.0.0"
|
||||||
|
category:
|
||||||
|
- category_name
|
||||||
|
icon: ""
|
||||||
|
handles: []
|
||||||
|
init_param_schema: {}
|
||||||
|
|
||||||
|
class:
|
||||||
|
module: "unilabos.resources.my_module:MyResource"
|
||||||
|
type: pylabrobot # 或 python
|
||||||
|
```
|
||||||
|
|
||||||
|
### PyLabRobot 资源示例
|
||||||
|
|
||||||
|
```yaml
|
||||||
|
BIOYOND_Electrolyte_6VialCarrier:
|
||||||
|
category:
|
||||||
|
- bottle_carriers
|
||||||
|
- bioyond
|
||||||
|
class:
|
||||||
|
module: "unilabos.resources.bioyond.bottle_carriers:BIOYOND_Electrolyte_6VialCarrier"
|
||||||
|
type: pylabrobot
|
||||||
|
version: "1.0.0"
|
||||||
|
```
|
||||||
|
|
||||||
|
## 状态类型映射
|
||||||
|
|
||||||
|
Python 类型到 ROS 消息类型的映射:
|
||||||
|
|
||||||
|
| Python 类型 | ROS 消息类型 |
|
||||||
|
|------------|-------------|
|
||||||
|
| `str` | `String` |
|
||||||
|
| `bool` | `Bool` |
|
||||||
|
| `int` | `Int64` |
|
||||||
|
| `float` | `Float64` |
|
||||||
|
| `list` | `String` (序列化) |
|
||||||
|
| `dict` | `String` (序列化) |
|
||||||
|
|
||||||
|
## 自动完善注册表
|
||||||
|
|
||||||
|
使用 `--complete_registry` 参数自动生成 schema:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
python -m unilabos.app.main --complete_registry
|
||||||
|
```
|
||||||
|
|
||||||
|
这会:
|
||||||
|
1. 扫描设备类的方法签名
|
||||||
|
2. 自动生成 `auto-` 前缀的动作
|
||||||
|
3. 生成 JSON Schema
|
||||||
|
4. 更新 YAML 文件
|
||||||
|
|
||||||
|
## 验证规则
|
||||||
|
|
||||||
|
1. **device_type_id** 必须唯一
|
||||||
|
2. **module** 路径必须正确可导入
|
||||||
|
3. **status_types** 的类型必须是有效的 ROS 消息类型
|
||||||
|
4. **schema** 必须是有效的 JSON Schema
|
||||||
|
|
||||||
|
## 示例:完整设备配置
|
||||||
|
|
||||||
|
```yaml
|
||||||
|
virtual_stirrer:
|
||||||
|
category:
|
||||||
|
- virtual_device
|
||||||
|
description: "虚拟搅拌器设备"
|
||||||
|
version: "1.0.0"
|
||||||
|
icon: "icon_stirrer.webp"
|
||||||
|
handles: []
|
||||||
|
init_param_schema: {}
|
||||||
|
|
||||||
|
class:
|
||||||
|
module: "unilabos.devices.virtual.virtual_stirrer:VirtualStirrer"
|
||||||
|
type: python
|
||||||
|
|
||||||
|
status_types:
|
||||||
|
status: String
|
||||||
|
operation_mode: String
|
||||||
|
current_speed: Float64
|
||||||
|
is_stirring: Bool
|
||||||
|
remaining_time: Float64
|
||||||
|
|
||||||
|
action_value_mappings:
|
||||||
|
auto-initialize:
|
||||||
|
type: UniLabJsonCommandAsync
|
||||||
|
goal: {}
|
||||||
|
feedback: {}
|
||||||
|
result: {}
|
||||||
|
schema:
|
||||||
|
title: "initialize参数"
|
||||||
|
type: object
|
||||||
|
properties:
|
||||||
|
goal:
|
||||||
|
type: object
|
||||||
|
properties: {}
|
||||||
|
feedback: {}
|
||||||
|
result: {}
|
||||||
|
required:
|
||||||
|
- goal
|
||||||
|
|
||||||
|
stir:
|
||||||
|
type: UniLabJsonCommandAsync
|
||||||
|
goal:
|
||||||
|
stir_time: stir_time
|
||||||
|
stir_speed: stir_speed
|
||||||
|
settling_time: settling_time
|
||||||
|
feedback:
|
||||||
|
current_speed: current_speed
|
||||||
|
remaining_time: remaining_time
|
||||||
|
result:
|
||||||
|
success: success
|
||||||
|
goal_default:
|
||||||
|
stir_time: 60.0
|
||||||
|
stir_speed: 300.0
|
||||||
|
settling_time: 30.0
|
||||||
|
handles: {}
|
||||||
|
schema:
|
||||||
|
title: "stir参数"
|
||||||
|
description: "搅拌操作"
|
||||||
|
type: object
|
||||||
|
properties:
|
||||||
|
goal:
|
||||||
|
type: object
|
||||||
|
properties:
|
||||||
|
stir_time:
|
||||||
|
type: number
|
||||||
|
description: "搅拌时间(秒)"
|
||||||
|
stir_speed:
|
||||||
|
type: number
|
||||||
|
description: "搅拌速度(RPM)"
|
||||||
|
settling_time:
|
||||||
|
type: number
|
||||||
|
description: "沉降时间(秒)"
|
||||||
|
required:
|
||||||
|
- stir_time
|
||||||
|
- stir_speed
|
||||||
|
feedback:
|
||||||
|
type: object
|
||||||
|
properties:
|
||||||
|
current_speed:
|
||||||
|
type: number
|
||||||
|
remaining_time:
|
||||||
|
type: number
|
||||||
|
result:
|
||||||
|
type: object
|
||||||
|
properties:
|
||||||
|
success:
|
||||||
|
type: boolean
|
||||||
|
required:
|
||||||
|
- goal
|
||||||
|
```
|
||||||
233
.cursor/rules/ros-integration.mdc
Normal file
233
.cursor/rules/ros-integration.mdc
Normal file
@@ -0,0 +1,233 @@
|
|||||||
|
---
|
||||||
|
description: ROS 2 集成开发规范
|
||||||
|
globs: ["unilabos/ros/**/*.py", "**/*_node.py"]
|
||||||
|
---
|
||||||
|
|
||||||
|
# ROS 2 集成开发规范
|
||||||
|
|
||||||
|
## 概述
|
||||||
|
|
||||||
|
Uni-Lab-OS 使用 ROS 2 作为设备通信中间件,基于 rclpy 实现。
|
||||||
|
|
||||||
|
## 核心组件
|
||||||
|
|
||||||
|
### BaseROS2DeviceNode
|
||||||
|
|
||||||
|
设备节点基类,提供:
|
||||||
|
- ROS Topic 自动发布(状态属性)
|
||||||
|
- Action Server 自动创建(设备动作)
|
||||||
|
- 资源管理服务
|
||||||
|
- 异步任务调度
|
||||||
|
|
||||||
|
```python
|
||||||
|
from unilabos.ros.nodes.base_device_node import BaseROS2DeviceNode
|
||||||
|
```
|
||||||
|
|
||||||
|
### 消息转换器
|
||||||
|
|
||||||
|
```python
|
||||||
|
from unilabos.ros.msgs.message_converter import (
|
||||||
|
convert_to_ros_msg,
|
||||||
|
convert_from_ros_msg_with_mapping,
|
||||||
|
msg_converter_manager,
|
||||||
|
ros_action_to_json_schema,
|
||||||
|
ros_message_to_json_schema,
|
||||||
|
)
|
||||||
|
```
|
||||||
|
|
||||||
|
## 设备与 ROS 集成
|
||||||
|
|
||||||
|
### post_init 方法
|
||||||
|
|
||||||
|
设备类必须实现 `post_init` 方法接收 ROS 节点:
|
||||||
|
|
||||||
|
```python
|
||||||
|
class MyDevice:
|
||||||
|
_ros_node: BaseROS2DeviceNode
|
||||||
|
|
||||||
|
def post_init(self, ros_node: BaseROS2DeviceNode):
|
||||||
|
"""ROS节点注入"""
|
||||||
|
self._ros_node = ros_node
|
||||||
|
```
|
||||||
|
|
||||||
|
### 状态属性发布
|
||||||
|
|
||||||
|
设备的 `@property` 属性会自动发布为 ROS Topic:
|
||||||
|
|
||||||
|
```python
|
||||||
|
class MyDevice:
|
||||||
|
@property
|
||||||
|
def temperature(self) -> float:
|
||||||
|
return self._temperature
|
||||||
|
|
||||||
|
# 自动发布到 /{namespace}/temperature Topic
|
||||||
|
```
|
||||||
|
|
||||||
|
### Topic 配置装饰器
|
||||||
|
|
||||||
|
```python
|
||||||
|
from unilabos.utils.decorator import topic_config
|
||||||
|
|
||||||
|
class MyDevice:
|
||||||
|
@property
|
||||||
|
@topic_config(period=1.0, print_publish=False, qos=10)
|
||||||
|
def fast_data(self) -> float:
|
||||||
|
"""高频数据 - 每秒发布一次"""
|
||||||
|
return self._fast_data
|
||||||
|
|
||||||
|
@property
|
||||||
|
@topic_config(period=5.0)
|
||||||
|
def slow_data(self) -> str:
|
||||||
|
"""低频数据 - 每5秒发布一次"""
|
||||||
|
return self._slow_data
|
||||||
|
```
|
||||||
|
|
||||||
|
### 订阅装饰器
|
||||||
|
|
||||||
|
```python
|
||||||
|
from unilabos.utils.decorator import subscribe
|
||||||
|
|
||||||
|
class MyDevice:
|
||||||
|
@subscribe(topic="/external/sensor_data", qos=10)
|
||||||
|
def on_sensor_data(self, msg):
|
||||||
|
"""订阅外部Topic"""
|
||||||
|
self._sensor_value = msg.data
|
||||||
|
```
|
||||||
|
|
||||||
|
## 异步操作
|
||||||
|
|
||||||
|
### 使用 ROS 节点睡眠
|
||||||
|
|
||||||
|
```python
|
||||||
|
# 推荐:使用ROS节点的睡眠方法
|
||||||
|
await self._ros_node.sleep(1.0)
|
||||||
|
|
||||||
|
# 不推荐:直接使用asyncio(可能导致回调阻塞)
|
||||||
|
await asyncio.sleep(1.0)
|
||||||
|
```
|
||||||
|
|
||||||
|
### 获取事件循环
|
||||||
|
|
||||||
|
```python
|
||||||
|
from unilabos.ros.x.rclpyx import get_event_loop
|
||||||
|
|
||||||
|
loop = get_event_loop()
|
||||||
|
```
|
||||||
|
|
||||||
|
## 消息类型
|
||||||
|
|
||||||
|
### unilabos_msgs 包
|
||||||
|
|
||||||
|
```python
|
||||||
|
from unilabos_msgs.msg import Resource
|
||||||
|
from unilabos_msgs.srv import (
|
||||||
|
ResourceAdd,
|
||||||
|
ResourceDelete,
|
||||||
|
ResourceUpdate,
|
||||||
|
ResourceList,
|
||||||
|
SerialCommand,
|
||||||
|
)
|
||||||
|
from unilabos_msgs.action import SendCmd
|
||||||
|
```
|
||||||
|
|
||||||
|
### Resource 消息结构
|
||||||
|
|
||||||
|
```python
|
||||||
|
Resource:
|
||||||
|
id: str
|
||||||
|
name: str
|
||||||
|
category: str
|
||||||
|
type: str
|
||||||
|
parent: str
|
||||||
|
children: List[str]
|
||||||
|
config: str # JSON字符串
|
||||||
|
data: str # JSON字符串
|
||||||
|
sample_id: str
|
||||||
|
pose: Pose
|
||||||
|
```
|
||||||
|
|
||||||
|
## 日志适配器
|
||||||
|
|
||||||
|
```python
|
||||||
|
from unilabos.utils.log import info, debug, warning, error, trace
|
||||||
|
|
||||||
|
class MyDevice:
|
||||||
|
def __init__(self):
|
||||||
|
# 创建设备专属日志器
|
||||||
|
self.logger = logging.getLogger(f"MyDevice.{self.device_id}")
|
||||||
|
```
|
||||||
|
|
||||||
|
ROSLoggerAdapter 同时向自定义日志和 ROS 日志发送消息。
|
||||||
|
|
||||||
|
## Action Server
|
||||||
|
|
||||||
|
设备动作自动创建为 ROS Action Server:
|
||||||
|
|
||||||
|
```yaml
|
||||||
|
# 在注册表中配置
|
||||||
|
action_value_mappings:
|
||||||
|
my_action:
|
||||||
|
type: UniLabJsonCommandAsync # 异步Action
|
||||||
|
goal: {...}
|
||||||
|
feedback: {...}
|
||||||
|
result: {...}
|
||||||
|
```
|
||||||
|
|
||||||
|
### Action 类型
|
||||||
|
|
||||||
|
- **UniLabJsonCommand**: 同步动作
|
||||||
|
- **UniLabJsonCommandAsync**: 异步动作(支持feedback)
|
||||||
|
|
||||||
|
## 服务客户端
|
||||||
|
|
||||||
|
```python
|
||||||
|
from rclpy.client import Client
|
||||||
|
|
||||||
|
# 调用其他节点的服务
|
||||||
|
response = await self._ros_node.call_service(
|
||||||
|
service_name="/other_node/service",
|
||||||
|
request=MyServiceRequest(...)
|
||||||
|
)
|
||||||
|
```
|
||||||
|
|
||||||
|
## 命名空间
|
||||||
|
|
||||||
|
设备节点使用命名空间隔离:
|
||||||
|
|
||||||
|
```
|
||||||
|
/{device_id}/ # 设备命名空间
|
||||||
|
/{device_id}/status # 状态Topic
|
||||||
|
/{device_id}/temperature # 温度Topic
|
||||||
|
/{device_id}/my_action # 动作Server
|
||||||
|
```
|
||||||
|
|
||||||
|
## 调试
|
||||||
|
|
||||||
|
### 查看 Topic
|
||||||
|
|
||||||
|
```bash
|
||||||
|
ros2 topic list
|
||||||
|
ros2 topic echo /{device_id}/status
|
||||||
|
```
|
||||||
|
|
||||||
|
### 查看 Action
|
||||||
|
|
||||||
|
```bash
|
||||||
|
ros2 action list
|
||||||
|
ros2 action info /{device_id}/my_action
|
||||||
|
```
|
||||||
|
|
||||||
|
### 查看 Service
|
||||||
|
|
||||||
|
```bash
|
||||||
|
ros2 service list
|
||||||
|
ros2 service call /{device_id}/resource_list unilabos_msgs/srv/ResourceList
|
||||||
|
```
|
||||||
|
|
||||||
|
## 最佳实践
|
||||||
|
|
||||||
|
1. **状态属性命名**: 使用蛇形命名法(snake_case)
|
||||||
|
2. **Topic 频率**: 根据数据变化频率调整,避免过高频率
|
||||||
|
3. **Action 反馈**: 长时间操作提供进度反馈
|
||||||
|
4. **错误处理**: 使用 try-except 捕获并记录错误
|
||||||
|
5. **资源清理**: 在 cleanup 方法中正确清理资源
|
||||||
357
.cursor/rules/testing-patterns.mdc
Normal file
357
.cursor/rules/testing-patterns.mdc
Normal file
@@ -0,0 +1,357 @@
|
|||||||
|
---
|
||||||
|
description: 测试开发规范
|
||||||
|
globs: ["tests/**/*.py", "**/test_*.py"]
|
||||||
|
---
|
||||||
|
|
||||||
|
# 测试开发规范
|
||||||
|
|
||||||
|
## 目录结构
|
||||||
|
|
||||||
|
```
|
||||||
|
tests/
|
||||||
|
├── __init__.py
|
||||||
|
├── devices/ # 设备测试
|
||||||
|
│ └── liquid_handling/
|
||||||
|
│ └── test_transfer_liquid.py
|
||||||
|
├── resources/ # 资源测试
|
||||||
|
│ ├── test_bottle_carrier.py
|
||||||
|
│ └── test_resourcetreeset.py
|
||||||
|
├── ros/ # ROS消息测试
|
||||||
|
│ └── msgs/
|
||||||
|
│ ├── test_basic.py
|
||||||
|
│ ├── test_conversion.py
|
||||||
|
│ └── test_mapping.py
|
||||||
|
└── workflow/ # 工作流测试
|
||||||
|
└── merge_workflow.py
|
||||||
|
```
|
||||||
|
|
||||||
|
## 测试框架
|
||||||
|
|
||||||
|
使用 pytest 作为测试框架:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 运行所有测试
|
||||||
|
pytest tests/
|
||||||
|
|
||||||
|
# 运行特定测试文件
|
||||||
|
pytest tests/resources/test_bottle_carrier.py
|
||||||
|
|
||||||
|
# 运行特定测试函数
|
||||||
|
pytest tests/resources/test_bottle_carrier.py::test_bottle_carrier
|
||||||
|
|
||||||
|
# 显示详细输出
|
||||||
|
pytest -v tests/
|
||||||
|
|
||||||
|
# 显示打印输出
|
||||||
|
pytest -s tests/
|
||||||
|
```
|
||||||
|
|
||||||
|
## 测试文件模板
|
||||||
|
|
||||||
|
```python
|
||||||
|
import pytest
|
||||||
|
from typing import List, Dict, Any
|
||||||
|
|
||||||
|
# 导入被测试的模块
|
||||||
|
from unilabos.resources.bioyond.bottle_carriers import (
|
||||||
|
BIOYOND_Electrolyte_6VialCarrier,
|
||||||
|
)
|
||||||
|
from unilabos.resources.bioyond.bottles import (
|
||||||
|
BIOYOND_PolymerStation_Solid_Vial,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class TestBottleCarrier:
|
||||||
|
"""BottleCarrier 测试类"""
|
||||||
|
|
||||||
|
def setup_method(self):
|
||||||
|
"""每个测试方法前执行"""
|
||||||
|
self.carrier = BIOYOND_Electrolyte_6VialCarrier("test_carrier")
|
||||||
|
|
||||||
|
def teardown_method(self):
|
||||||
|
"""每个测试方法后执行"""
|
||||||
|
pass
|
||||||
|
|
||||||
|
def test_carrier_creation(self):
|
||||||
|
"""测试载架创建"""
|
||||||
|
assert self.carrier.name == "test_carrier"
|
||||||
|
assert len(self.carrier.sites) == 6
|
||||||
|
|
||||||
|
def test_bottle_placement(self):
|
||||||
|
"""测试瓶子放置"""
|
||||||
|
bottle = BIOYOND_PolymerStation_Solid_Vial("test_bottle")
|
||||||
|
# 测试逻辑...
|
||||||
|
assert bottle.name == "test_bottle"
|
||||||
|
|
||||||
|
|
||||||
|
def test_standalone_function():
|
||||||
|
"""独立测试函数"""
|
||||||
|
result = some_function()
|
||||||
|
assert result is True
|
||||||
|
|
||||||
|
|
||||||
|
# 参数化测试
|
||||||
|
@pytest.mark.parametrize("input,expected", [
|
||||||
|
("5 min", 300.0),
|
||||||
|
("1 h", 3600.0),
|
||||||
|
("120", 120.0),
|
||||||
|
(60, 60.0),
|
||||||
|
])
|
||||||
|
def test_time_parsing(input, expected):
|
||||||
|
"""测试时间解析"""
|
||||||
|
from unilabos.compile.utils.unit_parser import parse_time_input
|
||||||
|
assert parse_time_input(input) == expected
|
||||||
|
|
||||||
|
|
||||||
|
# 异常测试
|
||||||
|
def test_invalid_input_raises_error():
|
||||||
|
"""测试无效输入抛出异常"""
|
||||||
|
with pytest.raises(ValueError) as exc_info:
|
||||||
|
invalid_function("bad_input")
|
||||||
|
assert "invalid" in str(exc_info.value).lower()
|
||||||
|
|
||||||
|
|
||||||
|
# 跳过条件测试
|
||||||
|
@pytest.mark.skipif(
|
||||||
|
not os.environ.get("ROS_DISTRO"),
|
||||||
|
reason="需要ROS环境"
|
||||||
|
)
|
||||||
|
def test_ros_feature():
|
||||||
|
"""需要ROS环境的测试"""
|
||||||
|
pass
|
||||||
|
```
|
||||||
|
|
||||||
|
## 设备测试
|
||||||
|
|
||||||
|
### 虚拟设备测试
|
||||||
|
|
||||||
|
```python
|
||||||
|
import pytest
|
||||||
|
import asyncio
|
||||||
|
from unittest.mock import MagicMock, AsyncMock
|
||||||
|
|
||||||
|
from unilabos.devices.virtual.virtual_stirrer import VirtualStirrer
|
||||||
|
|
||||||
|
|
||||||
|
class TestVirtualStirrer:
|
||||||
|
"""VirtualStirrer 测试"""
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def stirrer(self):
|
||||||
|
"""创建测试用搅拌器"""
|
||||||
|
device = VirtualStirrer(
|
||||||
|
device_id="test_stirrer",
|
||||||
|
config={"max_speed": 1500.0, "min_speed": 50.0}
|
||||||
|
)
|
||||||
|
|
||||||
|
# Mock ROS节点
|
||||||
|
mock_node = MagicMock()
|
||||||
|
mock_node.sleep = AsyncMock(return_value=None)
|
||||||
|
device.post_init(mock_node)
|
||||||
|
|
||||||
|
return device
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_initialize(self, stirrer):
|
||||||
|
"""测试初始化"""
|
||||||
|
result = await stirrer.initialize()
|
||||||
|
assert result is True
|
||||||
|
assert stirrer.status == "待机中"
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_stir_action(self, stirrer):
|
||||||
|
"""测试搅拌动作"""
|
||||||
|
await stirrer.initialize()
|
||||||
|
|
||||||
|
result = await stirrer.stir(
|
||||||
|
stir_time=5.0,
|
||||||
|
stir_speed=300.0,
|
||||||
|
settling_time=2.0
|
||||||
|
)
|
||||||
|
|
||||||
|
assert result is True
|
||||||
|
assert stirrer.operation_mode == "Completed"
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_stir_invalid_speed(self, stirrer):
|
||||||
|
"""测试无效速度"""
|
||||||
|
await stirrer.initialize()
|
||||||
|
|
||||||
|
# 速度超出范围
|
||||||
|
result = await stirrer.stir(
|
||||||
|
stir_time=5.0,
|
||||||
|
stir_speed=2000.0, # 超过max_speed
|
||||||
|
settling_time=0.0
|
||||||
|
)
|
||||||
|
|
||||||
|
assert result is False
|
||||||
|
assert "错误" in stirrer.status
|
||||||
|
```
|
||||||
|
|
||||||
|
### 异步测试配置
|
||||||
|
|
||||||
|
```python
|
||||||
|
# conftest.py
|
||||||
|
import pytest
|
||||||
|
import asyncio
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture(scope="session")
|
||||||
|
def event_loop():
|
||||||
|
"""创建事件循环"""
|
||||||
|
loop = asyncio.get_event_loop_policy().new_event_loop()
|
||||||
|
yield loop
|
||||||
|
loop.close()
|
||||||
|
```
|
||||||
|
|
||||||
|
## 资源测试
|
||||||
|
|
||||||
|
```python
|
||||||
|
import pytest
|
||||||
|
from unilabos.resources.resource_tracker import (
|
||||||
|
ResourceTreeSet,
|
||||||
|
ResourceTreeInstance,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def test_resource_tree_creation():
|
||||||
|
"""测试资源树创建"""
|
||||||
|
tree_set = ResourceTreeSet()
|
||||||
|
|
||||||
|
# 添加资源
|
||||||
|
resource = {"id": "res_1", "name": "Resource 1"}
|
||||||
|
tree_set.add_resource(resource)
|
||||||
|
|
||||||
|
# 验证
|
||||||
|
assert len(tree_set.all_nodes) == 1
|
||||||
|
assert tree_set.get_resource("res_1") is not None
|
||||||
|
|
||||||
|
|
||||||
|
def test_resource_tree_merge():
|
||||||
|
"""测试资源树合并"""
|
||||||
|
local_set = ResourceTreeSet()
|
||||||
|
remote_set = ResourceTreeSet()
|
||||||
|
|
||||||
|
# 设置数据...
|
||||||
|
|
||||||
|
local_set.merge_remote_resources(remote_set)
|
||||||
|
|
||||||
|
# 验证合并结果...
|
||||||
|
```
|
||||||
|
|
||||||
|
## ROS 消息测试
|
||||||
|
|
||||||
|
```python
|
||||||
|
import pytest
|
||||||
|
from unilabos.ros.msgs.message_converter import (
|
||||||
|
convert_to_ros_msg,
|
||||||
|
convert_from_ros_msg_with_mapping,
|
||||||
|
msg_converter_manager,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def test_message_conversion():
|
||||||
|
"""测试消息转换"""
|
||||||
|
# Python -> ROS
|
||||||
|
python_data = {"id": "test", "value": 42}
|
||||||
|
ros_msg = convert_to_ros_msg(python_data, MyMsgType)
|
||||||
|
|
||||||
|
assert ros_msg.id == "test"
|
||||||
|
assert ros_msg.value == 42
|
||||||
|
|
||||||
|
# ROS -> Python
|
||||||
|
result = convert_from_ros_msg_with_mapping(ros_msg, mapping)
|
||||||
|
assert result["id"] == "test"
|
||||||
|
```
|
||||||
|
|
||||||
|
## 协议测试
|
||||||
|
|
||||||
|
```python
|
||||||
|
import pytest
|
||||||
|
import networkx as nx
|
||||||
|
from unilabos.compile.stir_protocol import (
|
||||||
|
generate_stir_protocol,
|
||||||
|
extract_vessel_id,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def topology_graph():
|
||||||
|
"""创建测试拓扑图"""
|
||||||
|
G = nx.DiGraph()
|
||||||
|
G.add_node("flask_1", **{"class": "flask"})
|
||||||
|
G.add_node("stirrer_1", **{"class": "virtual_stirrer"})
|
||||||
|
G.add_edge("stirrer_1", "flask_1")
|
||||||
|
return G
|
||||||
|
|
||||||
|
|
||||||
|
def test_generate_stir_protocol(topology_graph):
|
||||||
|
"""测试搅拌协议生成"""
|
||||||
|
actions = generate_stir_protocol(
|
||||||
|
G=topology_graph,
|
||||||
|
vessel="flask_1",
|
||||||
|
time="5 min",
|
||||||
|
stir_speed=300.0
|
||||||
|
)
|
||||||
|
|
||||||
|
assert len(actions) == 1
|
||||||
|
assert actions[0]["device_id"] == "stirrer_1"
|
||||||
|
assert actions[0]["action_name"] == "stir"
|
||||||
|
|
||||||
|
|
||||||
|
def test_extract_vessel_id():
|
||||||
|
"""测试vessel_id提取"""
|
||||||
|
# 字典格式
|
||||||
|
assert extract_vessel_id({"id": "flask_1"}) == "flask_1"
|
||||||
|
|
||||||
|
# 字符串格式
|
||||||
|
assert extract_vessel_id("flask_2") == "flask_2"
|
||||||
|
|
||||||
|
# 空值
|
||||||
|
assert extract_vessel_id("") == ""
|
||||||
|
```
|
||||||
|
|
||||||
|
## 测试标记
|
||||||
|
|
||||||
|
```python
|
||||||
|
# 慢速测试
|
||||||
|
@pytest.mark.slow
|
||||||
|
def test_long_running():
|
||||||
|
pass
|
||||||
|
|
||||||
|
# 需要网络
|
||||||
|
@pytest.mark.network
|
||||||
|
def test_network_call():
|
||||||
|
pass
|
||||||
|
|
||||||
|
# 需要ROS
|
||||||
|
@pytest.mark.ros
|
||||||
|
def test_ros_feature():
|
||||||
|
pass
|
||||||
|
```
|
||||||
|
|
||||||
|
运行特定标记的测试:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
pytest -m "not slow" # 排除慢速测试
|
||||||
|
pytest -m ros # 仅ROS测试
|
||||||
|
```
|
||||||
|
|
||||||
|
## 覆盖率
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 生成覆盖率报告
|
||||||
|
pytest --cov=unilabos tests/
|
||||||
|
|
||||||
|
# HTML报告
|
||||||
|
pytest --cov=unilabos --cov-report=html tests/
|
||||||
|
```
|
||||||
|
|
||||||
|
## 最佳实践
|
||||||
|
|
||||||
|
1. **测试命名**: `test_{功能}_{场景}_{预期结果}`
|
||||||
|
2. **独立性**: 每个测试独立运行,不依赖其他测试
|
||||||
|
3. **Mock外部依赖**: 使用 unittest.mock 模拟外部服务
|
||||||
|
4. **参数化**: 使用 `@pytest.mark.parametrize` 减少重复代码
|
||||||
|
5. **fixtures**: 使用 fixtures 共享测试设置
|
||||||
|
6. **断言清晰**: 每个断言只验证一件事
|
||||||
353
.cursor/rules/unilabos-project.mdc
Normal file
353
.cursor/rules/unilabos-project.mdc
Normal file
@@ -0,0 +1,353 @@
|
|||||||
|
---
|
||||||
|
description: Uni-Lab-OS 实验室自动化平台开发规范 - 核心规则
|
||||||
|
globs: ["**/*.py", "**/*.yaml", "**/*.json"]
|
||||||
|
---
|
||||||
|
|
||||||
|
# Uni-Lab-OS 项目开发规范
|
||||||
|
|
||||||
|
## 项目概述
|
||||||
|
|
||||||
|
Uni-Lab-OS 是一个实验室自动化操作系统,用于连接和控制各种实验设备,实现实验工作流的自动化和标准化。
|
||||||
|
|
||||||
|
## 技术栈
|
||||||
|
|
||||||
|
- **Python 3.11** - 核心开发语言
|
||||||
|
- **ROS 2** - 设备通信中间件 (rclpy)
|
||||||
|
- **Conda/Mamba** - 包管理 (robostack-staging, conda-forge)
|
||||||
|
- **FastAPI** - Web API 服务
|
||||||
|
- **WebSocket** - 实时通信
|
||||||
|
- **NetworkX** - 拓扑图管理
|
||||||
|
- **YAML** - 配置和注册表定义
|
||||||
|
- **PyLabRobot** - 实验室自动化库集成
|
||||||
|
- **pytest** - 测试框架
|
||||||
|
- **asyncio** - 异步编程
|
||||||
|
|
||||||
|
## 项目结构
|
||||||
|
|
||||||
|
```
|
||||||
|
unilabos/
|
||||||
|
├── app/ # 应用入口、Web服务、后端
|
||||||
|
├── compile/ # 协议编译器 (stir, add, filter 等)
|
||||||
|
├── config/ # 配置管理
|
||||||
|
├── devices/ # 设备驱动 (真实/虚拟)
|
||||||
|
├── device_comms/ # 设备通信协议
|
||||||
|
├── device_mesh/ # 3D网格和可视化
|
||||||
|
├── registry/ # 设备和资源类型注册表 (YAML)
|
||||||
|
├── resources/ # 资源定义
|
||||||
|
├── ros/ # ROS 2 集成
|
||||||
|
├── utils/ # 工具函数
|
||||||
|
└── workflow/ # 工作流管理
|
||||||
|
```
|
||||||
|
|
||||||
|
## 代码规范
|
||||||
|
|
||||||
|
### Python 风格
|
||||||
|
|
||||||
|
1. **类型注解**:所有函数必须使用类型注解
|
||||||
|
```python
|
||||||
|
def transfer_liquid(
|
||||||
|
source: str,
|
||||||
|
destination: str,
|
||||||
|
volume: float,
|
||||||
|
**kwargs
|
||||||
|
) -> List[Dict[str, Any]]:
|
||||||
|
```
|
||||||
|
|
||||||
|
2. **Docstring**:使用 Google 风格的文档字符串
|
||||||
|
```python
|
||||||
|
def initialize(self) -> bool:
|
||||||
|
"""
|
||||||
|
初始化设备
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
bool: 初始化是否成功
|
||||||
|
"""
|
||||||
|
```
|
||||||
|
|
||||||
|
3. **导入顺序**:
|
||||||
|
- 标准库
|
||||||
|
- 第三方库
|
||||||
|
- ROS 相关 (rclpy, unilabos_msgs)
|
||||||
|
- 项目内部模块
|
||||||
|
|
||||||
|
### 异步编程
|
||||||
|
|
||||||
|
1. 设备操作方法使用 `async def`
|
||||||
|
2. 使用 `await self._ros_node.sleep()` 而非 `asyncio.sleep()`
|
||||||
|
3. 长时间运行操作需提供进度反馈
|
||||||
|
|
||||||
|
```python
|
||||||
|
async def stir(self, stir_time: float, stir_speed: float, **kwargs) -> bool:
|
||||||
|
"""执行搅拌操作"""
|
||||||
|
start_time = time_module.time()
|
||||||
|
while True:
|
||||||
|
elapsed = time_module.time() - start_time
|
||||||
|
remaining = max(0, stir_time - elapsed)
|
||||||
|
|
||||||
|
self.data.update({
|
||||||
|
"remaining_time": remaining,
|
||||||
|
"status": f"搅拌中: {stir_speed} RPM"
|
||||||
|
})
|
||||||
|
|
||||||
|
if remaining <= 0:
|
||||||
|
break
|
||||||
|
await self._ros_node.sleep(1.0)
|
||||||
|
return True
|
||||||
|
```
|
||||||
|
|
||||||
|
### 日志规范
|
||||||
|
|
||||||
|
使用项目自定义日志系统:
|
||||||
|
|
||||||
|
```python
|
||||||
|
from unilabos.utils.log import logger, info, debug, warning, error, trace
|
||||||
|
|
||||||
|
# 在设备类中使用
|
||||||
|
self.logger = logging.getLogger(f"DeviceName.{self.device_id}")
|
||||||
|
self.logger.info("设备初始化完成")
|
||||||
|
```
|
||||||
|
|
||||||
|
## 设备驱动开发
|
||||||
|
|
||||||
|
### 设备类结构
|
||||||
|
|
||||||
|
```python
|
||||||
|
from unilabos.ros.nodes.base_device_node import BaseROS2DeviceNode
|
||||||
|
|
||||||
|
class MyDevice:
|
||||||
|
"""设备驱动类"""
|
||||||
|
|
||||||
|
_ros_node: BaseROS2DeviceNode
|
||||||
|
|
||||||
|
def __init__(self, device_id: str = None, config: Dict[str, Any] = None, **kwargs):
|
||||||
|
self.device_id = device_id or "unknown_device"
|
||||||
|
self.config = config or {}
|
||||||
|
self.data = {} # 设备状态数据
|
||||||
|
|
||||||
|
def post_init(self, ros_node: BaseROS2DeviceNode):
|
||||||
|
"""ROS节点注入"""
|
||||||
|
self._ros_node = ros_node
|
||||||
|
|
||||||
|
async def initialize(self) -> bool:
|
||||||
|
"""初始化设备"""
|
||||||
|
pass
|
||||||
|
|
||||||
|
async def cleanup(self) -> bool:
|
||||||
|
"""清理设备"""
|
||||||
|
pass
|
||||||
|
|
||||||
|
# 状态属性 - 自动发布为 ROS Topic
|
||||||
|
@property
|
||||||
|
def status(self) -> str:
|
||||||
|
return self.data.get("status", "待机")
|
||||||
|
```
|
||||||
|
|
||||||
|
### 状态属性装饰器
|
||||||
|
|
||||||
|
```python
|
||||||
|
from unilabos.utils.decorator import topic_config
|
||||||
|
|
||||||
|
class MyDevice:
|
||||||
|
@property
|
||||||
|
@topic_config(period=1.0, qos=10) # 每秒发布一次
|
||||||
|
def temperature(self) -> float:
|
||||||
|
return self._temperature
|
||||||
|
```
|
||||||
|
|
||||||
|
### 虚拟设备
|
||||||
|
|
||||||
|
虚拟设备放置在 `unilabos/devices/virtual/` 目录下,命名为 `virtual_*.py`
|
||||||
|
|
||||||
|
## 注册表配置
|
||||||
|
|
||||||
|
### 设备注册表 (YAML)
|
||||||
|
|
||||||
|
位置: `unilabos/registry/devices/*.yaml`
|
||||||
|
|
||||||
|
```yaml
|
||||||
|
my_device_type:
|
||||||
|
category:
|
||||||
|
- my_category
|
||||||
|
description: "设备描述"
|
||||||
|
version: "1.0.0"
|
||||||
|
class:
|
||||||
|
module: "unilabos.devices.my_device:MyDevice"
|
||||||
|
type: python
|
||||||
|
status_types:
|
||||||
|
status: String
|
||||||
|
temperature: Float64
|
||||||
|
action_value_mappings:
|
||||||
|
auto-initialize:
|
||||||
|
type: UniLabJsonCommandAsync
|
||||||
|
goal: {}
|
||||||
|
feedback: {}
|
||||||
|
result: {}
|
||||||
|
schema: {...}
|
||||||
|
```
|
||||||
|
|
||||||
|
### 资源注册表 (YAML)
|
||||||
|
|
||||||
|
位置: `unilabos/registry/resources/**/*.yaml`
|
||||||
|
|
||||||
|
```yaml
|
||||||
|
my_container:
|
||||||
|
category:
|
||||||
|
- container
|
||||||
|
class:
|
||||||
|
module: "unilabos.resources.my_resource:MyContainer"
|
||||||
|
type: pylabrobot
|
||||||
|
version: "1.0.0"
|
||||||
|
```
|
||||||
|
|
||||||
|
## 协议编译器
|
||||||
|
|
||||||
|
位置: `unilabos/compile/*_protocol.py`
|
||||||
|
|
||||||
|
### 协议生成函数模板
|
||||||
|
|
||||||
|
```python
|
||||||
|
from typing import List, Dict, Any, Union
|
||||||
|
import networkx as nx
|
||||||
|
|
||||||
|
def generate_my_protocol(
|
||||||
|
G: nx.DiGraph,
|
||||||
|
vessel: Union[str, dict],
|
||||||
|
param1: float = 0.0,
|
||||||
|
**kwargs
|
||||||
|
) -> List[Dict[str, Any]]:
|
||||||
|
"""
|
||||||
|
生成操作协议序列
|
||||||
|
|
||||||
|
Args:
|
||||||
|
G: 物理拓扑图
|
||||||
|
vessel: 容器ID或字典
|
||||||
|
param1: 参数1
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
List[Dict]: 动作序列
|
||||||
|
"""
|
||||||
|
# 提取vessel_id
|
||||||
|
vessel_id = vessel if isinstance(vessel, str) else vessel.get("id", "")
|
||||||
|
|
||||||
|
# 查找设备
|
||||||
|
device_id = find_connected_device(G, vessel_id)
|
||||||
|
|
||||||
|
# 生成动作
|
||||||
|
action_sequence = [{
|
||||||
|
"device_id": device_id,
|
||||||
|
"action_name": "my_action",
|
||||||
|
"action_kwargs": {
|
||||||
|
"vessel": {"id": vessel_id},
|
||||||
|
"param1": float(param1)
|
||||||
|
}
|
||||||
|
}]
|
||||||
|
|
||||||
|
return action_sequence
|
||||||
|
```
|
||||||
|
|
||||||
|
## 测试规范
|
||||||
|
|
||||||
|
### 测试文件位置
|
||||||
|
|
||||||
|
- 单元测试: `tests/` 目录
|
||||||
|
- 设备测试: `tests/devices/`
|
||||||
|
- 资源测试: `tests/resources/`
|
||||||
|
- ROS消息测试: `tests/ros/msgs/`
|
||||||
|
|
||||||
|
### 测试命名
|
||||||
|
|
||||||
|
```python
|
||||||
|
# tests/devices/my_device/test_my_device.py
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
def test_device_initialization():
|
||||||
|
"""测试设备初始化"""
|
||||||
|
pass
|
||||||
|
|
||||||
|
def test_device_action():
|
||||||
|
"""测试设备动作"""
|
||||||
|
pass
|
||||||
|
```
|
||||||
|
|
||||||
|
## 错误处理
|
||||||
|
|
||||||
|
```python
|
||||||
|
from unilabos.utils.exception import UniLabException
|
||||||
|
|
||||||
|
try:
|
||||||
|
result = await device.execute_action()
|
||||||
|
except ValueError as e:
|
||||||
|
self.logger.error(f"参数错误: {e}")
|
||||||
|
self.data["status"] = "错误: 参数无效"
|
||||||
|
return False
|
||||||
|
except Exception as e:
|
||||||
|
self.logger.error(f"执行失败: {e}")
|
||||||
|
raise
|
||||||
|
```
|
||||||
|
|
||||||
|
## 配置管理
|
||||||
|
|
||||||
|
```python
|
||||||
|
from unilabos.config.config import BasicConfig, HTTPConfig
|
||||||
|
|
||||||
|
# 读取配置
|
||||||
|
port = BasicConfig.port
|
||||||
|
is_host = BasicConfig.is_host_mode
|
||||||
|
|
||||||
|
# 配置文件: local_config.py
|
||||||
|
```
|
||||||
|
|
||||||
|
## 常用工具
|
||||||
|
|
||||||
|
### 单例模式
|
||||||
|
|
||||||
|
```python
|
||||||
|
from unilabos.utils.decorator import singleton
|
||||||
|
|
||||||
|
@singleton
|
||||||
|
class MyManager:
|
||||||
|
pass
|
||||||
|
```
|
||||||
|
|
||||||
|
### 类型检查
|
||||||
|
|
||||||
|
```python
|
||||||
|
from unilabos.utils.type_check import NoAliasDumper
|
||||||
|
|
||||||
|
yaml.dump(data, f, Dumper=NoAliasDumper)
|
||||||
|
```
|
||||||
|
|
||||||
|
### 导入管理
|
||||||
|
|
||||||
|
```python
|
||||||
|
from unilabos.utils.import_manager import get_class
|
||||||
|
|
||||||
|
device_class = get_class("unilabos.devices.my_device:MyDevice")
|
||||||
|
```
|
||||||
|
|
||||||
|
## Git 提交规范
|
||||||
|
|
||||||
|
提交信息格式:
|
||||||
|
```
|
||||||
|
<type>(<scope>): <subject>
|
||||||
|
|
||||||
|
<body>
|
||||||
|
```
|
||||||
|
|
||||||
|
类型:
|
||||||
|
- `feat`: 新功能
|
||||||
|
- `fix`: 修复bug
|
||||||
|
- `docs`: 文档更新
|
||||||
|
- `refactor`: 重构
|
||||||
|
- `test`: 测试相关
|
||||||
|
- `chore`: 构建/工具相关
|
||||||
|
|
||||||
|
示例:
|
||||||
|
```
|
||||||
|
feat(devices): 添加虚拟搅拌器设备
|
||||||
|
|
||||||
|
- 实现VirtualStirrer类
|
||||||
|
- 支持定时搅拌和持续搅拌模式
|
||||||
|
- 添加速度验证逻辑
|
||||||
|
```
|
||||||
188
.cursorignore
Normal file
188
.cursorignore
Normal file
@@ -0,0 +1,188 @@
|
|||||||
|
# ============================================================
|
||||||
|
# Uni-Lab-OS Cursor Ignore 配置,控制 Cursor AI 的文件索引范围
|
||||||
|
# ============================================================
|
||||||
|
|
||||||
|
# ==================== 敏感配置文件 ====================
|
||||||
|
# 本地配置(可能包含密钥)
|
||||||
|
**/local_config.py
|
||||||
|
test_config.py
|
||||||
|
local_test*.py
|
||||||
|
|
||||||
|
# 环境变量和密钥
|
||||||
|
.env
|
||||||
|
.env.*
|
||||||
|
**/.certs/
|
||||||
|
*.pem
|
||||||
|
*.key
|
||||||
|
credentials.json
|
||||||
|
secrets.yaml
|
||||||
|
|
||||||
|
# ==================== 二进制和 3D 模型文件 ====================
|
||||||
|
# 3D 模型文件(无需索引)
|
||||||
|
*.stl
|
||||||
|
*.dae
|
||||||
|
*.glb
|
||||||
|
*.gltf
|
||||||
|
*.obj
|
||||||
|
*.fbx
|
||||||
|
*.blend
|
||||||
|
|
||||||
|
# URDF/Xacro 机器人描述文件(大型XML)
|
||||||
|
*.xacro
|
||||||
|
|
||||||
|
# 图片文件
|
||||||
|
*.png
|
||||||
|
*.jpg
|
||||||
|
*.jpeg
|
||||||
|
*.gif
|
||||||
|
*.webp
|
||||||
|
*.ico
|
||||||
|
*.svg
|
||||||
|
*.bmp
|
||||||
|
|
||||||
|
# 压缩包
|
||||||
|
*.zip
|
||||||
|
*.tar
|
||||||
|
*.tar.gz
|
||||||
|
*.tgz
|
||||||
|
*.bz2
|
||||||
|
*.rar
|
||||||
|
*.7z
|
||||||
|
|
||||||
|
# ==================== Python 生成文件 ====================
|
||||||
|
__pycache__/
|
||||||
|
*.py[cod]
|
||||||
|
*$py.class
|
||||||
|
*.so
|
||||||
|
*.pyd
|
||||||
|
*.egg
|
||||||
|
*.egg-info/
|
||||||
|
.eggs/
|
||||||
|
dist/
|
||||||
|
build/
|
||||||
|
*.manifest
|
||||||
|
*.spec
|
||||||
|
|
||||||
|
# ==================== IDE 和编辑器 ====================
|
||||||
|
.idea/
|
||||||
|
.vscode/
|
||||||
|
*.swp
|
||||||
|
*.swo
|
||||||
|
*~
|
||||||
|
.#*
|
||||||
|
|
||||||
|
# ==================== 测试和覆盖率 ====================
|
||||||
|
.pytest_cache/
|
||||||
|
.coverage
|
||||||
|
.coverage.*
|
||||||
|
htmlcov/
|
||||||
|
.tox/
|
||||||
|
.nox/
|
||||||
|
coverage.xml
|
||||||
|
*.cover
|
||||||
|
|
||||||
|
# ==================== 虚拟环境 ====================
|
||||||
|
.venv/
|
||||||
|
venv/
|
||||||
|
env/
|
||||||
|
ENV/
|
||||||
|
|
||||||
|
# ==================== ROS 2 生成文件 ====================
|
||||||
|
# ROS 构建目录
|
||||||
|
build/
|
||||||
|
install/
|
||||||
|
log/
|
||||||
|
logs/
|
||||||
|
devel/
|
||||||
|
|
||||||
|
# ROS 消息生成
|
||||||
|
msg_gen/
|
||||||
|
srv_gen/
|
||||||
|
msg/*Action.msg
|
||||||
|
msg/*ActionFeedback.msg
|
||||||
|
msg/*ActionGoal.msg
|
||||||
|
msg/*ActionResult.msg
|
||||||
|
msg/*Feedback.msg
|
||||||
|
msg/*Goal.msg
|
||||||
|
msg/*Result.msg
|
||||||
|
msg/_*.py
|
||||||
|
srv/_*.py
|
||||||
|
build_isolated/
|
||||||
|
devel_isolated/
|
||||||
|
|
||||||
|
# ROS 动态配置
|
||||||
|
*.cfgc
|
||||||
|
/cfg/cpp/
|
||||||
|
/cfg/*.py
|
||||||
|
|
||||||
|
# ==================== 项目特定目录 ====================
|
||||||
|
# 工作数据目录
|
||||||
|
unilabos_data/
|
||||||
|
|
||||||
|
# 临时和输出目录
|
||||||
|
temp/
|
||||||
|
output/
|
||||||
|
cursor_docs/
|
||||||
|
configs/
|
||||||
|
|
||||||
|
# 文档构建
|
||||||
|
docs/_build/
|
||||||
|
/site
|
||||||
|
|
||||||
|
# ==================== 大型数据文件 ====================
|
||||||
|
# 点云数据
|
||||||
|
*.pcd
|
||||||
|
|
||||||
|
# GraphML 图形文件
|
||||||
|
*.graphml
|
||||||
|
|
||||||
|
# 日志文件
|
||||||
|
*.log
|
||||||
|
|
||||||
|
# 数据库
|
||||||
|
*.sqlite3
|
||||||
|
*.db
|
||||||
|
|
||||||
|
# Jupyter 检查点
|
||||||
|
.ipynb_checkpoints/
|
||||||
|
|
||||||
|
# ==================== 设备网格资源 ====================
|
||||||
|
# 3D 网格文件目录(包含大量 STL/DAE 文件)
|
||||||
|
unilabos/device_mesh/devices/**/*.stl
|
||||||
|
unilabos/device_mesh/devices/**/*.dae
|
||||||
|
unilabos/device_mesh/resources/**/*.stl
|
||||||
|
unilabos/device_mesh/resources/**/*.glb
|
||||||
|
unilabos/device_mesh/resources/**/*.xacro
|
||||||
|
|
||||||
|
# RViz 配置
|
||||||
|
*.rviz
|
||||||
|
|
||||||
|
# ==================== 系统文件 ====================
|
||||||
|
.DS_Store
|
||||||
|
Thumbs.db
|
||||||
|
desktop.ini
|
||||||
|
|
||||||
|
# ==================== 锁文件 ====================
|
||||||
|
poetry.lock
|
||||||
|
Pipfile.lock
|
||||||
|
pdm.lock
|
||||||
|
package-lock.json
|
||||||
|
yarn.lock
|
||||||
|
|
||||||
|
# ==================== 类型检查缓存 ====================
|
||||||
|
.mypy_cache/
|
||||||
|
.dmypy.json
|
||||||
|
.pytype/
|
||||||
|
.pyre/
|
||||||
|
pyrightconfig.json
|
||||||
|
|
||||||
|
# ==================== 其他 ====================
|
||||||
|
# Catkin
|
||||||
|
CATKIN_IGNORE
|
||||||
|
|
||||||
|
# Eclipse/Qt
|
||||||
|
.project
|
||||||
|
.cproject
|
||||||
|
CMakeLists.txt.user
|
||||||
|
*.user
|
||||||
|
qtcreator-*
|
||||||
@@ -2,7 +2,6 @@ import json
|
|||||||
import logging
|
import logging
|
||||||
import traceback
|
import traceback
|
||||||
import uuid
|
import uuid
|
||||||
import xml.etree.ElementTree as ET
|
|
||||||
from typing import Any, Dict, List
|
from typing import Any, Dict, List
|
||||||
|
|
||||||
import networkx as nx
|
import networkx as nx
|
||||||
@@ -25,7 +24,15 @@ class SimpleGraph:
|
|||||||
|
|
||||||
def add_edge(self, source, target, **attrs):
|
def add_edge(self, source, target, **attrs):
|
||||||
"""添加边"""
|
"""添加边"""
|
||||||
edge = {"source": source, "target": target, **attrs}
|
# edge = {"source": source, "target": target, **attrs}
|
||||||
|
edge = {
|
||||||
|
"source": source, "target": target,
|
||||||
|
"source_node_uuid": source,
|
||||||
|
"target_node_uuid": target,
|
||||||
|
"source_handle_io": "source",
|
||||||
|
"target_handle_io": "target",
|
||||||
|
**attrs
|
||||||
|
}
|
||||||
self.edges.append(edge)
|
self.edges.append(edge)
|
||||||
|
|
||||||
def to_dict(self):
|
def to_dict(self):
|
||||||
@@ -42,6 +49,7 @@ class SimpleGraph:
|
|||||||
"multigraph": False,
|
"multigraph": False,
|
||||||
"graph": {},
|
"graph": {},
|
||||||
"nodes": nodes_list,
|
"nodes": nodes_list,
|
||||||
|
"edges": self.edges,
|
||||||
"links": self.edges,
|
"links": self.edges,
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -58,495 +66,8 @@ def extract_json_from_markdown(text: str) -> str:
|
|||||||
return text
|
return text
|
||||||
|
|
||||||
|
|
||||||
def convert_to_type(val: str) -> Any:
|
|
||||||
"""将字符串值转换为适当的数据类型"""
|
|
||||||
if val == "True":
|
|
||||||
return True
|
|
||||||
if val == "False":
|
|
||||||
return False
|
|
||||||
if val == "?":
|
|
||||||
return None
|
|
||||||
if val.endswith(" g"):
|
|
||||||
return float(val.split(" ")[0])
|
|
||||||
if val.endswith("mg"):
|
|
||||||
return float(val.split("mg")[0])
|
|
||||||
elif val.endswith("mmol"):
|
|
||||||
return float(val.split("mmol")[0]) / 1000
|
|
||||||
elif val.endswith("mol"):
|
|
||||||
return float(val.split("mol")[0])
|
|
||||||
elif val.endswith("ml"):
|
|
||||||
return float(val.split("ml")[0])
|
|
||||||
elif val.endswith("RPM"):
|
|
||||||
return float(val.split("RPM")[0])
|
|
||||||
elif val.endswith(" °C"):
|
|
||||||
return float(val.split(" ")[0])
|
|
||||||
elif val.endswith(" %"):
|
|
||||||
return float(val.split(" ")[0])
|
|
||||||
return val
|
|
||||||
|
|
||||||
|
|
||||||
def refactor_data(data: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
|
|
||||||
"""统一的数据重构函数,根据操作类型自动选择模板"""
|
|
||||||
refactored_data = []
|
|
||||||
|
|
||||||
# 定义操作映射,包含生物实验和有机化学的所有操作
|
|
||||||
OPERATION_MAPPING = {
|
|
||||||
# 生物实验操作
|
|
||||||
"transfer_liquid": "SynBioFactory-liquid_handler.prcxi-transfer_liquid",
|
|
||||||
"transfer": "SynBioFactory-liquid_handler.biomek-transfer",
|
|
||||||
"incubation": "SynBioFactory-liquid_handler.biomek-incubation",
|
|
||||||
"move_labware": "SynBioFactory-liquid_handler.biomek-move_labware",
|
|
||||||
"oscillation": "SynBioFactory-liquid_handler.biomek-oscillation",
|
|
||||||
# 有机化学操作
|
|
||||||
"HeatChillToTemp": "SynBioFactory-workstation-HeatChillProtocol",
|
|
||||||
"StopHeatChill": "SynBioFactory-workstation-HeatChillStopProtocol",
|
|
||||||
"StartHeatChill": "SynBioFactory-workstation-HeatChillStartProtocol",
|
|
||||||
"HeatChill": "SynBioFactory-workstation-HeatChillProtocol",
|
|
||||||
"Dissolve": "SynBioFactory-workstation-DissolveProtocol",
|
|
||||||
"Transfer": "SynBioFactory-workstation-TransferProtocol",
|
|
||||||
"Evaporate": "SynBioFactory-workstation-EvaporateProtocol",
|
|
||||||
"Recrystallize": "SynBioFactory-workstation-RecrystallizeProtocol",
|
|
||||||
"Filter": "SynBioFactory-workstation-FilterProtocol",
|
|
||||||
"Dry": "SynBioFactory-workstation-DryProtocol",
|
|
||||||
"Add": "SynBioFactory-workstation-AddProtocol",
|
|
||||||
}
|
|
||||||
|
|
||||||
UNSUPPORTED_OPERATIONS = ["Purge", "Wait", "Stir", "ResetHandling"]
|
|
||||||
|
|
||||||
for step in data:
|
|
||||||
operation = step.get("action")
|
|
||||||
if not operation or operation in UNSUPPORTED_OPERATIONS:
|
|
||||||
continue
|
|
||||||
|
|
||||||
# 处理重复操作
|
|
||||||
if operation == "Repeat":
|
|
||||||
times = step.get("times", step.get("parameters", {}).get("times", 1))
|
|
||||||
sub_steps = step.get("steps", step.get("parameters", {}).get("steps", []))
|
|
||||||
for i in range(int(times)):
|
|
||||||
sub_data = refactor_data(sub_steps)
|
|
||||||
refactored_data.extend(sub_data)
|
|
||||||
continue
|
|
||||||
|
|
||||||
# 获取模板名称
|
|
||||||
template = OPERATION_MAPPING.get(operation)
|
|
||||||
if not template:
|
|
||||||
# 自动推断模板类型
|
|
||||||
if operation.lower() in ["transfer", "incubation", "move_labware", "oscillation"]:
|
|
||||||
template = f"SynBioFactory-liquid_handler.biomek-{operation}"
|
|
||||||
else:
|
|
||||||
template = f"SynBioFactory-workstation-{operation}Protocol"
|
|
||||||
|
|
||||||
# 创建步骤数据
|
|
||||||
step_data = {
|
|
||||||
"template": template,
|
|
||||||
"description": step.get("description", step.get("purpose", f"{operation} operation")),
|
|
||||||
"lab_node_type": "Device",
|
|
||||||
"parameters": step.get("parameters", step.get("action_args", {})),
|
|
||||||
}
|
|
||||||
refactored_data.append(step_data)
|
|
||||||
|
|
||||||
return refactored_data
|
|
||||||
|
|
||||||
|
|
||||||
def build_protocol_graph(
|
|
||||||
labware_info: List[Dict[str, Any]], protocol_steps: List[Dict[str, Any]], workstation_name: str
|
|
||||||
) -> SimpleGraph:
|
|
||||||
"""统一的协议图构建函数,根据设备类型自动选择构建逻辑"""
|
|
||||||
G = SimpleGraph()
|
|
||||||
resource_last_writer = {}
|
|
||||||
LAB_NAME = "SynBioFactory"
|
|
||||||
|
|
||||||
protocol_steps = refactor_data(protocol_steps)
|
|
||||||
|
|
||||||
# 检查协议步骤中的模板来判断协议类型
|
|
||||||
has_biomek_template = any(
|
|
||||||
("biomek" in step.get("template", "")) or ("prcxi" in step.get("template", ""))
|
|
||||||
for step in protocol_steps
|
|
||||||
)
|
|
||||||
|
|
||||||
if has_biomek_template:
|
|
||||||
# 生物实验协议图构建
|
|
||||||
for labware_id, labware in labware_info.items():
|
|
||||||
node_id = str(uuid.uuid4())
|
|
||||||
|
|
||||||
labware_attrs = labware.copy()
|
|
||||||
labware_id = labware_attrs.pop("id", labware_attrs.get("name", f"labware_{uuid.uuid4()}"))
|
|
||||||
labware_attrs["description"] = labware_id
|
|
||||||
labware_attrs["lab_node_type"] = (
|
|
||||||
"Reagent" if "Plate" in str(labware_id) else "Labware" if "Rack" in str(labware_id) else "Sample"
|
|
||||||
)
|
|
||||||
labware_attrs["device_id"] = workstation_name
|
|
||||||
|
|
||||||
G.add_node(node_id, template=f"{LAB_NAME}-host_node-create_resource", **labware_attrs)
|
|
||||||
resource_last_writer[labware_id] = f"{node_id}:labware"
|
|
||||||
|
|
||||||
# 处理协议步骤
|
|
||||||
prev_node = None
|
|
||||||
for i, step in enumerate(protocol_steps):
|
|
||||||
node_id = str(uuid.uuid4())
|
|
||||||
G.add_node(node_id, **step)
|
|
||||||
|
|
||||||
# 添加控制流边
|
|
||||||
if prev_node is not None:
|
|
||||||
G.add_edge(prev_node, node_id, source_port="ready", target_port="ready")
|
|
||||||
prev_node = node_id
|
|
||||||
|
|
||||||
# 处理物料流
|
|
||||||
params = step.get("parameters", {})
|
|
||||||
if "sources" in params and params["sources"] in resource_last_writer:
|
|
||||||
source_node, source_port = resource_last_writer[params["sources"]].split(":")
|
|
||||||
G.add_edge(source_node, node_id, source_port=source_port, target_port="labware")
|
|
||||||
|
|
||||||
if "targets" in params:
|
|
||||||
resource_last_writer[params["targets"]] = f"{node_id}:labware"
|
|
||||||
|
|
||||||
# 添加协议结束节点
|
|
||||||
end_id = str(uuid.uuid4())
|
|
||||||
G.add_node(end_id, template=f"{LAB_NAME}-liquid_handler.biomek-run_protocol")
|
|
||||||
if prev_node is not None:
|
|
||||||
G.add_edge(prev_node, end_id, source_port="ready", target_port="ready")
|
|
||||||
|
|
||||||
else:
|
|
||||||
# 有机化学协议图构建
|
|
||||||
WORKSTATION_ID = workstation_name
|
|
||||||
|
|
||||||
# 为所有labware创建资源节点
|
|
||||||
for item_id, item in labware_info.items():
|
|
||||||
# item_id = item.get("id") or item.get("name", f"item_{uuid.uuid4()}")
|
|
||||||
node_id = str(uuid.uuid4())
|
|
||||||
|
|
||||||
# 判断节点类型
|
|
||||||
if item.get("type") == "hardware" or "reactor" in str(item_id).lower():
|
|
||||||
if "reactor" not in str(item_id).lower():
|
|
||||||
continue
|
|
||||||
lab_node_type = "Sample"
|
|
||||||
description = f"Prepare Reactor: {item_id}"
|
|
||||||
liquid_type = []
|
|
||||||
liquid_volume = []
|
|
||||||
else:
|
|
||||||
lab_node_type = "Reagent"
|
|
||||||
description = f"Add Reagent to Flask: {item_id}"
|
|
||||||
liquid_type = [item_id]
|
|
||||||
liquid_volume = [1e5]
|
|
||||||
|
|
||||||
G.add_node(
|
|
||||||
node_id,
|
|
||||||
template=f"{LAB_NAME}-host_node-create_resource",
|
|
||||||
description=description,
|
|
||||||
lab_node_type=lab_node_type,
|
|
||||||
res_id=item_id,
|
|
||||||
device_id=WORKSTATION_ID,
|
|
||||||
class_name="container",
|
|
||||||
parent=WORKSTATION_ID,
|
|
||||||
bind_locations={"x": 0.0, "y": 0.0, "z": 0.0},
|
|
||||||
liquid_input_slot=[-1],
|
|
||||||
liquid_type=liquid_type,
|
|
||||||
liquid_volume=liquid_volume,
|
|
||||||
slot_on_deck="",
|
|
||||||
role=item.get("role", ""),
|
|
||||||
)
|
|
||||||
resource_last_writer[item_id] = f"{node_id}:labware"
|
|
||||||
|
|
||||||
last_control_node_id = None
|
|
||||||
|
|
||||||
# 处理协议步骤
|
|
||||||
for step in protocol_steps:
|
|
||||||
node_id = str(uuid.uuid4())
|
|
||||||
G.add_node(node_id, **step)
|
|
||||||
|
|
||||||
# 控制流
|
|
||||||
if last_control_node_id is not None:
|
|
||||||
G.add_edge(last_control_node_id, node_id, source_port="ready", target_port="ready")
|
|
||||||
last_control_node_id = node_id
|
|
||||||
|
|
||||||
# 物料流
|
|
||||||
params = step.get("parameters", {})
|
|
||||||
input_resources = {
|
|
||||||
"Vessel": params.get("vessel"),
|
|
||||||
"ToVessel": params.get("to_vessel"),
|
|
||||||
"FromVessel": params.get("from_vessel"),
|
|
||||||
"reagent": params.get("reagent"),
|
|
||||||
"solvent": params.get("solvent"),
|
|
||||||
"compound": params.get("compound"),
|
|
||||||
"sources": params.get("sources"),
|
|
||||||
"targets": params.get("targets"),
|
|
||||||
}
|
|
||||||
|
|
||||||
for target_port, resource_name in input_resources.items():
|
|
||||||
if resource_name and resource_name in resource_last_writer:
|
|
||||||
source_node, source_port = resource_last_writer[resource_name].split(":")
|
|
||||||
G.add_edge(source_node, node_id, source_port=source_port, target_port=target_port)
|
|
||||||
|
|
||||||
output_resources = {
|
|
||||||
"VesselOut": params.get("vessel"),
|
|
||||||
"FromVesselOut": params.get("from_vessel"),
|
|
||||||
"ToVesselOut": params.get("to_vessel"),
|
|
||||||
"FiltrateOut": params.get("filtrate_vessel"),
|
|
||||||
"reagent": params.get("reagent"),
|
|
||||||
"solvent": params.get("solvent"),
|
|
||||||
"compound": params.get("compound"),
|
|
||||||
"sources_out": params.get("sources"),
|
|
||||||
"targets_out": params.get("targets"),
|
|
||||||
}
|
|
||||||
|
|
||||||
for source_port, resource_name in output_resources.items():
|
|
||||||
if resource_name:
|
|
||||||
resource_last_writer[resource_name] = f"{node_id}:{source_port}"
|
|
||||||
|
|
||||||
return G
|
|
||||||
|
|
||||||
|
|
||||||
def draw_protocol_graph(protocol_graph: SimpleGraph, output_path: str):
|
|
||||||
"""
|
|
||||||
(辅助功能) 使用 networkx 和 matplotlib 绘制协议工作流图,用于可视化。
|
|
||||||
"""
|
|
||||||
if not protocol_graph:
|
|
||||||
print("Cannot draw graph: Graph object is empty.")
|
|
||||||
return
|
|
||||||
|
|
||||||
G = nx.DiGraph()
|
|
||||||
|
|
||||||
for node_id, attrs in protocol_graph.nodes.items():
|
|
||||||
label = attrs.get("description", attrs.get("template", node_id[:8]))
|
|
||||||
G.add_node(node_id, label=label, **attrs)
|
|
||||||
|
|
||||||
for edge in protocol_graph.edges:
|
|
||||||
G.add_edge(edge["source"], edge["target"])
|
|
||||||
|
|
||||||
plt.figure(figsize=(20, 15))
|
|
||||||
try:
|
|
||||||
pos = nx.nx_agraph.graphviz_layout(G, prog="dot")
|
|
||||||
except Exception:
|
|
||||||
pos = nx.shell_layout(G) # Fallback layout
|
|
||||||
|
|
||||||
node_labels = {node: data["label"] for node, data in G.nodes(data=True)}
|
|
||||||
nx.draw(
|
|
||||||
G,
|
|
||||||
pos,
|
|
||||||
with_labels=False,
|
|
||||||
node_size=2500,
|
|
||||||
node_color="skyblue",
|
|
||||||
node_shape="o",
|
|
||||||
edge_color="gray",
|
|
||||||
width=1.5,
|
|
||||||
arrowsize=15,
|
|
||||||
)
|
|
||||||
nx.draw_networkx_labels(G, pos, labels=node_labels, font_size=8, font_weight="bold")
|
|
||||||
|
|
||||||
plt.title("Chemical Protocol Workflow Graph", size=15)
|
|
||||||
plt.savefig(output_path, dpi=300, bbox_inches="tight")
|
|
||||||
plt.close()
|
|
||||||
print(f" - Visualization saved to '{output_path}'")
|
|
||||||
|
|
||||||
|
|
||||||
from networkx.drawing.nx_agraph import to_agraph
|
|
||||||
import re
|
|
||||||
|
|
||||||
COMPASS = {"n","e","s","w","ne","nw","se","sw","c"}
|
|
||||||
|
|
||||||
def _is_compass(port: str) -> bool:
|
|
||||||
return isinstance(port, str) and port.lower() in COMPASS
|
|
||||||
|
|
||||||
def draw_protocol_graph_with_ports(protocol_graph, output_path: str, rankdir: str = "LR"):
|
|
||||||
"""
|
|
||||||
使用 Graphviz 端口语法绘制协议工作流图。
|
|
||||||
- 若边上的 source_port/target_port 是 compass(n/e/s/w/...),直接用 compass。
|
|
||||||
- 否则自动为节点创建 record 形状并定义命名端口 <portname>。
|
|
||||||
最终由 PyGraphviz 渲染并输出到 output_path(后缀决定格式,如 .png/.svg/.pdf)。
|
|
||||||
"""
|
|
||||||
if not protocol_graph:
|
|
||||||
print("Cannot draw graph: Graph object is empty.")
|
|
||||||
return
|
|
||||||
|
|
||||||
# 1) 先用 networkx 搭建有向图,保留端口属性
|
|
||||||
G = nx.DiGraph()
|
|
||||||
for node_id, attrs in protocol_graph.nodes.items():
|
|
||||||
label = attrs.get("description", attrs.get("template", node_id[:8]))
|
|
||||||
# 保留一个干净的“中心标签”,用于放在 record 的中间槽
|
|
||||||
G.add_node(node_id, _core_label=str(label), **{k:v for k,v in attrs.items() if k not in ("label",)})
|
|
||||||
|
|
||||||
edges_data = []
|
|
||||||
in_ports_by_node = {} # 收集命名输入端口
|
|
||||||
out_ports_by_node = {} # 收集命名输出端口
|
|
||||||
|
|
||||||
for edge in protocol_graph.edges:
|
|
||||||
u = edge["source"]
|
|
||||||
v = edge["target"]
|
|
||||||
sp = edge.get("source_port")
|
|
||||||
tp = edge.get("target_port")
|
|
||||||
|
|
||||||
# 记录到图里(保留原始端口信息)
|
|
||||||
G.add_edge(u, v, source_port=sp, target_port=tp)
|
|
||||||
edges_data.append((u, v, sp, tp))
|
|
||||||
|
|
||||||
# 如果不是 compass,就按“命名端口”先归类,等会儿给节点造 record
|
|
||||||
if sp and not _is_compass(sp):
|
|
||||||
out_ports_by_node.setdefault(u, set()).add(str(sp))
|
|
||||||
if tp and not _is_compass(tp):
|
|
||||||
in_ports_by_node.setdefault(v, set()).add(str(tp))
|
|
||||||
|
|
||||||
# 2) 转为 AGraph,使用 Graphviz 渲染
|
|
||||||
A = to_agraph(G)
|
|
||||||
A.graph_attr.update(rankdir=rankdir, splines="true", concentrate="false", fontsize="10")
|
|
||||||
A.node_attr.update(shape="box", style="rounded,filled", fillcolor="lightyellow", color="#999999", fontname="Helvetica")
|
|
||||||
A.edge_attr.update(arrowsize="0.8", color="#666666")
|
|
||||||
|
|
||||||
# 3) 为需要命名端口的节点设置 record 形状与 label
|
|
||||||
# 左列 = 输入端口;中间 = 核心标签;右列 = 输出端口
|
|
||||||
for n in A.nodes():
|
|
||||||
node = A.get_node(n)
|
|
||||||
core = G.nodes[n].get("_core_label", n)
|
|
||||||
|
|
||||||
in_ports = sorted(in_ports_by_node.get(n, []))
|
|
||||||
out_ports = sorted(out_ports_by_node.get(n, []))
|
|
||||||
|
|
||||||
# 如果该节点涉及命名端口,则用 record;否则保留原 box
|
|
||||||
if in_ports or out_ports:
|
|
||||||
def port_fields(ports):
|
|
||||||
if not ports:
|
|
||||||
return " " # 必须留一个空槽占位
|
|
||||||
# 每个端口一个小格子,<p> name
|
|
||||||
return "|".join(f"<{re.sub(r'[^A-Za-z0-9_:.|-]', '_', p)}> {p}" for p in ports)
|
|
||||||
|
|
||||||
left = port_fields(in_ports)
|
|
||||||
right = port_fields(out_ports)
|
|
||||||
|
|
||||||
# 三栏:左(入) | 中(节点名) | 右(出)
|
|
||||||
record_label = f"{{ {left} | {core} | {right} }}"
|
|
||||||
node.attr.update(shape="record", label=record_label)
|
|
||||||
else:
|
|
||||||
# 没有命名端口:普通盒子,显示核心标签
|
|
||||||
node.attr.update(label=str(core))
|
|
||||||
|
|
||||||
# 4) 给边设置 headport / tailport
|
|
||||||
# - 若端口为 compass:直接用 compass(e.g., headport="e")
|
|
||||||
# - 若端口为命名端口:使用在 record 中定义的 <port> 名(同名即可)
|
|
||||||
for (u, v, sp, tp) in edges_data:
|
|
||||||
e = A.get_edge(u, v)
|
|
||||||
|
|
||||||
# Graphviz 属性:tail 是源,head 是目标
|
|
||||||
if sp:
|
|
||||||
if _is_compass(sp):
|
|
||||||
e.attr["tailport"] = sp.lower()
|
|
||||||
else:
|
|
||||||
# 与 record label 中 <port> 名一致;特殊字符已在 label 中做了清洗
|
|
||||||
e.attr["tailport"] = re.sub(r'[^A-Za-z0-9_:.|-]', '_', str(sp))
|
|
||||||
|
|
||||||
if tp:
|
|
||||||
if _is_compass(tp):
|
|
||||||
e.attr["headport"] = tp.lower()
|
|
||||||
else:
|
|
||||||
e.attr["headport"] = re.sub(r'[^A-Za-z0-9_:.|-]', '_', str(tp))
|
|
||||||
|
|
||||||
# 可选:若想让边更贴边缘,可设置 constraint/spline 等
|
|
||||||
# e.attr["arrowhead"] = "vee"
|
|
||||||
|
|
||||||
# 5) 输出
|
|
||||||
A.draw(output_path, prog="dot")
|
|
||||||
print(f" - Port-aware workflow rendered to '{output_path}'")
|
|
||||||
|
|
||||||
|
|
||||||
def flatten_xdl_procedure(procedure_elem: ET.Element) -> List[ET.Element]:
|
|
||||||
"""展平嵌套的XDL程序结构"""
|
|
||||||
flattened_operations = []
|
|
||||||
TEMP_UNSUPPORTED_PROTOCOL = ["Purge", "Wait", "Stir", "ResetHandling"]
|
|
||||||
|
|
||||||
def extract_operations(element: ET.Element):
|
|
||||||
if element.tag not in ["Prep", "Reaction", "Workup", "Purification", "Procedure"]:
|
|
||||||
if element.tag not in TEMP_UNSUPPORTED_PROTOCOL:
|
|
||||||
flattened_operations.append(element)
|
|
||||||
|
|
||||||
for child in element:
|
|
||||||
extract_operations(child)
|
|
||||||
|
|
||||||
for child in procedure_elem:
|
|
||||||
extract_operations(child)
|
|
||||||
|
|
||||||
return flattened_operations
|
|
||||||
|
|
||||||
|
|
||||||
def parse_xdl_content(xdl_content: str) -> tuple:
|
|
||||||
"""解析XDL内容"""
|
|
||||||
try:
|
|
||||||
xdl_content_cleaned = "".join(c for c in xdl_content if c.isprintable())
|
|
||||||
root = ET.fromstring(xdl_content_cleaned)
|
|
||||||
|
|
||||||
synthesis_elem = root.find("Synthesis")
|
|
||||||
if synthesis_elem is None:
|
|
||||||
return None, None, None
|
|
||||||
|
|
||||||
# 解析硬件组件
|
|
||||||
hardware_elem = synthesis_elem.find("Hardware")
|
|
||||||
hardware = []
|
|
||||||
if hardware_elem is not None:
|
|
||||||
hardware = [{"id": c.get("id"), "type": c.get("type")} for c in hardware_elem.findall("Component")]
|
|
||||||
|
|
||||||
# 解析试剂
|
|
||||||
reagents_elem = synthesis_elem.find("Reagents")
|
|
||||||
reagents = []
|
|
||||||
if reagents_elem is not None:
|
|
||||||
reagents = [{"name": r.get("name"), "role": r.get("role", "")} for r in reagents_elem.findall("Reagent")]
|
|
||||||
|
|
||||||
# 解析程序
|
|
||||||
procedure_elem = synthesis_elem.find("Procedure")
|
|
||||||
if procedure_elem is None:
|
|
||||||
return None, None, None
|
|
||||||
|
|
||||||
flattened_operations = flatten_xdl_procedure(procedure_elem)
|
|
||||||
return hardware, reagents, flattened_operations
|
|
||||||
|
|
||||||
except ET.ParseError as e:
|
|
||||||
raise ValueError(f"Invalid XDL format: {e}")
|
|
||||||
|
|
||||||
|
|
||||||
def convert_xdl_to_dict(xdl_content: str) -> Dict[str, Any]:
|
|
||||||
"""
|
|
||||||
将XDL XML格式转换为标准的字典格式
|
|
||||||
|
|
||||||
Args:
|
|
||||||
xdl_content: XDL XML内容
|
|
||||||
|
|
||||||
Returns:
|
|
||||||
转换结果,包含步骤和器材信息
|
|
||||||
"""
|
|
||||||
try:
|
|
||||||
hardware, reagents, flattened_operations = parse_xdl_content(xdl_content)
|
|
||||||
if hardware is None:
|
|
||||||
return {"error": "Failed to parse XDL content", "success": False}
|
|
||||||
|
|
||||||
# 将XDL元素转换为字典格式
|
|
||||||
steps_data = []
|
|
||||||
for elem in flattened_operations:
|
|
||||||
# 转换参数类型
|
|
||||||
parameters = {}
|
|
||||||
for key, val in elem.attrib.items():
|
|
||||||
converted_val = convert_to_type(val)
|
|
||||||
if converted_val is not None:
|
|
||||||
parameters[key] = converted_val
|
|
||||||
|
|
||||||
step_dict = {
|
|
||||||
"operation": elem.tag,
|
|
||||||
"parameters": parameters,
|
|
||||||
"description": elem.get("purpose", f"Operation: {elem.tag}"),
|
|
||||||
}
|
|
||||||
steps_data.append(step_dict)
|
|
||||||
|
|
||||||
# 合并硬件和试剂为统一的labware_info格式
|
|
||||||
labware_data = []
|
|
||||||
labware_data.extend({"id": hw["id"], "type": "hardware", **hw} for hw in hardware)
|
|
||||||
labware_data.extend({"name": reagent["name"], "type": "reagent", **reagent} for reagent in reagents)
|
|
||||||
|
|
||||||
return {
|
|
||||||
"success": True,
|
|
||||||
"steps": steps_data,
|
|
||||||
"labware": labware_data,
|
|
||||||
"message": f"Successfully converted XDL to dict format. Found {len(steps_data)} steps and {len(labware_data)} labware items.",
|
|
||||||
}
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
error_msg = f"XDL conversion failed: {str(e)}"
|
|
||||||
logger.error(error_msg)
|
|
||||||
return {"error": error_msg, "success": False}
|
|
||||||
|
|
||||||
|
|
||||||
def create_workflow(
|
def create_workflow(
|
||||||
|
|||||||
@@ -439,7 +439,7 @@ class MessageProcessor:
|
|||||||
self.connected = True
|
self.connected = True
|
||||||
self.reconnect_count = 0
|
self.reconnect_count = 0
|
||||||
|
|
||||||
logger.info(f"[MessageProcessor] Connected to {self.websocket_url}")
|
logger.trace(f"[MessageProcessor] Connected to {self.websocket_url}")
|
||||||
|
|
||||||
# 启动发送协程
|
# 启动发送协程
|
||||||
send_task = asyncio.create_task(self._send_handler())
|
send_task = asyncio.create_task(self._send_handler())
|
||||||
@@ -517,7 +517,7 @@ class MessageProcessor:
|
|||||||
|
|
||||||
async def _send_handler(self):
|
async def _send_handler(self):
|
||||||
"""处理发送队列中的消息"""
|
"""处理发送队列中的消息"""
|
||||||
logger.debug("[MessageProcessor] Send handler started")
|
logger.trace("[MessageProcessor] Send handler started")
|
||||||
|
|
||||||
try:
|
try:
|
||||||
while self.connected and self.websocket:
|
while self.connected and self.websocket:
|
||||||
@@ -1026,7 +1026,7 @@ class QueueProcessor:
|
|||||||
|
|
||||||
def _run(self):
|
def _run(self):
|
||||||
"""运行队列处理主循环"""
|
"""运行队列处理主循环"""
|
||||||
logger.debug("[QueueProcessor] Queue processor started")
|
logger.trace("[QueueProcessor] Queue processor started")
|
||||||
|
|
||||||
while self.is_running:
|
while self.is_running:
|
||||||
try:
|
try:
|
||||||
@@ -1236,7 +1236,6 @@ class WebSocketClient(BaseCommunicationClient):
|
|||||||
else:
|
else:
|
||||||
url = f"{scheme}://{parsed.netloc}/api/v1/ws/schedule"
|
url = f"{scheme}://{parsed.netloc}/api/v1/ws/schedule"
|
||||||
|
|
||||||
logger.debug(f"[WebSocketClient] URL: {url}")
|
|
||||||
return url
|
return url
|
||||||
|
|
||||||
def start(self) -> None:
|
def start(self) -> None:
|
||||||
@@ -1249,13 +1248,11 @@ class WebSocketClient(BaseCommunicationClient):
|
|||||||
logger.error("[WebSocketClient] WebSocket URL not configured")
|
logger.error("[WebSocketClient] WebSocket URL not configured")
|
||||||
return
|
return
|
||||||
|
|
||||||
logger.info(f"[WebSocketClient] Starting connection to {self.websocket_url}")
|
|
||||||
|
|
||||||
# 启动两个核心线程
|
# 启动两个核心线程
|
||||||
self.message_processor.start()
|
self.message_processor.start()
|
||||||
self.queue_processor.start()
|
self.queue_processor.start()
|
||||||
|
|
||||||
logger.info("[WebSocketClient] All threads started")
|
logger.trace("[WebSocketClient] All threads started")
|
||||||
|
|
||||||
def stop(self) -> None:
|
def stop(self) -> None:
|
||||||
"""停止WebSocket客户端"""
|
"""停止WebSocket客户端"""
|
||||||
|
|||||||
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")
|
||||||
@@ -258,7 +258,7 @@ class BioyondResourceSynchronizer(ResourceSynchronizer):
|
|||||||
logger.info(f"[同步→Bioyond] ➕ 物料不存在于 Bioyond,将创建新物料并入库")
|
logger.info(f"[同步→Bioyond] ➕ 物料不存在于 Bioyond,将创建新物料并入库")
|
||||||
|
|
||||||
# 第1步:从配置中获取仓库配置
|
# 第1步:从配置中获取仓库配置
|
||||||
warehouse_mapping = self.bioyond_config.get("warehouse_mapping", {})
|
warehouse_mapping = self.workstation.bioyond_config.get("warehouse_mapping", {})
|
||||||
|
|
||||||
# 确定目标仓库名称
|
# 确定目标仓库名称
|
||||||
parent_name = None
|
parent_name = None
|
||||||
|
|||||||
@@ -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: 有液
|
||||||
|
|||||||
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
|
||||||
@@ -4,6 +4,8 @@ import os
|
|||||||
import sys
|
import sys
|
||||||
import inspect
|
import inspect
|
||||||
import importlib
|
import importlib
|
||||||
|
import threading
|
||||||
|
from concurrent.futures import ThreadPoolExecutor, as_completed
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import Any, Dict, List, Union, Tuple
|
from typing import Any, Dict, List, Union, Tuple
|
||||||
|
|
||||||
@@ -60,6 +62,7 @@ class Registry:
|
|||||||
self.device_module_to_registry = {}
|
self.device_module_to_registry = {}
|
||||||
self.resource_type_registry = {}
|
self.resource_type_registry = {}
|
||||||
self._setup_called = False # 跟踪setup是否已调用
|
self._setup_called = False # 跟踪setup是否已调用
|
||||||
|
self._registry_lock = threading.Lock() # 多线程加载时的锁
|
||||||
# 其他状态变量
|
# 其他状态变量
|
||||||
# self.is_host_mode = False # 移至BasicConfig中
|
# self.is_host_mode = False # 移至BasicConfig中
|
||||||
|
|
||||||
@@ -177,8 +180,7 @@ class Registry:
|
|||||||
"result": {},
|
"result": {},
|
||||||
"schema": test_latency_schema,
|
"schema": test_latency_schema,
|
||||||
"goal_default": {
|
"goal_default": {
|
||||||
arg["name"]: arg["default"]
|
arg["name"]: arg["default"] for arg in test_latency_method_info.get("args", [])
|
||||||
for arg in test_latency_method_info.get("args", [])
|
|
||||||
},
|
},
|
||||||
"handles": {},
|
"handles": {},
|
||||||
},
|
},
|
||||||
@@ -262,18 +264,26 @@ class Registry:
|
|||||||
# 标记setup已被调用
|
# 标记setup已被调用
|
||||||
self._setup_called = True
|
self._setup_called = True
|
||||||
|
|
||||||
def load_resource_types(self, path: os.PathLike, complete_registry: bool, upload_registry: bool):
|
def _load_single_resource_file(
|
||||||
abs_path = Path(path).absolute()
|
self, file: Path, complete_registry: bool, upload_registry: bool
|
||||||
resource_path = abs_path / "resources"
|
) -> Tuple[Dict[str, Any], Dict[str, Any], bool]:
|
||||||
files = list(resource_path.glob("*/*.yaml"))
|
"""
|
||||||
logger.debug(f"[UniLab Registry] resources: {resource_path.exists()}, total: {len(files)}")
|
加载单个资源文件 (线程安全)
|
||||||
current_resource_number = len(self.resource_type_registry) + 1
|
|
||||||
for i, file in enumerate(files):
|
Returns:
|
||||||
|
(data, complete_data, is_valid): 资源数据, 完整数据, 是否有效
|
||||||
|
"""
|
||||||
|
try:
|
||||||
with open(file, encoding="utf-8", mode="r") as f:
|
with open(file, encoding="utf-8", mode="r") as f:
|
||||||
data = yaml.safe_load(io.StringIO(f.read()))
|
data = yaml.safe_load(io.StringIO(f.read()))
|
||||||
|
except Exception as e:
|
||||||
|
logger.warning(f"[UniLab Registry] 读取资源文件失败: {file}, 错误: {e}")
|
||||||
|
return {}, {}, False
|
||||||
|
|
||||||
|
if not data:
|
||||||
|
return {}, {}, False
|
||||||
|
|
||||||
complete_data = {}
|
complete_data = {}
|
||||||
if data:
|
|
||||||
# 为每个资源添加文件路径信息
|
|
||||||
for resource_id, resource_info in data.items():
|
for resource_id, resource_info in data.items():
|
||||||
if "version" not in resource_info:
|
if "version" not in resource_info:
|
||||||
resource_info["version"] = "1.0.0"
|
resource_info["version"] = "1.0.0"
|
||||||
@@ -301,28 +311,68 @@ class Registry:
|
|||||||
if len(class_info) and "module" in class_info:
|
if len(class_info) and "module" in class_info:
|
||||||
if class_info.get("type") == "pylabrobot":
|
if class_info.get("type") == "pylabrobot":
|
||||||
res_class = get_class(class_info["module"])
|
res_class = get_class(class_info["module"])
|
||||||
if callable(res_class) and not isinstance(
|
if callable(res_class) and not isinstance(res_class, type):
|
||||||
res_class, type
|
|
||||||
): # 有的是类,有的是函数,这里暂时只登记函数类的
|
|
||||||
res_instance = res_class(res_class.__name__)
|
res_instance = res_class(res_class.__name__)
|
||||||
res_ulr = tree_to_list([resource_plr_to_ulab(res_instance)])
|
res_ulr = tree_to_list([resource_plr_to_ulab(res_instance)])
|
||||||
resource_info["config_info"] = res_ulr
|
resource_info["config_info"] = res_ulr
|
||||||
resource_info["registry_type"] = "resource"
|
resource_info["registry_type"] = "resource"
|
||||||
resource_info["file_path"] = str(file.absolute()).replace("\\", "/")
|
resource_info["file_path"] = str(file.absolute()).replace("\\", "/")
|
||||||
|
|
||||||
complete_data = dict(sorted(complete_data.items()))
|
complete_data = dict(sorted(complete_data.items()))
|
||||||
complete_data = copy.deepcopy(complete_data)
|
complete_data = copy.deepcopy(complete_data)
|
||||||
|
|
||||||
if complete_registry:
|
if complete_registry:
|
||||||
|
try:
|
||||||
with open(file, "w", encoding="utf-8") as f:
|
with open(file, "w", encoding="utf-8") as f:
|
||||||
yaml.dump(complete_data, f, allow_unicode=True, default_flow_style=False, Dumper=NoAliasDumper)
|
yaml.dump(complete_data, f, allow_unicode=True, default_flow_style=False, Dumper=NoAliasDumper)
|
||||||
|
except Exception as e:
|
||||||
|
logger.warning(f"[UniLab Registry] 写入资源文件失败: {file}, 错误: {e}")
|
||||||
|
|
||||||
|
return data, complete_data, True
|
||||||
|
|
||||||
|
def load_resource_types(self, path: os.PathLike, complete_registry: bool, upload_registry: bool):
|
||||||
|
abs_path = Path(path).absolute()
|
||||||
|
resource_path = abs_path / "resources"
|
||||||
|
files = list(resource_path.glob("*/*.yaml"))
|
||||||
|
logger.debug(f"[UniLab Registry] resources: {resource_path.exists()}, total: {len(files)}")
|
||||||
|
|
||||||
|
if not files:
|
||||||
|
return
|
||||||
|
|
||||||
|
# 使用线程池并行加载
|
||||||
|
max_workers = min(8, len(files))
|
||||||
|
results = []
|
||||||
|
|
||||||
|
with ThreadPoolExecutor(max_workers=max_workers) as executor:
|
||||||
|
future_to_file = {
|
||||||
|
executor.submit(self._load_single_resource_file, file, complete_registry, upload_registry): file
|
||||||
|
for file in files
|
||||||
|
}
|
||||||
|
for future in as_completed(future_to_file):
|
||||||
|
file = future_to_file[future]
|
||||||
|
try:
|
||||||
|
data, complete_data, is_valid = future.result()
|
||||||
|
if is_valid:
|
||||||
|
results.append((file, data))
|
||||||
|
except Exception as e:
|
||||||
|
logger.warning(f"[UniLab Registry] 处理资源文件异常: {file}, 错误: {e}")
|
||||||
|
|
||||||
|
# 线程安全地更新注册表
|
||||||
|
current_resource_number = len(self.resource_type_registry) + 1
|
||||||
|
with self._registry_lock:
|
||||||
|
for i, (file, data) in enumerate(results):
|
||||||
self.resource_type_registry.update(data)
|
self.resource_type_registry.update(data)
|
||||||
logger.trace( # type: ignore
|
logger.trace(
|
||||||
f"[UniLab Registry] Resource-{current_resource_number} File-{i+1}/{len(files)} "
|
f"[UniLab Registry] Resource-{current_resource_number} File-{i+1}/{len(results)} "
|
||||||
+ f"Add {list(data.keys())}"
|
+ f"Add {list(data.keys())}"
|
||||||
)
|
)
|
||||||
current_resource_number += 1
|
current_resource_number += 1
|
||||||
else:
|
|
||||||
logger.debug(f"[UniLab Registry] Res File-{i+1}/{len(files)} Not Valid YAML File: {file.absolute()}")
|
# 记录无效文件
|
||||||
|
valid_files = {r[0] for r in results}
|
||||||
|
for file in files:
|
||||||
|
if file not in valid_files:
|
||||||
|
logger.debug(f"[UniLab Registry] Res File Not Valid YAML File: {file.absolute()}")
|
||||||
|
|
||||||
def _extract_class_docstrings(self, module_string: str) -> Dict[str, str]:
|
def _extract_class_docstrings(self, module_string: str) -> Dict[str, str]:
|
||||||
"""
|
"""
|
||||||
@@ -674,32 +724,34 @@ class Registry:
|
|||||||
"handles": {},
|
"handles": {},
|
||||||
}
|
}
|
||||||
|
|
||||||
def load_device_types(self, path: os.PathLike, complete_registry: bool):
|
def _load_single_device_file(
|
||||||
# return
|
self, file: Path, complete_registry: bool, get_yaml_from_goal_type
|
||||||
abs_path = Path(path).absolute()
|
) -> Tuple[Dict[str, Any], Dict[str, Any], bool, List[str]]:
|
||||||
devices_path = abs_path / "devices"
|
"""
|
||||||
device_comms_path = abs_path / "device_comms"
|
加载单个设备文件 (线程安全)
|
||||||
files = list(devices_path.glob("*.yaml")) + list(device_comms_path.glob("*.yaml"))
|
|
||||||
logger.trace( # type: ignore
|
|
||||||
f"[UniLab Registry] devices: {devices_path.exists()}, device_comms: {device_comms_path.exists()}, "
|
|
||||||
+ f"total: {len(files)}"
|
|
||||||
)
|
|
||||||
current_device_number = len(self.device_type_registry) + 1
|
|
||||||
from unilabos.app.web.utils.action_utils import get_yaml_from_goal_type
|
|
||||||
|
|
||||||
for i, file in enumerate(files):
|
Returns:
|
||||||
|
(data, complete_data, is_valid, device_ids): 设备数据, 完整数据, 是否有效, 设备ID列表
|
||||||
|
"""
|
||||||
|
try:
|
||||||
with open(file, encoding="utf-8", mode="r") as f:
|
with open(file, encoding="utf-8", mode="r") as f:
|
||||||
data = yaml.safe_load(io.StringIO(f.read()))
|
data = yaml.safe_load(io.StringIO(f.read()))
|
||||||
|
except Exception as e:
|
||||||
|
logger.warning(f"[UniLab Registry] 读取设备文件失败: {file}, 错误: {e}")
|
||||||
|
return {}, {}, False, []
|
||||||
|
|
||||||
|
if not data:
|
||||||
|
return {}, {}, False, []
|
||||||
|
|
||||||
complete_data = {}
|
complete_data = {}
|
||||||
action_str_type_mapping = {
|
action_str_type_mapping = {
|
||||||
"UniLabJsonCommand": "UniLabJsonCommand",
|
"UniLabJsonCommand": "UniLabJsonCommand",
|
||||||
"UniLabJsonCommandAsync": "UniLabJsonCommandAsync",
|
"UniLabJsonCommandAsync": "UniLabJsonCommandAsync",
|
||||||
}
|
}
|
||||||
status_str_type_mapping = {}
|
status_str_type_mapping = {}
|
||||||
if data:
|
device_ids = []
|
||||||
# 在添加到注册表前处理类型替换
|
|
||||||
for device_id, device_config in data.items():
|
for device_id, device_config in data.items():
|
||||||
# 添加文件路径信息 - 使用规范化的完整文件路径
|
|
||||||
if "version" not in device_config:
|
if "version" not in device_config:
|
||||||
device_config["version"] = "1.0.0"
|
device_config["version"] = "1.0.0"
|
||||||
if "category" not in device_config:
|
if "category" not in device_config:
|
||||||
@@ -717,10 +769,7 @@ class Registry:
|
|||||||
if "init_param_schema" not in device_config:
|
if "init_param_schema" not in device_config:
|
||||||
device_config["init_param_schema"] = {}
|
device_config["init_param_schema"] = {}
|
||||||
if "class" in device_config:
|
if "class" in device_config:
|
||||||
if (
|
if "status_types" not in device_config["class"] or device_config["class"]["status_types"] is None:
|
||||||
"status_types" not in device_config["class"]
|
|
||||||
or device_config["class"]["status_types"] is None
|
|
||||||
):
|
|
||||||
device_config["class"]["status_types"] = {}
|
device_config["class"]["status_types"] = {}
|
||||||
if (
|
if (
|
||||||
"action_value_mappings" not in device_config["class"]
|
"action_value_mappings" not in device_config["class"]
|
||||||
@@ -738,25 +787,17 @@ class Registry:
|
|||||||
)
|
)
|
||||||
for status_name, status_type in device_config["class"]["status_types"].items():
|
for status_name, status_type in device_config["class"]["status_types"].items():
|
||||||
if isinstance(status_type, tuple) or status_type in ["Any", "None", "Unknown"]:
|
if isinstance(status_type, tuple) or status_type in ["Any", "None", "Unknown"]:
|
||||||
status_type = "String" # 替换成ROS的String,便于显示
|
status_type = "String"
|
||||||
device_config["class"]["status_types"][status_name] = status_type
|
device_config["class"]["status_types"][status_name] = status_type
|
||||||
try:
|
try:
|
||||||
target_type = self._replace_type_with_class(
|
target_type = self._replace_type_with_class(status_type, device_id, f"状态 {status_name}")
|
||||||
status_type, device_id, f"状态 {status_name}"
|
|
||||||
)
|
|
||||||
except ROSMsgNotFound:
|
except ROSMsgNotFound:
|
||||||
continue
|
continue
|
||||||
if target_type in [
|
if target_type in [dict, list]:
|
||||||
dict,
|
|
||||||
list,
|
|
||||||
]: # 对于嵌套类型返回的对象,暂时处理成字符串,无法直接进行转换
|
|
||||||
target_type = String
|
target_type = String
|
||||||
status_str_type_mapping[status_type] = target_type
|
status_str_type_mapping[status_type] = target_type
|
||||||
device_config["class"]["status_types"] = dict(
|
device_config["class"]["status_types"] = dict(sorted(device_config["class"]["status_types"].items()))
|
||||||
sorted(device_config["class"]["status_types"].items())
|
|
||||||
)
|
|
||||||
if complete_registry:
|
if complete_registry:
|
||||||
# 保存原有的 action 配置(用于保留 schema 的 description 和 handles 等)
|
|
||||||
old_action_configs = {}
|
old_action_configs = {}
|
||||||
for action_name, action_config in device_config["class"]["action_value_mappings"].items():
|
for action_name, action_config in device_config["class"]["action_value_mappings"].items():
|
||||||
old_action_configs[action_name] = action_config
|
old_action_configs[action_name] = action_config
|
||||||
@@ -766,7 +807,6 @@ class Registry:
|
|||||||
for k, v in device_config["class"]["action_value_mappings"].items()
|
for k, v in device_config["class"]["action_value_mappings"].items()
|
||||||
if not k.startswith("auto-")
|
if not k.startswith("auto-")
|
||||||
}
|
}
|
||||||
# 处理动作值映射
|
|
||||||
device_config["class"]["action_value_mappings"].update(
|
device_config["class"]["action_value_mappings"].update(
|
||||||
{
|
{
|
||||||
f"auto-{k}": {
|
f"auto-{k}": {
|
||||||
@@ -778,18 +818,15 @@ class Registry:
|
|||||||
v["args"],
|
v["args"],
|
||||||
k,
|
k,
|
||||||
v.get("return_annotation"),
|
v.get("return_annotation"),
|
||||||
# 传入旧的 schema 以保留字段 description
|
|
||||||
old_action_configs.get(f"auto-{k}", {}).get("schema"),
|
old_action_configs.get(f"auto-{k}", {}).get("schema"),
|
||||||
),
|
),
|
||||||
"goal_default": {i["name"]: i["default"] for i in v["args"]},
|
"goal_default": {i["name"]: i["default"] for i in v["args"]},
|
||||||
# 保留原有的 handles 配置
|
|
||||||
"handles": old_action_configs.get(f"auto-{k}", {}).get("handles", []),
|
"handles": old_action_configs.get(f"auto-{k}", {}).get("handles", []),
|
||||||
"placeholder_keys": {
|
"placeholder_keys": {
|
||||||
i["name"]: (
|
i["name"]: (
|
||||||
"unilabos_resources"
|
"unilabos_resources"
|
||||||
if i["type"] == "unilabos.registry.placeholder_type:ResourceSlot"
|
if i["type"] == "unilabos.registry.placeholder_type:ResourceSlot"
|
||||||
or i["type"]
|
or i["type"] == ("list", "unilabos.registry.placeholder_type:ResourceSlot")
|
||||||
== ("list", "unilabos.registry.placeholder_type:ResourceSlot")
|
|
||||||
else "unilabos_devices"
|
else "unilabos_devices"
|
||||||
)
|
)
|
||||||
for i in v["args"]
|
for i in v["args"]
|
||||||
@@ -802,14 +839,12 @@ class Registry:
|
|||||||
]
|
]
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
# 不生成已配置action的动作
|
|
||||||
for k, v in enhanced_info["action_methods"].items()
|
for k, v in enhanced_info["action_methods"].items()
|
||||||
if k not in device_config["class"]["action_value_mappings"]
|
if k not in device_config["class"]["action_value_mappings"]
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
# 恢复原有的 description 信息(非 auto- 开头的动作)
|
|
||||||
for action_name, old_config in old_action_configs.items():
|
for action_name, old_config in old_action_configs.items():
|
||||||
if action_name in device_config["class"]["action_value_mappings"]: # 有一些会被删除
|
if action_name in device_config["class"]["action_value_mappings"]:
|
||||||
old_schema = old_config.get("schema", {})
|
old_schema = old_config.get("schema", {})
|
||||||
if "description" in old_schema and old_schema["description"]:
|
if "description" in old_schema and old_schema["description"]:
|
||||||
device_config["class"]["action_value_mappings"][action_name]["schema"][
|
device_config["class"]["action_value_mappings"][action_name]["schema"][
|
||||||
@@ -838,7 +873,6 @@ class Registry:
|
|||||||
action_config["handles"] = {}
|
action_config["handles"] = {}
|
||||||
if "type" in action_config:
|
if "type" in action_config:
|
||||||
action_type_str: str = action_config["type"]
|
action_type_str: str = action_config["type"]
|
||||||
# 通过Json发放指令,而不是通过特殊的ros action进行处理
|
|
||||||
if not action_type_str.startswith("UniLabJsonCommand"):
|
if not action_type_str.startswith("UniLabJsonCommand"):
|
||||||
try:
|
try:
|
||||||
target_type = self._replace_type_with_class(
|
target_type = self._replace_type_with_class(
|
||||||
@@ -856,31 +890,78 @@ class Registry:
|
|||||||
logger.warning(
|
logger.warning(
|
||||||
f"[UniLab Registry] 设备 {device_id} 的动作 {action_name} 类型为空,跳过替换"
|
f"[UniLab Registry] 设备 {device_id} 的动作 {action_name} 类型为空,跳过替换"
|
||||||
)
|
)
|
||||||
complete_data[device_id] = copy.deepcopy(dict(sorted(device_config.items()))) # 稍后dump到文件
|
complete_data[device_id] = copy.deepcopy(dict(sorted(device_config.items())))
|
||||||
for status_name, status_type in device_config["class"]["status_types"].items():
|
for status_name, status_type in device_config["class"]["status_types"].items():
|
||||||
device_config["class"]["status_types"][status_name] = status_str_type_mapping[status_type]
|
device_config["class"]["status_types"][status_name] = status_str_type_mapping[status_type]
|
||||||
for action_name, action_config in device_config["class"]["action_value_mappings"].items():
|
for action_name, action_config in device_config["class"]["action_value_mappings"].items():
|
||||||
if action_config["type"] not in action_str_type_mapping:
|
if action_config["type"] not in action_str_type_mapping:
|
||||||
continue
|
continue
|
||||||
action_config["type"] = action_str_type_mapping[action_config["type"]]
|
action_config["type"] = action_str_type_mapping[action_config["type"]]
|
||||||
# 添加内置的驱动命令动作
|
|
||||||
self._add_builtin_actions(device_config, device_id)
|
self._add_builtin_actions(device_config, device_id)
|
||||||
device_config["file_path"] = str(file.absolute()).replace("\\", "/")
|
device_config["file_path"] = str(file.absolute()).replace("\\", "/")
|
||||||
device_config["registry_type"] = "device"
|
device_config["registry_type"] = "device"
|
||||||
logger.trace( # type: ignore
|
device_ids.append(device_id)
|
||||||
f"[UniLab Registry] Device-{current_device_number} File-{i+1}/{len(files)} Add {device_id} "
|
|
||||||
|
complete_data = dict(sorted(complete_data.items()))
|
||||||
|
complete_data = copy.deepcopy(complete_data)
|
||||||
|
try:
|
||||||
|
with open(file, "w", encoding="utf-8") as f:
|
||||||
|
yaml.dump(complete_data, f, allow_unicode=True, default_flow_style=False, Dumper=NoAliasDumper)
|
||||||
|
except Exception as e:
|
||||||
|
logger.warning(f"[UniLab Registry] 写入设备文件失败: {file}, 错误: {e}")
|
||||||
|
|
||||||
|
return data, complete_data, True, device_ids
|
||||||
|
|
||||||
|
def load_device_types(self, path: os.PathLike, complete_registry: bool):
|
||||||
|
abs_path = Path(path).absolute()
|
||||||
|
devices_path = abs_path / "devices"
|
||||||
|
device_comms_path = abs_path / "device_comms"
|
||||||
|
files = list(devices_path.glob("*.yaml")) + list(device_comms_path.glob("*.yaml"))
|
||||||
|
logger.trace(
|
||||||
|
f"[UniLab Registry] devices: {devices_path.exists()}, device_comms: {device_comms_path.exists()}, "
|
||||||
|
+ f"total: {len(files)}"
|
||||||
|
)
|
||||||
|
|
||||||
|
if not files:
|
||||||
|
return
|
||||||
|
|
||||||
|
from unilabos.app.web.utils.action_utils import get_yaml_from_goal_type
|
||||||
|
|
||||||
|
# 使用线程池并行加载
|
||||||
|
max_workers = min(8, len(files))
|
||||||
|
results = []
|
||||||
|
|
||||||
|
with ThreadPoolExecutor(max_workers=max_workers) as executor:
|
||||||
|
future_to_file = {
|
||||||
|
executor.submit(self._load_single_device_file, file, complete_registry, get_yaml_from_goal_type): file
|
||||||
|
for file in files
|
||||||
|
}
|
||||||
|
for future in as_completed(future_to_file):
|
||||||
|
file = future_to_file[future]
|
||||||
|
try:
|
||||||
|
data, complete_data, is_valid, device_ids = future.result()
|
||||||
|
if is_valid:
|
||||||
|
results.append((file, data, device_ids))
|
||||||
|
except Exception as e:
|
||||||
|
logger.warning(f"[UniLab Registry] 处理设备文件异常: {file}, 错误: {e}")
|
||||||
|
|
||||||
|
# 线程安全地更新注册表
|
||||||
|
current_device_number = len(self.device_type_registry) + 1
|
||||||
|
with self._registry_lock:
|
||||||
|
for file, data, device_ids in results:
|
||||||
|
self.device_type_registry.update(data)
|
||||||
|
for device_id in device_ids:
|
||||||
|
logger.trace(
|
||||||
|
f"[UniLab Registry] Device-{current_device_number} Add {device_id} "
|
||||||
+ f"[{data[device_id].get('name', '未命名设备')}]"
|
+ f"[{data[device_id].get('name', '未命名设备')}]"
|
||||||
)
|
)
|
||||||
current_device_number += 1
|
current_device_number += 1
|
||||||
complete_data = dict(sorted(complete_data.items()))
|
|
||||||
complete_data = copy.deepcopy(complete_data)
|
# 记录无效文件
|
||||||
with open(file, "w", encoding="utf-8") as f:
|
valid_files = {r[0] for r in results}
|
||||||
yaml.dump(complete_data, f, allow_unicode=True, default_flow_style=False, Dumper=NoAliasDumper)
|
for file in files:
|
||||||
self.device_type_registry.update(data)
|
if file not in valid_files:
|
||||||
else:
|
logger.debug(f"[UniLab Registry] Device File Not Valid YAML File: {file.absolute()}")
|
||||||
logger.debug(
|
|
||||||
f"[UniLab Registry] Device File-{i+1}/{len(files)} Not Valid YAML File: {file.absolute()}"
|
|
||||||
)
|
|
||||||
|
|
||||||
def obtain_registry_device_info(self):
|
def obtain_registry_device_info(self):
|
||||||
devices = []
|
devices = []
|
||||||
|
|||||||
@@ -46,3 +46,16 @@ BIOYOND_PolymerStation_8StockCarrier:
|
|||||||
init_param_schema: {}
|
init_param_schema: {}
|
||||||
registry_type: resource
|
registry_type: resource
|
||||||
version: 1.0.0
|
version: 1.0.0
|
||||||
|
BIOYOND_PolymerStation_TipBox:
|
||||||
|
category:
|
||||||
|
- bottle_carriers
|
||||||
|
- tip_racks
|
||||||
|
class:
|
||||||
|
module: unilabos.resources.bioyond.bottle_carriers:BIOYOND_PolymerStation_TipBox
|
||||||
|
type: pylabrobot
|
||||||
|
description: BIOYOND_PolymerStation_TipBox (4x6布局,24个枪头孔位)
|
||||||
|
handles: []
|
||||||
|
icon: ''
|
||||||
|
init_param_schema: {}
|
||||||
|
registry_type: resource
|
||||||
|
version: 1.0.0
|
||||||
|
|||||||
@@ -82,14 +82,3 @@ BIOYOND_PolymerStation_Solution_Beaker:
|
|||||||
icon: ''
|
icon: ''
|
||||||
init_param_schema: {}
|
init_param_schema: {}
|
||||||
version: 1.0.0
|
version: 1.0.0
|
||||||
BIOYOND_PolymerStation_TipBox:
|
|
||||||
category:
|
|
||||||
- bottles
|
|
||||||
- tip_boxes
|
|
||||||
class:
|
|
||||||
module: unilabos.resources.bioyond.bottles:BIOYOND_PolymerStation_TipBox
|
|
||||||
type: pylabrobot
|
|
||||||
handles: []
|
|
||||||
icon: ''
|
|
||||||
init_param_schema: {}
|
|
||||||
version: 1.0.0
|
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
from pylabrobot.resources import create_homogeneous_resources, Coordinate, ResourceHolder, create_ordered_items_2d
|
from pylabrobot.resources import create_homogeneous_resources, Coordinate, ResourceHolder, create_ordered_items_2d, Container
|
||||||
|
|
||||||
from unilabos.resources.itemized_carrier import BottleCarrier
|
from unilabos.resources.itemized_carrier import BottleCarrier
|
||||||
from unilabos.resources.bioyond.bottles import (
|
from unilabos.resources.bioyond.bottles import (
|
||||||
@@ -9,6 +9,28 @@ from unilabos.resources.bioyond.bottles import (
|
|||||||
BIOYOND_PolymerStation_Reagent_Bottle,
|
BIOYOND_PolymerStation_Reagent_Bottle,
|
||||||
BIOYOND_PolymerStation_Flask,
|
BIOYOND_PolymerStation_Flask,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def BIOYOND_PolymerStation_Tip(name: str, size_x: float = 8.0, size_y: float = 8.0, size_z: float = 50.0) -> Container:
|
||||||
|
"""创建单个枪头资源
|
||||||
|
|
||||||
|
Args:
|
||||||
|
name: 枪头名称
|
||||||
|
size_x: 枪头宽度 (mm)
|
||||||
|
size_y: 枪头长度 (mm)
|
||||||
|
size_z: 枪头高度 (mm)
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Container: 枪头容器
|
||||||
|
"""
|
||||||
|
return Container(
|
||||||
|
name=name,
|
||||||
|
size_x=size_x,
|
||||||
|
size_y=size_y,
|
||||||
|
size_z=size_z,
|
||||||
|
category="tip",
|
||||||
|
model="BIOYOND_PolymerStation_Tip",
|
||||||
|
)
|
||||||
# 命名约定:试剂瓶-Bottle,烧杯-Beaker,烧瓶-Flask,小瓶-Vial
|
# 命名约定:试剂瓶-Bottle,烧杯-Beaker,烧瓶-Flask,小瓶-Vial
|
||||||
|
|
||||||
|
|
||||||
@@ -322,3 +344,88 @@ def BIOYOND_Electrolyte_1BottleCarrier(name: str) -> BottleCarrier:
|
|||||||
carrier.num_items_z = 1
|
carrier.num_items_z = 1
|
||||||
carrier[0] = BIOYOND_PolymerStation_Solution_Beaker(f"{name}_beaker_1")
|
carrier[0] = BIOYOND_PolymerStation_Solution_Beaker(f"{name}_beaker_1")
|
||||||
return carrier
|
return carrier
|
||||||
|
|
||||||
|
|
||||||
|
def BIOYOND_PolymerStation_TipBox(
|
||||||
|
name: str,
|
||||||
|
size_x: float = 127.76, # 枪头盒宽度
|
||||||
|
size_y: float = 85.48, # 枪头盒长度
|
||||||
|
size_z: float = 100.0, # 枪头盒高度
|
||||||
|
barcode: str = None,
|
||||||
|
) -> BottleCarrier:
|
||||||
|
"""创建4×6枪头盒 (24个枪头) - 使用 BottleCarrier 结构
|
||||||
|
|
||||||
|
Args:
|
||||||
|
name: 枪头盒名称
|
||||||
|
size_x: 枪头盒宽度 (mm)
|
||||||
|
size_y: 枪头盒长度 (mm)
|
||||||
|
size_z: 枪头盒高度 (mm)
|
||||||
|
barcode: 条形码
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
BottleCarrier: 包含24个枪头孔位的枪头盒载架
|
||||||
|
|
||||||
|
布局说明:
|
||||||
|
- 4行×6列 (A-D, 1-6)
|
||||||
|
- 枪头孔位间距: 18mm (x方向) × 18mm (y方向)
|
||||||
|
- 起始位置居中对齐
|
||||||
|
- 索引顺序: 列优先 (0=A1, 1=B1, 2=C1, 3=D1, 4=A2, ...)
|
||||||
|
"""
|
||||||
|
# 枪头孔位参数
|
||||||
|
num_cols = 6 # 1-6 (x方向)
|
||||||
|
num_rows = 4 # A-D (y方向)
|
||||||
|
tip_diameter = 8.0 # 枪头孔位直径
|
||||||
|
tip_spacing_x = 18.0 # 列间距 (增加到18mm,更宽松)
|
||||||
|
tip_spacing_y = 18.0 # 行间距 (增加到18mm,更宽松)
|
||||||
|
|
||||||
|
# 计算起始位置 (居中对齐)
|
||||||
|
total_width = (num_cols - 1) * tip_spacing_x + tip_diameter
|
||||||
|
total_height = (num_rows - 1) * tip_spacing_y + tip_diameter
|
||||||
|
start_x = (size_x - total_width) / 2
|
||||||
|
start_y = (size_y - total_height) / 2
|
||||||
|
|
||||||
|
# 使用 create_ordered_items_2d 创建孔位
|
||||||
|
# create_ordered_items_2d 返回的 key 是数字索引: 0, 1, 2, ...
|
||||||
|
# 顺序是列优先: 先y后x (即 0=A1, 1=B1, 2=C1, 3=D1, 4=A2, 5=B2, ...)
|
||||||
|
sites = create_ordered_items_2d(
|
||||||
|
klass=ResourceHolder,
|
||||||
|
num_items_x=num_cols,
|
||||||
|
num_items_y=num_rows,
|
||||||
|
dx=start_x,
|
||||||
|
dy=start_y,
|
||||||
|
dz=5.0,
|
||||||
|
item_dx=tip_spacing_x,
|
||||||
|
item_dy=tip_spacing_y,
|
||||||
|
size_x=tip_diameter,
|
||||||
|
size_y=tip_diameter,
|
||||||
|
size_z=50.0, # 枪头深度
|
||||||
|
)
|
||||||
|
|
||||||
|
# 更新 sites 中每个 ResourceHolder 的名称
|
||||||
|
for k, v in sites.items():
|
||||||
|
v.name = f"{name}_{v.name}"
|
||||||
|
|
||||||
|
# 创建枪头盒载架
|
||||||
|
# 注意:不设置 category,使用默认的 "bottle_carrier",这样前端会显示为完整的矩形载架
|
||||||
|
tip_box = BottleCarrier(
|
||||||
|
name=name,
|
||||||
|
size_x=size_x,
|
||||||
|
size_y=size_y,
|
||||||
|
size_z=size_z,
|
||||||
|
sites=sites, # 直接使用数字索引的 sites
|
||||||
|
model="BIOYOND_PolymerStation_TipBox",
|
||||||
|
)
|
||||||
|
|
||||||
|
# 设置自定义属性
|
||||||
|
tip_box.barcode = barcode
|
||||||
|
tip_box.tip_count = 24 # 4行×6列
|
||||||
|
tip_box.num_items_x = num_cols
|
||||||
|
tip_box.num_items_y = num_rows
|
||||||
|
tip_box.num_items_z = 1
|
||||||
|
|
||||||
|
# ⭐ 枪头盒不需要放入子资源
|
||||||
|
# 与其他 carrier 不同,枪头盒在 Bioyond 中是一个整体
|
||||||
|
# 不需要追踪每个枪头的状态,保持为空的 ResourceHolder 即可
|
||||||
|
# 这样前端会显示24个空槽位,可以用于放置枪头
|
||||||
|
|
||||||
|
return tip_box
|
||||||
|
|||||||
@@ -116,7 +116,9 @@ def BIOYOND_PolymerStation_TipBox(
|
|||||||
size_z: float = 100.0, # 枪头盒高度
|
size_z: float = 100.0, # 枪头盒高度
|
||||||
barcode: str = None,
|
barcode: str = None,
|
||||||
):
|
):
|
||||||
"""创建4×6枪头盒 (24个枪头)
|
"""创建4×6枪头盒 (24个枪头) - 使用 BottleCarrier 结构
|
||||||
|
|
||||||
|
注意:此函数已弃用,请使用 bottle_carriers.py 中的版本
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
name: 枪头盒名称
|
name: 枪头盒名称
|
||||||
@@ -126,55 +128,11 @@ def BIOYOND_PolymerStation_TipBox(
|
|||||||
barcode: 条形码
|
barcode: 条形码
|
||||||
|
|
||||||
Returns:
|
Returns:
|
||||||
TipBoxCarrier: 包含24个枪头孔位的枪头盒
|
BottleCarrier: 包含24个枪头孔位的枪头盒载架
|
||||||
"""
|
"""
|
||||||
from pylabrobot.resources import Container, Coordinate
|
# 重定向到 bottle_carriers.py 中的实现
|
||||||
|
from unilabos.resources.bioyond.bottle_carriers import BIOYOND_PolymerStation_TipBox as TipBox_Carrier
|
||||||
# 创建枪头盒容器
|
return TipBox_Carrier(name=name, size_x=size_x, size_y=size_y, size_z=size_z, barcode=barcode)
|
||||||
tip_box = Container(
|
|
||||||
name=name,
|
|
||||||
size_x=size_x,
|
|
||||||
size_y=size_y,
|
|
||||||
size_z=size_z,
|
|
||||||
category="tip_rack",
|
|
||||||
model="BIOYOND_PolymerStation_TipBox_4x6",
|
|
||||||
)
|
|
||||||
|
|
||||||
# 设置自定义属性
|
|
||||||
tip_box.barcode = barcode
|
|
||||||
tip_box.tip_count = 24 # 4行×6列
|
|
||||||
tip_box.num_items_x = 6 # 6列
|
|
||||||
tip_box.num_items_y = 4 # 4行
|
|
||||||
|
|
||||||
# 创建24个枪头孔位 (4行×6列)
|
|
||||||
# 假设孔位间距为 9mm
|
|
||||||
tip_spacing_x = 9.0 # 列间距
|
|
||||||
tip_spacing_y = 9.0 # 行间距
|
|
||||||
start_x = 14.38 # 第一个孔位的x偏移
|
|
||||||
start_y = 11.24 # 第一个孔位的y偏移
|
|
||||||
|
|
||||||
for row in range(4): # A, B, C, D
|
|
||||||
for col in range(6): # 1-6
|
|
||||||
spot_name = f"{chr(65 + row)}{col + 1}" # A1, A2, ..., D6
|
|
||||||
x = start_x + col * tip_spacing_x
|
|
||||||
y = start_y + row * tip_spacing_y
|
|
||||||
|
|
||||||
# 创建枪头孔位容器
|
|
||||||
tip_spot = Container(
|
|
||||||
name=spot_name,
|
|
||||||
size_x=8.0, # 单个枪头孔位大小
|
|
||||||
size_y=8.0,
|
|
||||||
size_z=size_z - 10.0, # 略低于盒子高度
|
|
||||||
category="tip_spot",
|
|
||||||
)
|
|
||||||
|
|
||||||
# 添加到枪头盒
|
|
||||||
tip_box.assign_child_resource(
|
|
||||||
tip_spot,
|
|
||||||
location=Coordinate(x=x, y=y, z=0)
|
|
||||||
)
|
|
||||||
|
|
||||||
return tip_box
|
|
||||||
|
|
||||||
|
|
||||||
def BIOYOND_PolymerStation_Flask(
|
def BIOYOND_PolymerStation_Flask(
|
||||||
|
|||||||
@@ -42,7 +42,7 @@ def canonicalize_nodes_data(
|
|||||||
Returns:
|
Returns:
|
||||||
ResourceTreeSet: 标准化后的资源树集合
|
ResourceTreeSet: 标准化后的资源树集合
|
||||||
"""
|
"""
|
||||||
print_status(f"{len(nodes)} Resources loaded:", "info")
|
print_status(f"{len(nodes)} Resources loaded", "info")
|
||||||
|
|
||||||
# 第一步:基本预处理(处理graphml的label字段)
|
# 第一步:基本预处理(处理graphml的label字段)
|
||||||
outer_host_node_id = None
|
outer_host_node_id = None
|
||||||
@@ -759,6 +759,9 @@ def resource_bioyond_to_plr(bioyond_materials: list[dict], type_mapping: Dict[st
|
|||||||
bottle = plr_material[number] = initialize_resource(
|
bottle = plr_material[number] = initialize_resource(
|
||||||
{"name": f'{detail["name"]}_{number}', "class": reverse_type_mapping[typeName][0]}, resource_type=ResourcePLR
|
{"name": f'{detail["name"]}_{number}', "class": reverse_type_mapping[typeName][0]}, resource_type=ResourcePLR
|
||||||
)
|
)
|
||||||
|
# 只有具有 tracker 属性的容器才设置液体信息(如 Bottle, Well)
|
||||||
|
# ResourceHolder 等不支持液体追踪的容器跳过
|
||||||
|
if hasattr(bottle, "tracker"):
|
||||||
bottle.tracker.liquids = [
|
bottle.tracker.liquids = [
|
||||||
(detail["name"], float(detail.get("quantity", 0)) if detail.get("quantity") else 0)
|
(detail["name"], float(detail.get("quantity", 0)) if detail.get("quantity") else 0)
|
||||||
]
|
]
|
||||||
@@ -770,6 +773,8 @@ def resource_bioyond_to_plr(bioyond_materials: list[dict], type_mapping: Dict[st
|
|||||||
# 只对有 capacity 属性的容器(液体容器)处理液体追踪
|
# 只对有 capacity 属性的容器(液体容器)处理液体追踪
|
||||||
if hasattr(plr_material, 'capacity'):
|
if hasattr(plr_material, 'capacity'):
|
||||||
bottle = plr_material[0] if plr_material.capacity > 0 else plr_material
|
bottle = plr_material[0] if plr_material.capacity > 0 else plr_material
|
||||||
|
# 确保 bottle 有 tracker 属性才设置液体信息
|
||||||
|
if hasattr(bottle, "tracker"):
|
||||||
bottle.tracker.liquids = [
|
bottle.tracker.liquids = [
|
||||||
(material["name"], float(material.get("quantity", 0)) if material.get("quantity") else 0)
|
(material["name"], float(material.get("quantity", 0)) if material.get("quantity") else 0)
|
||||||
]
|
]
|
||||||
@@ -801,24 +806,29 @@ def resource_bioyond_to_plr(bioyond_materials: list[dict], type_mapping: Dict[st
|
|||||||
wh_name = loc.get("whName")
|
wh_name = loc.get("whName")
|
||||||
logger.debug(f"[物料位置] {unique_name} 尝试放置到 warehouse: {wh_name} (Bioyond坐标: x={loc.get('x')}, y={loc.get('y')}, z={loc.get('z')})")
|
logger.debug(f"[物料位置] {unique_name} 尝试放置到 warehouse: {wh_name} (Bioyond坐标: x={loc.get('x')}, y={loc.get('y')}, z={loc.get('z')})")
|
||||||
|
|
||||||
|
# Bioyond坐标映射 (重要!): x→行(1=A,2=B...), y→列(1=01,2=02...), z→层(通常=1)
|
||||||
|
# 必须在warehouse映射之前先获取坐标,以便后续调整
|
||||||
|
x = loc.get("x", 1) # 行号 (1-based: 1=A, 2=B, 3=C, 4=D)
|
||||||
|
y = loc.get("y", 1) # 列号 (1-based: 1=01, 2=02, 3=03...)
|
||||||
|
z = loc.get("z", 1) # 层号 (1-based, 通常为1)
|
||||||
|
|
||||||
# 特殊处理: Bioyond的"堆栈1"需要映射到"堆栈1左"或"堆栈1右"
|
# 特殊处理: Bioyond的"堆栈1"需要映射到"堆栈1左"或"堆栈1右"
|
||||||
# 根据列号(x)判断: 1-4映射到左侧, 5-8映射到右侧
|
# 根据列号(y)判断: 1-4映射到左侧, 5-8映射到右侧
|
||||||
if wh_name == "堆栈1":
|
if wh_name == "堆栈1":
|
||||||
x_val = loc.get("x", 1)
|
if 1 <= y <= 4:
|
||||||
if 1 <= x_val <= 4:
|
|
||||||
wh_name = "堆栈1左"
|
wh_name = "堆栈1左"
|
||||||
elif 5 <= x_val <= 8:
|
elif 5 <= y <= 8:
|
||||||
wh_name = "堆栈1右"
|
wh_name = "堆栈1右"
|
||||||
|
y = y - 4 # 调整列号: 5-8映射到1-4
|
||||||
else:
|
else:
|
||||||
logger.warning(f"物料 {material['name']} 的列号 x={x_val} 超出范围,无法映射到堆栈1左或堆栈1右")
|
logger.warning(f"物料 {material['name']} 的列号 y={y} 超出范围,无法映射到堆栈1左或堆栈1右")
|
||||||
continue
|
continue
|
||||||
|
|
||||||
# 特殊处理: Bioyond的"站内Tip盒堆栈"也需要进行拆分映射
|
# 特殊处理: Bioyond的"站内Tip盒堆栈"也需要进行拆分映射
|
||||||
if wh_name == "站内Tip盒堆栈":
|
if wh_name == "站内Tip盒堆栈":
|
||||||
y_val = loc.get("y", 1)
|
if y == 1:
|
||||||
if y_val == 1:
|
|
||||||
wh_name = "站内Tip盒堆栈(右)"
|
wh_name = "站内Tip盒堆栈(右)"
|
||||||
elif y_val in [2, 3]:
|
elif y in [2, 3]:
|
||||||
wh_name = "站内Tip盒堆栈(左)"
|
wh_name = "站内Tip盒堆栈(左)"
|
||||||
y = y - 1 # 调整列号,因为左侧仓库对应的 Bioyond y=2 实际上是它的第1列
|
y = y - 1 # 调整列号,因为左侧仓库对应的 Bioyond y=2 实际上是它的第1列
|
||||||
|
|
||||||
@@ -826,15 +836,6 @@ def resource_bioyond_to_plr(bioyond_materials: list[dict], type_mapping: Dict[st
|
|||||||
warehouse = deck.warehouses[wh_name]
|
warehouse = deck.warehouses[wh_name]
|
||||||
logger.debug(f"[Warehouse匹配] 找到warehouse: {wh_name} (容量: {warehouse.capacity}, 行×列: {warehouse.num_items_x}×{warehouse.num_items_y})")
|
logger.debug(f"[Warehouse匹配] 找到warehouse: {wh_name} (容量: {warehouse.capacity}, 行×列: {warehouse.num_items_x}×{warehouse.num_items_y})")
|
||||||
|
|
||||||
# Bioyond坐标映射 (重要!): x→行(1=A,2=B...), y→列(1=01,2=02...), z→层(通常=1)
|
|
||||||
x = loc.get("x", 1) # 行号 (1-based: 1=A, 2=B, 3=C, 4=D)
|
|
||||||
y = loc.get("y", 1) # 列号 (1-based: 1=01, 2=02, 3=03...)
|
|
||||||
z = loc.get("z", 1) # 层号 (1-based, 通常为1)
|
|
||||||
|
|
||||||
# 如果是右侧堆栈,需要调整列号 (5→1, 6→2, 7→3, 8→4)
|
|
||||||
if wh_name == "堆栈1右":
|
|
||||||
y = y - 4 # 将5-8映射到1-4
|
|
||||||
|
|
||||||
# 特殊处理竖向warehouse(站内试剂存放堆栈、测量小瓶仓库)
|
# 特殊处理竖向warehouse(站内试剂存放堆栈、测量小瓶仓库)
|
||||||
# 这些warehouse使用 vertical-col-major 布局
|
# 这些warehouse使用 vertical-col-major 布局
|
||||||
if wh_name in ["站内试剂存放堆栈", "测量小瓶仓库(测密度)"]:
|
if wh_name in ["站内试剂存放堆栈", "测量小瓶仓库(测密度)"]:
|
||||||
|
|||||||
@@ -18,3 +18,9 @@ def register():
|
|||||||
from unilabos.devices.liquid_handling.rviz_backend import UniLiquidHandlerRvizBackend
|
from unilabos.devices.liquid_handling.rviz_backend import UniLiquidHandlerRvizBackend
|
||||||
from unilabos.devices.liquid_handling.laiyu.backend.laiyu_v_backend import UniLiquidHandlerLaiyuBackend
|
from unilabos.devices.liquid_handling.laiyu.backend.laiyu_v_backend import UniLiquidHandlerLaiyuBackend
|
||||||
|
|
||||||
|
# noinspection PyUnresolvedReferences
|
||||||
|
from unilabos.resources.bioyond.decks import (
|
||||||
|
BIOYOND_PolymerReactionStation_Deck,
|
||||||
|
BIOYOND_PolymerPreparationStation_Deck,
|
||||||
|
BIOYOND_YB_Deck,
|
||||||
|
)
|
||||||
|
|||||||
@@ -69,8 +69,8 @@ class ResourceDict(BaseModel):
|
|||||||
klass: str = Field(alias="class", description="Resource class name")
|
klass: str = Field(alias="class", description="Resource class name")
|
||||||
pose: ResourceDictPosition = Field(description="Resource position", default_factory=ResourceDictPosition)
|
pose: ResourceDictPosition = Field(description="Resource position", default_factory=ResourceDictPosition)
|
||||||
config: Dict[str, Any] = Field(description="Resource configuration")
|
config: Dict[str, Any] = Field(description="Resource configuration")
|
||||||
data: Dict[str, Any] = Field(description="Resource data")
|
data: Dict[str, Any] = Field(description="Resource data, eg: container liquid data")
|
||||||
extra: Dict[str, Any] = Field(description="Extra data")
|
extra: Dict[str, Any] = Field(description="Extra data, eg: slot index")
|
||||||
|
|
||||||
@field_serializer("parent_uuid")
|
@field_serializer("parent_uuid")
|
||||||
def _serialize_parent(self, parent_uuid: Optional["ResourceDict"]):
|
def _serialize_parent(self, parent_uuid: Optional["ResourceDict"]):
|
||||||
@@ -341,6 +341,7 @@ class ResourceTreeSet(object):
|
|||||||
"deck": "deck",
|
"deck": "deck",
|
||||||
"tip_rack": "tip_rack",
|
"tip_rack": "tip_rack",
|
||||||
"tip_spot": "tip_spot",
|
"tip_spot": "tip_spot",
|
||||||
|
"tip": "tip", # 添加 tip 类型支持
|
||||||
"tube": "tube",
|
"tube": "tube",
|
||||||
"bottle_carrier": "bottle_carrier",
|
"bottle_carrier": "bottle_carrier",
|
||||||
}
|
}
|
||||||
|
|||||||
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": []
|
||||||
|
}
|
||||||
0
unilabos/workflow/__init__.py
Normal file
0
unilabos/workflow/__init__.py
Normal file
241
unilabos/workflow/from_python_script.py
Normal file
241
unilabos/workflow/from_python_script.py
Normal file
@@ -0,0 +1,241 @@
|
|||||||
|
import ast
|
||||||
|
import json
|
||||||
|
from typing import Dict, List, Any, Tuple, Optional
|
||||||
|
|
||||||
|
from .common import WorkflowGraph, RegistryAdapter
|
||||||
|
|
||||||
|
Json = Dict[str, Any]
|
||||||
|
|
||||||
|
# ---------------- Converter ----------------
|
||||||
|
|
||||||
|
class DeviceMethodConverter:
|
||||||
|
"""
|
||||||
|
- 字段统一:resource_name(原 device_class)、template_name(原 action_key)
|
||||||
|
- params 单层;inputs 使用 'params.' 前缀
|
||||||
|
- SimpleGraph.add_workflow_node 负责变量连线与边
|
||||||
|
"""
|
||||||
|
def __init__(self, device_registry: Optional[Dict[str, Any]] = None):
|
||||||
|
self.graph = WorkflowGraph()
|
||||||
|
self.variable_sources: Dict[str, Dict[str, Any]] = {} # var -> {node_id, output_name}
|
||||||
|
self.instance_to_resource: Dict[str, Optional[str]] = {} # 实例名 -> resource_name
|
||||||
|
self.node_id_counter: int = 0
|
||||||
|
self.registry = RegistryAdapter(device_registry or {})
|
||||||
|
|
||||||
|
# ---- helpers ----
|
||||||
|
def _new_node_id(self) -> int:
|
||||||
|
nid = self.node_id_counter
|
||||||
|
self.node_id_counter += 1
|
||||||
|
return nid
|
||||||
|
|
||||||
|
def _assign_targets(self, targets) -> List[str]:
|
||||||
|
names: List[str] = []
|
||||||
|
import ast
|
||||||
|
if isinstance(targets, ast.Tuple):
|
||||||
|
for elt in targets.elts:
|
||||||
|
if isinstance(elt, ast.Name):
|
||||||
|
names.append(elt.id)
|
||||||
|
elif isinstance(targets, ast.Name):
|
||||||
|
names.append(targets.id)
|
||||||
|
return names
|
||||||
|
|
||||||
|
def _extract_device_instantiation(self, node) -> Optional[Tuple[str, str]]:
|
||||||
|
import ast
|
||||||
|
if not isinstance(node.value, ast.Call):
|
||||||
|
return None
|
||||||
|
callee = node.value.func
|
||||||
|
if isinstance(callee, ast.Name):
|
||||||
|
class_name = callee.id
|
||||||
|
elif isinstance(callee, ast.Attribute) and isinstance(callee.value, ast.Name):
|
||||||
|
class_name = callee.attr
|
||||||
|
else:
|
||||||
|
return None
|
||||||
|
if isinstance(node.targets[0], ast.Name):
|
||||||
|
instance = node.targets[0].id
|
||||||
|
return instance, class_name
|
||||||
|
return None
|
||||||
|
|
||||||
|
def _extract_call(self, call) -> Tuple[str, str, Dict[str, Any], str]:
|
||||||
|
import ast
|
||||||
|
owner_name, method_name, call_kind = "", "", "func"
|
||||||
|
if isinstance(call.func, ast.Attribute):
|
||||||
|
method_name = call.func.attr
|
||||||
|
if isinstance(call.func.value, ast.Name):
|
||||||
|
owner_name = call.func.value.id
|
||||||
|
call_kind = "instance" if owner_name in self.instance_to_resource else "class_or_module"
|
||||||
|
elif isinstance(call.func.value, ast.Attribute) and isinstance(call.func.value.value, ast.Name):
|
||||||
|
owner_name = call.func.value.attr
|
||||||
|
call_kind = "class_or_module"
|
||||||
|
elif isinstance(call.func, ast.Name):
|
||||||
|
method_name = call.func.id
|
||||||
|
call_kind = "func"
|
||||||
|
|
||||||
|
def pack(node):
|
||||||
|
if isinstance(node, ast.Name):
|
||||||
|
return {"type": "variable", "value": node.id}
|
||||||
|
if isinstance(node, ast.Constant):
|
||||||
|
return {"type": "constant", "value": node.value}
|
||||||
|
if isinstance(node, ast.Dict):
|
||||||
|
return {"type": "dict", "value": self._parse_dict(node)}
|
||||||
|
if isinstance(node, ast.List):
|
||||||
|
return {"type": "list", "value": self._parse_list(node)}
|
||||||
|
return {"type": "raw", "value": ast.unparse(node) if hasattr(ast, "unparse") else str(node)}
|
||||||
|
|
||||||
|
args: Dict[str, Any] = {}
|
||||||
|
pos: List[Any] = []
|
||||||
|
for a in call.args:
|
||||||
|
pos.append(pack(a))
|
||||||
|
for kw in call.keywords:
|
||||||
|
args[kw.arg] = pack(kw.value)
|
||||||
|
if pos:
|
||||||
|
args["_positional"] = pos
|
||||||
|
return owner_name, method_name, args, call_kind
|
||||||
|
|
||||||
|
def _parse_dict(self, node) -> Dict[str, Any]:
|
||||||
|
import ast
|
||||||
|
out: Dict[str, Any] = {}
|
||||||
|
for k, v in zip(node.keys, node.values):
|
||||||
|
if isinstance(k, ast.Constant):
|
||||||
|
key = str(k.value)
|
||||||
|
if isinstance(v, ast.Name):
|
||||||
|
out[key] = f"var:{v.id}"
|
||||||
|
elif isinstance(v, ast.Constant):
|
||||||
|
out[key] = v.value
|
||||||
|
elif isinstance(v, ast.Dict):
|
||||||
|
out[key] = self._parse_dict(v)
|
||||||
|
elif isinstance(v, ast.List):
|
||||||
|
out[key] = self._parse_list(v)
|
||||||
|
return out
|
||||||
|
|
||||||
|
def _parse_list(self, node) -> List[Any]:
|
||||||
|
import ast
|
||||||
|
out: List[Any] = []
|
||||||
|
for elt in node.elts:
|
||||||
|
if isinstance(elt, ast.Name):
|
||||||
|
out.append(f"var:{elt.id}")
|
||||||
|
elif isinstance(elt, ast.Constant):
|
||||||
|
out.append(elt.value)
|
||||||
|
elif isinstance(elt, ast.Dict):
|
||||||
|
out.append(self._parse_dict(elt))
|
||||||
|
elif isinstance(elt, ast.List):
|
||||||
|
out.append(self._parse_list(elt))
|
||||||
|
return out
|
||||||
|
|
||||||
|
def _normalize_var_tokens(self, x: Any) -> Any:
|
||||||
|
if isinstance(x, str) and x.startswith("var:"):
|
||||||
|
return {"__var__": x[4:]}
|
||||||
|
if isinstance(x, list):
|
||||||
|
return [self._normalize_var_tokens(i) for i in x]
|
||||||
|
if isinstance(x, dict):
|
||||||
|
return {k: self._normalize_var_tokens(v) for k, v in x.items()}
|
||||||
|
return x
|
||||||
|
|
||||||
|
def _make_params_payload(self, resource_name: Optional[str], template_name: str, call_args: Dict[str, Any]) -> Dict[str, Any]:
|
||||||
|
input_keys = self.registry.get_action_input_keys(resource_name, template_name) if resource_name else []
|
||||||
|
defaults = self.registry.get_action_goal_default(resource_name, template_name) if resource_name else {}
|
||||||
|
params: Dict[str, Any] = dict(defaults)
|
||||||
|
|
||||||
|
def unpack(p):
|
||||||
|
t, v = p.get("type"), p.get("value")
|
||||||
|
if t == "variable":
|
||||||
|
return {"__var__": v}
|
||||||
|
if t == "dict":
|
||||||
|
return self._normalize_var_tokens(v)
|
||||||
|
if t == "list":
|
||||||
|
return self._normalize_var_tokens(v)
|
||||||
|
return v
|
||||||
|
|
||||||
|
for k, p in call_args.items():
|
||||||
|
if k == "_positional":
|
||||||
|
continue
|
||||||
|
params[k] = unpack(p)
|
||||||
|
|
||||||
|
pos = call_args.get("_positional", [])
|
||||||
|
if pos:
|
||||||
|
if input_keys:
|
||||||
|
for i, p in enumerate(pos):
|
||||||
|
if i >= len(input_keys):
|
||||||
|
break
|
||||||
|
name = input_keys[i]
|
||||||
|
if name in params:
|
||||||
|
continue
|
||||||
|
params[name] = unpack(p)
|
||||||
|
else:
|
||||||
|
for i, p in enumerate(pos):
|
||||||
|
params[f"arg_{i}"] = unpack(p)
|
||||||
|
return params
|
||||||
|
|
||||||
|
# ---- handlers ----
|
||||||
|
def _on_assign(self, stmt):
|
||||||
|
import ast
|
||||||
|
inst = self._extract_device_instantiation(stmt)
|
||||||
|
if inst:
|
||||||
|
instance, code_class = inst
|
||||||
|
resource_name = self.registry.resolve_resource_by_classname(code_class)
|
||||||
|
self.instance_to_resource[instance] = resource_name
|
||||||
|
return
|
||||||
|
|
||||||
|
if isinstance(stmt.value, ast.Call):
|
||||||
|
owner, method, call_args, kind = self._extract_call(stmt.value)
|
||||||
|
if kind == "instance":
|
||||||
|
device_key = owner
|
||||||
|
resource_name = self.instance_to_resource.get(owner)
|
||||||
|
else:
|
||||||
|
device_key = owner
|
||||||
|
resource_name = self.registry.resolve_resource_by_classname(owner)
|
||||||
|
|
||||||
|
module = self.registry.get_device_module(resource_name)
|
||||||
|
params = self._make_params_payload(resource_name, method, call_args)
|
||||||
|
|
||||||
|
nid = self._new_node_id()
|
||||||
|
self.graph.add_workflow_node(
|
||||||
|
nid,
|
||||||
|
device_key=device_key,
|
||||||
|
resource_name=resource_name, # ✅
|
||||||
|
module=module,
|
||||||
|
template_name=method, # ✅
|
||||||
|
params=params,
|
||||||
|
variable_sources=self.variable_sources,
|
||||||
|
add_ready_if_no_vars=True,
|
||||||
|
prev_node_id=(nid - 1) if nid > 0 else None,
|
||||||
|
)
|
||||||
|
|
||||||
|
out_vars = self._assign_targets(stmt.targets[0])
|
||||||
|
for var in out_vars:
|
||||||
|
self.variable_sources[var] = {"node_id": nid, "output_name": "result"}
|
||||||
|
|
||||||
|
def _on_expr(self, stmt):
|
||||||
|
import ast
|
||||||
|
if not isinstance(stmt.value, ast.Call):
|
||||||
|
return
|
||||||
|
owner, method, call_args, kind = self._extract_call(stmt.value)
|
||||||
|
if kind == "instance":
|
||||||
|
device_key = owner
|
||||||
|
resource_name = self.instance_to_resource.get(owner)
|
||||||
|
else:
|
||||||
|
device_key = owner
|
||||||
|
resource_name = self.registry.resolve_resource_by_classname(owner)
|
||||||
|
|
||||||
|
module = self.registry.get_device_module(resource_name)
|
||||||
|
params = self._make_params_payload(resource_name, method, call_args)
|
||||||
|
|
||||||
|
nid = self._new_node_id()
|
||||||
|
self.graph.add_workflow_node(
|
||||||
|
nid,
|
||||||
|
device_key=device_key,
|
||||||
|
resource_name=resource_name, # ✅
|
||||||
|
module=module,
|
||||||
|
template_name=method, # ✅
|
||||||
|
params=params,
|
||||||
|
variable_sources=self.variable_sources,
|
||||||
|
add_ready_if_no_vars=True,
|
||||||
|
prev_node_id=(nid - 1) if nid > 0 else None,
|
||||||
|
)
|
||||||
|
|
||||||
|
def convert(self, python_code: str):
|
||||||
|
tree = ast.parse(python_code)
|
||||||
|
for stmt in tree.body:
|
||||||
|
if isinstance(stmt, ast.Assign):
|
||||||
|
self._on_assign(stmt)
|
||||||
|
elif isinstance(stmt, ast.Expr):
|
||||||
|
self._on_expr(stmt)
|
||||||
|
return self
|
||||||
131
unilabos/workflow/from_xdl.py
Normal file
131
unilabos/workflow/from_xdl.py
Normal file
@@ -0,0 +1,131 @@
|
|||||||
|
from typing import List, Any, Dict
|
||||||
|
import xml.etree.ElementTree as ET
|
||||||
|
|
||||||
|
|
||||||
|
def convert_to_type(val: str) -> Any:
|
||||||
|
"""将字符串值转换为适当的数据类型"""
|
||||||
|
if val == "True":
|
||||||
|
return True
|
||||||
|
if val == "False":
|
||||||
|
return False
|
||||||
|
if val == "?":
|
||||||
|
return None
|
||||||
|
if val.endswith(" g"):
|
||||||
|
return float(val.split(" ")[0])
|
||||||
|
if val.endswith("mg"):
|
||||||
|
return float(val.split("mg")[0])
|
||||||
|
elif val.endswith("mmol"):
|
||||||
|
return float(val.split("mmol")[0]) / 1000
|
||||||
|
elif val.endswith("mol"):
|
||||||
|
return float(val.split("mol")[0])
|
||||||
|
elif val.endswith("ml"):
|
||||||
|
return float(val.split("ml")[0])
|
||||||
|
elif val.endswith("RPM"):
|
||||||
|
return float(val.split("RPM")[0])
|
||||||
|
elif val.endswith(" °C"):
|
||||||
|
return float(val.split(" ")[0])
|
||||||
|
elif val.endswith(" %"):
|
||||||
|
return float(val.split(" ")[0])
|
||||||
|
return val
|
||||||
|
|
||||||
|
|
||||||
|
def flatten_xdl_procedure(procedure_elem: ET.Element) -> List[ET.Element]:
|
||||||
|
"""展平嵌套的XDL程序结构"""
|
||||||
|
flattened_operations = []
|
||||||
|
TEMP_UNSUPPORTED_PROTOCOL = ["Purge", "Wait", "Stir", "ResetHandling"]
|
||||||
|
|
||||||
|
def extract_operations(element: ET.Element):
|
||||||
|
if element.tag not in ["Prep", "Reaction", "Workup", "Purification", "Procedure"]:
|
||||||
|
if element.tag not in TEMP_UNSUPPORTED_PROTOCOL:
|
||||||
|
flattened_operations.append(element)
|
||||||
|
|
||||||
|
for child in element:
|
||||||
|
extract_operations(child)
|
||||||
|
|
||||||
|
for child in procedure_elem:
|
||||||
|
extract_operations(child)
|
||||||
|
|
||||||
|
return flattened_operations
|
||||||
|
|
||||||
|
|
||||||
|
def parse_xdl_content(xdl_content: str) -> tuple:
|
||||||
|
"""解析XDL内容"""
|
||||||
|
try:
|
||||||
|
xdl_content_cleaned = "".join(c for c in xdl_content if c.isprintable())
|
||||||
|
root = ET.fromstring(xdl_content_cleaned)
|
||||||
|
|
||||||
|
synthesis_elem = root.find("Synthesis")
|
||||||
|
if synthesis_elem is None:
|
||||||
|
return None, None, None
|
||||||
|
|
||||||
|
# 解析硬件组件
|
||||||
|
hardware_elem = synthesis_elem.find("Hardware")
|
||||||
|
hardware = []
|
||||||
|
if hardware_elem is not None:
|
||||||
|
hardware = [{"id": c.get("id"), "type": c.get("type")} for c in hardware_elem.findall("Component")]
|
||||||
|
|
||||||
|
# 解析试剂
|
||||||
|
reagents_elem = synthesis_elem.find("Reagents")
|
||||||
|
reagents = []
|
||||||
|
if reagents_elem is not None:
|
||||||
|
reagents = [{"name": r.get("name"), "role": r.get("role", "")} for r in reagents_elem.findall("Reagent")]
|
||||||
|
|
||||||
|
# 解析程序
|
||||||
|
procedure_elem = synthesis_elem.find("Procedure")
|
||||||
|
if procedure_elem is None:
|
||||||
|
return None, None, None
|
||||||
|
|
||||||
|
flattened_operations = flatten_xdl_procedure(procedure_elem)
|
||||||
|
return hardware, reagents, flattened_operations
|
||||||
|
|
||||||
|
except ET.ParseError as e:
|
||||||
|
raise ValueError(f"Invalid XDL format: {e}")
|
||||||
|
|
||||||
|
|
||||||
|
def convert_xdl_to_dict(xdl_content: str) -> Dict[str, Any]:
|
||||||
|
"""
|
||||||
|
将XDL XML格式转换为标准的字典格式
|
||||||
|
|
||||||
|
Args:
|
||||||
|
xdl_content: XDL XML内容
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
转换结果,包含步骤和器材信息
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
hardware, reagents, flattened_operations = parse_xdl_content(xdl_content)
|
||||||
|
if hardware is None:
|
||||||
|
return {"error": "Failed to parse XDL content", "success": False}
|
||||||
|
|
||||||
|
# 将XDL元素转换为字典格式
|
||||||
|
steps_data = []
|
||||||
|
for elem in flattened_operations:
|
||||||
|
# 转换参数类型
|
||||||
|
parameters = {}
|
||||||
|
for key, val in elem.attrib.items():
|
||||||
|
converted_val = convert_to_type(val)
|
||||||
|
if converted_val is not None:
|
||||||
|
parameters[key] = converted_val
|
||||||
|
|
||||||
|
step_dict = {
|
||||||
|
"operation": elem.tag,
|
||||||
|
"parameters": parameters,
|
||||||
|
"description": elem.get("purpose", f"Operation: {elem.tag}"),
|
||||||
|
}
|
||||||
|
steps_data.append(step_dict)
|
||||||
|
|
||||||
|
# 合并硬件和试剂为统一的labware_info格式
|
||||||
|
labware_data = []
|
||||||
|
labware_data.extend({"id": hw["id"], "type": "hardware", **hw} for hw in hardware)
|
||||||
|
labware_data.extend({"name": reagent["name"], "type": "reagent", **reagent} for reagent in reagents)
|
||||||
|
|
||||||
|
return {
|
||||||
|
"success": True,
|
||||||
|
"steps": steps_data,
|
||||||
|
"labware": labware_data,
|
||||||
|
"message": f"Successfully converted XDL to dict format. Found {len(steps_data)} steps and {len(labware_data)} labware items.",
|
||||||
|
}
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
error_msg = f"XDL conversion failed: {str(e)}"
|
||||||
|
return {"error": error_msg, "success": False}
|
||||||
Reference in New Issue
Block a user