companion/src/printdialog.cpp

   1 /*
   2  * Copyright (C) OpenTX
   3  *
   4  * Based on code named
   5  *   th9x - http://code.google.com/p/th9x
   6  *   er9x - http://code.google.com/p/er9x
   7  *   gruvin9x - http://code.google.com/p/gruvin9x
   8  *
   9  * License GPLv2: http://www.gnu.org/licenses/gpl-2.0.html
  10  *
  11  * This program is free software; you can redistribute it and/or modify
  12  * it under the terms of the GNU General Public License version 2 as
  13  * published by the Free Software Foundation.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  */
  20
  21 #include "printdialog.h"
  22 #include "ui_printdialog.h"
  23 #include "helpers.h"
  24 #include "styleeditdialog.h"
  25 #include <QPrinter>
  26 #include <QPrintDialog>
  27
  28 PrintDialog::PrintDialog(QWidget *parent, Firmware * firmware, GeneralSettings & generalSettings, ModelData & model, const QString & filename) :
  29   QDialog(parent),
  30   firmware(firmware),
  31   generalSettings(generalSettings),
  32   model(model),
  33   printfilename(filename),
  34   ui(new Ui::PrintDialog),
  35   multiModelPrinter(firmware)
  36 {
  37   ui->setupUi(this);
  38   setWindowIcon(CompanionIcon("print.png"));
  39   setWindowTitle(model.name);
  40   multiModelPrinter.setModel(0, &model, &generalSettings);
  41   ui->textEdit->setHtml(multiModelPrinter.print(ui->textEdit->document()));
  42   if (!printfilename.isEmpty()) {
  43     printToFile();
  44     QTimer::singleShot(0, this, SLOT(autoClose()));
  45   }
  46 }
  47
  48 void PrintDialog::closeEvent(QCloseEvent *event)
  49 {
  50 }
  51
  52 PrintDialog::~PrintDialog()
  53 {
  54   delete ui;
  55 }
  56
  57 void PrintDialog::on_printButton_clicked()
  58 {
  59   QPrinter printer;
  60   printer.setPageMargins(10.0,10.0,10.0,10.0,printer.Millimeter);
  61   QPrintDialog *dialog = new QPrintDialog(&printer, this);
  62   dialog->setWindowTitle(tr("Print Document"));
  63   if (dialog->exec() != QDialog::Accepted)
  64     return;
  65   ui->textEdit->print(&printer);
  66 }
  67
  68 void PrintDialog::on_printFileButton_clicked()
  69 {
  70   QString fn = QFileDialog::getSaveFileName(this,tr("Select PDF output file"),QString(),tr("ODF files (*.odt);;PDF Files(*.pdf);;HTML-Files (*.htm *.html);;All Files (*)"));
  71   if (fn.isEmpty())
  72     return;
  73   if (! (fn.endsWith(".odt", Qt::CaseInsensitive) || fn.endsWith(".pdf", Qt::CaseInsensitive) || fn.endsWith(".htm", Qt::CaseInsensitive) || fn.endsWith(".html", Qt::CaseInsensitive)) )
  74     fn += ".pdf"; // default
  75   if (fn.endsWith(".pdf", Qt::CaseInsensitive)) {
  76     QPrinter printer;
  77     printer.setPageMargins(10.0,10.0,10.0,10.0,printer.Millimeter);
  78     printer.setOutputFormat(QPrinter::PdfFormat);
  79     printer.setColorMode(QPrinter::Color);
  80     printer.setOutputFileName(fn);
  81     ui->textEdit->print(&printer);
  82   }
  83   else {
  84     QTextDocumentWriter writer(fn);
  85      writer.write(ui->textEdit->document());
  86   }
  87 }
  88
  89 void PrintDialog::printToFile()
  90 {
  91   if (printfilename.isEmpty())
  92     return;
  93   if (! (printfilename.endsWith(".odt", Qt::CaseInsensitive) || printfilename.endsWith(".pdf", Qt::CaseInsensitive) || printfilename.endsWith(".htm", Qt::CaseInsensitive) || printfilename.endsWith(".html", Qt::CaseInsensitive)) )
  94     printfilename += ".pdf"; // default
  95   if (printfilename.endsWith(".pdf", Qt::CaseInsensitive)) {
  96     QPrinter printer;
  97     printer.setPageMargins(10.0,10.0,10.0,10.0,printer.Millimeter);
  98     printer.setOutputFormat(QPrinter::PdfFormat);
  99     printer.setColorMode(QPrinter::Color);
 100     printer.setOutputFileName(printfilename);
 101     ui->textEdit->print(&printer);
 102   }
 103   else {
 104     QTextDocumentWriter writer(printfilename);
 105     writer.write(ui->textEdit->document());
 106   }
 107 }
 108
 109 void PrintDialog::autoClose()
 110 {
 111   this->close();
 112 }
 113
 114 void PrintDialog::on_styleButton_clicked()
 115 {
 116   StyleEditDialog *g = new StyleEditDialog(this, MODEL_PRINT_CSS);
 117   if (g->exec() == QDialog::Accepted)
 118     ui->textEdit->setHtml(multiModelPrinter.print(ui->textEdit->document()));
 119 }