diff --git a/main.py b/main.py index e751948..003c2ca 100644 --- a/main.py +++ b/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(): diff --git a/webview_app.py b/webview_app.py index 90eb6f2..58c8a44 100644 --- a/webview_app.py +++ b/webview_app.py @@ -183,7 +183,7 @@ class EGMWebApp: logger.info("开始执行 EGM 计算...") # 调用 main.py 的核心计算函数 - result = run_egm() + result = run_egm(para) self.add_log("info", "EGM 计算完成")