summaryrefslogtreecommitdiff
path: root/Assets/ThirdParty/VRM/VRMShaders/ShaderProperty/Editor/ShaderPropMenu.cs
blob: b13dec97eabfbb80ccefe47ced038d0915d7be4a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
using System.Reflection;
using System.Linq;
using UnityEditor;
using UnityEngine;
using System.IO;
using UniGLTF.ShaderPropExporter;
using System.Collections.Generic;

namespace UniGLTF
{
    public static class ShaderPropMenu
    {
#if VRM_DEVELOP
        [MenuItem("VRM/ShaderProperty/PreExport ShaderProps")]
#endif
        public static void PreExport()
        {
            foreach (var fi in typeof(PreExportShaders).GetFields(
                BindingFlags.Static
                | BindingFlags.Public
                | BindingFlags.NonPublic))
            {
                var attr = fi.GetCustomAttributes(true).FirstOrDefault(y => y is PreExportShadersAttribute);
                if (attr != null)
                {
                    var supportedShaders = fi.GetValue(null) as SupportedShader[];
                    foreach (var supported in supportedShaders)
                    {
                        PreExport(supported);
                    }
                }
            }
        }

        static string EscapeShaderName(string name)
        {
            return name.Replace("/", "_").Replace(" ", "_");
        }

        static string ExportDir
        {
            get
            {
                return Application.dataPath + "/VRM/ShaderProperty/Runtime";
            }
        }

        static void PreExport(SupportedShader supportedShader)
        {
            var shader = Shader.Find(supportedShader.ShaderName);
            var props = ShaderProps.FromShader(shader);

            var path = Path.Combine(ExportDir, supportedShader.TargetFolder);
            path = Path.Combine(path, EscapeShaderName(supportedShader.ShaderName) + ".cs").Replace("\\", "/");
            Debug.LogFormat("PreExport: {0}", path);
            File.WriteAllText(path, ToString(props, shader.name));
        }

        static string ToString(ShaderProps props, string shaderName)
        {
            var list = new List<string>();
            foreach (var prop in props.Properties)
            {
                list.Add(string.Format("new ShaderProperty(\"{0}\", ShaderPropertyType.{1})\r\n", prop.Key, prop.ShaderPropertyType));
            }

            return string.Format(@"using System.Collections.Generic;


namespace UniGLTF.ShaderPropExporter
{{
    public static partial class PreShaderPropExporter
    {{
        [PreExportShader]
        static KeyValuePair<string, ShaderProps> {0} 
        {{
            get 
            {{
                return new KeyValuePair<string, ShaderProps>(
                    ""{1}"",
                    new ShaderProps
                    {{
                        Properties = new ShaderProperty[]{{
{2}
                        }}
                    }}
                );
            }}
        }}
    }}
}}
"
, EscapeShaderName(shaderName)
, shaderName
, string.Join(",", list.ToArray()));
        }

    }
}