1 files changed, 18 insertions, 16 deletions

diff --git a/packages/sundials/src/Main.hs b/packages/sundials/src/Main.hs
index d1f6755..5e51372 100644
--- a/packages/sundials/src/Main.hs
+++ b/packages/sundials/src/Main.hs
@@ -1,5 +1,6 @@
-module Main where
+{-# OPTIONS_GHC -Wall #-}
 
+import qualified Data.Vector.Storable as V
 import           Numeric.Sundials.Arkode.ODE
 import           Numeric.LinearAlgebra
 
@@ -8,10 +9,11 @@ import qualified Diagrams.Prelude as D
 import           Diagrams.Backend.Rasterific
 
 import           Control.Lens
+import           Data.List (zip4)
 
-brusselator :: a -> [Double] -> [Double]
-brusselator _t x = [ a - (w + 1) * u + v * u^(2::Int)
-                   , w * u - v * u^(2::Int)
+
+brusselator _t x = [ a - (w + 1) * u + v * u^2
+                   , w * u - v * u^2
                    , (b - w) / eps - w * u
                    ]
   where
@@ -22,7 +24,6 @@ brusselator _t x = [ a - (w + 1) * u + v * u^(2::Int)
     v = x !! 1
     w = x !! 2
 
-stiffish :: Double -> [Double] -> [Double]
 stiffish t v = [ lamda * u + 1.0 / (1.0 + t * t) - lamda * atan t ]
   where
     lamda = -100.0
@@ -44,14 +45,15 @@ kSaxis xs = P.r2Axis &~ do
 
 main :: IO ()
 main = do
-  do let res = odeSolve brusselator [1.2, 3.1, 3.0] (fromList [0.0, 0.1 .. 10.0])
-     putStrLn $ show res
-     renderRasterific "diagrams/brusselator.png"
-                      (D.dims2D 500.0 500.0)
-                      (renderAxis $ lSaxis $ [0.0, 0.1 .. 10.0]:(toLists $ tr res))
-
-  do let res = odeSolve stiffish [0.0] (fromList [0.0, 0.1 .. 10.0])
-     putStrLn $ show res
-     renderRasterific "diagrams/stiffish.png"
-                      (D.dims2D 500.0 500.0)
-                      (renderAxis $ kSaxis $ zip [0.0, 0.1 .. 10.0] (concat $ toLists res))
+  let res = odeSolve brusselator [1.2, 3.1, 3.0] (fromList [0.0, 0.1 .. 10.0])
+  putStrLn $ show res
+  renderRasterific "diagrams/brusselator.png"
+                   (D.dims2D 500.0 500.0)
+                   (renderAxis $ lSaxis $ [0.0, 0.1 .. 10.0]:(toLists $ tr res))
+
+  let res = odeSolve stiffish [0.0] (fromList [0.0, 0.1 .. 10.0])
+  putStrLn $ show res
+  renderRasterific "diagrams/stiffish.png"
+                   (D.dims2D 500.0 500.0)
+                   (renderAxis $ kSaxis $ zip [0.0, 0.1 .. 10.0] (concat $ toLists res))
+