using System.Linq;

using System.Reflection;

using Soco.ShaderVariantsCollection;

using UnityEditor;

using UnityEditor.Rendering;

using UnityEngine.Rendering;

// 用于联动变体收集 和 变体剔除工具

// 如果没有使用Soco变体剔除工具，此执行器应删除

public class SocoVariantStripAssociate : Soco.ShaderVariantsCollection.IExecutable

{

public override void Execute(ShaderVariantCollectionMapper mapper)

{

var m_ShaderType = typeof(ShaderSnippetData).GetField("m_ShaderType", BindingFlags.Instance | BindingFlags.NonPublic);

var m_PassType = typeof(ShaderSnippetData).GetField("m_PassType", BindingFlags.Instance | BindingFlags.NonPublic);

var m_PassName = typeof(ShaderSnippetData).GetField("m_PassName", BindingFlags.Instance | BindingFlags.NonPublic);

var stripConfigs =

com.cyou.soco.ShaderVariantsStripper.ShaderVariantsStripperCode.LoadConfigs();

foreach (var shader in mapper.shaders)

{

int variantIndex = 0;

var variants = mapper.GetShaderVariants(shader).ToArray();

foreach (var variant in variants)

{

EditorUtility.DisplayProgressBar("变体剔除", $"正在处理{shader.name} {variantIndex}/{variants.Length}变体",

(float)variantIndex / (float)variants.Length);

variantIndex++;

ShaderSnippetData snipperData = new ShaderSnippetData();

//这里没对RayTracing、Mesh Shader的剔除做相关处理，默认所有Shader Pass都有VertexShader

m_ShaderType.SetValueDirect(__makeref(snipperData), ShaderType.Vertex);

m_PassType.SetValueDirect(__makeref(snipperData), variant.passType);

m_PassName.SetValueDirect(__makeref(snipperData), "");

ShaderCompilerData data = new ShaderCompilerData();

data.shaderKeywordSet = new ShaderKeywordSet();

foreach (var keyword in variant.keywords)

{

data.shaderKeywordSet.Enable(new ShaderKeyword(keyword));

}

if (com.cyou.soco.ShaderVariantsStripper.ShaderVariantsStripperCode.IsVariantStrip(

shader, snipperData, data, stripConfigs))

{

mapper.RemoveVariant(variant);

}

}

}

EditorUtility.ClearProgressBar();

}

}