Unit Arrays (Spawning Method)

globals
    private constant integer MAX_LEVELS     = 20
    private constant integer MAX_UNIT_TYPES = 20 //Should be 1 more than the actual max, so you can use the 0th index for the count
    
    integer array UnitIds[MAX_LEVELS][MAX_UNIT_TYPES]
endglobals

function InitUnitTypes takes nothing returns nothing
    set UnitIds[1][0] = 5 //Level 1 has 5 unit types
    set UnitIds[1][1] = 'u000'
    set UnitIds[1][2] = 'u001'
    set UnitIds[1][3] = 'u002'
    set UnitIds[1][4] = 'u003'
    set UnitIds[1][5] = 'u004'
    
    set UnitIds[2][0] = 6 //Level 2 has 6 unit types
    set UnitIds[2][1] = 'u010'
    set UnitIds[2][2] = 'u011'
    set UnitIds[2][3] = 'u012'
    set UnitIds[2][4] = 'u013'
    set UnitIds[2][5] = 'u014'
    set UnitIds[2][6] = 'u015'
    
    set UnitIds[3][0] = 6 //Level 3 has 3 unit types
    set UnitIds[3][1] = 'u110'
    set UnitIds[3][2] = 'u111'
    set UnitIds[3][3] = 'u112'
    
    //You get the idea...
endfunction

function UseUnitTypes takes integer level returns nothing
    //Iterate over all the unit types for a level
    local integer i = 1 //We start at 1 since 0 is the UnitId count for each level
    
    loop
      exitwhen i > UnitIds[level][0]
        //Do something with UnitIds[level][i]
        set i = i + 1
    endloop
    
    //Or get a random UnitId from the level
    set i = GetRandomInt(1, UnitIds[0])
endfunction

SpawnSystemAddUnitType takes integer level, integer ut returns boolean
    local integer i = UnitIds[level][0] + 1
    if i>=MAX_UNIT_TYPES then
        debug call BJDebugMsg("Spawn System error: MAX_UNIT_TYPES is set too low for level "+I2S(level)+".")
        return false
    endif
    set UnitIds[level][0] = i
    set UnitIds[level][i] = ut
    return true
endfunction