aboutsummaryrefslogtreecommitdiff
path: root/extension/src/MapSingleton.cpp
diff options
context:
space:
mode:
author Hop311 <hop3114@gmail.com>2023-03-30 23:50:50 +0200
committer Spartan322 <Megacake1234@gmail.com>2023-04-14 17:16:02 +0200
commitc0d76b78d3762e6eec3ed1c62618be84c5b7559b (patch)
treeacfcbeedd5a47136acdf883e791a297200b7d1b8 /extension/src/MapSingleton.cpp
parent1f04a7827ae377372cb491ff0257a47d0d4c2967 (diff)
Add terrain map
With Directional movement using WASD With Directional movement using arrow keys With Click-Drag movement using middle mouse button With Province identifiers With Province shape loading With Province rendering With Province selection With Province overview panel With Color lookup texture
Diffstat (limited to 'extension/src/MapSingleton.cpp')
-rw-r--r--extension/src/MapSingleton.cpp256
1 files changed, 256 insertions, 0 deletions
diff --git a/extension/src/MapSingleton.cpp b/extension/src/MapSingleton.cpp
new file mode 100644
index 0000000..0f5fe7c
--- /dev/null
+++ b/extension/src/MapSingleton.cpp
@@ -0,0 +1,256 @@
+#include "MapSingleton.hpp"
+
+#include <godot_cpp/variant/utility_functions.hpp>
+#include <godot_cpp/classes/file_access.hpp>
+#include <godot_cpp/classes/json.hpp>
+
+using namespace godot;
+using namespace OpenVic2;
+
+MapSingleton* MapSingleton::singleton = nullptr;
+
+void MapSingleton::_bind_methods() {
+ ClassDB::bind_method(D_METHOD("load_province_identifier_file", "file_path"), &MapSingleton::load_province_identifier_file);
+ ClassDB::bind_method(D_METHOD("load_province_shape_file", "file_path"), &MapSingleton::load_province_shape_file);
+ ClassDB::bind_method(D_METHOD("get_province_identifier_from_pixel_coords", "coords"), &MapSingleton::get_province_identifier_from_pixel_coords);
+ ClassDB::bind_method(D_METHOD("get_width"), &MapSingleton::get_width);
+ ClassDB::bind_method(D_METHOD("get_height"), &MapSingleton::get_height);
+ ClassDB::bind_method(D_METHOD("get_province_index_image"), &MapSingleton::get_province_index_image);
+ ClassDB::bind_method(D_METHOD("get_province_colour_image"), &MapSingleton::get_province_colour_image);
+}
+
+MapSingleton* MapSingleton::get_singleton() {
+ return singleton;
+}
+
+MapSingleton::MapSingleton() {
+ ERR_FAIL_COND(singleton != nullptr);
+ singleton = this;
+}
+
+MapSingleton::~MapSingleton() {
+ ERR_FAIL_COND(singleton != this);
+ singleton = nullptr;
+}
+
+Error MapSingleton::load_province_identifier_file(String const& file_path) {
+ UtilityFunctions::print("Loading identifier file: ", file_path);
+ Ref<FileAccess> file = FileAccess::open(file_path, FileAccess::ModeFlags::READ);
+ Error err = FileAccess::get_open_error();
+ if (err != OK || file.is_null()) {
+ UtilityFunctions::push_error("Failed to load province identifier file: ", file_path);
+ return err == OK ? FAILED : err;
+ }
+ String json_string = file->get_as_text();
+ Ref<JSON> json;
+ json.instantiate();
+ err = json->parse(json_string);
+ if (err) {
+ UtilityFunctions::push_error("Failed to parse province identifier file as JSON: ", file_path,
+ "\nError at line ", json->get_error_line(), ": ", json->get_error_message());
+ return err;
+ }
+ Variant json_var = json->get_data();
+ Variant::Type type = json_var.get_type();
+ if (type != Variant::DICTIONARY) {
+ UtilityFunctions::push_error("Invalid province identifier JSON: root has type ",
+ Variant::get_type_name(type), " (expected Dictionary)");
+ return FAILED;
+ }
+ Dictionary prov_dict = json_var;
+ Array prov_identifiers = prov_dict.keys();
+ for (int idx = 0; idx < prov_identifiers.size(); ++idx) {
+ String const& identifier = prov_identifiers[idx];
+ Variant const& colour_var = prov_dict[identifier];
+ if (identifier.is_empty()) {
+ UtilityFunctions::push_error("Empty province identifier with colour: ", colour_var);
+ err = FAILED;
+ continue;
+ }
+ static const String prov_prefix = "prov_";
+ if (!identifier.begins_with(prov_prefix))
+ UtilityFunctions::push_warning("Province identifier missing prefix: ", identifier);
+ type = colour_var.get_type();
+ Province::colour_t colour = Province::NULL_COLOUR;
+ if (type == Variant::ARRAY) {
+ Array colour_array = colour_var;
+ if (colour_array.size() == 3) {
+ for (int jdx = 0; jdx < 3; ++jdx) {
+ Variant var = colour_array[jdx];
+ if (var.get_type() != Variant::FLOAT) {
+ colour = Province::NULL_COLOUR;
+ break;
+ }
+ double colour_double = var;
+ if (std::trunc(colour_double) != colour_double) {
+ colour = Province::NULL_COLOUR;
+ break;
+ }
+ int64_t colour_int = static_cast<int64_t>(colour_double);
+ if (colour_int < 0 || colour_int > 255) {
+ colour = Province::NULL_COLOUR;
+ break;
+ }
+ colour = (colour << 8) | colour_int;
+ }
+ }
+ } else if (type == Variant::STRING) {
+ String colour_string = colour_var;
+ if (colour_string.is_valid_hex_number()) {
+ int64_t colour_int = colour_string.hex_to_int();
+ if (0 <= colour_int && colour_int <= 0xFFFFFF)
+ colour = colour_int;
+ }
+ }
+ if (colour == Province::NULL_COLOUR) {
+ UtilityFunctions::push_error("Invalid province identifier colour for ", identifier, ": ", colour_var);
+ err = FAILED;
+ continue;
+ }
+ std::string error_message;
+ if (!map.add_province(identifier.utf8().get_data(), colour, error_message)) {
+ UtilityFunctions::push_error(error_message.c_str());
+ err = FAILED;
+ }
+ }
+ map.lock_provinces();
+ return err;
+}
+
+static Province::colour_t colour_at(PackedByteArray const& colour_data_array, int32_t idx) {
+ return (colour_data_array[idx * 3] << 16) | (colour_data_array[idx * 3 + 1] << 8) | colour_data_array[idx * 3 + 2];
+}
+
+Error MapSingleton::load_province_shape_file(String const& file_path) {
+ if (province_shape_image.is_valid()) {
+ UtilityFunctions::push_error("Province shape file has already been loaded, cannot load: ", file_path);
+ return FAILED;
+ }
+ province_shape_image.instantiate();
+ Error err = province_shape_image->load(file_path);
+ if (err != OK) {
+ UtilityFunctions::push_error("Failed to load province shape file: ", file_path);
+ province_shape_image.unref();
+ return err;
+ }
+ width = province_shape_image->get_width();
+ height = province_shape_image->get_height();
+ if (width < 1 || height < 1) {
+ UtilityFunctions::push_error("Invalid dimensions (", width, "x", height, ") for province shape file: ", file_path);
+ err = FAILED;
+ }
+ static const Image::Format expected_format = Image::FORMAT_RGB8;
+ Image::Format format = province_shape_image->get_format();
+ if (format != expected_format) {
+ UtilityFunctions::push_error("Invalid format (", format, ", should be ", expected_format, ") for province shape file: ", file_path);
+ err = FAILED;
+ }
+ if (err) {
+ province_shape_image.unref();
+ return err;
+ }
+
+ std::vector<bool> province_checklist(map.get_province_count());
+
+ PackedByteArray shape_data_array = province_shape_image->get_data(), index_data_array;
+ index_data_array.resize(width * height * sizeof(Province::index_t));
+ Province::index_t* index_data = reinterpret_cast<Province::index_t*>(index_data_array.ptrw());
+
+ for (int32_t y = 0; y < height; ++y) {
+ for (int32_t x = 0; x < width; ++x) {
+ const int32_t idx = x + y * width;
+ const Province::colour_t colour = colour_at(shape_data_array, idx);
+ if (colour == Province::NULL_COLOUR) {
+ index_data[idx] = Province::NULL_INDEX;
+ continue;
+ }
+ if (x > 0) {
+ const int32_t jdx = idx - 1;
+ if (colour_at(shape_data_array, jdx) == colour) {
+ index_data[idx] = index_data[jdx];
+ continue;
+ }
+ }
+ if (y > 0) {
+ const int32_t jdx = idx - width;
+ if (colour_at(shape_data_array, jdx) == colour) {
+ index_data[idx] = index_data[jdx];
+ continue;
+ }
+ }
+ const Province* province = map.get_province_by_colour(colour);
+ if (province) {
+ Province::index_t index = province->get_index();
+ index_data[idx] = index;
+ province_checklist[index - 1] = true;
+ continue;
+ }
+ UtilityFunctions::push_error("Unrecognised province colour ", Province::colour_to_hex_string(colour).c_str(), " at (", x, ", ", y, ")");
+ err = FAILED;
+ index_data[idx] = Province::NULL_INDEX;
+ }
+ }
+
+ for (size_t idx = 0; idx < province_checklist.size(); ++idx) {
+ if (!province_checklist[idx]) {
+ Province* province = map.get_province_by_index(idx + 1);
+ if (province) UtilityFunctions::push_error("Province missing from shape image: ", province->to_string().c_str());
+ else UtilityFunctions::push_error("Province missing for index: ", static_cast<int32_t>(idx + 1));
+ err = FAILED;
+ }
+ }
+
+ province_index_image = Image::create_from_data(width, height, false, Image::FORMAT_RG8, index_data_array);
+ if (province_index_image.is_null()) {
+ UtilityFunctions::push_error("Failed to create province ID image");
+ err = FAILED;
+ }
+
+ PackedByteArray colour_data_array;
+ colour_data_array.resize((Province::MAX_INDEX + 1) * 3);
+ for (size_t idx = 1; idx <= map.get_province_count(); ++idx) {
+ const Province* province = map.get_province_by_index(idx);
+ if (province) {
+ const Province::colour_t colour = province->get_colour();
+ colour_data_array[3 * idx + 0] = (colour >> 16) & 0xFF;
+ colour_data_array[3 * idx + 1] = (colour >> 8) & 0xFF;
+ colour_data_array[3 * idx + 2] = colour & 0xFF;
+ } else UtilityFunctions::push_error("Missing province at index ", static_cast<int32_t>(idx));
+ }
+ static const int32_t PROVINCE_INDEX_SQRT = 1 << (sizeof(Province::index_t) * 4);
+ province_colour_image = Image::create_from_data(PROVINCE_INDEX_SQRT, PROVINCE_INDEX_SQRT, false, Image::FORMAT_RGB8, colour_data_array);
+ if (province_colour_image.is_null()) {
+ UtilityFunctions::push_error("Failed to create province colour image");
+ err = FAILED;
+ }
+
+ return err;
+}
+
+String MapSingleton::get_province_identifier_from_pixel_coords(Vector2i const& coords) {
+ if (province_index_image.is_valid()) {
+ const PackedByteArray index_data_array = province_index_image->get_data();
+ const Province::index_t* index_data = reinterpret_cast<const Province::index_t*>(index_data_array.ptr());
+ const int32_t x_mod_w = UtilityFunctions::posmod(coords.x, width);
+ const int32_t y_mod_h = UtilityFunctions::posmod(coords.y, height);
+ const Province* province = map.get_province_by_index(index_data[x_mod_w + y_mod_h * width]);
+ if (province) return province->get_identifier().c_str();
+ }
+ return String{};
+}
+
+int32_t MapSingleton::get_width() const {
+ return width;
+}
+
+int32_t MapSingleton::get_height() const {
+ return height;
+}
+
+Ref<Image> MapSingleton::get_province_index_image() const {
+ return province_index_image;
+}
+
+Ref<Image> MapSingleton::get_province_colour_image() const {
+ return province_colour_image;
+}