Add a getter for sleeping for offload

This allows wakelock releases to be state based
instead of transition based.

#exo-offload

PiperOrigin-RevId: 337286501

library/core/src/main/java/com/google/android/exoplayer2/ExoPlayer.java

@@ -631,7 +631,7 @@ public ExoPlayer build() {
   boolean getPauseAtEndOfMediaItems();
 
   /**
-   * Sets whether audio offload scheduling is enabled. If enabled, ExoPlayer's main loop will as
+   * Sets whether audio offload scheduling is enabled. If enabled, ExoPlayer's main loop will run as
    * rarely as possible when playing an audio stream using audio offload.
    *
    * <p>Only use this scheduling mode if the player is not displaying anything to the user. For
@@ -660,9 +660,19 @@ public ExoPlayer build() {
    *   <li>The {@link AudioSink} is playing with an offload {@link AudioTrack}.
    * </ul>
    *
+   * <p>The state where ExoPlayer main loop has been paused to save power during offload playback
+   * can be queried with {@link #experimentalIsSleepingForOffload()}.
+   *
    * <p>This method is experimental, and will be renamed or removed in a future release.
    *
    * @param offloadSchedulingEnabled Whether to enable offload scheduling.
    */
   void experimentalSetOffloadSchedulingEnabled(boolean offloadSchedulingEnabled);
+
+  /**
+   * Returns whether the player has paused its main loop to save power in offload scheduling mode.
+   *
+   * @see #experimentalSetOffloadSchedulingEnabled(boolean)
+   */
+  boolean experimentalIsSleepingForOffload();
 }

library/core/src/main/java/com/google/android/exoplayer2/ExoPlayerImpl.java

@@ -208,6 +208,11 @@ public void experimentalSetOffloadSchedulingEnabled(boolean offloadSchedulingEna
     internalPlayer.experimentalSetOffloadSchedulingEnabled(offloadSchedulingEnabled);
   }
 
+  @Override
+  public boolean experimentalIsSleepingForOffload() {
+    return playbackInfo.sleepingForOffload;
+  }
+
   @Override
   @Nullable
   public AudioComponent getAudioComponent() {

library/core/src/main/java/com/google/android/exoplayer2/SimpleExoPlayer.java

@@ -732,9 +732,16 @@ protected SimpleExoPlayer(Builder builder) {
 
   @Override
   public void experimentalSetOffloadSchedulingEnabled(boolean offloadSchedulingEnabled) {
+    verifyApplicationThread();
     player.experimentalSetOffloadSchedulingEnabled(offloadSchedulingEnabled);
   }
 
+  @Override
+  public boolean experimentalIsSleepingForOffload() {
+    verifyApplicationThread();
+    return player.experimentalIsSleepingForOffload();
+  }
+
   @Override
   @Nullable
   public AudioComponent getAudioComponent() {
@@ -2154,7 +2161,9 @@ private void updateWakeAndWifiLock() {
     switch (playbackState) {
       case Player.STATE_READY:
       case Player.STATE_BUFFERING:
-        wakeLockManager.setStayAwake(getPlayWhenReady());
+        boolean isSleeping = experimentalIsSleepingForOffload();
+        wakeLockManager.setStayAwake(getPlayWhenReady() && !isSleeping);
+        // The wifi lock is not released while sleeping to avoid interrupting downloads.
         wifiLockManager.setStayAwake(getPlayWhenReady());
         break;
       case Player.STATE_ENDED:
@@ -2495,12 +2504,7 @@ public void onPlayWhenReadyChanged(
 
     @Override
     public void onExperimentalSleepingForOffloadChanged(boolean sleepingForOffload) {
-      if (sleepingForOffload) {
-        // The wifi lock is not released to avoid interrupting downloads.
-        wakeLockManager.setStayAwake(false);
-      } else {
-        updateWakeAndWifiLock();
-      }
+      updateWakeAndWifiLock();
     }
   }
 }

library/core/src/test/java/com/google/android/exoplayer2/ExoPlayerTest.java

@@ -8327,6 +8327,7 @@ public void enableOffloadScheduling_isEnable_playerSleeps() throws Exception {
     sleepRenderer.sleepOnNextRender();
 
     runUntilSleepingForOffload(player, /* expectedSleepForOffload= */ true);
+    assertThat(player.experimentalIsSleepingForOffload()).isTrue();
   }
 
   @Test
@@ -8345,6 +8346,7 @@ public void enableOffloadScheduling_isEnable_playerSleeps() throws Exception {
     player.experimentalSetOffloadSchedulingEnabled(false); // Force the player to exit offload sleep
 
     runUntilSleepingForOffload(player, /* expectedSleepForOffload= */ false);
+    assertThat(player.experimentalIsSleepingForOffload()).isFalse();
     runUntilPlaybackState(player, Player.STATE_ENDED);
   }
 
@@ -8362,6 +8364,7 @@ public void wakeupListenerWhileSleepingForOffload_isWokenUp_renderingResumes() t
     sleepRenderer.wakeup();
 
     runUntilSleepingForOffload(player, /* expectedSleepForOffload= */ false);
+    assertThat(player.experimentalIsSleepingForOffload()).isFalse();
     runUntilPlaybackState(player, Player.STATE_ENDED);
   }
 

testutils/src/main/java/com/google/android/exoplayer2/testutil/StubExoPlayer.java

@@ -460,4 +460,9 @@ public boolean getPauseAtEndOfMediaItems() {
   public void experimentalSetOffloadSchedulingEnabled(boolean offloadSchedulingEnabled) {
     throw new UnsupportedOperationException();
   }
+
+  @Override
+  public boolean experimentalIsSleepingForOffload() {
+    throw new UnsupportedOperationException();
+  }
 }