refactor: add setReady

Tresjs · Jun 22, 2024 · 543fbfa · 543fbfa
1 parent e1df8a9
commit 543fbfa
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src/composables/useTresContextProvider/index.ts b/src/composables/useTresContextProvider/index.ts
@@ -212,9 +212,11 @@ export function useTresContextProvider({
 
   const { on: onTresReady, cancel: cancelTresReady } = useTresReady(ctx)!
 
+  ctx.loop.setReady(false)
+  ctx.loop.start()
   onTresReady(() => {
     emit('ready', ctx)
-    ctx.loop.start()
+    ctx.loop.setReady(true)
   })
 
   onUnmounted(() => {

diff --git a/src/core/loop.ts b/src/core/loop.ts
@@ -40,6 +40,7 @@ export interface RendererLoop {
   isActive: Ref<boolean>
   isRenderPaused: Ref<boolean>
   setContext: (newContext: Record<string, any>) => void
+  setReady: (isReady: boolean) => void
 }
 
 export function createRenderLoop(): RendererLoop {
@@ -52,6 +53,7 @@ export function createRenderLoop(): RendererLoop {
   const subscribersBefore = createPriorityEventHook<LoopCallbackWithCtx>()
   const subscriberRender = createPriorityEventHook<LoopCallbackWithCtx>()
   const subscribersAfter = createPriorityEventHook<LoopCallbackWithCtx>()
+  let isReady = true
 
   // Context to be passed to callbacks
   let context: Record<string, any> = {}
@@ -110,6 +112,10 @@ export function createRenderLoop(): RendererLoop {
   }
 
   function loop() {
+    if (!isReady) {
+      animationFrameId = requestAnimationFrame(loop)
+      return
+    }
     const delta = clock.getDelta()
     const elapsed = clock.getElapsedTime()
     const snapshotCtx = {
@@ -157,5 +163,6 @@ export function createRenderLoop(): RendererLoop {
     isRenderPaused,
     isActive,
     setContext,
+    setReady: (b: boolean) => isReady = b,
   }
 }