obs-StreamFX/source/encoders/handlers/prores_aw_handler.cpp

134 lines
5.2 KiB
C++
Raw Normal View History

2020-01-13 00:52:30 +00:00
// 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 <array>
2020-01-13 21:40:15 +00:00
#include "../codecs/prores.hpp"
2020-01-13 00:52:30 +00:00
#include "ffmpeg/tools.hpp"
#include "plugin.hpp"
#include "utility.hpp"
extern "C" {
#include <obs-module.h>
}
2020-01-13 21:40:15 +00:00
using namespace encoder::ffmpeg::handler;
2020-01-13 00:52:30 +00:00
2020-01-13 21:40:15 +00:00
void prores_aw_handler::override_colorformat(AVPixelFormat& target_format, obs_data_t* settings, const AVCodec* codec,
AVCodecContext*)
2020-01-13 00:52:30 +00:00
{
static const std::array<std::pair<encoder::codec::prores::profile, AVPixelFormat>,
static_cast<size_t>(encoder::codec::prores::profile::_COUNT)>
profile_to_format_map{
std::pair{encoder::codec::prores::profile::APCO, AV_PIX_FMT_YUV422P10},
std::pair{encoder::codec::prores::profile::APCS, AV_PIX_FMT_YUV422P10},
std::pair{encoder::codec::prores::profile::APCN, AV_PIX_FMT_YUV422P10},
std::pair{encoder::codec::prores::profile::APCH, AV_PIX_FMT_YUV422P10},
std::pair{encoder::codec::prores::profile::AP4H, AV_PIX_FMT_YUV444P10},
std::pair{encoder::codec::prores::profile::AP4X, AV_PIX_FMT_YUV444P10},
};
2020-01-13 00:52:30 +00:00
const std::int64_t profile_id = obs_data_get_int(settings, P_PRORES_PROFILE);
for (auto kv : profile_to_format_map) {
if (kv.first == static_cast<encoder::codec::prores::profile>(profile_id)) {
target_format = kv.second;
2020-01-13 00:52:30 +00:00
break;
}
}
}
2020-01-13 21:40:15 +00:00
void prores_aw_handler::get_defaults(obs_data_t* settings, const AVCodec*, AVCodecContext*, bool)
2020-01-13 00:52:30 +00:00
{
obs_data_set_default_int(settings, P_PRORES_PROFILE, 0);
}
bool encoder::ffmpeg::handler::prores_aw_handler::has_pixel_format_support(ffmpeg_factory* instance)
{
return false;
}
2020-01-13 00:52:30 +00:00
inline const char* profile_to_name(const AVProfile* ptr)
{
switch (static_cast<encoder::codec::prores::profile>(ptr->profile)) {
case encoder::codec::prores::profile::APCO:
2020-01-13 21:40:15 +00:00
return D_TRANSLATE(P_PRORES_PROFILE_APCO);
case encoder::codec::prores::profile::APCS:
2020-01-13 21:40:15 +00:00
return D_TRANSLATE(P_PRORES_PROFILE_APCS);
case encoder::codec::prores::profile::APCN:
2020-01-13 21:40:15 +00:00
return D_TRANSLATE(P_PRORES_PROFILE_APCN);
case encoder::codec::prores::profile::APCH:
2020-01-13 21:40:15 +00:00
return D_TRANSLATE(P_PRORES_PROFILE_APCH);
case encoder::codec::prores::profile::AP4H:
2020-01-13 21:40:15 +00:00
return D_TRANSLATE(P_PRORES_PROFILE_AP4H);
case encoder::codec::prores::profile::AP4X:
2020-01-13 21:40:15 +00:00
return D_TRANSLATE(P_PRORES_PROFILE_AP4X);
2020-01-13 00:52:30 +00:00
default:
return ptr->name;
}
}
2020-01-13 21:40:15 +00:00
void prores_aw_handler::get_properties(obs_properties_t* props, const AVCodec* codec, AVCodecContext* context, bool)
2020-01-13 00:52:30 +00:00
{
if (!context) {
2020-01-13 21:40:15 +00:00
auto p = obs_properties_add_list(props, P_PRORES_PROFILE, D_TRANSLATE(P_PRORES_PROFILE), OBS_COMBO_TYPE_LIST,
OBS_COMBO_FORMAT_INT);
obs_property_set_long_description(p, D_TRANSLATE(D_DESC(P_PRORES_PROFILE)));
2020-01-13 00:52:30 +00:00
for (auto ptr = codec->profiles; ptr->profile != FF_PROFILE_UNKNOWN; ptr++) {
obs_property_list_add_int(p, profile_to_name(ptr), static_cast<int64_t>(ptr->profile));
}
} else {
obs_property_set_enabled(obs_properties_get(props, P_PRORES_PROFILE), false);
}
}
2020-01-13 21:40:15 +00:00
void prores_aw_handler::update(obs_data_t* settings, const AVCodec*, AVCodecContext* context)
2020-01-13 00:52:30 +00:00
{
context->profile = static_cast<int>(obs_data_get_int(settings, P_PRORES_PROFILE));
}
2020-01-13 21:40:15 +00:00
void prores_aw_handler::log_options(obs_data_t* settings, const AVCodec* codec, AVCodecContext* context)
2020-01-13 00:52:30 +00:00
{
2020-01-13 21:40:15 +00:00
LOG_INFO("[%s] Apple ProRes:", codec->name);
::ffmpeg::tools::print_av_option_string(context, "profile", " Profile", [&codec](int64_t v) {
2020-01-13 00:52:30 +00:00
int val = static_cast<int>(v);
for (auto ptr = codec->profiles; (ptr->profile != FF_PROFILE_UNKNOWN) && (ptr != nullptr); ptr++) {
if (ptr->profile == val) {
return std::string(profile_to_name(ptr));
}
}
return std::string("<Unknown>");
});
}
2020-01-13 21:40:15 +00:00
void prores_aw_handler::process_avpacket(AVPacket& packet, const AVCodec*, AVCodecContext*)
2020-01-13 00:52:30 +00:00
{
//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);
}