kmlexportplugin.cpp

Go to the documentation of this file.

/*
 * 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 3 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, see <http://www.gnu.org/licenses/>
 */

#include "kmlexportplugin.h"
#include <QDebug>
#include <QtPlugin>
#include <QStringList>
#include <QDir>
#include <QFileDialog>
#include <QList>
#include <QMessageBox>
#include <QWriteLocker>

#include <extensionsystem/pluginmanager.h>
#include <QKeySequence>
#include "uavobjects/uavobjectmanager.h"

#include "kmlexport.h"

KmlExportPlugin::KmlExportPlugin() {}

bool KmlExportPlugin::initialize(const QStringList &args, QString *errMsg)
{
    Q_UNUSED(args);
    Q_UNUSED(errMsg);

    // Add Menu entry
    Core::ActionManager *am = Core::ICore::instance()->actionManager();
    Core::ActionContainer *ac = am->actionContainer(Core::Constants::M_TOOLS);

    // Command to convert log file to KML
    exportToKmlCmd = am->registerAction(new QAction(this), "KmlExport.ExportToKML",
                                        QList<int>() << Core::Constants::C_GLOBAL_ID);
    exportToKmlCmd->action()->setText("Export logfile to KML");

    ac->menu()->addSeparator();
    ac->appendGroup("KML Export");
    ac->addAction(exportToKmlCmd, "KML Export");

    connect(exportToKmlCmd->action(), SIGNAL(triggered(bool)), this, SLOT(exportToKML()));

    return true;
}

void KmlExportPlugin::exportToKML()
{
    // Get input file
    //
    QString inputFileName = QFileDialog::getOpenFileName(
        static_cast<QWidget *>(Core::ICore::instance)()->mainWindow(), tr("Open file"), QString(""),
        tr("dRonin Log Files (*.drlog *.tll)"));

    if (inputFileName.isEmpty())
        return;

    // Set up input filter.
    QString filters = tr("Keyhole Markup Language (compressed) (*.kmz);; Keyhole Markup Language "
                         "(uncompressed) (*.kml)");
    bool proceed_flag = false;
    QString outputFileName;
    QString localizedOutputFileName;

    // Get output file. Suggest to user that output have same base name and location as input file.
    while (proceed_flag == false) {
        outputFileName = QFileDialog::getSaveFileName(
            static_cast<QWidget *>(Core::ICore::instance)()->mainWindow(), tr("Export log"),
            inputFileName.split(".", QString::SkipEmptyParts).at(0), filters);

        if (outputFileName.isEmpty()) {
            qDebug() << "No KML file name given.";
            return;
        } else if (QFileInfo(outputFileName).suffix() == NULL) {
            qDebug() << "No KML file extension: " << outputFileName;
            QMessageBox::critical(new QWidget(), "No file extension",
                                  "Filename must have .kml or .kmz extension.");
        } else if (QFileInfo(outputFileName).suffix().toLower() != "kml"
                   && QFileInfo(outputFileName).suffix().toLower() != "kmz") {
            qDebug() << "Incorrect KML file extension: " << QFileInfo(outputFileName).suffix();
            QMessageBox::critical(new QWidget(), "Incorrect file extension",
                                  "Filename must have .kml or .kmz extension.");
        } else if (outputFileName.toLocal8Bit() == (QByteArray)NULL) {
            qDebug() << "Unsupported characters in path: " << outputFileName;
            QMessageBox::critical(new QWidget(), "Unsupported characters",
                                  "Not all uni-code characters are supported. Please choose a path "
                                  "and file name that uses only the standard latin alphabet.");
        } else {
            proceed_flag = true;

            // Due to limitations in the KML library, we must convert the output file name into
            // local 8-bit characters.
            localizedOutputFileName = outputFileName.toLocal8Bit();
        }
    }

    // Create kmlExport instance, and trigger export
    KmlExport kmlExport(inputFileName, localizedOutputFileName);
    kmlExport.exportToKML();
}

void KmlExportPlugin::extensionsInitialized() {}

void KmlExportPlugin::shutdown()
{
    // Do nothing
}