refactor: 将参数对象作为参数传递给run_egm函数
修改run_egm函数,使其接收参数对象而不是依赖全局变量,并更新相关调用代码
This commit is contained in:
10
main.py
10
main.py
@@ -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():
|
||||
|
||||
Reference in New Issue
Block a user