Upload

lua/pac3/core/client/parts/material.lua

@@ -0,0 +1,609 @@
+--[[
+| This file was obtained through the combined efforts
+| of Madbluntz & Plymouth Antiquarian Society.
+|
+| Credits: lifestorm, Gregory Wayne Rossel JR.,
+| 	Maloy, DrPepper10 @ RIP, Atle!
+|
+| Visit for more: https://plymouth.thetwilightzone.ru/
+--]]
+
+local shader_params = include("pac3/libraries/shader_params.lua")
+
+local mat_hdr_level = GetConVar("mat_hdr_level")
+
+local material_flags = {
+	debug = bit.lshift(1, 0),
+	no_debug_override = bit.lshift(1, 1),
+	no_draw = bit.lshift(1, 2),
+	use_in_fillrate_mode = bit.lshift(1, 3),
+	vertexcolor = bit.lshift(1, 4),
+	vertexalpha = bit.lshift(1, 5),
+	selfillum = bit.lshift(1, 6),
+	additive = bit.lshift(1, 7),
+	alphatest = bit.lshift(1, 8),
+	multipass = bit.lshift(1, 9),
+	znearer = bit.lshift(1, 10),
+	model = bit.lshift(1, 11),
+	flat = bit.lshift(1, 12),
+	nocull = bit.lshift(1, 13),
+	nofog = bit.lshift(1, 14),
+	ignorez = bit.lshift(1, 15),
+	decal = bit.lshift(1, 16),
+	envmapsphere = bit.lshift(1, 17),
+	noalphamod = bit.lshift(1, 18),
+	envmapcameraspace = bit.lshift(1, 19),
+	basealphaenvmapmask = bit.lshift(1, 20),
+	translucent = bit.lshift(1, 21),
+	normalmapalphaenvmapmask = bit.lshift(1, 22),
+	needs_software_skinning = bit.lshift(1, 23),
+	opaquetexture = bit.lshift(1, 24),
+	envmapmode = bit.lshift(1, 25),
+	suppress_decals = bit.lshift(1, 26),
+	halflambert = bit.lshift(1, 27),
+	wireframe = bit.lshift(1, 28),
+	allowalphatocoverage = bit.lshift(1, 29),
+	ignore_alpha_modulation = bit.lshift(1, 30),
+}
+
+local function TableToFlags(flags, valid_flags)
+	if isstring(flags) then
+		flags = {flags}
+	end
+
+	local out = 0
+
+	for k, v in pairs(flags) do
+		if v then
+			local flag = valid_flags[v] or valid_flags[k]
+			if not flag then
+				error("invalid flag", 2)
+			end
+
+			out = bit.bor(out, tonumber(flag))
+		end
+	end
+
+	return out
+end
+
+local function FlagsToTable(flags, valid_flags)
+
+	if not flags then return valid_flags.default_valid_flag end
+
+	local out = {}
+
+	for k, v in pairs(valid_flags) do
+		if bit.band(flags, v) > 0 then
+			out[k] = true
+		end
+	end
+
+	return out
+end
+
+local shader_name_translate = {
+	vertexlitgeneric = "3d",
+	unlitgeneric = "2d",
+	eyerefract = "eye refract",
+}
+
+for shader_name, groups in pairs(shader_params.shaders) do
+	for group_name, base_group in pairs(shader_params.base) do
+		if groups[group_name] then
+			for k,v in pairs(base_group) do
+				if not groups[group_name][k] then
+					groups[group_name][k] = v
+				end
+			end
+		else
+			groups[group_name] = base_group
+		end
+	end
+end
+
+for shader_name, groups in pairs(shader_params.shaders) do
+	local temp = CreateMaterial(tostring({}), shader_name, {})
+
+	local BUILDER, PART = pac.PartTemplate("base")
+
+	PART.ClassName = "material_" .. (shader_name_translate[shader_name] or shader_name)
+	PART.Description = shader_name
+
+	PART.ProperColorRange = true
+
+	if shader_name == "vertexlitgeneric" then
+		PART.FriendlyName = "material"
+		PART.Group = {'modifiers', 'model', 'entity'}
+	else
+		PART.FriendlyName = "material " .. shader_name
+		PART.Group = "advanced"
+	end
+
+	PART.Icon = "icon16/paintcan.png"
+
+	BUILDER:StartStorableVars()
+
+	BUILDER:SetPropertyGroup("generic")
+
+	-- move this to tools or something
+	BUILDER:GetSet("LoadVmt", "", {editor_panel = "material"})
+	function PART:SetLoadVmt(path)
+		if not path or path == "" then return end
+
+		local str = file.Read("materials/" .. path .. ".vmt", "GAME")
+
+		if not str then return end
+
+		local vmt = util.KeyValuesToTable(str)
+		local shader = str:match("^(.-)%{"):gsub("%p", ""):Trim()
+
+
+		for k,v in pairs(self:GetVars()) do
+			local param = PART.ShaderParams[k]
+			if param and param.default ~= nil then
+				self["Set" .. k](self, param.default)
+			end
+			if param and param.type == "texture" then
+				self["Set" .. k](self, "")
+			end
+		end
+
+		print(str)
+		print("======")
+		PrintTable(vmt)
+		print("======")
+
+		for k,v in pairs(vmt) do
+			if k:StartWith("$") then k = k:sub(2) end
+
+			local func = self["Set" .. k]
+			if func then
+				local info = PART.ShaderParams[k]
+
+				if isstring(v) then
+					if v:find("[", nil, true) then
+						v = Vector(v:gsub("[%[%]]", ""):gsub("%s+", " "):Trim())
+
+						if isnumber(info.default) then
+							v = v.x
+						end
+					elseif v:find("{", nil, true) then
+						v = Vector(v:gsub("[%{%}]", ""):gsub("%s+", " "):Trim())
+
+						if info.type == "color" then
+							v = v / 255
+						end
+					end
+				end
+
+				if isnumber(v) then
+					if info.type == "bool" or info.is_flag then
+						v = v == 1
+					end
+				end
+
+				func(self, v)
+			else
+				pac.Message("cannot convert material parameter " .. k)
+			end
+		end
+	end
+
+	BUILDER:GetSet("MaterialOverride", "all", {enums = function(self, str)
+
+		local materials = {}
+
+		if pace.current_part:GetOwner():IsValid() then
+			materials = pace.current_part:GetOwner():GetMaterials()
+		end
+
+		table.insert(materials, "all")
+
+		local tbl = {}
+
+		for _, v in ipairs(materials) do
+			v = v:match(".+/(.+)") or v
+			tbl[v] = v:lower()
+		end
+
+		return tbl
+	end})
+
+	local function update_submaterial(self, remove, parent)
+		pac.RunNextFrameSimple(function()
+			if not IsValid(self) and not remove then return end
+			local name = self:GetName()
+
+			for _, part in ipairs(self:GetRootPart():GetChildrenList()) do
+				if part.GetMaterials then
+					for _, path in ipairs(part.Materials:Split(";")) do
+						if path == name then
+							part:SetMaterials(part.Materials)
+							break
+						end
+					end
+				end
+			end
+
+			local str = self.MaterialOverride
+			parent = parent or self:GetParent()
+
+			local num = 0
+
+			if parent:IsValid() then
+				if tonumber(str) then
+					num = tonumber(str)
+				elseif str ~= "all" and parent:GetOwner():IsValid() then
+					for i, v in ipairs(parent:GetOwner():GetMaterials()) do
+						if (v:match(".+/(.+)") or v):lower() == str:lower() then
+							num = i
+							break
+						end
+					end
+				end
+
+				parent.material_override = parent.material_override or {}
+				parent.material_override[num] = parent.material_override[num] or {}
+
+				for _, stack in pairs(parent.material_override) do
+					for i, v in ipairs(stack) do
+						if v == self then
+							table.remove(stack, i)
+							break
+						end
+					end
+				end
+
+				if not remove then
+					table.insert(parent.material_override[num], self)
+				end
+			end
+
+		end)
+	end
+
+	function PART:Initialize()
+		self.translation_vector = Vector()
+		self.rotation_angle = Angle(0, 0, 0)
+	end
+
+
+	function PART:GetNiceName()
+		local path = ""
+
+		if shader_name == "refract" then
+			path = self:Getnormalmap()
+		elseif shader_name == "eyerefract" then
+			path = self:Getiris()
+		else
+			path = self:Getbasetexture()
+		end
+
+		path = path:gsub("%%(..)", function(char)
+			local num = tonumber("0x" .. char)
+			if num then
+				return string.char(num)
+			end
+		end)
+
+		local name = ("/".. path):match(".+/(.-)%.") or ("/".. path):match(".+/(.+)")
+		local nice_name = (pac.PrettifyName(name) or "no texture") .. " | " .. shader_name
+
+		return nice_name
+	end
+
+	function PART:SetMaterialOverride(num)
+		self.MaterialOverride = num
+
+		update_submaterial(self)
+	end
+
+	function PART:OnThink()
+		if self:GetOwner():IsValid() then
+			local materials = self:GetOwner():GetMaterials()
+			if materials and #materials ~= self.last_material_count then
+				update_submaterial(self)
+				self.last_material_count = #materials
+			end
+		end
+	end
+
+	PART.ShaderParams = {}
+	PART.TransformVars = {}
+
+	local sorted_groups = {}
+	for k, v in pairs(groups) do
+		table.insert(sorted_groups, {k = k, v = v})
+	end
+	table.sort(sorted_groups, function(a, b) return a.k:lower() < b.k:lower() end)
+
+	for _, v in ipairs(sorted_groups) do
+		local group, params =  v.k, v.v
+
+		local sorted_params = {}
+		for k, v in pairs(params) do
+			table.insert(sorted_params, {k = k, v = v})
+		end
+		table.sort(sorted_params, function(a, b) return a.k:lower() < b.k:lower() end)
+
+		for _, v in ipairs(sorted_params) do
+			local key, info = v.k, v.v
+
+			PART.ShaderParams[key] = info
+
+			if info.is_flag and group == "generic" then
+				BUILDER:SetPropertyGroup("flags")
+			else
+				BUILDER:SetPropertyGroup(group)
+			end
+
+			if info.default == nil then
+				if info.type == "vec3" then
+					info.default = Vector(0,0,0)
+				elseif info.type == "color" then
+					info.default = Vector(1,1,1)
+				elseif info.type == "float" then
+					info.default = 0
+				elseif info.type == "vec2" then
+					info.default = Vector(0, 0)
+				end
+			end
+
+			local property_name = key
+
+			local description = (info.description or "") .. " ($" .. key .. ")"
+
+			if info.type == "matrix" then
+				local position_key = property_name .. "Position"
+				local scale_key = property_name .. "Scale"
+				local angle_key = property_name .. "Angle"
+				local angle_center_key = property_name .. "AngleCenter"
+
+				local friendly_name = info.friendly:gsub("Transform", "")
+				BUILDER:GetSet(position_key, Vector(0, 0, 0), {editor_friendly = friendly_name .. "Position", description = description})
+				BUILDER:GetSet(scale_key, Vector(1, 1, 1), {editor_friendly = friendly_name .. "Scale", description = description})
+				BUILDER:GetSet(angle_key, 0, {editor_panel = "number", editor_friendly = friendly_name .. "Angle", description = description})
+				BUILDER:GetSet(angle_center_key, Vector(0.5, 0.5, 0), {editor_friendly = friendly_name .. "AngleCenter", description = description})
+
+				PART.TransformVars[position_key] = true
+				PART.TransformVars[scale_key] = true
+				PART.TransformVars[angle_key] = true
+				PART.TransformVars[angle_center_key] = true
+
+				local shader_key = "$" .. key
+
+				local function setup_matrix(self)
+					self.matrix = self.matrix or Matrix()
+
+					self.matrix:Identity()
+					self.matrix:Translate(self.translation_vector)
+
+					self.matrix:Translate(self[angle_center_key])
+					self.matrix:Rotate(self.rotation_angle)
+					self.matrix:Translate(-self[angle_center_key])
+
+					self.matrix:SetScale(self[scale_key])
+				end
+
+				PART["Set" .. position_key] = function(self, vec)
+					self[position_key] = vec
+
+
+					self.translation_vector.x = self[position_key].x
+					self.translation_vector.y = self[position_key].y
+
+					setup_matrix(self)
+
+					self:GetRawMaterial():SetMatrix(shader_key, self.matrix)
+				end
+
+				PART["Set" .. scale_key] = function(self, vec)
+					self[scale_key] = vec
+					setup_matrix(self)
+
+					self:GetRawMaterial():SetMatrix(shader_key, self.matrix)
+				end
+
+				PART["Set" .. angle_key] = function(self, num)
+					self[angle_key] = num
+
+					self.rotation_angle.y = self[angle_key]*360
+
+					setup_matrix(self)
+
+					self:GetRawMaterial():SetMatrix(shader_key, self.matrix)
+				end
+
+				PART["Set" .. angle_center_key] = function(self, vec)
+					self[angle_center_key] = vec
+
+					setup_matrix(self)
+
+					self:GetRawMaterial():SetMatrix(shader_key, self.matrix)
+				end
+			elseif info.type == "texture" then
+				local getnohdr = "Get" .. property_name .. "NoHDR"
+
+				if info.partial_hdr then
+					BUILDER:GetSet(property_name .. "NoHDR", false, {
+						editor_friendly = info.friendly .. " No HDR",
+						description = "Disables bound param when HDR is enabled",
+					})
+				end
+
+				info.default = info.default or ""
+
+				BUILDER:GetSet(property_name, info.default, {
+					editor_panel = "textures",
+					editor_friendly = info.friendly,
+					description = description,
+					shader_param_info = info,
+				})
+
+				local key = "$" .. key
+
+				PART["Set" .. property_name .. "NoHDR"] = function(self, val)
+					self[property_name .. "NoHDR"] = val
+					PART["Set" .. property_name](self, self[property_name])
+				end
+
+				PART["Set" .. property_name] = function(self, val)
+					self[property_name] = val
+
+					if val == "" or info.partial_hdr and mat_hdr_level:GetInt() > 0 and self[getnohdr](self) then
+						self:GetRawMaterial():SetUndefined(key)
+						self:GetRawMaterial():Recompute()
+					else
+						if not pac.resource.DownloadTexture(val, function(tex, frames)
+							if frames then
+								self.vtf_frame_limit = self.vtf_frame_limit or {}
+								self.vtf_frame_limit[property_name] = frames
+							end
+							self:GetRawMaterial():SetTexture(key, tex)
+						end, self:GetPlayerOwner()) then
+							self:GetRawMaterial():SetTexture(key, val)
+
+							local texture = self:GetRawMaterial():GetTexture(key)
+
+							if texture then
+								self.vtf_frame_limit = self.vtf_frame_limit or {}
+								self.vtf_frame_limit[property_name] = texture:GetNumAnimationFrames()
+							end
+						end
+					end
+				end
+			else
+				BUILDER:GetSet(property_name, info.default, {
+					editor_friendly = info.friendly,
+					enums = info.enums,
+					description = description,
+					editor_sensitivity = (info.type == "vec3" or info.type == "color") and 0.25 or nil,
+					editor_panel = (info.type == "color" and "color2") or (property_name == "model" and "boolean") or nil,
+					editor_round = info.type == "integer",
+				})
+
+				local flag_key = key
+				local key = "$" .. key
+
+				if isnumber(info.default) then
+					PART["Set" .. property_name] = function(self, val)
+						self[property_name] = val
+						local mat = self:GetRawMaterial()
+						mat:SetFloat(key, val)
+						if info.recompute then
+							mat:Recompute()
+						end
+					end
+					if property_name:lower():find("frame") then
+						PART["Set" .. property_name] = function(self, val)
+							self[property_name] = val
+							if self.vtf_frame_limit and info.linked and self.vtf_frame_limit[info.linked] then
+								self:GetRawMaterial():SetInt(key, math.abs(val)%self.vtf_frame_limit[info.linked])
+							else
+								self:GetRawMaterial():SetInt(key, val)
+							end
+						end
+					end
+				elseif isbool(info.default) then
+					if info.is_flag then
+						PART["Set" .. property_name] = function(self, val)
+							self[property_name] = val
+
+							local mat = self:GetRawMaterial()
+
+							local tbl = FlagsToTable(mat:GetInt("$flags"), material_flags)
+							tbl[flag_key] = val
+							mat:SetInt("$flags", TableToFlags(tbl, material_flags))
+
+							mat:Recompute()
+						end
+					else
+						PART["Set" .. property_name] = function(self, val)
+							if isvector(val) then
+								val = (val == Vector(1,1,1)) and true or false
+							end
+
+							self[property_name] = val
+							local mat = self:GetRawMaterial()
+
+							mat:SetInt(key, val and 1 or 0)
+							if info.recompute then mat:Recompute() end
+						end
+					end
+				elseif isvector(info.default) or info.type == "vec3" or info.type == "vec2" then
+					PART["Set" .. property_name] = function(self, val)
+						if isstring(val) then val = Vector() end
+						self[property_name] = val
+						local mat = self:GetRawMaterial()
+						mat:SetVector(key, val)
+						if info.recompute then mat:Recompute() end
+					end
+				elseif info.type == "vec4" then
+					-- need vec4 type
+					PART["Set" .. property_name] = function(self, val)
+
+						local x,y,z,w
+						if isstring(val) then
+							x,y,z,w = unpack(val:Split(" "))
+							x = tonumber(x) or 0
+							y = tonumber(y) or 0
+							z = tonumber(z) or 0
+							w = tonumber(w) or 0
+						elseif isvector(val) then
+							x,y,z = val.x, val.y, val.z
+							w = 0
+						else
+							x, y, z, w = 0, 0, 0, 0
+						end
+
+						self[property_name] = ("%f %f %f %f"):format(x, y, z, w)
+						local mat = self:GetRawMaterial()
+						mat:SetString(key, ("[%f %f %f %f]"):format(x,y,z,w))
+
+						if info.recompute then mat:Recompute() end
+					end
+				end
+			end
+		end
+	end
+
+	BUILDER:EndStorableVars()
+
+	function PART:GetRawMaterial()
+		if not self.Materialm then
+			self.material_name = tostring({})
+			local mat = pac.CreateMaterial(self.material_name, shader_name, {})
+			self.Materialm = mat
+
+			for k,v in pairs(self:GetVars()) do
+				if PART.ShaderParams[k] and PART.ShaderParams[k].default ~= nil then
+					self["Set" .. k](self, v)
+				end
+			end
+		end
+		return self.Materialm
+	end
+
+	function PART:OnParent(parent)
+		update_submaterial(self)
+	end
+
+	function PART:OnRemove()
+		update_submaterial(self, true)
+	end
+
+	function PART:OnUnParent(parent)
+		update_submaterial(self, true, parent)
+	end
+
+	function PART:OnHide()
+		update_submaterial(self, true)
+	end
+
+	function PART:OnShow()
+		update_submaterial(self)
+	end
+
+	BUILDER:Register()
+end