#include "../../patcher_x86.hpp"
#include <cmath>
#include <stdio.h>

Patcher* _P;
PatcherInstance* _PI;

int BuildDepends_Castle[] =
{
0, -1,
1, 0, -1,
2, 1, -1,
3, 2, -1,
6, -1,
5, -1,
22, 5, -1,
14, -1,
15, 14, -1,
16, -1,
17, 6, -1,
7, -1,
8, 7, -1,
9, 8, -1,
11, 5, -1,
12, 11, 16, 0, 14, -1,
13, 12, 9, -1,
30, 7, -1,
37, 30, -1,
31, 30, -1,
38, 31, -1,
33, 30, 16, -1,
40, 33, -1,
32, 33, -1,
18, 32, -1,
39, 32, -1,
19, 39, -1,
21, 33, -1,
35, 21, -1,
42, 35, -1,
34, 33, 0, -1,
41, 34, -1,
36, 34, 1, -1,
43, 36, 2, -1,
26, -1,
-100
};

int BuildDepends_Inferno[] =
{
7, -1,
21, 7, -1,
8, 7, -1,
22, 8, -1,
9, 8, -1,
5, -1,
16, -1,
14, -1,
15, 14, -1,
0, -1,
23, 0, -1,
1, 0, -1,
2, 1, -1,
3, 2, -1,
4, 3, -1,
11, 5, -1,
12, 11, 16, 14, 0, -1,
13, 12, 9, -1,
30, 7, -1,
18, 30, -1,
37, 30, -1,
19, 37, -1,
32, 30, -1,
24, 32, -1,
39, 32, -1,
25, 39, -1,
31, 30, -1,
38, 31, -1,
33, 31, -1,
40, 33, -1,
35, 33, 0, -1,
42, 35, -1,
34, 33, -1,
41, 34, 0, -1,
36, 35, 33, 5, -1,
43, 36, -1,
26, -1,
-100
};

int BuildDepends_Rampart[] =
{
7, -1,
8, 7, -1,
9, 8, -1,
5, -1,
16, -1,
14, -1,
15, 14, -1,
0, -1,
1, 0, -1,
2, 1, -1,
3, 2, -1,
4, 3, -1,
17, -1,
21, 17, -1,
11, 5, -1,
12, 11, 0, 16, 14, -1,
13, 12, 9, -1,
30, 7, -1,
37, 30, -1,
31, 30, -1,
18, 31, -1,
22, 18, -1,
38, 31, -1,
19, 38, -1,
32, 30, -1,
39, 32, -1,
34, 32, -1,
24, 34, -1,
41, 34, -1,
25, 41, -1,
33, 32, -1,
40, 33, -1,
35, 33, 34, -1,
42, 35, -1,
36, 35, 0, -1,
43, 36, 2, -1,
26, -1,
-100
};

int BuildDepends_Dungeon[] =
{
7, -1,
8, 7, -1,
9, 8, -1,
22, -1,
23, -1,
5, -1,
16, -1,
14, -1,
15, 14, -1,
17, 14, -1,
0, -1,
21, 0, -1,
1, 0, -1,
2, 1, -1,
3, 2, -1,
4, 3, -1,
11, 5, -1,
12, 11, 16, 14, 0, -1,
13, 12, 9, -1,
30, 7, -1,
18, 30, -1,
37, 30, -1,
19, 37, -1,
32, 30, -1,
39, 32, -1,
31, 30, -1,
38, 31, -1,
33, 32, 31, -1,
40, 33, -1,
34, 33, -1,
41, 34, -1,
35, 33, -1,
42, 35, -1,
36, 34, 35, 0, -1,
43, 36, 2, -1,
26, -1,
-100
};

int BuildDepends_Stronghold[] =
{
7, -1,
17, 7, -1,
23, 7, -1,
8, 7, -1,
9, 8, -1,
5, -1,
16, -1,
22, 16, -1,
14, -1,
15, 14, -1,
21, 14, -1,
0, -1,
1, 0, -1,
2, 1, -1,
11, 5, -1,
12, 11, 0, 14, 16, -1,
13, 12, 9, -1,
30, 7, -1,
18, 30, -1,
37, 30, -1,
19, 37, -1,
31, 30, -1,
38, 31, 37, -1,
34, 31, -1,
41, 34, -1,
36, 34, 35, -1,
43, 36, -1,
32, 30, -1,
39, 32, 16, -1,
33, 32, -1,
40, 33, 0, -1,
35, 32, 34, -1,
42, 35, -1,
26, -1,
-100
};

int SecondarySkills_Mysticism[] =
{
1, 7, 14, 20
};

int SecondarySkills_Estates[] =
{
0, 250, 500, 750
};

float SecondarySkills_Necromancy[] =
{
0, 0.15, 0.30, 0.45
};

float SecondarySkills_Resistance[] =
{
0, 0.15, 0.25, 0.35
};

float SecondarySkills_Sorcery[] =
{
0, 0.1, 0.2, 0.3
};

float SecondarySkills_Learning[] =
{
0, 0.15, 0.3, 0.45
};



int __stdcall spellSlow(LoHook* h, HookContext* c)
{
float speed = (float)(*(int*)(c->ecx +196));
c->eax = floor( (speed * (*(float*)(c->ecx + 1224))) + 0.5 );

c->return_address = 0x448A2E;
return NO_EXEC_DEFAULT;
}

int __stdcall spellSlowMelee(LoHook* h, HookContext* c)
{
float speed = (float)(*(int*)(c->esi +196));
c->eax = floor( (speed * (*(float*)(c->esi + 1224))) + 0.5 );

c->return_address = 0x441E2D;
return NO_EXEC_DEFAULT;
}

int __stdcall libraryVisit(LoHook* h, HookContext* c)
{
if (*(char*)(c->esi + 201 + 4) > 0) {
char bonus = 1;

*(char*)(c->esi + 1142) += bonus; // атака
*(char*)(c->esi + 1143) += bonus; // защита
*(char*)(c->esi + 1144) += bonus; // колд. сила
*(char*)(c->esi + 1145) += bonus; // знания
}

if (*(char*)(c->esi + 201 + 4) > 0) {
char moreleBonus = *(char*)(c->esi + 201 + 4);
*(char*)(c->esi + 282) += moreleBonus;
}

return EXEC_DEFAULT;
}

float __stdcall getArtArmyValueModifier(int art[])
{
for (int i = 0; i < 19; ++i) {
if ( art[i] == 66 ) {
return 30;
}
}

return 0;
}

float __stdcall getArmyValueModifierOld(int diplomacyLevel, HookContext* c)
{
float armyModifier[4] = {0.00, 0.10, 0.15, 19.00};
float currentModifier = ( (diplomacyLevel >= 0) && (diplomacyLevel <= 3) ? armyModifier[diplomacyLevel] : 1 );

return currentModifier;
}

float __stdcall getArmyValueModifier(int baseHeroOffset)
{
int diplomacyLevel = (int)(*(char*)(baseHeroOffset + 201 + 4));
float armyModifier[4] = {0.00, 0.10, 0.2, 0.3};
float currentModifier = ( (diplomacyLevel >= 0) && (diplomacyLevel <= 3) ? armyModifier[diplomacyLevel] : 0 );

float armyArtModifier = 0;
for (int i = 0; i < 19; ++i) {
if ( *(int*)(baseHeroOffset + 301 + (i << 3)) == 66 ) {
armyArtModifier += 0.1;
}

if ( *(int*)(baseHeroOffset + 301 + (i << 3)) == 67 ) {
armyArtModifier += 0.1;
}

if ( *(int*)(baseHeroOffset + 301 + (i << 3)) == 68 ) {
armyArtModifier += 0.1;
}
}

return currentModifier + armyArtModifier;
}

int __stdcall setArmyValue(LoHook* h, HookContext* c)
{
int baseHeroOffset = c->edi;
float armyModifier = getArmyValueModifier(baseHeroOffset);
(int)c->eax *= (1 + armyModifier);

return EXEC_DEFAULT;
}


int __stdcall setArmyValuePreview(LoHook* h, HookContext* c)
{
int baseHeroOffset = c->esi;
float armyModifier = getArmyValueModifier(baseHeroOffset);
(int)c->eax *= (1 + armyModifier);

return EXEC_DEFAULT;
}

int __stdcall setArmyValueSplit(LoHook* h, HookContext* c)
{
int baseHeroOffset = *(int*)(c->ebp + 8);
float armyModifier = getArmyValueModifier(baseHeroOffset);
(int)c->eax *= (1 + armyModifier);

return EXEC_DEFAULT;
}

int __stdcall setSurrenderButton(LoHook* h, HookContext* c)
{
int heroOffset = *(int*)(c->esi + c->eax * 4 + 0x53CC);

if ( heroOffset == 0 ) {
return EXEC_DEFAULT;
}

char diplomacyLevel = *(char*)(*(int*)(c->esi + c->eax * 4 + 0x53CC) + 201 + 4);

if ( diplomacyLevel > 0 ) {
c->return_address = 0x477EB5;
return NO_EXEC_DEFAULT;
}

return EXEC_DEFAULT;
}

int __stdcall setSurrenderPrice(LoHook* h, HookContext* c)
{
//if ( *(int*)(*(int*)0x699420 + 0x132C0) == 0 ) {
int combatManager = *(int*)0x699420;
if ( *(int*)(combatManager + (0x14F4 - *(int*)(combatManager + 0x132C0)) * 4) == 0 ) {
switch ( (char)c->ecx ) {
case 1: *(float*)&c->edx = 0.7f; break;
case 2: *(float*)&c->edx = 0.8f; break;
case 3: *(float*)&c->edx = 0.9f; break;
default: c->edx = 0;
}
c->return_address = 0x4E480B;

return NO_EXEC_DEFAULT;
}

return EXEC_DEFAULT;
}

int __stdcall setSurrenderMinPrice(LoHook* h, HookContext* c)
{
int cmAddr = *(int*)0x699420; // адрес combatManager
int side = *(int*)(cmAddr + 0x132C0); // сторона в битве (0 - левая, 1 - правая)

float minimumPrice;
if ( *(int*)(cmAddr + (0x14F4 - side) * 4) == 0 ) minimumPrice = 1.0f; else minimumPrice = 0.9f;

if ( *(float*)(c->ebp - 4) > minimumPrice ) {
*(float*)(c->ebp - 4) = minimumPrice;
}

c->return_address = 0x4E493F;

return NO_EXEC_DEFAULT;
}

int __stdcall setLibraryVisitLevel(LoHook* h, HookContext* c)
{
int allLevelBonuses = 0;
if (*(short*)(c->esi + 85) < 10) {

for (int i = 0; i < 19; ++i) {
switch ( *(int*)(c->esi + 301 + (i << 3)) ) {
case 66: allLevelBonuses += 1; break;
case 67: allLevelBonuses += 1; break;
case 68: allLevelBonuses += 1; break;
}
}

if (*(char*)(c->esi + 201 + 4) > 0) {
allLevelBonuses += *(char*)(c->esi + 201 + 4) * 2;
}
}

int currentLevel = *(int*)(c->esi + 85);

c->eax = currentLevel + allLevelBonuses;
c->return_address = 0x4A2F55;

return NO_EXEC_DEFAULT;
}

int __stdcall setBaseHeroSpeed(LoHook* h, HookContext* c)
{
c->ecx += *(char*)(c->ebx + 201 + 0);
return EXEC_DEFAULT;
}

int __stdcall setBankUnitBonus(LoHook* h, HookContext* c)
{
int cmAddr = *(int*)0x699420;
int heroOffset = *(int*)(cmAddr + 0x53CC);

if (*(int*)(c->ebp + 16) != heroOffset) {
return EXEC_DEFAULT;
}

if ( *(char*)(*(int*)0x699420 + 0x53C5) == 1) { // проверка что бой в банке существ

if(*(char*)(heroOffset + 201 + 19) > 0) { // проверка на тактику
*(int*)(c->ebx + 196) += *(char*)(heroOffset + 201 + 19);
}

if(*(char*)(heroOffset + 201 + 20) > 0) { //проверка на артиллерию
if ( *(char*)(heroOffset + 301 + (13 << 3)) != -1 ) { //13 - слот на кукле для баллисты. -1 - пустой слот
*(int*)(c->ebx + 200) += *(char*)(heroOffset + 201 + 20);
}
}

if(*(char*)(heroOffset + 201 + 27) > 0) { //проверка на первую помощь
if ( *(char*)(heroOffset + 301 + (15 << 3)) == 6 ) { //15 - слот на кукле для палатки. 6 - айди палатки
*(int*)(c->ebx + 204) += *(char*)(heroOffset + 201 + 27);
}
}
}

return EXEC_DEFAULT;
}

int __stdcall resetTacticsUnitSpeedBonus(LoHook* h, HookContext* c)
{
int cmAddr = *(int*)0x699420;
if ( *(char*)(cmAddr + 0x53C5) == 1) { // проверка что бой в банке существ
char tacticLevel = *(char*)(*(int*)(cmAddr + 0x53CC) + 201 + 19);
int roundNumber = *(int*)(cmAddr + 0x132A0) - 30000;

if ((tacticLevel > 0) && (roundNumber == 1)) {
if ( c->ebx == cmAddr + 0x132A0 ) {
*(int*)(c->ecx + 196) -= tacticLevel;
}
}
}

return EXEC_DEFAULT;
}

int __stdcall spellPray(LoHook* h, HookContext* c)
{
c->ecx += 2;
c->ebx += 2;

int cmAddr = *(int*)0x699420;
char waterLevel = *(char*)(*(int*)(cmAddr + 0x53CC) + 201 + 16);

if (waterLevel > 1) {
*(int*)(c->esi + 0x458) = 1;
}

return EXEC_DEFAULT;
}

int __stdcall spellPrayEnd(LoHook* h, HookContext* c)
{
c->ecx -= 2;
c->ebx -= 2;

return EXEC_DEFAULT;
}

int __stdcall spellPrayBonus(LoHook* h, HookContext* c)
{
int spellDuration = *(int*)(c->ecx + 0x258); // Prayer
bool isUndead = *(int*)(c->ecx + 0x84) & 0x40000;
if (spellDuration != 0 && !isUndead) { // проверяем Prayer и бит Undead
c->eax = 1;
}

return EXEC_DEFAULT;
}

int __stdcall preCalculateDmg(LoHook* h, HookContext* c)
{
int cmAddr = *(int*)0x699420;
int heroOffset = *(int*)(cmAddr + 0x53CC);

return EXEC_DEFAULT;
}

int __stdcall setFirstAidBonus(LoHook* h, HookContext* c)
{
int cmAddr = *(int*)0x699420;
int heroOffset = *(int*)(cmAddr + 0x53CC);

if(*(char*)(heroOffset + 201 + 27) > 0) { //проверка на первую помощь
if ( *(char*)(heroOffset + 301 + (15 << 3)) == 6 ) { //15 - слот на кукле для палатки. 6 - айди палатки
*(int*)(c->ebx + 204) += *(char*)(heroOffset + 201 + 27) * 2;
}
}

return EXEC_DEFAULT;
}

int __stdcall setFirstAidShield(LoHook* h, HookContext* c)
{
int cmAddr = *(int*)0x699420;
float stackBonusModificator = 5.0f;


int currentSide = *(int*)(cmAddr + 0x132C0);
int heroAddr = *(int*)(cmAddr + currentSide * 4 + 0x53CC); // адрес героя с учётом стороны
char firstAidLevel = *(char*)(heroAddr + 201 + 27);

int heroID = *(int*)(heroAddr + 26);
int specInfoAddr = *(int*)0x679C80;
bool firstAidSpec = *(int*)(specInfoAddr + heroID * 40) == 0 && *(int*)(specInfoAddr + heroID * 40 + 4) == 27;


int specBonus = 0;
if (firstAidSpec) {
specBonus += 1;
}

int unitLevel = *(int*)(c->esi + 0x78) + 1 + specBonus;

int firstAidBonus = (5 + firstAidLevel * 5) * unitLevel;

int countMonsters = *(int*)(c->esi + 0x4C);
float countUnitsForShield = ceil(countMonsters / stackBonusModificator);
int countUnitBonus = countUnitsForShield * unitLevel;

int totalSheildCount = firstAidBonus + countUnitBonus;
if (*(int*)(c->esi + 0x58) == 0 && !firstAidSpec) {
totalSheildCount = totalSheildCount / 2;
}

*(int*)(c->esi + 0x94) = totalSheildCount;

return EXEC_DEFAULT;
}

int __stdcall absorbDamage(LoHook* h, HookContext* c)
{
if ( c->ecx <= *(int*)(c->esi + 0x94) ) {
*(int*)(c->esi + 0x94) -= c->ecx;
c->ecx = 1;
} else {
c->ecx -= *(int*)(c->esi + 0x94);
*(int*)(c->esi + 0x94) = 0;
}

return EXEC_DEFAULT;
}

int __stdcall getStackHp(LoHook* h, HookContext* c)
{
int unitAddr = *(int*)(c->ebp - 0x20);
int hpShieldValue = *(int*)(unitAddr + 0x94);

if ( hpShieldValue > 0 ) {
sprintf( (char*)0x697428, "%d [%d]", *(int*)(c->ebp + 8), hpShieldValue ); // display left hp [shield]
} else {
return EXEC_DEFAULT;
}

c->esp -= 12; // adjust esp*
c->return_address = 0x5F64C2;

return NO_EXEC_DEFAULT;
}

int __stdcall checkAllowHeal(LoHook* h, HookContext* c)
{
int cmAddr = *(int*)0x699420;
int currentSide = *(int*)(cmAddr + 0x132C0);
int heroAddr = *(int*)(cmAddr + currentSide * 4 + 0x53CC); // адрес героя с учётом стороны

if (*(char*)(heroAddr + 201 + 27) == 0) {
return EXEC_DEFAULT;
}

c->return_address = 0x4738C3;

return NO_EXEC_DEFAULT;
}

BOOL APIENTRY DllMain( HMODULE hModule,
DWORD ul_reason_for_call,
LPVOID lpReserved
)
{
static _bool_ plugin_On = 0;
switch (ul_reason_for_call)
{
case DLL_PROCESS_ATTACH:
if (!plugin_On)
{
plugin_On = 1;

_P = GetPatcher();
_PI = _P->CreateInstance("HD.Plugin.DedMorozzz");



*(int*)0x4EB816 = (int)BuildDepends_Castle;
*(int*)0x4EB8B3 = (int)BuildDepends_Rampart;
*(int*)0x4EBA39 = (int)BuildDepends_Inferno;
*(int*)0x4EBA5C = (int)BuildDepends_Dungeon;
*(int*)0x4EBA70 = (int)BuildDepends_Stronghold;

*(int*)0x4E41C8 = (int)SecondarySkills_Mysticism;
*(int*)0x4E461E = (int)SecondarySkills_Estates;
*(int*)0x4E4AC4 = (int)SecondarySkills_Learning;
*(int*)0x4E5B2C = (int)SecondarySkills_Sorcery;
*(int*)0x4E4967 = (int)SecondarySkills_Resistance;

//Necromancy
*(int*)0x4E3F59 = (int)SecondarySkills_Necromancy;
*(int*)0x4E3FE7 = 0x63EAE4; //amulet 0.05
*(int*)0x4E402D = 0x63B8D0; //cloak 0.10
*(int*)0x4E4074 = 0x63EB28; //boots 0.15
*(int*)0x4E4100 = 0x63B8D0; //amplifier 0.10

//specialization bonuses. Set 0.02
*(int*)0x4E3F92 = 0x63AC60; //Necromancy
*(int*)0x4E4F1E = 0x63AC60; //Logistic
*(int*)0x4E4569 = 0x63AC60; //Attack
*(int*)0x4E45C9 = 0x63AC60; //Defence

_PI->WriteLoHook(0x448A1A, spellSlow);
_PI->WriteLoHook(0x441E19, spellSlowMelee);

_PI->WriteDword((0x417236 + 1), 3); // при ПКМ
_PI->WriteHexPatch((0x4A755D), "EB"); // проходим мимо бесплатного присоединения
_PI->WriteHexPatch((0x4A75A9), "EB"); // проходим мимо платного присоединения

_PI->WriteHexPatch(0x417234, "EB"); // проходим мимо бесплатного присоединения (preview)
_PI->WriteHexPatch(0x417246, "EB"); // проходим мимо платного присоединения (preview)


_PI->WriteLoHook(0x4A7441, setArmyValue);
_PI->WriteLoHook(0x4171E6, setArmyValuePreview);
_PI->WriteLoHook(0x4AC35E, setArmyValueSplit);
_PI->WriteLoHook(0x477EA9, setSurrenderButton);
_PI->WriteLoHook(0x4E4804, setSurrenderPrice);
_PI->WriteLoHook(0x4E4931, setSurrenderMinPrice);


_PI->WriteLoHook(0x4A2F92, libraryVisit);
_PI->WriteLoHook(0x4A2F47, setLibraryVisitLevel);

_PI->WriteLoHook(0x4E4ED9, setBaseHeroSpeed);

_PI->WriteLoHook(0x43D492, setBankUnitBonus);
_PI->WriteLoHook(0x475A3D, resetTacticsUnitSpeedBonus);

_PI->WriteLoHook(0x44493C, spellPray);
_PI->WriteLoHook(0x4443F1, spellPrayEnd);
_PI->WriteLoHook(0x442F6A, spellPrayBonus);

_PI->WriteLoHook(0x478552, setFirstAidShield);
_PI->WriteLoHook(0x443DBB, absorbDamage);
_PI->WriteLoHook(0x5F64AF, getStackHp);

_PI->WriteHexPatch(0x47609A, "9090");
_PI->WriteLoHook(0x4738AC, checkAllowHeal);

/*
----------------------------------------
--- CmManager. Base address = 699420 ---
----------------------------------------
*/


}
break;

case DLL_THREAD_ATTACH:
case DLL_THREAD_DETACH:
case DLL_PROCESS_DETACH:
break;
}
return TRUE;
}

int __stdcall setFirstAidShield(LoHook* h, HookContext* c)
{
    int cmAddr =  *(int*)0x699420;
    char firstAidLevel = *(char*)(*(int*)(cmAddr + 0x53CC) + 201 + 27);
    int unitLevel = *(int*)(c->esi + 0x78) + 1;
    *(int*)(c->esi + 0x94) = unitLevel * firstAidLevel * 10;
    
    return EXEC_DEFAULT;
}

int __stdcall absorbDamage(LoHook* h, HookContext* c)
{
    if ( c->ecx <= *(int*)(c->esi + 0x94) ) {
        *(int*)(c->esi + 0x94) -= c->ecx;
        c->ecx = 1;
    } else {
        c->ecx -= *(int*)(c->esi + 0x94);
        *(int*)(c->esi + 0x94) = 0;
    }
    
    return EXEC_DEFAULT;
}

int __stdcall getStackHp(LoHook* h, HookContext* c)
{
    int unitAddr = *(int*)(c->ebp - 0x20);    // получаем адрес отряда, чьё информационное окно смотрим
    int hpShieldValue = *(int*)(unitAddr + 0x94);    // получаем значение щита

    if ( hpShieldValue > 0 ) {
        sprintf( (char*)0x697428, "%d [%d]", *(int*)(c->ebp + 8), hpShieldValue );    // отображаем остаток_здоровья [щит]
    }
    else return EXEC_DEFAULT;    // отображаем по умолчанию: остаток_здоровья

    c->esp -= 12;    // adjust esp*
    c->return_address = 0x5F64C2;

    return NO_EXEC_DEFAULT;
}

_PI->WriteLoHook(0x478552, setFirstAidShield);
            _PI->WriteLoHook(0x443DBB, absorbDamage);
            _PI->WriteLoHook(0x5F64AF, getStackHp);

int __stdcall setBankUnitBonus(LoHook* h, HookContext* c)
{
    if ( *(char*)(*(int*)0x699420 + 0x53C5) == 1) { // проверка что бой в банке существ
        if(*(char*)(c->ecx + 201 + 19) > 0) { // проверка на тактику
            *(int*)(c->esi+80) += *(char*)(c->ecx + 201 + 19);
        }
    }

    return EXEC_DEFAULT;
}

int __stdcall resetTacticsUnitSpeedBonus(LoHook* h, HookContext* c)
{
    int cmAddr =  *(int*)0x699420;
    if ( *(char*)(cmAddr + 0x53C5) == 1) { // проверка что бой в банке существ
        char tacticLevel = *(char*)(*(int*)(*(int*)0x699420 + 0x53CC) + 201 + 19);
        int roundNumber = *(int*)(cmAddr + 0x132A0) - 30000;

        if ((tacticLevel > 0) && (roundNumber == 1)) {
            if ( c->ebx == cmAddr + 0x132A0 ) {
                *(int*)(c->ecx + 196) -= tacticLevel;
            }
        }
    }

    return EXEC_DEFAULT;
}

int __stdcall setBankUnitBonus(LoHook* h, HookContext* c)
{
    if ( *(char*)(*(int*)0x699420 + 0x53C5) == 1) { // проверка что бой в банке существ
        if(*(char*)(c->ecx + 201 + 19) > 0) { // проверка на тактику
            *(int*)(c->esi+80) += *(char*)(c->ecx + 201 + 19);
        }

        if(*(int*)(c->ecx + 201 + 20) > 0) { //проверка на артиллерию
            if ( *(int*)(c->ecx + 301 + (13 << 3)) == 4 ) { //13 - слот на кукле для баллисты. 4 - айди баллисты
                *(int*)(c->esi + 84) += *(int*)(c->ecx + 201 + 20);
            }
        }

        if(*(int*)(c->ecx + 201 + 27) > 0) { //проверка на палатку
            if ( *(int*)(c->ecx + 301 + (15 << 3)) == 6 ) { //15 - слот на кукле для палатки. 6 - айди палатки
                *(int*)(c->esi + 88) += *(int*)(c->ecx + 201 + 27) * 2;
            }
        }
    }

    return EXEC_DEFAULT;
}

int __stdcall resetTacticsUnitSpeedBonus(LoHook* h, HookContext* c)
{
    int cmAddr =  *(int*)0x699420;
    if ( *(char*)(cmAddr + 0x53C5) == 1) { // проверка что бой в банке существ
        char tacticLevel = *(char*)(*(int*)(*(int*)0x699420 + 0x53CC) + 201 + 19);
        int roundNumber = *(int*)(cmAddr + 0x132A0) - 30000;

        if ((tacticLevel > 0) && (roundNumber == 1)) {
            if ( c->ebx == cmAddr + 0x132A0 ) {
                *(int*)(c->ecx + 196) -= tacticLevel;
            }
        }
    }

    return EXEC_DEFAULT;
}

float __stdcall getArmyValueModifier(int baseHeroOffset)
{
    int diplomacyLevel = (int)(*(char*)(baseHeroOffset + 201 + 4));
    float armyModifier[4] = {0.00, 0.10, 0.15, 19.00};
    float currentModifier = ( (diplomacyLevel >= 0) && (diplomacyLevel <= 3) ? armyModifier[diplomacyLevel] : 1 );

    float armyArtModifier = 0;
    for (int i = 0; i < 19; ++i) {
        if ( *(int*)(baseHeroOffset + 297 + (i << 3) + 4) == 66 ) {
            armyArtModifier += 40.05;
        }
    }

    return currentModifier + armyArtModifier;
}

int __stdcall setArmyValue(LoHook* h, HookContext* c)
{
    int baseHeroOffset = c->edi;
    float armyModifier = getArmyValueModifier(baseHeroOffset);
    (int)c->eax *= (1 + armyModifier);

    return EXEC_DEFAULT;
}


int __stdcall setArmyValuePreview(LoHook* h, HookContext* c)
{
    int baseHeroOffset = c->esi;
    float armyModifier = getArmyValueModifier(baseHeroOffset);
    (int)c->eax *= (1 + armyModifier);

    return EXEC_DEFAULT;
}

int __stdcall setArmyValueSplit(LoHook* h, HookContext* c)
{
    int baseHeroOffset = *(int*)(c->ebp + 8);
    float armyModifier = getArmyValueModifier(baseHeroOffset);
    (int)c->eax *= (1 + armyModifier);

    return EXEC_DEFAULT;
}

float __stdcall getArmyValueModificator(int diplomatyLevel)
{

    float armyModificator = 1;

    if (diplomatyLevel > 0) {
        switch (diplomatyLevel) {
            case 1: {
                armyModificator = 1.1;
                break;
            }
            
            case 2: {
                armyModificator = 1.15;
                break;
            }
            
            case 3: {
                armyModificator = 20;
                break;
            }
        }
    }

    return armyModificator;

}

int __stdcall setArmyValue(LoHook* h, HookContext* c)
{
    int diplomatyLevel = (int)(*(char*)(c->edi + 201 + 4));
    float armyModificator = getArmyValueModificator(diplomatyLevel);
    (int)c->eax *= armyModificator;

    return EXEC_DEFAULT;
}


int __stdcall setArmyValuePreview(LoHook* h, HookContext* c)
{
    int diplomatyLevel = (int)(*(char*)(c->esi + 201 + 4));
    float armyModificator = getArmyValueModificator(diplomatyLevel);
    (int)c->eax *= armyModificator;

    return EXEC_DEFAULT;
}