Restore atlas single file pdf export

qgis · Jan 5, 2018 · 427da5c · 427da5c
1 parent 9751c77
commit 427da5c
Show file tree

Hide file tree

Showing 5 changed files with 290 additions and 33 deletions.
diff --git a/python/core/layout/qgslayoutexporter.sip b/python/core/layout/qgslayoutexporter.sip
@@ -259,6 +259,20 @@ Returns a result code indicating whether the export was successful or an
 error was encountered.
 %End
 
+    static ExportResult exportToPdf( QgsAbstractLayoutIterator *iterator, const QString &fileName,
+                                     const QgsLayoutExporter::PdfExportSettings &settings,
+                                     QString &error /Out/, QgsFeedback *feedback = 0 );
+%Docstring
+Exports a layout ``iterator`` to a single PDF file, with the specified export ``settings``.
+
+The ``fileName`` argument gives the destination file name for the output PDF.
+
+Returns a result code indicating whether the export was successful or an
+error was encountered. If an error was obtained then ``error`` will be set
+to the error description.
+%End
+
+
 
     struct SvgExportSettings
     {

diff --git a/src/app/layout/qgslayoutatlaswidget.cpp b/src/app/layout/qgslayoutatlaswidget.cpp
@@ -172,9 +172,8 @@ void QgsLayoutAtlasWidget::mAtlasSingleFileCheckBox_stateChanged( int state )
     mAtlasFilenamePatternEdit->setEnabled( true );
     mAtlasFilenameExpressionButton->setEnabled( true );
   }
-#if 0 //TODO
-  mAtlas->setSingleFile( state == Qt::Checked );
-#endif
+
+  mLayout->setCustomProperty( QStringLiteral( "singleFile" ), state == Qt::Checked );
 }
 
 void QgsLayoutAtlasWidget::mAtlasSortFeatureCheckBox_stateChanged( int state )
@@ -334,11 +333,10 @@ void QgsLayoutAtlasWidget::updateGuiElements()
   mAtlasFilenamePatternEdit->setText( mAtlas->filenameExpression() );
   mAtlasHideCoverageCheckBox->setCheckState( mAtlas->hideCoverage() ? Qt::Checked : Qt::Unchecked );
 
-#if 0 //TODO
-  mAtlasSingleFileCheckBox->setCheckState( mAtlas->singleFile() ? Qt::Checked : Qt::Unchecked );
-  mAtlasFilenamePatternEdit->setEnabled( !mAtlas->singleFile() );
-  mAtlasFilenameExpressionButton->setEnabled( !mAtlas->singleFile() );
-#endif
+  bool singleFile = mLayout->customProperty( "singleFile", true ).toBool();
+  mAtlasSingleFileCheckBox->setCheckState( singleFile ? Qt::Checked : Qt::Unchecked );
+  mAtlasFilenamePatternEdit->setEnabled( !singleFile );
+  mAtlasFilenameExpressionButton->setEnabled( !singleFile );
 
   mAtlasSortFeatureCheckBox->setCheckState( mAtlas->sortFeatures() ? Qt::Checked : Qt::Unchecked );
   mAtlasSortFeatureDirectionButton->setEnabled( mAtlas->sortFeatures() );

diff --git a/src/app/layout/qgslayoutdesignerdialog.cpp b/src/app/layout/qgslayoutdesignerdialog.cpp
@@ -2291,8 +2291,172 @@ void QgsLayoutDesignerDialog::exportAtlasToSvg()
 
 void QgsLayoutDesignerDialog::exportAtlasToPdf()
 {
+  QgsLayoutAtlas *printAtlas = atlas();
+  if ( !printAtlas || !printAtlas->enabled() )
+    return;
+
   loadAtlasPredefinedScalesFromProject();
-//TODO
+  if ( containsWmsLayers() )
+  {
+    showWmsPrintingWarning();
+  }
+
+  if ( requiresRasterization() )
+  {
+    showRasterizationWarning();
+  }
+
+  if ( containsAdvancedEffects() && ( mLayout->customProperty( QStringLiteral( "forceVector" ), false ).toBool() ) )
+  {
+    showForceVectorWarning();
+  }
+
+  bool singleFile = mLayout->customProperty( QStringLiteral( "singleFile" ), true ).toBool();
+
+  QString outputFileName;
+  QgsSettings settings;
+  if ( singleFile )
+  {
+    QString lastUsedFile = settings.value( QStringLiteral( "UI/lastSaveAsPdfFile" ), QStringLiteral( "qgis.pdf" ) ).toString();
+    QFileInfo file( lastUsedFile );
+
+    QgsLayoutAtlas *printAtlas = atlas();
+    if ( printAtlas && printAtlas->enabled() && mActionAtlasPreview->isChecked() )
+    {
+      outputFileName = QDir( file.path() ).filePath( QgsFileUtils::stringToSafeFilename( printAtlas->currentFilename() ) + QStringLiteral( ".pdf" ) );
+    }
+    else
+    {
+      outputFileName = file.path() + '/' + QgsFileUtils::stringToSafeFilename( mLayout->name() ) + QStringLiteral( ".pdf" );
+    }
+
+#ifdef Q_OS_MAC
+    QgisApp::instance()->activateWindow();
+    this->raise();
+#endif
+    outputFileName = QFileDialog::getSaveFileName(
+                       this,
+                       tr( "Export to PDF" ),
+                       outputFileName,
+                       tr( "PDF Format" ) + " (*.pdf *.PDF)" );
+    this->activateWindow();
+    if ( outputFileName.isEmpty() )
+    {
+      return;
+    }
+
+    if ( !outputFileName.endsWith( QLatin1String( ".pdf" ), Qt::CaseInsensitive ) )
+    {
+      outputFileName += QLatin1String( ".pdf" );
+    }
+    settings.setValue( QStringLiteral( "UI/lastSaveAsPdfFile" ), outputFileName );
+  }
+
+  mView->setPaintingEnabled( false );
+  QApplication::setOverrideCursor( Qt::BusyCursor );
+
+  QgsLayoutExporter::PdfExportSettings pdfSettings;
+  pdfSettings.rasterizeWholeImage = mLayout->customProperty( QStringLiteral( "rasterize" ), false ).toBool();
+  pdfSettings.forceVectorOutput = mLayout->customProperty( QStringLiteral( "forceVector" ), false ).toBool();
+
+  QFileInfo fi( outputFileName );
+
+  QString error;
+  std::unique_ptr< QgsFeedback > feedback = qgis::make_unique< QgsFeedback >();
+  std::unique_ptr< QProgressDialog > progressDialog = qgis::make_unique< QProgressDialog >( tr( "Rendering maps..." ), tr( "Abort" ), 0, 100, this );
+  progressDialog->setWindowTitle( tr( "Exporting Atlas" ) );
+  connect( feedback.get(), &QgsFeedback::progressChanged, this, [ & ]( double progress )
+  {
+    progressDialog->setValue( progress );
+    progressDialog->setLabelText( feedback->property( "progress" ).toString() ) ;
+
+#ifdef Q_OS_LINUX
+    // For some reason on Windows hasPendingEvents() always return true,
+    // but one iteration is actually enough on Windows to get good interactivity
+    // whereas on Linux we must allow for far more iterations.
+    // For safety limit the number of iterations
+    int nIters = 0;
+    while ( QCoreApplication::hasPendingEvents() && ++nIters < 100 )
+#endif
+    {
+      QCoreApplication::processEvents();
+    }
+
+  } );
+  connect( progressDialog.get(), &QProgressDialog::canceled, this, [ & ]
+  {
+    feedback->cancel();
+  } );
+
+  QgsLayoutExporter::ExportResult result = QgsLayoutExporter::Success;
+  if ( singleFile )
+  {
+    result = QgsLayoutExporter::exportToPdf( printAtlas, outputFileName, pdfSettings, error, feedback.get() );
+  }
+  else
+  {
+
+  }
+
+  switch ( result )
+  {
+    case QgsLayoutExporter::Success:
+    {
+      if ( singleFile )
+      {
+        mMessageBar->pushMessage( tr( "Export atlas" ),
+                                  tr( "Successfully exported atlas to <a href=\"%1\">%2</a>" ).arg( QUrl::fromLocalFile( fi.path() ).toString(), outputFileName ),
+                                  QgsMessageBar::SUCCESS, 0 );
+      }
+      else
+      {
+        mMessageBar->pushMessage( tr( "Export atlas" ),
+                                  tr( "Successfully exported atlas to <a href=\"%1\">%2</a>" ).arg( QUrl::fromLocalFile( outputFileName ).toString(), outputFileName ),
+                                  QgsMessageBar::SUCCESS, 0 );
+      }
+      break;
+    }
+
+    case QgsLayoutExporter::FileError:
+      QMessageBox::warning( this, tr( "Export atlas" ),
+                            error, QMessageBox::Ok,
+                            QMessageBox::Ok );
+      break;
+
+    case QgsLayoutExporter::SvgLayerError:
+      // no meaning
+      break;
+
+    case QgsLayoutExporter::PrintError:
+      QMessageBox::warning( this, tr( "Export atlas" ),
+                            tr( "Could not create print device." ),
+                            QMessageBox::Ok,
+                            QMessageBox::Ok );
+      break;
+
+
+    case QgsLayoutExporter::MemoryError:
+      QMessageBox::warning( this, tr( "Memory Allocation Error" ),
+                            tr( "Exporting the PDF "
+                                "resulted in a memory overflow.\n\n"
+                                "Please try a lower resolution or a smaller paper size." ),
+                            QMessageBox::Ok, QMessageBox::Ok );
+      break;
+
+    case QgsLayoutExporter::IteratorError:
+      QMessageBox::warning( this, tr( "Atlas Export Error" ),
+                            tr( "Error encountered while exporting atlas" ),
+                            QMessageBox::Ok,
+                            QMessageBox::Ok );
+      break;
+
+    case QgsLayoutExporter::Canceled:
+      // no meaning here
+      break;
+  }
+
+  mView->setPaintingEnabled( true );
+  QApplication::restoreOverrideCursor();
 }
 
 void QgsLayoutDesignerDialog::paste()
@@ -2412,7 +2576,7 @@ void QgsLayoutDesignerDialog::createAtlasWidget()
 
   QgsPrintLayout *printLayout = qobject_cast< QgsPrintLayout * >( mLayout );
   QgsLayoutAtlas *atlas = printLayout->atlas();
-  QgsLayoutAtlasWidget *atlasWidget = new QgsLayoutAtlasWidget( mGeneralDock, printLayout );
+  QgsLayoutAtlasWidget *atlasWidget = new QgsLayoutAtlasWidget( mAtlasDock, printLayout );
   atlasWidget->setMessageBar( mMessageBar );
   mAtlasDock->setWidget( atlasWidget );
   mAtlasDock->show();