Module:GameData: Difference between revisions

Module:GameData (view source)

Revision as of 19:58, 26 May 2024

443 bytes added , 6 months ago

Break up recipe row functions, add comments, use recipe ID to determine component/building

Chorus

trusted

167

edits

@@ Line 17: / Line 17: @@
 }
 local OilSeepID = 7
+-- exported functions
 function funcs.recipesMaking(frame)
@@ Line 27: / Line 29: @@
 function recipesMakingName(name, frame)
+	local rtable = {
+		frame:expandTemplate{title='ProductionChainTable/head', args={}},
+	}
 	local item = itemByName(name)
-	local recipes = recipesMakingByID(item.ID)
+	for _, recipe in pairs(recipesMakingByID(item.ID)) do
+		table.insert(rtable, recipeRow(recipe, frame))
+	end
 	local vein = veinMakingByID(item.ID)
-	local gas = deduplicateGiants(gasGiantsMakingByID(item.ID))
+	if vein ~= nil then
-	local isocean = isItemOcean(item.ID)
+		table.insert(rtable, veinRow(item, vein, frame))
-	return tableRecipes(item, recipes, vein, gas, isocean, frame)
+	end
+	for _, giant in pairs(deduplicateGiants(gasGiantsMakingByID(item.ID))) do
+		table.insert(rtable, gasGiantRow(item, giant, frame))
+	end
+	if isItemOcean(item.ID) then
+		table.insert(rtable, oceanRow(item, ocean, frame))
+	end
+	table.insert(rtable, '|}')
+	return table.concat(rtable, '\n')
 end
@@ Line 45: / Line 60: @@
 function recipesUsingName(name, frame)
 	local output = {}
+	local header = frame:expandTemplate{title='ProductionChainTable/head', args={}}
 	local item = itemByName(name)
 	local components, buildings = recipesUsingByID(item.ID)
 	if next(components) then
 		table.insert(output, '=== Components ===')
-		table.insert(output, tableRecipes(item, components, nil, nil, false, frame))
+		table.insert(output, header)
+		for _, recipe in pairs(components) do
+			table.insert(output, recipeRow(recipe, frame))
+		end
+		table.insert(output, '|}')
 	end
 	if next(buildings) then
 		table.insert(output, '=== Buildings ===')
-		table.insert(output, tableRecipes(item, buildings, nil, nil, false, frame))
+		table.insert(output, header)
+		for _, recipe in pairs(buildings) do
+			table.insert(output, recipeRow(recipe, frame))
+		end
+		table.insert(output, '|}')
 	end
 	return table.concat(output, '\n')
@@ Line 93: / Line 117: @@
 end
-function tableRecipes(item, recipes, vein, gas, isocean, frame)
+-- make rows and recipe data for tables
-	local result = {}
-	for _, recipe in pairs(recipes) do
+function recipeRow(recipe, frame)
-		local data = {
+	local rdata = {
-			Building = titleCase(machines[recipe.Type]),
+		Building = titleCase(machines[recipe.Type]),
-			Recipe = itemRecipe(recipe, frame)
+	}
-		}
+	if recipe.Type == 'Fractionate' then
-		if recipe.Handcraft then
+		rdata.Recipe = fractionateRecipe(recipe, frame)
-			data.Replicator = 'Yes'
+	else
-		else
+		rdata.Recipe = itemRecipe(recipe, frame)
-			data.Replicator = 'No'
-		end
-		local tech = technologyForRecipeID(recipe.ID)
-		if tech ~= nil then
-			data.Technology = titleCase(tech.Name)
-		else
-			data.Technology = ''
-		end
-		table.insert(result, '| ' .. frame:expandTemplate{title='ProductionChain', args=data})
 	end
-	if vein and vein.ID == OilSeepID then -- I don't have a better test for oil right now
+	if recipe.Handcraft then
-		local extractor = itemByName(machines.Oil)
+		rdata.Replicator = 'Yes'
-		local extractorTech = technologyForRecipeID(recipesMakingByID(extractor.ID)[1].ID)
+	else
-		local oildata = {
+		rdata.Replicator = 'No'
-			Building = titleCase(extractor.Name),
-			Replicator = 'No',
-			Technology = titleCase(extractorTech.Name),
-			Recipe = oilRecipe(item, vein, frame),
-		}
-		table.insert(result, '| ' .. frame:expandTemplate{title='ProductionChain', args=oildata})
-	elseif vein then
-		local miner = itemByName(machines.Mine)
-		local minerTech = technologyForRecipeID(recipesMakingByID(miner.ID)[1].ID)
-		local veindata = {
-			Building = titleCase(miner.Name),
-			Replicator = 'Mine',
-			Technology = titleCase(minerTech.Name),
-			Recipe = veinRecipe(item, vein, frame),
-		}
-		table.insert(result, '| ' .. frame:expandTemplate{title='ProductionChain', args=veindata})
 	end
-	if gas then
+	local tech = technologyForRecipeID(recipe.ID)
-		local collector = itemByName(machines.Gas)
+	if tech ~= nil then
-		local collectorTech = technologyForRecipeID(recipesMakingByID(collector.ID)[1].ID)
+		rdata.Technology = titleCase(tech.Name)
-		for _, giant in pairs(gas) do
+	else
-			local gasdata = {
+		rdata.Technology = ''
-				Building = titleCase(collector.Name),
-				Replicator = 'Mine',
-				Technology = titleCase(collectorTech.Name),
-				Recipe = gasGiantRecipe(giant, frame)
-			}
-			table.insert(result, '| ' .. frame:expandTemplate{title='ProductionChain', args=gasdata})
-		end
 	end
-	if isocean then
+	return '|-\n| ' .. frame:expandTemplate{title='ProductionChain', args=rdata}
-		local pump = itemByName(machines.Ocean)
-		local pumpTech = technologyForRecipeID(recipesMakingByID(pump.ID)[1].ID)
-		local oceandata = {
-			Building = titleCase(pump.Name),
-			Replicator = 'No',
-			Technology = titleCase(pumpTech.Name),
-			Recipe = oceanRecipe(item, frame)
-		}
-		table.insert(result, '| ' .. frame:expandTemplate{title='ProductionChain', args=oceandata})
-	end
-	local prefix = frame:expandTemplate{title='ProductionChainTable/head', args={}}
-	return string.format('%s\n%s\n|}', prefix, table.concat(result, '\n|-\n'))
 end
 function itemRecipe(recipe, frame)
-	if recipe.Type == 'Fractionate' then
-		return fractionateRecipe(recipe, frame)
-	end
 	local rdata = {
 		CraftTime = string.format('%s s', tostring(recipe.TimeSpend/60)),
@@ Line 194: / Line 171: @@
 	end
 	return frame:expandTemplate{title='ItemRecipe', args=rdata}
+end
+function veinRow(item, vein, frame)
+	if vein.ID == OilSeepID then -- I don't have a better test for oil right now
+		local extractor = itemByName(machines.Oil)
+		local extractorTech = technologyForRecipeID(recipesMakingByID(extractor.ID)[1].ID)
+		local oildata = {
+			Building = titleCase(extractor.Name),
+			Replicator = 'No',
+			Technology = titleCase(extractorTech.Name),
+			Recipe = oilRecipe(item, vein, frame),
+		}
+		return '|-\n| ' .. frame:expandTemplate{title='ProductionChain', args=oildata}
+	else
+		local miner = itemByName(machines.Mine)
+		local minerTech = technologyForRecipeID(recipesMakingByID(miner.ID)[1].ID)
+		local veindata = {
+			Building = titleCase(miner.Name),
+			Replicator = 'Mine',
+			Technology = titleCase(minerTech.Name),
+			Recipe = veinRecipe(item, vein, frame),
+		}
+		return '|-\n| ' .. frame:expandTemplate{title='ProductionChain', args=veindata}
+	end
 end
@@ Line 217: / Line 218: @@
 	}
 	return frame:expandTemplate{title='ItemRecipe', args=odata}
+end
+function gasGiantRow(item, giant, frame)
+	local collector = itemByName(machines.Gas)
+	local collectorTech = technologyForRecipeID(recipesMakingByID(collector.ID)[1].ID)
+	local giantdata = {
+		Building = titleCase(collector.Name),
+		Replicator = 'Mine',
+		Technology = titleCase(collectorTech.Name),
+		Recipe = gasGiantRecipe(giant, frame)
+	}
+	return '|-\n| ' .. frame:expandTemplate{title='ProductionChain', args=giantdata}
 end
@@ Line 231: / Line 244: @@
 	end
 	return frame:expandTemplate{title='ItemRecipe', args=gdata}
+end
+function oceanRow(item, ocean, frame)
+	local pump = itemByName(machines.Ocean)
+	local pumpTech = technologyForRecipeID(recipesMakingByID(pump.ID)[1].ID)
+	local oceandata = {
+		Building = titleCase(pump.Name),
+		Replicator = 'No',
+		Technology = titleCase(pumpTech.Name),
+		Recipe = oceanRecipe(item, frame)
+	}
+	return '|-\n| ' .. frame:expandTemplate{title='ProductionChain', args=oceandata}
 end
@@ Line 243: / Line 268: @@
 	return frame:expandTemplate{title='ItemRecipe', args=odata}
 end
+-- get data from protosets
 function itemByName(name)
@@ Line 280: / Line 307: @@
 		for _, itemid in pairs(recipe.Items) do
 			if itemid == id then
-				if itemByID(recipe.Results[1]).CanBuild then
+				if itemByID(recipe.Results[1]).ID > 2000 then
 					table.insert(buildings, recipe)
 				else
@@ Line 333: / Line 360: @@
 	return false
 end
+-- deduplication
 function deduplicateGiants(giants)
@@ Line 348: / Line 377: @@
 	return dedup
 end
+-- display formatting
 function titleCase(str)