simple-vec3-0.1.0.0: src/Data/Vec3/TUnboxed.hs

{-# LANGUAGE GeneralizedNewtypeDeriving #-}
{-# LANGUAGE TypeFamilies #-}

module Data.Vec3.TUnboxed
    ( TUVec3(..)
    )

where

import Prelude hiding (reverse)

import Data.Vector.Unboxed as VU
import Data.Vector.Generic as VG
import Data.Vector.Generic.Mutable as VG

import Data.Vec3.Class


-- | 'Vec3' implementation with 'Data.Vector.Unboxed.Unbox' instance
-- based on tuples, suitable for use with "Data.Vector.Unboxed".
--
-- This represents 3-vector as a triple of doubles, using the default
-- Unbox instance for tuples as provided by "Data.Vector.Unboxed",
-- which wraps a vector of tuples as a tuple of vectors.
--
-- @
-- interface:  [d1 (x, y, z); d2 (x, y, z) ...], length = N
--                  |  |  |       |  |  |
-- storage(x): [d1x-+  |  | ; d2x-+  |  |  ...], length = N
-- storage(y): [d1y----+  | ; d2y----+  |  ...], length = N
-- storage(z): [d1z-------+ ; d2z-------+  ...], length = N
-- @
newtype TUVec3 = TUVec3 (Double, Double, Double)
                deriving (Eq, Show,
                          VG.Vector VU.Vector,
                          VG.MVector VU.MVector,
                          VU.Unbox)


instance Vec3 TUVec3 where
    newtype Matrix TUVec3 = TUMatrix (TUVec3, TUVec3, TUVec3)
                            deriving (Eq, Show,
                                      VG.Vector VU.Vector,
                                      VG.MVector VU.MVector,
                                      VU.Unbox)


    fromXYZ v = TUVec3 v
    {-# INLINE fromXYZ #-}

    toXYZ (TUVec3 v) = v
    {-# INLINE toXYZ #-}

    fromRows (r1, r2, r3) = TUMatrix (r1, r2, r3)
    {-# INLINE fromRows #-}

    toRows (TUMatrix (r1, r2, r3)) = (r1, r2, r3)
    {-# INLINE toRows #-}