aboutsummaryrefslogtreecommitdiff
path: root/src/openvic-simulation/modifier/ModifierSum.cpp
blob: 676e872ee226701677940033f0f04acc02ad52cb (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
119
120
121
122
123
#include "ModifierSum.hpp"

#include "openvic-simulation/modifier/Modifier.hpp"

#include "openvic-simulation/country/CountryInstance.hpp"
#include "openvic-simulation/map/ProvinceInstance.hpp"

using namespace OpenVic;

std::string_view ModifierSum::source_to_string(modifier_source_t const& source) {
   return std::visit(
      [](HasGetIdentifier auto const* has_identifier) -> std::string_view {
         return has_identifier->get_identifier();
      },
      source
   );
}

std::string ModifierSum::modifier_entry_t::to_string() const {
   return StringUtils::append_string_views(
      "[", modifier->get_identifier(), ", ", multiplier.to_string(), ", ", source_to_string(source), ", ",
      ModifierEffect::target_to_string(excluded_targets), "]"
   );
}

void ModifierSum::clear() {
   modifiers.clear();
   value_sum.clear();
}

bool ModifierSum::empty() {
   return modifiers.empty();
}

fixed_point_t ModifierSum::get_effect(ModifierEffect const& effect, bool* effect_found) const {
   return value_sum.get_effect(effect, effect_found);
}

fixed_point_t ModifierSum::get_effect_nullcheck(ModifierEffect const* effect, bool* effect_found) const {
   return value_sum.get_effect_nullcheck(effect, effect_found);
}

bool ModifierSum::has_effect(ModifierEffect const& effect) const {
   return value_sum.has_effect(effect);
}

void ModifierSum::add_modifier(
   Modifier const& modifier, modifier_source_t const& source, fixed_point_t multiplier,
   ModifierEffect::target_t excluded_targets
) {
   using enum ModifierEffect::target_t;

   // We could test that excluded_targets != ALL_TARGETS, but in practice it's always
   // called with an explcit/hardcoded value and so won't ever exclude everything.
   if (multiplier != fixed_point_t::_0()) {
      modifiers.emplace_back(&modifier, multiplier, source, excluded_targets);
      value_sum.multiply_add_exclude_targets(modifier, multiplier, excluded_targets);
   }
}

void ModifierSum::add_modifier_nullcheck(
   Modifier const* modifier, modifier_source_t const& source, fixed_point_t multiplier,
   ModifierEffect::target_t excluded_targets
) {
   if (modifier != nullptr) {
      add_modifier(*modifier, source, multiplier, excluded_targets);
   }
}

void ModifierSum::add_modifier_sum(ModifierSum const& modifier_sum) {
   modifiers.insert(modifiers.end(), modifier_sum.modifiers.begin(), modifier_sum.modifiers.end());
   value_sum += modifier_sum.value_sum;
}

void ModifierSum::add_modifier_sum_exclude_targets(
   ModifierSum const& modifier_sum, ModifierEffect::target_t excluded_targets
) {
   // We could test that excluded_targets != ALL_TARGETS, but in practice it's always
   // called with an explcit/hardcoded value and so won't ever exclude everything.
   for (modifier_entry_t const& modifier_entry : modifier_sum.modifiers) {
      add_modifier(
         *modifier_entry.modifier, modifier_entry.source, modifier_entry.multiplier,
         modifier_entry.excluded_targets | excluded_targets
      );
   }
}

void ModifierSum::add_modifier_sum_exclude_source(ModifierSum const& modifier_sum, modifier_source_t const& excluded_source) {
   for (modifier_entry_t const& modifier_entry : modifier_sum.modifiers) {
      if (modifier_entry.source != excluded_source) {
         add_modifier(
            *modifier_entry.modifier, modifier_entry.source, modifier_entry.multiplier, modifier_entry.excluded_targets
         );
      }
   }
}

// TODO - include value_sum[effect] in result? Early return if lookup in value_sum fails?

void ModifierSum::push_contributing_modifiers(
   ModifierEffect const& effect, std::vector<modifier_entry_t>& contributions
) const {
   using enum ModifierEffect::target_t;

   for (modifier_entry_t const& modifier_entry : modifiers) {
      if (ModifierEffect::excludes_targets(effect.get_targets(), modifier_entry.excluded_targets)) {
         bool effect_found = false;
         const fixed_point_t value = modifier_entry.modifier->get_effect(effect, &effect_found);

         if (effect_found) {
            contributions.push_back(modifier_entry);
         }
      }
   }
}

std::vector<ModifierSum::modifier_entry_t> ModifierSum::get_contributing_modifiers(ModifierEffect const& effect) const {
   std::vector<modifier_entry_t> contributions;

   push_contributing_modifiers(effect, contributions);

   return contributions;
}