module Numeric.DataFrame.IO
( MutableFrame (), IODataFrame ()
, newDataFrame, copyDataFrame, copyMutableDataFrame
, unsafeFreezeDataFrame
, freezeDataFrame, thawDataFrame
, writeDataFrame, readDataFrame
, writeDataFrameOff, readDataFrameOff
) where
import GHC.Prim (RealWorld)
import GHC.Types (Int (..), IO (..))
import Numeric.Commons
import Numeric.DataFrame.Type
import Numeric.DataFrame.Mutable
import Numeric.Dimensions
import Numeric.Scalar
newtype IODataFrame t (ns :: [Nat]) = IODataFrame (MDataFrame RealWorld t (ns :: [Nat]))
newDataFrame :: forall t (ns :: [Nat])
. ( PrimBytes t, Dimensions ns)
=> IO (IODataFrame t ns)
newDataFrame = IODataFrame <$> IO (newDataFrame# @t @ns)
copyDataFrame :: forall t (as :: [Nat]) (bs :: [Nat]) (asbs :: [Nat])
. ( PrimBytes (DataFrame t as)
, ConcatList as bs asbs
, Dimensions bs
)
=> DataFrame t as -> Idx bs -> IODataFrame t asbs -> IO ()
copyDataFrame df ei (IODataFrame mdf) = IO (copyDataFrame# df ei mdf)
copyMutableDataFrame :: forall t (as :: [Nat]) (bs :: [Nat]) (asbs :: [Nat])
. ( PrimBytes t
, ConcatList as bs asbs
, Dimensions bs
)
=> IODataFrame t as -> Idx bs -> IODataFrame t asbs -> IO ()
copyMutableDataFrame (IODataFrame mdfA) ei (IODataFrame mdfB)
= IO (copyMDataFrame# mdfA ei mdfB)
unsafeFreezeDataFrame :: forall t (ns :: [Nat])
. PrimBytes (DataFrame t ns)
=> IODataFrame t ns -> IO (DataFrame t ns)
unsafeFreezeDataFrame (IODataFrame mdf) = IO (unsafeFreezeDataFrame# mdf)
freezeDataFrame :: forall t (ns :: [Nat])
. PrimBytes (DataFrame t ns)
=> IODataFrame t ns -> IO (DataFrame t ns)
freezeDataFrame (IODataFrame mdf) = IO (freezeDataFrame# mdf)
thawDataFrame :: forall t (ns :: [Nat])
. PrimBytes (DataFrame t ns)
=> DataFrame t ns -> IO (IODataFrame t ns)
thawDataFrame df = IODataFrame <$> IO (thawDataFrame# df)
writeDataFrame :: forall t (ns :: [Nat])
. ( MutableFrame t ns, Dimensions ns )
=> IODataFrame t ns -> Idx ns -> Scalar t -> IO ()
writeDataFrame (IODataFrame mdf) ei = IO . writeDataFrame# mdf ei . unScalar
readDataFrame :: forall t (ns :: [Nat])
. ( MutableFrame t ns, Dimensions ns )
=> IODataFrame t ns -> Idx ns -> IO (Scalar t)
readDataFrame (IODataFrame mdf) = fmap scalar . IO . readDataFrame# mdf
writeDataFrameOff :: forall t (ns :: [Nat])
. ( MutableFrame t ns, Dimensions ns )
=> IODataFrame t ns -> Int -> Scalar t -> IO ()
writeDataFrameOff (IODataFrame mdf) (I# i) x = IO $ \s -> (# writeDataFrameOff# mdf i (unScalar x) s, () #)
readDataFrameOff :: forall t (ns :: [Nat])
. ( MutableFrame t ns, Dimensions ns )
=> IODataFrame t ns -> Int -> IO (Scalar t)
readDataFrameOff (IODataFrame mdf) (I# i) = scalar <$> IO (readDataFrameOff# mdf i)