media/codec2/components/vpx/C2SoftVpxEnc.h - LeafOS-Project/android_frameworks_av - Gitiles

 /*
  * Copyright 2018 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.
  */

 #ifndef ANDROID_C2_SOFT_VPX_ENC_H__
 #define ANDROID_C2_SOFT_VPX_ENC_H__

 #include <media/stagefright/foundation/MediaDefs.h>

 #include <C2PlatformSupport.h>
 #include <Codec2BufferUtils.h>
 #include <SimpleC2Component.h>
 #include <SimpleC2Interface.h>
 #include <util/C2InterfaceHelper.h>

 #include "vpx/vpx_encoder.h"
 #include "vpx/vpx_codec.h"
 #include "vpx/vpx_image.h"
 #include "vpx/vp8cx.h"

 namespace android {

 // TODO: These defs taken from deprecated OMX_VideoExt.h. Move these definitions
 // to a new header file and include it.

 /** Maximum number of temporal layers */
 #define MAXTEMPORALLAYERS 3

 /** temporal layer patterns */
 typedef enum TemporalPatternType {
     VPXTemporalLayerPatternNone = 0,
     VPXTemporalLayerPatternWebRTC = 1,
     VPXTemporalLayerPatternMax = 0x7FFFFFFF
 } TemporalPatternType;

 // Base class for a VPX Encoder Component
 //
 // Only following encoder settings are available (codec specific settings might
 // be available in the sub-classes):
 //    - video resolution
 //    - target bitrate
 //    - rate control (constant / variable)
 //    - frame rate
 //    - error resilience
 //    - reconstruction & loop filters (g_profile)
 //
 // Only following color formats are recognized
 //    - C2PlanarLayout::TYPE_RGB
 //    - C2PlanarLayout::TYPE_RGBA
 //
 // Following settings are not configurable by the client
 //    - encoding deadline is realtime
 //    - multithreaded encoding utilizes a number of threads equal
 // to online cpu's available
 //    - the algorithm interface for encoder is decided by the sub-class in use
 //    - fractional bits of frame rate is discarded
 //    - timestamps are in microseconds, therefore encoder timebase is fixed
 // to 1/1000000

 struct C2SoftVpxEnc : public SimpleC2Component {
     class IntfImpl;

     C2SoftVpxEnc(const char* name, c2_node_id_t id,
                  const std::shared_ptr<IntfImpl>& intfImpl);

     // From SimpleC2Component
     c2_status_t onInit() override final;
     c2_status_t onStop() override final;
     void onReset() override final;
     void onRelease() override final;
     c2_status_t onFlush_sm() override final;

     void process(
             const std::unique_ptr<C2Work> &work,
             const std::shared_ptr<C2BlockPool> &pool) override final;
     c2_status_t drain(
             uint32_t drainMode,
             const std::shared_ptr<C2BlockPool> &pool) override final;

  protected:
      std::shared_ptr<IntfImpl> mIntf;
      virtual ~C2SoftVpxEnc();

      // Initializes vpx encoder with available settings.
      status_t initEncoder();

      // Populates mCodecInterface with codec specific settings.
      virtual void setCodecSpecificInterface() = 0;

      // Sets codec specific configuration.
      virtual void setCodecSpecificConfiguration() = 0;

      // Sets codec specific encoder controls.
      virtual vpx_codec_err_t setCodecSpecificControls() = 0;

      // Get current encode flags.
      virtual vpx_enc_frame_flags_t getEncodeFlags();

      enum TemporalReferences {
          // For 1 layer case: reference all (last, golden, and alt ref), but only
          // update last.
          kTemporalUpdateLastRefAll = 12,
          // First base layer frame for 3 temporal layers, which updates last and
          // golden with alt ref dependency.
          kTemporalUpdateLastAndGoldenRefAltRef = 11,
          // First enhancement layer with alt ref dependency.
          kTemporalUpdateGoldenRefAltRef = 10,
          // First enhancement layer with alt ref dependency.
          kTemporalUpdateGoldenWithoutDependencyRefAltRef = 9,
          // Base layer with alt ref dependency.
          kTemporalUpdateLastRefAltRef = 8,
          // Highest enhacement layer without dependency on golden with alt ref
          // dependency.
          kTemporalUpdateNoneNoRefGoldenRefAltRef = 7,
          // Second layer and last frame in cycle, for 2 layers.
          kTemporalUpdateNoneNoRefAltref = 6,
          // Highest enhancement layer.
          kTemporalUpdateNone = 5,
          // Second enhancement layer.
          kTemporalUpdateAltref = 4,
          // Second enhancement layer without dependency on previous frames in
          // the second enhancement layer.
          kTemporalUpdateAltrefWithoutDependency = 3,
          // First enhancement layer.
          kTemporalUpdateGolden = 2,
          // First enhancement layer without dependency on previous frames in
          // the first enhancement layer.
          kTemporalUpdateGoldenWithoutDependency = 1,
          // Base layer.
          kTemporalUpdateLast = 0,
      };
      enum {
          kMaxTemporalPattern = 8
      };

      // vpx specific opaque data structure that
      // stores encoder state
      vpx_codec_ctx_t* mCodecContext;

      // vpx specific data structure that
      // stores encoder configuration
      vpx_codec_enc_cfg_t* mCodecConfiguration;

      // vpx specific read-only data structure
      // that specifies algorithm interface (e.g. vp8)
      vpx_codec_iface_t* mCodecInterface;

      // align stride to the power of 2
      int32_t mStrideAlign;

      // Color format for the input port
      vpx_img_fmt_t mColorFormat;

      // Bitrate control mode, either constant or variable
      vpx_rc_mode mBitrateControlMode;

      // Parameter that denotes whether error resilience
      // is enabled in encoder
      bool mErrorResilience;

      // Minimum (best quality) quantizer
      uint32_t mMinQuantizer;

      // Maximum (worst quality) quantizer
      uint32_t mMaxQuantizer;

      // Number of coding temporal layers to be used.
      size_t mTemporalLayers;

      // Temporal layer bitrare ratio in percentage
      float_t mTemporalLayerBitrateRatio[MAXTEMPORALLAYERS];

      // Temporal pattern type
      TemporalPatternType mTemporalPatternType;

      // Temporal pattern length
      size_t mTemporalPatternLength;

      // Temporal pattern current index
      size_t mTemporalPatternIdx;

      // Frame type temporal pattern
      TemporalReferences mTemporalPattern[kMaxTemporalPattern];

      // Last input buffer timestamp
      uint64_t mLastTimestamp;

      // Number of input frames
      int64_t mNumInputFrames;

      // Conversion buffer is needed to input to
      // yuv420 planar format.
      MemoryBlock mConversionBuffer;

      // Signalled EOS
      bool mSignalledOutputEos;

      // Signalled Error
      bool mSignalledError;

     // configurations used by component in process
     // (TODO: keep this in intf but make them internal only)
     std::shared_ptr<C2StreamPictureSizeInfo::input> mSize;
     std::shared_ptr<C2StreamIntraRefreshTuning::output> mIntraRefresh;
     std::shared_ptr<C2StreamFrameRateInfo::output> mFrameRate;
     std::shared_ptr<C2StreamBitrateInfo::output> mBitrate;
     std::shared_ptr<C2StreamBitrateModeTuning::output> mBitrateMode;
     std::shared_ptr<C2StreamRequestSyncFrameTuning::output> mRequestSync;
     std::shared_ptr<C2StreamTemporalLayeringTuning::output> mLayering;
     std::shared_ptr<C2StreamPictureQuantizationTuning::output> mQpBounds;

      C2_DO_NOT_COPY(C2SoftVpxEnc);
 };

 namespace {

 #ifdef VP9
 constexpr char COMPONENT_NAME[] = "c2.android.vp9.encoder";
 const char *MEDIA_MIMETYPE_VIDEO = MEDIA_MIMETYPE_VIDEO_VP9;
 #else
 constexpr char COMPONENT_NAME[] = "c2.android.vp8.encoder";
 const char *MEDIA_MIMETYPE_VIDEO = MEDIA_MIMETYPE_VIDEO_VP8;
 #endif

 } // namepsace

 class C2SoftVpxEnc::IntfImpl : public SimpleInterface<void>::BaseParams {
    public:
     explicit IntfImpl(const std::shared_ptr<C2ReflectorHelper> &helper);
     static C2R BitrateSetter(bool mayBlock, C2P<C2StreamBitrateInfo::output> &me);

     static C2R SizeSetter(bool mayBlock, const C2P<C2StreamPictureSizeInfo::input> &oldMe,
                           C2P<C2StreamPictureSizeInfo::input> &me);

     static C2R ProfileLevelSetter(bool mayBlock, C2P<C2StreamProfileLevelInfo::output>& me,
                                   const C2P<C2StreamPictureSizeInfo::input>& size,
                                   const C2P<C2StreamFrameRateInfo::output>& frameRate,
                                   const C2P<C2StreamBitrateInfo::output>& bitrate);

     static C2R LayeringSetter(bool mayBlock, C2P<C2StreamTemporalLayeringTuning::output>& me);

     static C2R PictureQuantizationSetter(bool mayBlock,
                                          C2P<C2StreamPictureQuantizationTuning::output> &me);

     // unsafe getters
     std::shared_ptr<C2StreamPictureSizeInfo::input> getSize_l() const { return mSize; }
     std::shared_ptr<C2StreamIntraRefreshTuning::output> getIntraRefresh_l() const {
         return mIntraRefresh;
     }
     std::shared_ptr<C2StreamFrameRateInfo::output> getFrameRate_l() const { return mFrameRate; }
     std::shared_ptr<C2StreamBitrateInfo::output> getBitrate_l() const { return mBitrate; }
     std::shared_ptr<C2StreamBitrateModeTuning::output> getBitrateMode_l() const {
         return mBitrateMode;
     }
     std::shared_ptr<C2StreamRequestSyncFrameTuning::output> getRequestSync_l() const {
         return mRequestSync;
     }
     std::shared_ptr<C2StreamTemporalLayeringTuning::output> getTemporalLayers_l() const {
         return mLayering;
     }
     std::shared_ptr<C2StreamColorAspectsInfo::output> getCodedColorAspects_l() const {
         return mCodedColorAspects;
     }
     std::shared_ptr<C2StreamPictureQuantizationTuning::output> getPictureQuantization_l() const {
         return mPictureQuantization;
     }
     uint32_t getSyncFramePeriod() const;
     static C2R ColorAspectsSetter(bool mayBlock, C2P<C2StreamColorAspectsInfo::input> &me);
     static C2R CodedColorAspectsSetter(bool mayBlock, C2P<C2StreamColorAspectsInfo::output> &me,
                                        const C2P<C2StreamColorAspectsInfo::input> &coded);

    private:
     std::shared_ptr<C2StreamUsageTuning::input> mUsage;
     std::shared_ptr<C2StreamPictureSizeInfo::input> mSize;
     std::shared_ptr<C2StreamFrameRateInfo::output> mFrameRate;
     std::shared_ptr<C2StreamTemporalLayeringTuning::output> mLayering;
     std::shared_ptr<C2StreamIntraRefreshTuning::output> mIntraRefresh;
     std::shared_ptr<C2StreamRequestSyncFrameTuning::output> mRequestSync;
     std::shared_ptr<C2StreamSyncFrameIntervalTuning::output> mSyncFramePeriod;
     std::shared_ptr<C2StreamBitrateInfo::output> mBitrate;
     std::shared_ptr<C2StreamBitrateModeTuning::output> mBitrateMode;
     std::shared_ptr<C2StreamProfileLevelInfo::output> mProfileLevel;
     std::shared_ptr<C2StreamColorAspectsInfo::input> mColorAspects;
     std::shared_ptr<C2StreamColorAspectsInfo::output> mCodedColorAspects;
     std::shared_ptr<C2StreamPictureQuantizationTuning::output> mPictureQuantization;
 };

 }  // namespace android

 #endif  // ANDROID_C2_SOFT_VPX_ENC_H__
	/*
	* Copyright 2018 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.
	*/

	#ifndef ANDROID_C2_SOFT_VPX_ENC_H__
	#define ANDROID_C2_SOFT_VPX_ENC_H__

	#include <media/stagefright/foundation/MediaDefs.h>

	#include <C2PlatformSupport.h>
	#include <Codec2BufferUtils.h>
	#include <SimpleC2Component.h>
	#include <SimpleC2Interface.h>
	#include <util/C2InterfaceHelper.h>

	#include "vpx/vpx_encoder.h"
	#include "vpx/vpx_codec.h"
	#include "vpx/vpx_image.h"
	#include "vpx/vp8cx.h"

	namespace android {

	// TODO: These defs taken from deprecated OMX_VideoExt.h. Move these definitions
	// to a new header file and include it.

	/** Maximum number of temporal layers */
	#define MAXTEMPORALLAYERS 3

	/** temporal layer patterns */
	typedef enum TemporalPatternType {
	VPXTemporalLayerPatternNone = 0,
	VPXTemporalLayerPatternWebRTC = 1,
	VPXTemporalLayerPatternMax = 0x7FFFFFFF
	} TemporalPatternType;

	// Base class for a VPX Encoder Component
	//
	// Only following encoder settings are available (codec specific settings might
	// be available in the sub-classes):
	// - video resolution
	// - target bitrate
	// - rate control (constant / variable)
	// - frame rate
	// - error resilience
	// - reconstruction & loop filters (g_profile)
	//
	// Only following color formats are recognized
	// - C2PlanarLayout::TYPE_RGB
	// - C2PlanarLayout::TYPE_RGBA
	//
	// Following settings are not configurable by the client
	// - encoding deadline is realtime
	// - multithreaded encoding utilizes a number of threads equal
	// to online cpu's available
	// - the algorithm interface for encoder is decided by the sub-class in use
	// - fractional bits of frame rate is discarded
	// - timestamps are in microseconds, therefore encoder timebase is fixed
	// to 1/1000000

	struct C2SoftVpxEnc : public SimpleC2Component {
	class IntfImpl;

	C2SoftVpxEnc(const char* name, c2_node_id_t id,
	const std::shared_ptr<IntfImpl>& intfImpl);

	// From SimpleC2Component
	c2_status_t onInit() override final;
	c2_status_t onStop() override final;
	void onReset() override final;
	void onRelease() override final;
	c2_status_t onFlush_sm() override final;

	void process(
	const std::unique_ptr<C2Work> &work,
	const std::shared_ptr<C2BlockPool> &pool) override final;
	c2_status_t drain(
	uint32_t drainMode,
	const std::shared_ptr<C2BlockPool> &pool) override final;

	protected:
	std::shared_ptr<IntfImpl> mIntf;
	virtual ~C2SoftVpxEnc();

	// Initializes vpx encoder with available settings.
	status_t initEncoder();

	// Populates mCodecInterface with codec specific settings.
	virtual void setCodecSpecificInterface() = 0;

	// Sets codec specific configuration.
	virtual void setCodecSpecificConfiguration() = 0;

	// Sets codec specific encoder controls.
	virtual vpx_codec_err_t setCodecSpecificControls() = 0;

	// Get current encode flags.
	virtual vpx_enc_frame_flags_t getEncodeFlags();

	enum TemporalReferences {
	// For 1 layer case: reference all (last, golden, and alt ref), but only
	// update last.
	kTemporalUpdateLastRefAll = 12,
	// First base layer frame for 3 temporal layers, which updates last and
	// golden with alt ref dependency.
	kTemporalUpdateLastAndGoldenRefAltRef = 11,
	// First enhancement layer with alt ref dependency.
	kTemporalUpdateGoldenRefAltRef = 10,
	// First enhancement layer with alt ref dependency.
	kTemporalUpdateGoldenWithoutDependencyRefAltRef = 9,
	// Base layer with alt ref dependency.
	kTemporalUpdateLastRefAltRef = 8,
	// Highest enhacement layer without dependency on golden with alt ref
	// dependency.
	kTemporalUpdateNoneNoRefGoldenRefAltRef = 7,
	// Second layer and last frame in cycle, for 2 layers.
	kTemporalUpdateNoneNoRefAltref = 6,
	// Highest enhancement layer.
	kTemporalUpdateNone = 5,
	// Second enhancement layer.
	kTemporalUpdateAltref = 4,
	// Second enhancement layer without dependency on previous frames in
	// the second enhancement layer.
	kTemporalUpdateAltrefWithoutDependency = 3,
	// First enhancement layer.
	kTemporalUpdateGolden = 2,
	// First enhancement layer without dependency on previous frames in
	// the first enhancement layer.
	kTemporalUpdateGoldenWithoutDependency = 1,
	// Base layer.
	kTemporalUpdateLast = 0,
	};
	enum {
	kMaxTemporalPattern = 8
	};

	// vpx specific opaque data structure that
	// stores encoder state
	vpx_codec_ctx_t* mCodecContext;

	// vpx specific data structure that
	// stores encoder configuration
	vpx_codec_enc_cfg_t* mCodecConfiguration;

	// vpx specific read-only data structure
	// that specifies algorithm interface (e.g. vp8)
	vpx_codec_iface_t* mCodecInterface;

	// align stride to the power of 2
	int32_t mStrideAlign;

	// Color format for the input port
	vpx_img_fmt_t mColorFormat;

	// Bitrate control mode, either constant or variable
	vpx_rc_mode mBitrateControlMode;

	// Parameter that denotes whether error resilience
	// is enabled in encoder
	bool mErrorResilience;

	// Minimum (best quality) quantizer
	uint32_t mMinQuantizer;

	// Maximum (worst quality) quantizer
	uint32_t mMaxQuantizer;

	// Number of coding temporal layers to be used.
	size_t mTemporalLayers;

	// Temporal layer bitrare ratio in percentage
	float_t mTemporalLayerBitrateRatio[MAXTEMPORALLAYERS];

	// Temporal pattern type
	TemporalPatternType mTemporalPatternType;

	// Temporal pattern length
	size_t mTemporalPatternLength;

	// Temporal pattern current index
	size_t mTemporalPatternIdx;

	// Frame type temporal pattern
	TemporalReferences mTemporalPattern[kMaxTemporalPattern];

	// Last input buffer timestamp
	uint64_t mLastTimestamp;

	// Number of input frames
	int64_t mNumInputFrames;

	// Conversion buffer is needed to input to
	// yuv420 planar format.
	MemoryBlock mConversionBuffer;

	// Signalled EOS
	bool mSignalledOutputEos;

	// Signalled Error
	bool mSignalledError;

	// configurations used by component in process
	// (TODO: keep this in intf but make them internal only)
	std::shared_ptr<C2StreamPictureSizeInfo::input> mSize;
	std::shared_ptr<C2StreamIntraRefreshTuning::output> mIntraRefresh;
	std::shared_ptr<C2StreamFrameRateInfo::output> mFrameRate;
	std::shared_ptr<C2StreamBitrateInfo::output> mBitrate;
	std::shared_ptr<C2StreamBitrateModeTuning::output> mBitrateMode;
	std::shared_ptr<C2StreamRequestSyncFrameTuning::output> mRequestSync;
	std::shared_ptr<C2StreamTemporalLayeringTuning::output> mLayering;
	std::shared_ptr<C2StreamPictureQuantizationTuning::output> mQpBounds;

	C2_DO_NOT_COPY(C2SoftVpxEnc);
	};

	namespace {

	#ifdef VP9
	constexpr char COMPONENT_NAME[] = "c2.android.vp9.encoder";
	const char *MEDIA_MIMETYPE_VIDEO = MEDIA_MIMETYPE_VIDEO_VP9;
	#else
	constexpr char COMPONENT_NAME[] = "c2.android.vp8.encoder";
	const char *MEDIA_MIMETYPE_VIDEO = MEDIA_MIMETYPE_VIDEO_VP8;
	#endif

	} // namepsace

	class C2SoftVpxEnc::IntfImpl : public SimpleInterface<void>::BaseParams {
	public:
	explicit IntfImpl(const std::shared_ptr<C2ReflectorHelper> &helper);
	static C2R BitrateSetter(bool mayBlock, C2P<C2StreamBitrateInfo::output> &me);

	static C2R SizeSetter(bool mayBlock, const C2P<C2StreamPictureSizeInfo::input> &oldMe,
	C2P<C2StreamPictureSizeInfo::input> &me);

	static C2R ProfileLevelSetter(bool mayBlock, C2P<C2StreamProfileLevelInfo::output>& me,
	const C2P<C2StreamPictureSizeInfo::input>& size,
	const C2P<C2StreamFrameRateInfo::output>& frameRate,
	const C2P<C2StreamBitrateInfo::output>& bitrate);

	static C2R LayeringSetter(bool mayBlock, C2P<C2StreamTemporalLayeringTuning::output>& me);

	static C2R PictureQuantizationSetter(bool mayBlock,
	C2P<C2StreamPictureQuantizationTuning::output> &me);

	// unsafe getters
	std::shared_ptr<C2StreamPictureSizeInfo::input> getSize_l() const { return mSize; }
	std::shared_ptr<C2StreamIntraRefreshTuning::output> getIntraRefresh_l() const {
	return mIntraRefresh;
	}
	std::shared_ptr<C2StreamFrameRateInfo::output> getFrameRate_l() const { return mFrameRate; }
	std::shared_ptr<C2StreamBitrateInfo::output> getBitrate_l() const { return mBitrate; }
	std::shared_ptr<C2StreamBitrateModeTuning::output> getBitrateMode_l() const {
	return mBitrateMode;
	}
	std::shared_ptr<C2StreamRequestSyncFrameTuning::output> getRequestSync_l() const {
	return mRequestSync;
	}
	std::shared_ptr<C2StreamTemporalLayeringTuning::output> getTemporalLayers_l() const {
	return mLayering;
	}
	std::shared_ptr<C2StreamColorAspectsInfo::output> getCodedColorAspects_l() const {
	return mCodedColorAspects;
	}
	std::shared_ptr<C2StreamPictureQuantizationTuning::output> getPictureQuantization_l() const {
	return mPictureQuantization;
	}
	uint32_t getSyncFramePeriod() const;
	static C2R ColorAspectsSetter(bool mayBlock, C2P<C2StreamColorAspectsInfo::input> &me);
	static C2R CodedColorAspectsSetter(bool mayBlock, C2P<C2StreamColorAspectsInfo::output> &me,
	const C2P<C2StreamColorAspectsInfo::input> &coded);

	private:
	std::shared_ptr<C2StreamUsageTuning::input> mUsage;
	std::shared_ptr<C2StreamPictureSizeInfo::input> mSize;
	std::shared_ptr<C2StreamFrameRateInfo::output> mFrameRate;
	std::shared_ptr<C2StreamTemporalLayeringTuning::output> mLayering;
	std::shared_ptr<C2StreamIntraRefreshTuning::output> mIntraRefresh;
	std::shared_ptr<C2StreamRequestSyncFrameTuning::output> mRequestSync;
	std::shared_ptr<C2StreamSyncFrameIntervalTuning::output> mSyncFramePeriod;
	std::shared_ptr<C2StreamBitrateInfo::output> mBitrate;
	std::shared_ptr<C2StreamBitrateModeTuning::output> mBitrateMode;
	std::shared_ptr<C2StreamProfileLevelInfo::output> mProfileLevel;
	std::shared_ptr<C2StreamColorAspectsInfo::input> mColorAspects;
	std::shared_ptr<C2StreamColorAspectsInfo::output> mCodedColorAspects;
	std::shared_ptr<C2StreamPictureQuantizationTuning::output> mPictureQuantization;
	};

	} // namespace android

	#endif // ANDROID_C2_SOFT_VPX_ENC_H__