mirror of
https://github.com/ComfyFactory/ComfyFactorio.git
synced 2024-12-28 23:06:38 +02:00
109 lines
2.9 KiB
Lua
109 lines
2.9 KiB
Lua
local LootRaffle = require 'functions.loot_raffle'
|
|
|
|
local Public = {}
|
|
local random = math.random
|
|
local abs = math.abs
|
|
local floor = math.floor
|
|
local sqrt = math.sqrt
|
|
|
|
local blacklist = {
|
|
['atomic-bomb'] = true,
|
|
['cargo-wagon'] = true,
|
|
['car'] = true,
|
|
['tank'] = true,
|
|
['spidertron'] = true,
|
|
['locomotive'] = true,
|
|
['artillery-wagon'] = true,
|
|
['artillery-turret'] = true,
|
|
['landfill'] = true,
|
|
['fluid-wagon'] = true,
|
|
['pistol'] = true
|
|
}
|
|
|
|
function Public.get_distance(position)
|
|
local difficulty = sqrt(position.x ^ 2 + position.y ^ 2) * 0.0001
|
|
return difficulty
|
|
end
|
|
|
|
function Public.add(surface, position, chest)
|
|
local budget = 48 + abs(position.y) * 1.75
|
|
budget = budget * random(25, 175) * 0.01
|
|
|
|
if random(1, 128) == 1 then
|
|
budget = budget * 4
|
|
chest = 'crash-site-chest-' .. random(1, 2)
|
|
end
|
|
if random(1, 256) == 1 then
|
|
budget = budget * 4
|
|
chest = 'crash-site-chest-' .. random(1, 2)
|
|
end
|
|
|
|
budget = floor(budget) + 1
|
|
|
|
local amount = random(1, 5)
|
|
local base_amount = 12 * amount
|
|
local distance_mod = Public.get_distance(position)
|
|
|
|
local result = base_amount + budget + distance_mod
|
|
|
|
local c = game.entity_prototypes[chest]
|
|
local slots = c.get_inventory_size(defines.inventory.chest)
|
|
|
|
local item_stacks = LootRaffle.roll(result, slots, blacklist)
|
|
local container = surface.create_entity({name = chest, position = position, force = 'neutral'})
|
|
for _, item_stack in pairs(item_stacks) do
|
|
container.insert(item_stack)
|
|
end
|
|
container.minable = false
|
|
|
|
for _ = 1, 3, 1 do
|
|
if random(1, 8) == 1 then
|
|
container.insert({name = 'explosives', count = random(25, 50)})
|
|
else
|
|
break
|
|
end
|
|
end
|
|
end
|
|
|
|
function Public.add_rare(surface, position, chest, magic)
|
|
local budget = magic * 48 + abs(position.y) * 1.75
|
|
budget = budget * random(25, 175) * 0.01
|
|
|
|
if random(1, 128) == 1 then
|
|
budget = budget * 6
|
|
chest = 'crash-site-chest-' .. random(1, 2)
|
|
end
|
|
if random(1, 128) == 1 then
|
|
budget = budget * 6
|
|
chest = 'crash-site-chest-' .. random(1, 2)
|
|
end
|
|
|
|
local amount = random(1, 5)
|
|
local base_amount = 12 * amount
|
|
local distance_mod = Public.get_distance(position)
|
|
|
|
budget = floor(budget) + 1
|
|
|
|
local result = base_amount + budget + distance_mod
|
|
|
|
local c = game.entity_prototypes[chest]
|
|
local slots = c.get_inventory_size(defines.inventory.chest)
|
|
|
|
local item_stacks = LootRaffle.roll(result, slots, blacklist)
|
|
local container = surface.create_entity({name = chest, position = position, force = 'neutral'})
|
|
for _, item_stack in pairs(item_stacks) do
|
|
container.insert(item_stack)
|
|
end
|
|
container.minable = false
|
|
|
|
for _ = 1, 3, 1 do
|
|
if random(1, 8) == 1 then
|
|
container.insert({name = 'explosives', count = random(25, 50)})
|
|
else
|
|
break
|
|
end
|
|
end
|
|
end
|
|
|
|
return Public
|