use inferred animation bindings from aigc

Author: cn0809

spx-gui/src/apis/aigc.ts

@@ -13,6 +13,7 @@ import {
   type UniversalUrl
 } from './common'
 import type { AssetExtraSettings, AssetType } from './asset'
+import type { State } from '@/models/sprite.ts'
 
 export type ProjectSettings = {
   name: string
@@ -277,6 +278,7 @@ export async function enrichAssetSettings(
 export type SpriteContentSettings = {
   costumes: CostumeSettings[]
   animations: AnimationSettings[]
+  animationBindings?: Record<State, string | undefined>
 }
 
 export async function genSpriteContentSettings(

spx-gui/src/models/gen/aigc-mock.ts

@@ -17,6 +17,7 @@ import {
 } from '@/apis/aigc'
 import { AnimationLoopMode, ArtStyle, BackdropCategory, Perspective, SpriteCategory } from '@/apis/common'
 import * as aigcApis from '@/apis/aigc'
+import { State } from '@/models/sprite.ts'
 
 export function setupAigcMock() {
   vi.mock('@/apis/aigc', { spy: true })
@@ -167,7 +168,14 @@ export class MockAigcApis {
           loopMode: AnimationLoopMode.Loopable,
           referenceFrameUrl: null
         }
-      ]
+      ],
+      animationBindings: {
+        [State.Default]: 'walk',
+        [State.Die]: undefined,
+        [State.Step]: 'jump',
+        [State.Turn]: undefined,
+        [State.Glide]: undefined
+      }
     }
   }
 

spx-gui/src/models/gen/sprite-gen.test.ts

@@ -6,6 +6,7 @@ import * as fileHelpers from '@/models/common/file'
 import { makeProject } from '../common/test'
 import { setupAigcMock, MockAigcApis } from './aigc-mock'
 import { SpriteGen } from './sprite-gen'
+import { State } from '../sprite'
 import type { CostumeGen } from './costume-gen'
 import type { AnimationGen } from './animation-gen'
 import { RotationStyle } from '../sprite'
@@ -145,6 +146,8 @@ describe('SpriteGen', () => {
     expect(sprite.animations.length).toBe(2)
     expect(sprite.animations[0].name).toBe('walk')
     expect(sprite.animations[1].name).toBe('jump')
+    expect(sprite.getAnimationBoundStates(sprite.animations[0].id)).toEqual([State.Default])
+    expect(sprite.getAnimationBoundStates(sprite.animations[1].id)).toEqual([State.Step])
   })
 
   it('should validate sprite name correctly', async () => {

spx-gui/src/models/gen/sprite-gen.ts

@@ -38,6 +38,7 @@ export class SpriteGen extends Disposable {
   private genImagesTask: Task<TaskType.GenerateCostume>
   private genImagesPhase: Phase<File[]>
   private prepareContentPhase: Phase<void>
+  private inferredAnimationBindings: Record<State, string | undefined> | null = null
 
   constructor(i18n: I18n, project: Project, initialDescription = '') {
     super()
@@ -181,9 +182,17 @@ export class SpriteGen extends Disposable {
           generatedAnimations.forEach((a) => {
             if (!sprite.animations.includes(a)) {
               sprite.addAnimation(a)
-              // Auto bind to default state if no default animation is set
-              if (sprite.getDefaultAnimation() == null) {
-                sprite.setAnimationBoundStates(a.id, [State.Default])
+              // Auto bind to states based on inferred animation bindings if available.
+              // If the animation name has been changed by user, the binding may not be applied.
+              const inferredBindings = this.inferredAnimationBindings
+              if (inferredBindings != null) {
+                const animationName = a.name
+                const statesToBind = Object.entries(inferredBindings)
+                  .filter(([, name]) => name === animationName)
+                  .map(([state]) => state as State)
+                if (statesToBind.length > 0) {
+                  sprite.setAnimationBoundStates(a.id, statesToBind)
+                }
               }
             }
           })
@@ -195,10 +204,6 @@ export class SpriteGen extends Disposable {
     )
   }
 
-  get isPreparePhase() {
-    return this.contentPreparingState.status !== 'finished'
-  }
-
   get contentPreparingState() {
     return this.prepareContentPhase.state
   }
@@ -223,6 +228,7 @@ export class SpriteGen extends Disposable {
 
       // Generate additional costumes & animations
       const settings = await genSpriteContentSettings(this.settings)
+      this.inferredAnimationBindings = settings.animationBindings ?? null
       this.costumes.push(
         ...settings.costumes.map((s) => new CostumeGen(this, project, { ...s, referenceCostumeId: defaultCostume.id }))
       )