Improve numerical issues

bioDS · Jul 25, 2024 · 6c83bf4 · 6c83bf4
1 parent d224eca
commit 6c83bf4
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/phylonco-beast/src/main/java/phylonco/beast/evolution/populationmodel/LogisticGrowth.java b/phylonco-beast/src/main/java/phylonco/beast/evolution/populationmodel/LogisticGrowth.java
@@ -85,16 +85,18 @@ public double getPopSize(double t) {
     @Override
     public double getIntensity(double t) {
         if (t == 0) return 0;
-        UnivariateFunction function = time -> 1 / getPopSize(time);
+        UnivariateFunction function = time -> 1 / Math.max(getPopSize(time), 1e-20);
         IterativeLegendreGaussIntegrator integrator = new IterativeLegendreGaussIntegrator(5, 1.0e-12, 1.0e-8, 2, 10000);
+        double intensity = 0;
         try {
-            return integrator.integrate(Integer.MAX_VALUE, function, 0, t);
+            intensity = integrator.integrate(100000, function, 0, t);
         } catch (TooManyEvaluationsException ex) {
-            System.err.println("Numerical Integration Error: " + ex.getMessage());
-            return 0;
+            return intensity;
         }
+        return intensity;
     }
 
+
     @Override
     public double getInverseIntensity(double x) {
         return 0;