Rampant/libs/BaseUtils.lua

local baseUtils = {}

-- imports

local mapUtils = require("MapUtils")
local constants = require("Constants")

-- constants

local BASE_DISTANCE_THRESHOLD = constants.BASE_DISTANCE_THRESHOLD

local BASE_ALIGNMENT_NEUTRAL = constants.BASE_ALIGNMENT_NEUTRAL

local CHUNK_BASE = constants.CHUNK_BASE

local MAGIC_MAXIMUM_NUMBER = constants.MAGIC_MAXIMUM_NUMBER

-- imported functions

local euclideanDistancePoints = mapUtils.euclideanDistancePoints

local tableRemove = table.remove

-- module code

local function calculateBaseMean(base)
    local chunks = base.chunks
    -- local xs = 0
    -- local ys = 0
    local strength = 0
    for _,aChunk in pairs(chunks) do
	-- xs = xs + aChunk.cX
	-- ys = ys + aChunk.cY
	--strength = strength + aChunk[ENEMY_BASE_GENERATOR]
    end
    -- local count = #chunks
    -- if (count ~= 0) then
    -- 	base.cX = xs / count
    -- 	base.cY = ys / count
    -- end
    base.strength = strength
end

function baseUtils.annexChunk(natives, chunk, tick, surface)
    local bases = natives.bases
    local annex = nil
    local closest = MAGIC_MAXIMUM_NUMBER
    for i=1,#bases do
	local base = bases[i]
	local distance = euclideanDistancePoints(base.cX, base.cY, chunk.cX, chunk.cY)
	if (distance <= BASE_DISTANCE_THRESHOLD) and (distance < closest) then
	    closest = distance
	    annex = base
	end
    end
    if (annex ~= nil) then
	return annex
    end
    return nil
end

function baseUtils.removeNest(natives, chunk)
    -- local base = chunk[CHUNK_BASE]
    -- local chunks = base.chunks
    -- for i=1, #chunks do
    -- 	local aChunk = chunks[i]
    -- 	if (chunk == aChunk) then
    -- 	    tableRemove(chunks, i)
    -- 	    break
    -- 	end
    -- end
    -- if (#chunks == 0) then
    -- 	local bases = natives.bases
    -- 	for i=1, #bases do
    -- 	    local aBase = bases[i]
    -- 	    if (base == aBase) then
    -- 		tableRemove(bases, i)
    -- 		break
    -- 	    end
    -- 	end
    -- else
    -- 	calculateBaseMean(base)
    -- end
    -- chunk[CHUNK_BASE] = nil
end

function baseUtils.buildOrder(base)
    local pattern = base.pattern
    
end

function baseUtils.createBase(natives, chunk, tick, surface)
    local bases = natives.bases
    local chunks = {}
    local cX = chunk.cX
    local cY = chunk.cY
    local base = {
	cX = cX,
	cY = cY,
	created = tick,
	alignment = { BASE_ALIGNMENT_NEUTRAL },
	nests = {},
	worms = {},
	chunks = chunks,
	upgradePoints = 0,
	growth = tick,
	pattern = 0,
	level = 0
    }
    chunks[#chunks+1] = chunk
    chunk[CHUNK_BASE] = base
    bases[#bases+1] = base
    local basePlacement = {
	name="biter-spawner",
	position={
	    chunk.pX,
	    chunk.pY
	}
    }
    print("creating base")
    if (surface.can_place_entity(basePlacement)) then
	print("created base")
	local nest = surface.create_entity(basePlacement)
	--addRemoveEnemyEntity
    end
    return base
end

function baseUtils.addBaseToChunk(chunk, entity, base)
    local bases = chunk[CHUNK_BASE]
    bases[#bases+1] = base
    base.nests[entity.unit_number] = entity
end

function baseUtils.removeBaseFromChunk(chunk, entity, base)
    if (chunk[ENEMY_BUILDING_COUNT] - 1 <= 0) then
	local bases = chunk[CHUNK_BASE]
	local baseIndex
	for i=1,#bases do
	    local b = bases[i]
	    if (b == base) then
		table.remove(bases, i)
		return true
	    end
	end
    else
	base.nests[entity.unit_number] = nil
    end
    return false
end

-- function baseUtils.mergeBases(natives)
--     local bases = natives.bases
--     for x=#bases,1,-1 do
-- 	local outerBase = bases[x]
-- 	if (outerBase ~= nil) then
-- 	    local outerChunks = outerBase.chunks
-- 	    for y=x-1,1,-1 do
-- 		local innerBase = bases[y]
-- 		if (innerBase ~= nil) then
-- 		    if (euclideanDistancePoints(outerBase.cX, outerBase.cY, innerBase.cX, innerBase.cY) < BASE_DISTANCE_THRESHOLD) then
-- 			local innerChunks = innerBase.chunks
-- 			for i=1,#innerChunks do
-- 			    local innerChunk = innerChunks[i]
-- 			    outerChunks[#outerChunks+1] = innerChunk
-- 			    innerChunk[CHUNK_BASE] = outerBase
-- 			end
-- 			bases[y] = nil
-- 		    end
-- 		end 
-- 	    end
-- 	end
--     end
--     for x=#bases,1,-1 do
-- 	if (bases[x] == nil) then
-- 	    tableRemove(bases, x)
-- 	else
-- 	    calculateBaseMean(bases[x])
-- 	end
--     end
-- end

return baseUtils