Added custom render queue exporter.

AkaiMage · Mar 18, 2019 · 0280f4a · 0280f4a
1 parent a1adcef
commit 0280f4a
Showing 1 changed file with 66 additions and 1 deletion.
diff --git a/Editor/CancerspaceInspector.cs b/Editor/CancerspaceInspector.cs
@@ -1,6 +1,8 @@
 using UnityEditor;
 using UnityEngine;
 using System;
+using System.IO;
+using System.Text.RegularExpressions;
 
 public class CancerspaceInspector : ShaderGUI {
 	public enum BlendMode {
@@ -25,6 +27,9 @@ public static class Styles {
 		public static string screenColorAdjustmentsTitle = "Screen Color Adjustment";
 		public static string screenTransformTitle = "Screen Transformations";
 		public static string miscSettingsTitle = "Misc";
+		public static string renderQueueExportTitle = "Custom Render Queue Exporter";
+		public static string customRenderQueueSliderText = "Custom Render Queue";
+		public static string exportCustomRenderQueueButtonText = "Export shader with queue and replace in this material";
 
 		public static readonly string[] blendNames = Enum.GetNames(typeof(BlendMode));
 	}
@@ -89,7 +94,9 @@ public CSCategory(string nname, GUIStyle sstyle, CSCategorySetup ssetupDelegate)
 	protected MaterialProperty screenRotationAngle;
 
 	protected MaterialProperty mirrorReflectionMode;
-	protected MaterialProperty customRenderQueue;
+
+	protected int customRenderQueue;
+	protected bool initialized;
 
 	public void FindProperties(MaterialProperty[] props) {
 		categoryExpansionFlags = FindProperty("_InspectorCategoryExpansionFlags", props);
@@ -145,6 +152,11 @@ public void FindProperties(MaterialProperty[] props) {
 	public override void OnGUI(MaterialEditor materialEditor, MaterialProperty[] properties) {
 		FindProperties(properties);
 
+		if (!initialized) {
+			customRenderQueue = (materialEditor.target as Material).shader.renderQueue;
+			initialized = true;
+		}
+
 		GUIStyle defaultStyle = new GUIStyle(EditorStyles.foldout);
 		defaultStyle.fontStyle = FontStyle.Bold;
 		defaultStyle.onNormal = EditorStyles.boldLabel.onNormal;
@@ -212,10 +224,59 @@ public override void OnGUI(MaterialEditor materialEditor, MaterialProperty[] pro
 				me.ShaderProperty(zTest, zTest.displayName);
 				me.ShaderProperty(mirrorReflectionMode, mirrorReflectionMode.displayName);
 			}),
+
+			new CSCategory(Styles.renderQueueExportTitle, defaultStyle, me => {
+				Material material = me.target as Material;
+
+				customRenderQueue = EditorGUILayout.IntSlider(Styles.customRenderQueueSliderText, customRenderQueue, 0, 5000);
+				if (GUILayout.Button(Styles.exportCustomRenderQueueButtonText)) {
+					int relativeQueue = customRenderQueue - ((int) UnityEngine.Rendering.RenderQueue.Transparent);
+					string newQueueString = "Transparent" + (relativeQueue >= 0 ? "+" : "") + relativeQueue;
+					string newShaderPath = "RedMage/Cancerspace Queue " + customRenderQueue;
+
+					string shaderPath = AssetDatabase.GetAssetPath(material.shader.GetInstanceID());
+					string outputLocation = EditorUtility.SaveFilePanelInProject("Save new shader file", "CancerspaceQueue" + customRenderQueue, "shader", "Please enter a file name to save the custom shader to.");
+
+					try {
+						using (StreamWriter sw = new StreamWriter(outputLocation)) {
+							using (StreamReader sr = new StreamReader(shaderPath)) {
+								string line;
+								while ((line = sr.ReadLine()) != null) {
+									if (line.Contains("\"Transparent+")) {
+										Regex rx = new Regex(@"Transparent[+-]\d+", RegexOptions.Compiled);
+										MatchCollection matches = rx.Matches(line);
+										foreach (Match match in matches) {
+											line = line.Replace(match.Value, newQueueString);
+										}
+									} else if (line.Contains("RedMage/Cancerspace")) {
+										Regex rx = new Regex("\"[^\"]+\"", RegexOptions.Compiled);
+										MatchCollection matches = rx.Matches(line);
+										foreach (Match match in matches) {
+											line = line.Replace(match.Value, "\"" + newShaderPath + "\"");
+										}
+									}
+									sw.Write(line);
+									sw.WriteLine();
+								}
+							}
+						}
+					} catch (Exception e) {
+						Debug.Log("AAAGAGHH WHAT? HOW? WHY??? WHAT ARE YOU DOING? Shader file could not be read / written.");
+						return;
+					}
+
+					AssetDatabase.Refresh();
+
+					material.shader = Shader.Find(newShaderPath);
+
+					AssetDatabase.SaveAssets();
+				}
+			}),
 		};
 
 		EditorGUIUtility.labelWidth = 0f;
 
+
 		int oldflags = BitConverter.ToInt32(BitConverter.GetBytes(categoryExpansionFlags.floatValue), 0);
 		int newflags = 0;
 		for (int i = 0; i < categories.Length; ++i) {
@@ -228,6 +289,10 @@ public override void OnGUI(MaterialEditor materialEditor, MaterialProperty[] pro
 			}
 		}
 		categoryExpansionFlags.floatValue = BitConverter.ToSingle(BitConverter.GetBytes(newflags), 0);
+
+
+		GUI.enabled = false;
+		materialEditor.RenderQueueField();
 	}
 
 	void BlendModePopup(MaterialEditor materialEditor) {