{-# LANGUAGE ImportQualifiedPost #-}
module Data.Array.Mixed.Internal.Arith where

import Data.Array.Internal qualified as OI
import Data.Array.Internal.RankedG qualified as RG
import Data.Array.Internal.RankedS qualified as RS

import Data.Array.Strided qualified as AS


fromO :: RS.Array n a -> AS.Array n a
fromO (RS.A (RG.A sh (OI.T strides offset vec))) = AS.Array sh strides offset vec

toO :: AS.Array n a -> RS.Array n a
toO (AS.Array sh strides offset vec) = RS.A (RG.A sh (OI.T strides offset vec))

liftO1 :: (AS.Array n a -> AS.Array n' b)
       -> RS.Array n a -> RS.Array n' b
liftO1 f = toO . f . fromO

liftO2 :: (AS.Array n a -> AS.Array n1 b -> AS.Array n2 c)
       -> RS.Array n a -> RS.Array n1 b -> RS.Array n2 c
liftO2 f x y = toO (f (fromO x) (fromO y))