| // Copyright 2018 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 MEDIA_GPU_VAAPI_VP8_VAAPI_VIDEO_ENCODER_DELEGATE_H_ |
| #define MEDIA_GPU_VAAPI_VP8_VAAPI_VIDEO_ENCODER_DELEGATE_H_ |
| |
| #include <list> |
| #include <vector> |
| |
| #include "base/macros.h" |
| #include "media/base/video_bitrate_allocation.h" |
| #include "media/gpu/vaapi/vaapi_video_encoder_delegate.h" |
| #include "media/gpu/vp8_picture.h" |
| #include "media/gpu/vp8_reference_frame_vector.h" |
| #include "media/parsers/vp8_parser.h" |
| |
| namespace media { |
| class VaapiWrapper; |
| |
| class VP8VaapiVideoEncoderDelegate : public VaapiVideoEncoderDelegate { |
| public: |
| struct EncodeParams { |
| EncodeParams(); |
| |
| // Produce a keyframe at least once per this many frames. |
| size_t kf_period_frames; |
| |
| // Bitrate allocation in bps. |
| VideoBitrateAllocation bitrate_allocation; |
| |
| // Framerate in FPS. |
| uint32_t framerate; |
| |
| // Bitrate window size in ms. |
| unsigned int cpb_window_size_ms; |
| |
| // Coded picture buffer size in bits. |
| unsigned int cpb_size_bits; |
| |
| // Quantization parameter. They are vp8 ac/dc indices and their ranges are |
| // 0-127. |
| uint8_t initial_qp; |
| uint8_t min_qp; |
| uint8_t max_qp; |
| |
| bool error_resilient_mode; |
| }; |
| |
| VP8VaapiVideoEncoderDelegate(scoped_refptr<VaapiWrapper> vaapi_wrapper, |
| base::RepeatingClosure error_cb); |
| |
| VP8VaapiVideoEncoderDelegate(const VP8VaapiVideoEncoderDelegate&) = delete; |
| VP8VaapiVideoEncoderDelegate& operator=(const VP8VaapiVideoEncoderDelegate&) = |
| delete; |
| |
| ~VP8VaapiVideoEncoderDelegate() override; |
| |
| // VaapiVideoEncoderDelegate implementation. |
| bool Initialize(const VideoEncodeAccelerator::Config& config, |
| const VaapiVideoEncoderDelegate::Config& ave_config) override; |
| bool UpdateRates(const VideoBitrateAllocation& bitrate_allocation, |
| uint32_t framerate) override; |
| gfx::Size GetCodedSize() const override; |
| size_t GetMaxNumOfRefFrames() const override; |
| std::vector<gfx::Size> GetSVCLayerResolutions() override; |
| bool PrepareEncodeJob(EncodeJob* encode_job) override; |
| |
| private: |
| void InitializeFrameHeader(); |
| void UpdateFrameHeader(bool keyframe); |
| void UpdateReferenceFrames(scoped_refptr<VP8Picture> picture); |
| void Reset(); |
| |
| scoped_refptr<VP8Picture> GetPicture(EncodeJob* job); |
| |
| bool SubmitFrameParameters( |
| EncodeJob* job, |
| const EncodeParams& encode_params, |
| scoped_refptr<VP8Picture> pic, |
| const Vp8ReferenceFrameVector& ref_frames, |
| const std::array<bool, kNumVp8ReferenceBuffers>& ref_frames_used); |
| |
| gfx::Size visible_size_; |
| gfx::Size coded_size_; // Macroblock-aligned. |
| |
| // Frame count since last keyframe, reset to 0 every keyframe period. |
| size_t frame_num_ = 0; |
| |
| EncodeParams current_params_; |
| |
| Vp8FrameHeader current_frame_hdr_; |
| Vp8ReferenceFrameVector reference_frames_; |
| }; |
| |
| } // namespace media |
| |
| #endif // MEDIA_GPU_VAAPI_VP8_VAAPI_VIDEO_ENCODER_DELEGATE_H_ |