open_pub/Uni-Lab-OS
1
0
Fork 0
mirror of https://github.com/deepmodeling/Uni-Lab-OS synced 2026-04-27 08:50:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

修改角度读取方式

Browse Source
This commit is contained in:
zhangshixiang
2026-04-14 15:58:37 +08:00
parent c001f6a151
commit a45db46cf1
8 changed files with 254 additions and 201 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
unilabos/resources/graphio.py
View File

@@ -593,7 +593,7 @@ def resource_ulab_to_plr(resource: dict, plr_model=False) -> "ResourcePLR":
"size_y": resource["config"].get("size_y", 0),
"size_z": resource["config"].get("size_z", 0),
"location": {**resource["position"], "type": "Coordinate"},
"rotation": {"x": 0, "y": 0, "z": 0, "type": "Rotation"}, # Resource如果没有rotation，是plr版本太低
"rotation": {resource["config"].get("rotation", {"x": 0, "y": 0, "z": 0, "type": "Rotation"})}, # Resource如果没有rotation，是plr版本太低
"category": resource["type"],
"model": resource["config"].get("model", None), # resource中deck没有model
"children": (

6
unilabos/resources/resource_tracker.py
View File

@@ -244,6 +244,12 @@ class ResourceDictInstance(object):
height= content["config"].get("size_y", 0),
depth= content["config"].get("size_z", 0),
)
if "rotation" not in pose:
pose["rotation"] = ResourceDictPositionObjectType(
x=content["config"].get("rotation", {}).get("x", 0),
y=content["config"].get("rotation", {}).get("y", 0),
z=content["config"].get("rotation", {}).get("z", 0),
)
content["pose"] = pose
try:
res_dict = ResourceDict.model_validate(content)