37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
|
# midware/drivers/__init__.py
|
|||
|
|
"""
|
|||
|
|
总线驱动模块：统一管理所有总线驱动的导入和初始化
|
|||
|
|
"""
|
|||
|
|
from .base_driver import BaseBusDriver
|
|||
|
|
from .bus_1553b import Bus1553BDriver
|
|||
|
|
from .bus_can import BusCANDriver
|
|||
|
|
from .bus_uart import BusUARTDriver
|
|||
|
|
from .bus_ad import BusADDriver
|
|||
|
|
from .bus_oc import BusOCDriver
|
|||
|
|
from .bus_network import BusNetworkDriver
|
|||
|
|
from loguru import logger
|
|||
|
|
|
|||
|
|
# 驱动映射表：协议名称 → 驱动类（与bus_udp_decoupler.py中的协议名对齐）
|
|||
|
|
DRIVER_MAPPING = {
|
|||
|
|
"1553B": Bus1553BDriver,
|
|||
|
|
"CAN": BusCANDriver,
|
|||
|
|
"UART": BusUARTDriver,
|
|||
|
|
"AD": BusADDriver,
|
|||
|
|
"OC": BusOCDriver,
|
|||
|
|
"网络": BusNetworkDriver
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
def create_bus_driver(protocol: str, config: dict = None) -> BaseBusDriver:
|
|||
|
|
"""
|
|||
|
|
工厂函数：根据协议名称创建驱动实例
|
|||
|
|
:param protocol: 协议名称（如"1553B"）
|
|||
|
|
:param config: 驱动配置
|
|||
|
|
:return: 驱动实例
|
|||
|
|
"""
|
|||
|
|
if protocol not in DRIVER_MAPPING:
|
|||
|
|
raise ValueError(f"未知协议 {protocol}，无对应驱动")
|
|||
|
|
|
|||
|
|
driver_class = DRIVER_MAPPING[protocol]
|
|||
|
|
driver = driver_class(config=config)
|
|||
|
|
logger.info(f"创建 {protocol} 驱动实例成功：{driver.__class__.__name__}")
|
|||
|
|
return driver
|