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', []))
131 self.once = self.data.get('once', False)
132 self.acquire_win = self.data.get('acquire_win', False)
133 self.lose_health = self.data.get('lose_health', False)
135 def perform_action(self, board, location):
136 bits = board.player.bits
137 bits.set_bits(self.set_bits)
138 bits.toggle_bits(self.toggle_bits)
139 bits.clear_bits(self.clear_bits)
141 sound.play_sound('yipee.ogg')
142 board.acquire_win_token()
144 sound.play_sound('awwww.ogg')
147 location.actions.remove(self)
149 def get_glyphs(self):
152 glyphs.append(ACTION_GLYPHS.WINTOKEN)
154 glyphs.append(ACTION_GLYPHS.DAMAGE)
156 glyphs.append(ACTION_GLYPHS.SET_BITS)
158 glyphs.append(ACTION_GLYPHS.CLEAR_BITS)
160 glyphs.append(ACTION_GLYPHS.TOGGLE_BITS)
163 def get_text(self, location=None):
164 if 'message' in self.data:
165 return super(GenericBits, self).get_text()
168 parts.append("Gain a {WINTOKEN}.")
170 parts.append("Lose {HEALTH}.")
171 for template, bits in [
172 ('Set %s.', self.set_bits), ('Clear %s.', self.clear_bits),
173 ('Toggle %s', self.toggle_bits)]:
175 parts.append(template % (bit_glyphs(bits)))
177 parts.append('Usable once only.')
178 return " ".join(parts)
181 class ShiftBits(LocationAction):
182 TEXT = "Barrel-shift player bits %(shift_glyph)s %(shift)s."
183 GLYPHS = (ACTION_GLYPHS.SHIFT_LEFT,)
185 def perform_action(self, board, location):
186 shift = self.data['shift']
187 if self.data['direction'] == 'left':
188 board.player.bits.shift_bits_left(shift)
190 board.player.bits.shift_bits_right(shift)
193 class LoseHealthOrMSBAndSetBits(LocationAction):
194 TEXT = "Lose {HEALTH} or {MSB}, then set %(location_bits)s."
195 GLYPHS = (ACTION_GLYPHS.SET_BITS,)
196 MSB_GLYPH = ACTION_GLYPHS.DAMAGE
198 def perform_action(self, board, location):
199 if not self.check_and_clear_MSB(board.player):
200 sound.play_sound('awwww.ogg')
202 board.player.bits.set_bits(location.bitwise_operand)
205 class AcquireWinToken(LocationAction):
206 TEXT = "Gain {WINTOKEN}, then clear {RED,GREEN,BLUE}."
207 GLYPHS = (ACTION_GLYPHS.WINTOKEN,)
209 def perform_action(self, board, location):
210 sound.play_sound('yipee.ogg')
211 board.acquire_win_token()
212 board.player.bits.clear_bits(set([
213 BITS.RED, BITS.GREEN, BITS.BLUE,
217 class GainHealth(LocationAction):
218 TEXT = "Gain {HEALTH}."
219 GLYPHS = (ACTION_GLYPHS.HEAL,)
221 def perform_action(self, board, location):
222 sound.play_sound('aha.ogg')
226 class GainHealthAndClearBitsOrMSB(LocationAction):
227 TEXT = "Gain {HEALTH}, then clear %(location_bits)s or {MSB}."
228 GLYPHS = (ACTION_GLYPHS.HEAL,)
229 MSB_GLYPH = ACTION_GLYPHS.CLEAR_BITS
231 def perform_action(self, board, location):
232 sound.play_sound('aha.ogg')
234 if not self.check_and_clear_MSB(board.player):
235 board.player.bits.clear_bits(location.bitwise_operand)
238 class ShiftLocations(LocationAction):
239 TEXT = "Shift current %(rowcol)s %(direction)s."
240 GLYPHS = (ACTION_GLYPHS.CHANGE_BOARD,)
242 def perform_action(self, board, location):
243 sound.play_sound('grind.ogg')
244 board.shift_locations(self.data['direction'])
247 class RotateLocations(LocationAction):
248 TEXT = "Rotate adjacent tiles %(rot_direction_name)s."
249 GLYPHS = (ACTION_GLYPHS.CHANGE_BOARD,)
251 def perform_action(self, board, location):
252 sound.play_sound('grind.ogg')
253 board.rotate_locations(self.data['rot_direction'])
256 class AllowChessMove(LocationAction):
257 TEXT = "Move like a %(chesspiece_name)s for one turn."
258 GLYPHS = (ACTION_GLYPHS.MOVEMENT,)
260 def perform_action(self, board, location):
261 if self.data['chesspiece'] in CHESS_PIECES:
262 chesspiece = CHESS_PIECES[self.data['chesspiece']]
263 board.allow_chess_move(chesspiece)
266 class AllowChessMoveIfMSB(LocationAction):
268 "Clear {MSB} and move like a %(chesspiece_name)s for one turn if it "
270 MSB_GLYPH = ACTION_GLYPHS.MOVEMENT
272 def perform_action(self, board, location):
273 if self.data['chesspiece'] in CHESS_PIECES:
274 if self.check_and_clear_MSB(board.player):
275 chesspiece = CHESS_PIECES[self.data['chesspiece']]
276 board.allow_chess_move(chesspiece)
279 class GainMSB(LocationAction):
281 GLYPHS = (ACTION_GLYPHS.MSB,)
283 def perform_action(self, board, location):
284 board.player.bits.set_bit(BITS.MSB)