aboutsummaryrefslogtreecommitdiff
path: root/src/openvic-simulation/misc/SongChance.cpp
blob: 94fb57163651fb8ff8d111e0c0dfaf44aa81f13a (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
#include "SongChance.hpp"

using namespace OpenVic;
using namespace OpenVic::NodeTools;

SongChance::SongChance(
   size_t new_index, std::string_view new_filename, ConditionalWeight&& new_chance
) : HasIdentifier { std::to_string(new_index) }, file_name { new_filename }, chance { std::move(new_chance) } {}

bool SongChance::parse_scripts(DefinitionManager const& definition_manager) {
   return chance.parse_scripts(definition_manager);
}

bool SongChanceManager::load_songs_file(ast::NodeCPtr root) {
   bool ret = true;

   ret &= expect_dictionary_reserve_length(
      song_chances,
      [this](std::string_view key, ast::NodeCPtr value) -> bool {
         if (key != "song") {
            Logger::error("Invalid song declaration ", key);
            return false;
         }
         std::string_view name {};
         ConditionalWeight chance { scope_t::COUNTRY, scope_t::COUNTRY, scope_t::NO_SCOPE };

         bool ret = expect_dictionary_keys(
            "name", ONE_EXACTLY, expect_string(assign_variable_callback(name)),
            "chance", ONE_EXACTLY, chance.expect_conditional_weight(ConditionalWeight::FACTOR)
         )(value);

         ret &= song_chances.add_item({ song_chances.size(), name, std::move(chance) });
         return ret;
      }
   )(root);

   if (song_chances.size() == 0) {
      Logger::error("No songs found in Songs.txt");
      return false;
   }

   return ret;
}

bool SongChanceManager::parse_scripts(DefinitionManager const& definition_manager) {
   bool ret = true;
   for (SongChance& songChance : song_chances.get_items()) {
      ret &= songChance.parse_scripts(definition_manager);
   }
   return ret;
}