UniVRM/Scripts/Format/Editor/VRMExporterMenu.cs

using System.Collections.Generic;
using System.Linq;
using UnityEditor;
using UnityEngine;
using UniGLTF;
using System.IO;
using System.Text;


namespace VRM
{
    public class VRMExporterWizard : ScriptableWizard
    {
        const string EXTENSION = ".vrm";

        public Animator Target;
        VRMMeta m_meta;

        public string Title;

        public string Author;

        public bool ForceTPose;

        public bool PoseFreeze;

        public static void CreateWizard()
        {
            var wiz = ScriptableWizard.DisplayWizard<VRMExporterWizard>(
                "VRM Exporter", "Export");
            var go= Selection.activeObject as GameObject;
            if (go != null)
            {
                wiz.Target = go.GetComponent<Animator>();
            }

            // update checkbox
            var desc = wiz.Target.GetComponent<VRMHumanoidDescription>();
            if (desc == null)
            {
                wiz.ForceTPose = true;
                wiz.PoseFreeze = true;
            }

            var meta = wiz.Target.GetComponent<VRMMeta>();
            if (meta != null && meta.Meta != null)
            {
                wiz.Title = meta.Meta.Title;
                wiz.Author = meta.Meta.Author;
            }
            else
            {
                wiz.Title = wiz.Target.name;
            }

            wiz.OnWizardUpdate();
        }

        string m_dir;
        string Dir
        {
            get
            {
                if (m_dir == null)
                {
                    m_dir = Path.GetFullPath(Application.dataPath); ;
                }
                return m_dir;
            }
            set
            {
                m_dir = value;
            }
        }

        void OnWizardCreate()
        {
            // save dialog
            Debug.LogFormat("OnWizardCreate: {0}", Dir);
            var path = EditorUtility.SaveFilePanel(
                    "Save vrm",
                    Dir,
                    Target.name + EXTENSION,
                    EXTENSION.Substring(1));
            if (string.IsNullOrEmpty(path))
            {
                return;
            }
            Dir = Path.GetDirectoryName(path);

            // export
            var target = Target.gameObject;
            if (PoseFreeze)
            {
                Undo.RecordObjects(Target.transform.Traverse().ToArray(), "before normalize");
                var map = new Dictionary<Transform, Transform>();
                target = VRM.BoneNormalizer.Execute(Target.gameObject, map, ForceTPose);
                VRMHumanoidNorimalizerMenu.CopyVRMComponents(Target.gameObject, target, map);
                Undo.PerformUndo();
            }

            var sw = System.Diagnostics.Stopwatch.StartNew();

            var vrm = VRMExporter.Export(target);
            vrm.extensions.VRM.meta.title = Title;
            vrm.extensions.VRM.meta.author = Author;

            var bytes = vrm.ToGlbBytes();
            File.WriteAllBytes(path, bytes);

            Debug.LogFormat("Export elapsed {0}", sw.Elapsed);

            if (Target.gameObject!=target)
            {
                GameObject.DestroyImmediate(target);
            }

            if (path.StartsWithUnityAssetPath())
            {
                AssetDatabase.ImportAsset(path.ToUnityRelativePath());
            }
        }

        void OnWizardUpdate()
        {
            isValid = true;
            var helpBuilder = new StringBuilder();
            var errorBuilder = new StringBuilder();

            if (Target == null)
            {
                isValid = false;
                errorBuilder.Append("Require animator. ");
            }
            else if (Target.avatar == null)
            {
                isValid = false;
                errorBuilder.Append("Require animator.avatar. ");
            }
            else if (!Target.avatar.isValid)
            {
                isValid = false;
                errorBuilder.Append("Animator.avatar is not valid. ");
            }
            else if (!Target.avatar.isHuman)
            {
                isValid = false;
                errorBuilder.Append("Animator.avatar is not humanoid. Please change model's AnimationType to humanoid. ");
            }


            if (string.IsNullOrEmpty(Title))
            {
                isValid = false;
                errorBuilder.Append("Require Title. ");
            }

            if (string.IsNullOrEmpty(Author))
            {
                isValid = false;
                errorBuilder.Append("Require Author. ");
            }

            helpString = helpBuilder.ToString();
            errorString = errorBuilder.ToString();
        }
    }

    public static class VRMExporterMenu
    {
        const string CONVERT_HUMANOID_KEY = "VRM/export humanoid";

        [MenuItem(CONVERT_HUMANOID_KEY, true, 1)]
        private static bool ExportValidate()
        {
            var root = Selection.activeObject as GameObject;
            if (root == null)
            {
                return false;
            }

            var animator = root.GetComponent<Animator>();
            if (animator == null)
            {
                return false;
            }

            return true;
        }

        [MenuItem(CONVERT_HUMANOID_KEY, false, 1)]
        private static void ExportFromMenu()
        {
            VRMExporterWizard.CreateWizard();
        }
    }
}