refactor: 将参数对象作为参数传递给run_egm函数

修改run_egm函数,使其接收参数对象而不是依赖全局变量,并更新相关调用代码
This commit is contained in:
dmy
2026-03-02 23:06:34 +08:00
parent 759b8b9a25
commit 630598d498
2 changed files with 7 additions and 5 deletions

10
main.py
View File

@@ -60,11 +60,13 @@ def read_parameter(toml_file_path):
para.max_i = toml_optional["max_i"]
def run_egm() -> dict:
def run_egm(para: Parameter) -> dict:
"""
执行 EGM 计算的核心函数,可被外部调用。
假设参数已通过 para 全局对象设置好。
返回计算结果字典
Args:
para: 参数对象,包含所有计算所需的参数
Returns:
计算结果字典。
"""
parameter_display(para)
h_whole = para.h_arm[0] # 挂点高
@@ -379,7 +381,7 @@ def egm():
sys.exit(0)
logger.info(f"读取文件{toml_file_path}")
read_parameter(toml_file_path)
run_egm()
run_egm(para)
def speed():