refactor: get PixelData

homuler · Nov 29, 2020 · 9d386dc · 9d386dc
1 parent 682b6a8
commit 9d386dc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/Assets/MediaPipe/Examples/Scripts/SceneDirector.cs b/Assets/MediaPipe/Examples/Scripts/SceneDirector.cs
@@ -157,10 +157,7 @@ IEnumerator RunGraph() {
         break;
       }
 
-      var colors = webCamScreenController.GetPixels32();
-      var width = webCamScreenController.Width();
-      var height = webCamScreenController.Height();
-      var pixelData = new PixelData(colors, width, height);
+      var pixelData = webCamScreenController.GetPixelData();
 
       graph.PushInput(pixelData).AssertOk();
       graph.RenderOutput(webCamScreenController, pixelData);

diff --git a/Assets/MediaPipe/Examples/Scripts/WebCamScreenController.cs b/Assets/MediaPipe/Examples/Scripts/WebCamScreenController.cs
@@ -55,6 +55,10 @@ public Color32[] GetPixels32() {
     return IsPlaying() ? webCamTexture.GetPixels32(pixelData) : null;
   }
 
+  public PixelData GetPixelData() {
+    return new PixelData(GetPixels32(), Width(), Height());
+  }
+
   public Texture2D GetScreen() {
     return outputTexture;
   }