AnimeFunsion/modular-avatar
1
0
Fork 0
mirror of https://github.com/bdunderscore/modular-avatar.git synced 2025-01-30 18:22:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

ParameterPolicy: Detect parameter driver parameters

Browse Source 
Closes: #60
This commit is contained in:
bd_ 2022-11-07 20:28:43 -08:00
parent 8bb1969ecf
commit ec4a5db0c5
1 changed files with 41 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
Packages/net.fushizen.modular-avatar/Editor/ParameterPolicy.cs
View File

@ -4,6 +4,7 @@ using System.Collections.Immutable;
using System.Linq;
using UnityEditor.Animations;
using UnityEngine;
using VRC.SDK3.Avatars.Components;
using VRC.SDK3.Avatars.ScriptableObjects;
using VRC.SDK3.Dynamics.Contact.Components;
using VRC.SDK3.Dynamics.PhysBone.Components;
@ -212,10 +213,48 @@ namespace net.fushizen.modular_avatar.core.editor
foreach (var parameter in controller.parameters)
{
if (VRCSDKParameters.Contains(parameter.name)) continue;
AddDetectedParameter(parameter.name);
}
HashSet<AnimatorStateMachine> visitedStateMachines = new HashSet<AnimatorStateMachine>();
Queue<AnimatorStateMachine> stateMachines =
new Queue<AnimatorStateMachine>(controller.layers.Select(l => l.stateMachine));
while (stateMachines.Count > 0)
{
var sm = stateMachines.Dequeue();
if (visitedStateMachines.Contains(sm)) continue;
visitedStateMachines.Add(sm);
foreach (var subStateMachine in sm.stateMachines)
{
stateMachines.Enqueue(subStateMachine.stateMachine);
}
foreach (var state in sm.states)
{
foreach (var behavior in state.state.behaviours)
{
if (behavior is VRCAvatarParameterDriver driver)
{
foreach (var p in driver.parameters)
{
AddDetectedParameter(p.name);
AddDetectedParameter(p.source);
AddDetectedParameter(p.destParam as string);
AddDetectedParameter(p.sourceParam as string);
}
}
}
}
}
void AddDetectedParameter(string parameterName)
{
if (string.IsNullOrEmpty(parameterName) || VRCSDKParameters.Contains(parameterName)) return;
var param = new DetectedParameter()
{
OriginalName = parameter.name,
OriginalName = parameterName,
IsPrefix = false
};
parameters[param.MapKey] = param;