/*
 * Copyright (C) 2015 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.android.tv.tuner.exoplayer;

import android.content.Context;

import com.google.android.exoplayer.SampleSource;
import com.google.android.exoplayer.TrackRenderer;
import com.google.android.exoplayer.upstream.DataSource;
import com.android.tv.tuner.exoplayer.MpegTsPlayer.RendererBuilder;
import com.android.tv.tuner.exoplayer.MpegTsPlayer.RendererBuilderCallback;
import com.android.tv.tuner.exoplayer.ac3.Ac3PassthroughTrackRenderer;
import com.android.tv.tuner.exoplayer.buffer.BufferManager;
import com.android.tv.tuner.tvinput.PlaybackBufferListener;

/**
 * Builder for renderer objects for {@link MpegTsPlayer}.
 */
public class MpegTsRendererBuilder implements RendererBuilder {
    private final Context mContext;
    private final BufferManager mBufferManager;
    private final PlaybackBufferListener mBufferListener;

    public MpegTsRendererBuilder(Context context, BufferManager bufferManager,
            PlaybackBufferListener bufferListener) {
        mContext = context;
        mBufferManager = bufferManager;
        mBufferListener = bufferListener;
    }

    @Override
    public void buildRenderers(MpegTsPlayer mpegTsPlayer, DataSource dataSource,
            RendererBuilderCallback callback) {
        // Build the video and audio renderers.
        SampleExtractor extractor = dataSource == null ?
                new MpegTsSampleExtractor(mBufferManager, mBufferListener) :
                new MpegTsSampleExtractor(dataSource, mBufferManager, mBufferListener);
        SampleSource sampleSource = new MpegTsSampleSource(extractor);
        MpegTsVideoTrackRenderer videoRenderer = new MpegTsVideoTrackRenderer(mContext,
                sampleSource, mpegTsPlayer.getMainHandler(), mpegTsPlayer);
        // TODO: Only using Ac3PassthroughTrackRenderer for A/V sync issue. We will use
        // {@link Ac3TrackRenderer} when we use ExoPlayer's extractor.
        TrackRenderer audioRenderer = new Ac3PassthroughTrackRenderer(sampleSource,
                mpegTsPlayer.getMainHandler(), mpegTsPlayer);
        Cea708TextTrackRenderer textRenderer = new Cea708TextTrackRenderer(sampleSource);

        TrackRenderer[] renderers = new TrackRenderer[MpegTsPlayer.RENDERER_COUNT];
        renderers[MpegTsPlayer.TRACK_TYPE_VIDEO] = videoRenderer;
        renderers[MpegTsPlayer.TRACK_TYPE_AUDIO] = audioRenderer;
        renderers[MpegTsPlayer.TRACK_TYPE_TEXT] = textRenderer;
        callback.onRenderers(null, renderers);
    }
}