Module:GameData: Difference between revisions

Module:GameData (view source)

Revision as of 17:43, 27 April 2024

2,212 bytes added , 7 months ago

Add vein and gas giant support, and use local variables

Chorus

trusted

167

edits

@@ Line 11: / Line 11: @@
 	Fractionate='Fractionator',
 	Research='Matrix Lab',
+	Mine='Mining Machine',
+	Gas='Orbital Collector',
 }
 function funcs.recipesMaking(frame)
-	name = frame:getParent().args[1]
+	return recipesMakingName(frame:getParent().args[1], frame)
-	item = itemByName(name)
-	recipes = recipesMakingByID(item.ID)
-	return tableRecipes(recipes, frame)
 end
 function funcs.recipesMakingDirect(frame)
-	name = frame.args[1]
+	return recipesMakingName(frame.args[1], frame)
-	item = itemByName(name)
+end
-	recipes = recipesMakingByID(item.ID)
-	return tableRecipes(recipes, frame)
+function recipesMakingName(name, frame)
+	local item = itemByName(name)
+	local recipes = recipesMakingByID(item.ID)
+	local vein = veinMakingByID(item.ID)
+	local gas = deduplicateGiants(gasGiantsMakingByID(item.ID))
+	return tableRecipes(item, recipes, vein, gas, frame)
 end
 function funcs.recipesUsing(frame)
-	output = {}
+	return recipesUsingName(frame:getParent().args[1], frame)
-	name = frame:getParent().args[1]
-	item = itemByName(name)
-	components, buildings = recipesUsingByID(item.ID)
-	if next(components) then
-		table.insert(output, '=== Components ===')
-		table.insert(output, tableRecipes(components, frame))
-	end
-	if next(buildings) then
-		table.insert(output, '=== Buildings ===')
-		table.insert(output, tableRecipes(buildings, frame))
-	end
-	return table.concat(output, '\n')
 end
 function funcs.recipesUsingDirect(frame)
-	output = {}
+	return recipesUsingName(frame.args[1], frame)
-	name = frame.args[1]
+end
-	item = itemByName(name)
-	components, buildings = recipesUsingByID(item.ID)
+function recipesUsingName(name, frame)
+	local output = {}
+	local item = itemByName(name)
+	local components, buildings = recipesUsingByID(item.ID)
 	if next(components) then
 		table.insert(output, '=== Components ===')
-		table.insert(output, tableRecipes(components, frame))
+		table.insert(output, tableRecipes(item, components, nil, nil, frame))
 	end
 	if next(buildings) then
 		table.insert(output, '=== Buildings ===')
-		table.insert(output, tableRecipes(buildings, frame))
+		table.insert(output, tableRecipes(item, buildings, nil, nil, frame))
 	end
 	return table.concat(output, '\n')
 end
-function tableRecipes(recipes, frame)
+function tableRecipes(item, recipes, vein, gas, frame)
-	result = {}
+	local result = {}
 	for _, recipe in pairs(recipes) do
-		itemdata = {}
+		local itemdata = {}
 		itemdata.CraftTime = string.format('%s s', tostring(recipe.TimeSpend/60))
 		for i, itemid in ipairs(recipe.Results) do
@@ Line 72: / Line 67: @@
 			itemdata[string.format('In%dQty', i)] = tostring(recipe.ItemCounts[i])
 		end
-		data = {
+		local data = {
-			Building = machines[recipe.Type],
+			Building = titleCase(machines[recipe.Type]),
 			Recipe = frame:expandTemplate{title='ItemRecipe', args=itemdata},
 		}
@@ Line 81: / Line 76: @@
 			data.Replicator = 'No'
 		end
-		tech = technologyForRecipeID(recipe.ID)
+		local tech = technologyForRecipeID(recipe.ID)
 		if tech ~= nil then
 			data.Technology = titleCase(tech.Name)
@@ Line 89: / Line 84: @@
 		table.insert(result, '| ' .. frame:expandTemplate{title='ProductionChain', args=data})
 	end
-	prefix = frame:expandTemplate{title='ProductionChainTable/head', args={}}
+	if vein then
+		local miner = itemByName(machines.Mine)
+		local minerTech = technologyForRecipeID(recipesMakingByID(miner.ID)[1].ID)
+		local vname = string.gsub(vein.Name, '%sVeins$', ' Vein')
+		local idata = {
+			In1 = titleCase(vname),
+			In1Qty = '1',
+			Out1 = titleCase(item.Name),
+			Out1Qty = '1',
+			CraftTime = '2 s',
+		}
+		local vrecipe = frame:expandTemplate{title='ItemRecipe', args=idata}
+		local veindata = {
+			Building = titleCase(miner.Name),
+			Replicator = 'Mine',
+			Technology = titleCase(minerTech.Name),
+			Recipe = vrecipe,
+		}
+		table.insert(result, '| ' .. frame:expandTemplate{title='ProductionChain', args=veindata})
+	end
+	if gas then
+		local collector = itemByName(machines.Gas)
+		local collectorTech = technologyForRecipeID(recipesMakingByID(collector.ID)[1].ID)
+		for _, giant in pairs(gas) do
+			itemrecipe = {
+				CraftTime = '?',
+				In1 = titleCase(giant.DisplayName),
+				In1Qty = '',
+			}
+			for i, productID in ipairs(giant.GasItems) do
+				product = itemByID(productID)
+				itemrecipe[string.format('Out%d', i)] = titleCase(product.Name)
+				itemrecipe[string.format('Out%dQty', i)] = '1'
+			end
+			gasdata = {
+				Building = titleCase(collector.Name),
+				Replicator = 'Mine',
+				Technology = titleCase(collectorTech.Name),
+				Recipe = frame:expandTemplate{title='ItemRecipe', args=itemrecipe}
+			}
+			table.insert(result, '| ' .. frame:expandTemplate{title='ProductionChain', args=gasdata})
+		end
+	end
+	local prefix = frame:expandTemplate{title='ProductionChainTable/head', args={}}
 	return string.format('%s\n%s\n|}', prefix, table.concat(result, '\n|-\n'))
 end
 function itemByName(name)
-	lame = string.lower(name)
+	local lame = string.lower(name)
 	for _, item in pairs(protosets.ItemProtoSet.dataArray) do
 		if string.lower(item.Name) == lame then
@@ Line 112: / Line 150: @@
 function recipesMakingByID(id)
-	result = {}
+	local result = {}
 	for _, recipe in pairs(protosets.RecipeProtoSet.dataArray) do
 		for _, itemid in pairs(recipe.Results) do
@@ Line 124: / Line 162: @@
 end
 function recipesUsingByID(id)
-	components = {}
+	local components = {}
-	buildings = {}
+	local buildings = {}
 	for _, recipe in pairs(protosets.RecipeProtoSet.dataArray) do
 		for _, itemid in pairs(recipe.Items) do
@@ Line 150: / Line 188: @@
 	end
 	return nil
+end
+function veinMakingByID(id)
+	for _, vein in pairs(protosets.VeinProtoSet.dataArray) do
+		if vein.MiningItem == id then
+			return vein
+		end
+	end
+	return nil
+end
+function gasGiantsMakingByID(id)
+	local giants = {}
+	for _, planet in pairs(protosets.ThemeProtoSet.dataArray) do
+		for _, item in pairs(planet.GasItems) do
+			if item == id then
+				table.insert(giants, planet)
+				break
+			end
+		end
+	end
+	return giants
+end
+function deduplicateGiants(giants)
+	local dedup = {}
+	for _, planet in pairs(giants) do
+		new = true
+		for _, exist in pairs(dedup) do
+			if string.lower(planet.DisplayName) == string.lower(exist.DisplayName) then
+				new = false
+				break
+			end
+		end
+		if new then table.insert(dedup, planet) end
+	end
+	return dedup
 end
 function titleCase(str)
-	start, _ = string.gsub(str, '^%l', string.upper)
+	local start = string.gsub(str, '^%l', string.upper)
-	title, _ = string.gsub(start, '[-%s]%l', string.upper)
+	local title = string.gsub(start, '[-%s]%l', string.upper)
 	return title
 end
 return funcs