...

4 files changed, 289 insertions, 122 deletions

diff --git a/Makefile b/Makefile
index bce0a0d..96e75d7 100644
--- a/Makefile
+++ b/Makefile
@@ -27,7 +27,7 @@ YAWS ?= yaws
 YAWS_OPTS ?= $(ERL_NAME_VS_SNAME) query_node -erlarg '$(ERL_OPTS)'
 
 DIALYZER ?= dialyzer
-DIALYZER_OPTS ?=
+DIALYZER_OPTS ?= -I $(INCLUDE_DIR)
 
 M4 ?= m4
 M4_OPTS ?=
diff --git a/include/base_attributes.hrl b/include/base_attributes.hrl
index 2a5254e..749e84a 100644
--- a/include/base_attributes.hrl
+++ b/include/base_attributes.hrl
@@ -98,10 +98,10 @@
    )
 ).
 
--define(ATTRIBUTE_DEFENSE_SCORE_MIN,   0).
--define(ATTRIBUTE_DEFENSE_SCORE_MAX,   300).
--define(ATTRIBUTE_DEFENSE_SCORE_BASE,  50).
--define(ATTRIBUTE_DEFENSE_SCORE_COST,  1).
+-define(ATTRIBUTE_DEFENSE_SCORE_MIN,      0).
+-define(ATTRIBUTE_DEFENSE_SCORE_MAX,      300).
+-define(ATTRIBUTE_DEFENSE_SCORE_DEFAULT,  50).
+-define(ATTRIBUTE_DEFENSE_SCORE_COST,     1).
 -define
 (
    ATTRIBUTE_DEFENSE_SCORE_MAX_POINTS,
@@ -111,10 +111,10 @@
    )
 ).
 
--define(ATTRIBUTE_ATTACK_SCORE_MIN,    0).
--define(ATTRIBUTE_ATTACK_SCORE_MAX,    300).
--define(ATTRIBUTE_ATTACK_SCORE_BASE,   50).
--define(ATTRIBUTE_ATTACK_SCORE_COST,   1).
+-define(ATTRIBUTE_ATTACK_SCORE_MIN,       0).
+-define(ATTRIBUTE_ATTACK_SCORE_MAX,       300).
+-define(ATTRIBUTE_ATTACK_SCORE_DEFAULT,   50).
+-define(ATTRIBUTE_ATTACK_SCORE_COST,      1).
 -define
 (
    ATTRIBUTE_ATTACK_SCORE_MAX_POINTS,
diff --git a/src/balancer/blc_damage_type.erl b/src/balancer/blc_damage_type.erl
new file mode 100644
index 0000000..340b4ff
--- /dev/null
+++ b/src/balancer/blc_damage_type.erl
@@ -0,0 +1,117 @@
+-module(blc_damage_type).
+
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%% TYPES %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+-type entry() :: {shr_damage_type:type(), non_neg_integer()}.
+-type coefficient() :: {shr_damage_type:type(), 0..100}.
+
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%% EXPORTS %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+-export_type([entry/0, coefficient/0]).
+
+-export
+(
+   [
+      sort_entries/1,
+      compute_score/1,
+      apply_score_modifier/3,
+      generate_entries_from_score/2
+   ]
+).
+
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%% LOCAL FUNCTIONS %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+
+
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%% EXPORTED FUNCTIONS %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+-spec sort_entries (list(entry())) -> list(entry()).
+sort_entries (Entries) ->
+   lists:sort
+   (
+      fun ({_NameA, ValueA}, {_NameB, ValueB}) -> (ValueA >= ValueB) end,
+      Entries
+   ).
+
+-spec compute_score (list(entry())) -> non_neg_integer().
+compute_score (SortedEntries) ->
+   {_LastIndex, Result} =
+      lists:foldl
+      (
+         fun ({_Name, Value}, {Index, Current}) ->
+            {(Index + 1), (Current + (Index * Value))}
+         end,
+         {1, 0},
+         SortedEntries
+      ),
+
+   Result.
+
+-spec apply_score_modifier
+   (
+      non_neg_integer(),
+      (-1 | 1),
+      list(entry())
+   )
+   -> list(entry()).
+apply_score_modifier (AbsModifier, Mod, S0SortedEntries) ->
+   {S1SortedEntries, {_EndIndex, EndModifier}} =
+      lists:mapfoldl
+      (
+         fun ({Name, S0Value}, {Index, RemainingModifier}) ->
+            case ((RemainingModifier >= Index) and (S0Value > 0)) of
+               true ->
+                  {
+                     {Name, (S0Value + Mod)},
+                     {(Index + 1), (RemainingModifier - Index)}
+                  };
+
+               false -> {{Name, S0Value}, {(Index + 1), RemainingModifier}}
+            end
+         end,
+         {1, AbsModifier},
+         S0SortedEntries
+      ),
+
+   case (EndModifier > 0) of
+      false -> S1SortedEntries;
+      true -> apply_score_modifier(EndModifier, Mod, S1SortedEntries)
+   end.
+
+-spec generate_entries_from_score
+   (
+      non_neg_integer(),
+      list(coefficient())
+   )
+   -> list(entry()).
+generate_entries_from_score (TargetScore, SortedRatios) ->
+   {Distribution, _LastIndex} =
+      lists:foldl
+      (
+         fun ({_Name, Value}, {Cumul, Index}) ->
+            {(Cumul + (Value * Index)), (Index + 1)}
+         end,
+         {0, 1},
+         SortedRatios
+      ),
+
+   Base = (TargetScore / (Distribution / 100)),
+
+   UnderperformingEntries =
+      lists:map
+      (
+         fun ({Name, Value}) -> {Name, trunc(Base * (Value / 100))} end,
+         SortedRatios
+      ),
+
+   MissingScore = (TargetScore - compute_score(UnderperformingEntries)),
+
+   case (MissingScore >= 0) of
+      true -> apply_score_modifier(MissingScore, 1, UnderperformingEntries);
+      false ->
+         apply_score_modifier((-1 * MissingScore), -1, UnderperformingEntries)
+   end.
diff --git a/src/balancer/struct/blc_armor.erl b/src/balancer/struct/blc_armor.erl
index d3c3daf..46189d2 100644
--- a/src/balancer/struct/blc_armor.erl
+++ b/src/balancer/struct/blc_armor.erl
@@ -4,36 +4,38 @@
 
 -define
 (
-   ATTRIBUTE_ARMOR_POINTS,
+   SPENDABLE_ARMOR_POINTS,
    (
       (
          ?ATTRIBUTE_DAMAGE_MODIFIER_COST
-         * (?ATTRIBUTE_DAMAGE_MODIFIER_BASE - ?ATTRIBUTE_DAMAGE_MODIFIER_MIN)
+         * (?ATTRIBUTE_DAMAGE_MODIFIER_DEFAULT - ?ATTRIBUTE_DAMAGE_MODIFIER_MIN)
+      )
+      +
+      (
+         ?ATTRIBUTE_DODGE_CHANCE_COST
+         * (?ATTRIBUTE_DODGE_CHANCE_DEFAULT - ?ATTRIBUTE_DODGE_CHANCE_MIN)
       )
       +
       (
          ?ATTRIBUTE_MOVEMENT_POINTS_COST
-         * (?ATTRIBUTE_MOVEMENT_POINTS_BASE - ?ATTRIBUTE_MOVEMENT_POINTS_MIN)
+         * (?ATTRIBUTE_MOVEMENT_POINTS_DEFAULT - ?ATTRIBUTE_MOVEMENT_POINTS_MIN)
       )
       +
       (
          ?ATTRIBUTE_HEALTH_COST
-         * (?ATTRIBUTE_HEALTH_BASE - ?ATTRIBUTE_HEALTH_MIN)
+         * (?ATTRIBUTE_HEALTH_DEFAULT - ?ATTRIBUTE_HEALTH_MIN)
       )
       +
       (
          ?ATTRIBUTE_DEFENSE_SCORE_COST
-         * (?ATTRIBUTE_DEFENSE_SCORE_BASE - ?ATTRIBUTE_DEFENSE_SCORE_MIN)
+         * (?ATTRIBUTE_DEFENSE_SCORE_DEFAULT - ?ATTRIBUTE_DEFENSE_SCORE_MIN)
       )
-      + (?ATTRIBUTE_DODGE_COST * (?ATTRIBUTE_DODGE_BASE - ?ATTRIBUTE_DODGE_MIN))
    )
 ).
 
 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
 %% TYPES %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
--type defense_entry() :: {atom(), non_neg_integer()}.
-
 -record
 (
    proto_armor,
@@ -41,8 +43,9 @@
       health :: non_neg_integer(),
       damage_modifier :: non_neg_integer(),
       dodge :: non_neg_integer(),
-      defense :: list(defense_entry()),
-      defense_coef :: list(defense_entry()),
+      mvt_points :: non_neg_integer(),
+      defense :: list(blc_damage_type:entry()),
+      defense_coef :: list(blc_damage_type:coefficient()),
       defense_score :: non_neg_integer()
    }
 ).
@@ -55,111 +58,35 @@
 -export_type([proto_armor/0]).
 
 % FIXME: quick debug
--compile(export_all).
+-export
+(
+   [
+      increase_health_by/2,
+      increase_damage_modifier_by/2,
+      increase_dodge_chance_by/2,
+      increase_defense_score_by/2,
+      increase_movement_points_by/2,
+      increase_health_for/2,
+      increase_damage_modifier_for/2,
+      increase_dodge_chance_for/2,
+      increase_movement_points_for/2,
+      increase_defense_score_for/2,
+      set_defense_coefficients/2
+   ]
+).
+
+-export
+(
+   [
+      new/1,
+      get_spendable_points/0
+   ]
+).
 
 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
 %% LOCAL FUNCTIONS %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
 
--spec sort_defense_entries (list(defense_entry())) -> list(defense_entry()).
-sort_defense_entries (Entries) ->
-   lists:sort
-   (
-      fun ({_NameA, ValueA}, {_NameB, ValueB}) ->
-         (ValueA >= ValueB)
-      end,
-      Entries
-   ).
-
--spec calc_defense_score (list(defense_entry())) -> non_neg_integer().
-calc_defense_score (Defense) ->
-   {_LastIndex, Result} =
-      lists:foldl
-      (
-         fun ({_NameA, ValueA}, {Index, Current}) ->
-            {(Index + 1), (Current + (Index * ValueA))}
-         end,
-         {1, 0},
-         Defense
-      ),
-
-   Result.
-
--spec apply_defense_score_modifier
-   (
-      non_neg_integer(),
-      integer(),
-      list(defense_entry())
-   )
-   -> list(defense_entry()).
-apply_defense_score_modifier (AbsModifier, Mod, S0DescSortedDefense) ->
-   {S1DescSortedDefense, {_EndIndex, EndModifier}} =
-      lists:mapfoldl
-      (
-         fun ({Name, S0Value}, {Index, RemainingModifier}) ->
-            case ((RemainingModifier >= Index) and (S0Value > 0)) of
-               true ->
-                  {
-                     {Name, (S0Value + Mod)},
-                     {
-                        (Index + 1),
-                        RemainingModifier - Index
-                     }
-                  };
-
-               false -> {{Name, S0Value}, {(Index + 1), RemainingModifier}}
-            end
-         end,
-         {1, AbsModifier},
-         S0DescSortedDefense
-      ),
-
-   case (EndModifier > 0) of
-      false -> S1DescSortedDefense;
-      true ->
-         apply_defense_score_modifier
-         (
-            EndModifier,
-            Mod,
-            S1DescSortedDefense
-         )
-   end.
-
--spec generate_defense_worth
-   (
-      non_neg_integer(),
-      list(defense_entry())
-   )
-   -> list(defense_entry()).
-generate_defense_worth (TargetScore, SortedRatios) ->
-   [{T0, V0}, {T1, V1}, {T2, V2}] = SortedRatios,
-   Distribution = ((V0 + 2 * V1 + 3 * V2) / 100),
-   Base = TargetScore / Distribution,
-   UnderperformingDefense =
-      [
-         {T0, trunc(Base * (V0/100))},
-         {T1, trunc(Base * (V1/100))},
-         {T2, trunc(Base * (V2/100))}
-      ],
-   MissingScore = TargetScore - calc_defense_score(UnderperformingDefense),
-   case (MissingScore >= 0) of
-      true ->
-         apply_defense_score_modifier
-         (
-            MissingScore,
-            1,
-            UnderperformingDefense
-         );
-
-      false ->
-         apply_defense_score_modifier
-         (
-            (-1 * MissingScore),
-            -1,
-            UnderperformingDefense
-         )
-   end.
-
 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
 %% EXPORTED FUNCTIONS %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
@@ -244,6 +171,31 @@ increase_dodge_chance_by (Amount, Armor) ->
          }
    end.
 
+-spec increase_movement_points_by
+   (
+      non_neg_integer(),
+      proto_armor()
+   )
+   -> {proto_armor(), non_neg_integer()}.
+increase_movement_points_by (Amount, Armor) ->
+   NewMvtPoints = Armor#proto_armor.mvt_points + Amount,
+   case (NewMvtPoints > ?ATTRIBUTE_MOVEMENT_POINTS_MAX) of
+      true ->
+         {
+            Armor#proto_armor{ mvt_points = ?ATTRIBUTE_MOVEMENT_POINTS_MAX },
+            (
+               (?ATTRIBUTE_MOVEMENT_POINTS_MAX - Armor#proto_armor.mvt_points)
+               * ?ATTRIBUTE_MOVEMENT_POINTS_COST
+            )
+         };
+
+      false ->
+         {
+            Armor#proto_armor{ mvt_points = NewMvtPoints },
+            (Amount * ?ATTRIBUTE_MOVEMENT_POINTS_COST)
+         }
+   end.
+
 -spec increase_defense_score_by
    (
       non_neg_integer(),
@@ -251,7 +203,7 @@ increase_dodge_chance_by (Amount, Armor) ->
    )
    -> {proto_armor(), non_neg_integer()}.
 increase_defense_score_by (Amount, Armor) ->
-   NewDefenseScore = Armor#proto_armor.defense_score + Amount,
+   NewDefenseScore = (Armor#proto_armor.defense_score + Amount),
    case (NewDefenseScore > ?ATTRIBUTE_DEFENSE_SCORE_MAX) of
       true ->
          {
@@ -259,7 +211,7 @@ increase_defense_score_by (Amount, Armor) ->
             {
                defense_score = ?ATTRIBUTE_DEFENSE_SCORE_MAX,
                defense =
-                  generate_defense_worth
+                  blc_damage_type:generate_entries_from_score
                   (
                      NewDefenseScore,
                      Armor#proto_armor.defense_coef
@@ -277,7 +229,7 @@ increase_defense_score_by (Amount, Armor) ->
             {
                defense_score = NewDefenseScore,
                defense =
-                  generate_defense_worth
+                  blc_damage_type:generate_entries_from_score
                   (
                      NewDefenseScore,
                      Armor#proto_armor.defense_coef
@@ -286,3 +238,101 @@ increase_defense_score_by (Amount, Armor) ->
             (Amount * ?ATTRIBUTE_DEFENSE_SCORE_COST)
          }
    end.
+
+-spec set_defense_coefficients
+   (
+      list(blc_damage_type:coefficient()),
+      proto_armor()
+   )
+   -> proto_armor().
+set_defense_coefficients (Coefficients, Armor) ->
+   {Result, 0} =
+      increase_defense_score_by
+      (
+         0,
+         Armor#proto_armor
+         {
+            defense_coef = blc_damage_type:sort_entries(Coefficients)
+         }
+      ),
+
+   Result.
+
+-spec new (list(blc_damage_type:coefficient())) -> proto_armor().
+new (Coefficients) ->
+   {Result, _DefenseScoreIncreaseCost} =
+      increase_defense_score_by
+      (
+         ?ATTRIBUTE_DEFENSE_SCORE_MIN,
+         #proto_armor
+         {
+            health = ?ATTRIBUTE_HEALTH_MIN,
+            damage_modifier = ?ATTRIBUTE_DAMAGE_MODIFIER_MIN,
+            dodge = ?ATTRIBUTE_DODGE_CHANCE_MIN,
+            mvt_points = ?ATTRIBUTE_MOVEMENT_POINTS_MIN,
+            defense = [],
+            defense_coef = blc_damage_type:sort_entries(Coefficients),
+            defense_score = 0
+         }
+      ),
+
+   Result.
+
+-spec increase_health_for
+   (
+      non_neg_integer(),
+      proto_armor()
+   )
+   -> {proto_armor(), non_neg_integer()}.
+increase_health_for (GivenPoints, Armor) ->
+   AmountOfIncrease = trunc(GivenPoints / ?ATTRIBUTE_HEALTH_COST),
+   {Result, SpentPoints} = increase_health_by(AmountOfIncrease, Armor),
+   {Result, (GivenPoints - SpentPoints)}.
+
+-spec increase_damage_modifier_for
+   (
+      non_neg_integer(),
+      proto_armor()
+   )
+   -> {proto_armor(), non_neg_integer()}.
+increase_damage_modifier_for (GivenPoints, Armor) ->
+   AmountOfIncrease = trunc(GivenPoints / ?ATTRIBUTE_DAMAGE_MODIFIER_COST),
+   {Result, SpentPoints} = increase_damage_modifier_by(AmountOfIncrease, Armor),
+   {Result, (GivenPoints - SpentPoints)}.
+
+-spec increase_dodge_chance_for
+   (
+      non_neg_integer(),
+      proto_armor()
+   )
+   -> {proto_armor(), non_neg_integer()}.
+increase_dodge_chance_for (GivenPoints, Armor) ->
+   AmountOfIncrease = trunc(GivenPoints / ?ATTRIBUTE_DODGE_CHANCE_COST),
+   {Result, SpentPoints} = increase_dodge_chance_by(AmountOfIncrease, Armor),
+   {Result, (GivenPoints - SpentPoints)}.
+
+-spec increase_movement_points_for
+   (
+      non_neg_integer(),
+      proto_armor()
+   )
+   -> {proto_armor(), non_neg_integer()}.
+increase_movement_points_for (GivenPoints, Armor) ->
+   AmountOfIncrease = trunc(GivenPoints / ?ATTRIBUTE_MOVEMENT_POINTS_COST),
+   {Result, SpentPoints} = increase_movement_points_by(AmountOfIncrease, Armor),
+   {Result, (GivenPoints - SpentPoints)}.
+
+-spec increase_defense_score_for
+   (
+      non_neg_integer(),
+      proto_armor()
+   )
+   -> {proto_armor(), non_neg_integer()}.
+increase_defense_score_for (GivenPoints, Armor) ->
+   AmountOfIncrease = trunc(GivenPoints / ?ATTRIBUTE_DEFENSE_SCORE_COST),
+   {Result, SpentPoints} = increase_defense_score_by(AmountOfIncrease, Armor),
+   {Result, (GivenPoints - SpentPoints)}.
+
+
+-spec get_spendable_points () -> non_neg_integer().
+get_spendable_points () -> ?SPENDABLE_ARMOR_POINTS.