Package horizons :: Package ai :: Package aiplayer :: Package building :: Module firestation
[hide private]
[frames] | no frames]

Source Code for Module horizons.ai.aiplayer.building.firestation

 1  # ################################################### 
 2  # Copyright (C) 2008-2017 The Unknown Horizons Team 
 3  # team@unknown-horizons.org 
 4  # This file is part of Unknown Horizons. 
 5  # 
 6  # Unknown Horizons is free software; you can redistribute it and/or modify 
 7  # it under the terms of the GNU General Public License as published by 
 8  # the Free Software Foundation; either version 2 of the License, or 
 9  # (at your option) any later version. 
10  # 
11  # This program is distributed in the hope that it will be useful, 
12  # but WITHOUT ANY WARRANTY; without even the implied warranty of 
13  # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the 
14  # GNU General Public License for more details. 
15  # 
16  # You should have received a copy of the GNU General Public License 
17  # along with this program; if not, write to the 
18  # Free Software Foundation, Inc., 
19  # 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA 
20  # ################################################### 
21   
22  from horizons.ai.aiplayer.basicbuilder import BasicBuilder 
23  from horizons.ai.aiplayer.building import AbstractBuilding 
24  from horizons.ai.aiplayer.buildingevaluator import BuildingEvaluator 
25  from horizons.ai.aiplayer.constants import BUILDING_PURPOSE 
26  from horizons.constants import BUILDINGS 
27 28 29 -class AbstractFireStation(AbstractBuilding):
30 - def iter_potential_locations(self, settlement_manager):
31 spots_in_settlement = settlement_manager.settlement.buildability_cache.cache[(2, 2)] 32 village_builder = settlement_manager.village_builder 33 for coords in village_builder.special_building_assignments[BUILDING_PURPOSE.FIRE_STATION].keys(): 34 if coords not in spots_in_settlement or village_builder.plan[coords][1][0] > village_builder.current_section: 35 continue 36 object = settlement_manager.settlement.ground_map[coords].object 37 if object is None or object.buildable_upon: 38 yield (coords[0], coords[1], 0)
39 40 @property
41 - def producer_building(self):
42 """Fire stations don't produce any resources.""" 43 return False
44 45 @property
46 - def evaluator_class(self):
48 49 @classmethod
50 - def register_buildings(cls):
52
53 54 -class FireStationEvaluator(BuildingEvaluator):
55 need_collector_connection = False 56 record_plan_change = False 57 58 @classmethod
59 - def create(cls, production_builder, x, y, orientation):
60 settlement_manager = production_builder.settlement_manager 61 village_builder = settlement_manager.village_builder 62 builder = BasicBuilder.create(BUILDINGS.FIRE_STATION, (x, y), orientation) 63 64 assigned_residences = village_builder.special_building_assignments[BUILDING_PURPOSE.FIRE_STATION][(x, y)] 65 total = len(assigned_residences) 66 not_serviced = 0 67 for residence_coords in assigned_residences: 68 if village_builder.plan[residence_coords][0] == BUILDING_PURPOSE.RESIDENCE: 69 not_serviced += 1 70 71 if not_serviced <= 0 or not_serviced < total * settlement_manager.owner.personality_manager.get('AbstractFireStation').fraction_of_assigned_residences_built: 72 return None 73 74 return FireStationEvaluator(village_builder, builder, not_serviced)
75 76 @property
77 - def purpose(self):
79 80 81 AbstractFireStation.register_buildings() 82