Handle case when guess is NaN

haskell · Mar 20, 2018 · 51ccc98 · 51ccc98
1 parent 50de88c
commit 51ccc98
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/Numeric/RootFinding.hs b/Numeric/RootFinding.hs
@@ -370,9 +370,10 @@ newtonRaphsonIterations (lo,guess,hi) function
     (flo,_) = function lo
     (fhi,_) = function hi
     -- Ensure that initial guess is within bracket
-    guess'  | guess < lo && guess < hi = (lo + hi) / 2
-            | guess > lo && guess > hi = (lo + hi) / 2
-            | otherwise                = guess
+    guess'
+      | guess >= lo && guess <= hi = guess
+      | guess >= hi && guess <= lo = guess
+      | otherwise                  = (lo + hi) / 2
     --
     go xA x xB
       | f  == 0   = [NewtonRoot x]