library SAS initializer InitSAS

globals
    private integer array s4a
    private integer array c4a
endglobals

function GetUnitBonusAttack takes unit w returns integer
    local integer i = 1
    local integer r = 0
    loop
        exitwhen (i > 9)
        if (GetUnitAbilityLevel(w, s4a[i]) != 0) then
            set r = r + c4a[i]
        endif
        set i = i + 1
    endloop
    return r
endfunction

function AddUnitBonusAttack takes unit w, integer c returns nothing
    local integer i = 1
    set c = GetUnitBonusAttack(w) + c
    loop
        exitwhen (i > 9)
        call UnitRemoveAbility(w, s4a[i])
        set i = i + 1
    endloop
    set i = 1
    loop
        exitwhen (i > 9)
        if (c4a[i] <= c) then
            call UnitAddAbility(w, s4a[i])
            set c = c - c4a[i]
        endif
        set i = i + 1
    endloop
endfunction

private function InitSAS takes nothing returns nothing
    //Создаём базу данных -> заполняем массивы...
endfunction

endlibrary