aboutsummaryrefslogtreecommitdiff
path: root/src/openvic-simulation/map/Map.hpp
blob: d5157ef4ad9fc7a18263e936f4d3c6741ef02df2 (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
#pragma once

#include <functional>

#include <openvic-dataloader/csv/LineObject.hpp>

#include "openvic-simulation/map/Region.hpp"

namespace OpenVic {

   struct Mapmode : HasIdentifier {
      friend struct Map;

      using colour_func_t = std::function<colour_t(Map const&, Province const&)>;
      using index_t = size_t;

   private:
      const index_t index;
      const colour_func_t colour_func;

      Mapmode(const std::string_view new_identifier, index_t new_index, colour_func_t new_colour_func);

   public:
      static const Mapmode ERROR_MAPMODE;

      Mapmode(Mapmode&&) = default;

      index_t get_index() const;
      colour_t get_colour(Map const& map, Province const& province) const;
   };

   struct GoodManager;

   /* REQUIREMENTS:
    * MAP-4
    */
   struct Map {
      using terrain_t = uint8_t;
      using terrain_variant_map_t = std::map<colour_t, terrain_t>;

#pragma pack(push, 1)
      struct shape_pixel_t {
         Province::index_t index;
         terrain_t terrain;
      };
#pragma pack(pop)
   private:
      using colour_index_map_t = std::map<colour_t, Province::index_t>;

      IdentifierRegistry<Province> provinces;
      IdentifierRegistry<Region> regions;
      IdentifierRegistry<Mapmode> mapmodes;
      ProvinceSet water_provinces;

      size_t width = 0, height = 0;
      std::vector<shape_pixel_t> province_shape_image;
      colour_index_map_t colour_index_map;
      Province::index_t max_provinces = Province::MAX_INDEX;
      Province::index_t selected_province = Province::NULL_INDEX;

      Pop::pop_size_t highest_province_population, total_map_population;

      Province::index_t get_index_from_colour(colour_t colour) const;

   public:
      Map();

      bool add_province(const std::string_view identifier, colour_t colour);
      void lock_provinces();
      bool set_water_province(const std::string_view identifier);
      bool set_water_province_list(std::vector<std::string_view> const& list);
      void lock_water_provinces();
      bool add_region(const std::string_view identifier, std::vector<std::string_view> const& province_identifiers);
      void lock_regions();

      size_t get_province_count() const;
      std::vector<Province> const& get_provinces() const;
      Province* get_province_by_index(Province::index_t index);
      Province const* get_province_by_index(Province::index_t index) const;
      Province* get_province_by_identifier(const std::string_view identifier);
      Province const* get_province_by_identifier(const std::string_view identifier) const;
      Province::index_t get_province_index_at(size_t x, size_t y) const;
      bool set_max_provinces(Province::index_t new_max_provinces);
      Province::index_t get_max_provinces() const;
      void set_selected_province(Province::index_t index);
      Province::index_t get_selected_province_index() const;
      Province const* get_selected_province() const;

      Region* get_region_by_identifier(const std::string_view identifier);
      Region const* get_region_by_identifier(const std::string_view identifier) const;
      size_t get_region_count() const;
      std::vector<Region> const& get_regions() const;

      bool generate_province_shape_image(size_t new_width, size_t new_height, uint8_t const* colour_data,
         uint8_t const* terrain_data, terrain_variant_map_t const& terrain_variant_map, bool detailed_errors);
      size_t get_width() const;
      size_t get_height() const;
      std::vector<shape_pixel_t> const& get_province_shape_image() const;

      bool add_mapmode(const std::string_view identifier, Mapmode::colour_func_t colour_func);
      IDENTIFIER_REGISTRY_ACCESSORS(Mapmode, mapmode)
      static constexpr size_t MAPMODE_COLOUR_SIZE = 4;
      bool generate_mapmode_colours(Mapmode::index_t index, uint8_t* target) const;

      bool setup(GoodManager const& good_manager, BuildingManager const& building_manager, PopManager const& pop_manager);

      void update_highest_province_population();
      Pop::pop_size_t get_highest_province_population() const;
      void update_total_map_population();
      Pop::pop_size_t get_total_map_population() const;

      void update_state(Date const& today);
      void tick(Date const& today);

      bool load_province_definitions(std::vector<ovdl::csv::LineObject> const& lines);
      bool load_region_file(ast::NodeCPtr root);
   };
}