Eq and Ord instances
[darcs-mirrors/vector.git] / Data / Vector / Unboxed.hs
index bd2250d..703ce5f 100644 (file)
@@ -1,10 +1,22 @@
 {-# LANGUAGE MagicHash, UnboxedTuples, FlexibleInstances, MultiParamTypeClasses #-}
 
+-- |
+-- Module      : Data.Vector.Unboxed
+-- Copyright   : (c) Roman Leshchinskiy 2008
+-- License     : BSD-style
+--
+-- Maintainer  : rl@cse.unsw.edu.au
+-- Stability   : experimental
+-- Portability : non-portable
+-- 
+-- Unboxed vectors based on 'Unbox'.
+--
+
 module Data.Vector.Unboxed (
-  Vector(..), module Data.Vector.Base
+  Vector(..), module Data.Vector.IVector
 ) where
 
-import           Data.Vector.Base
+import           Data.Vector.IVector
 import qualified Data.Vector.Unboxed.Mutable as Mut
 import           Data.Vector.Unboxed.Unbox
 
@@ -18,17 +30,15 @@ data Vector a = Vector {-# UNPACK #-} !Int
                        {-# UNPACK #-} !Int
                                       ByteArray#
 
-instance Unbox a => Base Vector a where
-  {-# INLINE create #-}
-  create init = runST (do
-      Mut.Vector i n marr# <- init
-      ST (\s# -> case unsafeFreezeByteArray# marr# s# of
-                   (# s2#, arr# #) -> (# s2#, Vector i n arr# #)
-         )
-    )
+instance Unbox a => IVector Vector a where
+  {-# INLINE vnew #-}
+  vnew init = runST (do
+                       Mut.Vector i n marr# <- init
+                       ST (\s# -> case unsafeFreezeByteArray# marr# s# of
+                            (# s2#, arr# #) -> (# s2#, Vector i n arr# #)))
 
-  {-# INLINE length #-}
-  length (Vector _ n _) = n
+  {-# INLINE vlength #-}
+  vlength (Vector _ n _) = n
 
   {-# INLINE unsafeSlice #-}
   unsafeSlice (Vector i _ arr#) j n = Vector (i+j) n arr#
@@ -36,3 +46,11 @@ instance Unbox a => Base Vector a where
   {-# INLINE unsafeIndex #-}
   unsafeIndex (Vector (I# i#) _ arr#) (I# j#) f = f (at# arr# (i# +# j#))
 
+instance (Unbox a, Eq a) => Eq (Vector a) where
+  {-# INLINE (==) #-}
+  (==) = eq
+
+instance (Unbox a, Ord a) => Ord (Vector a) where
+  {-# INLINE compare #-}
+  compare = cmp
+