// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "webkit/media/filter_helpers.h"

#include "base/bind.h"
#include "media/base/filter_collection.h"
#include "media/base/message_loop_factory.h"
#include "media/filters/chunk_demuxer.h"
#include "media/filters/dummy_demuxer.h"
#include "media/filters/ffmpeg_audio_decoder.h"
#include "media/filters/ffmpeg_demuxer.h"
#include "media/filters/ffmpeg_video_decoder.h"
#include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebURL.h"
#include "webkit/media/media_stream_client.h"

namespace webkit_media {

// Constructs and adds the default audio/video decoders to |filter_collection|.
static void AddDefaultDecodersToCollection(
    media::MessageLoopFactory* message_loop_factory,
    media::FilterCollection* filter_collection,
    scoped_refptr<media::FFmpegVideoDecoder>* ffmpeg_video_decoder) {
  filter_collection->AddAudioDecoder(new media::FFmpegAudioDecoder(
      base::Bind(&media::MessageLoopFactory::GetMessageLoop,
                 base::Unretained(message_loop_factory),
                 "AudioDecoderThread")));
  *ffmpeg_video_decoder = new media::FFmpegVideoDecoder(
      base::Bind(&media::MessageLoopFactory::GetMessageLoop,
                 base::Unretained(message_loop_factory),
                 "VideoDecoderThread"));
  filter_collection->AddVideoDecoder(*ffmpeg_video_decoder);
}

bool BuildMediaStreamCollection(const WebKit::WebURL& url,
                                MediaStreamClient* client,
                                media::MessageLoopFactory* message_loop_factory,
                                media::FilterCollection* filter_collection) {
  if (!client)
    return false;

  scoped_refptr<media::VideoDecoder> video_decoder = client->GetVideoDecoder(
      url, message_loop_factory);
  if (!video_decoder)
    return false;

  // Remove any "traditional" decoders (e.g. GpuVideoDecoder) from the
  // collection.
  // NOTE: http://crbug.com/110800 is about replacing this ad-hockery with
  // something more designed.
  scoped_refptr<media::VideoDecoder> old_videodecoder;
  do {
    filter_collection->SelectVideoDecoder(&old_videodecoder);
  } while (old_videodecoder);

  filter_collection->AddVideoDecoder(video_decoder);

  // TODO(vrk/wjia): Setting true for local_source is under the assumption
  // that the MediaStream represents a local webcam. This will need to
  // change in the future when GetVideoDecoder is no longer hardcoded to
  // only return CaptureVideoDecoders.
  //
  // See http://crbug.com/120426 for details.
  filter_collection->SetDemuxer(new media::DummyDemuxer(true, false, true));

  return true;
}

bool BuildMediaSourceCollection(
    const WebKit::WebURL& url,
    const WebKit::WebURL& media_source_url,
    media::ChunkDemuxerClient* client,
    media::MessageLoopFactory* message_loop_factory,
    media::FilterCollection* filter_collection,
    scoped_refptr<media::FFmpegVideoDecoder>* video_decoder) {
  if (media_source_url.isEmpty() || url != media_source_url)
    return false;

  filter_collection->SetDemuxer(new media::ChunkDemuxer(client));

  AddDefaultDecodersToCollection(message_loop_factory, filter_collection,
                                 video_decoder);
  return true;
}

void BuildDefaultCollection(
    const scoped_refptr<media::DataSource>& data_source,
    bool local_source,
    media::MessageLoopFactory* message_loop_factory,
    media::FilterCollection* filter_collection,
    scoped_refptr<media::FFmpegVideoDecoder>* video_decoder) {
  filter_collection->SetDemuxer(new media::FFmpegDemuxer(
      message_loop_factory->GetMessageLoop("PipelineThread"),
      data_source,
      local_source));

  AddDefaultDecodersToCollection(message_loop_factory, filter_collection,
                                 video_decoder);
}

}  // webkit_media