Mercurial > sypikslang

annotate gamelib/gamestate.py @ 80:a40a76012bd7

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Refactor message handling so we can deal with in the gui
author Neil Muller <drnlmuller@gmail.com>
date Wed, 09 May 2012 12:08:44 +0200
parents 364ff3479ef2
children 59afe9f92383
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
16
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
1 # -*- coding: utf-8 -*-
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
2 # vim:fileencoding=utf-8 ai ts=4 sts=4 et sw=4
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
3
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
4 """The actual game state object"""
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
5
45
1e8f7e694f0c Refactor missions and sciences a bit to reduce duplication.
Jeremy Thurgood <firxen@gmail.com>
parents: 42
diff changeset
6 from gamelib import missions, lab
1e8f7e694f0c Refactor missions and sciences a bit to reduce duplication.
Jeremy Thurgood <firxen@gmail.com>
parents: 42
diff changeset
7 from gamelib.game_base import get_subclasses
80
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
8 from gamelib.constants import NEW_SCIENCE, NEW_SCHEMATIC
16
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
9
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
10
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
11 class Game(object):
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
12
22
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
13 def __init__(self, init_data=None):
16
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
14 self.money = 1000
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
15 # Will be updated on the next turn
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
16 self.points = 0
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
17 self.reputation = 0
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
18 # Missions being attempted
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
19 self.cur_missions = []
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
20 # Science allocation for the current turn
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
21 self.cur_allocation = []
22
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
22 self.lab = None
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
23 self.missions = []
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
24 if init_data:
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
25 self._load_data(init_data)
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
26 else:
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
27 self.lab = lab.Lab()
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
28 # instantiate all the available missions
45
1e8f7e694f0c Refactor missions and sciences a bit to reduce duplication.
Jeremy Thurgood <firxen@gmail.com>
parents: 42
diff changeset
29 for cls in get_subclasses(missions.Mission):
42
47c7e96cf9c8 Change initial mission creation logic to look for leaf classes
Neil Muller <drnlmuller@gmail.com>
parents: 41
diff changeset
30 self.missions.append(cls())
16
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
31
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
32 def start_turn(self):
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
33 # Make more flexible?
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
34 self.points += 3
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
35 self.cur_missions = []
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
36 self.cur_allocation = []
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
37
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
38 def get_available_equipment(self):
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
39 """Return a list of equipment we can produce and afford"""
25
27570aca5d17 Fix some stupid bugs
Neil Muller <drnlmuller@gmail.com>
parents: 23
diff changeset
40 available = [x for x in self.lab.science
49
373c57ab4140 Product -> Schematic.
Jeremy Thurgood <firxen@gmail.com>
parents: 45
diff changeset
41 if x.SCIENCE_TYPE == 'schematic' and x.COST <= self.money]
16
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
42 return available
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
43
33
12c33aac7684 Gate certain missions by reputation.
Jeremy Thurgood <firxen@gmail.com>
parents: 27
diff changeset
44 def get_available_missions(self):
12c33aac7684 Gate certain missions by reputation.
Jeremy Thurgood <firxen@gmail.com>
parents: 27
diff changeset
45 """Return a list of missions we can feasibly attempt"""
41
e285b1e31a08 Add can_attempt method for future flexibility
Neil Muller <drnlmuller@gmail.com>
parents: 33
diff changeset
46 available = [x for x in self.missions if x.can_attempt(self)]
33
12c33aac7684 Gate certain missions by reputation.
Jeremy Thurgood <firxen@gmail.com>
parents: 27
diff changeset
47 return available
12c33aac7684 Gate certain missions by reputation.
Jeremy Thurgood <firxen@gmail.com>
parents: 27
diff changeset
48
63
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 49
diff changeset
49 def get_available_points(self):
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 49
diff changeset
50 return self.points - len(self.cur_allocation)
364ff3479ef2 Remove duplicate point tracking. Rename classes for future refactoring
Neil Muller <drnlmuller@gmail.com>
parents: 49
diff changeset
51
16
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
52 def end_turn(self):
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
53 # Attempt the missions
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
54 mission_results = []
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
55 for mission, equipment in self.cur_missions:
25
27570aca5d17 Fix some stupid bugs
Neil Muller <drnlmuller@gmail.com>
parents: 23
diff changeset
56 mission_results.append(mission.attempt(equipment, self))
16
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
57 # Do the science
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
58 self.points -= len(self.cur_allocation)
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
59 if self.points < 0:
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
60 raise RuntimeError('Spent too many points')
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
61 new_stuff = self.lab.spend_points(self.cur_allocation, self.points)
27
4d9610c59efa Reset points after spending them.
Jeremy Thurgood <firxen@gmail.com>
parents: 25
diff changeset
62 self.points = 0
16
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
63 # Process mission results
80
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
64 messages = []
16
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
65 for result in mission_results:
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
66 result.apply(self)
80
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
67 messages.append((result.outcome, result.msg))
16
af1bfeb648cb Sketch in gamestate
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
68 for science in new_stuff:
23
f6a3b213857b Fix up some game state logic, add very basic REPL game interface.
Jeremy Thurgood <firxen@gmail.com>
parents: 22
diff changeset
69 # FIXME: Update UI better.
80
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
70 if science.SCIENCE_TYPE == 'research':
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
71 messages.append((NEW_SCIENCE,
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
72 "You've started a new line of research in %s"
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
73 % science.NAME))
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
74 else:
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
75 messages.append((NEW_SCHEMATIC,
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
76 "You've developed blueprints for a %s" % science.NAME))
a40a76012bd7 Refactor message handling so we can deal with in the gui
Neil Muller <drnlmuller@gmail.com>
parents: 63
diff changeset
77 return messages
22
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
78
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
79 def save_data(self):
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
80 """Serialize the game state into a dict"""
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
81 data = {}
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
82 data['money'] = self.money
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
83 data['reputation'] = self.reputation
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
84 data['points'] = self.points
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
85 data['lab'] = self.lab.serialize()
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
86 # Save mission state
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
87 data['missions'] = {}
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
88 for mission in self.missions:
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
89 miss_name = type(mission).__name__
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
90 data['missions'][miss_name] = mission.save_data()
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
91 return data
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
92
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
93 def _load_data(self, data):
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
94 """Restore the game state"""
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
95 self.money = data['money']
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
96 self.reputation = data['reputation']
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
97 self.points = data['points']
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
98 self.lab = lab.Lab(data['lab'])
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
99 for mis_class in missions.Mission.__subclasses__():
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
100 miss_name = mis_class.__name__
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
101 if miss_name in data['missions']:
296ce36fa7d9 Serialize and unserialize game state and missions
Neil Muller <drnlmuller@gmail.com>
parents: 16
diff changeset
102 self.missions.append(mis_class(data['missions'][miss_name]))