QGCParamWidget.cc 15.7 KB
Newer Older
1
2
3
4
/*=====================================================================

QGroundControl Open Source Ground Control Station

pixhawk's avatar
pixhawk committed
5
(c) 2009, 2010 QGROUNDCONTROL PROJECT <http://www.qgroundcontrol.org>
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

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>
 */

29
30
#include <QGridLayout>
#include <QPushButton>
31
32
#include <QFileDialog>
#include <QFile>
pixhawk's avatar
pixhawk committed
33
34
35
36

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

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

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

63
    horizontalLayout->addWidget(tree, 0, 0, 1, 3);
lm's avatar
lm committed
64
65
66
    QPushButton* refreshButton = new QPushButton(tr("Refresh"));
    connect(refreshButton, SIGNAL(clicked()), this, SLOT(requestParameterList()));
    horizontalLayout->addWidget(refreshButton, 1, 0);
67

lm's avatar
lm committed
68
    QPushButton* setButton = new QPushButton(tr("Transmit"));
69
70
71
    connect(setButton, SIGNAL(clicked()), this, SLOT(setParameters()));
    horizontalLayout->addWidget(setButton, 1, 1);

lm's avatar
lm committed
72
    QPushButton* writeButton = new QPushButton(tr("Write (ROM)"));
73
74
75
    connect(writeButton, SIGNAL(clicked()), this, SLOT(writeParameters()));
    horizontalLayout->addWidget(writeButton, 1, 2);

lm's avatar
lm committed
76
77
78
79
    QPushButton* readButton = new QPushButton(tr("Read (ROM)"));
    connect(readButton, SIGNAL(clicked()), this, SLOT(readParameters()));
    horizontalLayout->addWidget(readButton, 2, 2);

80
81
82
83
84
85
86
87
    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);

88
    // Set layout
pixhawk's avatar
pixhawk committed
89
90
91
92
93
94
95
96
    this->setLayout(horizontalLayout);

    // Set header
    QStringList headerItems;
    headerItems.append("Parameter");
    headerItems.append("Value");
    tree->setHeaderLabels(headerItems);
    tree->setColumnCount(2);
97
    tree->setExpandsOnDoubleClick(true);
98
99
100

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

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

107
108
109
110
/**
 * @return The MAV of this widget. Unless the MAV object has been destroyed, this
 *         pointer is never zero.
 */
pixhawk's avatar
pixhawk committed
111
112
113
114
115
116
117
118
119
120
121
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
 */
122
void QGCParamWidget::addComponent(int uas, int component, QString componentName)
pixhawk's avatar
pixhawk committed
123
{
124
    Q_UNUSED(uas);
pixhawk's avatar
pixhawk committed
125
    if (components->contains(component))
pixhawk's avatar
pixhawk committed
126
    {
pixhawk's avatar
pixhawk committed
127
128
129
130
131
132
133
134
135
136
137
138
139
140
        // 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
141
142
        tree->addTopLevelItem(comp);
        tree->update();
143
144
145
146
147
148
149
150
151
152
        // 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
153
154
155
    }
}

156
157
158
159
160
161
/**
 * @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
162
{
163
    Q_UNUSED(uas);
164
    // Reference to item in tree
165
    QTreeWidgetItem* parameterItem = NULL;
pixhawk's avatar
pixhawk committed
166
167
168
169

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

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

        // 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
221
            parameterItem->setFlags(parameterItem->flags() | Qt::ItemIsEditable);
222
        }
223
    }
pixhawk's avatar
pixhawk committed
224
    else
225
    {
pixhawk's avatar
pixhawk committed
226
227
228
229
230
231
232
233
234
        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";
235
236
                parameterItem = child;
                parameterItem->setData(1, Qt::DisplayRole, value);
pixhawk's avatar
pixhawk committed
237
238
239
240
241
242
                found = true;
            }
        }

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

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

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

274
void QGCParamWidget::parameterItemChanged(QTreeWidgetItem* current, int column)
lm's avatar
lm committed
275
{
276
    if (current && column > 0)
lm's avatar
lm committed
277
    {
278
279
280
281
282
283
284
285
286
287
288
289
290
        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
291
        {
292
293
294
295
296
            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
297
            {
298
299
300
                if (ok)
                {
                    qDebug() << "PARAM CHANGED: COMP:" << key << "KEY:" << str << "VALUE:" << value;
301
                    // Changed values list
302
                    if (map->contains(str)) map->remove(str);
303
                    map->insert(str, value);
304
305
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

                    // 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
377

lm's avatar
lm committed
378
379
380
381
382
383
384
385
386
                    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
387
                    addParameter(wpParams.at(0).toInt(), wpParams.at(1).toInt(), wpParams.at(2), wpParams.at(3).toDouble());
lm's avatar
lm committed
388
389

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

lm's avatar
lm committed
397
398
399
400
401
402
403
404
                        // 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());

405
                        //qDebug() << "MARKING COMP" << wpParams.at(1).toInt() << "PARAM" << wpParams.at(2) << "VALUE" << (float)wpParams.at(3).toDouble() << "AS CHANGED";
lm's avatar
lm committed
406
407

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

pixhawk's avatar
pixhawk committed
409

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

lm's avatar
lm committed
417
418
}

419

420
421
422
423
424
425
426
/**
 * @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)
{
427
    emit parameterChanged(component, parameterName, value);
428
    qDebug() << "SENT PARAM" << parameterName << value;
429
430
}

431
432
433
/**
 * Set all parameter in the parameter tree on the MAV
 */
434
435
void QGCParamWidget::setParameters()
{
436
437
438
439
440
441
442
443
444
445
446
    // 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)
            {
447
                setParameter(compid, j.key(), j.value());
448
449
450
451
            }
        }
    }

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

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

void QGCParamWidget::readParameters()
{
    mav->readParametersFromStorage();
468
469
}

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