133 lines
4.6 KiB
Python
133 lines
4.6 KiB
Python
import serial
|
||
import time
|
||
import logging
|
||
import os
|
||
import random
|
||
from typing import Optional
|
||
|
||
class SerialSimulator:
|
||
"""串口数据模拟器"""
|
||
|
||
def __init__(self, port: str, baudrate: int):
|
||
"""
|
||
初始化串口模拟器
|
||
Args:
|
||
port: 串口名称
|
||
baudrate: 波特率
|
||
"""
|
||
self.port = port
|
||
self.baudrate = baudrate
|
||
self.serial: Optional[serial.Serial] = None
|
||
self.test_data: Optional[bytes] = None
|
||
self.bytes_sent = 0
|
||
self.start_time = None
|
||
self.end_time = None
|
||
|
||
# 配置日志
|
||
log_dir = "logs"
|
||
if not os.path.exists(log_dir):
|
||
os.makedirs(log_dir)
|
||
|
||
logging.basicConfig(
|
||
level=logging.INFO,
|
||
format='%(asctime)s - %(levelname)s - %(message)s'
|
||
)
|
||
self.logger = logging.getLogger(__name__)
|
||
|
||
def load_test_data(self) -> bool:
|
||
"""加载测试数据"""
|
||
try:
|
||
filename = os.path.join("test_data", "test_frames.bin")
|
||
with open(filename, 'rb') as f:
|
||
self.test_data = f.read()
|
||
self.logger.info(f"已加载测试数据: {len(self.test_data)} 字节")
|
||
return True
|
||
except Exception as e:
|
||
self.logger.error(f"加载测试数据失败: {e}")
|
||
return False
|
||
|
||
def connect(self) -> bool:
|
||
"""连接串口"""
|
||
try:
|
||
self.serial = serial.Serial(
|
||
port=self.port,
|
||
baudrate=self.baudrate,
|
||
timeout=0.1,
|
||
write_timeout=10 # 添加写入超时
|
||
)
|
||
self.logger.info(f"成功连接到串口 {self.port}")
|
||
return True
|
||
except Exception as e:
|
||
self.logger.error(f"连接串口失败: {e}")
|
||
return False
|
||
|
||
def disconnect(self):
|
||
"""断开串口连接"""
|
||
if self.serial and self.serial.is_open:
|
||
self.serial.close()
|
||
self.logger.info("已断开串口连接")
|
||
|
||
def send_data(self) -> bool:
|
||
"""发送数据"""
|
||
if not self.test_data:
|
||
if not self.load_test_data():
|
||
return False
|
||
|
||
if not self.serial or not self.serial.is_open:
|
||
if not self.connect():
|
||
return False
|
||
|
||
try:
|
||
data_length = len(self.test_data)
|
||
current_pos = 0
|
||
self.start_time = time.time()
|
||
|
||
# 记录发送开始信息
|
||
self.logger.info(f"开始发送数据: {data_length} 字节")
|
||
self.logger.info(f"串口参数: {self.port}, {self.baudrate} baud")
|
||
|
||
while current_pos < data_length:
|
||
# 随机大小的数据包（1-1024字节）
|
||
packet_size = random.randint(1, 1024)
|
||
# 确保不超过剩余数据长度
|
||
packet_size = min(packet_size, data_length - current_pos)
|
||
|
||
# 发送数据包
|
||
packet = self.test_data[current_pos:current_pos + packet_size]
|
||
self.serial.write(packet)
|
||
current_pos += packet_size
|
||
self.bytes_sent += packet_size
|
||
|
||
# 随机延迟（0-10ms）
|
||
time.sleep(random.uniform(0, 0.01))
|
||
|
||
# 每发送1MB数据打印一次进度
|
||
if self.bytes_sent % (1024*1024) < packet_size:
|
||
progress = (self.bytes_sent / data_length) * 100
|
||
elapsed_time = time.time() - self.start_time
|
||
speed = self.bytes_sent / elapsed_time / 1024 # KB/s
|
||
self.logger.info(f"发送进度: {progress:.1f}% ({self.bytes_sent}/{data_length} 字节), "
|
||
f"速度: {speed:.1f} KB/s")
|
||
|
||
self.end_time = time.time()
|
||
total_time = self.end_time - self.start_time
|
||
avg_speed = self.bytes_sent / total_time / 1024 # KB/s
|
||
|
||
# 记录发送完成信息
|
||
self.logger.info(f"数据发送完成")
|
||
self.logger.info(f"总时间: {total_time:.2f} 秒")
|
||
self.logger.info(f"平均速度: {avg_speed:.1f} KB/s")
|
||
self.logger.info(f"发送字节数: {self.bytes_sent}")
|
||
|
||
return True
|
||
|
||
except Exception as e:
|
||
self.logger.error(f"发送数据错误: {e}")
|
||
return False
|
||
finally:
|
||
self.disconnect()
|
||
|
||
if __name__ == "__main__":
|
||
# 使用示例
|
||
simulator = SerialSimulator(port="COM1", baudrate=4000000)
|
||
simulator.send_data() |