aboutsummaryrefslogtreecommitdiff
path: root/src/openvic-simulation/map/State.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/openvic-simulation/map/State.cpp')
-rw-r--r--src/openvic-simulation/map/State.cpp79
1 files changed, 79 insertions, 0 deletions
diff --git a/src/openvic-simulation/map/State.cpp b/src/openvic-simulation/map/State.cpp
new file mode 100644
index 0000000..1a7f10d
--- /dev/null
+++ b/src/openvic-simulation/map/State.cpp
@@ -0,0 +1,79 @@
+#include "State.hpp"
+
+#include "Map.hpp"
+
+using namespace OpenVic;
+
+State::State(Country const* owner, Province const* capital, Region::provinces_t&& provinces, Province::colony_status_t colony_status)
+ : owner { owner }, capital { capital }, provinces { std::move(provinces) }, colony_status { colony_status } {}
+
+StateSet::StateSet(Region const* new_region) {
+ if (region->get_meta()) {
+ Logger::error("Cannot use meta region as state template!");
+ }
+ region = new_region;
+
+ std::vector<Region::provinces_t> temp_provinces;
+ bool in_state = false;
+
+ for (const auto province : region->get_provinces()) {
+ // add to existing state if shared owner & status...
+ for (auto& provinces : temp_provinces) {
+ if (provinces[0] == province) {
+ provinces.push_back(province);
+ in_state = true;
+ break;
+ }
+ }
+ if (in_state) {
+ in_state = false;
+ } else {
+ // ...otherwise start a new state
+ temp_provinces.push_back({ province });
+ }
+ }
+
+ for (auto& provinces : temp_provinces) {
+ states.push_back({ /* TODO: capital province logic */ provinces[0]->get_owner(), provinces[0], std::move(provinces), provinces[0]->get_colony_status() });
+ }
+
+ // Go back and assign each new state to its provinces.
+ for (const auto& state : states) {
+ for (auto province : state.get_provinces()) {
+ province->set_state(&state);
+ }
+ }
+}
+
+bool StateSet::add_state(State&& state) {
+ const auto existing = std::find(states.begin(), states.end(), state);
+ if (existing != states.end()) {
+ Logger::error("Attempted to add existing state!");
+ return false;
+ }
+ states.push_back(std::move(state));
+ return true;
+}
+
+bool StateSet::remove_state(State const* state) {
+ const auto existing = std::find(states.begin(), states.end(), *state);
+ if (existing == states.end()) {
+ Logger::error("Attempted to remove non-existant state!");
+ return false;
+ }
+ states.erase(existing);
+ return true;
+}
+
+StateSet::states_t& StateSet::get_states() {
+ return states;
+}
+
+void StateManager::generate_states(Map const& map) {
+ regions.clear();
+ regions.reserve(map.get_region_count());
+ for(const auto& region : map.get_regions()) {
+ regions.push_back(StateSet(&region));
+ }
+ Logger::info("Generated states.");
+} \ No newline at end of file