Motown: ffmpeg implementations of framework 'parts'

services/media/framework_ffmpeg/ffmpeg_video_decoder.cc

+// Copyright 2016 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 "base/logging.h"
+#include "services/media/framework_ffmpeg/ffmpeg_video_decoder.h"
+
+namespace mojo {
+namespace media {
+
+FfmpegVideoDecoder::FfmpegVideoDecoder(AvCodecContextPtr av_codec_context) :
+    FfmpegDecoderBase(std::move(av_codec_context)) {
+  DCHECK(context());
+}
+
+FfmpegVideoDecoder::~FfmpegVideoDecoder() {}
+
+int FfmpegVideoDecoder::Decode(
+    PayloadAllocator* allocator,
+    bool* frame_decoded_out) {
+  DCHECK(allocator);
+  DCHECK(frame_decoded_out);
+  DCHECK(context());
+  DCHECK(frame());
+
+  int frame_decoded = 0;
+  int input_bytes_used = avcodec_decode_video2(
+    context().get(),
+    frame().get(),
+    &frame_decoded,
+    &packet());
+  *frame_decoded_out = frame_decoded != 0;
+  return input_bytes_used;
+}
+
+PacketPtr FfmpegVideoDecoder::CreateOutputPacket(PayloadAllocator* allocator) {
+  DCHECK(allocator);
+  DCHECK(frame());
+
+  // End of stream is indicated when we're draining and produce no packet.
+  // TODO(dalesat): This is just a copy of the audio version.
+  return Packet::Create(
+      frame()->pts,
+      frame()->pkt_duration,
+      false,
+      packet_size_,
+      frame()->data[0],
+      allocator);
+}
+
+PacketPtr FfmpegVideoDecoder::CreateOutputEndOfStreamPacket() {
+  // TODO(dalesat): Presentation time for this packet.
+  return Packet::CreateEndOfStream(0);
+}
+
+int FfmpegVideoDecoder::AllocateBufferForAvFrame(
+    AVCodecContext* av_codec_context,
+    AVFrame* av_frame,
+    int flags) {
+  // It's important to use av_codec_context here rather than context(),
+  // because av_codec_context is different for different threads when we're
+  // decoding on multiple threads. If this code is moved to an instance method,
+  // be sure to avoid using context().
+
+  // TODO(dalesat): Not sure why/if this is needed.
+  //int result = av_image_check_size(
+  //    av_codec_context->width,
+  //    av_codec_context->height,
+  //    0,
+  //    NULL);
+  //if (result < 0) {
+  //  DCHECK(false) << "av_image_check_size failed";
+  //  return result;
+  //}
+
+  // TODO(dalesat): Not sure why this is needed.
+  int coded_width =
+      std::max(av_codec_context->width, av_codec_context->coded_width);
+  int coded_height =
+      std::max(av_codec_context->height, av_codec_context->coded_height);
+  DCHECK_EQ(coded_width, av_codec_context->coded_width) <<
+      "coded width is less than width";
+  DCHECK_EQ(coded_height, av_codec_context->coded_height) <<
+      "coded height is less than height";
+
+  // TODO(dalesat): Fill in av_frame->data and av_frame->data for each plane.
+
+  av_frame->width = coded_width;
+  av_frame->height = coded_height;
+  av_frame->format = av_codec_context->pix_fmt;
+  av_frame->reordered_opaque = av_codec_context->reordered_opaque;
+
+  av_frame->buf[0] = av_buffer_create(
+      av_frame->data[0], // Because this is the first chunk in the buffer.
+      0, // TODO(dalesat): Provide this.
+      ReleaseBufferForAvFrame,
+      nullptr, // opaque
+      0); // flags
+
+  return 0;
+}
+
+void FfmpegVideoDecoder::ReleaseBufferForAvFrame(
+  void* opaque, uint8_t* buffer) {
+  // Nothing to do.
+  // TODO(dalesat): Can we get rid of this method altogether?
+}
+
+} // namespace media
+} // namespace mojo