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

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

 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  from horizons.entities import Entities 
28 29 30 -class AbstractTobacconist(AbstractBuilding):
31 @property
32 - def evaluator_class(self):
34 35 @classmethod
36 - def register_buildings(cls):
38
39 40 -class TobacconistEvaluator(BuildingEvaluator):
41 @classmethod
42 - def create(cls, area_builder, x, y, orientation):
43 builder = BasicBuilder.create(BUILDINGS.TOBACCONIST, (x, y), orientation) 44 45 distance_to_farm = None 46 for building in area_builder.settlement.buildings_by_id.get(BUILDINGS.FARM, []): 47 distance = builder.position.distance(building.position) 48 if distance <= Entities.buildings[BUILDINGS.TOBACCONIST].radius: 49 tobacco_producer = False 50 for provider in building.get_providers(): 51 if isinstance(provider, Entities.buildings[BUILDINGS.TOBACCO_FIELD]): 52 tobacco_producer = True 53 break 54 if tobacco_producer: 55 distance_to_farm = distance if distance_to_farm is None or distance < distance_to_farm else distance_to_farm 56 57 distance_to_collector = cls._distance_to_nearest_collector(area_builder, builder) 58 if distance_to_collector is None: 59 return None # require tobacconists to have a collector building in range 60 61 personality = area_builder.owner.personality_manager.get('TobacconistEvaluator') 62 distance_penalty = Entities.buildings[BUILDINGS.TOBACCONIST].radius * personality.distance_penalty 63 64 alignment = cls._get_alignment(area_builder, builder.position.tuple_iter()) 65 distance = cls._weighted_distance(distance_to_collector, [(personality.farm_distance_importance, distance_to_farm)], distance_penalty) 66 value = float(Entities.buildings[BUILDINGS.TOBACCONIST].radius) / distance + alignment * personality.alignment_importance 67 return TobacconistEvaluator(area_builder, builder, value)
68 69 @property
70 - def purpose(self):
72 73 74 AbstractTobacconist.register_buildings() 75