From ef39d317d560c9e22ebf42e5768a7dab05ed8474 Mon Sep 17 00:00:00 2001
From: huangdarwin <huangdarwin@google.com>
Date: Mon, 24 Jul 2023 16:57:19 +0100
Subject: [PATCH] Compositor: Signal onEnded when frames finish compositing.

PiperOrigin-RevId: 550576032
---
 .../src/main/java/androidx/media3/effect/VideoCompositor.java  | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/libraries/effect/src/main/java/androidx/media3/effect/VideoCompositor.java b/libraries/effect/src/main/java/androidx/media3/effect/VideoCompositor.java
index ecd21ccd4b..019f3c1774 100644
--- a/libraries/effect/src/main/java/androidx/media3/effect/VideoCompositor.java
+++ b/libraries/effect/src/main/java/androidx/media3/effect/VideoCompositor.java
@@ -238,6 +238,9 @@ public final class VideoCompositor {
       InputFrameInfo inputFrameInfo = framesToComposite.get(i);
       inputFrameInfo.releaseCallback.release(inputFrameInfo.presentationTimeUs);
     }
+    if (allInputsEnded && inputSources.get(PRIMARY_INPUT_ID).frameInfos.isEmpty()) {
+      listener.onEnded();
+    }
   }
 
   private synchronized boolean isReadyToComposite() {