rinkhals: gamelib/gameboard.py comparison

Sanity check and fix up buildings on map after loading.

author	Simon Cross <hodgestar@gmail.com>
date	Mon, 31 Aug 2009 21:22:10 +0000
parents	edc15ce8fa30
children	6b8ac424da83

comparison

equal deleted inserted replaced

-:edc15ce8fa30
+:9171d9b9ab35
 self.foxes = []
 self.buildings = []
 self.cash = 0
 self.add_cash(constants.STARTING_CASH)
+self.fix_buildings()
 def create_disp(self):
 width, height = pygame.display.get_surface().get_size()
 tbl = gui.Table()
 tbl.tr()
 self.toolbar = ToolBar(self)
 self.tv.sprites.remove(building)
 def add_cash(self, amount):
 self.cash += amount
 self.toolbar.update_cash_counter(self.cash)
+def fix_buildings(self):
+"""Go through the level map looking for buildings that haven't
+been added to self.buildings and adding them.
+Where partial buildings exist (i.e. places where the building
+cannot fit on the available tiles) the building is added anyway
+to the top left corner.
+Could be a lot faster.
+"""
+tile_to_building = dict((b.TILE_NO, b) for b in buildings.BUILDINGS)
+print tile_to_building
+w, h = self.tv.size
+for x in xrange(w):
+for y in xrange(h):
+tile_pos = (x, y)
+tile_no = self.tv.get(tile_pos)
+if tile_no not in tile_to_building:
+continue
+covered = False
+for building in self.buildings:
+if building.covers(tile_pos):
+covered = True
+break
+if covered:
+continue
+building_cls = tile_to_building[tile_no]
+building = building_cls(tile_pos)
+building.remove(self.tv)
+building.place(self.tv)
+self.add_building(building)

Mercurial > rinkhals