X-Git-Url: https://git.ctpug.org.za/?p=naja.git;a=blobdiff_plain;f=naja%2Factions.py;h=e1ec1e0a40b2cf58bbc741e0fcdd71cbc966602f;hp=c96a219a6ec28e9382d0abc81c88d8736f7498b1;hb=0dcb7a00f2ad80c4b544e3e04dce4b008dac1ad7;hpb=bd20e189a992894f8c20226b127fdec426de6425 diff --git a/naja/actions.py b/naja/actions.py index c96a219..e1ec1e0 100644 --- a/naja/actions.py +++ b/naja/actions.py @@ -1,6 +1,6 @@ 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): @@ -16,6 +16,15 @@ 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() @@ -71,6 +80,10 @@ class LocationAction(object): '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." @@ -86,8 +99,7 @@ class LoseHealthOrMSB(LocationAction): 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): @@ -98,6 +110,23 @@ class SetBits(LocationAction): board.player.bits.set_bits(location.bitwise_operand) +class ClearBits(LocationAction): + TEXT = "Clear %(location_bits)s." + GLYPHS = (ACTION_GLYPHS.CLEAR_BITS,) + + def perform_action(self, board, location): + 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,) @@ -106,6 +135,91 @@ class ToggleBits(LocationAction): 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,) @@ -142,6 +256,16 @@ class AcquireWinToken(LocationAction): ])) +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}." GLYPHS = (ACTION_GLYPHS.HEAL,) @@ -169,7 +293,15 @@ class ShiftLocations(LocationAction): def perform_action(self, board, location): sound.play_sound('grind.ogg') - board.shift_locations(self.data['direction']) + board.shift_locations( + self.data['direction'], + self.data.get('skip_current', True)) + if self.data.get('move_player', False): + pos = { + 'NORTH': (0, -1), 'SOUTH': (0, 1), + 'EAST': (1, 0), 'WEST': (-1, 0), + }.get(self.data['direction'], (0, 0)) + board.player.force_position(pos, delta=True) class RotateLocations(LocationAction): @@ -192,7 +324,9 @@ class AllowChessMove(LocationAction): class AllowChessMoveIfMSB(LocationAction): - TEXT = "Clear {MSB} and move like a %(chesspiece_name)s for one turn." + TEXT = ( + "If {MSB} is set, unset {MSB} and move like a " + "%(chesspiece_name)s for one turn. Otherwise do nothing.") MSB_GLYPH = ACTION_GLYPHS.MOVEMENT def perform_action(self, board, location):