Pdf.cpp

Go to the documentation of this file.

/******************************************************************************************************
 * (C) 2014 markummitchell@github.com. This file is part of Engauge Digitizer, which is released      *
 * under GNU General Public License version 2 (GPLv2) or (at your option) any later version. See file *
 * LICENSE or go to gnu.org/licenses for details. Distribution requires prior written permission.     *
 ******************************************************************************************************/
 
#include "DlgImportCroppingPdf.h"
#include "ImportCroppingUtilPdf.h"
#include "Pdf.h"
#include "poppler-qt6.h"
#include <QApplication>
#include <QImage>
#include <QString>
 
using namespace Poppler;
 
const int X_TOP_LEFT = 0, Y_TOP_LEFT = 0;
const int WIDTH = -1, HEIGHT = -1; // Negative values give full page
const int FIRST_PAGE_1_BASED = 1;
 
Pdf::Pdf ()
{
}
 
PdfReturn Pdf::load (const QString &fileName,
                     QImage &image,
                     int resolution,
                     ImportCropping importCropping,
                     bool isErrorReportRegressionTest) const
{
  std::unique_ptr<Document> document;
 
  ImportCroppingUtilPdf importCroppingUtil;
  bool cropping = importCroppingUtil.applyImportCropping (isErrorReportRegressionTest,
                                                          fileName,
                                                          importCropping,
                                                          document);
 
  PdfReturn rtn;
  QApplication::setOverrideCursor(Qt::BusyCursor); // Since loading can be slow
  if (cropping) {
 
    rtn = loadWithCropping (std::move(document),
                            image,
                            resolution);
 
  } else {
 
    rtn = loadWithoutCropping (fileName,
                               image,
                               resolution);
 
  }
  QApplication::restoreOverrideCursor();
 
  return rtn;
}
 
PdfReturn Pdf::loadWithCropping (std::unique_ptr<Document> document,
                                 QImage &image,
                                 int resolution) const
{
  PdfReturn pdfReturn = PDF_RETURN_FAILED;
 
  // Get page and extent. At this point it is always true that the image can be read
  DlgImportCroppingPdf dlg (*document,
                            resolution);
  if (dlg.exec() == QDialog::Accepted) {
 
    // Returned image is null if it could not be read
    image = dlg.image ();
 
    if (!image.isNull()) {
      pdfReturn = PDF_RETURN_SUCCESS;
    }
 
  } else {
    pdfReturn = PDF_RETURN_CANCELED;
  }
 
  return pdfReturn;
}
 
PdfReturn Pdf::loadWithoutCropping (const QString &fileName,
                                    QImage &image,
                                    int resolution) const
{
  PdfReturn pdfReturn = PDF_RETURN_FAILED;
 
  // Simple check to prevent complaints from poppler code
  if (fileName.right (4).toLower () == ".pdf") {
 
    // Try to read the file
    std::unique_ptr<Document> document = Document::load (fileName);
 
    if (document != nullptr) {
      if (!document->isLocked ()) {
 
        std::unique_ptr<Page> page = document->page (FIRST_PAGE_1_BASED - 1);
        if (page != nullptr) {
 
          image = page->renderToImage (resolution,
                                       resolution,
                                       X_TOP_LEFT,
                                       Y_TOP_LEFT,
                                       WIDTH,
                                       HEIGHT);
 
          if (!image.isNull()) {
            pdfReturn = PDF_RETURN_SUCCESS;
          }
        }
      }
    }
  }
 
  return pdfReturn;
}