92 lines
2.2 KiB
Python
92 lines
2.2 KiB
Python
"""MIL SDK 核心数据模型"""
|
|
from dataclasses import dataclass, field
|
|
from typing import Any
|
|
|
|
|
|
@dataclass
|
|
class DataLog:
|
|
"""仿真数据日志记录
|
|
|
|
Attributes:
|
|
time: 时间戳（秒）
|
|
value: 信号值（可以是任意类型）
|
|
"""
|
|
time: float = 0.0
|
|
value: str = ""
|
|
|
|
|
|
@dataclass
|
|
class SignalData:
|
|
"""信号数据封装
|
|
|
|
Attributes:
|
|
signal_type: 信号类型
|
|
column: 列索引
|
|
datalog: 数据日志列表
|
|
"""
|
|
# signal_type: str | None = None
|
|
column: int = 0
|
|
attributes: dict[str, str] = field(default_factory=dict)
|
|
datalog: list[DataLog] = field(default_factory=list)
|
|
|
|
|
|
def to_dict(self) -> dict[str, Any]:
|
|
"""转换为字典格式
|
|
|
|
Returns:
|
|
包含 signal_type、column、datalog 的字典
|
|
"""
|
|
return {
|
|
"column": self.column,
|
|
"attributes": self.attributes,
|
|
"datalog": self.datalog
|
|
}
|
|
|
|
|
|
@dataclass
|
|
class ExcelDataResult:
|
|
"""Excel 数据读取结果封装
|
|
|
|
提供对 Excel 数据的类型安全访问，隐藏内部实现细节
|
|
|
|
Attributes:
|
|
sheet_name: 工作表名称
|
|
source_row: Source: Input 所在行号
|
|
signals: 信号名称到信号数据的映射
|
|
"""
|
|
sheet_name: str = "Scenario1"
|
|
source_row: int = 0
|
|
signals: dict[str, SignalData] = field(default_factory=dict)
|
|
|
|
def get_signal(self, name: str) -> SignalData | None:
|
|
"""获取指定信号的数据
|
|
|
|
Args:
|
|
name: 信号名称
|
|
|
|
Returns:
|
|
信号数据对象，如果不存在返回 None
|
|
"""
|
|
return self.signals.get(name)
|
|
|
|
def get_signal_names(self) -> list[str]:
|
|
"""获取所有信号名称
|
|
|
|
Returns:
|
|
信号名称列表
|
|
"""
|
|
return list(self.signals.keys())
|
|
|
|
def to_dict(self) -> dict[str, Any]:
|
|
"""转换为字典格式（兼容旧 API）
|
|
|
|
Returns:
|
|
包含所有信号的字典，保留原有的数据结构
|
|
"""
|
|
result = {
|
|
"sheet_name": self.sheet_name,
|
|
"source_row": self.source_row
|
|
}
|
|
for name, signal in self.signals.items():
|
|
result[name] = signal.to_dict()
|
|
return result |