Parse the Factorio recipe files to create a CSV of recipes

factorio-recipe-parser.lua

local write = io.write
function print_me(write,...)
	local n = select("#",...)
	for i = 1,n do
		local v = tostring(select(i,...))
		write(v)
		if i~=n then write'\t' end
	end
	write'\n'
end
function print(...)
	print_me(write,...)
end
function error(...)
	local function write(...)
		io.stderr:write(...)
	end
	print_me(write,...)
end

-- remember about trailing / (or \\ on Windows™)
local basepath='/home/j000/.local/share/Steam/steamapps/common/Factorio/data/base/'

-- global variables
local constraints = {}
local images = {}
local skip = {}
local nodes = {}

function mark(array, what)
	if type(what) ~= 'table' then
		array[what] = true
		return
	end
	for _, value in ipairs(what) do
		mark(array, value)
	end
end

function check_node(name)
	if nodes[name] ~= false then
		nodes[name] = true
	end
end

function make_node(name, attributes)
	nodes[name] = false
	-- just name
	if (type(attributes) ~= 'table') then
		error(name .. ' does not have attributes')
		attributes = {}
	end
	-- open attributes
	write('\t"' .. name .. '" [')
	-- loop through attributes
	for k, v in pairs(attributes) do
		write(k .. '="' .. v .. '", ')
	end
	-- clear label
	if (not attributes['label'] and images[name] ~= '') then
		write('label=" ", ')
	end
	-- add images
	if not attributes['image'] then
		write('image="' .. (images[name] or (name .. '.png')) .. '", ')
		--write('labelloc="b", ')
	end
	-- close attributes
	write('];\n')
end

function make_edge(from, to, attributes)
	if (type(attributes) ~= 'table') then
		error(from .. ' -> ' .. to .. ' does not have attributes')
		attributes = {}
	end
	write('\t"' .. from .. '" -> "' .. to .. '" [')
	for k, v in pairs(attributes) do
		write(k .. '="' .. v .. '", ')
	end
	if constraints[from] and not constraints[to] then
		write('constraint=false, /*style=invis,*/ ')
	end
	write('];\n')
end

function get_recipes()
	local file_name = ''

	data = {}
	data["extend"] = function (data, t)
		print('// '..file_name)
		for n, recipe in ipairs(t) do
			local name = recipe["name"]
			if not skip[name] then
				local image = recipe["icon"]
				if image then
					image = basepath .. string.sub(image, 10)
				end
				-- we have array of results
				if recipe["results"] then
					-- just one result with same name as product (i.e. water, petroleum gas -> sulfur -> sulfur)
					if recipe["results"][2] == nil
						and ((recipe["results"][1][1] and recipe["results"][1][1] == name)
						or (recipe["results"][1]["name"] and recipe["results"][1]["name"] == name)) then
						make_node(name, {image = image})
					else
						make_node(name, {image = image, shape = 'box'})
						for i, result in ipairs(recipe["results"]) do
							local rname = result[1] or result["name"]
							local ramt = result[2] or result["amount"]
							check_node(rname)
							make_edge(name, rname, {label = ramt})
						end
					end
				else
					make_node(name, {image = image})
				end
				for i, component in ipairs(recipe["ingredients"]) do
					local cname = component[1] or component["name"]
					local camt = component[2] or component["amount"]
					check_node(cname)
					make_edge(cname, name, {label = camt})
				end
			end
		end
	end

	local files = {
		'ammo',
		'capsule',
		'demo-furnace-recipe',
		'demo-recipe',
		'demo-turret',
		'equipment',
		'fluid-recipe',
		'furnace-recipe',
		'inserter',
		'module',
		'recipe',
		'turret',
	}

	mark(constraints, {
		--'electronic-circuit',
		--'advanced-circuit',
		--'processing-unit',

		'empty-barrel',
		'crude-oil-barrel',
		--'empty-crude-oil-barrel',
		--'fill-crude-oil-barrel'
	})

	-- not usable in-game items as of 0.14.21
	mark(skip, {
		'small-plane',
		'loader',
		'fast-loader',
		'express-loader',
		'player-port',
		'electric-energy-interface'
	})

	-- images have different names
	images['railgun-dart'] = 'railgun-ammo.png'
	images['low-density-structure'] = 'rocket-structure.png'
	-- no image
	images['discharge-defense-remote'] = ''
	images['electric-energy-interface'] = ''

	for i, f in ipairs(files) do
		file_name = f
		dofile(basepath .. "prototypes/recipe/" .. f .. ".lua")
	end

	for k, v in pairs(nodes) do
		if v then
			--error('making node '..k)
			make_node(k, {})
		end
	end
end

print([[
digraph factorio {
	imagepath="]]..basepath..[[graphics/icons/fluid:]]..basepath..[[graphics/icons";
	graph [
		//compound=true,
		//ordering="out", // preserve order from file
	];
	//rankdir=LR; // draw from left to right, not top to bottom
	ranksep="0.1 equally" // space between "ranks"
	//size="8,5"; // image size
	//ratio="0.625"; // image ratio
	overlap=false; // don't overlap nodes
	//style=invis; // hide clusters' borders
	maxiter=100000; // number of iterations
]])
get_recipes()
print([[
	// some positioning
	//{ rank=min; "coal"; "iron-ore"; "copper-ore"; "stone"; "raw-wood"; "water"; "crude-oil" }
	//{ rank=same; "iron-plate"; "copper-plate"; "wood"; "landfill"; "stone-brick" }
	//{ rank=same; "lubricant"; "heavy-oil"; "heavy-oil-cracking"; "light-oil"; "light-oil-cracking"; "petroleum-gas" }
	//{ rank=same; "splitter"; "underground-belt"; "loader"; "fast-transport-belt" }
	//{ rank=same; "empty-barrel"; "crude-oil-barrel"; "empty-crude-oil-barrel"; "fill-crude-oil-barrel" }
	subgraph cluster_barrels {
		{ rank=same; "empty-barrel"; "crude-oil-barrel"; }
		"empty-crude-oil-barrel"; "fill-crude-oil-barrel"
	}
}
]])

recipes.csv