rinkhals: gamelib/buildings.py annotate

annotate gamelib/buildings.py @ 64:99fbb652ce8d

Refactor buildings so that new ones can be added just by adding a class to buildings.py.

author	Simon Cross <hodgestar@gmail.com>
date	Mon, 31 Aug 2009 20:25:11 +0000
parents	1047ccd22dac
children	7e9c8ad06d32

rev	line source
43 fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	1 """Classes for various buildings in the game."""
fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	2
fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	3 from pgu.vid import Sprite
fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	4
fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	5 import imagecache
57 08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	6 import tiles
43 fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	7
fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	8 class Building(Sprite):
fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	9 """Base class for buildings"""
fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	10
57 08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	11 GRASSLAND = tiles.REVERSE_TILE_MAP['grassland']
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	12
64 99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	13 def __init__(self, pos):
57 08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	14 """Initial image, tile vid position, size and tile number for building."""
64 99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	15 self.day_image = imagecache.load_image(self.IMAGE)
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	16 self.night_image = imagecache.load_image(self.IMAGE, ('night',))
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	17 self.pos = pos
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	18 self.size = self.SIZE
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	19 self.tile_no = self.TILE_NO
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	20 self._buy_price = self.BUY_PRICE
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	21
43 fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	22 # Create the building somewhere far off screen
64 99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	23 Sprite.__init__(self, self.day_image, (-1000, -1000))
57 08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	24
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	25 def tile_positions(self):
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	26 """Return pairs of (x, y) tile positions for each of the tile positions
64 99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	27 occupied by the building.
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	28 """
57 08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	29 xpos, ypos = self.pos
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	30 xsize, ysize = self.size
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	31
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	32 for dx in xrange(xsize):
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	33 for dy in xrange(ysize):
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	34 yield (xpos + dx, ypos + dy)
43 fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	35
fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	36 def loop(self, tv, _sprite):
fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	37 ppos = tv.tile_to_view(self.pos)
fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	38 self.rect.x = ppos[0]
fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	39 self.rect.y = ppos[1]
fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	40
fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	41 def move(self, state):
fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	42 """Given the game state, return a new position for the object"""
fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	43 # Default is not to move
fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	44 return self.pos
fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	45
57 08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	46 def place(self, tv):
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	47 """Check that the building can be placed at its current position
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	48 and place it if possible.
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	49 """
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	50 xpos, ypos = self.pos
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	51 xsize, ysize = self.size
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	52
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	53 # check that all spaces under the structure are grassland
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	54 for tile_pos in self.tile_positions():
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	55 if not tv.get(tile_pos) == self.GRASSLAND:
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	56 return False
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	57
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	58 # place tile
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	59 for tile_pos in self.tile_positions():
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	60 tv.set(tile_pos, self.tile_no)
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	61
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	62 return True
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	63
64 99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	64 def buy_price(self):
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	65 return self._buy_price
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	66
57 08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	67 def sun(self, sun_on):
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	68 if sun_on:
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	69 self.setimage(self.day_image)
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	70 else:
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	71 self.setimage(self.night_image)
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	72
64 99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	73
43 fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	74 class HenHouse(Building):
fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	75 """A HenHouse."""
fb5be14ea930 Start of building sprites. Simon Cross <hodgestar@gmail.com> parents: diff changeset	76
64 99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	77 TILE_NO = tiles.REVERSE_TILE_MAP['henhouse']
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	78 BUY_PRICE = 100
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	79 SIZE = (3, 2)
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	80 IMAGE = 'sprites/henhouse.png'
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	81 NAME = 'Hen House'
57 08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	82
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	83
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	84 class GuardTower(Building):
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	85 """A GuardTower."""
08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	86
64 99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	87 TILE_NO = tiles.REVERSE_TILE_MAP['guardtower']
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	88 BUY_PRICE = 200
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	89 SIZE = (2, 2)
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	90 IMAGE = 'sprites/watchtower.png'
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	91 NAME = 'Watch Tower'
57 08665fa60345 Implement henhouses and henhouse adding. Simon Cross <hodgestar@gmail.com> parents: 43 diff changeset	92
64 99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	93 def is_building(obj):
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	94 """Return true if obj is a build class."""
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	95 return hasattr(obj, "NAME")
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	96
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	97 BUILDINGS = []
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	98 for name in dir():
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	99 obj = eval(name)
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	100 try:
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	101 if is_building(obj):
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	102 BUILDINGS.append(obj)
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	103 except TypeError:
99fbb652ce8d Refactor buildings so that new ones can be added just by adding a class to buildings.py. Simon Cross <hodgestar@gmail.com> parents: 63 diff changeset	104 pass

Mercurial > rinkhals

annotate gamelib/buildings.py @ 64:99fbb652ce8d