191 lines
7.2 KiB
Python
191 lines
7.2 KiB
Python
# -*- coding: utf-8 -*-
|
||
import sys
|
||
import os
|
||
import io
|
||
import time
|
||
|
||
# 设置标准输出和标准错误的编码为UTF-8
|
||
sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8')
|
||
sys.stderr = io.TextIOWrapper(sys.stderr.buffer, encoding='utf-8')
|
||
|
||
from loguru import logger
|
||
from midware.ui.main_ui import start_main_ui
|
||
from midware.config.base_config import init_base_config
|
||
#from midware.core.data_forward import init_forward_service
|
||
from midware.config.excel_config import load_csv_config
|
||
from midware.core.error_check import check_checksum, verify_frame_integrity
|
||
from midware.network.udp_handler import udp_server
|
||
from midware.network.protocol_dispatcher import dispatch_udp_data
|
||
|
||
import midware.config.base_config as base_config
|
||
|
||
# from midware.config.base_config import init_base_config
|
||
# from midware.config.base_config import DEVICE_CONFIG_DICT
|
||
# 先导入配置核心模块
|
||
from midware.config.base_config import init_base_config, update_device_config, get_device_config
|
||
# 后导入UDP模块（此时udp_server还是None）
|
||
from midware.network.udp_handler import MultiUDPServer, udp_server
|
||
|
||
# 添加工程根目录到环境变量
|
||
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
|
||
'''
|
||
1. 程序入口（main.py）
|
||
负责初始化日志、加载配置、启动 UI 和核心服务，实现程序一键启动
|
||
'''
|
||
# 初始化日志
|
||
def init_logger():
|
||
logger.add(
|
||
"logs/run_{time:YYYY-MM-DD}.log",
|
||
rotation="daily",
|
||
level="INFO",
|
||
format="{time} | {level} | {module}:{line} | {message}"
|
||
)
|
||
logger.add(
|
||
"logs/error_{time:YYYY-MM-DD}.log",
|
||
rotation="daily",
|
||
level="ERROR",
|
||
format="{time} | {level} | {module}:{line} | {message}"
|
||
)
|
||
return logger
|
||
|
||
if __name__ == "__main__":
|
||
# 初始化日志
|
||
log = init_logger()
|
||
log.info("虚拟仿真平台中间层软件启动")
|
||
logger.info("========== 虚拟仿真平台中间层软件 启动 ==========")
|
||
##试一下2026/2/27
|
||
csv_path = "config_files/device_config.csv"
|
||
|
||
DEVICE_CONFIG_DICT = load_csv_config(csv_path)
|
||
|
||
#try:
|
||
# 初始化基础配置（外设、UDP、总线）
|
||
# 步骤1：初始化外设配置（核心！先初始化配置，再实例化其他模块）
|
||
init_base_config()
|
||
|
||
device_config = get_device_config()
|
||
if not device_config:
|
||
logger.error("配置加载为空，工程无法启动，退出！")
|
||
exit(1)
|
||
|
||
# 初始化数据转发核心服务
|
||
#init_forward_service()
|
||
|
||
if not base_config.DEVICE_CONFIG_DICT:
|
||
logger.error("外设配置加载失败，工程启动终止")
|
||
else:
|
||
logger.info("外设配置加载成功")
|
||
print("程序运行到这里了")
|
||
|
||
# 步骤2：实例化UDP单例服务（此时配置已初始化，能读取到有效数据）
|
||
global udp_server
|
||
udp_server = MultiUDPServer()
|
||
logger.info(f"工程启动成功｜外设数量：{len(device_config)}｜UDP监听端口：{list(udp_server.udp_sockets.keys())}")
|
||
|
||
# 初始化UDP服务器
|
||
#udp_server.__init__() # 确保先初始化
|
||
|
||
'''
|
||
# # 初始化UDP服务器
|
||
# udp_server.init() # 确保先初始化
|
||
# 启动主UI
|
||
start_main_ui()
|
||
log.info("软件运行中...")
|
||
except Exception as e:
|
||
log.error(f"程序启动失败：{str(e)}", exc_info=True)
|
||
sys.exit(1)
|
||
'''
|
||
|
||
# 步骤3：工程主循环（阻塞运行，处理UDP数据）
|
||
try:
|
||
while True:
|
||
# 非阻塞接收所有UDP端口数据
|
||
recv_data = udp_server.recv_multi_udp_hex()
|
||
if recv_data:
|
||
# 后续对接协议解析层/总线适配层的逻辑写在这里
|
||
for data in recv_data:
|
||
port = data["port"]
|
||
dev_name = data["dev_name"]
|
||
protocol = data["protocol"]
|
||
raw_data = data["raw_data"]
|
||
if len(raw_data) > 0:
|
||
#打印原始数据（测试用）
|
||
print(f"接收到 {dev_name} 的原始数据：{raw_data}, 协议类型：{protocol}")
|
||
|
||
# 根据协议类型进行数据解析和转发
|
||
print("后续根据设备的协议类型进行数据解析和转发...")
|
||
|
||
#判断协议类型，采用不同解析和转发方法
|
||
if protocol == "UART":
|
||
pass
|
||
|
||
# 示例：直接回传数据到故障注入软件（测试用）
|
||
# 2. 分发处理数据（核心调用）
|
||
dispatch_udp_data(recv_data)
|
||
|
||
# 示例：直接回传数据到故障注入软件（测试用）
|
||
udp_server.send_to_fault(raw_data, dev_name)
|
||
# 主循环间隔1us，不影响实时性
|
||
time.sleep(1e-6)
|
||
except KeyboardInterrupt:
|
||
logger.info("用户终止工程，开始释放资源...")
|
||
finally:
|
||
# 退出时关闭UDP服务
|
||
udp_server.close()
|
||
logger.info("========== 虚拟仿真平台中间层软件 退出 ==========")
|
||
|
||
|
||
# if name == "main":
|
||
#test_config = load_excel_config("../../config_files/device_config.xlsx")
|
||
#test_config = load_excel_config("config_files/device_config.xlsx")
|
||
|
||
# test_config = load_csv_config("config_files/device_config.csv") ##暂时注释这行
|
||
# print(test_config)
|
||
'''
|
||
#测试校验和代码（单独运行该文件时执行）
|
||
|
||
#测试 UART 校验：原始数据 + b'x9c\x47'（crc16_modbus 校验位）
|
||
uart_data = b"test_uart_data" + b'\x9c\x47'
|
||
print (f"UART 校验结果：{check_checksum (uart_data, 'UART', 'Uart0')}")
|
||
#测试 CAN 校验：原始数据（10 字节）+b'\x1a\x2b'（crc16_ccitt 校验位），总长度 12≤13（11+2）
|
||
can_data = b"test_can_123" + b'\x1a\x2b'
|
||
print (f"CAN 校验结果：{check_checksum (can_data, 'CAN', 'CAN1')}")
|
||
print (f"CAN 帧完整性结果：{verify_frame_integrity (can_data, 'CAN', 'CAN1')}")
|
||
#测试 AD 校验：原始数据 + b'\x5f'（累加和校验位）
|
||
ad_data = b"test_ad" + b'\x5f'
|
||
print (f"AD 校验结果：{check_checksum (ad_data, 'AD', 'AD0')}")
|
||
#测试失败场景：校验位错误
|
||
wrong_checksum_data = b"test_uart_data" + b'\x00\x00'
|
||
print (f"UART 校验失败场景结果：{check_checksum (wrong_checksum_data, 'UART', 'Uart0')}")
|
||
'''
|
||
|
||
try:
|
||
# 初始化基础配置
|
||
init_base_config()
|
||
|
||
# 初始化UDP服务器
|
||
udp_server.init() # 确保先初始化
|
||
|
||
# 测试数据接收
|
||
data_list = udp_server.recv_multi_udp_hex()
|
||
for data in data_list:
|
||
dev_name = data["dev_name"]
|
||
raw_data = data["raw_data"]
|
||
# 对每个外设的原始数据进行协议处理...
|
||
|
||
# 2. 向故障注入软件发送数据
|
||
if len(raw_data) > 0:
|
||
udp_server.send_to_fault(raw_data, dev_name="Uart0")
|
||
|
||
# 3. 外设配置更新后，重新加载UDP端口
|
||
udp_server.reload_sockets()
|
||
|
||
# 4. 程序退出时关闭所有套接字
|
||
udp_server.close()
|
||
|
||
except Exception as e:
|
||
log.error(f"程序启动失败：{str(e)}", exc_info=True)
|
||
sys.exit(1)
|
||
|
||
|
||
|