mainwindow.cpp

/*
  11/01/2022 https://github.com/su8/kurger
  This program is free software; you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation; either version 2 of the License, or
  (at your option) any later version.

  This program 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 General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with this program; if not, write to the Free Software
  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
  MA 02110-1301, USA.
*/
#include <iostream>
#include <string>
#include <cstdio>
#include <cstdlib>
#include <sys/stat.h>
#include <sys/types.h>

#include <QApplication>
#include <QFileDialog>
#include <QMessageBox>
#include <QIcon>
#include <QDesktopServices>
#include <QUrl>
/* Tray icon
#include <QSystemTrayIcon>
#include <QMenu>
#include <QAction>
#include <QCloseEvent>*/

#include "mainwindow.h"
#include "./ui_mainwindow.h"

static void RaiseMsg(const QString &msg1, const QString &msg2, unsigned short int isWarning);
static void pdf2img(const char *str);
static size_t indexLastSep(const char *str);

#define VLA 4999

static std::string PdfFile = "";
static char lastDirectory[VLA+1] = {'\0'};
static Ui::MainWindow *UI;
/* Tray icon
static QAction *exitAction;
static QMenu *trayIconMenu;
static QSystemTrayIcon *sysTrayIcon;*/

MainWindow::MainWindow(QWidget *parent)
    : QMainWindow(parent)
    , ui(new Ui::MainWindow)
{
    ui->setupUi(this);
    UI = ui;

    ui->comboBox->addItems({"png", "jpg", "bmp", "tiff"});
    ui->comboBox_2->addItems({"png16m", "pngalpha", "pnggray", "jpeg", "jpegcmyk", "jpeggray", "bmp16m", "bmpgray", "tiff24nc", "tiffgray"});

    /* Tray icon, based on https://amin-ahmadi.com/2019/06/20/implementing-minimize-to-tray-functionality-using-qt-c
    closing = 0U;

    exitAction = new QAction(tr("&Exit"), this);
    connect(exitAction, &QAction::triggered, [this]() { closing = 1U; close(); });

    trayIconMenu = new QMenu(this);
    trayIconMenu->addAction(exitAction);
    sysTrayIcon = new QSystemTrayIcon(this);
    sysTrayIcon->setContextMenu(trayIconMenu);
    sysTrayIcon->setIcon(QIcon("media/icon.xpm"));
    sysTrayIcon->show();

    connect(sysTrayIcon, &QSystemTrayIcon::activated, [this](auto reason)
    {
        if (reason == QSystemTrayIcon::Trigger)
        {
            if (isVisible()) { hide(); }
            else { show(); activateWindow(); }
        }
    });
*/
}

MainWindow::~MainWindow()
{
    //delete UI;

    /* Tray icon
    delete sysTrayIcon;
    delete trayIconMenu;
    delete exitAction; */

    delete ui;
}

int main(int argc, char *argv[])
{
    QApplication a(argc, argv);
    MainWindow w;
    w.show();
    return a.exec();
}

/* Tray icon
void MainWindow::closeEvent(QCloseEvent *event)
{
    if (closing == 1U)
    {
        event->accept();
    }
    else
    {
        this->hide();
        event->ignore();
    }
}
*/

void MainWindow::on_pushButton_clicked()
{
    if (UI->spinBox_2->value() > UI->spinBox_3->value())
    {
        RaiseMsg("Reversed Numbers", "From page can't be greater than To page.", 1U);
        return;
    }
    if (PdfFile == "")
    {
        RaiseMsg("No pdf selected", "You must select some PDF file first and then click this button.", 1U);
        return;
    }
    pdf2img(PdfFile.c_str());
}

void MainWindow::on_pushButton_2_clicked()
{
    char dialogFile[VLA+1] = {'\0'};

    if (UI->spinBox_2->value() > UI->spinBox_3->value())
    {
        RaiseMsg("Reversed Numbers", "From page can't be greater than To page.", 1U);
        return;
    }
    QString filename = QFileDialog::getOpenFileName(
                nullptr,
                QObject::tr("Open PDF file"),
                (*lastDirectory) ? lastDirectory : QDir::currentPath(),
                QObject::tr("PDF File (*.pdf)"));
    if ((!filename.isEmpty()) && (!filename.isNull()))
    {
        snprintf(dialogFile, VLA, "%s", filename.toStdString().c_str());
        snprintf(lastDirectory, VLA, "%s", dialogFile);
        lastDirectory[indexLastSep(dialogFile)] = '\0';
        PdfFile = filename.toStdString();
        pdf2img(dialogFile);
    }
}

static void RaiseMsg(const QString &msg1, const QString &msg2, unsigned short int isWarning)
{
    QMessageBox msg;
    msg.setText(msg2);
    msg.setIcon((isWarning == 1U) ? QMessageBox::Warning : QMessageBox::Information);
    msg.setWindowTitle(msg1);
    msg.setWindowIcon(static_cast<QIcon>("media/icon.xpm"));
    msg.exec();
}

void MainWindow::on_comboBox_currentIndexChanged(int index)
{
    unsigned short int sdevCombo = 0;

    if (index == 0)
    {
        sdevCombo = 0;
    }
    else if (index == 1)
    {
        sdevCombo = 3;
    }
    else
    {
        sdevCombo = (index == 2) ? 6 : 8;
    }
    UI->comboBox_2->setCurrentIndex(sdevCombo);
}

void MainWindow::on_comboBox_2_currentIndexChanged(int index)
{
    unsigned short int sdevCombo = 0, imageCombo = 0;

    if (index == 0)
        sdevCombo = 0;
    else if (index == 1)
        sdevCombo = 1;
    else if (index == 2)
        sdevCombo = 2;
    else    /* 'active' not in png group */
    {
        imageCombo = 1; /* The jpg group */
        if (index == 3)
            sdevCombo = 3;
        else if (index == 4)
            sdevCombo = 4;
        else if (index == 5)
            sdevCombo = 5;
        else    /* 'active' not in jpg group */
        {
            imageCombo = 2; /* The bmp group */
            if (index == 6)
                sdevCombo = 6;
            else if (index == 7)
                sdevCombo = 7;
            else    /* 'active' not in bmp group */
            {
                imageCombo = 3; /* The tiff group */
                sdevCombo = (index == 8) ? 8 : 9;
            }
        }
    }
    UI->comboBox->setCurrentIndex(imageCombo);
    UI->comboBox_2->setCurrentIndex(sdevCombo);
}

/*
   return the last separator '/' index number,
   we will use this index number to create our own
   `basename' alternative in C++
*/
static size_t indexLastSep(const char *str)
{
    const char *ptr = str;
    size_t sepIndex = 0;
    size_t x = 0;

    for (; *ptr; x++, ptr++)
    {
        if ('/' == *ptr)
        {
            sepIndex = x; /* keep in mind that we use loop */
        }
    }
    return sepIndex;
}

static void pdf2img(const char *str)
{
    char imageCombo[10] = {'\0'};
    char sdeviceCombo[10] = {'\0'};
    char pdfName[VLA+1] = {'\0'};
    char BaseName[VLA+1] = {'\0'};
    char params[VLA+1] = {'\0'};
    char ren1[VLA+1] = {'\0'};
    char ren2[VLA+1] = {'\0'};
    char created_dir[VLA+1] = {'\0'};
    int spin1 = UI->spinBox_2->value();
    int spin2 = UI->spinBox_3->value();
    int smallRange = (spin2 - spin1) + 2;
    int bigRange = spin1;
    int y = 0;
    size_t dirnameLen = indexLastSep(str);
    size_t x = 0;
    size_t z = 0;
    size_t fit2 = strlen(str) - 4; /* exclude the .pdf file extension */
    struct stat DiR;

    if (1850U <= fit2)
    {
        RaiseMsg("Warning!", "The given filename is too long!", 1U);
        return;
    }
    if (spin1 > spin2)
    {
        RaiseMsg("Reversed Numbers", "From page can't be greater than To page.", 1U);
        return;
    }

    snprintf(imageCombo, 10, "%s", UI->comboBox->currentText().toStdString().c_str());
    snprintf(sdeviceCombo, 10, "%s", UI->comboBox_2->currentText().toStdString().c_str());

    snprintf(pdfName, VLA, "%s", str);
    for (z = 0, x = dirnameLen+1; x < fit2; x++, z++)
    {
        BaseName[z] = pdfName[x]; /* /path/to/some.pdf -> some      */
    }
    BaseName[z] = '\0';

    if (250U < z)
    {
        RaiseMsg("Warning!", "The given filename is too long!", 1U);
        return;
    }

    snprintf(created_dir, VLA, "%s_converted", pdfName);
    stat(created_dir, &DiR);
    if (0 == S_ISDIR(DiR.st_mode))
    {
#if defined(__OpenBSD__) || defined(__linux__) || defined(__FreeBSD__) || defined(__NetBSD__)
        mkdir(created_dir, 0700);
#define GS "gs"
#else
        mkdir(created_dir);
#define GS "C:\\gs\\bin\\gswin64.exe"
#endif /*__OpenBSD__ || __linux__ || __FreeBSD__ || __NetBSD__ */
    }

    snprintf(params, VLA, GS " -dBATCH -dNOPAUSE -dQUIET -dFirstPage=%d -dLastPage=%d "
                    "-sOutputFile=\"%s\"_pAge_%%01d.%s -sDEVICE=%s -r%d "
                    "-dGraphicsAlphaBits=4 -sBandListStorage=memory "
                    "-dBufferSpace=99000 -dNumRenderingThreads=8 \"%s\"",
      spin1, spin2, pdfName, imageCombo, sdeviceCombo, UI->spinBox->value(), str);
    RaiseMsg("Please wait...", "Please wait until we convert the requested images.", 0U);
    system(params);

    for (y = 1; y < smallRange; y++, bigRange++)
    {
      snprintf(ren1, VLA, "%s_pAge_%d.%s", pdfName, y, imageCombo);
      snprintf(ren2, VLA, "%s/%s_page_%d.%s", created_dir, BaseName, bigRange, imageCombo);
      remove(ren2);
      rename(ren1, ren2);
    }
    QDesktopServices::openUrl(QUrl::fromLocalFile(static_cast<QString>(created_dir)));
}