Firstborn/Assets/AwesomeTechnologies/VegetationStudioPro/Runtime/BillboardSystem/BillboardShaderDetector.cs

using System.Collections;
using System.Collections.Generic;
using AwesomeTechnologies.VegetationSystem;
using UnityEngine;

namespace AwesomeTechnologies.Billboards
{
    public class BillboardShaderDetector
    {
        public static Shader GetDiffuceBillboardAtlasShader(GameObject prefab)
        {
            string shaderName = GetShaderName(prefab);
            if (shaderName.ToLower().Contains("speedtree")) return Shader.Find("AwesomeTechnologies/Billboards/RenderDiffuseAtlasSpeedtree");
            if (shaderName.ToLower().Contains("tree creator")) return Shader.Find("AwesomeTechnologies/Billboards/RenderDiffuseAtlasTreeCreator");
            if (shaderName.Contains("CTI")) return Shader.Find("AwesomeTechnologies/Billboards/RenderDiffuseAtlasCTI");

            switch (shaderName)
            {
                default:
                    return Shader.Find("AwesomeTechnologies/Billboards/RenderDiffuseAtlasNormal");
            }                    
        }

        public static Shader GetNormalBillboardAtlasShader(GameObject prefab)
        {           
            string shaderName = GetShaderName(prefab);
            if (shaderName.ToLower().Contains("tree creator")) return Shader.Find("AwesomeTechnologies/Billboards/RenderNormalsAtlasTreeCreator");
            if (shaderName.Contains("CTI")) return Shader.Find("AwesomeTechnologies/Billboards/RenderNormalsAtlasCTI");

            switch (shaderName)
            {
                default:
                    return Shader.Find("AwesomeTechnologies/Billboards/RenderNormalsAtlas");
            }
        }

        private static string GetShaderName(GameObject prefab)
        {
            GameObject selectedVegetationModel = MeshUtils.SelectMeshObject(prefab, LODLevel.LOD0);
            MeshRenderer meshrenderer = selectedVegetationModel.GetComponentInChildren<MeshRenderer>();
            if (!meshrenderer || !meshrenderer.sharedMaterial) return "";

            Shader shader = meshrenderer.sharedMaterial.shader;
            return shader.name;
        }
    }
}