scope GravitonAcceleration // requires Damage, Timer32, GTrigger, GroupUtils

//===========================================================================
//                          CONFIGURABLES
//===========================================================================

globals
    private constant integer ABILID = 'ABGA' // raw code of ability "Graviton Acceleration"
    private constant integer DUMMYID = 'grAc' // raw code of unit "Graviton Acceleration Dummy"
    
    private constant real MAXSPEED = 200.0 // maximum drag speed
    private constant real SPINRATE = 2.0 // number of spins in the center area per second
    
    private constant string LIGHTNING = "DRAM" // lightning effect used, small note that lightning effect
                                               // can generate lag when dealing with mass units
    private constant boolean ALLOWLIGHT = true // true if lightning effect is used
    
    private constant attacktype ATK = ATTACK_TYPE_NORMAL // attack type of damage
    private constant damagetype DMG = DAMAGE_TYPE_MAGIC // damage type of damage
    private constant weapontype WEP = WEAPON_TYPE_WHOKNOWS // weapon type of damage
endglobals

private function GetCenterAoe takes integer level returns real
    return 100.0 // radius of center area
endfunction

private function GetAoe takes integer level returns real
    return 600.0 // area of effect of spell
endfunction

private function GetAbsorbAmount takes integer level returns real
    return 10.0 + (10.0 * level) // amount of mana absorbed per second
endfunction

private function GetDuration takes integer level returns real
    return 5.0 // duration of spell
endfunction
native UnitAlive takes unit id returns boolean

globals
    public group Checkgroup = CreateGroup()
endglobals

private struct Data
    integer level
    real x
    real y
    real duration
    real amount
    real angle
    real center
    real aoe
    group track
    unit caster
    unit dummy
    static Data tempData
    static hashtable Hash = InitHashtable()
    
    private static method groupFunc takes nothing returns nothing
        local thistype this = tempData
        local unit u = GetEnumUnit()
        local real x = GetUnitX(u)
        local real y = GetUnitY(u)
        local real dx = x - this.x
        local real dy = y - this.y
        local real a
        local real mana
        local real dist = SquareRoot(dx * dx + dy * dy)
        local integer id = GetHandleId(u)
        local boolean check
        
        if dist > this.aoe or not UnitAlive(u) then
            call DestroyLightning(LoadLightningHandle(Hash, this, id))
            call GroupRemoveUnit(this.track, u)
            call GroupRemoveUnit(Checkgroup, u)
        else
            set mana = GetUnitState(u, UNIT_STATE_MANA)
            if mana >= this.amount then
                call SetUnitState(u, UNIT_STATE_MANA, mana - this.amount)
            else
                call UnitDamageTargetEx(this.caster, u, this.amount, true, false, ATK, DMG, WEP)
                call SetUnitState(u, UNIT_STATE_MANA, 0)
            endif
            set check = mana >= 1
            set a = LoadReal(Hash, id, 0) + (this.angle)
            set dx = this.x + this.center * Cos(a)
            set dy = this.y + this.center * Sin(a)
            call MoveLightningEx(LoadLightningHandle(Hash, this, id), true, dx, dy, 0, x, y, GetUnitFlyHeight(u) + 50.0)
            call SaveReal(Hash, id, 0, a)
            if dist > this.center then
                set a = Atan2(this.y - y, this.x - x)
                if check then
                    set dist = MAXSPEED * T32_PERIOD
                else
                    set dist = MAXSPEED * (1.0 - (mana / GetUnitState(u, UNIT_STATE_MAX_MANA))) * T32_PERIOD
                endif
                set dx = x + dist * Cos(a)
                set dy = y + dist * Sin(a)
            endif
            if check then
                call SetUnitX(u, dx)
                call SetUnitY(u, dy)
            else
                call SetUnitPosition(u, dx, dy)
            endif
        endif
        set u = null
    endmethod
    
    private static method filterFunc takes nothing returns boolean
        local thistype this = tempData
        local unit u = GetFilterUnit()
        local real a
        local real x
        local real y
        
        if IsUnitEnemy(u, GetOwningPlayer(this.caster)) and UnitAlive(u) and not IsUnitInGroup(u, this.track) then
            set x = GetUnitX(u)
            set y = GetUnitY(u)
            set a = Atan2(y - this.y, x - this.x)
            static if ALLOWLIGHT then
                call SaveLightningHandle(Hash, this, GetHandleId(u), AddLightning(LIGHTNING, true, this.x + this.center + Cos(a), this.y + this.center + Sin(a), x, y))
            endif
            call SaveReal(Hash, GetHandleId(u), 0, a)
            call GroupAddUnit(this.track, u)
            call GroupAddUnit(Checkgroup, u)
        endif
        set u = null
        return false
    endmethod
    
    private static method groupFlush takes nothing returns nothing
        local thistype this = tempData
        local unit u = GetEnumUnit()
        local integer id = GetHandleId(u)
        
        static if ALLOWLIGHT then
            call DestroyLightning(LoadLightningHandle(Hash, this, id))
        endif
        call GroupRemoveUnit(Checkgroup, u)
        set u = null
    endmethod
    
    private method periodic takes nothing returns nothing
        set tempData = this
        if this.duration <= 0 then
            call KillUnit(this.dummy)
            call ForGroup(this.track, function thistype.groupFlush)
            call FlushChildHashtable(Hash, this)
            call ReleaseGroup(this.track)
            call this.stopPeriodic()
            call this.deallocate()
        else
            call GroupEnumUnitsInArea(ENUM_GROUP, this.x, this.y, this.aoe, Filter(function thistype.filterFunc))
            call ForGroup(this.track, function thistype.groupFunc)
            set this.duration = this.duration - T32_PERIOD
        endif
    endmethod
    implement T32x
    
    static method actions takes nothing returns nothing
        local thistype this = thistype.allocate()
        
        set this.caster = GetTriggerUnit()
        set this.level = GetUnitAbilityLevel(this.caster, ABILID)
        set this.x = GetSpellTargetX()
        set this.y = GetSpellTargetY()
        set this.duration = GetDuration(this.level)
        set this.amount = GetAbsorbAmount(this.level) * T32_PERIOD
        set this.angle = bj_PI * 2 * (T32_PERIOD / SPINRATE)
        set this.center = GetCenterAoe(this.level)
        set this.aoe = GetAoe(this.level)
        set this.track = NewGroup()
        set this.dummy = CreateUnit(GetOwningPlayer(this.caster), DUMMYID, this.x, this.y, 0)
        
        call this.startPeriodic()
    endmethod
    
    private static method onInit takes nothing returns nothing
        local trigger t = CreateTrigger()
        
        call GT_RegisterStartsEffectEvent(t, ABILID)
        call TriggerAddAction(t, function thistype.actions)
        set t = null
        
        //preload
        call RemoveUnit(CreateUnit(Player(13), DUMMYID, 0, 0, 0))
    endmethod
endstruct

endscope

function Trig_Trigger_Conditions takes nothing returns boolean
    if ( not ( GetSpellAbilityId() == 'ApuQ' ) ) then
        return false
    endif
    if ( not ( GetRandomInt(1, 100) >= 50 ) ) then
        return false
    endif
    return true
endfunction

function Trig_Trigger_Func004001003 takes nothing returns boolean
    return ( IsUnitType(GetTriggerUnit(), UNIT_TYPE_STRUCTURE) == false )
endfunction

function Trig_Trigger_Func004A takes nothing returns nothing
endfunction

function Trig_Trigger_Actions takes nothing returns nothing
    call ForGroupBJ( GetUnitsInRangeOfLocMatching(400.00, GetSpellTargetLoc(), Condition(function Trig_Trigger_Func004001003)), function Trig_Trigger_Func004A )
endfunction

//===========================================================================
function InitTrig_Trigger takes nothing returns nothing
    set gg_trg_Trigger = CreateTrigger(  )
    call TriggerRegisterAnyUnitEventBJ( gg_trg_Trigger, EVENT_PLAYER_UNIT_SPELL_EFFECT )
    call TriggerAddCondition( gg_trg_Trigger, Condition( function Trig_Trigger_Conditions ) )
    call TriggerAddAction( gg_trg_Trigger, function Trig_Trigger_Actions )
endfunction