Allow setting a LoadControl on CompositionPlayer

PiperOrigin-RevId: 795460388

Author: ychaparov

libraries/transformer/src/main/java/androidx/media3/transformer/CompositionPlayer.java

@@ -61,7 +61,9 @@
 import androidx.media3.effect.DefaultVideoFrameProcessor;
 import androidx.media3.effect.SingleInputVideoGraph;
 import androidx.media3.effect.TimestampAdjustment;
+import androidx.media3.exoplayer.DefaultLoadControl;
 import androidx.media3.exoplayer.ExoPlayer;
+import androidx.media3.exoplayer.LoadControl;
 import androidx.media3.exoplayer.RendererCapabilities;
 import androidx.media3.exoplayer.RendererCapabilities.Capabilities;
 import androidx.media3.exoplayer.analytics.AnalyticsCollector;
@@ -130,6 +132,7 @@ public static final class Builder {
     private VideoGraph.@MonotonicNonNull Factory videoGraphFactory;
 
     private @MonotonicNonNull GlObjectsProvider glObjectsProvider;
+    private LoadControl loadControl;
     private boolean enableReplayableCache;
     private long lateThresholdToDropInputUs;
     private boolean built;
@@ -148,6 +151,7 @@ public Builder(Context context) {
               .setMaxOutputSize(GlUtil.MAX_BITMAP_DECODING_SIZE);
       videoPrewarmingEnabled = true;
       lateThresholdToDropInputUs = LATE_US_TO_DROP_INPUT_FRAME;
+      loadControl = new DefaultLoadControl();
       clock = Clock.DEFAULT;
     }
 
@@ -290,6 +294,21 @@ public Builder setGlObjectsProvider(GlObjectsProvider glObjectsProvider) {
       return this;
     }
 
+    /**
+     * Sets the {@link LoadControl} that will be used by the player to control buffering of all
+     * {@linkplain EditedMediaItem#mediaItem media items} in a {@link Composition}.
+     *
+     * <p>By default, a {@link DefaultLoadControl} is used.
+     *
+     * @param loadControl A {@link LoadControl}.
+     * @return This builder, for convenience.
+     */
+    @CanIgnoreReturnValue
+    public Builder setLoadControl(LoadControl loadControl) {
+      this.loadControl = loadControl;
+      return this;
+    }
+
     /**
      * Sets whether to enable replayable cache.
      *
@@ -402,6 +421,7 @@ public CompositionPlayer build() {
   private final VideoGraph.Factory videoGraphFactory;
   private final boolean videoPrewarmingEnabled;
   private final HandlerWrapper compositionInternalListenerHandler;
+  private final LoadControl loadControl;
   private final boolean enableReplayableCache;
   private final long lateThresholdToDropInputUs;
 
@@ -454,6 +474,7 @@ private CompositionPlayer(Builder builder) {
     videoGraphFactory = checkNotNull(builder.videoGraphFactory);
     videoPrewarmingEnabled = builder.videoPrewarmingEnabled;
     compositionInternalListenerHandler = clock.createHandler(builder.looper, /* callback= */ null);
+    loadControl = builder.loadControl;
     this.enableReplayableCache = builder.enableReplayableCache;
     lateThresholdToDropInputUs = builder.lateThresholdToDropInputUs;
     videoTracksSelected = new SparseBooleanArray();
@@ -1477,6 +1498,7 @@ private SequencePlayerHolder(
               .setPlaybackLooper(playbackLooper)
               .setRenderersFactory(renderersFactory)
               .setHandleAudioBecomingNoisy(true)
+              .setLoadControl(loadControl)
               .setClock(clock)
               // Use dynamic scheduling to show the first video/image frame more promptly when the
               // player is paused (which is common in editing applications).

libraries/transformer/src/test/java/androidx/media3/transformer/CompositionPlayerTest.java

@@ -48,6 +48,8 @@
 import androidx.media3.common.util.ConditionVariable;
 import androidx.media3.common.util.Log;
 import androidx.media3.common.util.NullableType;
+import androidx.media3.exoplayer.DefaultLoadControl;
+import androidx.media3.exoplayer.analytics.PlayerId;
 import androidx.media3.exoplayer.audio.AudioSink;
 import androidx.media3.exoplayer.audio.DefaultAudioSink;
 import androidx.media3.exoplayer.audio.ForwardingAudioSink;
@@ -946,6 +948,22 @@ public void setScrubbingModeEnabled_updatesPlaybackSuppressionReason() throws Ti
         .isEqualTo(PLAYBACK_SUPPRESSION_REASON_SCRUBBING);
   }
 
+  @Test
+  public void prepare_withCustomLoadControl_preparesTheLoadControl() throws Exception {
+    CustomLoadControl customLoadControl = new CustomLoadControl();
+    CompositionPlayer.Builder playerBuilder = createCompositionPlayerBuilder();
+    playerBuilder.setLoadControl(customLoadControl);
+    CompositionPlayer player = playerBuilder.build();
+
+    player.setComposition(buildComposition());
+    player.prepare();
+    TestPlayerRunHelper.runUntilPlaybackState(player, Player.STATE_READY);
+
+    assertThat(customLoadControl.prepared).isTrue();
+
+    player.release();
+  }
+
   private static CompositionPlayer buildCompositionPlayer() {
     return createCompositionPlayerBuilder().build();
   }
@@ -970,4 +988,14 @@ private static Composition buildComposition() {
         new EditedMediaItemSequence.Builder(editedMediaItem1, editedMediaItem2).build();
     return new Composition.Builder(sequence).build();
   }
+
+  private static final class CustomLoadControl extends DefaultLoadControl {
+    public boolean prepared;
+
+    @Override
+    public void onPrepared(PlayerId playerId) {
+      prepared = true;
+      super.onPrepared(playerId);
+    }
+  }
 }