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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
|
#if UNITY_EDITOR
using UnityEngine;
using nadena.dev.ndmf;
using nadena.dev.ndmf.builtin;
using nadena.dev.ndmf.localization;
using nadena.dev.ndmf.VRChat;
using VRC.SDK3.Avatars.ScriptableObjects;
using UnityEditor;
using System;
using System.Collections.Generic;
using System.Linq;
using UnityEditor.Animations;
[assembly: ExportsPlugin(typeof(YOTS.YOTSNDMFGenerator))]
namespace YOTS
{
public class YOTSNDMFGenerator : Plugin<YOTSNDMFGenerator> {
private readonly Localizer localizer = new Localizer("en-us", () =>
new List<(string, Func<string, string>)> {
("en-us", key => key)
});
public override string DisplayName => "YOTS Animator Generator";
protected override void Configure() {
// First pass: Retrieve and stash configuration
InPhase(BuildPhase.Resolving)
.Run("Cache YOTS Config", ctx => {
var config = ctx.AvatarRootObject.GetComponentInChildren<YOTSNDMFConfig>();
if (config == null) {
ctx.GetState<YOTSBuildState>().skipGeneration = true;
Debug.Log("No YOTS config found - skipping.");
return;
}
if (config.jsonConfig == null) {
ctx.GetState<YOTSBuildState>().skipGeneration = true;
ErrorReport.WithContextObject(ctx.AvatarRootObject, () => {
ErrorReport.ReportException(
new Exception("No YOTS config found"),
"Missing required YOTS configuration"
);
});
return;
}
ctx.GetState<YOTSBuildState>().jsonConfig = config.jsonConfig.text;
})
// Shoutsout anatawa12/AvatarOptimizer
.BeforePass(RemoveEditorOnlyPass.Instance);
// Second pass: Generate and merge animator
InPhase(BuildPhase.Transforming)
.Run("Generate YOTS Animator", ctx => {
var config = ctx.GetState<YOTSBuildState>();
if (config.skipGeneration) {
return;
}
if (config == null) {
ErrorReport.WithContextObject(ctx.AvatarRootObject, () => {
ErrorReport.ReportException(
new Exception("No YOTS config component found"),
"Missing required YOTS configuration"
);
});
return;
}
if (config.jsonConfig == null) {
ErrorReport.WithContextObject(ctx.AvatarRootObject, () => {
ErrorReport.ReportException(
new Exception("Missing JSON config file"),
"YOTS config component is missing required JSON configuration"
);
});
return;
}
// Get menu and parameters
var descriptor = ctx.AvatarDescriptor;
if (descriptor == null) {
ErrorReport.WithContextObject(ctx.AvatarRootObject, () => {
ErrorReport.ReportException(
new Exception("Avatar descriptor is missing"),
"Cannot find VRC Avatar Descriptor"
);
});
return;
}
RuntimeAnimatorController originalAnimator = descriptor.baseAnimationLayers[4].animatorController;
var menu = descriptor.expressionsMenu;
var parameters = descriptor.expressionParameters;
if (parameters == null || menu == null) {
ErrorReport.WithContextObject(descriptor, () => {
ErrorReport.ReportException(
new Exception("Missing required VRC assets"),
"Avatar is missing required Expression Parameters or Menu"
);
});
return;
}
// Create copies so the originals don't get modified
menu = DeepCopyMenu(menu);
parameters = UnityEngine.Object.Instantiate(parameters);
descriptor.expressionsMenu = menu;
descriptor.expressionParameters = parameters;
// Generate the YOTS animator.
RuntimeAnimatorController generatedAnimator = YOTSCore.GenerateAnimator(
config.jsonConfig,
parameters,
menu
);
if (generatedAnimator == null) {
ErrorReport.WithContextObject(ctx.AvatarRootObject, () => {
ErrorReport.ReportException(
new Exception("Failed to generate animator"),
"YOTS animator generation failed"
);
});
return;
}
// If no original animator, just assign the generated one.
if (originalAnimator == null) {
descriptor.baseAnimationLayers[4].animatorController = generatedAnimator;
return;
}
// Else append the generated animator to the original.
AnimatorController originalController = originalAnimator as AnimatorController;
AnimatorController generatedController = generatedAnimator as AnimatorController;
MergeAnimatorControllers(localizer, originalController, generatedController);
descriptor.baseAnimationLayers[4].animatorController = originalController;
});
}
// Simply append generated params and layers to the original animator.
private static void MergeAnimatorControllers(Localizer localizer, AnimatorController original, AnimatorController generated) {
// Merge parameters from generated into original.
foreach (var genParam in generated.parameters) {
// This is an O(m*n) check but m and n should be small enough to not matter.
if (original.parameters.Any(p => p.name == genParam.name)) {
ErrorReport.WithContextObject(original, () => {
ErrorReport.ReportException(
new Exception($"Parameter '{genParam.name}' already exists"),
"Parameter name conflict in animator"
);
});
return;
}
original.AddParameter(genParam);
}
// Append each YOTS layer after the original layers.
foreach (var genLayer in generated.layers) {
// This isn't strictly an error but if someone already has layers named
// YOTS_* that's probably not on purpose.
if (original.layers.Any(l => l.name == genLayer.name)) {
ErrorReport.WithContextObject(original, () => {
ErrorReport.ReportException(
new Exception($"Layer '{genLayer.name}' already exists"),
"Layer name conflict in animator"
);
});
return;
}
var newLayer = new AnimatorControllerLayer {
name = genLayer.name,
defaultWeight = genLayer.defaultWeight,
stateMachine = genLayer.stateMachine
};
original.AddLayer(newLayer);
}
}
private class YOTSBuildState {
public string jsonConfig;
public bool skipGeneration;
}
private static VRCExpressionsMenu DeepCopyMenu(VRCExpressionsMenu sourceMenu) {
var copiedMenu = UnityEngine.Object.Instantiate(sourceMenu);
// Deep copy all submenu references
for (int i = 0; i < copiedMenu.controls.Count; i++) {
var control = copiedMenu.controls[i];
if (control.type == VRCExpressionsMenu.Control.ControlType.SubMenu) {
control.subMenu = DeepCopyMenu(control.subMenu);
}
}
return copiedMenu;
}
}
}
#endif // UNITY_EDITOR
|
