feat: implement NDMF integration

CHANGELOG-PRERELEASE.md

@@ -9,6 +9,7 @@ The format is based on [Keep a Changelog].
 ## [Unreleased]
 ### Added
 - Support for Multi Frame BlendShapes `#333`
+- Support for [NDMF](https://ndmf.nadena.dev) integration `#375`
 
 ### Changed
 - Auto FreezeBlendShape now freezes meaningless BlendShape `#334`

Editor/OptimizerProcessor.cs

@@ -1,10 +1,10 @@
-
+#if !NADEMOFU
 using System;
-using Anatawa12.ApplyOnPlay;
 using Anatawa12.AvatarOptimizer.ErrorReporting;
 using UnityEngine;
 using VRC.SDK3.Avatars.Components;
 using VRC.SDKBase.Editor.BuildPipeline;
+using Anatawa12.ApplyOnPlay;
 
 namespace Anatawa12.AvatarOptimizer
 {
@@ -150,3 +150,4 @@ private static void DoProcessObject(OptimizerSession session)
         }
     }
 }
+#endif

Editor/OptimizerSession.cs

@@ -1,6 +1,8 @@
 using System;
 using System.Collections.Generic;
 using System.Linq;
+using Anatawa12.AvatarOptimizer.ndmf;
+using nadena.dev.ndmf;
 using UnityEditor;
 using UnityEngine;
 using Object = UnityEngine.Object;
@@ -15,6 +17,13 @@ internal class OptimizerSession
         public bool IsTest { get; }
         public ObjectMappingBuilder MappingBuilder { get; }
 
+#if NADEMOFU
+
+        public static implicit operator OptimizerSession(BuildContext context) 
+            => context.Extension<OptimizerContext>().session;
+
+#endif
+
         public OptimizerSession(GameObject rootObject, bool addToAsset, bool isTest) :
             this(rootObject, addToAsset ? Utils.CreateAssetFile() : null, isTest)
         {

Editor/Processors/OptimizerPlugin.cs

@@ -0,0 +1,73 @@
+using System;
+using Anatawa12.AvatarOptimizer.ErrorReporting;
+using Anatawa12.AvatarOptimizer.ndmf;
+using nadena.dev.ndmf;
+using nadena.dev.ndmf.builtin;
+
+[assembly: ExportsPlugin(typeof(OptimizerPlugin))]
+
+namespace Anatawa12.AvatarOptimizer.ndmf
+{
+    internal class OptimizerContext : IExtensionContext
+    {
+        private IDisposable buildReportScope;
+        internal OptimizerSession session;
+
+        public void OnActivate(BuildContext context)
+        {
+            buildReportScope = BuildReport.ReportingOnAvatar(context.AvatarDescriptor);
+            session = new OptimizerSession(context.AvatarRootObject, context.AssetContainer,
+                false);
+        }
+
+        public void OnDeactivate(BuildContext context)
+        {
+            session.MarkDirtyAll();
+            buildReportScope.Dispose();
+        }
+    }
+
+    internal class OptimizerPlugin : Plugin<OptimizerPlugin>
+    {
+        public override string DisplayName => "Anatawa12's Avatar Optimizer";
+
+        public override string QualifiedName => "com.anatawa12.avatar-optimizer";
+
+        protected override void Configure()
+        {
+            // Run early steps before EditorOnly objects are purged
+            InPhase(BuildPhase.Resolving)
+                .WithRequiredExtension(typeof(OptimizerContext), seq =>
+                {
+                    seq.Run("Early: UnusedBonesByReference",
+                            ctx => new Processors.UnusedBonesByReferencesToolEarlyProcessor().Process(ctx)
+                        )
+                        .Then.Run("Early: MakeChildren",
+                            ctx => new Processors.MakeChildrenProcessor(early: true).Process(ctx)
+                        )
+                        .BeforePass(RemoveEditorOnlyPass.Instance);
+                });
+
+            // Run everything else in the optimize phase
+            InPhase(BuildPhase.Optimizing)
+                .WithRequiredExtension(typeof(OptimizerContext), seq =>
+                {
+                    seq.Run("TraceAndOptimize",
+                            ctx => new Processors.TraceAndOptimizeProcessor().Process(ctx)
+                        )
+                        .Then.Run("ClearEndpointPosition",
+                            ctx => new Processors.ClearEndpointPositionProcessor().Process(ctx)
+                        )
+                        .Then.Run("MergePhysBone",
+                            ctx => new Processors.MergePhysBoneProcessor().Process(ctx)
+                        )
+                        .Then.Run("MakeChildrenProcessor",
+                            ctx => new Processors.MakeChildrenProcessor(early: false).Process(ctx)
+                        )
+                        .Then.Run("ApplyObjectMapping",
+                            ctx => new Processors.ApplyObjectMapping().Apply(ctx)
+                        );
+                });
+        }
+    }
+}

Editor/Utils.cs

@@ -3,7 +3,9 @@
 using System.Collections.Generic;
 using System.IO;
 using System.Linq;
+#if !NADEMOFU
 using Anatawa12.ApplyOnPlay;
+#endif
 using JetBrains.Annotations;
 using Unity.Collections;
 using UnityEditor;
@@ -321,7 +323,8 @@ public static void DeleteTemporalDirectory()
             AssetDatabase.DeleteAsset(TemporalDirPath);
             FileUtil.DeleteFileOrDirectory(TemporalDirPath);
         }
-
+
+#if !NADEMOFU
         [CanBeNull]
         public static DummyObject CreateOutputAssetFile(GameObject avatarGameObject, ApplyReason reason)
         {
@@ -334,6 +337,7 @@ public static DummyObject CreateOutputAssetFile(GameObject avatarGameObject, App
                     return CreateOutputAssetFile(avatarGameObject);
             }
         }
+#endif
 
         public static DummyObject CreateAssetFile()
         {

Editor/com.anatawa12.avatar-optimizer.editor.asmdef

@@ -9,7 +9,8 @@
         "GUID:f69eeb3e25674f4a9bd20e6d7e69e0e6",
         "GUID:2633ab9fa94544a69517fc9a1bc143c9",
         "GUID:b9880ca0b6584453a2627bd3c038759f",
-        "GUID:8542dbf824204440a818dbc2377cb4d6"
+        "GUID:8542dbf824204440a818dbc2377cb4d6",
+        "GUID:62ced99b048af7f4d8dfe4bed8373d76"
     ],
     "includePlatforms": [
         "Editor"
@@ -29,6 +30,12 @@
     ],
     "autoReferenced": false,
     "defineConstraints": [],
-    "versionDefines": [],
+    "versionDefines": [
+        {
+            "name": "nadena.dev.ndmf",
+            "expression": "[0.3.0,999.999.999]",
+            "define": "NADEMOFU"
+        }
+    ],
     "noEngineReferences": false
 }

Internal/ApplyOnPlay/Editor/ApplyOnPlayCallbackRegistry.cs

@@ -1,3 +1,4 @@
+#if !NADEMOFU
 using System;
 using System.Collections.Generic;
 using System.Linq;
@@ -101,4 +102,5 @@ private static void LogException(string message, params Exception[] exceptions)
             }
         }
     }
-}
+}
+#endif

Internal/ApplyOnPlay/Editor/ApplyOnPlayCaller.cs

@@ -1,3 +1,4 @@
+#if !NADEMOFU
 using System;
 using System.Diagnostics;
 using System.Linq;
@@ -143,4 +144,5 @@ public static void CallManualBakeFinalizer(IManualBakeFinalizer[] finalizers, Ga
             }
         }
     }
-}
+}
+#endif

Internal/ApplyOnPlay/Editor/ApplyOnPlayConfiguration.cs

@@ -1,3 +1,4 @@
+#if !NADEMOFU
 using System.Linq;
 using CustomLocalization4EditorExtension;
 using UnityEditor;
@@ -64,4 +65,5 @@ private void OnGUI()
             GUILayout.EndHorizontal();
         }
     }
-}
+}
+#endif

Internal/ApplyOnPlay/Editor/IApplyOnPlayCallback.cs

@@ -1,3 +1,4 @@
+#if !NADEMOFU
 using UnityEditor.Build;
 using UnityEngine;
 
@@ -33,3 +34,4 @@ public enum ApplyReason
         ManualBake,
     }
 }
+#endif

Internal/ApplyOnPlay/Editor/ManualBake.cs

@@ -1,3 +1,4 @@
+#if !NADEMOFU
 using UnityEditor;
 using UnityEngine;
 using VRC.SDK3.Avatars.Components;
@@ -45,3 +46,4 @@ private static void ExecuteManualBake()
         }
     }
 }
+#endif

Internal/ApplyOnPlay/Editor/ManualBakeFinallizer.cs

@@ -1,3 +1,4 @@
+#if !NADEMOFU
 using UnityEditor.Build;
 using UnityEngine;
 
@@ -27,3 +28,4 @@ public interface IManualBakeFinalizer : IOrderedCallback
         void FinalizeManualBake(GameObject original, GameObject cloned);
     }
 }
+#endif

Internal/ApplyOnPlay/Editor/RemoveEditorOnlyOnPlay.cs

@@ -1,3 +1,4 @@
+#if !NADEMOFU
 using UnityEngine;
 using Object = UnityEngine.Object;
 
@@ -17,4 +18,5 @@ public bool ApplyOnPlay(GameObject avatarGameObject, ApplyReason reason)
             return true;
         }
     }
-}
+}
+#endif

Internal/ApplyOnPlay/Editor/com.anatawa12.avatar-optimizer.internal.apply-on-play.editor.asmdef

@@ -21,8 +21,13 @@
     "versionDefines": [
         {
             "name": "nadena.dev.modular-avatar",
-            "expression": "[1.0.0,2.0.0)",
+            "expression": "[1.0.0,1.7.99999)",
             "define": "MODULAR_AVATAR"
+        },
+        {
+            "name": "nadena.dev.ndmf",
+            "expression": "[0.3.0,99999.0.0)",
+            "define": "NADEMOFU"
         }
     ],
     "noEngineReferences": false

Internal/ErrorReporter/Editor/ErrorReportingInitializerProcessor.cs

@@ -1,22 +1,29 @@
 using System.Collections.Generic;
-using Anatawa12.ApplyOnPlay;
 using UnityEngine;
 using VRC.SDK3.Avatars.Components;
 using VRC.SDKBase.Editor.BuildPipeline;
+#if !NADEMOFU
+using Anatawa12.ApplyOnPlay;
+#endif
 
 namespace Anatawa12.AvatarOptimizer.ErrorReporting
 {
     /// <summary>
     /// Initializes Error Reporting System
     /// </summary>
-    internal class ErrorReportingInitializerProcessor : IVRCSDKPreprocessAvatarCallback, IApplyOnPlayCallback, IManualBakeFinalizer
+    internal class ErrorReportingInitializerProcessor : IVRCSDKPreprocessAvatarCallback
+#if !NADEMOFU
+        , IApplyOnPlayCallback, IManualBakeFinalizer
+#endif
     {
         public int callbackOrder => int.MinValue;
 
         public string CallbackName => "Error Reporting Initialization";
         public string CallbackId => "com.anatawa12.error-reporting";
 
+#if !NADEMOFU
         public bool ApplyOnPlay(GameObject avatarGameObject, ApplyReason reason) => OnPreprocessAvatar(avatarGameObject);
+#endif
 
         public bool OnPreprocessAvatar(GameObject avatarGameObject)
         {

Internal/ErrorReporter/Editor/com.anatawa12.avatar-optimizer.internal.error-reporter.editor.asmdef

@@ -19,6 +19,12 @@
     ],
     "autoReferenced": false,
     "defineConstraints": [],
-    "versionDefines": [],
+    "versionDefines": [
+        {
+            "name": "nadena.dev.ndmf",
+            "expression": "[0.3.0,99999.0.0)",
+            "define": "NADEMOFU"
+        }
+    ],
     "noEngineReferences": false
 }