# Operate on individual bits
def check_bit(self, bit):
- return bool(self.bits & bit)
+ return bool(self.bits & (1 << bit))
def set_bit(self, bit):
- self.bits |= bit
+ self.bits |= (1 << bit)
def clear_bit(self, bit):
- self.bits &= (0xff ^ bit)
+ self.bits &= (0xff ^ (1 << bit))
def toggle_bit(self, bit):
- self.bits ^= bit
+ self.bits ^= (1 << bit)
# Operate on sets of bits
# TODO: Something cleaner than this.
x, y = self.position
if direction == BITS.NORTH:
- if y > 0:
+ if y > 0 and self.bits.check_bit(BITS.NORTH):
self.position = (x, y - 1)
return True
elif direction == BITS.SOUTH:
- if y < 4:
+ if y < 4 and self.bits.check_bit(BITS.SOUTH):
self.position = (x, y + 1)
return True
elif direction == BITS.EAST:
- if x < 4:
+ if x < 4 and self.bits.check_bit(BITS.EAST):
self.position = (x + 1, y)
return True
elif direction == BITS.WEST:
- if x > 0:
+ if x > 0 and self.bits.check_bit(BITS.WEST):
self.position = (x - 1, y)
return True