aboutsummaryrefslogtreecommitdiffstats
path: root/fg21sim/extragalactic
diff options
context:
space:
mode:
authorAaron LI <aaronly.me@outlook.com>2017-01-08 14:05:28 +0800
committerAaron LI <aly@aaronly.me>2017-06-01 16:33:39 +0800
commit480ae1b0caa2b5d8260df21dc7f541e6c79fdec6 (patch)
treec39ade13692ac553094a5eff29e3495140cd3ea3 /fg21sim/extragalactic
parentce23ecbd28487824bbae5ae3d9f20b7781e6cd32 (diff)
downloadfg21sim-480ae1b0caa2b5d8260df21dc7f541e6c79fdec6.tar.bz2
solver.py: Avoid possible overflow when w is too large
Diffstat (limited to 'fg21sim/extragalactic')
-rw-r--r--fg21sim/extragalactic/clusters/solver.py10
1 files changed, 10 insertions, 0 deletions
diff --git a/fg21sim/extragalactic/clusters/solver.py b/fg21sim/extragalactic/clusters/solver.py
index 9d4a781..21b9a68 100644
--- a/fg21sim/extragalactic/clusters/solver.py
+++ b/fg21sim/extragalactic/clusters/solver.py
@@ -216,6 +216,16 @@ class FokkerPlanckSolver:
C_mhalf = self.X_mhalf(C)
w_phalf = dx_phalf * B_phalf / C_phalf
w_mhalf = dx_mhalf * B_mhalf / C_mhalf
+ # Avoid overflow when w is too large
+ w_max = 300
+ with np.errstate(invalid="ignore"):
+ mask_phalf = (np.abs(w_phalf) > w_max)
+ mask_mhalf = (np.abs(w_mhalf) > w_max)
+ w_phalf[mask_phalf] = w_max * (np.sign(w_phalf[mask_phalf]))
+ w_mhalf[mask_mhalf] = w_max * (np.sign(w_mhalf[mask_mhalf]))
+ logger.warning("Fixed too large w values: " +
+ "%d+%d" % (len(mask_phalf), len(mask_mhalf)))
+ #
W_phalf = self.W(w_phalf)
W_mhalf = self.W(w_mhalf)
Wplus_phalf = W_phalf * np.exp(w_phalf/2)