Michael Fabian 'Xaymar' Dirks
129 lines
5.3 KiB
C++
129 lines
5.3 KiB
C++
// FFMPEG Video Encoder Integration for OBS Studio
|
|
// Copyright (c) 2019 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.
|
|
|
|
#include "prores_aw_handler.hpp"
|
|
#include "codecs/prores.hpp"
|
|
#include "plugin.hpp"
|
|
#include "utility.hpp"
|
|
|
|
extern "C" {
|
|
#include <obs-module.h>
|
|
}
|
|
|
|
INITIALIZER(prores_aw_handler_init)
|
|
{
|
|
obsffmpeg::initializers.push_back([]() {
|
|
obsffmpeg::register_codec_handler("prores_aw", std::make_shared<obsffmpeg::ui::prores_aw_handler>());
|
|
});
|
|
};
|
|
|
|
void obsffmpeg::ui::prores_aw_handler::override_colorformat(AVPixelFormat& target_format, obs_data_t* settings,
|
|
const AVCodec* codec, AVCodecContext*)
|
|
{
|
|
std::string profile = "";
|
|
|
|
int profile_id = static_cast<int>(obs_data_get_int(settings, P_PRORES_PROFILE));
|
|
for (auto ptr = codec->profiles; ptr->profile != FF_PROFILE_UNKNOWN; ptr++) {
|
|
if (ptr->profile == profile_id) {
|
|
profile = ptr->name;
|
|
break;
|
|
}
|
|
}
|
|
|
|
std::unordered_map<AVPixelFormat, std::list<std::string>> valid_formats = {
|
|
{AV_PIX_FMT_YUV422P10, {"apco", "apcs", "apcn", "apch"}}, {AV_PIX_FMT_YUV444P10, {"ap4h", "ap4x"}}};
|
|
|
|
for (auto kv : valid_formats) {
|
|
for (auto name : kv.second) {
|
|
if (profile == name) {
|
|
target_format = kv.first;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void obsffmpeg::ui::prores_aw_handler::get_defaults(obs_data_t* settings, const AVCodec*, AVCodecContext*, bool)
|
|
{
|
|
obs_data_set_default_int(settings, P_PRORES_PROFILE, 0);
|
|
}
|
|
|
|
void obsffmpeg::ui::prores_aw_handler::get_properties(obs_properties_t* props, const AVCodec* codec,
|
|
AVCodecContext* context, bool)
|
|
{
|
|
if (!context) {
|
|
auto p = obs_properties_add_list(props, P_PRORES_PROFILE, TRANSLATE(P_PRORES_PROFILE),
|
|
OBS_COMBO_TYPE_LIST, OBS_COMBO_FORMAT_INT);
|
|
obs_property_set_long_description(p, TRANSLATE(DESC(P_PRORES_PROFILE)));
|
|
for (auto ptr = codec->profiles; ptr->profile != FF_PROFILE_UNKNOWN; ptr++) {
|
|
if (strcmp("apco", ptr->name) == 0) {
|
|
obs_property_list_add_int(p, TRANSLATE(P_PRORES_PROFILE_APCO),
|
|
static_cast<int64_t>(ptr->profile));
|
|
} else if (strcmp("apcs", ptr->name) == 0) {
|
|
obs_property_list_add_int(p, TRANSLATE(P_PRORES_PROFILE_APCS),
|
|
static_cast<int64_t>(ptr->profile));
|
|
} else if (strcmp("apcn", ptr->name) == 0) {
|
|
obs_property_list_add_int(p, TRANSLATE(P_PRORES_PROFILE_APCN),
|
|
static_cast<int64_t>(ptr->profile));
|
|
} else if (strcmp("apch", ptr->name) == 0) {
|
|
obs_property_list_add_int(p, TRANSLATE(P_PRORES_PROFILE_APCH),
|
|
static_cast<int64_t>(ptr->profile));
|
|
} else if (strcmp("ap4h", ptr->name) == 0) {
|
|
obs_property_list_add_int(p, TRANSLATE(P_PRORES_PROFILE_AP4H),
|
|
static_cast<int64_t>(ptr->profile));
|
|
} else if (strcmp("ap4x", ptr->name) == 0) {
|
|
obs_property_list_add_int(p, TRANSLATE(P_PRORES_PROFILE_AP4X),
|
|
static_cast<int64_t>(ptr->profile));
|
|
} else {
|
|
obs_property_list_add_int(p, ptr->name, ptr->profile);
|
|
}
|
|
}
|
|
} else {
|
|
obs_property_set_enabled(obs_properties_get(props, P_PRORES_PROFILE), false);
|
|
}
|
|
}
|
|
|
|
void obsffmpeg::ui::prores_aw_handler::update(obs_data_t* settings, const AVCodec*, AVCodecContext* context)
|
|
{
|
|
context->profile = static_cast<int>(obs_data_get_int(settings, P_PRORES_PROFILE));
|
|
}
|
|
|
|
void obsffmpeg::ui::prores_aw_handler::log_options(obs_data_t* settings, const AVCodec* codec, AVCodecContext*)
|
|
{
|
|
for (auto ptr = codec->profiles; ptr->profile != FF_PROFILE_UNKNOWN; ptr++) {
|
|
if (ptr->profile == static_cast<int>(obs_data_get_int(settings, P_PRORES_PROFILE)))
|
|
PLOG_INFO("[%s] Profile: %s", codec->name, ptr->name);
|
|
}
|
|
}
|
|
|
|
void obsffmpeg::ui::prores_aw_handler::process_avpacket(AVPacket& packet, const AVCodec*, AVCodecContext*)
|
|
{
|
|
//FFmpeg Bug:
|
|
// When ProRes content is stored in Matroska, FFmpeg strips the size
|
|
// from the atom. Later when the ProRes content is demuxed from Matroska,
|
|
// FFmpeg creates an atom with the incorrect size, as the ATOM size
|
|
// should be content + atom, but FFmpeg set it to only be content. This
|
|
// difference leads to decoders to be off by 8 bytes.
|
|
//Fix (until FFmpeg stops being broken):
|
|
// Pad the packet with 8 bytes of 0x00.
|
|
|
|
av_grow_packet(&packet, 8);
|
|
}
|