public static void RefreshStep(object collection, int index)
{
var materials = (List <Material>)collection;
Stepper.title = "Updating " + materials.Count + " Materials";
Stepper.message = "Updating material : " + materials[index].name;
VariableMaterial.Refresh(true, materials[index]);
}
public void Reload()
{
this.parent = VariableMaterial.GetParentShader(this.material);
if (!this.parent.IsNull())
{
this.hash = this.parent.GetModifiedDate("MdyyHmmff") + "-" + this.material.shaderKeywords.Join(" ").ToMD5();
}
VariableMaterial.dirty = false;
this.shader = this.material.shader;
this.editor.Repaint();
}
public static void Refresh(bool delay, params UnityObject[] targets)
{
VariableMaterial.delay = delay;
foreach (var target in targets)
{
var material = (Material)target;
bool isFlat = material.shader != null && material.shader.name.Contains("#");
if (isFlat || VariableMaterial.IsBroken(material))
{
VariableMaterial.Flatten(target);
}
}
}
override public void OnGUI(MaterialEditor editor, MaterialProperty[] properties)
{
EditorUI.Reset();
this.editor = editor;
this.material = (Material)editor.target;
bool matching = this.shader == this.material.shader;
if (!matching || VariableMaterial.dirty)
{
this.Reload();
}
if (this.shader != null)
{
EditorGUILayout.BeginHorizontal();
string[] keywords = this.material.shaderKeywords;
bool isHook = this.shader.name.EndsWith("#");
bool isFlat = this.shader.name.Contains("#") && !isHook;
bool isUpdated = !isFlat || this.shader.name.Split("#")[1].Split(".")[0] == this.hash;
GUI.enabled = !this.parent.IsNull() && (isHook || this.parent.extension != "zshader");
if (isFlat && "Unflatten".ToLabel().DrawButton())
{
VariableMaterial.Unflatten(editor.targets);
}
if (!isFlat && "Flatten".ToLabel().DrawButton())
{
VariableMaterial.Flatten(true, editor.targets);
}
GUI.enabled = Event.current.shift || !isUpdated;
if ("Update".ToLabel().DrawButton())
{
VariableMaterial.force = true;
var materials = editor.targets.Cast <Material>().ToList();
Events.AddStepper("On Editor Update", VariableMaterialEditor.RefreshStep, materials, 50);
}
GUI.enabled = true;
EditorGUILayout.EndHorizontal();
this.DrawProperties(editor, properties);
//this.DrawSimpleProperties(editor,properties);
//editor.PropertiesDefaultGUI(properties);
if (GUI.changed)
{
editor.serializedObject.ApplyModifiedProperties();
ProxyEditor.SetDirty(editor.serializedObject.targetObject, false, true);
}
if (isFlat && !keywords.SequenceEqual(this.material.shaderKeywords))
{
VariableMaterial.Refresh(editor.target);
}
}
}
public static void RefreshEditor()
{
if (VariableMaterial.updates.GetInvocationList().Length > 1)
{
ProxyEditor.StartAssetEditing();
VariableMaterial.writes();
ProxyEditor.StopAssetEditing();
//ProxyEditor.SaveAssets();
ProxyEditor.RefreshAssets();
File.Refresh();
VariableMaterial.updates();
}
VariableMaterial.updates = () => {};
VariableMaterial.writes = () => {};
VariableMaterial.dirty = true;
VariableMaterial.delay = false;
ProxyEditor.RebuildInspectors();
}
public static List <Material> GetAll()
{
var materials = File.GetAssets <Material>("*.mat");
var variableMaterials = new List <Material>();
foreach (var material in materials)
{
if (material.IsNull())
{
continue;
}
string editorName = material.shader.GetVariable <string>("customEditor");
if (editorName.Contains("VariableMaterialEditor") || VariableMaterial.IsBroken(material))
{
variableMaterials.Add(material);
}
}
return(variableMaterials);
}
public static void Unflatten(params UnityObject[] targets)
{
string shaderName = "";
foreach (var target in targets)
{
var material = (Material)target;
FileData shaderFile = VariableMaterial.GetParentShader(target);
if (shaderFile.IsNull())
{
continue;
}
shaderName = shaderFile.fullName;
material.shader = shaderFile.GetAsset <Shader>();
}
VariableMaterial.dirty = true;
if (VariableMaterial.debug)
{
Log.Show("[VariableMaterial] " + shaderName + " -- " + targets.Length + " Unflattened.");
}
ProxyEditor.RebuildInspectors();
}
public static void RefreshAll()
{
var materials = VariableMaterial.GetAll();
Events.AddStepper("On Editor Update", VariableMaterialEditor.RefreshStep, materials, 50);
}
public static void Refresh(params UnityObject[] targets)
{
VariableMaterial.Refresh(false, targets);
}
public static void Flatten(params UnityObject[] targets)
{
string originalName = "";
foreach (var target in targets)
{
Material material = (Material)target;
FileData shaderFile = VariableMaterial.GetParentShader(target);
if (shaderFile.IsNull())
{
continue;
}
string timestamp = shaderFile.GetModifiedDate("MdyyHmmff");
string projectPath = Application.dataPath + "/" + "Shaders";
string hash = timestamp + "-" + material.shaderKeywords.Join(" ").ToMD5();
string folderPath = projectPath + "/" + shaderFile.name + "/";
string outputPath = folderPath + shaderFile.name + "#" + hash + ".shader";
Action update = () => {
ProxyEditor.RecordObject(material, "Variable Material - Shader Set");
material.EnableKeyword("VARIABLE_MATERIAL_" + shaderFile.name.ToUpper());
material.shader = File.GetAsset <Shader>(outputPath);
ProxyEditor.SetAssetDirty(material);
if (VariableMaterial.debug)
{
Log.Show("[VariableMaterial] Shader set " + outputPath);
}
};
if (!VariableMaterial.force && File.Exists(outputPath))
{
VariableMaterial.updates += update;
continue;
}
originalName = shaderFile.fullName;
string text = shaderFile.ReadText();
string shaderName = text.Parse("Shader ", "{").Trim(' ', '"');
if (shaderName.Contains("#"))
{
continue;
}
string output = "Shader " + '"' + "Hidden/" + shaderName + "#" + hash + '"' + "{\r\n";
var allowed = new Stack <bool?>();
int tabs = -1;
text = text.Replace("\\\r\n", "");
int lineNumber = 0;
foreach (string current in text.Split("\r\n").Skip(1))
{
lineNumber += 1;
if (current.IsEmpty())
{
continue;
}
string line = current;
bool hideBlock = allowed.Count > 0 && allowed.Peek() != true;
bool allowedBranch = line.ContainsAny("#else", "#elif") && allowed.Peek() != null;
bool ignoredBranch = line.Contains("@#");
//if(line.ContainsAny("[KeywordEnum","[Toggle")){continue;}
if (!ignoredBranch && line.Contains("#endif"))
{
allowed.Pop();
if (allowed.Count == 0)
{
tabs = -1;
}
continue;
}
if (hideBlock && !allowedBranch)
{
if (!ignoredBranch && line.ContainsAny("#if"))
{
allowed.Push(null);
}
continue;
}
if (ignoredBranch)
{
bool end = line.Contains("#end");
bool include = line.Contains("#include");
if (tabs < 0)
{
tabs = line.Length - line.TrimStart().Length;
}
if (end)
{
tabs -= 1;
}
line = new String('\t', tabs) + line.TrimStart();
output += line.Replace("@#", "#") + "\r\n";
if (!end && !include)
{
tabs += 1;
}
continue;
}
if (line.Contains("#include"))
{
line = line.Replace("#include \"", "#include \"../");
}
if (line.Contains("#pragma shader_feature"))
{
continue;
}
if (line.Contains("#pragma multi_compile "))
{
continue;
}
if (line.ContainsAny("#if", "#elif", "#else"))
{
bool useBlock = false;
if (line.ContainsAny("#else", "#elif"))
{
bool lastAllowed = allowed.Pop() == true;
if (lastAllowed)
{
allowed.Push(null);
continue;
}
useBlock = line.Contains("#else");
}
if (line.ContainsAny("#ifdef", "#ifndef"))
{
bool hasTerm = material.shaderKeywords.Contains(line.Trim().Split(" ").Last());
useBlock = line.Contains("#ifndef") ? !hasTerm : hasTerm;
}
else if (line.Contains("defined"))
{
string[] orBlocks = line.Trim().Trim("#if ").Trim("#elif ").Split("||");
foreach (string orBlock in orBlocks)
{
string[] andBlocks = orBlock.Split("&&");
foreach (string andBlock in andBlocks)
{
string term = andBlock.Parse("defined(", ")");
bool hasTerm = material.shaderKeywords.Contains(term);
useBlock = andBlock.Contains("!") ? !hasTerm : hasTerm;
if (!useBlock)
{
break;
}
}
if (useBlock)
{
break;
}
}
}
allowed.Push(useBlock);
if (useBlock && allowed.Count == 1 && tabs <= 0)
{
tabs = line.Length - line.TrimStart().Length;
}
continue;
}
if (tabs >= 1)
{
if (line.Contains("}") && !line.Contains("{"))
{
tabs -= 1;
}
line = new String('\t', tabs) + line.TrimStart();
if (line.Contains("{") && !line.Contains("}"))
{
tabs += 1;
}
}
output += line + "\r\n";
}
output = output.Replace("{\r\n\t\t\t}", "{}");
string pattern = output.Cut("{\r\n\t\t\t\treturn ", ";\r\n\t\t\t");
while (!pattern.IsEmpty())
{
string replace = pattern.Replace("\r\n", "").Replace("\t", "");
output = output.ReplaceFirst(pattern, replace);
pattern = output.Cut("{\r\n\t\t\t\treturn ", ";\r\n\t\t\t");
}
if (output != text)
{
Action write = () => {
File.Create(outputPath).Write(output);
};
VariableMaterial.writes += write;
VariableMaterial.updates += update;
}
}
if (VariableMaterial.debug)
{
Log.Show("[VariableMaterial] " + originalName + " -- " + targets.Length + " flattened.");
}
if (VariableMaterial.delay)
{
Call.Delay(VariableMaterial.RefreshEditor, 0.5f);
return;
}
VariableMaterial.RefreshEditor();
VariableMaterial.force = false;
}
public static void Flatten(bool force, params UnityObject[] targets)
{
VariableMaterial.force = force;
VariableMaterial.Flatten(targets);
}
