Package horizons :: Package ai :: Package aiplayer :: Package goal :: Module storagespace
[hide private]
[frames] | no frames]

Source Code for Module horizons.ai.aiplayer.goal.storagespace

 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.goal.improvecollectorcoverage import ImproveCollectorCoverageGoal 
23  from horizons.component.storagecomponent import StorageComponent 
24  from horizons.constants import RES 
25   
26   
27 -class StorageSpaceGoal(ImproveCollectorCoverageGoal):
28 - def get_personality_name(self):
29 return 'StorageSpaceGoal'
30
31 - def _need_more_storage(self):
32 limit = self.settlement.get_component(StorageComponent).inventory.get_limit(RES.FOOD) 33 if limit >= self.personality.max_required_storage_space: 34 return False 35 important_resources = [RES.FOOD, RES.TEXTILE, RES.LIQUOR] 36 for resource_id in important_resources: 37 if self.settlement.get_component(StorageComponent).inventory[resource_id] + self.personality.full_storage_threshold >= limit: 38 return True 39 return False
40
41 - def update(self):
42 if self._need_more_storage(): 43 super().update() 44 if not self._is_active: 45 self._is_active = True 46 self._problematic_buildings = self.production_builder.production_buildings 47 else: 48 self._is_active = False
49
50 - def execute(self):
51 result = self._build_extra_storage() 52 self._log_generic_build_result(result, 'storage space provider') 53 return self._translate_build_result(result)
54