TakeV
/
Factorio-Better-Air-Filtering
Archived
1
0
Fork 0
Code Issues Pull Requests 1 Packages Projects Releases Wiki Activity

disperse pollution back

This commit is contained in:
Joey De Pauw 2019-10-08 15:13:56 +02:00
parent a8ec7fa6f0
commit 71e7088258
4 changed files with 157 additions and 88 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

112
better-air-filtering/control.lua
View File

@ -21,7 +21,7 @@ function starts_with(str, start)
return str:sub(1, #start) == start
end
local function sign(x)
function sign(x)
if x == 0 then
return 1
else
@ -29,12 +29,12 @@ local function sign(x)
end
end
local function manhattan(x, y)
function manhattan(x, y)
-- Manhattan distance from origin to xy.
return math.abs(x) + math.abs(y)
end
local function positionToChunk(position)
function positionToChunk(position)
return { x = math.floor(position.x / 32), y = math.floor(position.y / 32) }
end
@ -77,6 +77,9 @@ function getAbsorptionRate(entity)
return math.min(getSpaceForPollution(entity), getSuctionRate(entity))
end
function pollutionInPollutedWater(amount)
return amount * 6 / 10
end
function getSpaceForPollution(entity)
if #entity.fluidbox < 1 then
@ -91,7 +94,7 @@ function getSpaceForPollution(entity)
return capacity - pollution
end
local function inRadius(filter, radius)
function inRadius(filter, radius)
if filter.name == "air-filter-machine-1" then
return radius <= 0
elseif filter.name == "air-filter-machine-2" then
@ -147,7 +150,7 @@ function absorbChunk(chunk)
end
end
local function stepsToOrigin(x, y)
function stepsToOrigin(x, y)
-- Provide coordinates of possible 1-steps toward (0, 0)
local steps = {}
if x ~= 0 then
@ -159,7 +162,7 @@ local function stepsToOrigin(x, y)
return steps
end
local function suctionUpdateChunk(chunkTo, dx, dy)
function suctionUpdateChunk(chunkTo, dx, dy)
local totalSuction = chunkTo:getTotalSuctionRate(manhattan(dx, dy))
if totalSuction == 0 then
@ -190,7 +193,7 @@ local function suctionUpdateChunk(chunkTo, dx, dy)
end
end
local function generateSuctionFunction(dx, dy)
function generateSuctionFunction(dx, dy)
local function suctionUpdate(event)
-- game.print("suck pollution " .. dx .. ", " .. dy)
@ -202,8 +205,7 @@ local function generateSuctionFunction(dx, dy)
return suctionUpdate
end
local function generateRadiusCoordinates(radius)
function generateRadiusCoordinates(radius)
local coords = {}
for signR = -1, 1, 2 do
for signX = -1, 1, 2 do
@ -220,7 +222,7 @@ local function generateRadiusCoordinates(radius)
return coords
end
local function generateRadiusSuctionFunctions(radius)
function generateRadiusSuctionFunctions(radius)
local functions = {}
for _, offset in pairs(generateRadiusCoordinates(radius)) do
@ -231,8 +233,7 @@ local function generateRadiusSuctionFunctions(radius)
return functions
end
local function generateFunctions()
function generateFunctions()
local functions = {}
table.insert(functions, absorbPollution)
@ -246,9 +247,7 @@ local function generateFunctions()
return functions
end
local function spreadOverTicks(functions, interval)
function spreadOverTicks(functions, interval)
local tickMap = {}
local funcs = {}
for index, f in pairs(functions) do
@ -412,7 +411,7 @@ function isAirFilterMachine(entity)
return starts_with(entity.name, "air-filter-machine-")
end
function OnEntityCreated(event)
function onEntityCreated(event)
if isAirFilterMachine(event.created_entity) then
local chunkPos = positionToChunk(event.created_entity.position)
local chunk = getFilteredChunk(event.created_entity.surface, chunkPos.x, chunkPos.y)
@ -420,17 +419,78 @@ function OnEntityCreated(event)
end
end
function OnEntityRemoved(event)
function onEntityRemoved(event)
-- Update map of air filters
if isAirFilterMachine(event.entity) then
local chunkPos = positionToChunk(event.entity.position)
local chunk = getFilteredChunk(event.entity.surface, chunkPos.x, chunkPos.y)
chunk:removeFilter(event.entity)
end
-- Disperse pollution back
local pollution = 0
-- from pollution (fluid)
pollution = pollution + event.entity.get_fluid_count("pollution")
-- from polluted water (fluid)
local pollutedWater = event.entity.get_fluid_count("polluted-water")
pollution = pollution + pollutionInPollutedWater(pollutedWater)
-- from current crafting op
--local crafting = "No"
--if event.entity.is_crafting() then crafting = "Yes" end
--game.print("crafting? " .. crafting)
--local recipe = event.entity.get_recipe()
--if recipe ~= nil and event.entity.is_crafting() then
-- for _, ingredient in pairs(recipe.ingredients) do
-- game.print("Ingredient: " .. ingredient.name)
-- if ingredient.name == "pollution" then
-- pollution = pollution + ingredient.amount
-- elseif ingredient.name == "polluted-water" then
-- pollution = pollution + pollutionInPollutedWater(ingredient.amount)
-- end
-- end
--end
if pollution > 0 then
game.print("Dispersing " .. pollution .. " pollution back")
event.entity.surface.pollute(event.entity.position, pollution)
end
script.on_event({ defines.events.on_built_entity, defines.events.on_robot_built_entity }, OnEntityCreated)
script.on_event({ defines.events.on_pre_player_mined_item, defines.events.on_robot_pre_mined, defines.events.on_entity_died }, OnEntityRemoved)
-- TODO disperse contents when recipe changes
end
function preEntityRemoved(event)
local isCrafting = event.entity.type == "assembling-machine" and event.entity.is_crafting()
-- disperse pollution in recipe back
if isCrafting then
local pollution = 0
local recipe = event.entity.get_recipe()
if recipe ~= nil then
for _, ingredient in pairs(recipe.ingredients) do
game.print("Ingredient: " .. ingredient.name)
if ingredient.name == "pollution" then
pollution = pollution + ingredient.amount
elseif ingredient.name == "polluted-water" then
pollution = pollution + pollutionInPollutedWater(ingredient.amount)
end
end
end
if pollution > 0 then
game.print("Dispersing " .. pollution .. " pollution back")
event.entity.surface.pollute(event.entity.position, pollution)
end
end
-- On entity died has no pre version -> call onEntityRemoved manually
if event.name == defines.events.on_entity_died then
onEntityRemoved(event)
end
end
function refreshMetatables()
for _, chunkListX in pairs(global.air_filtered_chunks_map) do
@ -456,18 +516,24 @@ function init()
chunk:addFilter(filter)
end
end
load()
end
function load()
refreshMetatables()
end
-- Set up callbacks
script.on_event({ defines.events.on_built_entity, defines.events.on_robot_built_entity }, onEntityCreated)
-- on_entity_died should trigger both functions -> called manually
script.on_event({ defines.events.on_player_mined_entity, defines.events.on_robot_mined_entity, defines.events.on_entity_died }, onEntityRemoved)
script.on_event({ defines.events.on_pre_player_mined_item, defines.events.on_pre_robot_mined_item, defines.events.on_entity_died }, preEntityRemoved)
local functions = generateFunctions()
local onTick = spreadOverTicks(functions, INTERVAL)
script.on_event(defines.events.on_tick, onTick)
end
script.on_load(load)
script.on_init(init)

3
better-air-filtering/prototypes/entity.lua
View File

@ -71,6 +71,7 @@ data:extend({
energy_usage = "100kW",
fixed_recipe = "filter-air",
ingredient_count = 1,
return_ingredients_on_change = true,
module_slots = 0,
allowed_effects=nil
},
@ -160,6 +161,7 @@ data:extend({
},
energy_usage = "50kW",
ingredient_count = 2,
return_ingredients_on_change = true,
module_slots = 0,
allowed_effects=nil
},
@ -247,6 +249,7 @@ data:extend({
},
energy_usage = "100kW",
ingredient_count = 2,
return_ingredients_on_change = true,
module_slots = 0,
allowed_effects=nil
}

4
better-air-filtering/prototypes/fluid.lua
View File

@ -10,7 +10,7 @@ data:extend({
icon = "__base__/graphics/icons/fluid/pollution.png",
icon_size = 32,
order = "a[fluid]-z[water]",
auto_barrel="false"
auto_barrel = false
},
{
type = "fluid",
@ -22,6 +22,6 @@ data:extend({
icon = "__better-air-filtering__/graphics/icons/fluid/pollution.png",
icon_size = 32,
order = "a[fluid]-z[pollution]",
auto_barrel="false"
auto_barrel = false
}
})

74
better-air-filtering/prototypes/recipe.lua
View File

@ -11,7 +11,7 @@ data:extend({
type = "recipe",
name = "air-filter-machine-1",
energy_required = 10.0,
enabled = "false",
enabled = false,
ingredients =
{
{ "assembling-machine-2", 1 },
@ -24,7 +24,7 @@ data:extend({
type = "recipe",
name = "air-filter-machine-2",
energy_required = 10.0,
enabled = "false",
enabled = false,
ingredients =
{
{ "air-filter-machine-1", 2 },
@ -36,7 +36,7 @@ data:extend({
type = "recipe",
name = "air-filter-machine-3",
energy_required = 10.0,
enabled = "false",
enabled = false,
ingredients =
{
{ "air-filter-machine-2", 2 },
@ -51,7 +51,7 @@ data:extend({
subgroup = "raw-material",
order = "f[plastic-bar]-f[expendable-air-filter]",
energy_required = 2,
enabled = "true",
enabled = true,
ingredients =
{
{ "coal", 5 },
@ -66,7 +66,7 @@ data:extend({
subgroup = "raw-material",
order = "f[plastic-bar]-f[air-filter]",
energy_required = 5,
enabled = "false",
enabled = true,
ingredients =
{
{ "coal", 10 },
@ -75,6 +75,35 @@ data:extend({
},
result = "air-filter"
},
{
type = "recipe",
name = "air-filter-recycling",
icons = {
{
icon = "__better-air-filtering__/graphics/icons/air-filter.png"
},
{
icon = "__better-air-filtering__/graphics/icons/used-air-filter-mask.png",
},
{
icon = "__better-air-filtering__/graphics/icons/recipe/recycle.png",
scale = 0.6,
shift = { 6, 6 }
},
},
icon_size = 32,
category = "crafting",
subgroup = "raw-material",
order = "f[air-filter]-f[air-filter-recycling]",
energy_required = 2,
enabled = true,
ingredients =
{
{ "used-air-filter", 1 },
{ "coal", 5 }
},
result = "air-filter"
},
{
type = "recipe",
name = "filter-air",
@ -94,7 +123,7 @@ data:extend({
subgroup = "raw-material",
order = "f[plastic-bar]-f[filter-air]",
energy_required = 1,
enabled = "true",
enabled = true,
ingredients =
{
{ type = "fluid", name = "pollution", amount = 2, fluidbox_index = 1 }
@ -120,7 +149,7 @@ data:extend({
subgroup = "raw-material",
order = "f[plastic-bar]-f[filter-air]",
energy_required = 5,
enabled = "true",
enabled = true,
ingredients =
{
{ type = "fluid", name = "pollution", amount = 20, fluidbox_index = 1 },
@ -136,41 +165,12 @@ data:extend({
subgroup = "raw-material",
order = "f[plastic-bar]-f[filter-air]",
energy_required = 1,
enabled = "true",
enabled = true,
ingredients =
{
{ type = "fluid", name = "pollution", amount = 6, fluidbox_index = 1 },
{ type = "fluid", name = "water", amount = 10, fluidbox_index = 2 }
},
results = { { type = "fluid", name = "polluted-water", amount = 10 } }
},
{
type = "recipe",
name = "air-filter-recycling",
icons = {
{
icon = "__better-air-filtering__/graphics/icons/air-filter.png"
},
{
icon = "__better-air-filtering__/graphics/icons/used-air-filter-mask.png",
},
{
icon = "__better-air-filtering__/graphics/icons/recipe/recycle.png",
scale = 0.6,
shift = { 6, 6 }
},
},
icon_size = 32,
category = "crafting",
subgroup = "raw-material",
order = "f[air-filter]-f[air-filter-recycling]",
energy_required = 5,
enabled = "false",
ingredients =
{
{ "used-air-filter", 1 },
{ "coal", 5 }
},
result = "air-filter"
}
})