#include "cj_types.j"

define
{
UnitDamageTarget = UserUnitDamageTarget
SPELL_MARK = 'A003' //РАВ-код способности "Метка (Damage System)"
SPELL_BUFF = 'B000' //РАВ-код эффекта для "Метка (Damage System)"
}

private constant trigger execute = CreateTrigger()
private constant group genum = CreateGroup()

private trigger array user
private boolexpr filter
private int count = 0

private damagetype bj_lastDamageType = null
private damagetype bj_userDamageType = null

private attacktype bj_lastAttackType = null
private attacktype bj_userAttackType = null

private unit bj_lastSourceDamage
private unit bj_lastTargetDamage
private real bj_lastCountsDamage


real GetDSCounts() {
return bj_lastCountsDamage
}

unit GetDSSource() {
return bj_lastSourceDamage
}

unit GetDSTarget() {
return bj_lastTargetDamage
}

private bool FilterEvent() {
if not (GetEventDamage() > 0.0) {return false}
bj_lastCountsDamage = GetEventDamage()
bj_lastSourceDamage = GetEventDamageSource()
bj_lastTargetDamage = GetTriggerUnit()
if (bj_lastDamageType == null && bj_lastAttackType == null) {
if (GetUnitAbilityLevel(bj_lastTargetDamage, SPELL_BUFF) != 0) {
bj_lastDamageType = DAMAGE_TYPE_NORMAL
if (IsUnitType(bj_lastSourceDamage, UNIT_TYPE_HERO) == true) {
bj_lastAttackType = ATTACK_TYPE_HERO
} else {
bj_lastAttackType = ATTACK_TYPE_MELEE
}
} else {
bj_lastDamageType = DAMAGE_TYPE_MAGIC
bj_lastAttackType = ATTACK_TYPE_MAGIC
}
}
UnitRemoveAbility(bj_lastTargetDamage, SPELL_BUFF)
int i = 0
while (i < count) {
if TriggerEvaluate(user[i]) {
TriggerExecute(user[i])
}; i++
}
bj_lastDamageType = null
bj_lastAttackType = null
return false
}

private bool CreateNewMember() {
unit targ = GetFilterUnit()
if (GetWidgetLife(targ) > 0.0 && GetUnitAbilityLevel(targ, SPELL_MARK) == 0) {
TriggerRegisterUnitEvent(execute, targ, EVENT_UNIT_DAMAGED)
UnitAddAbility(targ, SPELL_MARK)
}
targ = null
return false
}

int GetAttackTypeID (attacktype a_t) {
if (a_t == null) {return -1}
elseif (a_t == ATTACK_TYPE_NORMAL) {return 0}
elseif (a_t == ATTACK_TYPE_MELEE) {return 1}
elseif (a_t == ATTACK_TYPE_PIERCE) {return 2}
elseif (a_t == ATTACK_TYPE_SIEGE) {return 3}
elseif (a_t == ATTACK_TYPE_MAGIC) {return 4}
elseif (a_t == ATTACK_TYPE_CHAOS) {return 5}
elseif (a_t == ATTACK_TYPE_HERO) {return 6}
return -1
}

int GetDamageTypeID (damagetype d_t) {
if (d_t == null) {return -1}
elseif (d_t == DAMAGE_TYPE_UNKNOWN) {return 0}
elseif (d_t == DAMAGE_TYPE_NORMAL) {return 4}
elseif (d_t == DAMAGE_TYPE_ENHANCED) {return 5}
elseif (d_t == DAMAGE_TYPE_FIRE) {return 8}
elseif (d_t == DAMAGE_TYPE_COLD) {return 9}
elseif (d_t == DAMAGE_TYPE_LIGHTNING) {return 10}
elseif (d_t == DAMAGE_TYPE_POISON) {return 11}
elseif (d_t == DAMAGE_TYPE_DISEASE) {return 12}
elseif (d_t == DAMAGE_TYPE_DIVINE) {return 13}
elseif (d_t == DAMAGE_TYPE_MAGIC) {return 14}
elseif (d_t == DAMAGE_TYPE_SONIC) {return 15}
elseif (d_t == DAMAGE_TYPE_ACID) {return 16}
elseif (d_t == DAMAGE_TYPE_FORCE) {return 17}
elseif (d_t == DAMAGE_TYPE_DEATH) {return 18}
elseif (d_t == DAMAGE_TYPE_MIND) {return 19}
elseif (d_t == DAMAGE_TYPE_PLANT) {return 20}
elseif (d_t == DAMAGE_TYPE_DEFENSIVE) {return 21}
elseif (d_t == DAMAGE_TYPE_DEMOLITION) {return 22}
elseif (d_t == DAMAGE_TYPE_SLOW_POISON) {return 23}
elseif (d_t == DAMAGE_TYPE_SPIRIT_LINK) {return 24}
elseif (d_t == DAMAGE_TYPE_SHADOW_STRIKE) {return 25}
elseif (d_t == DAMAGE_TYPE_UNIVERSAL) {return 26}
return -1
}

damagetype GetEventDamageType() {
return bj_lastDamageType
}

attacktype GetEventAttackType() {
return bj_lastAttackType
}

bool UserUnitDamageTarget (unit c, unit t, float d, bool a_b, boolean r_b, attacktype a_t, damagetype d_t, weapontype w_t) {
bj_userAttackType = bj_lastAttackType; bj_lastAttackType = a_t
bj_userDamageType = bj_lastDamageType; bj_lastDamageType = d_t
undef UnitDamageTarget
bool result = UnitDamageTarget(c, t, d, a_b, r_b, a_t, d_t, w_t)
setdef UnitDamageTarget = UserUnitDamageTarget
bj_lastAttackType = bj_userAttackType
bj_lastDamageType = bj_userDamageType
return result
}

void TriggerRegisterAllUnitDamaged (trigger t) {
user[count] = t; count++
}

private void TimerLoop() {
GroupEnumUnitsInRect(genum, bj_mapInitialPlayableArea, filter)
}

private void InitSystem() {
filter = Condition(function CreateNewMember)
TriggerAddCondition(execute, Condition(function FilterEvent))
TimerStart(CreateTimer(), 0.1, true, function TimerLoop)
}

endlibrary