QGCParamWidget.cc 15.7 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
/*=====================================================================

QGroundControl Open Source Ground Control Station

(c) 2009, 2010 QGROUNDCONTROL/PIXHAWK PROJECT
<http://www.qgroundcontrol.org>
<http://pixhawk.ethz.ch>

This file is part of the QGROUNDCONTROL project

    QGROUNDCONTROL 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.

    QGROUNDCONTROL 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 QGROUNDCONTROL. If not, see <http://www.gnu.org/licenses/>.

======================================================================*/

/**
 * @file
 *   @brief Implementation of class QGCParamWidget
 *   @author Lorenz Meier <mail@qgroundcontrol.org>
 */

32 33
#include <QGridLayout>
#include <QPushButton>
34 35
#include <QFileDialog>
#include <QFile>
pixhawk's avatar
pixhawk committed
36 37 38 39

#include "QGCParamWidget.h"
#include "UASInterface.h"
#include <QDebug>
40
#include "QGC.h"
pixhawk's avatar
pixhawk committed
41

42 43 44 45
/**
 * @param uas MAV to set the parameters on
 * @param parent Parent widget
 */
pixhawk's avatar
pixhawk committed
46
QGCParamWidget::QGCParamWidget(UASInterface* uas, QWidget *parent) :
47 48
        QWidget(parent),
        mav(uas),
lm's avatar
lm committed
49
        components(new QMap<int, QTreeWidgetItem*>()),
pixhawk's avatar
pixhawk committed
50
        paramGroups(),
51 52
        changedValues(),
        parameters()
pixhawk's avatar
pixhawk committed
53 54 55
{
    // Create tree widget
    tree = new QTreeWidget(this);
56
    tree->setColumnWidth(0, 150);
pixhawk's avatar
pixhawk committed
57 58

    // Set tree widget as widget onto this component
59
    QGridLayout* horizontalLayout;
pixhawk's avatar
pixhawk committed
60
    //form->setAutoFillBackground(false);
61 62
    horizontalLayout = new QGridLayout(this);
    horizontalLayout->setSpacing(6);
pixhawk's avatar
pixhawk committed
63 64 65
    horizontalLayout->setMargin(0);
    horizontalLayout->setSizeConstraint(QLayout::SetMinimumSize);

66
    horizontalLayout->addWidget(tree, 0, 0, 1, 3);
67 68 69
    QPushButton* refreshButton = new QPushButton(tr("Refresh"));
    connect(refreshButton, SIGNAL(clicked()), this, SLOT(requestParameterList()));
    horizontalLayout->addWidget(refreshButton, 1, 0);
70

71
    QPushButton* setButton = new QPushButton(tr("Transmit"));
72 73 74
    connect(setButton, SIGNAL(clicked()), this, SLOT(setParameters()));
    horizontalLayout->addWidget(setButton, 1, 1);

75
    QPushButton* writeButton = new QPushButton(tr("Write (ROM)"));
76 77 78
    connect(writeButton, SIGNAL(clicked()), this, SLOT(writeParameters()));
    horizontalLayout->addWidget(writeButton, 1, 2);

79 80 81 82
    QPushButton* readButton = new QPushButton(tr("Read (ROM)"));
    connect(readButton, SIGNAL(clicked()), this, SLOT(readParameters()));
    horizontalLayout->addWidget(readButton, 2, 2);

83 84 85 86 87 88 89 90
    QPushButton* loadFileButton = new QPushButton(tr("Load File"));
    connect(loadFileButton, SIGNAL(clicked()), this, SLOT(loadParameters()));
    horizontalLayout->addWidget(loadFileButton, 2, 0);

    QPushButton* saveFileButton = new QPushButton(tr("Save File"));
    connect(saveFileButton, SIGNAL(clicked()), this, SLOT(saveParameters()));
    horizontalLayout->addWidget(saveFileButton, 2, 1);

91
    // Set layout
pixhawk's avatar
pixhawk committed
92 93 94 95 96 97 98 99
    this->setLayout(horizontalLayout);

    // Set header
    QStringList headerItems;
    headerItems.append("Parameter");
    headerItems.append("Value");
    tree->setHeaderLabels(headerItems);
    tree->setColumnCount(2);
100
    tree->setExpandsOnDoubleClick(true);
101 102 103

    // Connect signals/slots
    connect(this, SIGNAL(parameterChanged(int,QString,float)), mav, SLOT(setParameter(int,QString,float)));
104
    connect(tree, SIGNAL(itemChanged(QTreeWidgetItem*,int)), this, SLOT(parameterItemChanged(QTreeWidgetItem*,int)));
lm's avatar
lm committed
105

106 107
    // New parameters from UAS
    connect(uas, SIGNAL(parameterChanged(int,int,QString,float)), this, SLOT(addParameter(int,int,QString,float)));
pixhawk's avatar
pixhawk committed
108 109
}

110 111 112 113
/**
 * @return The MAV of this widget. Unless the MAV object has been destroyed, this
 *         pointer is never zero.
 */
pixhawk's avatar
pixhawk committed
114 115 116 117 118 119 120 121 122 123 124
UASInterface* QGCParamWidget::getUAS()
{
    return mav;
}

/**
 *
 * @param uas System which has the component
 * @param component id of the component
 * @param componentName human friendly name of the component
 */
125
void QGCParamWidget::addComponent(int uas, int component, QString componentName)
pixhawk's avatar
pixhawk committed
126
{
127
    Q_UNUSED(uas);
pixhawk's avatar
pixhawk committed
128
    if (components->contains(component))
pixhawk's avatar
pixhawk committed
129
    {
pixhawk's avatar
pixhawk committed
130 131 132 133 134 135 136 137 138 139 140 141 142 143
        // Update existing
        components->value(component)->setData(0, Qt::DisplayRole, componentName);
        components->value(component)->setData(1, Qt::DisplayRole, QString::number(component));
    }
    else
    {
        // Add new
        QStringList list;
        list.append(componentName);
        list.append(QString::number(component));
        QTreeWidgetItem* comp = new QTreeWidgetItem(list);
        components->insert(component, comp);
        // Create grouping and update maps
        paramGroups.insert(component, new QMap<QString, QTreeWidgetItem*>());
pixhawk's avatar
pixhawk committed
144 145
        tree->addTopLevelItem(comp);
        tree->update();
146 147 148 149 150 151 152 153 154 155
        // Create map in parameters
        if (!parameters.contains(component))
        {
            parameters.insert(component, new QMap<QString, float>());
        }
        // Create map in changed parameters
        if (!changedValues.contains(component))
        {
            changedValues.insert(component, new QMap<QString, float>());
        }
pixhawk's avatar
pixhawk committed
156 157 158
    }
}

159 160 161 162 163 164
/**
 * @param uas System which has the component
 * @param component id of the component
 * @param parameterName human friendly name of the parameter
 */
void QGCParamWidget::addParameter(int uas, int component, QString parameterName, float value)
pixhawk's avatar
pixhawk committed
165
{
166
    Q_UNUSED(uas);
167 168
    // Reference to item in tree
    QTreeWidgetItem* parameterItem;
pixhawk's avatar
pixhawk committed
169 170 171 172

    // Get component
    if (!components->contains(component))
    {
173
        addComponent(uas, component, "Component #" + QString::number(component));
pixhawk's avatar
pixhawk committed
174
    }
175

176 177 178 179 180 181 182
    // Replace value in map

    // FIXME
    if (parameters.value(component)->contains(parameterName)) parameters.value(component)->remove(parameterName);
    parameters.value(component)->insert(parameterName, value);


pixhawk's avatar
pixhawk committed
183 184 185
    QString splitToken = "_";
    // Check if auto-grouping can work
    if (parameterName.contains(splitToken))
186
    {
pixhawk's avatar
pixhawk committed
187 188 189
        QString parent = parameterName.section(splitToken, 0, 0, QString::SectionSkipEmpty);
        QMap<QString, QTreeWidgetItem*>* compParamGroups = paramGroups.value(component);
        if (!compParamGroups->contains(parent))
190
        {
pixhawk's avatar
pixhawk committed
191 192 193 194 195 196
            // Insert group item
            QStringList glist;
            glist.append(parent);
            QTreeWidgetItem* item = new QTreeWidgetItem(glist);
            compParamGroups->insert(parent, item);
            components->value(component)->addChild(item);
197
        }
198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223

        // Append child to group
        bool found = false;
        QTreeWidgetItem* parentItem = compParamGroups->value(parent);
        for (int i = 0; i < parentItem->childCount(); i++)
        {
            QTreeWidgetItem* child = parentItem->child(i);
            QString key = child->data(0, Qt::DisplayRole).toString();
            if (key == parameterName)
            {
                //qDebug() << "UPDATED CHILD";
                parameterItem = child;
                parameterItem->setData(1, Qt::DisplayRole, value);
                found = true;
            }
        }

        if (!found)
        {
            // Insert parameter into map
            QStringList plist;
            plist.append(parameterName);
            plist.append(QString::number(value));
            parameterItem = new QTreeWidgetItem(plist);

            compParamGroups->value(parent)->addChild(parameterItem);
lm's avatar
lm committed
224
            parameterItem->setFlags(parameterItem->flags() | Qt::ItemIsEditable);
225
        }
226
    }
pixhawk's avatar
pixhawk committed
227
    else
228
    {
pixhawk's avatar
pixhawk committed
229 230 231 232 233 234 235 236 237
        bool found = false;
        QTreeWidgetItem* parent = components->value(component);
        for (int i = 0; i < parent->childCount(); i++)
        {
            QTreeWidgetItem* child = parent->child(i);
            QString key = child->data(0, Qt::DisplayRole).toString();
            if (key == parameterName)
            {
                //qDebug() << "UPDATED CHILD";
238 239
                parameterItem = child;
                parameterItem->setData(1, Qt::DisplayRole, value);
pixhawk's avatar
pixhawk committed
240 241 242 243 244 245
                found = true;
            }
        }

        if (!found)
        {
246 247 248 249 250 251
            // Insert parameter into map
            QStringList plist;
            plist.append(parameterName);
            plist.append(QString::number(value));
            parameterItem = new QTreeWidgetItem(plist);

lm's avatar
lm committed
252 253
            components->value(component)->addChild(parameterItem);
            parameterItem->setFlags(parameterItem->flags() | Qt::ItemIsEditable);
pixhawk's avatar
pixhawk committed
254
        }
255
        //tree->expandAll();
256
    }
257
    // Reset background color
258 259 260
    parameterItem->setBackground(0, QBrush(QColor(0, 0, 0)));
    parameterItem->setBackground(1, Qt::NoBrush);
    //tree->update();
pixhawk's avatar
pixhawk committed
261
    if (changedValues.contains(component)) changedValues.value(component)->remove(parameterName);
pixhawk's avatar
pixhawk committed
262 263
}

264 265 266 267
/**
 * Send a request to deliver the list of onboard parameters
 * to the MAV.
 */
268 269
void QGCParamWidget::requestParameterList()
{
lm's avatar
lm committed
270
    // Clear view and request param list
271
    clear();
272
    parameters.clear();
lm's avatar
lm committed
273
    mav->requestParameters();
274 275
}

276
void QGCParamWidget::parameterItemChanged(QTreeWidgetItem* current, int column)
lm's avatar
lm committed
277
{
278
    if (current && column > 0)
lm's avatar
lm committed
279
    {
280 281 282 283 284 285 286 287 288 289 290 291 292
        QTreeWidgetItem* parent = current->parent();
        while (parent->parent() != NULL)
        {
            parent = parent->parent();
        }
        // Parent is now top-level component
        int key = components->key(parent);
        if (!changedValues.contains(key))
        {
            changedValues.insert(key, new QMap<QString, float>());
        }
        QMap<QString, float>* map = changedValues.value(key, NULL);
        if (map)
lm's avatar
lm committed
293
        {
294 295 296 297 298
            bool ok;
            QString str = current->data(0, Qt::DisplayRole).toString();
            float value = current->data(1, Qt::DisplayRole).toDouble(&ok);
            // Send parameter to MAV
            if (ok)
lm's avatar
lm committed
299
            {
300 301 302
                if (ok)
                {
                    qDebug() << "PARAM CHANGED: COMP:" << key << "KEY:" << str << "VALUE:" << value;
303
                    // Changed values list
304
                    if (map->contains(str)) map->remove(str);
305
                    map->insert(str, value);
306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378

                    // Check if the value was numerically changed
                    if (!parameters.value(key)->contains(str) || parameters.value(key)->value(str, 0.0f) != value)
                    {
                        current->setBackground(0, QBrush(QColor(QGC::colorGreen)));
                        current->setBackground(1, QBrush(QColor(QGC::colorGreen)));
                    }

                    // All parameters list
                    if (parameters.value(key)->contains(str)) parameters.value(key)->remove(str);
                    parameters.value(key)->insert(str, value);
                }
            }
        }
    }
}

void QGCParamWidget::saveParameters()
{    
    QString fileName = QFileDialog::getSaveFileName(this, tr("Save File"), "./parameters.txt", tr("Parameter File (*.txt)"));
    QFile file(fileName);
    if (!file.open(QIODevice::WriteOnly | QIODevice::Text))
    {
        return;
    }

    QTextStream in(&file);

    in << "# Onboard parameters for system " << mav->getUASName() << "\n";
    in << "#\n";
    in << "# MAV ID  COMPONENT ID  PARAM NAME  VALUE (FLOAT)\n";

    // Iterate through all components, through all parameters and emit them
    QMap<int, QMap<QString, float>*>::iterator i;
    for (i = parameters.begin(); i != parameters.end(); ++i)
    {
        // Iterate through the parameters of the component
        int compid = i.key();
        QMap<QString, float>* comp = i.value();
        {
            QMap<QString, float>::iterator j;
            for (j = comp->begin(); j != comp->end(); ++j)
            {
                in << mav->getUASID() << "\t" << compid << "\t" << j.key() << "\t" << j.value() << "\n";
                in.flush();
            }
        }
    }
    file.close();
}

void QGCParamWidget::loadParameters()
{
    QString fileName = QFileDialog::getOpenFileName(this, tr("Load File"), ".", tr("Parameter file (*.txt)"));
    QFile file(fileName);
    if (!file.open(QIODevice::ReadOnly | QIODevice::Text))
        return;

    // Clear list
    clear();

    QTextStream in(&file);
    while (!in.atEnd())
    {
        QString line = in.readLine();
        if (!line.startsWith("#"))
        {
            QStringList wpParams = line.split("\t");
            if (wpParams.size() == 4)
            {
                // Only load parameters for right mav
                if (mav->getUASID() == wpParams.at(0).toInt())
                {
pixhawk's avatar
pixhawk committed
379

lm's avatar
lm committed
380 381 382 383 384 385 386 387 388
                    bool changed = false;
                    int component = wpParams.at(1).toInt();
                    QString parameterName = wpParams.at(2);
                    if (!parameters.contains(component) || parameters.value(component)->value(parameterName, 0.0f) != (float)wpParams.at(3).toDouble())
                    {
                        changed = true;
                    }

                    // Set parameter value
389
                    addParameter(wpParams.at(0).toInt(), wpParams.at(1).toInt(), wpParams.at(2), wpParams.at(3).toDouble());
lm's avatar
lm committed
390 391

                    if (changed)
392
                    {
lm's avatar
lm committed
393 394
                        // Create changed values data structure if necessary
                        if (!changedValues.contains(wpParams.at(1).toInt()))
395
                        {
lm's avatar
lm committed
396
                            changedValues.insert(wpParams.at(1).toInt(), new QMap<QString, float>());
397 398
                        }

lm's avatar
lm committed
399 400 401 402 403 404 405 406 407 408 409
                        // Add to changed values
                        if (changedValues.value(wpParams.at(1).toInt())->contains(wpParams.at(2)))
                        {
                            changedValues.value(wpParams.at(1).toInt())->remove(wpParams.at(2));
                        }

                        changedValues.value(wpParams.at(1).toInt())->insert(wpParams.at(2), (float)wpParams.at(3).toDouble());

                        qDebug() << "MARKING COMP" << wpParams.at(1).toInt() << "PARAM" << wpParams.at(2) << "VALUE" << (float)wpParams.at(3).toDouble() << "AS CHANGED";

                        // Mark in UI
pixhawk's avatar
pixhawk committed
410

pixhawk's avatar
pixhawk committed
411

412
                    }
413
                }
lm's avatar
lm committed
414 415
            }
        }
416
    }
417 418
    file.close();

lm's avatar
lm committed
419 420
}

421

422 423 424 425 426 427 428
/**
 * @param component the subsystem which has the parameter
 * @param parameterName name of the parameter, as delivered by the system
 * @param value value of the parameter
 */
void QGCParamWidget::setParameter(int component, QString parameterName, float value)
{
429
    emit parameterChanged(component, parameterName, value);
430
    qDebug() << "SENT PARAM" << parameterName << value;
431 432
}

433 434 435
/**
 * Set all parameter in the parameter tree on the MAV
 */
436 437
void QGCParamWidget::setParameters()
{
438 439 440 441 442 443 444 445 446 447 448
    // Iterate through all components, through all parameters and emit them
    QMap<int, QMap<QString, float>*>::iterator i;
    for (i = changedValues.begin(); i != changedValues.end(); ++i)
    {
        // Iterate through the parameters of the component
        int compid = i.key();
        QMap<QString, float>* comp = i.value();
        {
            QMap<QString, float>::iterator j;
            for (j = comp->begin(); j != comp->end(); ++j)
            {
449
                setParameter(compid, j.key(), j.value());
450 451 452 453
            }
        }
    }

lm's avatar
lm committed
454
    changedValues.clear();
lm's avatar
lm committed
455
    qDebug() << __FILE__ << __LINE__ << "SETTING ALL PARAMETERS";
456 457
}

458 459 460 461
/**
 * Write the current onboard parameters from RAM into
 * permanent storage, e.g. EEPROM or harddisk
 */
462 463
void QGCParamWidget::writeParameters()
{
464 465 466 467 468 469
    mav->writeParametersToStorage();
}

void QGCParamWidget::readParameters()
{
    mav->readParametersFromStorage();
470 471
}

472 473 474
/**
 * Clear all data in the parameter widget
 */
pixhawk's avatar
pixhawk committed
475 476 477
void QGCParamWidget::clear()
{
    tree->clear();
lm's avatar
lm committed
478
    components->clear();
pixhawk's avatar
pixhawk committed
479
}