304 lines
12 KiB
C#
304 lines
12 KiB
C#
|
using System;
|
|||
|
|
using System.Collections.Generic;
|
|||
|
|
using System.IO;
|
|||
|
|
using System.Text;
|
|||
|
|
using System.Text.RegularExpressions;
|
|||
|
|
using Games.GlobeDefine;
|
|||
|
|
using UnityEditor;
|
|||
|
|
using UnityEngine;
|
|||
|
|
|
|||
|
|
public static class ShaderVariantTool
|
|||
|
|
{
|
|||
|
|
// public const string variantCollectionPath = "Assets/Project/Shader/ShaderVariants.shadervariants";
|
|||
|
|
// public const string shadowDepthKeyword = "SHADOWS_DEPTH";
|
|||
|
|
// public const string fogKeyword = "FOG_LINEAR";
|
|||
|
|
// public const string lightmapKeyword = "LIGHTMAP_ON";
|
|||
|
|
// public const string directionalKeyword = "DIRECTIONAL";
|
|||
|
|
//
|
|||
|
|
// private const string _fogCompileWord = "fog";
|
|||
|
|
// private const string _forwardBaseCompileWord = "fwdbase";
|
|||
|
|
|
|||
|
|
public static readonly string[] excludeKeywords =
|
|||
|
|
{
|
|||
|
|
"UNITY_HDR_ON",
|
|||
|
|
"DIRLIGHTMAP_COMBINED",
|
|||
|
|
"DIRLIGHTMAP_SEPARATE"
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
// public static void BuildVariantForShaders(IList<string> assetPaths)
|
|||
|
|
// {
|
|||
|
|
// var data = new List<ShaderVariantsSetting>();
|
|||
|
|
// for (var i = 0; i < assetPaths.Count; i++)
|
|||
|
|
// {
|
|||
|
|
// var path = assetPaths[i];
|
|||
|
|
// var shader = AssetDatabase.LoadAssetAtPath<Shader>(path);
|
|||
|
|
// var shaderSettings = CreateVariants(path, shader);
|
|||
|
|
// data.Add(shaderSettings);
|
|||
|
|
// }
|
|||
|
|
//
|
|||
|
|
// var collection = new ShaderVariantCollection();
|
|||
|
|
// for (var i = 0; i < data.Count; i++)
|
|||
|
|
// {
|
|||
|
|
// var varients = data[i].ParseToVarients();
|
|||
|
|
// for (var j = 0; j < varients.Length; j++)
|
|||
|
|
// collection.Add(varients[j]);
|
|||
|
|
// }
|
|||
|
|
//
|
|||
|
|
// AssetDatabase.CreateAsset(collection, variantCollectionPath);
|
|||
|
|
// AssetDatabase.SaveAssets();
|
|||
|
|
// AssetDatabase.Refresh();
|
|||
|
|
// Debug.LogWarning("Build Shader Variant Collection for " + assetPaths.Count);
|
|||
|
|
// }
|
|||
|
|
|
|||
|
|
// public static ShaderVariantsSetting CreateVariants(string assetPath, Shader shader)
|
|||
|
|
// {
|
|||
|
|
// var filePath = Application.dataPath.Open(assetPath.Substring("Assets".Length));
|
|||
|
|
// var keywordList = new List<string>();
|
|||
|
|
// var useForwardBase = false;
|
|||
|
|
// var useShadow = false;
|
|||
|
|
// var lines = File.ReadAllLines(filePath);
|
|||
|
|
// var regex = new Regex("(?<=#pragma multi_compile).+");
|
|||
|
|
// var shadowRegex = new Regex("\"LightMode\"\\s*=\\s*\"ShadowCaster\"");
|
|||
|
|
// foreach (var line in lines)
|
|||
|
|
// {
|
|||
|
|
// var match = regex.Match(line);
|
|||
|
|
// if (match.Success)
|
|||
|
|
// {
|
|||
|
|
// var result = match.Value.Trim('_', ' ');
|
|||
|
|
// // Unity内置Keyword
|
|||
|
|
// if (match.Value.StartsWith("_"))
|
|||
|
|
// {
|
|||
|
|
// switch (result)
|
|||
|
|
// {
|
|||
|
|
// case _fogCompileWord:
|
|||
|
|
// if (!keywordList.Contains(fogKeyword))
|
|||
|
|
// keywordList.Add(fogKeyword);
|
|||
|
|
// break;
|
|||
|
|
// case _forwardBaseCompileWord:
|
|||
|
|
// useForwardBase = true;
|
|||
|
|
// break;
|
|||
|
|
// }
|
|||
|
|
// }
|
|||
|
|
// else if (match.Value.Contains("__"))
|
|||
|
|
// {
|
|||
|
|
// if (!result.Contains(" ") && !excludeKeywords.Contain(result) && !keywordList.Contains(result))
|
|||
|
|
// keywordList.Add(result);
|
|||
|
|
// }
|
|||
|
|
// else
|
|||
|
|
// {
|
|||
|
|
// Debug.LogWarning("无法处理 " + line.TrimStart());
|
|||
|
|
// }
|
|||
|
|
// }
|
|||
|
|
// else
|
|||
|
|
// {
|
|||
|
|
// match = shadowRegex.Match(line);
|
|||
|
|
// if (match.Success)
|
|||
|
|
// useShadow = true;
|
|||
|
|
// }
|
|||
|
|
// }
|
|||
|
|
//
|
|||
|
|
// // 注：useForwardBase的物体，都是景物，仅仅需要烘培阴影
|
|||
|
|
// if (useForwardBase)
|
|||
|
|
// useShadow = false;
|
|||
|
|
// var fixedKeywords = useForwardBase ? new[] {lightmapKeyword, directionalKeyword} : new string[0];
|
|||
|
|
// var toggleKeywords = keywordList.ToArray();
|
|||
|
|
// return new ShaderVariantsSetting(shader.name, useForwardBase ? PassType.ForwardBase : PassType.Normal,
|
|||
|
|
// fixedKeywords, toggleKeywords, useShadow);
|
|||
|
|
// }
|
|||
|
|
|
|||
|
|
// public class ShaderVariantsSetting
|
|||
|
|
// {
|
|||
|
|
// public readonly string[] fixedKeywords;
|
|||
|
|
// public readonly PassType passType;
|
|||
|
|
// public readonly string shaderName;
|
|||
|
|
// public readonly bool shadow;
|
|||
|
|
// public readonly string[] toggleKeywords;
|
|||
|
|
//
|
|||
|
|
// public ShaderVariantsSetting(string name, PassType pass, string[] fixedMacro, string[] toggleMacro,
|
|||
|
|
// bool useShadow = false)
|
|||
|
|
// {
|
|||
|
|
// shaderName = name;
|
|||
|
|
// passType = pass;
|
|||
|
|
// fixedKeywords = fixedMacro;
|
|||
|
|
// toggleKeywords = toggleMacro;
|
|||
|
|
// shadow = useShadow;
|
|||
|
|
// }
|
|||
|
|
//
|
|||
|
|
// public ShaderVariantCollection.ShaderVariant[] ParseToVarients()
|
|||
|
|
// {
|
|||
|
|
// var shader = Shader.Find(shaderName);
|
|||
|
|
// if (shader == null)
|
|||
|
|
// throw new ArgumentException("无法获得Shader " + shaderName);
|
|||
|
|
// var end = 1 << toggleKeywords.Length;
|
|||
|
|
// var variants = new ShaderVariantCollection.ShaderVariant[end + (shadow ? 1 : 0)];
|
|||
|
|
// for (var i = 0; i < end; i++)
|
|||
|
|
// {
|
|||
|
|
// var index = 0;
|
|||
|
|
// var count = EditorCommonUtility.GetMaskCount(i);
|
|||
|
|
// var keywords = new string[count + fixedKeywords.Length];
|
|||
|
|
// for (var j = 0; j < toggleKeywords.Length; j++)
|
|||
|
|
// {
|
|||
|
|
// var num = j.ToFlag();
|
|||
|
|
// if (i.ContainFlag(num))
|
|||
|
|
// {
|
|||
|
|
// keywords[index] = toggleKeywords[j];
|
|||
|
|
// index++;
|
|||
|
|
// }
|
|||
|
|
// }
|
|||
|
|
//
|
|||
|
|
// for (var j = 0; j < fixedKeywords.Length; j++)
|
|||
|
|
// keywords[count + j] = fixedKeywords[j];
|
|||
|
|
// variants[i] = new ShaderVariantCollection.ShaderVariant
|
|||
|
|
// {
|
|||
|
|
// shader = shader,
|
|||
|
|
// passType = passType,
|
|||
|
|
// keywords = keywords
|
|||
|
|
// };
|
|||
|
|
// }
|
|||
|
|
//
|
|||
|
|
// if (shadow)
|
|||
|
|
// variants[end] = new ShaderVariantCollection.ShaderVariant
|
|||
|
|
// {
|
|||
|
|
// shader = shader,
|
|||
|
|
// passType = PassType.ShadowCaster,
|
|||
|
|
// keywords = new[] {shadowDepthKeyword}
|
|||
|
|
// };
|
|||
|
|
//
|
|||
|
|
// return variants;
|
|||
|
|
// }
|
|||
|
|
// }
|
|||
|
|
|
|||
|
|
#region DummyMaterial
|
|||
|
|
|
|||
|
|
public static void CreateAllDummyMaterials(List<string> assetPaths, string bundlePath)
|
|||
|
|
{
|
|||
|
|
var recordPath = GlobeVar.testConfigPath.Open("DummyMatRecord.txt");
|
|||
|
|
var create = true;
|
|||
|
|
assetPaths.Sort();
|
|||
|
|
var builder = new StringBuilder();
|
|||
|
|
foreach (var assetPath in assetPaths)
|
|||
|
|
builder.AppendLine(assetPath);
|
|||
|
|
var newHash = ClassifyBundles.GetStringHash(builder.ToString());
|
|||
|
|
if (File.Exists(recordPath))
|
|||
|
|
{
|
|||
|
|
var oldHash = File.ReadAllText(recordPath);
|
|||
|
|
if (oldHash.Equals(newHash, StringComparison.Ordinal))
|
|||
|
|
create = false;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (create)
|
|||
|
|
{
|
|||
|
|
var data = new List<DummyMaterialBuilder>();
|
|||
|
|
var dummyFilePath = EditorCommonUtility.AssetToFilePath(ClassifyBundles.dummyMaterialPath);
|
|||
|
|
if (Directory.Exists(dummyFilePath))
|
|||
|
|
Directory.Delete(dummyFilePath, true);
|
|||
|
|
Directory.CreateDirectory(dummyFilePath);
|
|||
|
|
AssetDatabase.Refresh();
|
|||
|
|
for (var i = 0; i < assetPaths.Count; i++)
|
|||
|
|
{
|
|||
|
|
var path = assetPaths[i];
|
|||
|
|
var shader = AssetDatabase.LoadAssetAtPath<Shader>(path);
|
|||
|
|
var shaderSettings = CreateDummyMaterials(path, shader);
|
|||
|
|
data.Add(shaderSettings);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for (var i = 0; i < data.Count; i++)
|
|||
|
|
{
|
|||
|
|
var variants = data[i];
|
|||
|
|
variants.SaveToVariants(bundlePath);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
File.WriteAllText(recordPath, newHash);
|
|||
|
|
|
|||
|
|
AssetDatabase.SaveAssets();
|
|||
|
|
AssetDatabase.Refresh();
|
|||
|
|
Debug.LogWarning("Build Shader Variant Collection for " + assetPaths.Count);
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
Debug.LogWarning("Skip Shader Variant as Hash matched!");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public static DummyMaterialBuilder CreateDummyMaterials(string assetPath, Shader shader)
|
|||
|
|
{
|
|||
|
|
var filePath = Application.dataPath.Open(assetPath.Substring("Assets".Length));
|
|||
|
|
var keywordList = new List<string>();
|
|||
|
|
var lines = File.ReadAllLines(filePath);
|
|||
|
|
var regex = new Regex("(?<=#pragma multi_compile).+");
|
|||
|
|
foreach (var line in lines)
|
|||
|
|
{
|
|||
|
|
var match = regex.Match(line);
|
|||
|
|
if (match.Success)
|
|||
|
|
{
|
|||
|
|
var result = match.Value.Trim('_', ' ');
|
|||
|
|
// Unity内置Keyword
|
|||
|
|
if (match.Value.StartsWith("_"))
|
|||
|
|
{
|
|||
|
|
}
|
|||
|
|
else if (match.Value.Contains("__"))
|
|||
|
|
{
|
|||
|
|
if (!result.Contains(" ") && !excludeKeywords.Contain(result) && !keywordList.Contains(result))
|
|||
|
|
keywordList.Add(result);
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
Debug.LogWarning("无法处理 " + line.TrimStart());
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 注：useForwardBase的物体，都是景物，仅仅需要烘培阴影
|
|||
|
|
return new DummyMaterialBuilder(shader, keywordList.ToArray());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public class DummyMaterialBuilder
|
|||
|
|
{
|
|||
|
|
public readonly Shader shader;
|
|||
|
|
public readonly string[] toggleKeywords;
|
|||
|
|
|
|||
|
|
public DummyMaterialBuilder(Shader name, string[] toggleMacro)
|
|||
|
|
{
|
|||
|
|
shader = name;
|
|||
|
|
toggleKeywords = toggleMacro;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void SaveToVariants(string bundleName)
|
|||
|
|
{
|
|||
|
|
if (!shader)
|
|||
|
|
throw new ArgumentException("无法获得Shader!");
|
|||
|
|
var end = 1 << toggleKeywords.Length;
|
|||
|
|
var variants = new Material[end];
|
|||
|
|
for (var i = 0; i < end; i++)
|
|||
|
|
{
|
|||
|
|
var index = 0;
|
|||
|
|
var count = EditorCommonUtility.GetMaskCount(i);
|
|||
|
|
var keywords = new string[count];
|
|||
|
|
for (var j = 0; j < toggleKeywords.Length; j++)
|
|||
|
|
{
|
|||
|
|
var num = j.ToFlag();
|
|||
|
|
if (i.ContainFlag(num))
|
|||
|
|
{
|
|||
|
|
keywords[index] = toggleKeywords[j];
|
|||
|
|
index++;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
variants[i] = new Material(shader);
|
|||
|
|
for (var j = 0; j < keywords.Length; j++)
|
|||
|
|
variants[i].EnableKeyword(keywords[j]);
|
|||
|
|
variants[i].name = shader.name + "_" + i;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for (var i = 0; i < variants.Length; i++)
|
|||
|
|
{
|
|||
|
|
var assetPath =
|
|||
|
|
ClassifyBundles.dummyMaterialPath.Open(ClassifyBundles.GetStringHash(variants[i].name) + AssetConst.materialExtension);
|
|||
|
|
AssetDatabase.CreateAsset(variants[i], assetPath);
|
|||
|
|
var importer = AssetImporter.GetAtPath(assetPath);
|
|||
|
|
importer.assetBundleName = bundleName;
|
|||
|
|
importer.assetBundleVariant = LoadAssetBundle.bundleFileExtension;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#endregion
|
|||
|
|
}
|