Bitmap processing

This commit is contained in:
Jakub Melka
2021-01-19 20:14:50 +01:00
parent d3827cfcc4
commit ea62e6ee32
5 changed files with 376 additions and 3 deletions

View File

@ -17,6 +17,7 @@
#include "pdftransparencyrenderer.h"
#include "pdfdocument.h"
#include "pdfcms.h"
namespace pdf
{
@ -37,7 +38,7 @@ PDFFloatBitmap::PDFFloatBitmap(size_t width, size_t height, PDFPixelFormat forma
{
Q_ASSERT(format.isValid());
m_data.resize(format.calculateBitmapDataLength(width, height), static_cast<PDFColorComponent>(0.0));
m_data.resize(format.calculateBitmapDataLength(width, height), static_cast<PDFColorComponent>(0.0f));
}
PDFColorBuffer PDFFloatBitmap::getPixel(size_t x, size_t y)
@ -46,6 +47,11 @@ PDFColorBuffer PDFFloatBitmap::getPixel(size_t x, size_t y)
return PDFColorBuffer(m_data.data() + index, m_pixelSize);
}
PDFColorBuffer PDFFloatBitmap::getPixels()
{
return PDFColorBuffer(m_data.data(), m_data.size());
}
const PDFColorComponent* PDFFloatBitmap::begin() const
{
return m_data.data();
@ -66,11 +72,146 @@ PDFColorComponent* PDFFloatBitmap::end()
return m_data.data() + m_data.size();
}
void PDFFloatBitmap::makeTransparent()
{
if (m_format.hasShapeChannel())
{
fillChannel(m_format.getShapeChannelIndex(), 0.0f);
}
if (m_format.hasOpacityChannel())
{
fillChannel(m_format.getOpacityChannelIndex(), 0.0f);
}
}
void PDFFloatBitmap::makeOpaque()
{
if (m_format.hasShapeChannel())
{
fillChannel(m_format.getShapeChannelIndex(), 1.0f);
}
if (m_format.hasOpacityChannel())
{
fillChannel(m_format.getOpacityChannelIndex(), 1.0f);
}
}
size_t PDFFloatBitmap::getPixelIndex(size_t x, size_t y) const
{
return (y * m_width + x) * m_pixelSize;
}
PDFFloatBitmap PDFFloatBitmap::extractProcessColors()
{
PDFPixelFormat format = PDFPixelFormat::createFormat(m_format.getProcessColorChannelCount(), 0, false, m_format.hasProcessColorsSubtractive());
PDFFloatBitmap result(getWidth(), getHeight(), format);
for (size_t x = 0; x < getWidth(); ++x)
{
for (size_t y = 0; y < getHeight(); ++y)
{
PDFColorBuffer sourceProcessColorBuffer = getPixel(x, y);
PDFColorBuffer targetProcessColorBuffer = result.getPixel(x, y);
Q_ASSERT(sourceProcessColorBuffer.size() >= targetProcessColorBuffer.size());
std::copy(sourceProcessColorBuffer.cbegin(), std::next(sourceProcessColorBuffer.cbegin(), targetProcessColorBuffer.size()), targetProcessColorBuffer.begin());
}
}
return result;
}
void PDFFloatBitmap::fillChannel(size_t channel, PDFColorComponent value)
{
// Do we have just one channel?
if (m_format.getChannelCount() == 1)
{
Q_ASSERT(channel == 0);
std::fill(m_data.begin(), m_data.end(), value);
return;
}
for (PDFColorComponent* pixel = begin(); pixel != end(); pixel += m_pixelSize)
{
pixel[channel] = value;
}
}
PDFFloatBitmapWithColorSpace::PDFFloatBitmapWithColorSpace()
{
}
PDFFloatBitmapWithColorSpace::PDFFloatBitmapWithColorSpace(size_t width, size_t height, PDFPixelFormat format, PDFColorSpacePointer blendColorSpace) :
PDFFloatBitmap(width, height, format),
m_colorSpace(blendColorSpace)
{
Q_ASSERT(!blendColorSpace || blendColorSpace->isBlendColorSpace());
}
PDFColorSpacePointer PDFFloatBitmapWithColorSpace::getColorSpace() const
{
return m_colorSpace;
}
void PDFFloatBitmapWithColorSpace::setColorSpace(const PDFColorSpacePointer& colorSpace)
{
m_colorSpace = colorSpace;
}
void PDFFloatBitmapWithColorSpace::convertToColorSpace(const PDFCMS* cms,
RenderingIntent intent,
const PDFColorSpacePointer& targetColorSpace,
PDFRenderErrorReporter* reporter)
{
Q_ASSERT(m_colorSpace);
if (m_colorSpace->equals(targetColorSpace.get()))
{
return;
}
const uint8_t targetDeviceColors = static_cast<uint8_t>(targetColorSpace->getColorComponentCount());
PDFPixelFormat newFormat = getPixelFormat();
newFormat.setProcessColors(targetDeviceColors);
newFormat.setProcessColorsSubtractive(targetDeviceColors == 4);
PDFFloatBitmap sourceProcessColors = extractProcessColors();
PDFFloatBitmap targetProcessColors(sourceProcessColors.getWidth(), sourceProcessColors.getHeight(), PDFPixelFormat::createFormat(targetDeviceColors, 0, false, newFormat.hasProcessColorsSubtractive()));
if (!PDFAbstractColorSpace::transform(m_colorSpace.data(), targetColorSpace.data(), cms, intent, sourceProcessColors.getPixels(), targetProcessColors.getPixels(), reporter))
{
reporter->reportRenderError(RenderErrorType::Error, PDFTranslationContext::tr("Transformation between blending color spaces failed."));
}
PDFFloatBitmapWithColorSpace temporary(getWidth(), getHeight(), newFormat, targetColorSpace);
for (size_t x = 0; x < getWidth(); ++x)
{
for (size_t y = 0; y < getHeight(); ++y)
{
PDFColorBuffer sourceProcessColorBuffer = targetProcessColors.getPixel(x, y);
PDFColorBuffer sourceSpotColorAndOpacityBuffer = getPixel(x, y);
PDFColorBuffer targetBuffer = temporary.getPixel(x, y);
Q_ASSERT(sourceProcessColorBuffer.size() <= targetBuffer.size());
// Copy process colors
auto targetIt = targetBuffer.begin();
targetIt = std::copy(sourceProcessColorBuffer.cbegin(), sourceProcessColorBuffer.cend(), targetIt);
Q_ASSERT(std::distance(targetIt, targetBuffer.end()) == temporary.getPixelFormat().getSpotColorChannelCount() + temporary.getPixelFormat().getAuxiliaryChannelCount());
auto sourceIt = std::next(sourceSpotColorAndOpacityBuffer.cbegin(), temporary.getPixelFormat().getProcessColorChannelCount());
targetIt = std::copy(sourceIt, sourceSpotColorAndOpacityBuffer.cend(), targetIt);
Q_ASSERT(targetIt == targetBuffer.cend());
}
}
*this = qMove(temporary);
}
PDFTransparencyRenderer::PDFTransparencyRenderer(const PDFPage* page,
const PDFDocument* document,
const PDFFontCache* fontCache,
@ -110,6 +251,8 @@ const PDFFloatBitmap& PDFTransparencyRenderer::endPaint()
Q_ASSERT(m_active);
m_pageTransparencyGroupGuard.reset();
m_active = false;
return *getImmediateBackdrop();
}
void PDFTransparencyRenderer::performPathPainting(const QPainterPath& path, bool stroke, bool fill, bool text, Qt::FillRule fillRule)
@ -134,10 +277,86 @@ void PDFTransparencyRenderer::performRestoreGraphicState(ProcessOrder order)
void PDFTransparencyRenderer::performBeginTransparencyGroup(ProcessOrder order, const PDFTransparencyGroup& transparencyGroup)
{
if (order == ProcessOrder::BeforeOperation)
{
PDFTransparencyGroupPainterData data;
data.group = transparencyGroup;
data.alphaIsShape = getGraphicState()->getAlphaIsShape();
data.alphaFill = getGraphicState()->getAlphaFilling();
data.alphaStroke = getGraphicState()->getAlphaStroking();
data.blendMode = getGraphicState()->getBlendMode();
data.blackPointCompensationMode = getGraphicState()->getBlackPointCompensationMode();
data.renderingIntent = getGraphicState()->getRenderingIntent();
data.blendColorSpace = transparencyGroup.colorSpacePointer;
if (!data.blendColorSpace)
{
data.blendColorSpace = getBlendColorSpace();
}
// Create initial backdrop, according to 11.4.8 of PDF 2.0 specification.
// If group is knockout, use initial backdrop.
PDFFloatBitmapWithColorSpace* oldBackdrop = getBackdrop();
data.initialBackdrop = *getBackdrop();
if (isTransparencyGroupIsolated())
{
// Make initial backdrop transparent
data.initialBackdrop.makeTransparent();
}
// Prepare soft mask
data.softMask = PDFFloatBitmap(oldBackdrop->getWidth(), oldBackdrop->getHeight(), PDFPixelFormat::createOpacityMask());
// TODO: Create soft mask
data.softMask.makeOpaque();
data.initialBackdrop.convertToColorSpace(getCMS(), data.renderingIntent, data.blendColorSpace, this);
data.immediateBackdrop = data.initialBackdrop;
m_transparencyGroupDataStack.emplace_back(qMove(data));
}
}
void PDFTransparencyRenderer::performEndTransparencyGroup(ProcessOrder order, const PDFTransparencyGroup& transparencyGroup)
{
}
PDFFloatBitmapWithColorSpace* PDFTransparencyRenderer::getInitialBackdrop()
{
return &m_transparencyGroupDataStack.back().initialBackdrop;
}
PDFFloatBitmapWithColorSpace* PDFTransparencyRenderer::getImmediateBackdrop()
{
return &m_transparencyGroupDataStack.back().immediateBackdrop;
}
PDFFloatBitmapWithColorSpace* PDFTransparencyRenderer::getBackdrop()
{
if (isTransparencyGroupKnockout())
{
return getInitialBackdrop();
}
else
{
return getImmediateBackdrop();
}
}
const PDFColorSpacePointer& PDFTransparencyRenderer::getBlendColorSpace() const
{
return m_transparencyGroupDataStack.back().blendColorSpace;
}
bool PDFTransparencyRenderer::isTransparencyGroupIsolated() const
{
return m_transparencyGroupDataStack.back().group.isolated;
}
bool PDFTransparencyRenderer::isTransparencyGroupKnockout() const
{
return m_transparencyGroupDataStack.back().group.knockout;
}
} // namespace pdf