self.required_bits = frozenset(bits)
self.data = data
+ def get_text(self):
+ substitutions = self.data.copy()
+ if 'direction' in self.data:
+ substitutions['rowcol'] = {
+ 'NORTH': 'column',
+ 'SOUTH': 'column',
+ 'EAST': 'row',
+ 'WEST': 'row',
+ }[self.data['direction']]
+ return self.TEXT % substitutions
+
def check_available(self, player):
return player.bits.check_bits(self.required_bits)
class AcquireWinToken(LocationAction):
- TEXT = "Acquire a win token, then clear all high bits."
+ TEXT = "Acquire a win token, then clear all key bits."
+
+ def perform_action(self, board, location):
+ board.acquire_win_token()
+ board.player.bits.clear_bits(set([
+ BITS.RED, BITS.GREEN, BITS.BLUE,
+ ]))
+
+
+class GainHealthAndClearBitsOrMSB(LocationAction):
+ TEXT = "Gain health, then clear bits specified by this location or MSB."
USES_MSB = True
def perform_action(self, board, location):
- if self.check_and_clear_MSB(board.player):
- board.acquire_win_token()
- board.player.bits.clear_bits(set([
- BITS.CYAN, BITS.MAGENTA, BITS.YELLOW,
- ]))
+ board.gain_health()
+ if not self.check_and_clear_MSB(board.player):
+ board.player.bits.clear_bits(location.bitwise_operand)
+
+
+class ShiftLocations(LocationAction):
+ TEXT = "Shift current %(rowcol)s %(direction)s."
+
+ def perform_action(self, board, location):
+ board.shift_locations(self.data['direction'])