mirror of
https://invent.kde.org/frameworks/kimageformats.git
synced 2025-06-03 17:08:08 -04:00
7899c27a80
- Added support for writing EXR files - Large image support (tested with 32GiB 16-bit image) - OpenEXR multithreaded read/write support through the use of line blocks - Build option to enable conversion to sRGB when read (disabled by default) - Multi-view image reading support - Options support: Size, ImageFormat, Quality, CompressRatio - Metadata write/read support via EXR attributes The plugin is now quite complete. It should also work on KF5 (not tested): if there is the will to include this MR also on KF5, let me know and I will do all the necessary tests.
121 lines
3.3 KiB
C++
121 lines
3.3 KiB
C++
/*
|
|
SPDX-FileCopyrightText: 2023 Mirco Miranda <mircomir@outlook.com>
|
|
|
|
SPDX-License-Identifier: LGPL-2.0-or-later
|
|
*/
|
|
|
|
#include "scanlineconverter_p.h"
|
|
#include <cstring>
|
|
|
|
ScanLineConverter::ScanLineConverter(const QImage::Format &targetFormat)
|
|
: _targetFormat(targetFormat)
|
|
{
|
|
}
|
|
|
|
ScanLineConverter::ScanLineConverter(const ScanLineConverter &other)
|
|
: _targetFormat(other._targetFormat)
|
|
, _colorSpace(other._colorSpace)
|
|
, _defaultColorSpace(other._defaultColorSpace)
|
|
{
|
|
}
|
|
|
|
ScanLineConverter &ScanLineConverter::operator=(const ScanLineConverter &other)
|
|
{
|
|
_targetFormat = other._targetFormat;
|
|
_colorSpace = other._colorSpace;
|
|
_defaultColorSpace = other._defaultColorSpace;
|
|
return (*this);
|
|
}
|
|
|
|
QImage::Format ScanLineConverter::targetFormat() const
|
|
{
|
|
return _targetFormat;
|
|
}
|
|
|
|
void ScanLineConverter::setTargetColorSpace(const QColorSpace &colorSpace)
|
|
{
|
|
_colorSpace = colorSpace;
|
|
}
|
|
|
|
QColorSpace ScanLineConverter::targetColorSpace() const
|
|
{
|
|
return _colorSpace;
|
|
}
|
|
|
|
void ScanLineConverter::setDefaultSourceColorSpace(const QColorSpace &colorSpace)
|
|
{
|
|
_defaultColorSpace = colorSpace;
|
|
}
|
|
|
|
QColorSpace ScanLineConverter::defaultSourceColorSpace() const
|
|
{
|
|
return _defaultColorSpace;
|
|
}
|
|
|
|
const uchar *ScanLineConverter::convertedScanLine(const QImage &image, qint32 y)
|
|
{
|
|
auto colorSpaceConversion = isColorSpaceConversionNeeded(image);
|
|
if (image.format() == _targetFormat && !colorSpaceConversion) {
|
|
return image.constScanLine(y);
|
|
}
|
|
if (image.width() != _tmpBuffer.width() || image.format() != _tmpBuffer.format()) {
|
|
_tmpBuffer = QImage(image.width(), 1, image.format());
|
|
}
|
|
if (_tmpBuffer.isNull()) {
|
|
return nullptr;
|
|
}
|
|
std::memcpy(_tmpBuffer.bits(), image.constScanLine(y), std::min(_tmpBuffer.bytesPerLine(), image.bytesPerLine()));
|
|
if (colorSpaceConversion) {
|
|
auto cs = image.colorSpace();
|
|
if (!cs.isValid()) {
|
|
cs = _defaultColorSpace;
|
|
}
|
|
_tmpBuffer.setColorSpace(cs);
|
|
_tmpBuffer.convertToColorSpace(_colorSpace);
|
|
}
|
|
_convBuffer = _tmpBuffer.convertToFormat(_targetFormat);
|
|
if (_convBuffer.isNull()) {
|
|
return nullptr;
|
|
}
|
|
return _convBuffer.constBits();
|
|
}
|
|
|
|
qsizetype ScanLineConverter::bytesPerLine() const
|
|
{
|
|
if (_convBuffer.isNull()) {
|
|
return 0;
|
|
}
|
|
return _convBuffer.bytesPerLine();
|
|
}
|
|
|
|
bool ScanLineConverter::isColorSpaceConversionNeeded(const QImage &image, const QColorSpace &targetColorSpace, const QColorSpace &defaultColorSpace)
|
|
{
|
|
if (image.depth() < 24) { // RGB 8 bit or grater only
|
|
return false;
|
|
}
|
|
auto sourceColorSpace = image.colorSpace();
|
|
if (!sourceColorSpace.isValid()) {
|
|
sourceColorSpace = defaultColorSpace;
|
|
}
|
|
if (!sourceColorSpace.isValid() || !targetColorSpace.isValid()) {
|
|
return false;
|
|
}
|
|
|
|
auto stf = sourceColorSpace.transferFunction();
|
|
auto spr = sourceColorSpace.primaries();
|
|
auto ttf = targetColorSpace.transferFunction();
|
|
auto tpr = targetColorSpace.primaries();
|
|
// clang-format off
|
|
if (stf == QColorSpace::TransferFunction::Custom ||
|
|
ttf == QColorSpace::TransferFunction::Custom ||
|
|
spr == QColorSpace::Primaries::Custom ||
|
|
tpr == QColorSpace::Primaries::Custom) {
|
|
return true;
|
|
}
|
|
// clang-format on
|
|
if (stf == ttf && spr == tpr) {
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|