XMLCommProtocolWidget.cc 5.64 KB
Newer Older
pixhawk's avatar
pixhawk committed
1 2 3
#include <QFileDialog>
#include <QTextBrowser>
#include <QMessageBox>
4
#include <QSettings>
pixhawk's avatar
pixhawk committed
5 6 7 8

#include "XMLCommProtocolWidget.h"
#include "ui_XMLCommProtocolWidget.h"
#include "MAVLinkXMLParser.h"
9
#include "MAVLinkSyntaxHighlighter.h"
10
#include "QGC.h"
pixhawk's avatar
pixhawk committed
11 12 13 14 15

#include <QDebug>
#include <iostream>

XMLCommProtocolWidget::XMLCommProtocolWidget(QWidget *parent) :
16 17
    QWidget(parent),
    m_ui(new Ui::XMLCommProtocolWidget)
pixhawk's avatar
pixhawk committed
18 19 20
{
    m_ui->setupUi(this);

21 22 23
    // Now set syntax highlighter
    highlighter = new MAVLinkSyntaxHighlighter(m_ui->xmlTextView->document());

pixhawk's avatar
pixhawk committed
24 25 26 27
    connect(m_ui->selectFileButton, SIGNAL(clicked()), this, SLOT(selectXMLFile()));
    connect(m_ui->selectOutputButton, SIGNAL(clicked()), this, SLOT(selectOutputDirectory()));
    connect(m_ui->generateButton, SIGNAL(clicked()), this, SLOT(generate()));
    connect(m_ui->saveButton, SIGNAL(clicked()), this, SLOT(save()));
28 29

    this->setVisible(false);
pixhawk's avatar
pixhawk committed
30 31 32 33 34
}

void XMLCommProtocolWidget::selectXMLFile()
{
    //QString fileName = QFileDialog::getOpenFileName(this, tr("Load Protocol Definition File"), ".", "*.xml");
35
    QSettings settings(QGC::COMPANYNAME, QGC::APPNAME);
36 37
    const QString mavlinkXML = "MAVLINK_XML_FILE";
    QString dirPath = settings.value(mavlinkXML, QCoreApplication::applicationDirPath() + "../").toString();
38
    QFileInfo dir(dirPath);
pixhawk's avatar
pixhawk committed
39
    QFileDialog dialog;
40
    dialog.setDirectory(dir.absoluteDir());
pixhawk's avatar
pixhawk committed
41 42 43 44
    dialog.setFileMode(QFileDialog::AnyFile);
    dialog.setFilter(tr("MAVLink XML (*.xml)"));
    dialog.setViewMode(QFileDialog::Detail);
    QStringList fileNames;
45
    if (dialog.exec()) {
pixhawk's avatar
pixhawk committed
46 47 48
        fileNames = dialog.selectedFiles();
    }

49
    if (fileNames.size() > 0) {
pixhawk's avatar
pixhawk committed
50 51
        m_ui->fileNameLabel->setText(fileNames.first());
        QFile file(fileNames.first());
52

53
        if (file.open(QIODevice::ReadOnly | QIODevice::Text)) {
pixhawk's avatar
pixhawk committed
54 55
            const QString instanceText(QString::fromUtf8(file.readAll()));
            setXML(instanceText);
56 57
            // Store filename for next time
            settings.setValue(mavlinkXML, QFileInfo(file).absoluteFilePath());
58
            settings.sync();
59
        } else {
pixhawk's avatar
pixhawk committed
60
            QMessageBox msgBox;
61
            msgBox.setText("Could not read XML file. Permission denied");
pixhawk's avatar
pixhawk committed
62 63 64 65 66 67 68 69 70 71
            msgBox.exec();
        }
    }
}

void XMLCommProtocolWidget::setXML(const QString& xml)
{
    m_ui->xmlTextView->setText(xml);
    QDomDocument doc;

72
    if (doc.setContent(xml)) {
73
        m_ui->validXMLLabel->setText(tr("<font color=\"green\">Valid XML file</font>"));
74
    } else {
75
        m_ui->validXMLLabel->setText(tr("<font color=\"red\">File is NOT valid XML, please fix in editor</font>"));
pixhawk's avatar
pixhawk committed
76 77
    }

78
    if (model != NULL) {
pixhawk's avatar
pixhawk committed
79 80 81 82 83 84 85 86 87 88
        m_ui->xmlTreeView->reset();
        //delete model;
    }
    model = new DomModel(doc, this);
    m_ui->xmlTreeView->setModel(model);
    m_ui->xmlTreeView->repaint();
}

void XMLCommProtocolWidget::selectOutputDirectory()
{
89
    QSettings settings(QGC::COMPANYNAME, QGC::APPNAME);
90 91
    const QString mavlinkOutputDir = "MAVLINK_OUTPUT_DIR";
    QString dirPath = settings.value(mavlinkOutputDir, QCoreApplication::applicationDirPath() + "../").toString();
pixhawk's avatar
pixhawk committed
92
    QFileDialog dialog;
93
    dialog.setDirectory(dirPath);
pixhawk's avatar
pixhawk committed
94 95 96
    dialog.setFileMode(QFileDialog::Directory);
    dialog.setViewMode(QFileDialog::Detail);
    QStringList fileNames;
97
    if (dialog.exec()) {
pixhawk's avatar
pixhawk committed
98 99 100
        fileNames = dialog.selectedFiles();
    }

101
    if (fileNames.size() > 0) {
pixhawk's avatar
pixhawk committed
102
        m_ui->outputDirNameLabel->setText(fileNames.first());
103
        // Store directory for next time
104
        settings.setValue(mavlinkOutputDir, QFileInfo(fileNames.first()).absoluteFilePath());
105
        settings.sync();
pixhawk's avatar
pixhawk committed
106 107 108 109 110 111
        //QFile file(fileName);
    }
}

void XMLCommProtocolWidget::generate()
{
112
    // Check if input file is present
113
    if (!QFileInfo(m_ui->fileNameLabel->text().trimmed()).isFile()) {
114 115 116 117 118
        QMessageBox::critical(this, tr("Please select an XML input file first"), tr("You have to select an input XML file before generating C files."), QMessageBox::Ok);
        return;
    }

    // Check if output dir is selected
119
    if (!QFileInfo(m_ui->outputDirNameLabel->text().trimmed()).isDir()) {
120 121 122 123
        QMessageBox::critical(this, tr("Please select output directory first"), tr("You have to select an output directory before generating C files."), QMessageBox::Ok);
        return;
    }

pixhawk's avatar
pixhawk committed
124 125
    // First save file
    save();
126 127
    // Clean log
    m_ui->compileLog->clear();
128

129 130 131
    MAVLinkXMLParser* parser = new MAVLinkXMLParser(m_ui->fileNameLabel->text().trimmed(), m_ui->outputDirNameLabel->text().trimmed());
    connect(parser, SIGNAL(parseState(QString)), m_ui->compileLog, SLOT(appendHtml(QString)));
    bool result = parser->generate();
132
    if (result) {
pixhawk's avatar
pixhawk committed
133 134 135
        QMessageBox msgBox;
        msgBox.setText(QString("The C code / headers have been generated in folder\n%1").arg(m_ui->outputDirNameLabel->text().trimmed()));
        msgBox.exec();
136
    } else {
137
        QMessageBox::critical(this, tr("C code generation failed, please see the compile log for further information"), QString("The C code / headers could not be written to folder\n%1").arg(m_ui->outputDirNameLabel->text().trimmed()), QMessageBox::Ok);
pixhawk's avatar
pixhawk committed
138
    }
139
    delete parser;
pixhawk's avatar
pixhawk committed
140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166
}

void XMLCommProtocolWidget::save()
{
    QFile file(m_ui->fileNameLabel->text().trimmed());
    setXML(m_ui->xmlTextView->document()->toPlainText().toUtf8());
    file.open(QIODevice::WriteOnly | QIODevice::Text);
    file.write(m_ui->xmlTextView->document()->toPlainText().toUtf8());
}

XMLCommProtocolWidget::~XMLCommProtocolWidget()
{
    delete model;
    delete m_ui;
}

void XMLCommProtocolWidget::changeEvent(QEvent *e)
{
    QWidget::changeEvent(e);
    switch (e->type()) {
    case QEvent::LanguageChange:
        m_ui->retranslateUi(this);
        break;
    default:
        break;
    }
}