123 lines
3.6 KiB
C++
123 lines
3.6 KiB
C++
/***
|
|
|
|
Olive - Non-Linear Video Editor
|
|
Copyright (C) 2020 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 "oiioutils.h"
|
|
#if OIIO_VERSION < 20112
|
|
#include "render/videoparams.h"
|
|
#endif
|
|
namespace olive {
|
|
|
|
void OIIOUtils::FrameToBuffer(const Frame* frame, OIIO::ImageBuf *buf)
|
|
{
|
|
#if OIIO_VERSION < 20112
|
|
//
|
|
// Workaround for OIIO bug that ignores destination stride in versions OLDER than 2.1.12
|
|
//
|
|
// See more: https://github.com/OpenImageIO/oiio/pull/2487
|
|
//
|
|
int width_in_bytes = frame->width() * VideoParams::GetBytesPerPixel(frame->format(), frame->channel_count());
|
|
|
|
for (int i=0;i<buf->spec().height;i++) {
|
|
memcpy(
|
|
#if OIIO_VERSION < 10903
|
|
reinterpret_cast<char*>(buf->localpixels()) + i * width_in_bytes,
|
|
#else
|
|
reinterpret_cast<char*>(buf->localpixels()) + i * buf->scanline_stride(),
|
|
#endif
|
|
frame->const_data() + i * frame->linesize_bytes(),
|
|
width_in_bytes);
|
|
}
|
|
#else
|
|
buf->set_pixels(OIIO::ROI(),
|
|
buf->spec().format,
|
|
frame->const_data(),
|
|
OIIO::AutoStride,
|
|
frame->linesize_bytes());
|
|
#endif
|
|
}
|
|
|
|
void OIIOUtils::BufferToFrame(OIIO::ImageBuf *buf, Frame* frame)
|
|
{
|
|
#if OIIO_VERSION < 20112
|
|
//
|
|
// Workaround for OIIO bug that ignores destination stride in versions OLDER than 2.1.12
|
|
//
|
|
// See more: https://github.com/OpenImageIO/oiio/pull/2487
|
|
//
|
|
int width_in_bytes = frame->width() * VideoParams::GetBytesPerPixel(frame->format(), frame->channel_count());
|
|
|
|
for (int i=0;i<buf->spec().height;i++) {
|
|
memcpy(frame->data() + i * frame->linesize_bytes(),
|
|
#if OIIO_VERSION < 10903
|
|
reinterpret_cast<const char*>(buf->localpixels()) + i * width_in_bytes,
|
|
#else
|
|
reinterpret_cast<const char*>(buf->localpixels()) + i * buf->scanline_stride(),
|
|
#endif
|
|
width_in_bytes);
|
|
}
|
|
#else
|
|
buf->get_pixels(OIIO::ROI(),
|
|
buf->spec().format,
|
|
frame->data(),
|
|
OIIO::AutoStride,
|
|
frame->linesize_bytes());
|
|
#endif
|
|
}
|
|
|
|
rational OIIOUtils::GetPixelAspectRatioFromOIIO(const OIIO::ImageSpec &spec)
|
|
{
|
|
return rational::fromDouble(spec.get_float_attribute("PixelAspectRatio", 1));
|
|
}
|
|
|
|
VideoParams::Format OIIOUtils::GetFormatFromOIIOBasetype(OIIO::TypeDesc::BASETYPE type)
|
|
{
|
|
switch (type) {
|
|
case OIIO::TypeDesc::UNKNOWN:
|
|
case OIIO::TypeDesc::NONE:
|
|
break;
|
|
|
|
case OIIO::TypeDesc::INT8:
|
|
case OIIO::TypeDesc::INT16:
|
|
case OIIO::TypeDesc::INT32:
|
|
case OIIO::TypeDesc::UINT32:
|
|
case OIIO::TypeDesc::INT64:
|
|
case OIIO::TypeDesc::UINT64:
|
|
case OIIO::TypeDesc::STRING:
|
|
case OIIO::TypeDesc::PTR:
|
|
case OIIO::TypeDesc::LASTBASE:
|
|
case OIIO::TypeDesc::DOUBLE:
|
|
qDebug() << "Tried to use unknown OIIO base type";
|
|
break;
|
|
|
|
case OIIO::TypeDesc::UINT8:
|
|
return VideoParams::kFormatUnsigned8;
|
|
case OIIO::TypeDesc::UINT16:
|
|
return VideoParams::kFormatUnsigned16;
|
|
case OIIO::TypeDesc::HALF:
|
|
return VideoParams::kFormatFloat16;
|
|
case OIIO::TypeDesc::FLOAT:
|
|
return VideoParams::kFormatFloat32;
|
|
}
|
|
|
|
return VideoParams::kFormatInvalid;
|
|
}
|
|
|
|
}
|