module Data.Array.Accelerate.Linear.V2 (
V2(..), R1(..), R2(..),
_yx,
ex, ey,
perp, angle,
) where
import Data.Array.Accelerate as A
import Data.Array.Accelerate.Smart
import Data.Array.Accelerate.Product
import Data.Array.Accelerate.Array.Sugar
import Data.Array.Accelerate.Linear.Lift
import Data.Array.Accelerate.Linear.Metric
import Data.Array.Accelerate.Linear.Type
import Data.Array.Accelerate.Linear.V1
import Data.Array.Accelerate.Linear.Vector
import Control.Lens
import Linear.V2 ( V2(..) )
import qualified Linear.V2 as L
import Prelude as P
perp :: forall a. A.Num a => Exp (V2 a) -> Exp (V2 a)
perp = lift1 (L.perp :: V2 (Exp a) -> V2 (Exp a))
angle :: A.Floating a => Exp a -> Exp (V2 a)
angle = lift . L.angle
class (L.R2 t, R1 t) => R2 t where
_y :: (Elt a, Box t a) => Lens' (Exp (t a)) (Exp a)
_y = liftLens (L._y :: Lens' (t (Exp a)) (Exp a))
_xy :: (Elt a, Box t a) => Lens' (Exp (t a)) (Exp (V2 a))
_xy = liftLens (L._xy :: Lens' (t (Exp a)) (V2 (Exp a)))
_yx :: forall t a. (R2 t, Elt a, Box t a) => Lens' (Exp (t a)) (Exp (V2 a))
_yx = liftLens (L._yx :: Lens' (t (Exp a)) (V2 (Exp a)))
ey :: R2 t => E t
ey = E _y
instance Metric V2
instance Additive V2
instance R1 V2
instance R2 V2
type instance EltRepr (V2 a) = EltRepr (a, a)
instance Elt a => Elt (V2 a) where
eltType _ = eltType (undefined :: (a,a))
toElt p = case toElt p of
(x, y) -> V2 x y
fromElt (V2 x y) = fromElt (x, y)
instance cst a => IsProduct cst (V2 a) where
type ProdRepr (V2 a) = ProdRepr (a,a)
fromProd p (V2 x y) = fromProd p (x,y)
toProd p t = case toProd p t of
(x, y) -> V2 x y
prod p _ = prod p (undefined :: (a,a))
instance (Lift Exp a, Elt (Plain a)) => Lift Exp (V2 a) where
type Plain (V2 a) = V2 (Plain a)
lift (V2 x y) = Exp $ Tuple $ NilTup `SnocTup` lift x `SnocTup` lift y
instance Elt a => Unlift Exp (V2 (Exp a)) where
unlift t = V2 (Exp $ SuccTupIdx ZeroTupIdx `Prj` t)
(Exp $ ZeroTupIdx `Prj` t)
instance A.Num a => P.Num (Exp (V2 a)) where
(+) = lift2 ((+) :: V2 (Exp a) -> V2 (Exp a) -> V2 (Exp a))
() = lift2 (() :: V2 (Exp a) -> V2 (Exp a) -> V2 (Exp a))
(*) = lift2 ((*) :: V2 (Exp a) -> V2 (Exp a) -> V2 (Exp a))
negate = lift1 (negate :: V2 (Exp a) -> V2 (Exp a))
signum = lift1 (signum :: V2 (Exp a) -> V2 (Exp a))
abs = lift1 (signum :: V2 (Exp a) -> V2 (Exp a))
fromInteger x = lift (fromInteger x :: V2 (Exp a))
instance A.Floating a => P.Fractional (Exp (V2 a)) where
(/) = lift2 ((/) :: V2 (Exp a) -> V2 (Exp a) -> V2 (Exp a))
recip = lift1 (recip :: V2 (Exp a) -> V2 (Exp a))
fromRational x = lift (fromRational x :: V2 (Exp a))
instance A.Floating a => P.Floating (Exp (V2 a)) where
pi = lift (pi :: V2 (Exp a))
log = lift1 (log :: V2 (Exp a) -> V2 (Exp a))
exp = lift1 (exp :: V2 (Exp a) -> V2 (Exp a))
sin = lift1 (sin :: V2 (Exp a) -> V2 (Exp a))
cos = lift1 (cos :: V2 (Exp a) -> V2 (Exp a))
tan = lift1 (tan :: V2 (Exp a) -> V2 (Exp a))
sinh = lift1 (sinh :: V2 (Exp a) -> V2 (Exp a))
cosh = lift1 (cosh :: V2 (Exp a) -> V2 (Exp a))
tanh = lift1 (tanh :: V2 (Exp a) -> V2 (Exp a))
asin = lift1 (asin :: V2 (Exp a) -> V2 (Exp a))
acos = lift1 (acos :: V2 (Exp a) -> V2 (Exp a))
atan = lift1 (atan :: V2 (Exp a) -> V2 (Exp a))
asinh = lift1 (asinh :: V2 (Exp a) -> V2 (Exp a))
acosh = lift1 (acosh :: V2 (Exp a) -> V2 (Exp a))
atanh = lift1 (atanh :: V2 (Exp a) -> V2 (Exp a))
instance (Elt a, Elt b) => Each (Exp (V2 a)) (Exp (V2 b)) (Exp a) (Exp b) where
each = liftLens (each :: Traversal (V2 (Exp a)) (V2 (Exp b)) (Exp a) (Exp b))