module Data.Array.Comfort.Storable.Unchecked.Monadic where import qualified Data.Array.Comfort.Shape as Shape import Data.Array.Comfort.Storable.Private (Array(Array)) import Foreign.Storable (Storable, ) import Foreign.Ptr (Ptr, ) import qualified Foreign.Marshal.Array.Guarded as Alloc import Control.Monad.Primitive (PrimMonad, unsafeIOToPrim) import Control.Monad (liftM) import Data.Tuple.HT (mapFst) unsafeCreate :: (PrimMonad m, Shape.C sh, Storable a) => sh -> (Ptr a -> IO ()) -> m (Array sh a) unsafeCreate :: forall (m :: * -> *) sh a. (PrimMonad m, C sh, Storable a) => sh -> (Ptr a -> IO ()) -> m (Array sh a) unsafeCreate sh sh Ptr a -> IO () f = forall (m :: * -> *) sh a. (PrimMonad m, C sh, Storable a) => sh -> (Int -> Ptr a -> IO ()) -> m (Array sh a) unsafeCreateWithSize sh sh forall a b. (a -> b) -> a -> b $ forall a b. a -> b -> a const Ptr a -> IO () f unsafeCreateWithSize :: (PrimMonad m, Shape.C sh, Storable a) => sh -> (Int -> Ptr a -> IO ()) -> m (Array sh a) unsafeCreateWithSize :: forall (m :: * -> *) sh a. (PrimMonad m, C sh, Storable a) => sh -> (Int -> Ptr a -> IO ()) -> m (Array sh a) unsafeCreateWithSize sh sh Int -> Ptr a -> IO () f = forall (m :: * -> *) a1 r. Monad m => (a1 -> r) -> m a1 -> m r liftM forall a b. (a, b) -> a fst forall a b. (a -> b) -> a -> b $ forall (m :: * -> *) sh a b. (PrimMonad m, C sh, Storable a) => sh -> (Int -> Ptr a -> IO b) -> m (Array sh a, b) unsafeCreateWithSizeAndResult sh sh Int -> Ptr a -> IO () f unsafeCreateWithSizeAndResult :: (PrimMonad m, Shape.C sh, Storable a) => sh -> (Int -> Ptr a -> IO b) -> m (Array sh a, b) unsafeCreateWithSizeAndResult :: forall (m :: * -> *) sh a b. (PrimMonad m, C sh, Storable a) => sh -> (Int -> Ptr a -> IO b) -> m (Array sh a, b) unsafeCreateWithSizeAndResult sh sh Int -> Ptr a -> IO b f = forall (m :: * -> *) a. PrimMonad m => IO a -> m a unsafeIOToPrim forall a b. (a -> b) -> a -> b $ let size :: Int size = forall sh. C sh => sh -> Int Shape.size sh sh in forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b fmap (forall a c b. (a -> c) -> (a, b) -> (c, b) mapFst (forall sh a. sh -> ForeignPtr a -> Array sh a Array sh sh)) forall a b. (a -> b) -> a -> b $ forall a b. Storable a => Int -> (Ptr a -> IO b) -> IO (ForeignPtr a, b) Alloc.create Int size forall a b. (a -> b) -> a -> b $ Int -> Ptr a -> IO b f Int size