library DamageDetection initializer DamDetect_Init, needs Tables, CasterSystem

//******************************************************************************** **********
//*
//* emjlr3's Damage Detection Engine
//*
//*
//* A fairly simple system that allows you to detect when units
//* in your map take damage, and do with them what you want. Use
//* the function DamDetect_Attacks and DamDetect_Spells to que your
//* functions to be ran at attack or other damage.
//*
//* Requires:
//* - "emjlr3s CS" setup correctly in your map
//* - The dummy orb ability in this map and its current buff updated in your map
//* - This trigger, ofcourse
//*
//******************************************************************************** **********

//================================================================================ =============================================
// Configuration :
//
globals
// Your dummy orb abilities rawcode
private constant integer DamDetect_Abil = 'A004'

// Your dummy orb abilities buffs rawcode
private constant integer DamDetect_Buff = 'B000'

endglobals

//================================================================================ =============================================
// Damage Detection Engine functions. Don't mess down here unless you know what you are doing!!!

globals
//needed globals:
private integer AttackCount = 0
private integer SpellCount = 0
private string array Attacks
private string array Spells
private string DamageType

private unit U
endglobals


//**********Main function to run the functions that need to be ran for each damage type
function DamDetect_Damaged takes nothing returns boolean
local integer i = 1
local string funcName
set U = GetTriggerUnit()

if GetEventDamage()>.1 then
if GetUnitAbilityLevel(U, DamDetect_Buff) > 0 then
call UnitRemoveAbility(U, DamDetect_Buff)
set DamageType = "attack"
loop
exitwhen i > AttackCount
call ExecuteFunc(Attacks[i])
set i = i + 1
endloop
else
set DamageType = "spell"
loop
exitwhen i > SpellCount
call ExecuteFunc(Spells[i])
set i = i + 1
endloop
endif
endif

return false
endfunction
//Returns the damage cause, incase you need to know
function DamDetect_GetDamageEventCause takes nothing returns string
return DamageType
endfunction
//Use this to ignore the damage done by the unit, avoids infinite loops in some cases
function DamDetect_Dummy takes unit u, string whichState returns nothing
if whichState=="ignore" then
call SetTableInt("DamDetect_"+I2S(CS_H2I(u)),"DamDetect_Dummy",1)
elseif whichState=="allow" then
call SetTableInt("DamDetect_"+I2S(CS_H2I(u)),"DamDetect_Dummy",0)
endif
endfunction

//**********Sets up the triggers to run when a unit takes damage, call either of these in the triggers init trigger
//for it to run when a unit takes damage
function DamDetect_Attacks takes string funcName returns nothing
//used for attack damage
set AttackCount = AttackCount + 1

set Attacks[AttackCount] = funcName
endfunction
function DamDetect_Spells takes string funcName returns nothing
//used for spell damage
set SpellCount = SpellCount + 1

set Spells[SpellCount] = funcName
endfunction

//**********Base functions for the system
//Filter for units you do not want to be checked to add or remove the ability when they enter the map, or when they die
function DamDetect_Filter takes nothing returns boolean
set U = GetTriggerUnit()
if GetUnitTypeId(U)==CasterUnitId then
return false
endif
if GetUnitTypeId(U)=='n02G' or GetUnitTypeId(U)=='e006' or GetUnitTypeId(U)=='e005' or GetUnitTypeId(U)=='h03J' then
return false
endif
return true
endfunction
//Add damage triggers for units who enter map
function DamDetect_AddTriggers takes nothing returns nothing
local trigger dam_trig = CreateTrigger()
set U = GetTriggerUnit()

if IsUnitType(U,UNIT_TYPE_STRUCTURE)==true then
call UnitAddAbility(U, DamDetect_Abil)
call UnitMakeAbilityPermanent(U,true,DamDetect_Abil)
else
call TriggerRegisterUnitEvent(dam_trig, U,EVENT_UNIT_DAMAGED)
call TriggerAddCondition(dam_trig,Condition(function DamDetect_Damaged))
call SetTableObject("DamDetect_"+I2S(CS_H2I(U)),"dam_trig",dam_trig)
endif

set dam_trig = null
endfunction
//Remove triggers from units who die
function DamDetect_RemoveTriggers takes nothing returns nothing
local unit u = GetTriggerUnit()
local string s

if GetWidgetLife(u)<.405 and GetUnitAbilityLevel(u,'A00U')==0 and IsUnitType(u,UNIT_TYPE_STRUCTURE)==false then
set s = "DamDetect_"+I2S(CS_H2I(u))
call DestroyTrigger(GetTableTrigger(s,"dam_trig"))
call FlushStoredMission(cscache,s)
set u = null
return
endif
endfunction
function DamDetect_Upgrades takes nothing returns nothing
local integer id

set U = GetTriggerUnit()
set id = GetUnitTypeId(U)

if id=='h00X' or id=='h01A' or id=='h01B' or id=='h01C' then //etc., for a tower that does not currently have the ability, add it
call UnitAddAbility(U, DamDetect_Abil)
call UnitMakeAbilityPermanent(U,true,DamDetect_Abil)
endif

if id=='h002' or id=='h00Y' or id=='h02X' or id=='h012' or id=='h014' or id=='h02F' or id=='h02P' or id=='n02G' then //etc., for a tower that has it, but we do not want it to anymore
call UnitMakeAbilityPermanent(U,false,DamDetect_Abil)
call UnitRemoveAbility(U,DamDetect_Abil)
endif
endfunction
//Basically just sets up the system for all units preplaced that may be used,
//for units that ever enter the map, and removes triggers when units die
function DamDetect_Init takes nothing returns nothing
local trigger entmap_trig = CreateTrigger()
local trigger death_trig = CreateTrigger()
local trigger upg_trig = CreateTrigger()
local group g = CreateGroup()

call GroupEnumUnitsInRect(g,bj_mapInitialPlayableArea,Condition(function DamDetect_Filter))
call ForGroup(g,function DamDetect_AddTriggers)

call TriggerRegisterEnterRectSimple(entmap_trig, bj_mapInitialPlayableArea)
call TriggerAddCondition(entmap_trig,Condition(function DamDetect_Filter))
call TriggerAddAction(entmap_trig,function DamDetect_AddTriggers)

call TriggerRegisterAnyUnitEventBJ(death_trig,EVENT_PLAYER_UNIT_DEATH)
call TriggerAddCondition(death_trig,Condition(function DamDetect_Filter))
call TriggerAddAction(death_trig,function DamDetect_RemoveTriggers)

call TriggerRegisterAnyUnitEventBJ(upg_trig,EVENT_PLAYER_UNIT_UPGRADE_FINISH)
call TriggerAddAction(upg_trig,function DamDetect_Upgrades)

set entmap_trig = null
set death_trig = null
set upg_trig = null
call DestroyGroup(g)
set g = null
endfunction

endlibrary