set interval to avoid block

iFwu · Oct 7, 2024 · 9baa0c3 · 9baa0c3
1 parent c817ce6
commit 9baa0c3
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/src/hooks/useOpenCV.ts b/src/hooks/useOpenCV.ts
@@ -23,21 +23,28 @@ export function useOpenCV() {
     }
   };
   useEffect(() => {
+    let intervalId: number;
     const initialize = async () => {
-      if (typeof cv !== 'undefined') {
+      if (window.cv) {
         await new Promise<void>((resolve) => {
           cv.onRuntimeInitialized = resolve;
         });
         await onOpenCVReady();
       } else {
-        document.addEventListener('opencv-loaded', onOpenCVReady);
+        window.addEventListener('opencv-loaded', onOpenCVReady);
+        intervalId = setInterval(() => {
+          if (!isLoading) {
+            onOpenCVReady();
+          }
+        }, 500);
       }
     }
 
     initialize();
 
     return () => {
-      document.removeEventListener('opencv-loaded', onOpenCVReady);
+      window.removeEventListener('opencv-loaded', onOpenCVReady);
+      clearInterval(intervalId);
     };
   }, []);
 

diff --git a/vite.config.ts b/vite.config.ts
@@ -28,7 +28,7 @@ export default defineConfig({
             injectTo: 'body',
             attrs: {
               async: true,
-              onload: 'cv.onRuntimeInitialized = () => { document.dispatchEvent(new Event(\'opencv-loaded\')); }',
+              onload: 'cv.onRuntimeInitialized = () => { window.dispatchEvent(new Event(\'opencv-loaded\')); }',
             },
           };
         },