UniVRM/Assets/VRM10/Runtime/Migration/MigrationVrm.cs

using System;
using System.Collections.Generic;
using UniGLTF;
using UniJSON;

namespace UniVRM10
{
    /// <summary>
    /// Convert vrm0 binary to vrm1 binary. Json processing
    /// </summary>
    public static class MigrationVrm
    {
        public static byte[] Migrate(byte[] src)
        {
            var data = new GlbBinaryParser(src, "migration").Parse();
            return Migrate(data);
        }

        static (int, int) GetVertexRange(SpanLike<int> indices)
        {
            var min = int.MaxValue; ;
            var max = 0;
            foreach (var i in indices)
            {
                if (i < min) min = i;
                if (i > max) max = i;
            }
            return (min, max);
        }

        public static byte[] Migrate(GltfData data)
        {
            // VRM0 -> Unity
            var model = ModelReader.Read(data, VrmLib.Coordinates.Vrm0);
            // Unity -> VRM1
            VrmLib.ModelExtensionsForCoordinates.ConvertCoordinate(model, VrmLib.Coordinates.Vrm1);

            var (gltf, bin) = new MeshUpdater(data).Update(model);
            gltf.extensions = null;
            return MigrateVrm(gltf, bin, data.Json.ParseAsJson()["extensions"]["VRM"]);
        }

        static byte[] MigrateVrm(glTF gltf, ArraySegment<byte> bin, JsonNode vrm0)
        {
            var meshToNode = CreateMeshToNode(gltf);

            {
                // vrm
                var vrm1 = new UniGLTF.Extensions.VRMC_vrm.VRMC_vrm();

                // meta (required)
                vrm1.Meta = MigrationVrmMeta.Migrate(gltf, vrm0["meta"]);
                // humanoid (required)
                vrm1.Humanoid = MigrationVrmHumanoid.Migrate(vrm0["humanoid"]);

                // blendshape (optional)
                if (vrm0.TryGet("blendShapeMaster", out JsonNode vrm0BlendShape))
                {
                    vrm1.Expressions = new UniGLTF.Extensions.VRMC_vrm.Expressions
                    {
                        Preset = new UniGLTF.Extensions.VRMC_vrm.Preset(),
                        Custom = new Dictionary<string, UniGLTF.Extensions.VRMC_vrm.Expression>(),
                    };
                    foreach (var (preset, customName, expression) in MigrationVrmExpression.Migrate(gltf, vrm0BlendShape, meshToNode))
                    {
                        switch (preset)
                        {
                            case ExpressionPreset.happy: vrm1.Expressions.Preset.Happy = expression; break;
                            case ExpressionPreset.angry: vrm1.Expressions.Preset.Angry = expression; break;
                            case ExpressionPreset.sad: vrm1.Expressions.Preset.Sad = expression; break;
                            case ExpressionPreset.relaxed: vrm1.Expressions.Preset.Relaxed = expression; break;
                            case ExpressionPreset.surprised: vrm1.Expressions.Preset.Surprised = expression; break;
                            case ExpressionPreset.aa: vrm1.Expressions.Preset.Aa = expression; break;
                            case ExpressionPreset.ih: vrm1.Expressions.Preset.Ih = expression; break;
                            case ExpressionPreset.ou: vrm1.Expressions.Preset.Ou = expression; break;
                            case ExpressionPreset.ee: vrm1.Expressions.Preset.Ee = expression; break;
                            case ExpressionPreset.oh: vrm1.Expressions.Preset.Oh = expression; break;
                            case ExpressionPreset.blink: vrm1.Expressions.Preset.Blink = expression; break;
                            case ExpressionPreset.blinkLeft: vrm1.Expressions.Preset.BlinkLeft = expression; break;
                            case ExpressionPreset.blinkRight: vrm1.Expressions.Preset.BlinkRight = expression; break;
                            case ExpressionPreset.lookUp: vrm1.Expressions.Preset.LookUp = expression; break;
                            case ExpressionPreset.lookDown: vrm1.Expressions.Preset.LookDown = expression; break;
                            case ExpressionPreset.lookLeft: vrm1.Expressions.Preset.LookLeft = expression; break;
                            case ExpressionPreset.lookRight: vrm1.Expressions.Preset.LookRight = expression; break;
                            case ExpressionPreset.neutral: vrm1.Expressions.Custom[customName] = expression; break;
                            case ExpressionPreset.custom: vrm1.Expressions.Custom[customName] = expression; break;
                            default: throw new NotImplementedException();
                        }
                    }
                }

                // lookat & firstperson (optional)
                if (vrm0.TryGet("firstPerson", out JsonNode vrm0FirstPerson))
                {
                    (vrm1.LookAt, vrm1.FirstPerson) = MigrationVrmLookAtAndFirstPerson.Migrate(gltf, vrm0FirstPerson);
                }

                UniGLTF.Extensions.VRMC_vrm.GltfSerializer.SerializeTo(ref gltf.extensions, vrm1);
            }

            // springBone & collider (optional)
            if (vrm0.TryGet("secondaryAnimation", out JsonNode vrm0SpringBone))
            {
                var springBone = MigrationVrmSpringBone.Migrate(gltf, vrm0SpringBone);
                UniGLTF.Extensions.VRMC_springBone.GltfSerializer.SerializeTo(ref gltf.extensions, springBone);
            }

            // Material
            {
                MigrationMaterials.Migrate(gltf, vrm0);
            }

            // Serialize whole glTF
            ArraySegment<byte> vrm1Json = default;
            {
                var f = new JsonFormatter();
                GltfSerializer.Serialize(f, gltf);
                vrm1Json = f.GetStoreBytes();
            }

            return Glb.Create(vrm1Json, bin).ToBytes();
        }

        public delegate int MeshIndexToNodeIndexFunc(int meshIndex);

        public static MeshIndexToNodeIndexFunc CreateMeshToNode(UniGLTF.glTF gltf)
        {
            return (int mesh) =>
            {
                for (int i = 0; i < gltf.nodes.Count; ++i)
                {
                    var node = gltf.nodes[i];
                    if (node.mesh == mesh)
                    {
                        return i;
                    }
                }
                return -1;
            };
        }

        public static void Check(JsonNode vrm0, UniGLTF.Extensions.VRMC_vrm.VRMC_vrm vrm1, MeshIndexToNodeIndexFunc meshToNode)
        {
            MigrationVrmMeta.Check(vrm0["meta"], vrm1.Meta);
            MigrationVrmHumanoid.Check(vrm0["humanoid"], vrm1.Humanoid);
            MigrationVrmExpression.Check(vrm0["blendShapeMaster"], vrm1.Expressions, meshToNode);
        }

        public static void Check(JsonNode vrm0, UniGLTF.Extensions.VRMC_springBone.VRMC_springBone vrm1, List<glTFNode> nodes)
        {
            // Migration.CheckSpringBone(vrm0["secondaryAnimation"], vrm1.sp)
        }
    }

}