mirror of
https://github.com/dptech-corp/Uni-Lab-OS.git
synced 2025-12-15 13:44:39 +00:00
60 lines
2.9 KiB
Python
60 lines
2.9 KiB
Python
import copy
|
||
from typing import Optional
|
||
|
||
from unilabos.registry.registry import lab_registry
|
||
from unilabos.ros.device_node_wrapper import ros2_device_node
|
||
from unilabos.ros.nodes.base_device_node import ROS2DeviceNode, DeviceInitError
|
||
from unilabos.ros.nodes.resource_tracker import ResourceDictInstance
|
||
from unilabos.utils import logger
|
||
from unilabos.utils.exception import DeviceClassInvalid
|
||
from unilabos.utils.import_manager import default_manager
|
||
|
||
|
||
|
||
def initialize_device_from_dict(device_id, device_config: ResourceDictInstance) -> Optional[ROS2DeviceNode]:
|
||
"""Initializes a device based on its configuration.
|
||
|
||
This function dynamically imports the appropriate device class and creates an instance of it using the provided device configuration.
|
||
It also sets up action clients for the device based on its action value mappings.
|
||
|
||
Args:
|
||
device_id (str): The unique identifier for the device.
|
||
device_config (dict): The configuration dictionary for the device, which includes the class type and other parameters.
|
||
|
||
Returns:
|
||
None
|
||
"""
|
||
d = None
|
||
device_class_config = device_config.res_content.klass
|
||
uid = device_config.res_content.uuid
|
||
if isinstance(device_class_config, str): # 如果是字符串，则直接去lab_registry中查找，获取class
|
||
if len(device_class_config) == 0:
|
||
raise DeviceClassInvalid(f"Device [{device_id}] class cannot be an empty string. {device_config}")
|
||
if device_class_config not in lab_registry.device_type_registry:
|
||
raise DeviceClassInvalid(f"Device [{device_id}] class {device_class_config} not found. {device_config}")
|
||
device_class_config = lab_registry.device_type_registry[device_class_config]["class"]
|
||
elif isinstance(device_class_config, dict):
|
||
raise DeviceClassInvalid(f"Device [{device_id}] class config should be type 'str' but 'dict' got. {device_config}")
|
||
if isinstance(device_class_config, dict):
|
||
DEVICE = default_manager.get_class(device_class_config["module"])
|
||
# 不管是ros2的实例，还是python的，都必须包一次，除了HostNode
|
||
DEVICE = ros2_device_node(
|
||
DEVICE,
|
||
status_types=device_class_config.get("status_types", {}),
|
||
device_config=device_config,
|
||
action_value_mappings=device_class_config.get("action_value_mappings", {}),
|
||
hardware_interface=device_class_config.get(
|
||
"hardware_interface",
|
||
{"name": "hardware_interface", "write": "send_command", "read": "read_data", "extra_info": []},
|
||
)
|
||
)
|
||
try:
|
||
d = DEVICE(
|
||
device_id=device_id, device_uuid=uid, driver_is_ros=device_class_config["type"] == "ros2", driver_params=device_config.res_content.config
|
||
)
|
||
except DeviceInitError as ex:
|
||
return d
|
||
else:
|
||
logger.warning(f"initialize device {device_id} failed, provided device_config: {device_config}")
|
||
return d
|