obs-StreamFX/source/ffmpeg/swscale.cpp

196 lines
4.2 KiB
C++
Raw Normal View History

// AUTOGENERATED COPYRIGHT HEADER START
// Copyright (C) 2020-2023 Michael Fabian 'Xaymar' Dirks <info@xaymar.com>
// Copyright (C) 2022 lainon <GermanAizek@yandex.ru>
// AUTOGENERATED COPYRIGHT HEADER END
2020-01-13 00:52:30 +00:00
#include "swscale.hpp"
#include "warning-disable.hpp"
2020-01-13 00:52:30 +00:00
#include <stdexcept>
#include "warning-enable.hpp"
2020-01-13 00:52:30 +00:00
2021-06-08 02:12:55 +00:00
using namespace streamfx::ffmpeg;
2020-01-13 00:52:30 +00:00
swscale::swscale() = default;
2020-01-13 21:40:15 +00:00
swscale::~swscale()
2020-01-13 00:52:30 +00:00
{
finalize();
}
void swscale::set_source_size(uint32_t width, uint32_t height)
2020-01-13 00:52:30 +00:00
{
source_size.first = width;
source_size.second = height;
}
2020-01-13 21:40:15 +00:00
void swscale::get_source_size(uint32_t& width, uint32_t& height)
2020-01-13 00:52:30 +00:00
{
width = this->source_size.first;
height = this->source_size.second;
}
2020-01-13 21:40:15 +00:00
std::pair<uint32_t, uint32_t> swscale::get_source_size()
2020-01-13 00:52:30 +00:00
{
return this->source_size;
}
uint32_t swscale::get_source_width()
2020-01-13 00:52:30 +00:00
{
return this->source_size.first;
}
uint32_t swscale::get_source_height()
2020-01-13 00:52:30 +00:00
{
return this->source_size.second;
}
2020-01-13 21:40:15 +00:00
void swscale::set_source_format(AVPixelFormat format)
2020-01-13 00:52:30 +00:00
{
source_format = format;
}
2020-01-13 21:40:15 +00:00
AVPixelFormat swscale::get_source_format()
2020-01-13 00:52:30 +00:00
{
return this->source_format;
}
2020-01-13 21:40:15 +00:00
void swscale::set_source_color(bool full_range, AVColorSpace space)
2020-01-13 00:52:30 +00:00
{
source_full_range = full_range;
source_colorspace = space;
}
2020-01-13 21:40:15 +00:00
void swscale::set_source_colorspace(AVColorSpace space)
2020-01-13 00:52:30 +00:00
{
this->source_colorspace = space;
}
2020-01-13 21:40:15 +00:00
AVColorSpace swscale::get_source_colorspace()
2020-01-13 00:52:30 +00:00
{
return this->source_colorspace;
}
2020-01-13 21:40:15 +00:00
void swscale::set_source_full_range(bool full_range)
2020-01-13 00:52:30 +00:00
{
this->source_full_range = full_range;
}
2020-01-13 21:40:15 +00:00
bool swscale::is_source_full_range()
2020-01-13 00:52:30 +00:00
{
return this->source_full_range;
}
void swscale::set_target_size(uint32_t width, uint32_t height)
2020-01-13 00:52:30 +00:00
{
target_size.first = width;
target_size.second = height;
}
2020-01-13 21:40:15 +00:00
void swscale::get_target_size(uint32_t& width, uint32_t& height)
2020-01-13 00:52:30 +00:00
{
width = target_size.first;
height = target_size.second;
}
2020-01-13 21:40:15 +00:00
std::pair<uint32_t, uint32_t> swscale::get_target_size()
2020-01-13 00:52:30 +00:00
{
return this->target_size;
}
uint32_t swscale::get_target_width()
2020-01-13 00:52:30 +00:00
{
return this->target_size.first;
}
uint32_t swscale::get_target_height()
2020-01-13 00:52:30 +00:00
{
return this->target_size.second;
}
2020-01-13 21:40:15 +00:00
void swscale::set_target_format(AVPixelFormat format)
2020-01-13 00:52:30 +00:00
{
target_format = format;
}
2020-01-13 21:40:15 +00:00
AVPixelFormat swscale::get_target_format()
2020-01-13 00:52:30 +00:00
{
return this->target_format;
}
2020-01-13 21:40:15 +00:00
void swscale::set_target_color(bool full_range, AVColorSpace space)
2020-01-13 00:52:30 +00:00
{
target_full_range = full_range;
target_colorspace = space;
}
2020-01-13 21:40:15 +00:00
void swscale::set_target_colorspace(AVColorSpace space)
2020-01-13 00:52:30 +00:00
{
this->target_colorspace = space;
}
2020-01-13 21:40:15 +00:00
AVColorSpace swscale::get_target_colorspace()
2020-01-13 00:52:30 +00:00
{
return this->target_colorspace;
}
2020-01-13 21:40:15 +00:00
void swscale::set_target_full_range(bool full_range)
2020-01-13 00:52:30 +00:00
{
this->target_full_range = full_range;
}
2020-01-13 21:40:15 +00:00
bool swscale::is_target_full_range()
2020-01-13 00:52:30 +00:00
{
return this->target_full_range;
}
2020-01-13 21:40:15 +00:00
bool swscale::initialize(int flags)
2020-01-13 00:52:30 +00:00
{
if (this->context) {
return false;
}
if (source_size.first == 0 || source_size.second == 0 || source_format == AV_PIX_FMT_NONE
2020-01-13 21:40:15 +00:00
|| source_colorspace == AVCOL_SPC_UNSPECIFIED) {
2020-01-13 00:52:30 +00:00
throw std::invalid_argument("not all source parameters were set");
}
if (target_size.first == 0 || target_size.second == 0 || target_format == AV_PIX_FMT_NONE
2020-01-13 21:40:15 +00:00
|| target_colorspace == AVCOL_SPC_UNSPECIFIED) {
2020-01-13 00:52:30 +00:00
throw std::invalid_argument("not all target parameters were set");
}
this->context =
sws_getContext(static_cast<int>(source_size.first), static_cast<int>(source_size.second), source_format,
static_cast<int>(target_size.first), static_cast<int>(target_size.second), target_format, flags,
nullptr, nullptr, nullptr);
2020-01-13 00:52:30 +00:00
if (!this->context) {
return false;
}
sws_setColorspaceDetails(this->context, sws_getCoefficients(source_colorspace), source_full_range ? 1 : 0,
2020-01-13 21:40:15 +00:00
sws_getCoefficients(target_colorspace), target_full_range ? 1 : 0, 1L << 16 | 0L,
1L << 16 | 0L, 1L << 16 | 0L);
2020-01-13 00:52:30 +00:00
return true;
}
2020-01-13 21:40:15 +00:00
bool swscale::finalize()
2020-01-13 00:52:30 +00:00
{
if (this->context) {
sws_freeContext(this->context);
this->context = nullptr;
return true;
}
return false;
}
int32_t swscale::convert(const uint8_t* const source_data[], const int source_stride[], int32_t source_row,
int32_t source_rows, uint8_t* const target_data[], const int target_stride[])
2020-01-13 00:52:30 +00:00
{
if (!this->context) {
return 0;
}
int height =
2020-01-13 21:40:15 +00:00
sws_scale(this->context, source_data, source_stride, source_row, source_rows, target_data, target_stride);
2020-01-13 00:52:30 +00:00
return height;
}