From fcb14494d7b275343904312cf31ad5d5f6b855b3 Mon Sep 17 00:00:00 2001
From: Magnus Norddahl <dpjudas@users.noreply.github.com>
Date: Sun, 15 Jul 2018 23:21:06 +0200
Subject: [PATCH] - improve the gldefs syntax by allowing the shader to be
 specified in the material section

---
 src/r_data/gldefs.cpp | 105 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 105 insertions(+)

diff --git a/src/r_data/gldefs.cpp b/src/r_data/gldefs.cpp
index 6035630309..04abc8a93a 100644
--- a/src/r_data/gldefs.cpp
+++ b/src/r_data/gldefs.cpp
@@ -1169,6 +1169,12 @@ class GLDefsParser
 		bool thiswad = false;
 		bool iwad = false;
 
+		UserShaderDesc usershader;
+		TArray<FString> texNameList;
+		TArray<FString> texNameIndex;
+		FString texnameDefs = "";
+		float speed = 1.f;
+
 		FTexture *textures[6];
 		const char *keywords[7] = { "brightmap", "normal", "specular", "metallic", "roughness", "ao", nullptr };
 		const char *notFound[6] = { "Brightmap", "Normalmap", "Specular texture", "Metallic texture", "Roughness texture", "Ambient occlusion texture" };
@@ -1218,6 +1224,62 @@ class GLDefsParser
 				if (tex)
 					tex->SpecularLevel = (float)sc.Float;
 			}
+			else if (sc.Compare("speed"))
+			{
+				sc.MustGetFloat();
+				speed = float(sc.Float);
+			}
+			else if (sc.Compare("shader"))
+			{
+				sc.MustGetString();
+				usershader.shader = sc.String;
+			}
+			else if (sc.Compare("texture"))
+			{
+				sc.MustGetString();
+				FString textureName = sc.String;
+				for (FString &texName : texNameList)
+				{
+					if (!texName.Compare(textureName))
+					{
+						sc.ScriptError("Trying to redefine custom hardware shader texture '%s' in texture '%s'\n", textureName.GetChars(), tex ? tex->Name.GetChars() : "(null)");
+					}
+				}
+				sc.MustGetString();
+				bool okay = false;
+				for (int i = 0; i < MAX_CUSTOM_HW_SHADER_TEXTURES; i++)
+				{
+					if (!tex->CustomShaderTextures[i])
+					{
+						tex->CustomShaderTextures[i] = TexMan.FindTexture(sc.String, ETextureType::Any, FTextureManager::TEXMAN_TryAny);
+						if (!tex->CustomShaderTextures[i])
+						{
+							sc.ScriptError("Custom hardware shader texture '%s' not found in texture '%s'\n", sc.String, tex ? tex->Name.GetChars() : "(null)");
+						}
+
+						texNameList.Push(textureName);
+						texNameIndex.Push(i);
+						okay = true;
+						break;
+					}
+				}
+				if (!okay)
+				{
+					sc.ScriptError("Error: out of texture units in texture '%s'", tex ? tex->Name.GetChars() : "(null)");
+				}
+			}
+			else if (sc.Compare("define"))
+			{
+				sc.MustGetString();
+				FString defineName = sc.String;
+				FString defineValue = "";
+				if (sc.CheckToken('='))
+				{
+					sc.MustGetString();
+					defineValue = sc.String;
+				}
+				texnameDefs.AppendFormat("#define %s %s\n", defineName.GetChars(), defineValue.GetChars());
+			}
 			else
 			{
 				for (int i = 0; keywords[i] != nullptr; i++)
@@ -1272,6 +1334,49 @@ class GLDefsParser
 
 		if (disable_fullbright_specified)
 			tex->bDisableFullbright = disable_fullbright;
+
+		if (usershader.shader.IsNotEmpty())
+		{
+			int firstUserTexture;
+			if (tex->Normal && tex->Specular)
+			{
+				usershader.shaderType = tex->Brightmap ? SHADER_SpecularBrightmap : SHADER_Specular;
+				firstUserTexture = tex->Brightmap ? 5 : 4;
+			}
+			else if (tex->Normal && tex->Metallic && tex->Roughness && tex->AmbientOcclusion)
+			{
+				usershader.shaderType = tex->Brightmap ? SHADER_PBRBrightmap : SHADER_PBR;
+				firstUserTexture = tex->Brightmap ? 7 : 6;
+			}
+			else
+			{
+				usershader.shaderType = tex->Brightmap ? SHADER_Brightmap : SHADER_Default;
+				firstUserTexture = tex->Brightmap ? 3 : 2;
+			}
+
+			for (unsigned int i = 0; i < texNameList.Size(); i++)
+			{
+				texnameDefs.AppendFormat("#define %s texture%d\n", texNameList[i].GetChars(), texNameIndex[i] + firstUserTexture);
+			}
+
+			if (tex->bWarped != 0)
+			{
+				Printf("Cannot combine warping with hardware shader on texture '%s'\n", tex->Name.GetChars());
+				return;
+			}
+			tex->shaderspeed = speed;
+			for (unsigned i = 0; i < usershaders.Size(); i++)
+			{
+				if (!usershaders[i].shader.CompareNoCase(usershader.shader) &&
+					usershaders[i].shaderType == usershader.shaderType &&
+					!usershaders[i].defines.Compare(texnameDefs))
+				{
+					tex->shaderindex = i + FIRST_USER_SHADER;
+					return;
+				}
+			}
+			tex->shaderindex = usershaders.Push(usershader) + FIRST_USER_SHADER;
+		}
 	}