Add method for reading 1 bit flags.

Author: GraylinKim

sc2reader/decoders.py

@@ -4,6 +4,7 @@
 from cStringIO import StringIO
 
 import struct
+import functools
 
 class ByteDecoder(object):
 
@@ -119,6 +120,7 @@ def __init__(self, contents):
 
         # Reduce the number of lookups required to read
         self._read = self._buffer.read
+        self.read_bool = functools.partial(self.read_bits, 1)
 
     def done(self):
         """ Returns true when all bits in the buffer have been used"""
@@ -294,7 +296,7 @@ def read_struct(self, datatype=None):
         elif datatype == 0x01: # bitarray, weird alignment requirements
             bits = self.read_vint()
             data = self.read_bits(bits)
-            
+
         elif datatype == 0x02: # blob
             length = self.read_vint()
             data = self.read_bytes(length)
@@ -327,7 +329,6 @@ def read_struct(self, datatype=None):
             data = self.read_vint()
 
         else:
-            if debug: print prefix
             raise TypeError("Unknown Data Structure: '%s'" % datatype)
 
         return data