duxingchen/WQ_GUI
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

测试修改

Browse Source
This commit is contained in:
DXC
2026-06-25 15:50:02 +08:00
parent 67aaaaa6b2
commit 43f50ec07b
16 changed files with 1098 additions and 1029 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/utils/band_math.py
View File

@ -99,8 +99,10 @@ class BandMathCalculator:
# 【新增安全防护】引入 numpy 命名空间，让 eval 引擎安全识别 nan 与 inf
import numpy as np
try:
# 即使 calc_expression 含有纯字符 nan，也能被 np.nan 安全接管
result = eval(calc_expression, {"__builtins__": None}, {"nan": np.nan, "inf": np.inf, "np": np})
# 【P0 修复】包 np.errstate 抑制除零 / 无效操作产生的 RuntimeWarning 洪水
with np.errstate(divide='ignore', invalid='ignore'):
# 即使 calc_expression 含有纯字符 nan，也能被 np.nan 安全接管
result = eval(calc_expression, {"__builtins__": None}, {"nan": np.nan, "inf": np.inf, "np": np})
except Exception as e:
print(f"⚠️ 警告：公式计算异常 ({e})，该点赋值为 nan")
result = np.nan

4
src/utils/water_index.py
View File

@ -117,7 +117,9 @@ class WaterQualityIndexCalculator:
calc_expr,
)
try:
r = eval(calc_expr, {"__builtins__": None}, {"nan": np.nan, "inf": np.inf, "np": np})
# 【P0 修复】包 np.errstate 抑制除零 / 无效操作产生的 RuntimeWarning 洪水
with np.errstate(divide='ignore', invalid='ignore'):
r = eval(calc_expr, {"__builtins__": None}, {"nan": np.nan, "inf": np.inf, "np": np})
except Exception:
r = np.nan
results.append(r)