library DamageEventLib
globals
    private constant group OrderGroup = CreateGroup( )
    private constant group TempG = CreateGroup( )
    private trigger trg  = null
    private unit bj_lastFilterUnit = null
    
    // specified
    private constant integer AbilityID   = 'Adef'
    private constant integer OrderIDUp   = 852055
    private constant integer OrderIDDown = 852056
endglobals

private function DamageEvent_Actions takes nothing returns nothing
    local real orientation
    local real lenght
    local real x
    local real y
    local real x1
    local real y1
    
    if IsUnitInGroup( GetTriggerUnit( ), OrderGroup ) then
        set udg_Damaged  = GetTriggerUnit( )
        set udg_Attacker = GetEventDamageSource( )
        set udg_Damage   = GetEventDamage( )
    
        set x1 = GetUnitX( udg_Damaged )
        set y1 = GetUnitY( udg_Damaged )
        set x  = GetUnitX( udg_Attacker )
        set y  = GetUnitY( udg_Attacker )
        set orientation = GetUnitFacing( udg_Damaged ) * bj_DEGTORAD 
        set lenght      = SquareRoot( ( x - x1 ) * ( x - x1 ) + ( y - y1 ) * ( y - y1 ) ) 
        set udg_Angle   = Acos( Cos( orientation ) * ( x - x1 ) / lenght + Sin( orientation ) * ( y - y1 ) / lenght ) * bj_RADTODEG
        
        call TriggerExecute( gg_trg_Negate )
    endif
endfunction

//===========================================================================

private function Order_Actions takes nothing returns nothing
    if GetUnitAbilityLevel( GetTriggerUnit( ), AbilityID ) > 0 then
        if GetIssuedOrderId( ) == OrderIDUp then
            call GroupAddUnit( OrderGroup, GetTriggerUnit( ) )
        elseif GetIssuedOrderId( ) == OrderIDDown then
            call GroupRemoveUnit( OrderGroup, GetTriggerUnit( ) )
        endif
    endif
endfunction

private function Death_Actions takes nothing returns nothing
    call GroupRemoveUnit( OrderGroup, GetTriggerUnit( ) )
endfunction

//===========================================================================
private function RegistEvent_2 takes nothing returns boolean
    set bj_lastFilterUnit = GetTriggerUnit( )
    if GetUnitAbilityLevel( bj_lastFilterUnit, 'Aloc' ) == 0 then
        call TriggerRegisterUnitEvent( trg, bj_lastFilterUnit, EVENT_UNIT_DAMAGED )
    endif
    return false
endfunction
private function RegistEvent_1 takes nothing returns boolean
    set bj_lastFilterUnit = GetFilterUnit( )
    if GetUnitAbilityLevel( bj_lastFilterUnit, 'Aloc' ) == 0 then
        call TriggerRegisterUnitEvent( trg, bj_lastFilterUnit, EVENT_UNIT_DAMAGED )
    endif
    return false
endfunction
private function RegistEvent takes nothing returns nothing
    if trg != null then
        call DestroyTrigger( trg )
    endif
    
    set trg = CreateTrigger( )
    call GroupEnumUnitsInRect( TempG, bj_mapInitialPlayableArea, Condition( function RegistEvent_1 ) )
    call TriggerAddCondition( trg, Condition( function DamageEvent_Actions ) )
    call TimerStart( GetExpiredTimer( ), 600.00, true, function RegistEvent )
endfunction

function InitTrig_DamageEvent takes nothing returns nothing
    local region rectRegion = CreateRegion( )
    local trigger t
    set gg_trg_DamageEvent  = CreateTrigger( )
    
    call RegionAddRect( rectRegion, bj_mapInitialPlayableArea )
    call TriggerRegisterEnterRegion( gg_trg_DamageEvent, rectRegion, null )
    call TriggerAddCondition( gg_trg_DamageEvent, Condition( function RegistEvent_2 ) )
    call TimerStart( CreateTimer( ), 0.00, false, function RegistEvent )
    
    set rectRegion = null
    
    set t = CreateTrigger( )
    call TriggerRegisterAnyUnitEventBJ( t, EVENT_PLAYER_UNIT_ISSUED_ORDER )
    call TriggerAddAction( t, function Order_Actions )
    
    set t = CreateTrigger( )
    call TriggerRegisterAnyUnitEventBJ( t, EVENT_PLAYER_UNIT_DEATH )
    call TriggerAddAction( t, function Death_Actions )
    
    set t = null
endfunction
endlibrary

//===========================================================================
// https://xgm.guru/p/wc3/negate-damage-lib
library NegateDamageLib
globals
    private constant group Group = CreateGroup()
    private constant timer Timer = CreateTimer()
    
    // MUST BE SPECIFIED
    public constant integer MaxLifeBonusAbility = 'A000'
    public constant integer MaxLifeBonus = 1000000
endglobals

native UnitAlive takes unit id returns boolean

private function ProcessUnits takes nothing returns nothing
    local unit u
    local real life2set
    
    loop
        set u = FirstOfGroup(Group)
        exitwhen u == null
        call GroupRemoveUnit(Group, u)
        
        if UnitAlive(u) then
            set life2set = GetWidgetLife(u)
            call UnitRemoveAbility(u, MaxLifeBonusAbility)
            call SetWidgetLife(u, life2set)
        else
            call UnitRemoveAbility(u, MaxLifeBonusAbility)
        endif
    endloop
endfunction

function NegateDamage takes unit u, real negated returns nothing
    local real life = GetWidgetLife(u)
    call GroupAddUnit(Group, u)
    call UnitAddAbility(u, MaxLifeBonusAbility)
    call SetWidgetLife(u, life + negated)
    
    call TimerStart(Timer, 0., false, function ProcessUnits)
endfunction

function GetUnitMaxHealth takes unit u returns real
    return GetUnitState(u, UNIT_STATE_MAX_LIFE) - MaxLifeBonus * GetUnitAbilityLevel(u, MaxLifeBonusAbility)
endfunction

endlibrary