Motown: ffmpeg implementations of framework 'parts'

services/media/framework_ffmpeg/ffmpeg_decoder_base.h

+// 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.
+
+#ifndef SERVICES_MEDIA_FRAMEWORK_FFMPEG_FFMPEG_DECODER_BASE_H_
+#define SERVICES_MEDIA_FRAMEWORK_FFMPEG_FFMPEG_DECODER_BASE_H_
+
+#include "services/media/framework/parts/decoder.h"
+#include "services/media/framework_ffmpeg/ffmpeg_type_converters.h"
+extern "C" {
+#include "third_party/ffmpeg/libavcodec/avcodec.h"
+}
+
+namespace mojo {
+namespace media {
+
+// Abstract base class for ffmpeg-based decoders.
+class FfmpegDecoderBase : public Decoder {
+ public:
+  FfmpegDecoderBase(AvCodecContextPtr av_codec_context);
+
+  ~FfmpegDecoderBase() override;
+
+  // Decoder implementation.
+  std::unique_ptr<StreamType> output_stream_type() override;
+
+  // Transform implementation.
+  void Flush() override;
+
+  bool TransformPacket(
+      const PacketPtr& input,
+      bool new_input,
+      PayloadAllocator* allocator,
+      PacketPtr* output) override;
+
+ protected:
+  struct AVFrameDeleter {
+    inline void operator()(AVFrame* ptr) const {
+      av_frame_free(&ptr);
+    }
+  };
+
+  // Decodes from av_packet_ into av_frame_. The result indicates how many
+  // bytes were consumed from av_packet_. *frame_decoded_out indicates whether
+  // av_frame_ contains a complete frame.
+  virtual int Decode(PayloadAllocator* allocator, bool* frame_decoded_out) = 0;
+
+  // Creates a Packet from av_frame_.
+  virtual PacketPtr CreateOutputPacket(PayloadAllocator* allocator) = 0;
+
+  // Creates an end-of-stream packet with no payload.
+  virtual PacketPtr CreateOutputEndOfStreamPacket() = 0;
+
+ protected:
+  // The ffmpeg codec context.
+  const AvCodecContextPtr& context() {
+    return av_codec_context_;
+  }
+
+  // Ffmpeg's representation of the input packet.
+  const AVPacket& packet() {
+    return av_packet_;
+  }
+
+  // Ffmpeg's representation of the output packet.
+  const std::unique_ptr<AVFrame, AVFrameDeleter>& frame() {
+    return av_frame_;
+  }
+
+ private:
+  // Prepares to process a new input packet.
+  void PrepareInputPacket(const PacketPtr& input);
+
+  // Finishes up after processing of an input packet has completed, possibly
+  // producing a zero-size end-of-stream packet. Returns true to indicate that
+  // a new input packet is required.
+  bool UnprepareInputPacket(const PacketPtr& input, PacketPtr* output);
+
+  AvCodecContextPtr av_codec_context_;
+  AVPacket av_packet_;
+  std::unique_ptr<AVFrame, AVFrameDeleter> av_frame_;
+};
+
+}  // namespace media
+}  // namespace mojo
+
+#endif // SERVICES_MEDIA_FRAMEWORK_FFMPEG_FFMPEG_DECODER_BASE_H_