+from naja.constants import BITS
class PlayerBits(object):
# Operate on sets of bits
- def check_bits(self, *bits):
+ def check_bits(self, bits):
return all(self.check_bit(bit) for bit in bits)
- def set_bits(self, *bits):
+ def set_bits(self, bits):
for bit in bits:
self.set_bit(bit)
- def clear_bits(self, *bits):
+ def clear_bits(self, bits):
for bit in bits:
self.clear_bit(bit)
- def toggle_bits(self, *bits):
+ def toggle_bits(self, bits):
for bit in bits:
self.toggle_bit(bit)
def __init__(self, bits, position):
self.bits = PlayerBits(bits)
self.position = position
+
+ @classmethod
+ def import_player(cls, definition):
+ return cls(definition['bits'], tuple(definition['position']))
+
+ def export(self):
+ return {
+ 'bits': self.bits.bits,
+ 'position': list(self.position),
+ }
+
+ def move(self, direction):
+ if not self.bits.check_bit(direction):
+ return False
+ # TODO: Something cleaner than this.
+ x, y = self.position
+ if direction == BITS.NORTH:
+ if y > 0 and self.bits.check_bit(BITS.NORTH):
+ self.position = (x, y - 1)
+ return True
+ elif direction == BITS.SOUTH:
+ if y < 4 and self.bits.check_bit(BITS.SOUTH):
+ self.position = (x, y + 1)
+ return True
+ elif direction == BITS.EAST:
+ if x < 4 and self.bits.check_bit(BITS.EAST):
+ self.position = (x + 1, y)
+ return True
+ elif direction == BITS.WEST:
+ if x > 0 and self.bits.check_bit(BITS.WEST):
+ self.position = (x - 1, y)
+ return True
+
+ return False