Модуль:TableOfChemicals

Материал из Space Station 14 Вики

Для документации этого модуля может быть создана страница Модуль:TableOfChemicals/doc

p = {}
p.chem = mw.text.jsonDecode(mw.title.new("User:KrasnoshchekovPavel/chem_prototypes.json"):getContent())
p.react = mw.text.jsonDecode(mw.title.new("User:KrasnoshchekovPavel/react_prototypes.json"):getContent())

p.fillTable = function(frame)
	
	local out = ""
	local group = frame.args.group
	local groups
	if group ~= nil and group ~= "" then
		groups = mw.text.split(group, ",")
	end
	
	local additional = frame.args.additional
	if additional ~= nil and additional ~= "" then
		
		local additionalChemPrototypeIds = mw.text.split(additional, ",")
		for _, chemPrototypeId in pairs(additionalChemPrototypeIds) do
			out = out .. fillChemistryRow(chemPrototypeId, frame) 
		end
	end
	
	for _, chemPrototype in pairs(p.chem) do
		
		if group == nil or group == "" or contains(groups, chemPrototype.group) then
			out = out .. fillChemistryRow(chemPrototype.id, frame) 
		end
	end
	
	return out
end

function contains(list, x)
	for _, v in ipairs(list) do
		if v == x then return true end
	end
	return false
end

p.fillChemistryRow = function(frame)
	
	return fillChemistryRow(frame.args.id, frame)
	
end

fillChemistryRow = function(chemPrototypeId, frame)
	
	local out = ""
	local chemPrototype = p.chem[chemPrototypeId]
	local templateArgs = {}
	templateArgs.id = chemPrototype.id
	templateArgs.name = chemPrototype.name
	templateArgs.description = chemPrototype.desc .. " На вид " .. chemPrototype.physicalDesc .. "."
	templateArgs.color = chemPrototype.color
	templateArgs.textColor = chemPrototype.textColor
	templateArgs.effects = getEffects(chemPrototype.id)
	templateArgs.recipes_count = tablelength(chemPrototype.recipes)
	
	templateArgs.reactants = ""
	templateArgs.products = ""
	templateArgs.action = ""
	
	if templateArgs.recipes_count == 0 then
		templateArgs.recipes_count = 1 -- Для заполнения параметра rowspan строки
		out = out .. frame:expandTemplate{ title = "Первая_строка_химического_вещества", args = templateArgs}
	else
		local firstReact = true
		for _, reactId in pairs(chemPrototype.recipes) do
			
			local reactPrototype = p.react[reactId]
			local reactants = {}
			local reactantTemplate = "%s [[#chem_%s|%s]]"
			
			for reactantId, reactantValue in pairs(reactPrototype.reactants) do
				
				local reactantChemData = p.chem[reactantId]
				local reactantText = string.format(reactantTemplate, reactantValue.amount, reactantChemData.id,  reactantChemData.name)
				table.insert(reactants, reactantText)
			end
			templateArgs.reactants = table.concat(reactants, "<br>")
			
			local products = {}
			local productTemplate = "%s [[#chem_%s|%s]]"
			
			for productId, productAmount in pairs(reactPrototype.products) do
				
				local productChemData = p.chem[productId]
				local productText = string.format(productTemplate, productAmount, productChemData.id,  productChemData.name)
				table.insert(products, productText)
			end
			templateArgs.products = table.concat(products, "<br>")
			
			templateArgs.actions = getActions(reactPrototype)
			
			local template = "Строка_химического_вещества"
			if firstReact then
				template = "Первая_строка_химического_вещества"
				firstReact = false
			end
			
			out = out .. frame:expandTemplate{ title = template, args = templateArgs}
		end
	end
	return out
end

getActions = function(reactPrototype)
	
	local actions = {}
	
	-- Температура
	local minMaxTemplate = "выше %sК и ниже %sК"
	local minTemplate = "выше %sК"
	local maxTemplate = "ниже %sК"
	
	local hasMin = reactPrototype.minTemp ~= 0
	local hasMax = reactPrototype.hasMax
	
	local tempString = ""
	if hasMax and hasMin then
		tempString = string.format(minMaxTemplate, reactPrototype.minTemp, reactPrototype.maxTemp)
	elseif hasMin then
		tempString = string.format(minTemplate, reactPrototype.minTemp)
	elseif hasMax then
		tempString = string.format(maxTemplate, reactPrototype.maxTemp)
	end
	
	for _, mixingCategory in pairs(reactPrototype.mixingCategories) do
		local image = getMixingImage(mixingCategory.id)
		if image ~= nil then
			table.insert(actions, string.format("[[File:%s|32px]]", image)) -- Картинка
		end
		
		table.insert(actions, mixingCategory.name .. " " .. tempString) -- Название
		
	end
	
	return table.concat(actions, "<br>")
	
end

getMixingImage = function(mixingId)
	
	local images = {
		["DummyMix"] = "Beaker"
	}
	
	return images[mixingId]
	
end

function getEffects(chemPrototypeId)
	
	local descriptions = {}
	
	local chemPrototype = p.chem[chemPrototypeId]
	if chemPrototype.metabolisms == nil then
		return ""
	end
	
	for _, metabolismGroup in pairs(chemPrototype.metabolisms) do
		for _, effect in pairs(metabolismGroup) do
			if effect.description ~= "" then
				table.insert(descriptions, effect.description)
			end
		end
	end
	return table.concat(descriptions, "<br>")
  
end

function tablelength(T)
  local count = 0
  for _ in pairs(T) do count = count + 1 end
  return count
end

function attachAsNewLine(originalStr, joinStr)
	return originalStr .. "\n" .. joinStr 
end

-- НЕ МОЕ
function p.geneffects(chemPrototypeId)
	local frame = mw.getCurrentFrame()
	local met = p.chem[chemPrototypeId].metabolisms
	if met == nil then
		return ""
	end
	local out = ""
	for k, v in pairs(met) do
		out = out .. ";<b>" .. k .. "</b> (" .. v.rate .. " единиц в секунду)\n" .. p.geneffectlist(v.effects, frame, v.rate)
	end
	return out
end

function p.geneffectlist(effects, frame, rate)
	local out = ""
	for l, w in pairs(effects) do
		-- Popup Message is ignored on purpose
		if w.id == "HealthChange" then
			out = out .. ":1" .. p.genhealthchange(w, rate, frame) .. "\n"
		elseif w.id == "AdjustReagent" then
			out = out .. ":2" .. p.genadjustreagent(w, rate, frame) .. "\n"
		elseif w.id == "FlammableReaction" then
			out = out .. ":3" .. p.genflammablereaction(w, frame) .. "\n"
		elseif w.id == "AdjustTemperature" then
			out = out .. ":4" .. p.genadjusttemperature(w, frame) .. "\n"
		elseif w.id == "GenericStatusEffect" then
			out = out .. ":5" .. p.gengenericstatuseffect(w, frame) .. "\n"
		elseif w.id == "ExplosionReactionEffect" then
			out = out .. ":6" .. p.genexplosionreactioneffect(w, frame) .. "\n"
		elseif w.id == "FoamAreaReactionEffect" then
			out = out .. ":7" .. p.genfoamareareactioneffect(w, frame) .. "\n"
		elseif w.id == "SmokeAreaReactionEffect" then
			out = out .. ":8" .. p.gensmokeareareactioneffect(w, frame) .. "\n"
		elseif w.id == "ModifyBleedAmount" then
			out = out .. ":9" .. p.genmodifybleedamount(w, frame) .. "\n"
		end
	end
	return out
end

function p.gensmokeareareactioneffect(r, frame)
	if r.conditions ~= nil then 
		conds = p.genconds(r.conditions, frame)
	end
	return frame:expandTemplate{ title = "SmokeAreaReactionEffect", args = { when = conds, prob = r.probability }}
end

function p.genfoamareareactioneffect(r, frame)
	if r.conditions ~= nil then 
		conds = p.genconds(r.conditions, frame)
	end
	return frame:expandTemplate{ title = "FoamAreaReactionEffect", args = { when = conds, prob = r.probability }}
end

function p.genexplosionreactioneffect(r, frame)
	if r.conditions ~= nil then 
		conds = p.genconds(r.conditions, frame)
	end
	return frame:expandTemplate{ title = "ExplosionReactionEffect", args = { when = conds, prob = r.probability }}
end

function p.gengenericstatuseffect(r, frame)
	if r.conditions ~= nil then 
		conds = p.genconds(r.conditions, frame)
	end
	return frame:expandTemplate{ title = "GenericStatusEffect", args = { key = r.Key, type = r.Type, time = r.Time, refresh = r.Refresh, when = conds, prob = r.probability }}
end

function p.genadjusttemperature(r, frame)
	if r.conditions ~= nil then 
		conds = p.genconds(r.conditions, frame)
	end
	return frame:expandTemplate{ title = "AdjustTemperature", args = { amount = r.Amount, when = conds, prob = r.probability }}
end

function p.genflammablereaction(r, frame)
	if r.conditions ~= nil then 
		conds = p.genconds(r.conditions, frame)
	end
	return frame:expandTemplate{ title = "FlammableReaction", args = { multiplier = r.Multiplier, when = conds, prob = r.probability }}
end

function p.genflammablereaction(r, frame)
	if r.conditions ~= nil then 
		conds = p.genconds(r.conditions, frame)
	end
	return frame:expandTemplate{ title = "FlammableReaction", args = { multiplier = r.Multiplier, when = conds, prob = r.probability }}
end

function p.genadjustreagent(r, rate, frame)
	if r.conditions ~= nil then 
		conds = p.genconds(r.conditions, frame)
	end
	return frame:expandTemplate{ title = "AdjustReagent", args = { amount = r.Amount, reagent = r.Reagent, when = conds, prob = r.probability }}
end

function p.genmodifybleedamount(r, frame)
	if r.conditions ~= nil then 
		conds = p.genconds(r.conditions, frame)
	end
	return frame:expandTemplate{ title = "ModifyBleedAmount", args = { amount = r.Amount, when = conds, prob = r.probability }}
end

function p.genhealthchange(h, rate, frame)
	local healst = {}
	local dealst = {}
	local r = 1.0 / rate
	if h.damage.types ~= nil then
	for k, v in pairs(h.damage.types) do
		if v < 0 then
			healst[k] = v * r
		else
			dealst[k] = v * r
		end
	end
	end
	
	if h.damage.groups ~= nil then
	for k, v in pairs(h.damage.groups) do
		if v < 0 then
			healst[k] = v * r
		else
			dealst[k] = v * r
		end
	end
	end
	
	local heals = hchangelist(healst, frame)
	local deals = hchangelist(dealst, frame)
	local conds = nil
	if h.conditions ~= nil then 
		conds = p.genconds(h.conditions, frame)
	end
	return frame:expandTemplate{ title = "HealthChange", args = { heals = heals, deals = deals, when = conds, prob = h.probability }}
end

function hchangelist(l, frame)
	out = ""
	local len = tablelength(l)
	local i = 0
	for k, v in pairs(l) do
		i = i + 1
		if len == i and i ~= 1 then
			out = out .. ", and "
		elseif i ~= 1 then
			out = out .. ", "
		end
		out = out .. hchange(k, v, frame)
	end
	return out
end

-- So we can make it fancy later
function hchange(ty, amnt, frame)
	return frame:expandTemplate{ title = "HealthModifier", args = { adj = amnt, kind = ty } }
end

function p.genconds(conds, frame)
	out = ""
	local len = tablelength(conds)
	local i = 0
	for k, v in pairs(conds) do
		i = i + 1
		if len == i and i ~= 1 then
			out = out .. ", and "
		elseif i ~= 1 then
			out = out .. ", "
		end
		out = out .. p.gencond(v, frame)
	end
	return out
end

function p.gencond(c, frame)
	if c.id == "TotalDamage" then
		return frame:expandTemplate{ title = "TotalDamage", args = { min = c.Min, max = c.Max } }
	elseif c.id == "ReagentThreshold" then
		return frame:expandTemplate{ title = "ReagentThreshold", args = { min = c.Min, max = c.Max, reagent = c.Reagent } }
	elseif c.id == "OrganType" then
		return frame:expandTemplate{ title = "OrganType", args = { shouldhave = c.ShouldHave, type = c.Type } }
	elseif c.id == "Temperature" then
		return frame:expandTemplate{ title = "Temperature", args = {  min = c.Min, max = c.Max } }
	end
	return ""
end

function getColor(chemPrototypeId)
	return mw.text.nowiki(p.chem[chemPrototypeId].color:sub(1, 7))
end

function getTextColor(chemPrototypeId)
	local basecol = p.chem[chemPrototypeId].color
	local red = tonumber(basecol:sub(2, 3), 16)
	local grn = tonumber(basecol:sub(4, 5), 16)
	local blu = tonumber(basecol:sub(6, 7), 16)
	local luminance = math.sqrt(0.241*red*red + 0.691*grn*grn + 0.068*blu*blu)
	if luminance > 100 then
		return mw.text.nowiki("#000")
	else
		return mw.text.nowiki("#FFF")
	end
end
-- НЕ МОЕ

return p