itsmattkc
6f0d72c05d
While planar audio cannot be used for playback, it's much easier for processing. This should lead to cleaner and easier to write code in the future.
254 lines
7.3 KiB
C++
254 lines
7.3 KiB
C++
/***
|
|
|
|
Olive - Non-Linear Video Editor
|
|
Copyright (C) 2019 Olive Team
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
***/
|
|
|
|
#include "pixelformat.h"
|
|
|
|
#include "OpenImageIO/imagebuf.h"
|
|
#include <QCoreApplication>
|
|
#include <QDebug>
|
|
#include <QFloat16>
|
|
|
|
#include "common/define.h"
|
|
#include "core.h"
|
|
|
|
bool PixelFormat::FormatHasAlphaChannel(const PixelFormat::Format &format)
|
|
{
|
|
switch (format) {
|
|
case PixelFormat::PIX_FMT_RGBA8:
|
|
case PixelFormat::PIX_FMT_RGBA16U:
|
|
case PixelFormat::PIX_FMT_RGBA16F:
|
|
case PixelFormat::PIX_FMT_RGBA32F:
|
|
return true;
|
|
case PixelFormat::PIX_FMT_RGB8:
|
|
case PixelFormat::PIX_FMT_RGB16U:
|
|
case PixelFormat::PIX_FMT_RGB16F:
|
|
case PixelFormat::PIX_FMT_RGB32F:
|
|
case PixelFormat::PIX_FMT_INVALID:
|
|
case PixelFormat::PIX_FMT_COUNT:
|
|
break;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
OIIO::TypeDesc::BASETYPE PixelFormat::GetOIIOTypeDesc(const PixelFormat::Format &format)
|
|
{
|
|
switch (format) {
|
|
case PixelFormat::PIX_FMT_RGB8:
|
|
case PixelFormat::PIX_FMT_RGBA8:
|
|
return OIIO::TypeDesc::UINT8;
|
|
case PixelFormat::PIX_FMT_RGB16U:
|
|
case PixelFormat::PIX_FMT_RGBA16U:
|
|
return OIIO::TypeDesc::UINT16;
|
|
case PixelFormat::PIX_FMT_RGB16F:
|
|
case PixelFormat::PIX_FMT_RGBA16F:
|
|
return OIIO::TypeDesc::HALF;
|
|
case PixelFormat::PIX_FMT_RGB32F:
|
|
case PixelFormat::PIX_FMT_RGBA32F:
|
|
return OIIO::TypeDesc::FLOAT;
|
|
case PixelFormat::PIX_FMT_INVALID:
|
|
case PixelFormat::PIX_FMT_COUNT:
|
|
break;
|
|
}
|
|
|
|
return OIIO::TypeDesc::UNKNOWN;
|
|
}
|
|
|
|
QString PixelFormat::GetName(const PixelFormat::Format &format)
|
|
{
|
|
switch (format) {
|
|
case PixelFormat::PIX_FMT_RGB8:
|
|
case PixelFormat::PIX_FMT_RGBA8:
|
|
return tr("8-bit");
|
|
case PixelFormat::PIX_FMT_RGB16U:
|
|
case PixelFormat::PIX_FMT_RGBA16U:
|
|
return tr("16-bit Integer");
|
|
case PixelFormat::PIX_FMT_RGB16F:
|
|
case PixelFormat::PIX_FMT_RGBA16F:
|
|
return tr("Half-Float (16-bit)");
|
|
case PixelFormat::PIX_FMT_RGB32F:
|
|
case PixelFormat::PIX_FMT_RGBA32F:
|
|
return tr("Full-Float (32-bit)");
|
|
case PixelFormat::PIX_FMT_INVALID:
|
|
case PixelFormat::PIX_FMT_COUNT:
|
|
break;
|
|
}
|
|
|
|
return tr("Unknown (%1)").arg(format);
|
|
}
|
|
|
|
PixelFormat* PixelFormat::instance_ = nullptr;
|
|
|
|
void PixelFormat::CreateInstance()
|
|
{
|
|
instance_ = new PixelFormat();
|
|
}
|
|
|
|
void PixelFormat::DestroyInstance()
|
|
{
|
|
delete instance_;
|
|
}
|
|
|
|
PixelFormat *PixelFormat::instance()
|
|
{
|
|
return instance_;
|
|
}
|
|
|
|
PixelFormat::Format PixelFormat::GetConfiguredFormatForMode(RenderMode::Mode mode)
|
|
{
|
|
return static_cast<PixelFormat::Format>(Core::GetPreferenceForRenderMode(mode, QStringLiteral("PixelFormat")).toInt());
|
|
}
|
|
|
|
void PixelFormat::SetConfiguredFormatForMode(RenderMode::Mode mode, PixelFormat::Format format)
|
|
{
|
|
if (format != GetConfiguredFormatForMode(mode)) {
|
|
Core::SetPreferenceForRenderMode(mode, QStringLiteral("PixelFormat"), format);
|
|
|
|
emit FormatChanged();
|
|
}
|
|
}
|
|
|
|
PixelFormat::Format PixelFormat::OIIOFormatToOliveFormat(OIIO::TypeDesc desc, bool has_alpha)
|
|
{
|
|
if (desc == OIIO::TypeDesc::UINT8) {
|
|
return has_alpha ? PixelFormat::PIX_FMT_RGBA8 : PixelFormat::PIX_FMT_RGB8;
|
|
} else if (desc == OIIO::TypeDesc::UINT16) {
|
|
return has_alpha ? PixelFormat::PIX_FMT_RGBA16U : PixelFormat::PIX_FMT_RGB16U;
|
|
} else if (desc == OIIO::TypeDesc::HALF) {
|
|
return has_alpha ? PixelFormat::PIX_FMT_RGBA16F : PixelFormat::PIX_FMT_RGB16F;
|
|
} else if (desc == OIIO::TypeDesc::FLOAT) {
|
|
return has_alpha ? PixelFormat::PIX_FMT_RGBA32F : PixelFormat::PIX_FMT_RGB32F;
|
|
}
|
|
|
|
return PixelFormat::PIX_FMT_INVALID;
|
|
}
|
|
|
|
/*PixelFormat::Info PixelFormat::GetPixelFormatInfo(const PixelFormat::Format &format)
|
|
{
|
|
PixelFormat::Info info;
|
|
|
|
switch (format) {
|
|
case PixelFormat::PIX_FMT_RGB8:
|
|
case PixelFormat::PIX_FMT_RGBA8:
|
|
info.name = tr("8-bit");
|
|
info.internal_format = (format == PixelFormat::PIX_FMT_RGB8) ? GL_RGB8 : GL_RGBA8;
|
|
info.gl_pixel_type = GL_UNSIGNED_BYTE;
|
|
info.oiio_desc = OIIO::TypeDesc::UINT8;
|
|
break;
|
|
case PixelFormat::PIX_FMT_RGB16U:
|
|
case PixelFormat::PIX_FMT_RGBA16U:
|
|
info.name = tr("16-bit Integer");
|
|
info.internal_format = (format == PixelFormat::PIX_FMT_RGB16U) ? GL_RGB8 : GL_RGBA16;
|
|
info.gl_pixel_type = GL_UNSIGNED_SHORT;
|
|
info.oiio_desc = OIIO::TypeDesc::UINT16;
|
|
break;
|
|
case PixelFormat::PIX_FMT_RGB16F:
|
|
case PixelFormat::PIX_FMT_RGBA16F:
|
|
info.name = tr("Half-Float (16-bit)");
|
|
info.internal_format = (format == PixelFormat::PIX_FMT_RGB8) ? GL_RGB8 : GL_RGBA16F;
|
|
info.gl_pixel_type = GL_HALF_FLOAT;
|
|
info.oiio_desc = OIIO::TypeDesc::HALF;
|
|
break;
|
|
case PixelFormat::PIX_FMT_RGB32F:
|
|
case PixelFormat::PIX_FMT_RGBA32F:
|
|
info.name = tr("Full-Float (32-bit)");
|
|
info.internal_format = (format == PixelFormat::PIX_FMT_RGB8) ? GL_RGB8 : GL_RGBA32F;
|
|
info.gl_pixel_type = GL_FLOAT;
|
|
info.oiio_desc = OIIO::TypeDesc::FLOAT;
|
|
break;
|
|
case PixelFormat::PIX_FMT_INVALID:
|
|
case PixelFormat::PIX_FMT_COUNT:
|
|
qFatal("Invalid pixel format requested");
|
|
}
|
|
|
|
info.pixel_format = GL_RGBA;
|
|
info.bytes_per_pixel = BytesPerPixel(format);
|
|
|
|
return info;
|
|
}*/
|
|
|
|
int PixelFormat::GetBufferSize(const PixelFormat::Format &format, const int &width, const int &height)
|
|
{
|
|
return BytesPerPixel(format) * width * height;
|
|
}
|
|
|
|
int PixelFormat::BytesPerPixel(const PixelFormat::Format &format)
|
|
{
|
|
return BytesPerChannel(format) * ChannelCount(format);
|
|
}
|
|
|
|
int PixelFormat::BytesPerChannel(const PixelFormat::Format &format)
|
|
{
|
|
switch (format) {
|
|
case PixelFormat::PIX_FMT_RGB8:
|
|
case PixelFormat::PIX_FMT_RGBA8:
|
|
return 1;
|
|
case PixelFormat::PIX_FMT_RGB16U:
|
|
case PixelFormat::PIX_FMT_RGB16F:
|
|
case PixelFormat::PIX_FMT_RGBA16U:
|
|
case PixelFormat::PIX_FMT_RGBA16F:
|
|
return 2;
|
|
case PixelFormat::PIX_FMT_RGB32F:
|
|
case PixelFormat::PIX_FMT_RGBA32F:
|
|
return 4;
|
|
case PixelFormat::PIX_FMT_INVALID:
|
|
case PixelFormat::PIX_FMT_COUNT:
|
|
break;
|
|
}
|
|
|
|
qFatal("Invalid pixel format requested");
|
|
|
|
// qFatal will abort so we won't get here, but this suppresses compiler warnings
|
|
return 0;
|
|
}
|
|
|
|
int PixelFormat::ChannelCount(const PixelFormat::Format &format)
|
|
{
|
|
if (PixelFormat::FormatHasAlphaChannel(format)) {
|
|
return kRGBAChannels;
|
|
} else {
|
|
return kRGBChannels;
|
|
}
|
|
}
|
|
|
|
FramePtr PixelFormat::ConvertPixelFormat(FramePtr frame, const PixelFormat::Format &dest_format)
|
|
{
|
|
if (frame->format() == dest_format) {
|
|
return frame;
|
|
}
|
|
|
|
FramePtr converted = Frame::Create();
|
|
|
|
// Copy parameters
|
|
converted->set_video_params(frame->video_params());
|
|
converted->set_timestamp(frame->timestamp());
|
|
converted->allocate();
|
|
|
|
OIIO::ImageBuf src(OIIO::ImageSpec(frame->width(), frame->height(), ChannelCount(frame->format()), GetOIIOTypeDesc(frame->format())), frame->data());
|
|
OIIO::ImageBuf dst(OIIO::ImageSpec(converted->width(), converted->height(), ChannelCount(converted->format()), GetOIIOTypeDesc(converted->format())), converted->data());
|
|
|
|
if (dst.copy_pixels(src)) {
|
|
return converted;
|
|
} else {
|
|
return nullptr;
|
|
}
|
|
}
|
|
|