[TD]refactor print routines out of MDIViewPage
This commit is contained in:
370
src/Mod/TechDraw/Gui/PagePrinter.cpp
Normal file
370
src/Mod/TechDraw/Gui/PagePrinter.cpp
Normal file
@@ -0,0 +1,370 @@
|
||||
/***************************************************************************
|
||||
* Copyright (c) 2023 WandererFan <wandererfan@gmail.com> *
|
||||
* *
|
||||
* This file is part of the FreeCAD CAx development system. *
|
||||
* *
|
||||
* This library is free software; you can redistribute it and/or *
|
||||
* modify it under the terms of the GNU Library General Public *
|
||||
* License as published by the Free Software Foundation; either *
|
||||
* version 2 of the License, or (at your option) any later version. *
|
||||
* *
|
||||
* This library is distributed in the hope that it will be useful, *
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
||||
* GNU Library General Public License for more details. *
|
||||
* *
|
||||
* You should have received a copy of the GNU Library General Public *
|
||||
* License along with this library; see the file COPYING.LIB. If not, *
|
||||
* write to the Free Software Foundation, Inc., 59 Temple Place, *
|
||||
* Suite 330, Boston, MA 02111-1307, USA *
|
||||
* *
|
||||
***************************************************************************/
|
||||
|
||||
#include "PreCompiled.h"
|
||||
|
||||
#ifndef _PreComp_
|
||||
#include <QApplication>
|
||||
#include <QMessageBox>
|
||||
#include <QPageLayout>
|
||||
#include <QPageSize>
|
||||
#include <QPaintEngine>
|
||||
#include <QPainter>
|
||||
#include <QPdfWriter>
|
||||
#include <QPrintDialog>
|
||||
#include <QPrintPreviewDialog>
|
||||
#include <QPrinter>
|
||||
#endif
|
||||
|
||||
#include <App/Application.h>
|
||||
#include <App/Document.h>
|
||||
#include <App/DocumentObject.h>
|
||||
#include <Base/Console.h>
|
||||
#include <Base/Stream.h>
|
||||
#include <Base/Tools.h>
|
||||
#include <Gui/Application.h>
|
||||
#include <Gui/Command.h>
|
||||
#include <Gui/Document.h>
|
||||
#include <Gui/ViewProvider.h>
|
||||
|
||||
#include <Mod/TechDraw/App/DrawPage.h>
|
||||
#include <Mod/TechDraw/App/DrawPagePy.h>
|
||||
#include <Mod/TechDraw/App/DrawTemplate.h>
|
||||
#include <Mod/TechDraw/App/Preferences.h>
|
||||
|
||||
#include "PagePrinter.h"
|
||||
#include "QGITemplate.h"
|
||||
#include "QGSPage.h"
|
||||
#include "Rez.h"
|
||||
#include "ViewProviderPage.h"
|
||||
|
||||
using namespace TechDrawGui;
|
||||
using namespace TechDraw;
|
||||
|
||||
/* TRANSLATOR TechDrawGui::PagePrinter */
|
||||
|
||||
//TYPESYSTEM_SOURCE_ABSTRACT(TechDrawGui::PagePrinter)
|
||||
|
||||
PagePrinter::PagePrinter(ViewProviderPage* pageVp)
|
||||
: m_vpPage(pageVp), m_orientation(QPageLayout::Landscape),
|
||||
m_paperSize(QPageSize::A4), m_pagewidth(0.0), m_pageheight(0.0)
|
||||
{
|
||||
}
|
||||
|
||||
void PagePrinter::setScene(QGSPage* scene)
|
||||
{
|
||||
m_scene = scene;
|
||||
}
|
||||
|
||||
void PagePrinter::setDocumentName(const std::string& name) { m_documentName = name; }
|
||||
|
||||
//**** printing routines
|
||||
|
||||
void PagePrinter::getPaperAttributes()
|
||||
{
|
||||
App::DocumentObject* obj = m_vpPage->getDrawPage()->Template.getValue();
|
||||
auto pageTemplate(dynamic_cast<TechDraw::DrawTemplate*>(obj));
|
||||
if (pageTemplate) {
|
||||
m_pagewidth = pageTemplate->Width.getValue();
|
||||
m_pageheight = pageTemplate->Height.getValue();
|
||||
}
|
||||
m_paperSize = QPageSize::id(QSizeF(m_pagewidth, m_pageheight), QPageSize::Millimeter,
|
||||
QPageSize::FuzzyOrientationMatch);
|
||||
if (m_pagewidth > m_pageheight) {
|
||||
m_orientation = QPageLayout::Landscape;
|
||||
}
|
||||
else {
|
||||
m_orientation = QPageLayout::Portrait;
|
||||
}
|
||||
}
|
||||
|
||||
void PagePrinter::printPdf(std::string file)
|
||||
{
|
||||
// Base::Console().Message("PP::printPdf(%s)\n", file.c_str());
|
||||
if (file.empty()) {
|
||||
Base::Console().Warning("PagePrinter - no file specified\n");
|
||||
return;
|
||||
}
|
||||
getPaperAttributes();
|
||||
|
||||
QString filename = QString::fromUtf8(file.data(), file.size());
|
||||
QPrinter printer(QPrinter::HighResolution);
|
||||
// setPdfVersion sets the printied PDF Version to comply with PDF/A-1b, more details under: https://www.kdab.com/creating-pdfa-documents-qt/
|
||||
// printer.setPdfVersion(QPagedPaintDevice::PdfVersion_A1b);
|
||||
printer.setFullPage(true);
|
||||
printer.setOutputFileName(filename);
|
||||
|
||||
if (m_paperSize == QPageSize::Ledger) {
|
||||
printer.setPageOrientation((QPageLayout::Orientation)(1 - m_orientation));//reverse 0/1
|
||||
}
|
||||
else {
|
||||
printer.setPageOrientation(m_orientation);
|
||||
}
|
||||
if (m_paperSize == QPageSize::Custom) {
|
||||
printer.setPageSize(QPageSize(QSizeF(m_pagewidth, m_pageheight), QPageSize::Millimeter));
|
||||
}
|
||||
else {
|
||||
printer.setPageSize(QPageSize(m_paperSize));
|
||||
}
|
||||
m_scene->setExportingPdf(true);
|
||||
print(&printer);
|
||||
m_scene->setExportingPdf(false);
|
||||
}
|
||||
|
||||
/// print the Page associated with the parent MDIViewPage
|
||||
void PagePrinter::print(QPrinter* printer)
|
||||
{
|
||||
// Base::Console().Message("PP::print(printer)\n");
|
||||
QPainter painter(printer);
|
||||
QPageLayout pageLayout = printer->pageLayout();
|
||||
|
||||
TechDraw::DrawPage* dp = m_vpPage->getDrawPage();
|
||||
double width = 297.0;//default to A4 Landscape 297 x 210
|
||||
double height = 210.0;
|
||||
setPageLayout(pageLayout, dp, width, height);
|
||||
printer->setPageLayout(pageLayout);
|
||||
|
||||
QRect targetRect = printer->pageLayout().fullRectPixels(printer->resolution());
|
||||
QRectF sourceRect(0.0, Rez::guiX(-height), Rez::guiX(width), Rez::guiX(height));
|
||||
if (!printer->outputFileName().isEmpty()) {
|
||||
// file name is not empty so we must be printing to pdf?
|
||||
m_scene->setExportingPdf(true);
|
||||
}
|
||||
renderPage(m_vpPage, painter, sourceRect, targetRect);
|
||||
painter.end();
|
||||
if (!printer->outputFileName().isEmpty()) {
|
||||
// file name is not empty so we must be printing to pdf?
|
||||
m_scene->setExportingPdf(false);
|
||||
}
|
||||
}
|
||||
|
||||
//static routine to print all pages in a document
|
||||
void PagePrinter::printAll(QPrinter* printer, App::Document* doc)
|
||||
{
|
||||
// Base::Console().Message("PP::printAll()\n");
|
||||
QPainter painter(printer);
|
||||
QPageLayout pageLayout = printer->pageLayout();
|
||||
bool firstTime = true;
|
||||
std::vector<App::DocumentObject*> docObjs =
|
||||
doc->getObjectsOfType(TechDraw::DrawPage::getClassTypeId());
|
||||
for (auto& obj : docObjs) {
|
||||
Gui::ViewProvider* vp = Gui::Application::Instance->getViewProvider(obj);
|
||||
if (!vp) {
|
||||
continue;// can't print this one
|
||||
}
|
||||
TechDrawGui::ViewProviderPage* vpp = dynamic_cast<TechDrawGui::ViewProviderPage*>(vp);
|
||||
if (!vpp) {
|
||||
continue;// can't print this one
|
||||
}
|
||||
|
||||
TechDraw::DrawPage* dp = static_cast<TechDraw::DrawPage*>(obj);
|
||||
double width = 297.0;//default to A4 Landscape 297 x 210
|
||||
double height = 210.0;
|
||||
setPageLayout(pageLayout, dp, width, height);
|
||||
printer->setPageLayout(pageLayout);
|
||||
|
||||
//for some reason the first page doesn't obey the pageLayout, so we have to print
|
||||
//a sacrificial blank page, but we make it a feature instead of a bug by printing a
|
||||
//table of contents on the sacrificial page.
|
||||
if (firstTime) {
|
||||
firstTime = false;
|
||||
printBannerPage(printer, painter, pageLayout, doc, docObjs);
|
||||
}
|
||||
|
||||
printer->newPage();
|
||||
QRectF sourceRect(0.0, Rez::guiX(-height), Rez::guiX(width), Rez::guiX(height));
|
||||
QRect targetRect = printer->pageLayout().fullRectPixels(printer->resolution());
|
||||
|
||||
renderPage(vpp, painter, sourceRect, targetRect);
|
||||
}
|
||||
painter.end();
|
||||
}
|
||||
|
||||
//static routine to print all pages in a document to pdf
|
||||
void PagePrinter::printAllPdf(QPrinter* printer, App::Document* doc)
|
||||
{
|
||||
// Base::Console().Message("PP::printAllPdf()\n");
|
||||
// setPdfVersion sets the printied PDF Version to comply with PDF/A-1b, more details under: https://www.kdab.com/creating-pdfa-documents-qt/
|
||||
printer->setPdfVersion(QPagedPaintDevice::PdfVersion_A1b);
|
||||
// Base::Console().Message("PP::printAllPdf()\n");
|
||||
QString outputFile = printer->outputFileName();
|
||||
QString documentName = QString::fromUtf8(doc->getName());
|
||||
QPdfWriter pdfWriter(outputFile);
|
||||
// setPdfVersion sets the printied PDF Version to comply with PDF/A-1b, more details under: https://www.kdab.com/creating-pdfa-documents-qt/
|
||||
pdfWriter.setPdfVersion(QPagedPaintDevice::PdfVersion_A1b);
|
||||
pdfWriter.setTitle(documentName);
|
||||
pdfWriter.setResolution(printer->resolution());
|
||||
QPainter painter(&pdfWriter);
|
||||
QPageLayout pageLayout = printer->pageLayout();
|
||||
|
||||
double dpmm = printer->resolution() / 25.4;
|
||||
bool firstTime = true;
|
||||
std::vector<App::DocumentObject*> docObjs =
|
||||
doc->getObjectsOfType(TechDraw::DrawPage::getClassTypeId());
|
||||
for (auto& obj : docObjs) {
|
||||
Gui::ViewProvider* vp = Gui::Application::Instance->getViewProvider(obj);
|
||||
if (!vp) {
|
||||
continue;// can't print this one
|
||||
}
|
||||
TechDrawGui::ViewProviderPage* vpp = dynamic_cast<TechDrawGui::ViewProviderPage*>(vp);
|
||||
if (!vpp) {
|
||||
continue;// can't print this one
|
||||
}
|
||||
|
||||
TechDraw::DrawPage* dp = static_cast<TechDraw::DrawPage*>(obj);
|
||||
double width = 297.0;//default to A4 Landscape 297 x 210
|
||||
double height = 210.0;
|
||||
setPageLayout(pageLayout, dp, width, height);
|
||||
pdfWriter.setPageLayout(pageLayout);
|
||||
|
||||
//for some reason the first page doesn't obey the pageLayout, so we have to print
|
||||
//a sacrificial blank page, but we make it a feature instead of a bug by printing a
|
||||
//table of contents on the sacrificial page.
|
||||
if (firstTime) {
|
||||
firstTime = false;
|
||||
printBannerPage(printer, painter, pageLayout, doc, docObjs);
|
||||
}
|
||||
pdfWriter.newPage();
|
||||
|
||||
QRectF sourceRect(0.0, Rez::guiX(-height), Rez::guiX(width), Rez::guiX(height));
|
||||
QRect targetRect(0, 0, width * dpmm, height * dpmm);
|
||||
|
||||
renderPage(vpp, painter, sourceRect, targetRect);
|
||||
}
|
||||
painter.end();
|
||||
}
|
||||
|
||||
//static
|
||||
void PagePrinter::printBannerPage(QPrinter* printer, QPainter& painter, QPageLayout& pageLayout,
|
||||
App::Document* doc, std::vector<App::DocumentObject*>& docObjs)
|
||||
{
|
||||
QFont savePainterFont = painter.font();
|
||||
QFont painterFont;
|
||||
painterFont.setFamily(Preferences::labelFontQString());
|
||||
int fontSizeMM = Preferences::labelFontSizeMM();
|
||||
double dpmm = printer->resolution() / 25.4;
|
||||
int fontSizePx = fontSizeMM * dpmm;
|
||||
painterFont.setPixelSize(fontSizePx);
|
||||
painter.setFont(painterFont);
|
||||
|
||||
//print a header
|
||||
QString docLine = QObject::tr("Document Name: ") + QString::fromUtf8(doc->getName());
|
||||
int leftMargin = pageLayout.margins().left() * dpmm + 5 * dpmm; //layout margin + 5mm
|
||||
int verticalPos = pageLayout.margins().top() * dpmm + 20 * dpmm;//layout margin + 20mm
|
||||
int verticalSpacing = 2; //double space
|
||||
painter.drawText(leftMargin, verticalPos, docLine);
|
||||
|
||||
//leave some blank space between document name and page entries
|
||||
verticalPos += 2 * verticalSpacing * fontSizePx;
|
||||
for (auto& obj : docObjs) {
|
||||
//print a line for each page
|
||||
QString pageLine = QString::fromUtf8(obj->getNameInDocument()) + QString::fromUtf8(" / ")
|
||||
+ QString::fromUtf8(obj->Label.getValue());
|
||||
painter.drawText(leftMargin, verticalPos, pageLine);
|
||||
verticalPos += verticalSpacing * fontSizePx;
|
||||
}
|
||||
painter.setFont(savePainterFont);//restore the original font
|
||||
}
|
||||
|
||||
//static
|
||||
void PagePrinter::renderPage(ViewProviderPage* vpp, QPainter& painter, QRectF& sourceRect,
|
||||
QRect& targetRect)
|
||||
{
|
||||
// Base::Console().Message("PP::renderPage()\n");
|
||||
//turn off view frames for print
|
||||
bool saveState = vpp->getFrameState();
|
||||
vpp->setFrameState(false);
|
||||
vpp->setTemplateMarkers(false);
|
||||
|
||||
//scene might be drawn in light text. we need to redraw in normal text.
|
||||
bool saveLightOnDark = Preferences::lightOnDark();
|
||||
if (Preferences::lightOnDark()) {
|
||||
Preferences::lightOnDark(false);
|
||||
vpp->getQGSPage()->redrawAllViews();
|
||||
}
|
||||
|
||||
vpp->getQGSPage()->refreshViews();
|
||||
vpp->getQGSPage()->render(&painter, targetRect, sourceRect);
|
||||
|
||||
// Reset
|
||||
vpp->setFrameState(saveState);
|
||||
vpp->setTemplateMarkers(saveState);
|
||||
Preferences::lightOnDark(saveLightOnDark);
|
||||
|
||||
vpp->getQGSPage()->refreshViews();
|
||||
}
|
||||
|
||||
//static
|
||||
void PagePrinter::setPageLayout(QPageLayout& pageLayout, TechDraw::DrawPage* dPage, double& width,
|
||||
double& height)
|
||||
{
|
||||
auto pageTemplate(dynamic_cast<TechDraw::DrawTemplate*>(dPage->Template.getValue()));
|
||||
if (pageTemplate) {
|
||||
width = pageTemplate->Width.getValue();
|
||||
height = pageTemplate->Height.getValue();
|
||||
}
|
||||
//Qt's page size determination assumes Portrait orientation. To get the right paper size
|
||||
//we need to ask in the proper form.
|
||||
QPageSize::PageSizeId paperSizeID =
|
||||
QPageSize::id(QSizeF(std::min(width, height), std::max(width, height)),
|
||||
QPageSize::Millimeter, QPageSize::FuzzyOrientationMatch);
|
||||
if (paperSizeID == QPageSize::Custom) {
|
||||
pageLayout.setPageSize(QPageSize(QSizeF(std::min(width, height), std::max(width, height)),
|
||||
QPageSize::Millimeter));
|
||||
}
|
||||
else {
|
||||
pageLayout.setPageSize(QPageSize(paperSizeID));
|
||||
}
|
||||
pageLayout.setOrientation((QPageLayout::Orientation)dPage->getOrientation());
|
||||
}
|
||||
|
||||
void PagePrinter::saveSVG(std::string file)
|
||||
{
|
||||
if (file.empty()) {
|
||||
Base::Console().Warning("PagePrinter - no file specified\n");
|
||||
return;
|
||||
}
|
||||
QString filename = QString::fromUtf8(file.data(), file.size());
|
||||
if (m_scene) {
|
||||
m_scene->saveSvg(filename);
|
||||
}
|
||||
}
|
||||
|
||||
void PagePrinter::saveDXF(std::string fileName)
|
||||
{
|
||||
TechDraw::DrawPage* page = m_vpPage->getDrawPage();
|
||||
std::string PageName = page->getNameInDocument();
|
||||
fileName = Base::Tools::escapeEncodeFilename(fileName);
|
||||
Gui::Command::openCommand(QT_TRANSLATE_NOOP("Command", "Save page to dxf"));
|
||||
Gui::Command::doCommand(Gui::Command::Doc, "import TechDraw");
|
||||
Gui::Command::doCommand(Gui::Command::Doc,
|
||||
"TechDraw.writeDXFPage(App.activeDocument().%s, u\"%s\")",
|
||||
PageName.c_str(), (const char*)fileName.c_str());
|
||||
Gui::Command::commitCommand();
|
||||
}
|
||||
|
||||
void PagePrinter::savePDF(std::string file)
|
||||
{
|
||||
// Base::Console().Message("PP::savePDF(%s)\n", file.c_str());
|
||||
printPdf(file);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user