Factorio Forums

Code: Select all

BUILDINGS={
    ["laboratory"]={need={beds=10},give={}},
    ['green-house']={need={beds=5},give={food=40}},
    ['water-well']={need={beds=1},give={water=50}},
    ['church']={need={beds=3},give={happiness=30}},
    ['habitat']={need={},give={beds=20}}
}
script.on_init(function()
    global.buildings={}
    global.active_index=0
    global.have={
        beds=100,
        food=100,
        water=100,
        happiness=100
    }
    global.satisfaction=1--multiplier for beds, pool
end)

local effects={
    add=function(effects)
        local have=global.have
        for k,v in pairs(effects.give) do
            have[k]=have[k]+v
        end
    end,
    rem=function(effects)
        local have=global.have
        for k,v in pairs(effects.give) do
            have[k]=have[k]-v
        end
    end
    take=function(effects,pool)
        local satisfied=true
        for k,v in pairs(effects.need) do
            pool[k]=pool[k]-v
            satisfied=pool[k]>0
        end
        return satisfied
    end,
}

building={
    add=function(ent)--create record for building comprised of its entity and effects
        local effects=BUILDINGS[ent.name]
        if not effects then return end
        local b={
            entity=ent,
            active=true,--this is shortcut to avoid calling factorio api without need, (maybe it is faster, not verified)
            effects=effects--this is shortcut to the location of the associated effects
        }
        effects.add(effects)
        table.insert(global.buildings,b)
    end,
    rem=function(ent)--remove building record by its entity
        if not BUILDINGS[ent.name] then return end
        local buildings=global.buildings
        for i=#buildings,1,-1 do
            if buildings[i].entity==e then
                effects.rem(buildings[i].effects)
                table.remove(buildings,i)
                break
            end
        end
    end,
    activate=function(building)
        if building.active then return end
        building.active=true
        building.entity.active=true
        effects.add(building.effects)
    end,
    deactivate=function(building)
        if not building.active then return end
        building.active=false
        building.entity.active=false
        effects.rem(building.effects)
    end,
}

local min=function(a,b) return (a<b) and a or b end
function recalculate_state()
    local have=global.have
    local pool={}--define available pool
    for k,v in pairs(have) do
        pool[k]=v
    end
    pool.beds=pool.beds*global.satisfaction
    
    local buildings=global.buildings
    for i=1,#buildings do--each building takes from pool
        local b=buildings[i]
        building.activate(b)
        local satisfied = effects.take(b.effects,pool)
        if not satisfied then building.deactivate(b) end--when no resources, work stops
    end
    
    local population=have.beds-pool.beds
    local wm=min ((pool.water-population),0)/population
    local fm=min ((pool.food-population),0)/population
    local hm=min ((pool.happiness-population),0)/population
    
    global.satisfaction=wm*fm*hm^2
end

local ev=defines.events
--without other mods intervention and without more complicated mechanics of resources in this mod
--these events are the only ones, that would cause the change in the state of the buildings
script.on_event({ev.on_built_entity,ev.on_robot_built_entity},function(event)
    building.add(event.created_entity)
    recalculate_state()
end)
script.on_event({ev.on_entity_died,ev.on_player_mined_entity,ev.on_robot_mined_entity},
function(event) 
    building.rem(event.entity)
    recalculate_state()
end)