summaryrefslogtreecommitdiff
path: root/src/openvic/map/Building.cpp
blob: f4432861d80346fd73c521c37a7c364c4bcec3f4 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
#include "Building.hpp"

#include <cassert>

#include "../utility/Logger.hpp"
#include "Province.hpp"

using namespace OpenVic;

Building::Building(BuildingType const& new_type)
   : HasIdentifier { new_type.get_identifier() },
     type { new_type } {}

bool Building::_can_expand() const {
   return level < type.get_max_level();
}

BuildingType const& Building::get_type() const {
   return type;
}

Building::level_t Building::get_level() const {
   return level;
}

Building::ExpansionState Building::get_expansion_state() const {
   return expansion_state;
}

Date const& Building::get_start_date() const {
   return start;
}

Date const& Building::get_end_date() const {
   return end;
}

float Building::get_expansion_progress() const {
   return expansion_progress;
}

return_t Building::expand() {
   if (expansion_state == ExpansionState::CanExpand) {
      expansion_state = ExpansionState::Preparing;
      expansion_progress = 0.0f;
      return SUCCESS;
   }
   return FAILURE;
}

/* REQUIREMENTS:
 * MAP-71, MAP-74, MAP-77
 */
void Building::update_state(Date const& today) {
   switch (expansion_state) {
      case ExpansionState::Preparing:
         start = today;
         end = start + type.get_build_time();
         break;
      case ExpansionState::Expanding:
         expansion_progress = static_cast<double>(today - start) / static_cast<double>(end - start);
         break;
      default: expansion_state = _can_expand() ? ExpansionState::CanExpand : ExpansionState::CannotExpand;
   }
}

void Building::tick(Date const& today) {
   if (expansion_state == ExpansionState::Preparing) {
      expansion_state = ExpansionState::Expanding;
   }
   if (expansion_state == ExpansionState::Expanding) {
      if (end <= today) {
         level++;
         expansion_state = ExpansionState::CannotExpand;
      }
   }
}

BuildingType::BuildingType(ovstring const& new_identifier, Building::level_t new_max_level, Timespan new_build_time)
   : HasIdentifier { new_identifier },
     max_level { new_max_level },
     build_time { new_build_time } {
   assert(max_level >= 0);
   assert(build_time >= 0);
}

Building::level_t BuildingType::get_max_level() const {
   return max_level;
}

Timespan BuildingType::get_build_time() const {
   return build_time;
}

BuildingManager::BuildingManager() : building_types { "building types" } {}

return_t BuildingManager::add_building_type(ovstring const& identifier, Building::level_t max_level, Timespan build_time) {
   if (identifier.empty()) {
      Logger::error("Invalid building type identifier - empty!");
      return FAILURE;
   }
   if (max_level < 0) {
      Logger::error("Invalid building type max level: ", max_level);
      return FAILURE;
   }
   if (build_time < 0) {
      Logger::error("Invalid building type build time: ", build_time);
      return FAILURE;
   }
   return building_types.add_item({ identifier, max_level, build_time });
}

void BuildingManager::lock_building_types() {
   building_types.lock();
}

BuildingType const* BuildingManager::get_building_type_by_identifier(ovstring const& identifier) const {
   return building_types.get_item_by_identifier(identifier);
}

return_t BuildingManager::generate_province_buildings(Province& province) const {
   province.reset_buildings();
   if (!building_types.is_locked()) {
      Logger::error("Cannot generate buildings until building types are locked!");
      return FAILURE;
   }
   if (province.is_water()) return SUCCESS;
   return_t ret = SUCCESS;
   for (BuildingType const& type : building_types.get_items())
      if (province.add_building(type) != SUCCESS) ret = FAILURE;
   province.lock_buildings();
   return ret;
}