1 files changed, 43 insertions, 19 deletions

diff --git a/src/com/android/tv/tuner/exoplayer/MpegTsPlayer.java b/src/com/android/tv/tuner/exoplayer/MpegTsPlayer.java
index 381b22e9..ba0edf20 100644
--- a/src/com/android/tv/tuner/exoplayer/MpegTsPlayer.java
+++ b/src/com/android/tv/tuner/exoplayer/MpegTsPlayer.java
@@ -39,8 +39,8 @@ import com.android.tv.common.SoftPreconditions;
 import com.android.tv.tuner.data.Cea708Data;
 import com.android.tv.tuner.data.Cea708Data.CaptionEvent;
 import com.android.tv.tuner.data.TunerChannel;
-import com.android.tv.tuner.exoplayer.ac3.Ac3PassthroughTrackRenderer;
-import com.android.tv.tuner.exoplayer.ac3.Ac3TrackRenderer;
+import com.android.tv.tuner.exoplayer.ac3.Ac3DefaultTrackRenderer;
+import com.android.tv.tuner.exoplayer.ac3.Ac3MediaCodecTrackRenderer;
 import com.android.tv.tuner.source.TsDataSource;
 import com.android.tv.tuner.source.TsDataSourceManager;
 import com.android.tv.tuner.tvinput.EventDetector;
@@ -48,11 +48,12 @@ import com.android.tv.tuner.tvinput.EventDetector;
 import java.lang.annotation.Retention;
 import java.lang.annotation.RetentionPolicy;
 
-/**
- * MPEG-2 TS stream player implementation using ExoPlayer.
- */
-public class MpegTsPlayer implements ExoPlayer.Listener, MediaCodecVideoTrackRenderer.EventListener,
-        Ac3PassthroughTrackRenderer.EventListener, Ac3TrackRenderer.Ac3EventListener {
+/** MPEG-2 TS stream player implementation using ExoPlayer. */
+public class MpegTsPlayer
+        implements ExoPlayer.Listener,
+                MediaCodecVideoTrackRenderer.EventListener,
+                Ac3DefaultTrackRenderer.EventListener,
+                Ac3MediaCodecTrackRenderer.Ac3EventListener {
     private int mCaptionServiceNumber = Cea708Data.EMPTY_SERVICE_NUMBER;
 
     /**
@@ -304,8 +305,10 @@ public class MpegTsPlayer implements ExoPlayer.Listener, MediaCodecVideoTrackRen
         SoftPreconditions.checkState(supportSmoothTrickPlay(playbackParams.getSpeed()));
         mPlayer.setPlayWhenReady(true);
         mTrickplayRunning = true;
-        if (mAudioRenderer instanceof Ac3PassthroughTrackRenderer) {
-            mPlayer.sendMessage(mAudioRenderer, Ac3PassthroughTrackRenderer.MSG_SET_PLAYBACK_SPEED,
+        if (mAudioRenderer instanceof Ac3DefaultTrackRenderer) {
+            mPlayer.sendMessage(
+                    mAudioRenderer,
+                    Ac3DefaultTrackRenderer.MSG_SET_PLAYBACK_SPEED,
                     playbackParams.getSpeed());
         } else {
             mPlayer.sendMessage(mAudioRenderer,
@@ -317,10 +320,9 @@ public class MpegTsPlayer implements ExoPlayer.Listener, MediaCodecVideoTrackRen
     private void stopSmoothTrickplay(boolean calledBySeek) {
         if (mTrickplayRunning) {
             mTrickplayRunning = false;
-            if (mAudioRenderer instanceof Ac3PassthroughTrackRenderer) {
-                mPlayer.sendMessage(mAudioRenderer,
-                        Ac3PassthroughTrackRenderer.MSG_SET_PLAYBACK_SPEED,
-                        1.0f);
+            if (mAudioRenderer instanceof Ac3DefaultTrackRenderer) {
+                mPlayer.sendMessage(
+                        mAudioRenderer, Ac3DefaultTrackRenderer.MSG_SET_PLAYBACK_SPEED, 1.0f);
             } else {
                 mPlayer.sendMessage(mAudioRenderer,
                         MediaCodecAudioTrackRenderer.MSG_SET_PLAYBACK_PARAMS,
@@ -423,8 +425,8 @@ public class MpegTsPlayer implements ExoPlayer.Listener, MediaCodecVideoTrackRen
      */
     public void setVolume(float volume) {
         mVolume = volume;
-        if (mAudioRenderer instanceof Ac3PassthroughTrackRenderer) {
-            mPlayer.sendMessage(mAudioRenderer, Ac3PassthroughTrackRenderer.MSG_SET_VOLUME, volume);
+        if (mAudioRenderer instanceof Ac3DefaultTrackRenderer) {
+            mPlayer.sendMessage(mAudioRenderer, Ac3DefaultTrackRenderer.MSG_SET_VOLUME, volume);
         } else {
             mPlayer.sendMessage(mAudioRenderer, MediaCodecAudioTrackRenderer.MSG_SET_VOLUME,
                     volume);
@@ -437,9 +439,9 @@ public class MpegTsPlayer implements ExoPlayer.Listener, MediaCodecVideoTrackRen
      * @param enable enables the audio when {@code true}, disables otherwise.
      */
     public void setAudioTrack(boolean enable) {
-        if (mAudioRenderer instanceof Ac3PassthroughTrackRenderer) {
-            mPlayer.sendMessage(mAudioRenderer, Ac3PassthroughTrackRenderer.MSG_SET_AUDIO_TRACK,
-                    enable ? 1 : 0);
+        if (mAudioRenderer instanceof Ac3DefaultTrackRenderer) {
+            mPlayer.sendMessage(
+                    mAudioRenderer, Ac3DefaultTrackRenderer.MSG_SET_AUDIO_TRACK, enable ? 1 : 0);
         } else {
             mPlayer.sendMessage(mAudioRenderer, MediaCodecAudioTrackRenderer.MSG_SET_VOLUME,
                     enable ? mVolume : 0.0f);
@@ -495,6 +497,28 @@ public class MpegTsPlayer implements ExoPlayer.Listener, MediaCodecVideoTrackRen
     }
 
     /**
+     * Returns the index of the currently selected track for the specified renderer.
+     *
+     * @param rendererIndex The index of the renderer.
+     * @return The selected track. A negative value or a value greater than or equal to the renderer's
+     *     track count indicates that the renderer is disabled.
+     */
+    public int getSelectedTrack(int rendererIndex) {
+        return mPlayer.getSelectedTrack(rendererIndex);
+    }
+
+    /**
+     * Returns the format of a track.
+     *
+     * @param rendererIndex The index of the renderer.
+     * @param trackIndex The index of the track.
+     * @return The format of the track.
+     */
+    public MediaFormat getTrackFormat(int rendererIndex, int trackIndex) {
+        return mPlayer.getTrackFormat(rendererIndex, trackIndex);
+    }
+
+    /**
      * Gets the main handler of the player.
      */
     /* package */ Handler getMainHandler() {
@@ -650,4 +674,4 @@ public class MpegTsPlayer implements ExoPlayer.Listener, MediaCodecVideoTrackRen
             }
         }
     }
-}
+}
\ No newline at end of file