1 files changed, 16 insertions, 5 deletions

diff --git a/packages/sundials/src/Main.hs b/packages/sundials/src/Main.hs
index fc48710..d1f35bd 100644
--- a/packages/sundials/src/Main.hs
+++ b/packages/sundials/src/Main.hs
@@ -21,6 +21,8 @@ import           System.IO.Unsafe (unsafePerformIO)
 
 import           Foreign.Storable (peekByteOff)
 
+import           Numeric.LinearAlgebra.HMatrix (Vector, Matrix)
+
 import qualified Types as T
 
 C.context (C.baseCtx <> C.vecCtx <> C.funCtx <> T.sunCtx)
@@ -91,10 +93,19 @@ brusselator _t x = V.fromList [ a - (w + 1) * u + v * u^2
     v = x V.! 1
     w = x V.! 2
 
-solveOdeC :: (CDouble -> V.Vector CDouble -> V.Vector CDouble) ->
-             V.Vector CDouble ->
-             Either CInt (V.Vector CDouble)
-solveOdeC fun f0 = unsafePerformIO $ do
+
+odeSolve :: (Double -> [Double] -> [Double]) -- ^ The RHS of the system \(\dot{y} = f(t,y)\)
+         -> [Double]                         -- ^ initial conditions
+         -> Vector Double                    -- ^ desired solution times
+         -> Matrix Double                    -- ^ solution
+odeSolve = undefined
+
+solveOdeC ::
+  (CDouble -> V.Vector CDouble -> V.Vector CDouble) -- ^ The RHS of the system \(\dot{y} = f(t,y)\)
+          -> V.Vector CDouble                       -- ^ Initial conditions
+          -> V.Vector CDouble                       -- ^ Desired solution times
+          -> Either CInt (V.Vector CDouble)         -- ^ Error code or solution
+solveOdeC fun f0 ts = unsafePerformIO $ do
   let dim = V.length f0
       nEq :: CLong
       nEq = fromIntegral dim
@@ -248,5 +259,5 @@ solveOdeC fun f0 = unsafePerformIO $ do
 
 main :: IO ()
 main = do
-  let res = solveOdeC (coerce brusselator) (V.fromList [1.2, 3.1, 3.0])
+  let res = solveOdeC (coerce brusselator) (V.fromList [1.2, 3.1, 3.0]) undefined
   putStrLn $ show res