from naja.constants import ACTION_GLYPHS, BITS, CHESS_PIECES
from naja.sound import sound
-from naja.utils import bit_glyphs, move_glyph
+from naja.utils import bit_glyphs, move_glyph, parse_bits
class LocationAction(object):
self.required_bits = required_bits
self.data = data
+ def sanity_check(self, location):
+ pass
+
+ def get_glyphs(self):
+ return self.GLYPHS
+
+ def get_msb_glyph(self):
+ return self.MSB_GLYPH
+
def get_text(self, location=None):
substitutions = self.data.copy()
'data': self.data,
'action_class': self.__class__.__name__}
+ def take_damage(self, board):
+ sound.play_sound('awwww.ogg')
+ board.lose_health()
+
class DoNothing(LocationAction):
TEXT = "No effect."
def perform_action(self, board, location):
if not self.check_and_clear_MSB(board.player):
- sound.play_sound('awwww.ogg')
- board.lose_health()
+ self.take_damage(board)
class SetBits(LocationAction):
board.player.bits.clear_bits(location.bitwise_operand)
+class ClearBitsAndHealth(LocationAction):
+ TEXT = "Clear %(location_bits)s and lose {HEALTH}."
+ GLYPHS = (ACTION_GLYPHS.CLEAR_BITS, ACTION_GLYPHS.DAMAGE)
+
+ def perform_action(self, board, location):
+ board.player.bits.clear_bits(location.bitwise_operand)
+ self.take_damage(board)
+
+
class ToggleBits(LocationAction):
TEXT = "Toggle %(location_bits)s."
GLYPHS = (ACTION_GLYPHS.TOGGLE_BITS,)
board.player.bits.toggle_bits(location.bitwise_operand)
+class ToggleBitsAndHarm(LocationAction):
+ TEXT = "Toggle %(location_bits)s and lose {HEALTH}."
+ GLYPHS = (ACTION_GLYPHS.TOGGLE_BITS, ACTION_GLYPHS.DAMAGE)
+
+ def perform_action(self, board, location):
+ board.player.bits.toggle_bits(location.bitwise_operand)
+ self.take_damage(board)
+
+
+class GenericBits(LocationAction):
+ GLYPHS = (ACTION_GLYPHS.SET_BITS, ACTION_GLYPHS.CLEAR_BITS)
+
+ def __init__(self, *args, **kw):
+ super(GenericBits, self).__init__(*args, **kw)
+ self.set_bits = parse_bits(self.data.get('set', []))
+ self.clear_bits = parse_bits(self.data.get('clear', []))
+ self.toggle_bits = parse_bits(self.data.get('toggle', []))
+ self.once = self.data.get('once', False)
+ self.acquire_win = self.data.get('acquire_win', False)
+ self.lose_health = self.data.get('lose_health', False)
+ self.gain_health = self.data.get('gain_health', False)
+
+ def sanity_check(self, location):
+ missing_bits = set()
+ missing_bits.update(self.set_bits - set(location.bitwise_operand))
+ missing_bits.update(self.clear_bits - set(location.bitwise_operand))
+ missing_bits.update(self.toggle_bits - set(location.bitwise_operand))
+ if missing_bits:
+ raise ValueError(
+ "Location %s missing bits %r"
+ % (location.card_name, sorted(list(missing_bits))))
+
+ def perform_action(self, board, location):
+ bits = board.player.bits
+ bits.set_bits(self.set_bits)
+ bits.toggle_bits(self.toggle_bits)
+ bits.clear_bits(self.clear_bits)
+ if self.acquire_win:
+ sound.play_sound('yipee.ogg')
+ board.acquire_win_token()
+ if self.lose_health:
+ sound.play_sound('awwww.ogg')
+ board.lose_health()
+ if self.gain_health:
+ sound.play_sound('aha.ogg')
+ board.gain_health()
+ if self.once:
+ location.actions.remove(self)
+
+ def get_glyphs(self):
+ glyphs = []
+ if self.acquire_win:
+ glyphs.append(ACTION_GLYPHS.WINTOKEN)
+ if self.lose_health:
+ glyphs.append(ACTION_GLYPHS.DAMAGE)
+ if self.gain_health:
+ glyphs.append(ACTION_GLYPHS.HEAL)
+ if self.set_bits:
+ glyphs.append(ACTION_GLYPHS.SET_BITS)
+ if self.clear_bits:
+ glyphs.append(ACTION_GLYPHS.CLEAR_BITS)
+ if self.toggle_bits:
+ glyphs.append(ACTION_GLYPHS.TOGGLE_BITS)
+ return tuple(glyphs)
+
+ def get_text(self, location=None):
+ if 'message' in self.data:
+ return super(GenericBits, self).get_text()
+ parts = []
+ if self.acquire_win:
+ parts.append("Gain {WINTOKEN}.")
+ if self.lose_health:
+ parts.append("Lose {HEALTH}.")
+ if self.gain_health:
+ parts.append("Gain {HEALTH}.")
+ for template, bits in [
+ ('Set %s.', self.set_bits), ('Clear %s.', self.clear_bits),
+ ('Toggle %s.', self.toggle_bits)]:
+ if bits:
+ parts.append(template % (bit_glyphs(bits)))
+ if self.once:
+ parts.append('Usable once only.')
+ return " ".join(parts)
+
+
class ShiftBits(LocationAction):
TEXT = "Barrel-shift player bits %(shift_glyph)s %(shift)s."
GLYPHS = (ACTION_GLYPHS.SHIFT_LEFT,)
BITS.RED, BITS.GREEN, BITS.BLUE,
]))
+class AcquireWinTokenAndLoseHealth(AcquireWinToken):
+ TEXT = "Gain {WINTOKEN}, lose {HEALTH}, then clear {RED,GREEN,BLUE}."
+ GLYPHS = (ACTION_GLYPHS.WINTOKEN, ACTION_GLYPHS.DAMAGE)
+
+ def perform_action(self, board, location):
+ self.take_damage(board)
+ super(AcquireWinTokenAndLoseHealth, self).perform_action(board,
+ location)
class GainHealth(LocationAction):
TEXT = "Gain {HEALTH}."