Rampant/libs/NEBaseUtils.lua

if neBaseUtilsG then
    return neBaseUtilsG
end
local ne = {}

-- imports

local constants = require("Constants")

-- imported constants

local BASE_ALIGNMENT_NE = constants.BASE_ALIGNMENT_NE
local ENABLED_BOBS_UNITS = constants.ENABLED_BOBS_UNITS

local BASE_ALIGNMENT_NE_BLUE = constants.BASE_ALIGNMENT_NE_BLUE
local BASE_ALIGNMENT_NE_RED = constants.BASE_ALIGNMENT_NE_RED
local BASE_ALIGNMENT_NE_PINK = constants.BASE_ALIGNMENT_NE_PINK
local BASE_ALIGNMENT_NE_GREEN = constants.BASE_ALIGNMENT_NE_GREEN
local BASE_ALIGNMENT_NE_YELLOW = constants.BASE_ALIGNMENT_NE_YELLOW

-- imported functions

local mMin = math.min

-- module code

local function fileEntity(baseAlignment, entity, evolutionTable, evo)
    local evoRequirement = mMin(evo or entity.prototype.build_base_evolution_requirement, 1)
    local eTable = evolutionTable[baseAlignment]
    if not eTable then
	eTable = {}
	evolutionTable[baseAlignment] = eTable
    end
    local aTable = eTable[evoRequirement]
    if not aTable then
	aTable = {}
	eTable[evoRequirement] = aTable
    end
    aTable[#aTable+1] = entity.name
end

function ne.processNEUnitClass(natives, surface)
    local position = { x = 0, y = 0 }

    local factionSet = {}

    local entity = surface.create_entity({
	    name = "biter-spawner",
	    position = position
    })
    fileEntity(BASE_ALIGNMENT_NE, entity, natives.evolutionTableUnitSpawner, 0.0)
    entity.destroy()

    entity = surface.create_entity({
	    name = "spitter-spawner",
	    position = position
    })
    fileEntity(BASE_ALIGNMENT_NE, entity, natives.evolutionTableUnitSpawner, 0.0)
    entity.destroy()


    if settings.startup["NE_Blue_Spawners"].value then
	entity = surface.create_entity({
		name = "ne-spawner-blue",
		position = position
	})
	fileEntity(BASE_ALIGNMENT_NE_BLUE, entity, natives.evolutionTableUnitSpawner, 0.0)
	factionSet[#factionSet+1] = BASE_ALIGNMENT_NE_BLUE
	entity.destroy()
    end

    if settings.startup["NE_Red_Spawners"].value then
	entity = surface.create_entity({
		name = "ne-spawner-red",
		position = position
	})
	fileEntity(BASE_ALIGNMENT_NE_RED, entity, natives.evolutionTableUnitSpawner, 0.0)
	factionSet[#factionSet+1] = BASE_ALIGNMENT_NE_RED
	entity.destroy()
    end

    if settings.startup["NE_Green_Spawners"].value then
	entity = surface.create_entity({
		name = "ne-spawner-green",
		position = position
	})
	fileEntity(BASE_ALIGNMENT_NE_GREEN, entity, natives.evolutionTableUnitSpawner, 0.0)
	factionSet[#factionSet+1] = BASE_ALIGNMENT_NE_GREEN
	entity.destroy()
    end


    if settings.startup["NE_Yellow_Spawners"].value then
	entity = surface.create_entity({
		name = "ne-spawner-yellow",
		position = position
	})
	fileEntity(BASE_ALIGNMENT_NE_YELLOW, entity, natives.evolutionTableUnitSpawner, 0.0)
	factionSet[#factionSet+1] = BASE_ALIGNMENT_NE_YELLOW
	entity.destroy()
    end

    if settings.startup["NE_Pink_Spawners"].value then
	entity = surface.create_entity({
		name = "ne-spawner-pink",
		position = position
	})
	fileEntity(BASE_ALIGNMENT_NE_PINK, entity, natives.evolutionTableUnitSpawner, 0.0)
	factionSet[#factionSet+1] = BASE_ALIGNMENT_NE_PINK
	entity.destroy()
    end

    factionSet[#factionSet+1] = BASE_ALIGNMENT_NE

    if ENABLED_BOBS_UNITS then
	entity = surface.create_entity({
		name = "bob-biter-spawner",
		position = position
	})
	fileEntity(BASE_ALIGNMENT_NE, entity, natives.evolutionTableUnitSpawner, 0.0)
	entity.destroy()

	entity = surface.create_entity({
		name = "bob-spitter-spawner",
		position = position
	})
	fileEntity(BASE_ALIGNMENT_NE, entity, natives.evolutionTableUnitSpawner, 0.0)
	entity.destroy()

	for _,alignment in ipairs(factionSet) do
	    entity = surface.create_entity({
		    name = "bob-big-fire-worm-turret",
		    position = position
	    })
	    fileEntity(alignment, entity, natives.evolutionTableWorm)
	    entity.destroy()

	    entity = surface.create_entity({
		    name = "bob-big-poison-worm-turret",
		    position = position
	    })
	    fileEntity(alignment, entity, natives.evolutionTableWorm)
	    entity.destroy()

	    entity = surface.create_entity({
		    name = "bob-big-piercing-worm-turret",
		    position = position
	    })
	    fileEntity(alignment, entity, natives.evolutionTableWorm)
	    entity.destroy()

	    entity = surface.create_entity({
		    name = "bob-big-electric-worm-turret",
		    position = position
	    })
	    fileEntity(alignment, entity, natives.evolutionTableWorm)
	    entity.destroy()

	    entity = surface.create_entity({
		    name = "bob-giant-worm-turret",
		    position = position
	    })
	    fileEntity(alignment, entity, natives.evolutionTableWorm)
	    entity.destroy()

	    entity = surface.create_entity({
		    name = "behemoth-worm-turret",
		    position = position
	    })
	    fileEntity(alignment, entity, natives.evolutionTableWorm)
	    entity.destroy()

	    entity = surface.create_entity({
		    name = "bob-big-explosive-worm-turret",
		    position = position
	    })
	    fileEntity(alignment, entity, natives.evolutionTableWorm)
	    entity.destroy()
	end
    end

    for _,alignment in ipairs(factionSet) do
	entity = surface.create_entity({
		name = "small-worm-turret",
		position = position
	})
	fileEntity(alignment, entity, natives.evolutionTableWorm)
	entity.destroy()

	entity = surface.create_entity({
		name = "medium-worm-turret",
		position = position
	})
	fileEntity(alignment, entity, natives.evolutionTableWorm)
	entity.destroy()

	entity = surface.create_entity({
		name = "big-worm-turret",
		position = position
	})
	fileEntity(alignment, entity, natives.evolutionTableWorm)
	entity.destroy()
    end
end

neBaseUtilsG = ne
return ne