197 lines
7.1 KiB
C++
197 lines
7.1 KiB
C++
// Copyright (c) 2021 Michael Fabian Dirks <info@xaymar.com>
|
|
//
|
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
// of this software and associated documentation files (the "Software"), to deal
|
|
// in the Software without restriction, including without limitation the rights
|
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
// copies of the Software, and to permit persons to whom the Software is
|
|
// furnished to do so, subject to the following conditions:
|
|
//
|
|
// The above copyright notice and this permission notice shall be included in all
|
|
// copies or substantial portions of the Software.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
// SOFTWARE.
|
|
|
|
#pragma once
|
|
#include "common.hpp"
|
|
#include <memory>
|
|
#include <queue>
|
|
#include "encoders/codecs/av1.hpp"
|
|
#include "obs/obs-encoder-factory.hpp"
|
|
#include "util/util-library.hpp"
|
|
#include "util/util-profiler.hpp"
|
|
|
|
#include <aom/aomcx.h>
|
|
|
|
namespace streamfx::encoder::aom::av1 {
|
|
class aom_av1_factory;
|
|
|
|
class aom_av1_instance : public obs::encoder_instance {
|
|
std::shared_ptr<aom_av1_factory> _factory;
|
|
|
|
aom_codec_iface_t* _iface;
|
|
aom_codec_ctx_t _ctx;
|
|
aom_codec_enc_cfg_t _cfg;
|
|
size_t _image_index;
|
|
std::vector<aom_image_t> _images;
|
|
aom_fixed_buf_t* _global_headers;
|
|
|
|
bool _initialized;
|
|
struct {
|
|
// Video (All Static)
|
|
uint16_t width;
|
|
uint16_t height;
|
|
struct {
|
|
uint32_t num;
|
|
uint32_t den;
|
|
} fps;
|
|
|
|
// Color (All Static)
|
|
aom_img_fmt color_format;
|
|
aom_color_primaries_t color_primaries;
|
|
aom_transfer_characteristics_t color_trc;
|
|
aom_matrix_coefficients_t color_matrix;
|
|
aom_color_range_t color_range;
|
|
bool monochrome;
|
|
|
|
// Encoder
|
|
codec::av1::profile profile; // Static
|
|
int8_t preset;
|
|
|
|
// Rate Control
|
|
aom_rc_mode rc_mode; // Static
|
|
int8_t rc_lookahead; // Static
|
|
int32_t rc_bitrate;
|
|
int32_t rc_bitrate_overshoot;
|
|
int32_t rc_bitrate_undershoot;
|
|
int8_t rc_quality;
|
|
int8_t rc_quantizer_min;
|
|
int8_t rc_quantizer_max;
|
|
int32_t rc_buffer_ms;
|
|
int32_t rc_buffer_initial_ms;
|
|
int32_t rc_buffer_optimal_ms;
|
|
|
|
// Key-Frames
|
|
aom_kf_mode kf_mode;
|
|
int32_t kf_distance_min;
|
|
int32_t kf_distance_max;
|
|
|
|
// Threads and Tiling (All Static)
|
|
int8_t threads;
|
|
int8_t rowmultithreading;
|
|
int8_t tile_columns;
|
|
int8_t tile_rows;
|
|
aom_tune_metric tune_metric;
|
|
aom_tune_content tune_content;
|
|
} _settings;
|
|
|
|
#ifdef ENABLE_PROFILING
|
|
std::shared_ptr<streamfx::util::profiler> _profiler_copy;
|
|
std::shared_ptr<streamfx::util::profiler> _profiler_encode;
|
|
std::shared_ptr<streamfx::util::profiler> _profiler_packet;
|
|
#endif
|
|
|
|
public:
|
|
aom_av1_instance(obs_data_t* settings, obs_encoder_t* self, bool is_hw);
|
|
virtual ~aom_av1_instance();
|
|
|
|
virtual void migrate(obs_data_t* settings, uint64_t version);
|
|
|
|
virtual bool update(obs_data_t* settings);
|
|
|
|
void log();
|
|
|
|
virtual bool get_extra_data(uint8_t** extra_data, size_t* size);
|
|
|
|
virtual bool get_sei_data(uint8_t** sei_data, size_t* size);
|
|
|
|
virtual void get_video_info(struct video_scale_info* info);
|
|
|
|
virtual bool encode_video(encoder_frame* frame, encoder_packet* packet, bool* received_packet);
|
|
};
|
|
|
|
class aom_av1_factory : public obs::encoder_factory<aom_av1_factory, aom_av1_instance> {
|
|
std::shared_ptr<::streamfx::util::library> _library;
|
|
|
|
public:
|
|
aom_av1_factory();
|
|
~aom_av1_factory();
|
|
|
|
const char* get_name() override;
|
|
|
|
void* create(obs_data_t* settings, obs_encoder_t* encoder, bool is_hw) override;
|
|
|
|
void get_defaults2(obs_data_t* data) override;
|
|
|
|
obs_properties_t* get_properties2(instance_t* data) override;
|
|
|
|
#ifdef ENABLE_FRONTEND
|
|
static bool on_manual_open(obs_properties_t* props, obs_property_t* property, void* data);
|
|
#endif
|
|
|
|
public:
|
|
// aom_codec.h
|
|
decltype(&aom_codec_version) libaom_codec_version;
|
|
decltype(&aom_codec_version_str) libaom_codec_version_str;
|
|
decltype(&aom_codec_version_extra_str) libaom_codec_version_extra_str;
|
|
decltype(&aom_codec_build_config) libaom_codec_build_config;
|
|
decltype(&aom_codec_iface_name) libaom_codec_iface_name;
|
|
decltype(&aom_codec_err_to_string) libaom_codec_err_to_string;
|
|
decltype(&aom_codec_error) libaom_codec_error;
|
|
decltype(&aom_codec_error_detail) libaom_codec_error_detail;
|
|
decltype(&aom_codec_destroy) libaom_codec_destroy;
|
|
decltype(&aom_codec_get_caps) libaom_codec_get_caps;
|
|
decltype(&aom_codec_control) libaom_codec_control;
|
|
decltype(&aom_codec_set_option) libaom_codec_set_option;
|
|
decltype(&aom_obu_type_to_string) libaom_obu_type_to_string;
|
|
|
|
// aom_integer.h
|
|
decltype(&aom_uleb_size_in_bytes) libaom_uleb_size_in_bytes;
|
|
decltype(&aom_uleb_decode) libaom_uleb_decode;
|
|
decltype(&aom_uleb_encode) libaom_uleb_encode;
|
|
decltype(&aom_uleb_encode_fixed_size) libaom_uleb_encode_fixed_size;
|
|
|
|
// aom_image.h
|
|
decltype(&aom_img_alloc) libaom_img_alloc;
|
|
decltype(&aom_img_wrap) libaom_img_wrap;
|
|
decltype(&aom_img_alloc_with_border) libaom_img_alloc_with_border;
|
|
decltype(&aom_img_set_rect) libaom_img_set_rect;
|
|
decltype(&aom_img_flip) libaom_img_flip;
|
|
decltype(&aom_img_free) libaom_img_free;
|
|
decltype(&aom_img_plane_width) libaom_img_plane_width;
|
|
decltype(&aom_img_plane_height) libaom_img_plane_height;
|
|
decltype(&aom_img_add_metadata) libaom_img_add_metadata;
|
|
decltype(&aom_img_get_metadata) libaom_img_get_metadata;
|
|
decltype(&aom_img_num_metadata) libaom_img_num_metadata;
|
|
decltype(&aom_img_remove_metadata) libaom_img_remove_metadata;
|
|
decltype(&aom_img_metadata_alloc) libaom_img_metadata_alloc;
|
|
decltype(&aom_img_metadata_free) libaom_img_metadata_free;
|
|
|
|
// aom_encoder.h
|
|
decltype(&aom_codec_enc_init_ver) libaom_codec_enc_init_ver;
|
|
decltype(&aom_codec_enc_config_default) libaom_codec_enc_config_default;
|
|
decltype(&aom_codec_enc_config_set) libaom_codec_enc_config_set;
|
|
decltype(&aom_codec_get_global_headers) libaom_codec_get_global_headers;
|
|
decltype(&aom_codec_encode) libaom_codec_encode;
|
|
decltype(&aom_codec_set_cx_data_buf) libaom_codec_set_cx_data_buf;
|
|
decltype(&aom_codec_get_cx_data) libaom_codec_get_cx_data;
|
|
decltype(&aom_codec_get_preview_frame) libaom_codec_get_preview_frame;
|
|
|
|
// aomcx.h
|
|
decltype(&aom_codec_av1_cx) libaom_codec_av1_cx;
|
|
|
|
public: // Singleton
|
|
static void initialize();
|
|
|
|
static void finalize();
|
|
|
|
static std::shared_ptr<aom_av1_factory> get();
|
|
};
|
|
} // namespace streamfx::encoder::aom::av1
|