import "Vector"

effect Python
    "Simantics/Python/Python"
    "org.simantics.pythonlink.PythonContext"

class PythonCompatible a where
    setPythonVariable :: String -> a -> <Python> ()
    getPythonVariable :: String -> <Python> a

importJava "org.simantics.pythonlink.NDArray" where
    data NDArray
    
    @JavaName "<init>"
    ndarray :: Vector Double -> NDArray
    @JavaName "<init>"
    ndarrayM :: Integer -> Integer -> Vector Double -> NDArray
    @JavaName "<init>"
    ndarrayN :: Vector Integer -> Vector Double -> NDArray
    
    @JavaName size
    ndarraySize :: NDArray -> Integer
    @JavaName dims
    ndarrayDims :: NDArray -> Vector Integer
    @JavaName getValues
    ndarrayValues :: NDArray -> Vector Double
    @JavaName getValue
    ndarrayElement :: NDArray -> Integer -> Double
    @JavaName getValue
    ndarrayElementM :: NDArray -> Integer -> Integer -> Double
    @JavaName getValue
    ndarrayElementN :: NDArray -> Vector Integer -> Double
    
    @JavaName toString
    ndarrayToString :: NDArray -> String
    @JavaName equals
    ndarrayEquals :: NDArray -> NDArray -> Boolean

instance Show NDArray where
    show = ndarrayToString

instance Eq NDArray where
    (==) = ndarrayEquals

importJava "org.simantics.pythonlink.PythonContext" where
    data PythonContext
    
    @JavaName close
    closePythonContext :: PythonContext -> <Proc> ()
    
    executePythonStatement :: String -> <Python> ()
    
    setPythonIntegerVariable :: String -> Integer -> <Python> ()
    setPythonIntegerArrayVariable :: String -> Vector Integer -> <Python> ()
    setPythonDoubleVariable :: String -> Double -> <Python> ()
    setPythonDoubleArrayVariable :: String -> Vector Double -> <Python> ()
    setPythonStringVariable :: String -> String -> <Python> ()
    setPythonStringArrayVariable :: String -> Vector String -> <Python> ()
    setPythonNDArrayVariable :: String -> NDArray -> <Python> ()
    
    getPythonIntegerVariable :: String -> <Python> Integer
    getPythonIntegerArrayVariable :: String -> <Python> Vector Integer
    getPythonDoubleVariable :: String -> <Python> Double
    getPythonDoubleArrayVariable :: String -> <Python> Vector Double
    getPythonStringVariable :: String -> <Python> String
    getPythonStringArrayVariable  :: String -> <Python> Vector String
    getPythonNDArrayVariable :: String -> <Python> NDArray
    
importJava "org.simantics.pythonlink.Python" where
    openPythonContext :: <Proc> PythonContext
    
    runPythonF :: (() -> <Python, e> a) -> <Proc, e> a
    runWithPythonContextF :: PythonContext -> (() -> <Python, e> a) -> <Proc, e> a

runPython :: (<Python, e> a) -> <Proc, e> a
runPython v = runPythonF (\_ -> v)

runWithPythonContext :: PythonContext -> (<Python, e> a) -> <Proc, e> a
runWithPythonContext python v = runWithPythonContextF python (\_ -> v) 

instance (VecComp a, PythonCompatible (Vector a)) => PythonCompatible [a] where
    setPythonVariable n v = setPythonVariable n (vector v)
    getPythonVariable = vectorToList . getPythonVariable
    
instance PythonCompatible Double where
    setPythonVariable = setPythonDoubleVariable
    getPythonVariable = getPythonDoubleVariable
    
instance PythonCompatible (Vector Double) where
    setPythonVariable = setPythonDoubleArrayVariable
    getPythonVariable = getPythonDoubleArrayVariable

instance PythonCompatible Integer where
    setPythonVariable = setPythonIntegerVariable
    getPythonVariable = getPythonIntegerVariable
    
instance PythonCompatible (Vector Integer) where
    setPythonVariable = setPythonIntegerArrayVariable
    getPythonVariable = getPythonIntegerArrayVariable
    
instance PythonCompatible String where
    setPythonVariable = setPythonStringVariable
    getPythonVariable = getPythonStringVariable
    
instance PythonCompatible (Vector String) where
    setPythonVariable = setPythonStringArrayVariable
    getPythonVariable = getPythonStringArrayVariable
    
instance PythonCompatible NDArray where
    setPythonVariable = setPythonNDArrayVariable
    getPythonVariable = getPythonNDArrayVariable