X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=src%2FPoint.hs;h=0e29d596e583786c99ca16e7a2922db4a616fdef;hb=7e59859d69d9c5bbe5d654566adea71265447756;hp=e5260ae06a943505e907439a4bff81ee7bbe03b8;hpb=248880a645548e5d1910814dd9023e2af80e16ac;p=spline3.git diff --git a/src/Point.hs b/src/Point.hs index e5260ae..0e29d59 100644 --- a/src/Point.hs +++ b/src/Point.hs @@ -1,34 +1,47 @@ {-# LANGUAGE FlexibleInstances #-} -module Point +module Point ( + Point(..), + dot, + scale ) where -import Comparisons +import Test.QuickCheck ( Arbitrary(..) ) -type Point = (Double, Double, Double) +-- | Represents a point in three dimensions. We use a custom type (as +-- opposed to a 3-tuple) so that we can make the coordinates strict. +-- +data Point = + Point !Double !Double !Double + deriving (Eq, Show) + + +instance Arbitrary Point where + arbitrary = do + (x,y,z) <- arbitrary + return $ Point x y z + instance Num Point where - (x1,y1,z1) + (x2,y2,z2) = (x1+x2, y1+y2, z1+z2) - (x1,y1,z1) - (x2,y2,z2) = (x1-x2, y1-y2, z1-z2) - (x1,y1,z1) * (x2,y2,z2) = (x1*x2, y1*y2, z1*z2) - abs (x, y, z) = (abs x, abs y, abs z) - signum (x, y, z) = (signum x, signum y, signum z) - fromInteger n = (fromInteger n, fromInteger n, fromInteger n) + (Point x1 y1 z1) + (Point x2 y2 z2) = Point (x1+x2) (y1+y2) (z1+z2) + (Point x1 y1 z1) - (Point x2 y2 z2) = Point (x1-x2) (y1-y2) (z1-z2) + (Point x1 y1 z1) * (Point x2 y2 z2) = Point (x1*x2) (y1*y2) (z1*z2) + abs (Point x y z) = Point (abs x) (abs y) (abs z) + signum (Point x y z) = Point (signum x) (signum y) (signum z) + fromInteger n = + Point coord coord coord + where + coord = fromInteger n -- | Scale a point by a constant. scale :: Point -> Double -> Point -scale (x, y, z) d = (x*d, y*d, z*d) - - --- | Returns the distance between p1 and p2. -distance :: Point -> Point -> Double -distance (x1, y1, z1) (x2, y2, z2) = - sqrt $ (x2 - x1)^(2::Int) + (y2 - y1)^(2::Int) + (z2 - z1)^(2::Int) +scale (Point x y z) d = Point (x*d) (y*d) (z*d) --- | Returns 'True' if p1 is close to (within 'epsilon' of) p2, --- 'False' otherwise. -is_close :: Point -> Point -> Bool -is_close p1 p2 = (distance p1 p2) ~= 0 +-- | Returns the dot product of two points (taken as three-vectors). +{-# INLINE dot #-} +dot :: Point -> Point -> Double +dot (Point x1 y1 z1) (Point x2 y2 z2) = + (x2 - x1)^(2::Int) + (y2 - y1)^(2::Int) + (z2 - z1)^(2::Int)