Render tool basics

2025-06-05 21:59:17 +02:00 · 2020-10-26 19:28:56 +01:00
parent 584a992da6
commit 428e5dc2ad
6 changed files with 251 additions and 40 deletions
--- a/PdfTool/pdftoolrender.cpp
+++ b/PdfTool/pdftoolrender.cpp
@ -20,5 +20,145 @@
 namespace pdftool
 {

+static PDFToolRender s_toolRenderApplication;
+static PDFToolBenchmark s_toolBenchmarkApplication;
+
+QString PDFToolRender::getStandardString(PDFToolAbstractApplication::StandardString standardString) const
+{
+    switch (standardString)
+    {
+        case Command:
+            return "render";
+
+        case Name:
+            return PDFToolTranslationContext::tr("Render document");
+
+        case Description:
+            return PDFToolTranslationContext::tr("Render selected pages of document into image files.");
+
+        default:
+            Q_ASSERT(false);
+            break;
+    }
+
+    return QString();
+}
+
+PDFToolAbstractApplication::Options PDFToolRender::getOptionsFlags() const
+{
+    return ConsoleFormat | OpenDocument | PageSelector | ImageWriterSettings | ImageExportSettingsFiles | ImageExportSettingsResolution | ColorManagementSystem | RenderFlags;
+}
+
+QString PDFToolBenchmark::getStandardString(PDFToolAbstractApplication::StandardString standardString) const
+{
+    switch (standardString)
+    {
+        case Command:
+            return "benchmark";
+
+        case Name:
+            return PDFToolTranslationContext::tr("Benchmark rendering");
+
+        case Description:
+            return PDFToolTranslationContext::tr("Benchmark page rendering (measure time, detect errors).");
+
+        default:
+            Q_ASSERT(false);
+            break;
+    }
+
+    return QString();
+}
+
+PDFToolAbstractApplication::Options PDFToolBenchmark::getOptionsFlags() const
+{
+    return ConsoleFormat | OpenDocument | PageSelector | ImageExportSettingsResolution | ColorManagementSystem | RenderFlags;
+}
+
+int PDFToolRenderBase::execute(const PDFToolOptions& options)
+{
+    pdf::PDFDocument document;
+    QByteArray sourceData;
+    if (!readDocument(options, document, &sourceData))
+    {
+        return ErrorDocumentReading;
+    }
+
+    QString parseError;
+    std::vector<pdf::PDFInteger> pages = options.getPageRange(document.getCatalog()->getPageCount(), parseError, true);
+
+    if (!parseError.isEmpty())
+    {
+        PDFConsole::writeError(parseError, options.outputCodec);
+        return ErrorInvalidArguments;
+    }
+
+    QString errorMessage;
+    Options optionFlags = getOptionsFlags();
+    if (!options.imageExportSettings.validate(&errorMessage, false, optionFlags.testFlag(ImageExportSettingsFiles), optionFlags.testFlag(ImageExportSettingsResolution)))
+    {
+        PDFConsole::writeError(errorMessage, options.outputCodec);
+        return ErrorInvalidArguments;
+    }
+
+    // We are ready to render the document
+    pdf::PDFOptionalContentActivity optionalContentActivity(&document, pdf::OCUsage::Export, nullptr);
+    m_cms = m_proxy->getCMSManager()->getCurrentCMS();
+    pdf::PDFRasterizerPool rasterizerPool(&document, m_proxy->getFontCache(), m_proxy->getCMSManager(),
+                                          &optionalContentActivity, m_proxy->getFeatures(), m_proxy->getMeshQualitySettings(),
+                                          pdf::PDFRasterizerPool::getDefaultRasterizerCount(), m_proxy->isUsingOpenGL(), m_proxy->getSurfaceFormat(), this);
+    connect(&rasterizerPool, &pdf::PDFRasterizerPool::renderError, this, &PDFRenderToImagesDialog::onRenderError);
+
+    auto imageSizeGetter = [&options](const pdf::PDFPage* page) -> QSize
+    {
+        Q_ASSERT(page);
+
+        switch (options.imageExportSettings.getResolutionMode())
+        {
+            case pdf::PDFPageImageExportSettings::ResolutionMode::DPI:
+            {
+                QSizeF size = page->getRotatedMediaBox().size() * pdf::PDF_POINT_TO_INCH * options.imageExportSettings.getDpiResolution();
+                return size.toSize();
+            }
+
+            case pdf::PDFPageImageExportSettings::ResolutionMode::Pixels:
+            {
+                int pixelResolution = options.imageExportSettings.getPixelResolution();
+                QSizeF size = page->getRotatedMediaBox().size().scaled(pixelResolution, pixelResolution, Qt::KeepAspectRatio);
+                return size.toSize();
+            }
+
+            default:
+            {
+                Q_ASSERT(false);
+                break;
+            }
+        }
+
+        return QSize();
+    };
+
+    auto processImage = [this](const pdf::PDFInteger pageIndex, QImage&& image)
+    {
+        QString fileName = m_imageExportSettings.getOutputFileName(pageIndex, m_imageWriterSettings.getCurrentFormat());
+
+        QImageWriter imageWriter(fileName, m_imageWriterSettings.getCurrentFormat());
+        imageWriter.setSubType(m_imageWriterSettings.getCurrentSubtype());
+        imageWriter.setCompression(m_imageWriterSettings.getCompression());
+        imageWriter.setQuality(m_imageWriterSettings.getQuality());
+        imageWriter.setGamma(m_imageWriterSettings.getGamma());
+        imageWriter.setOptimizedWrite(m_imageWriterSettings.hasOptimizedWrite());
+        imageWriter.setProgressiveScanWrite(m_imageWriterSettings.hasProgressiveScanWrite());
+
+        if (!imageWriter.write(image))
+        {
+            emit m_rasterizerPool->renderError(pdf::PDFRenderError(pdf::RenderErrorType::Error, tr("Can't write page image to file '%1', because: %2.").arg(fileName).arg(imageWriter.errorString())));
+        }
+    };
+
+    rasterizerPool.render(pageIndices, imageSizeGetter, processImage, m_progress);
+
+    return ExitSuccess;
+}

 }   // namespace pdftool