X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=src%2FBit.hs;h=5c8c5aa25903ba4332db63faaf55ccfd0e2e0f44;hb=e1d16438b44ecd962565756a828c9ed8014cf894;hp=d7fb679cfcaf079ebcf349b955d09a5b22da1f30;hpb=609ae9195c946f0be8d5105d8a590f9b32941ef7;p=hath.git

diff --git a/src/Bit.hs b/src/Bit.hs
index d7fb679..5c8c5aa 100644
--- a/src/Bit.hs
+++ b/src/Bit.hs
@@ -1,22 +1,35 @@
-module Bit where
+-- | The Bit module contains the Bit data type, which is essentially a
+--   renamed Boolean, and some convenience functions.
+module Bit
+where
 
-data Bit = None | Zero | One
-         deriving (Eq)
+import Test.QuickCheck (
+  Arbitrary,
+  arbitrary,
+  elements
+  )
+
+
+data Bit = Zero | One
+  deriving (Eq)
 
 instance Show Bit where
-    show None = "None"
-    show Zero = "0"
-    show One  = "1"
+  show Zero = "0"
+  show One  = "1"
+
+
+instance Arbitrary Bit where
+  arbitrary = elements [ Zero, One ]
+
 
-                
+-- | Convert a Bit to an Int.
 bit_to_int :: Bit -> Int
-bit_to_int None = -1
 bit_to_int Zero =  0
 bit_to_int One  =  1
 
--- If we are passed a '0' or '1', convert it appropriately. Otherwise,
--- default to None.
-bit_from_char :: Char -> Bit
-bit_from_char '0' = Zero
-bit_from_char '1' = One
-bit_from_char  _  = None
+-- | If we are passed a '0' or '1', convert it
+--   appropriately. Otherwise, return Nothing.
+bit_from_char :: Char -> Maybe Bit
+bit_from_char '0' = Just Zero
+bit_from_char '1' = Just One
+bit_from_char  _  = Nothing