refactor: setup gpu resources only once

homuler · Jan 31, 2021 · 4a65687 · 4a65687
1 parent 683b88f
commit 4a65687
Showing 1 changed file with 21 additions and 11 deletions.
diff --git a/Assets/MediaPipe/Examples/Scripts/SceneDirector.cs b/Assets/MediaPipe/Examples/Scripts/SceneDirector.cs
@@ -114,6 +114,9 @@ void StartGraph() {
       return;
     }
 
+    if (useGPU) {
+      SetupGpuResources();
+    }
     graphRunner = StartCoroutine(RunGraph());
   }
 
@@ -128,6 +131,24 @@ void StopGraph() {
     }
   }
 
+  void SetupGpuResources() {
+    if (gpuResources != null) {
+      Debug.Log("Gpu resources are already initialized");
+      return;
+    }
+
+    // TODO: have to wait for currentContext to be initialized.
+    if (currentContext == IntPtr.Zero) {
+      Debug.LogWarning("No EGL Context Found");
+    } else {
+      Debug.Log($"EGL Context Found ({currentContext})");
+    }
+
+    gpuResources = GpuResources.Create(currentContext).ConsumeValueOrDie();
+    gpuHelper = new GlCalculatorHelper();
+    gpuHelper.InitializeForTest(gpuResources);
+  }
+
   IEnumerator RunGraph() {
     yield return WaitForAssets();
 
@@ -155,17 +176,6 @@ IEnumerator RunGraph() {
     var graph = graphContainer.GetComponent<IDemoGraph<TextureFrame>>();
 
     if (useGPU) {
-      // TODO: have to wait for currentContext to be initialized.
-      if (currentContext == IntPtr.Zero) {
-        Debug.LogWarning("No EGL Context Found");
-      } else {
-        Debug.Log($"EGL Context Found ({currentContext})");
-      }
-
-      gpuResources = GpuResources.Create(currentContext).ConsumeValueOrDie();
-      gpuHelper = new GlCalculatorHelper();
-      gpuHelper.InitializeForTest(gpuResources);
-
       graph.Initialize(gpuResources, gpuHelper);
     } else {
       graph.Initialize();