feat:增加了根据跨膜压差修正泵功率的代码

models/uf-rl/env/uf_physics.py

@@ -363,8 +363,18 @@ class UFPhysicsModel:
         # 反洗时间（小时）
         t_bw_total_h = k_bw_per_ceb * t_bw_s / 3600.0
 
+        # 实际泵功率(kW)
+        simple_avg_tmp = (tmp_run_start + max_tmp_during_filtration) / 2
+        tmp_min = self.state_bounds.TMP0_min
+        tmp_max = self.state_bounds.global_TMP_hard_limit
+        simple_avg_tmp_clip = np.clip(simple_avg_tmp, tmp_min, tmp_max)
+
+        p_feed_kw_min = self.p.p_feed_kw_min
+        p_feed_kw_max = self.p.p_feed_kw_max
+        p_feed_kw = p_feed_kw_min + (simple_avg_tmp_clip - tmp_min) / (tmp_max - tmp_min) * (p_feed_kw_max - p_feed_kw_min)
+
         # 总能耗 (kWh)
-        E_total = t_feed_total_h * self.p.p_feed_kw+ t_bw_total_h * self.p.p_bw_kw
+        E_total = t_feed_total_h * p_feed_kw+ t_bw_total_h * self.p.p_bw_kw
 
         # 吨水电耗 (kWh/吨)
         ton_water_energy = E_total / max(V_net, 1e-12)