adding layout support

2025-12-08 16:08:07 -06:00 · 2022-10-17 01:09:55 -04:00
parent 7757a04f91
commit 1219a211f4
5 changed files with 168 additions and 77 deletions
--- a/cybersyn/scripts/layout.lua
+++ b/cybersyn/scripts/layout.lua
@@ -0,0 +1,151 @@
+
+
+
+function update_train_layout(map_data, train)
+	local carriages = train.entity.carriages
+	local layout = ""
+	local i = 1
+	local item_slot_capacity = 0
+	local fluid_capacity = 0
+	for _, carriage in pairs(carriages) do
+		if carriage.type == "cargo-wagon" then
+			layout = layout..TRAIN_LAYOUT_CARGO
+			local inv = carriage.get_inventory(defines.inventory.cargo_wagon)
+			item_slot_capacity = item_slot_capacity + #inv
+		elseif carriage.type == "fluid-wagon" then
+			layout = layout..TRAIN_LAYOUT_FLUID
+			fluid_capacity = fluid_capacity + carriage.prototype.fluid_capacity
+		--elseif carriage.type == "artillery-wagon" then
+			--layout = layout..TRAIN_LAYOUT_ARTILLERY
+		else
+			layout = layout..TRAIN_LAYOUT_NA
+		end
+		i = i + 1
+	end
+	local layout_id = 0
+	for id, cur_layout in pairs(map_data.layouts) do
+		if layout == cur_layout then
+			layout = cur_layout
+			layout_id = id
+			break
+		end
+	end
+	if layout_id == 0 then
+		--define new layout
+		layout_id = map_data.layout_top_id
+		map_data.layout_top_id = map_data.layout_top_id + 1
+
+		map_data.layouts[layout_id] = layout
+		map_data.layout_train_count[layout_id] = 1
+		for _, station in pairs(map_data.stations) do
+			if string.find(layout, station.layout_pattern) ~= nil then
+				station.accepted_layouts[layout_id] = true
+			end
+		end
+	else
+		map_data.layout_train_count[layout_id] = map_data.layout_train_count[layout_id] + 1
+	end
+	train.layout_id = layout_id
+	train.item_slot_capacity = item_slot_capacity
+	train.fluid_capacity = fluid_capacity
+end
+
+local area = require("__flib__.area")
+function reset_station_layout(map_data, station)
+	--station.entity
+	local station_rail = station.entity.connected_rail
+	local rail_direction_from_station
+	if station.entity.connected_rail_direction == defines.rail_direction.front then
+		rail_direction_from_station = defines.rail_direction.back
+	else
+		rail_direction_from_station = defines.rail_direction.front
+	end
+	local station_direction = station.entity.direction
+	local surface = station.entity.surface
+	local middle_x = station_rail.position.x
+	local middle_y = station_rail.position.y
+	local reach = LONGEST_INSERTER_REACH + 1 - DELTA
+	local search_area
+	local area_delta
+	local direction_filter
+	if station_direction == defines.direction.north then
+		search_area = {left_top = {x = middle_x - reach, y = middle_y}, right_bottom = {x = middle_x + reach, y = middle_y - 6}}
+		area_delta = {x = 0, y = -7}
+		direction_filter = {defines.direction.east, defines.direction.west}
+	elseif station_direction == defines.direction.east then
+		search_area = {left_top = {y = middle_y - reach, x = middle_x}, right_bottom = {y = middle_y + reach, x = middle_x - 6}}
+		area_delta = {y = 0, x = -7}
+		direction_filter = {defines.direction.north, defines.direction.south}
+	elseif station_direction == defines.direction.south then
+		search_area = {left_top = {x = middle_x - reach, y = middle_y + 6}, right_bottom = {x = middle_x + reach, y = middle_y}}
+		area_delta = {x = 0, y = 7}
+		direction_filter = {defines.direction.east, defines.direction.west}
+	elseif station_direction == defines.direction.west then
+		search_area = {left_top = {y = middle_y - reach, x = middle_x + 6}, right_bottom = {y = middle_y + reach, x = middle_x}}
+		area_delta = {y = 0, x = 7}
+		direction_filter = {defines.direction.north, defines.direction.south}
+	else
+		assert(false, "cybersyn: invalid station direction")
+	end
+	local length = 2
+	local pre_rail = station_rail
+	local layout_pattern = "^"
+	local layout_min_size = 10000
+	local type_filter = {"inserter", "pump"}
+	for i = 1, 100 do
+		local rail, rail_direction, rail_connection_direction = pre_rail.get_connected_rail({rail_direction = rail_direction_from_station, rail_connection_direction = defines.rail_connection_direction.straight})
+		if rail_connection_direction ~= defines.rail_connection_direction.straight or not rail.valid then
+			break
+		end
+		length = length + 2
+		if length%7 <= 1 then
+			local supports_cargo = false
+			local supports_fluid = false
+			local entities = surface.find_entities_filtered({
+				area = search_area,
+				type = type_filter,
+				direction = direction_filter,
+			})
+			for _, entity in pairs(entities) do
+				if entity.type == "inserter" then
+					--local pickup_pos = entity.prototype.inserter_pickup_position + entity.position
+					--local drop_pos = entity.prototype.inserter_drop_position + entity.position
+					--TODO: add further checks
+					supports_cargo = true
+				elseif entity.type == "pump" then
+					if entity.pump_rail_target then
+						supports_fluid = true
+					end
+				end
+			end
+
+			if supports_cargo then
+				if supports_fluid then
+					layout_pattern = layout_pattern..STATION_LAYOUT_BOTH
+				else
+					layout_pattern = layout_pattern..STATION_LAYOUT_CARGO
+				end
+			elseif supports_fluid then
+				layout_pattern = layout_pattern..STATION_LAYOUT_FLUID
+			else
+				layout_pattern = layout_pattern..STATION_LAYOUT_NA
+			end
+			if layout_min_size <= 0 then
+				layout_pattern = layout_pattern.."?"
+			else
+				layout_min_size = layout_min_size - 1
+			end
+			search_area = area.move(search_area, area_delta)
+		end
+	end
+	layout_pattern = layout_pattern..STATION_LAYOUT_NA.."*$"
+	station.layout_pattern = layout_pattern
+	local accepted_layouts = station.accepted_layouts
+	for id, layout in pairs(map_data.layouts) do
+		if string.find(layout, layout_pattern) ~= nil then
+			accepted_layouts[id] = true
+		else
+			accepted_layouts[id] = nil
+		end
+	end
+end