aboutsummaryrefslogtreecommitdiff
path: root/src/openvic-simulation/dataloader/Dataloader.hpp
blob: 1a8e9abe55d7c788573a0ce500e42722f48a828b (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
#pragma once

#include <filesystem>
#include <unordered_map>

#include <openvic-dataloader/csv/Parser.hpp>
#include <openvic-dataloader/v2script/Parser.hpp>

#include "openvic-simulation/dataloader/NodeTools.hpp"

namespace OpenVic {
   namespace fs = std::filesystem;

   struct GameManager;
   struct PopManager;
   struct UnitManager;
   struct GoodManager;

   class Dataloader {
   public:
      using path_vector_t = std::vector<fs::path>;

   private:
      path_vector_t roots;

      bool _load_pop_types(
         PopManager& pop_manager, UnitManager const& unit_manager, GoodManager const& good_manager
      ) const;
      bool _load_units(UnitManager& unit_manager, GoodManager const& good_manager) const;
      bool _load_map_dir(GameManager& game_manager) const;
      bool _load_oobs(GameManager& game_manager) const;
      bool _load_countries(GameManager& game_manager) const;
      bool _load_history(GameManager& game_manager) const;

   public:
      static ovdl::v2script::Parser parse_defines(fs::path const& path);
      static ovdl::v2script::Parser parse_lua_defines(fs::path const& path);
      static ovdl::csv::Windows1252Parser parse_csv(fs::path const& path);

      Dataloader() = default;

      /// @brief Searches for the Victoria 2 install directory
      ///
      /// @param hint_path A path to indicate a hint to assist in searching for the Victoria 2 install directory
      ///   Supports being supplied:
      ///      1. A valid Victoria 2 game directory (Victoria 2 directory that contains a v2game.exe file)
      ///      2. An Empty path: assumes a common Steam install structure per platform.
      ///         2b. If Windows, returns "HKEY_LOCAL_MACHINE\SOFTWARE\WOW6432Node\Paradox Interactive\Victoria 2"
      ///            "path" registry value
      ///         2c. If registry value returns an empty string, performs Steam checks below
      ///      3. A path to a root Steam install. (eg: C:\Program Files(x86)\Steam, ~/.steam/steam)
      ///      4. A path to a root Steam steamapps directory. (eg: C:\Program Files(x86)\Steam\steamapps,
      ///         ~/.steam/steam/steamapps)
      ///      5. A path to the root Steam libraryfolders.vdf, commonly in the root Steam steamapps directory.
      ///      6. A path to the Steam library directory that contains Victoria 2.
      ///      7. A path to a Steam library's steamapps directory that contains Victoria 2.
      ///      8. A path to a Steam library's steamapps/common directory that contains Victoria 2.
      ///      9. If any of these checks don't resolve to a valid Victoria 2 game directory when supplied
      ///         a non-empty hint_path, performs empty path behavior.
      /// @return fs::path The root directory of a valid Victoria 2 install, or an empty path.
      ///
      static fs::path search_for_game_path(fs::path hint_path = {});

      /* In reverse-load order, so base defines first and final loaded mod last */
      bool set_roots(path_vector_t const& new_roots);

      /* REQUIREMENTS:
       * DAT-24
       */
      fs::path lookup_file(fs::path const& path) const;
      path_vector_t lookup_files_in_dir(fs::path const& path, fs::path const& extension) const;
      bool apply_to_files_in_dir(
         fs::path const& path, fs::path const& extension, NodeTools::callback_t<fs::path const&> callback
      ) const;

      bool load_defines(GameManager& game_manager) const;
      bool load_pop_history(GameManager& game_manager, fs::path const& path) const;

      enum locale_t : size_t {
         English,
         French,
         German,
         Polish,
         Spanish,
         Italian,
         Swedish,
         Czech,
         Hungarian,
         Dutch,
         Portugese,
         Russian,
         Finnish,
         _LocaleCount
      };
      static constexpr char const* locale_names[_LocaleCount] = {
         "en_GB", "fr_FR", "de_DE", "pl_PL", "es_ES", "it_IT", "sv_SE", "cs_CZ", "hu_HU", "nl_NL", "pt_PT", "ru_RU", "fi_FI"
      };

      /* Args: key, locale, localisation */
      using localisation_callback_t = NodeTools::callback_t<std::string_view, locale_t, std::string_view>;
      bool load_localisation_files(
         localisation_callback_t callback, fs::path const& localisation_dir = "localisation"
      ) const;

   private:
      struct fshash {
         size_t operator()(const std::filesystem::path& p) const noexcept {
            return std::filesystem::hash_value(p);
         }
      };

      using hint_path_t = fs::path;
      using game_path_t = fs::path;
      inline static std::unordered_map<hint_path_t, game_path_t, fshash> _cached_paths;
   };
}