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 <QPrinter>
  25 #include <QPrintDialog>
  26
  27 PrintDialog::PrintDialog(QWidget *parent, Firmware * firmware, GeneralSettings & generalSettings, ModelData & model, const QString & filename) :
  28   QDialog(parent),
  29   firmware(firmware),
  30   generalSettings(generalSettings),
  31   model(model),
  32   printfilename(filename),
  33   ui(new Ui::PrintDialog),
  34   multimodelprinter(firmware)
  35 {
  36   ui->setupUi(this);
  37   setWindowIcon(CompanionIcon("print.png"));
  38   setWindowTitle(model.name);
  39   multimodelprinter.setModel(0, &model, &generalSettings);
  40   ui->textEdit->setHtml(multimodelprinter.print(ui->textEdit->document()));
  41   if (!printfilename.isEmpty()) {
  42     printToFile();
  43     QTimer::singleShot(0, this, SLOT(autoClose()));
  44   }
  45 }
  46
  47 void PrintDialog::closeEvent(QCloseEvent *event)
  48 {
  49 }
  50
  51 PrintDialog::~PrintDialog()
  52 {
  53   delete ui;
  54 }
  55
  56 void PrintDialog::on_printButton_clicked()
  57 {
  58   QPrinter printer;
  59   printer.setPageMargins(10.0,10.0,10.0,10.0,printer.Millimeter);
  60   QPrintDialog *dialog = new QPrintDialog(&printer, this);
  61   dialog->setWindowTitle(tr("Print Document"));
  62   if (dialog->exec() != QDialog::Accepted)
  63     return;
  64   ui->textEdit->print(&printer);
  65 }
  66
  67 void PrintDialog::on_printFileButton_clicked()
  68 {
  69   QString fn = QFileDialog::getSaveFileName(this,tr("Select PDF output file"),QString(),tr("ODF files (*.odt);;PDF Files(*.pdf);;HTML-Files (*.htm *.html);;All Files (*)"));
  70   if (fn.isEmpty())
  71     return;
  72   if (! (fn.endsWith(".odt", Qt::CaseInsensitive) || fn.endsWith(".pdf", Qt::CaseInsensitive) || fn.endsWith(".htm", Qt::CaseInsensitive) || fn.endsWith(".html", Qt::CaseInsensitive)) )
  73     fn += ".pdf"; // default
  74   if (fn.endsWith(".pdf", Qt::CaseInsensitive)) {
  75     QPrinter printer;
  76     printer.setPageMargins(10.0,10.0,10.0,10.0,printer.Millimeter);
  77     printer.setOutputFormat(QPrinter::PdfFormat);
  78     printer.setColorMode(QPrinter::Color);
  79     printer.setOutputFileName(fn);
  80     ui->textEdit->print(&printer);
  81   }
  82   else {
  83     QTextDocumentWriter writer(fn);
  84      writer.write(ui->textEdit->document());
  85   }
  86 }
  87
  88 void PrintDialog::printToFile()
  89 {
  90   if (printfilename.isEmpty())
  91     return;
  92   if (! (printfilename.endsWith(".odt", Qt::CaseInsensitive) || printfilename.endsWith(".pdf", Qt::CaseInsensitive) || printfilename.endsWith(".htm", Qt::CaseInsensitive) || printfilename.endsWith(".html", Qt::CaseInsensitive)) )
  93     printfilename += ".pdf"; // default
  94   if (printfilename.endsWith(".pdf", Qt::CaseInsensitive)) {
  95     QPrinter printer;
  96     printer.setPageMargins(10.0,10.0,10.0,10.0,printer.Millimeter);
  97     printer.setOutputFormat(QPrinter::PdfFormat);
  98     printer.setColorMode(QPrinter::Color);
  99     printer.setOutputFileName(printfilename);
 100     ui->textEdit->print(&printer);
 101   }
 102   else {
 103     QTextDocumentWriter writer(printfilename);
 104     writer.write(ui->textEdit->document());
 105   }
 106 }
 107
 108 void PrintDialog::autoClose()
 109 {
 110   this->close();
 111 }