#ifndef VAAPI_H264_ENCODER_IMPL_H_
#define VAAPI_H264_ENCODER_IMPL_H_
#include <memory>
#include <vector>
#include "absl/container/inlined_vector.h"
#include "api/transport/rtp/dependency_descriptor.h"
#include "api/video/i420_buffer.h"
#include "api/video/video_codec_constants.h"
#include "api/video_codecs/scalability_mode.h"
#include "api/video_codecs/video_encoder.h"
#include "common_video/h264/h264_bitstream_parser.h"
#include "modules/video_coding/codecs/h264/include/h264.h"
#include "modules/video_coding/svc/scalable_video_controller.h"
#include "modules/video_coding/utility/quality_scaler.h"
#include "vaapi_h264_encoder_wrapper.h"
namespace webrtc {
class VAAPIH264EncoderWrapper : public VideoEncoder {
public:
struct LayerConfig {
int simulcast_idx = 0;
int width = -1;
int height = -1;
bool sending = true;
bool key_frame_request = false;
float max_frame_rate = 0;
uint32_t target_bps = 0;
uint32_t max_bps = 0;
bool frame_dropping_on = false;
int key_frame_interval = 0;
int num_temporal_layers = 1;
void SetStreamState(bool send_stream);
};
public:
VAAPIH264EncoderWrapper(const webrtc::Environment& env,
const SdpVideoFormat& format);
~VAAPIH264EncoderWrapper() override;
int32_t InitEncode(const VideoCodec* codec_settings,
const Settings& settings) override;
int32_t RegisterEncodeCompleteCallback(
EncodedImageCallback* callback) override;
int32_t Release() override;
int32_t Encode(const VideoFrame& frame,
const std::vector<VideoFrameType>* frame_types) override;
void SetRates(const RateControlParameters& rc_parameters) override;
EncoderInfo GetEncoderInfo() const override;
private:
VAProfile GetVAProfile() const;
private:
const webrtc::Environment& env_;
EncodedImageCallback* encoded_image_callback_ = nullptr;
std::unique_ptr<livekit_ffi::VaapiH264EncoderWrapper> encoder_;
LayerConfig configuration_;
EncodedImage encoded_image_;
H264PacketizationMode packetization_mode_;
VideoCodec codec_;
void ReportInit();
void ReportError();
bool has_reported_init_ = false;
bool has_reported_error_ = false;
webrtc::H264BitstreamParser h264_bitstream_parser_;
const SdpVideoFormat format_;
H264Profile profile_ = H264Profile::kProfileConstrainedBaseline;
H264Level level_ = H264Level::kLevel1_b;
};
}
#endif