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_text(self, location=None):
139 if 'message' in self.data:
140 return super(GenericBits, self).get_text()
142 for template, bits in [
143 ('Set %s.', self.set_bits), ('Clear %s.', self.clear_bits),
144 ('Toggle %s', self.toggle_bits)]:
146 parts.append(template % (bit_glyphs(bits)))
147 return " ".join(parts)
150 class ShiftBits(LocationAction):
151 TEXT = "Barrel-shift player bits %(shift_glyph)s %(shift)s."
152 GLYPHS = (ACTION_GLYPHS.SHIFT_LEFT,)
154 def perform_action(self, board, location):
155 shift = self.data['shift']
156 if self.data['direction'] == 'left':
157 board.player.bits.shift_bits_left(shift)
159 board.player.bits.shift_bits_right(shift)
162 class LoseHealthOrMSBAndSetBits(LocationAction):
163 TEXT = "Lose {HEALTH} or {MSB}, then set %(location_bits)s."
164 GLYPHS = (ACTION_GLYPHS.SET_BITS,)
165 MSB_GLYPH = ACTION_GLYPHS.DAMAGE
167 def perform_action(self, board, location):
168 if not self.check_and_clear_MSB(board.player):
169 sound.play_sound('awwww.ogg')
171 board.player.bits.set_bits(location.bitwise_operand)
174 class AcquireWinToken(LocationAction):
175 TEXT = "Gain {WINTOKEN}, then clear {RED,GREEN,BLUE}."
176 GLYPHS = (ACTION_GLYPHS.WINTOKEN,)
178 def perform_action(self, board, location):
179 sound.play_sound('yipee.ogg')
180 board.acquire_win_token()
181 board.player.bits.clear_bits(set([
182 BITS.RED, BITS.GREEN, BITS.BLUE,
184 if self.data.get('once', False):
185 location.actions.remove(self)
188 class GainHealth(LocationAction):
189 TEXT = "Gain {HEALTH}."
190 GLYPHS = (ACTION_GLYPHS.HEAL,)
192 def perform_action(self, board, location):
193 sound.play_sound('aha.ogg')
197 class GainHealthAndClearBitsOrMSB(LocationAction):
198 TEXT = "Gain {HEALTH}, then clear %(location_bits)s or {MSB}."
199 GLYPHS = (ACTION_GLYPHS.HEAL,)
200 MSB_GLYPH = ACTION_GLYPHS.CLEAR_BITS
202 def perform_action(self, board, location):
203 sound.play_sound('aha.ogg')
205 if not self.check_and_clear_MSB(board.player):
206 board.player.bits.clear_bits(location.bitwise_operand)
209 class ShiftLocations(LocationAction):
210 TEXT = "Shift current %(rowcol)s %(direction)s."
211 GLYPHS = (ACTION_GLYPHS.CHANGE_BOARD,)
213 def perform_action(self, board, location):
214 sound.play_sound('grind.ogg')
215 board.shift_locations(self.data['direction'])
218 class RotateLocations(LocationAction):
219 TEXT = "Rotate adjacent tiles %(rot_direction_name)s."
220 GLYPHS = (ACTION_GLYPHS.CHANGE_BOARD,)
222 def perform_action(self, board, location):
223 sound.play_sound('grind.ogg')
224 board.rotate_locations(self.data['rot_direction'])
227 class AllowChessMove(LocationAction):
228 TEXT = "Move like a %(chesspiece_name)s for one turn."
229 GLYPHS = (ACTION_GLYPHS.MOVEMENT,)
231 def perform_action(self, board, location):
232 if self.data['chesspiece'] in CHESS_PIECES:
233 chesspiece = CHESS_PIECES[self.data['chesspiece']]
234 board.allow_chess_move(chesspiece)
237 class AllowChessMoveIfMSB(LocationAction):
239 "Clear {MSB} and move like a %(chesspiece_name)s for one turn if it "
241 MSB_GLYPH = ACTION_GLYPHS.MOVEMENT
243 def perform_action(self, board, location):
244 if self.data['chesspiece'] in CHESS_PIECES:
245 if self.check_and_clear_MSB(board.player):
246 chesspiece = CHESS_PIECES[self.data['chesspiece']]
247 board.allow_chess_move(chesspiece)
250 class GainMSB(LocationAction):
252 GLYPHS = (ACTION_GLYPHS.MSB,)
254 def perform_action(self, board, location):
255 board.player.bits.set_bit(BITS.MSB)