1 from naja.constants import ACTION_GLYPHS, BITS, CHESS_PIECES
2 from naja.sound import sound
3 from naja.utils import bit_glyphs, move_glyph, parse_bits
6 class LocationAction(object):
8 An action that may be performed on a location.
15 def __init__(self, required_bits, **data):
16 self.required_bits = required_bits
22 def get_msb_glyph(self):
25 def get_text(self, location=None):
26 substitutions = self.data.copy()
28 if 'shift' in self.data:
29 substitutions['shift'] = self.data['shift']
30 substitutions['shift_glyph'] = ('{SHIFT_%s}'
31 % self.data['direction'].upper())
32 elif 'direction' in self.data:
33 substitutions['rowcol'] = {
38 }[self.data['direction']]
39 substitutions['direction'] = '{%s}' % (substitutions['direction'],)
41 if 'chesspiece' in self.data:
42 substitutions['chesspiece_name'] = move_glyph(
43 self.data['chesspiece'])
45 if 'rot_direction' in self.data:
46 substitutions['rot_direction_name'] = '{%s}' % (
47 substitutions['rot_direction'],)
50 substitutions['location_bits'] = 'bits specified by this tile'
52 substitutions['location_bits'] = bit_glyphs(
53 location.bitwise_operand)
56 if self.data.get('message', None) is not None:
57 text = self.data['message']
59 return text % substitutions
61 def check_available(self, player):
62 return player.bits.check_bits(self.required_bits)
64 def perform_action(self, board, location):
65 raise NotImplementedError(
66 "%s does not implement perform_action()." % (type(self).__name__,))
68 def check_and_clear_MSB(self, player):
69 if player.bits.check_bit(BITS.MSB):
70 player.bits.clear_bit(BITS.MSB)
76 return {'required_bits': list(self.required_bits),
78 'action_class': self.__class__.__name__}
81 class DoNothing(LocationAction):
83 GLYPHS = (ACTION_GLYPHS.NOTHING,)
85 def perform_action(self, board, location):
89 class LoseHealthOrMSB(LocationAction):
90 TEXT = "Lose {HEALTH} or {MSB}."
91 MSB_GLYPH = ACTION_GLYPHS.DAMAGE
93 def perform_action(self, board, location):
94 if not self.check_and_clear_MSB(board.player):
95 sound.play_sound('awwww.ogg')
99 class SetBits(LocationAction):
100 TEXT = "Set %(location_bits)s."
101 GLYPHS = (ACTION_GLYPHS.SET_BITS,)
103 def perform_action(self, board, location):
104 board.player.bits.set_bits(location.bitwise_operand)
107 class ClearBits(LocationAction):
108 TEXT = "Clear %(location_bits)s."
109 GLYPHS = (ACTION_GLYPHS.CLEAR_BITS,)
111 def perform_action(self, board, location):
112 board.player.bits.clear_bits(location.bitwise_operand)
115 class ToggleBits(LocationAction):
116 TEXT = "Toggle %(location_bits)s."
117 GLYPHS = (ACTION_GLYPHS.TOGGLE_BITS,)
119 def perform_action(self, board, location):
120 board.player.bits.toggle_bits(location.bitwise_operand)
123 class GenericBits(LocationAction):
124 GLYPHS = (ACTION_GLYPHS.SET_BITS, ACTION_GLYPHS.CLEAR_BITS)
126 def __init__(self, *args, **kw):
127 super(GenericBits, self).__init__(*args, **kw)
128 self.set_bits = parse_bits(self.data.get('set', []))
129 self.clear_bits = parse_bits(self.data.get('clear', []))
130 self.toggle_bits = parse_bits(self.data.get('toggle', []))
132 def perform_action(self, board, location):
133 bits = board.player.bits
134 bits.set_bits(self.set_bits)
135 bits.toggle_bits(self.toggle_bits)
136 bits.clear_bits(self.clear_bits)
138 def get_glyphs(self):
141 glyphs.append(ACTION_GLYPHS.SET_BITS)
143 glyphs.append(ACTION_GLYPHS.CLEAR_BITS)
145 glyphs.append(ACTION_GLYPHS.TOGGLE_BITS)
148 def get_text(self, location=None):
149 if 'message' in self.data:
150 return super(GenericBits, self).get_text()
152 for template, bits in [
153 ('Set %s.', self.set_bits), ('Clear %s.', self.clear_bits),
154 ('Toggle %s', self.toggle_bits)]:
156 parts.append(template % (bit_glyphs(bits)))
157 return " ".join(parts)
160 class ShiftBits(LocationAction):
161 TEXT = "Barrel-shift player bits %(shift_glyph)s %(shift)s."
162 GLYPHS = (ACTION_GLYPHS.SHIFT_LEFT,)
164 def perform_action(self, board, location):
165 shift = self.data['shift']
166 if self.data['direction'] == 'left':
167 board.player.bits.shift_bits_left(shift)
169 board.player.bits.shift_bits_right(shift)
172 class LoseHealthOrMSBAndSetBits(LocationAction):
173 TEXT = "Lose {HEALTH} or {MSB}, then set %(location_bits)s."
174 GLYPHS = (ACTION_GLYPHS.SET_BITS,)
175 MSB_GLYPH = ACTION_GLYPHS.DAMAGE
177 def perform_action(self, board, location):
178 if not self.check_and_clear_MSB(board.player):
179 sound.play_sound('awwww.ogg')
181 board.player.bits.set_bits(location.bitwise_operand)
184 class AcquireWinToken(LocationAction):
185 TEXT = "Gain {WINTOKEN}, then clear {RED,GREEN,BLUE}."
186 GLYPHS = (ACTION_GLYPHS.WINTOKEN,)
188 def perform_action(self, board, location):
189 sound.play_sound('yipee.ogg')
190 board.acquire_win_token()
191 board.player.bits.clear_bits(set([
192 BITS.RED, BITS.GREEN, BITS.BLUE,
194 if self.data.get('once', False):
195 location.actions.remove(self)
198 class GainHealth(LocationAction):
199 TEXT = "Gain {HEALTH}."
200 GLYPHS = (ACTION_GLYPHS.HEAL,)
202 def perform_action(self, board, location):
203 sound.play_sound('aha.ogg')
207 class GainHealthAndClearBitsOrMSB(LocationAction):
208 TEXT = "Gain {HEALTH}, then clear %(location_bits)s or {MSB}."
209 GLYPHS = (ACTION_GLYPHS.HEAL,)
210 MSB_GLYPH = ACTION_GLYPHS.CLEAR_BITS
212 def perform_action(self, board, location):
213 sound.play_sound('aha.ogg')
215 if not self.check_and_clear_MSB(board.player):
216 board.player.bits.clear_bits(location.bitwise_operand)
219 class ShiftLocations(LocationAction):
220 TEXT = "Shift current %(rowcol)s %(direction)s."
221 GLYPHS = (ACTION_GLYPHS.CHANGE_BOARD,)
223 def perform_action(self, board, location):
224 sound.play_sound('grind.ogg')
225 board.shift_locations(self.data['direction'])
228 class RotateLocations(LocationAction):
229 TEXT = "Rotate adjacent tiles %(rot_direction_name)s."
230 GLYPHS = (ACTION_GLYPHS.CHANGE_BOARD,)
232 def perform_action(self, board, location):
233 sound.play_sound('grind.ogg')
234 board.rotate_locations(self.data['rot_direction'])
237 class AllowChessMove(LocationAction):
238 TEXT = "Move like a %(chesspiece_name)s for one turn."
239 GLYPHS = (ACTION_GLYPHS.MOVEMENT,)
241 def perform_action(self, board, location):
242 if self.data['chesspiece'] in CHESS_PIECES:
243 chesspiece = CHESS_PIECES[self.data['chesspiece']]
244 board.allow_chess_move(chesspiece)
247 class AllowChessMoveIfMSB(LocationAction):
249 "Clear {MSB} and move like a %(chesspiece_name)s for one turn if it "
251 MSB_GLYPH = ACTION_GLYPHS.MOVEMENT
253 def perform_action(self, board, location):
254 if self.data['chesspiece'] in CHESS_PIECES:
255 if self.check_and_clear_MSB(board.player):
256 chesspiece = CHESS_PIECES[self.data['chesspiece']]
257 board.allow_chess_move(chesspiece)
260 class GainMSB(LocationAction):
262 GLYPHS = (ACTION_GLYPHS.MSB,)
264 def perform_action(self, board, location):
265 board.player.bits.set_bit(BITS.MSB)