From 709264ea2cdcef8cb00da5d7791418e50006ce0d Mon Sep 17 00:00:00 2001
From: luzhuang <364439895@qq.com>
Date: Mon, 20 Mar 2023 11:23:25 +0800
Subject: [PATCH] `AnimatorControllerLoader` add StateMachineScript  support
 (#1378)

* feat: `AnimatorControllerLoader` add StateMachineScript  support
---
 packages/loader/src/AnimatorControllerLoader.ts | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/packages/loader/src/AnimatorControllerLoader.ts b/packages/loader/src/AnimatorControllerLoader.ts
index 7382163d5..28e925a7b 100644
--- a/packages/loader/src/AnimatorControllerLoader.ts
+++ b/packages/loader/src/AnimatorControllerLoader.ts
@@ -39,7 +39,8 @@ class AnimatorControllerLoader extends Loader<AnimatorController> {
                   clipStartNormalizedTime,
                   clipEndNormalizedTime,
                   isDefaultState,
-                  clip: clipData
+                  clip: clipData,
+                  scripts = []
                 } = stateData;
                 const state = stateMachine.addState(name);
                 isDefaultState && (stateMachine.defaultState = state);
@@ -47,6 +48,9 @@ class AnimatorControllerLoader extends Loader<AnimatorController> {
                 state.wrapMode = wrapMode;
                 state.clipStartTime = clipStartNormalizedTime;
                 state.clipEndTime = clipEndNormalizedTime;
+                scripts.forEach((script) => {
+                  state.addStateMachineScript(Loader.getClass(script));
+                });
                 if (clipData) {
                   promises.push(
                     new Promise((resolve) => {