include "cj_types.j" 
scope ForcePunch initializer init {
    define {
        private SPELL='A000' //RAW code ability
        private SPEED=20 //max speed
        private  MAX_D=600  //max distance
        private  EFFECT="Objects\\Spawnmodels\\Undead\\ImpaleTargetDust\\ImpaleTargetDust.mdl"
        // first effect on spell target location
        private  EFFECT2="Abilities\\Spells\\NightElf\\Taunt\\TauntCaster.mdl"
        // motion effect
        private  EFFECT3="Abilities\\Spells\\Human\\Polymorph\\PolyMorphTarget.mdl" 
        // effect for unit in range
        private  EFFECT4="Abilities\\Spells\\Human\\StormBolt\\StormBoltTarget.mdl"
        // stun effect
        private periodic = 0.035
        // periodic motion
        private RADIUS=250
        // . . . 
        private RADIUS_INC=50
        // increase radius per level
        private STUN_TIME=1
        // stun time
        private STUN_TIME_INC=1
        // stun time per level
    }
        /**************************************/
        define private XY1(u,x,y,a) = {SetUnitX(u, x+SPEED*Cos(a)); SetUnitY(u, y+SPEED*Sin(a))}
        define private XY(u) = {x=GetUnitX(u),y=GetUnitY(u)}
        define private TARG(targ,H,a,MAX_D,u,G) = {SaveReal(H, GetHandleId(targ),0, a);SaveReal(H, GetHandleId(targ),1, MAX_D);SaveUnitHandle(H, GetHandleId(targ),2, u);GroupAddUnit(G,targ)}
        /**************************************/
        group G=CreateGroup()
        group G2=CreateGroup()
        hashtable H=InitHashtable()
        /**************************************/
        private void stun () {
            unit u=GetEnumUnit()
            int i=GetHandleId(u)
            real time=LoadReal(H,i,4)-periodic
            real XY(u)
            if time>0 then
                SetUnitPosition(u,x,y)
                SaveReal(H, i,4, time)
            else
                effect e=LoadEffectHandle(H,i,5)
                DestroyEffect(e)
                GroupRemoveUnit(G2,u)
                FlushChildHashtable(H,i)
            endif
            u=null
            e=null
        }
        /**************************************/
        private void motion () {
            unit u=GetEnumUnit()
            int i=GetHandleId(u)
            real a=LoadReal(H,i,0),d=LoadReal(H,i,1),XY(u)
            if d>0 then
                XY1(u,x,y,a)
                SaveReal(H, i,1, d-SPEED)
                DestroyEffect(AddSpecialEffect(EFFECT,x,y))
            else
                SaveReal(H, i,4, STUN_TIME+STUN_TIME_INC*(GetUnitAbilityLevel(u,SPELL)))
                SaveEffectHandle(H, i,5, AddSpecialEffectTarget(EFFECT4,u,"overhead"))
                GroupRemoveUnit(G,u); GroupAddUnit(G2,u)
            endif
            u=null
        }
        /**************************************/
        private void period () {
            ForGroup(G,function motion)
            ForGroup(G2,function stun)
        }
        /**************************************/
        private bool cond () {
        return GetSpellAbilityId() == SPELL
        }
        /**************************************/
        private void action () {
            unit u=GetTriggerUnit()
            unit t=GetSpellTargetUnit()
            unit targ
            real X=GetUnitX(t)
            real Y=GetUnitY(t) 
            real a
            real x
            real y
            group g=CreateGroup()
            GroupEnumUnitsInRange(g , X, Y, RADIUS+RADIUS_INC*(GetUnitAbilityLevel(u,SPELL)), null)
            DestroyEffect(AddSpecialEffect(EFFECT2,X,Y))
            set targ=t
            x=GetUnitX(targ)
            y=GetUnitY(targ)
            a=Atan2(y-GetUnitY(u),x-GetUnitX(u))
            TARG(targ,H,a,MAX_D,u,G) 
            GroupRemoveUnit(g,targ)
            loop {
                targ = FirstOfGroup(g)
                x=GetUnitX(targ)
                y=GetUnitY(targ)
                a=Atan2(y-Y,x-X)
                exitwhen targ == null
                if  IsPlayerEnemy(GetOwningPlayer(targ), GetOwningPlayer(u)) == true and IsUnitType(targ, UNIT_TYPE_STRUCTURE) == false then
                    DestroyEffect(AddSpecialEffect(EFFECT3,x,y))
                    SaveReal(H, GetHandleId(targ),0, a)
                    SaveReal(H, GetHandleId(targ),1, MAX_D)
                    SaveUnitHandle(H, GetHandleId(targ),2, u)
                    GroupAddUnit(G,targ)
                endif
                GroupRemoveUnit(g, targ)
            }
            DestroyGroup(g)
            
            g=null; targ=null; u=null;t=null
        }
        /**************************************/
    private void init () {
        int i=0;trigger t=CreateTrigger();timer tr=CreateTimer()
        TimerStart(tr,periodic,true,function period)
        loop {
            exitwhen i==15
            TriggerRegisterPlayerUnitEvent(t, Player(i), EVENT_PLAYER_UNIT_SPELL_CAST, null)
            i += 1
        }
        TriggerAddCondition( t, Condition( function cond ) )
        TriggerAddAction( t, function action )
        tr=null;t=null
    }
}