linear_algebra_3d_vector.hs

import Data.Vector.Unboxed as VU

-- linear algebra
-- add
(<+>) :: Vector Double -> Vector Double -> Vector Double
(<+>) x y = VU.zipWith (+) x y

-- sub
(<->) :: Vector Double -> Vector Double -> Vector Double
(<->) x y = VU.zipWith (-) x y

-- dot product
(.*) :: Vector Double -> Vector Double -> Double
(.*) x y = VU.foldl1 (+) $ VU.zipWith (*) x y

-- cross product
(/*) :: Vector Double -> Vector Double -> Vector Double
(/*) x y = VU.zipWith (-) a' b'
    where
        i = fromList [1, 2, 0]
        z = fromList [0, 0, 0]
        a = VU.accumulate (*) x $ VU.zip i y
        b = VU.accumulate (*) y $ VU.zip i x
        a' = VU.accumulate (+) z $ VU.zip i a
        b' = VU.accumulate (+) z $ VU.zip i b