From 7f388d4f1955d2ce0e8995c38b2e2b7b89fb4f78 Mon Sep 17 00:00:00 2001
From: krasimir <krasimir@chalmers.se>
Date: Thu, 22 Jan 2009 15:01:41 +0000
Subject: [PATCH] optimized Data.Binary.Get.getWord8

---
 src/Data/Binary/Get.hs | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/Data/Binary/Get.hs b/src/Data/Binary/Get.hs
index d92567e45..bdc07d985 100644
--- a/src/Data/Binary/Get.hs
+++ b/src/Data/Binary/Get.hs
@@ -416,7 +416,15 @@ getPtr n = do
 
 -- | Read a Word8 from the monad state
 getWord8 :: Get Word8
-getWord8 = getPtr (sizeOf (undefined :: Word8))
+getWord8 = do
+    S s ss bytes <- get
+    case B.uncons s of
+      Just (w,rest) -> do put $! S rest ss (bytes + 1)
+                          return $! w
+      Nothing       -> case L.uncons ss of
+                         Just (w,rest) -> do put $! mkState rest (bytes + 1)
+                                             return $! w
+                         Nothing       -> fail "too few bytes"
 {-# INLINE getWord8 #-}
 
 -- | Read a Word16 in big endian format