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

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

 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.component.storagecomponent import StorageComponent 
27  from horizons.constants import BUILDINGS, RES 
28  from horizons.entities import Entities 
29 30 31 -class AbstractIronMine(AbstractBuilding):
32 - def iter_potential_locations(self, settlement_manager):
33 building_class = Entities.buildings[BUILDINGS.MOUNTAIN] 34 for building in settlement_manager.settlement.buildings_by_id.get(BUILDINGS.MOUNTAIN, []): 35 if building.get_component(StorageComponent).inventory[RES.RAW_IRON]: 36 coords = building.position.origin.to_tuple() 37 if coords in settlement_manager.production_builder.simple_collector_area_cache.cache[building_class.size]: 38 yield (coords[0], coords[1], (building.rotation - 45) // 90)
39 40 @property
41 - def evaluator_class(self):
43 44 @classmethod
45 - def register_buildings(cls):
47
48 49 -class IronMineEvaluator(BuildingEvaluator):
50 @classmethod
51 - def create(cls, area_builder, x, y, orientation):
52 builder = BasicBuilder.create(BUILDINGS.MINE, (x, y), orientation) 53 return IronMineEvaluator(area_builder, builder, 0)
54 55 @property
56 - def purpose(self):
58 59 60 AbstractIronMine.register_buildings() 61