LinechartWidget.cc 35.8 KB
Newer Older
pixhawk's avatar
pixhawk committed
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 32 33 34 35 36 37
/*=====================================================================

PIXHAWK Micro Air Vehicle Flying Robotics Toolkit

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

This file is part of the PIXHAWK project

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

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

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

/**
 * @file
 *   @brief Line chart plot widget
 *
 *   @author Lorenz Meier <mavteam@student.ethz.ch>
 *
 */

#include <QDebug>
#include <QWidget>
#include <QHBoxLayout>
#include <QGridLayout>
#include <QComboBox>
#include <QToolButton>
38
#include <QSizePolicy>
pixhawk's avatar
pixhawk committed
39 40 41 42 43 44 45
#include <QScrollBar>
#include <QLabel>
#include <QMenu>
#include <QSpinBox>
#include <QColor>
#include <QPalette>
#include <QFileDialog>
46 47
#include <QDesktopServices>
#include <QMessageBox>
pixhawk's avatar
pixhawk committed
48 49 50 51

#include "LinechartWidget.h"
#include "LinechartPlot.h"
#include "LogCompressor.h"
lm's avatar
lm committed
52
#include "MainWindow.h"
lm's avatar
lm committed
53
#include "QGC.h"
pixhawk's avatar
pixhawk committed
54 55 56
#include "MG.h"


57
LinechartWidget::LinechartWidget(int systemid, QWidget *parent) : QWidget(parent),
58 59 60 61 62 63 64 65 66 67 68 69 70 71 72
    sysid(systemid),
    activePlot(NULL),
    curvesLock(new QReadWriteLock()),
    plotWindowLock(),
    curveListIndex(0),
    curveListCounter(0),
    curveLabels(new QMap<QString, QLabel*>()),
    curveMeans(new QMap<QString, QLabel*>()),
    curveMedians(new QMap<QString, QLabel*>()),
    curveVariances(new QMap<QString, QLabel*>()),
    curveMenu(new QMenu(this)),
    logFile(new QFile()),
    logindex(1),
    logging(false),
    logStartTime(0),
lm's avatar
lm committed
73 74
    updateTimer(new QTimer()),
    selectedMAV(-1)
pixhawk's avatar
pixhawk committed
75 76 77
{
    // Add elements defined in Qt Designer
    ui.setupUi(this);
78
    this->setMinimumSize(200, 150);
pixhawk's avatar
pixhawk committed
79 80 81 82

    // Add and customize curve list elements (left side)
    curvesWidget = new QWidget(ui.curveListWidget);
    ui.curveListWidget->setWidget(curvesWidget);
83
    curvesWidgetLayout = new QGridLayout(curvesWidget);
pixhawk's avatar
pixhawk committed
84 85
    curvesWidgetLayout->setMargin(2);
    curvesWidgetLayout->setSpacing(4);
86
    //curvesWidgetLayout->setSizeConstraint(QSizePolicy::Expanding);
87
    curvesWidgetLayout->setAlignment(Qt::AlignTop);
88 89 90 91 92 93 94

    curvesWidgetLayout->setColumnStretch(0, 0);
    curvesWidgetLayout->setColumnStretch(1, 0);
    curvesWidgetLayout->setColumnStretch(2, 80);
    curvesWidgetLayout->setColumnStretch(3, 50);
    curvesWidgetLayout->setColumnStretch(4, 50);
    curvesWidgetLayout->setColumnStretch(5, 50);
95 96
//    horizontalLayout->setColumnStretch(median, 50);
    curvesWidgetLayout->setColumnStretch(6, 50);
97

pixhawk's avatar
pixhawk committed
98 99
    curvesWidget->setLayout(curvesWidgetLayout);

100 101 102 103 104 105
    // Create curve list headings
    QLabel* label;
    QLabel* value;
    QLabel* mean;
    QLabel* variance;

106 107
    connect(ui.recolorButton, SIGNAL(clicked()), this, SLOT(recolor()));
    connect(ui.shortNameCheckBox, SIGNAL(clicked(bool)), this, SLOT(setShortNames(bool)));
108

109 110
    int labelRow = curvesWidgetLayout->rowCount();

111 112 113
    selectAllCheckBox = new QCheckBox("", this);
    connect(selectAllCheckBox, SIGNAL(clicked(bool)), this, SLOT(selectAllCurves(bool)));
    curvesWidgetLayout->addWidget(selectAllCheckBox, labelRow, 0, 1, 2);
114 115

    label = new QLabel(this);
116
    label->setText("Name");
117
    curvesWidgetLayout->addWidget(label, labelRow, 2);
118 119

    // Value
120
    value = new QLabel(this);
121
    value->setText("Val");
122
    curvesWidgetLayout->addWidget(value, labelRow, 3);
123

124
    // Unit
125
    //curvesWidgetLayout->addWidget(new QLabel(tr("Unit")), labelRow, 4);
126

127
    // Mean
128
    mean = new QLabel(this);
129
    mean->setText("Mean");
130
    curvesWidgetLayout->addWidget(mean, labelRow, 5);
131 132

    // Variance
133
    variance = new QLabel(this);
134
    variance->setText("Variance");
135
    curvesWidgetLayout->addWidget(variance, labelRow, 6);
136

pixhawk's avatar
pixhawk committed
137 138
    // Create the layout
    createLayout();
139

140 141 142
    // And make sure we're listening for future style changes
    connect(MainWindow::instance(), SIGNAL(styleChanged()), this, SLOT(recolor()));

LM's avatar
LM committed
143
    updateTimer->setInterval(updateInterval);
144
    connect(updateTimer, SIGNAL(timeout()), this, SLOT(refresh()));
145
    connect(ui.uasSelectionBox, SIGNAL(currentIndexChanged(int)), this, SLOT(selectActiveSystem(int)));
146
    readSettings();
pixhawk's avatar
pixhawk committed
147 148
}

149 150 151
LinechartWidget::~LinechartWidget()
{
    writeSettings();
pixhawk's avatar
pixhawk committed
152
    stopLogging();
153 154
    if (activePlot) delete activePlot;
    activePlot = NULL;
pixhawk's avatar
pixhawk committed
155 156
}

157 158 159 160 161 162 163 164 165 166 167
void LinechartWidget::selectActiveSystem(int mav)
{
    // -1: Unitialized, 0: all
    if (mav != selectedMAV && (selectedMAV != -1))
    {
        // Delete all curves
        // FIXME
    }
    selectedMAV = mav;
}

168 169 170
void LinechartWidget::selectAllCurves(bool all)
{
    QMap<QString, QLabel*>::iterator i;
171
    for (i = curveLabels->begin(); i != curveLabels->end(); ++i) {
172 173 174 175
        activePlot->setVisible(i.key(), all);
    }
}

176 177 178 179
void LinechartWidget::writeSettings()
{
    QSettings settings;
    settings.beginGroup("LINECHART");
180 181
    bool enforceGT = (!autoGroundTimeSet && timeButton->isChecked()) ? true : false;
    if (timeButton) settings.setValue("ENFORCE_GROUNDTIME", enforceGT);
182
    if (ui.showUnitsCheckBox) settings.setValue("SHOW_UNITS", ui.showUnitsCheckBox->isChecked());
183
    if (ui.shortNameCheckBox) settings.setValue("SHORT_NAMES", ui.shortNameCheckBox->isChecked());
184 185 186 187 188 189 190 191 192
    settings.endGroup();
    settings.sync();
}

void LinechartWidget::readSettings()
{
    QSettings settings;
    settings.sync();
    settings.beginGroup("LINECHART");
193
    if (activePlot) {
194 195
        timeButton->setChecked(settings.value("ENFORCE_GROUNDTIME", timeButton->isChecked()).toBool());
        activePlot->enforceGroundTime(settings.value("ENFORCE_GROUNDTIME", timeButton->isChecked()).toBool());
196
        timeButton->setChecked(settings.value("ENFORCE_GROUNDTIME", timeButton->isChecked()).toBool());
197
        //userGroundTimeSet = settings.value("USER_GROUNDTIME", timeButton->isChecked()).toBool();
198
    }
199
    if (ui.showUnitsCheckBox) ui.showUnitsCheckBox->setChecked(settings.value("SHOW_UNITS", ui.showUnitsCheckBox->isChecked()).toBool());
200
    if (ui.shortNameCheckBox) ui.shortNameCheckBox->setChecked(settings.value("SHORT_NAMES", ui.shortNameCheckBox->isChecked()).toBool());
201 202 203
    settings.endGroup();
}

pixhawk's avatar
pixhawk committed
204 205 206 207 208 209
void LinechartWidget::createLayout()
{
    // Create actions
    createActions();

    // Setup the plot group box area layout
210 211 212
    QVBoxLayout* vlayout = new QVBoxLayout(ui.diagramGroupBox);
    vlayout->setSpacing(4);
    vlayout->setMargin(2);
pixhawk's avatar
pixhawk committed
213 214

    // Create plot container widget
215 216 217
    activePlot = new LinechartPlot(this, sysid);
    // Activate automatic scrolling
    activePlot->setAutoScroll(true);
pixhawk's avatar
pixhawk committed
218 219 220 221 222

    // TODO Proper Initialization needed
    //    activePlot = getPlot(0);
    //    plotContainer->setPlot(activePlot);

223 224 225 226
    vlayout->addWidget(activePlot);

    QHBoxLayout *hlayout = new QHBoxLayout;
    vlayout->addLayout(hlayout);
pixhawk's avatar
pixhawk committed
227 228 229

    // Logarithmic scaling button
    scalingLogButton = createButton(this);
230
    scalingLogButton->setText(tr("LOG"));
pixhawk's avatar
pixhawk committed
231
    scalingLogButton->setCheckable(true);
232 233
    scalingLogButton->setToolTip(tr("Set logarithmic scale for Y axis"));
    scalingLogButton->setWhatsThis(tr("Set logarithmic scale for Y axis"));
234
    hlayout->addWidget(scalingLogButton);
pixhawk's avatar
pixhawk committed
235 236 237

    // Averaging spin box
    averageSpinBox = new QSpinBox(this);
238 239
    averageSpinBox->setToolTip(tr("Sliding window size to calculate mean and variance"));
    averageSpinBox->setWhatsThis(tr("Sliding window size to calculate mean and variance"));
pixhawk's avatar
pixhawk committed
240
    averageSpinBox->setMinimum(2);
241 242
    averageSpinBox->setValue(200);
    setAverageWindow(200);
243
    averageSpinBox->setMaximum(9999);
244
    hlayout->addWidget(averageSpinBox);
pixhawk's avatar
pixhawk committed
245 246 247 248
    connect(averageSpinBox, SIGNAL(valueChanged(int)), this, SLOT(setAverageWindow(int)));

    // Log Button
    logButton = new QToolButton(this);
249 250
    logButton->setToolTip(tr("Start to log curve data into a CSV or TXT file"));
    logButton->setWhatsThis(tr("Start to log curve data into a CSV or TXT file"));
pixhawk's avatar
pixhawk committed
251
    logButton->setText(tr("Start Logging"));
252
    hlayout->addWidget(logButton);
pixhawk's avatar
pixhawk committed
253 254
    connect(logButton, SIGNAL(clicked()), this, SLOT(startLogging()));

255
    // Ground time button
256
    timeButton = new QCheckBox(this);
257
    timeButton->setText(tr("Ground Time"));
258 259
    timeButton->setToolTip(tr("Overwrite timestamp of data from vehicle with ground receive time. Helps if the plots are not visible because of missing or invalid onboard time."));
    timeButton->setWhatsThis(tr("Overwrite timestamp of data from vehicle with ground receive time. Helps if the plots are not visible because of missing or invalid onboard time."));
260
    hlayout->addWidget(timeButton);
261
    connect(timeButton, SIGNAL(clicked(bool)), activePlot, SLOT(enforceGroundTime(bool)));
262
    connect(timeButton, SIGNAL(clicked()), this, SLOT(writeSettings()));
263

264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286
    hlayout->addStretch();

    QLabel *timeScaleLabel = new QLabel("Time axis:");
    hlayout->addWidget(timeScaleLabel);

    timeScaleCmb = new QComboBox(this);
    timeScaleCmb->addItem("10 seconds", 10);
    timeScaleCmb->addItem("20 seconds", 20);
    timeScaleCmb->addItem("30 seconds", 30);
    timeScaleCmb->addItem("40 seconds", 40);
    timeScaleCmb->addItem("50 seconds", 50);
    timeScaleCmb->addItem("1 minute", 60);
    timeScaleCmb->addItem("2 minutes", 60*2);
    timeScaleCmb->addItem("3 minutes", 60*3);
    timeScaleCmb->addItem("4 minutes", 60*4);
    timeScaleCmb->addItem("5 minutes", 60*5);
    timeScaleCmb->addItem("10 minutes", 60*10);
    //timeScaleCmb->setSizeAdjustPolicy(QComboBox::AdjustToContents);
    timeScaleCmb->setMinimumContentsLength(12);

    hlayout->addWidget(timeScaleCmb);
    connect(timeScaleCmb, SIGNAL(currentIndexChanged(int)), this, SLOT(timeScaleChanged(int)));

287 288 289
    // Initialize the "Show units" checkbox. This is configured in the .ui file, so all
    // we do here is attach the clicked() signal.
    connect(ui.showUnitsCheckBox, SIGNAL(clicked()), this, SLOT(writeSettings()));
pixhawk's avatar
pixhawk committed
290

291 292 293 294 295 296 297
    // Add actions
    averageSpinBox->setValue(activePlot->getAverageWindow());

    // Connect notifications from the user interface to the plot
    connect(this, SIGNAL(curveRemoved(QString)), activePlot, SLOT(hideCurve(QString)));

    // Update scrollbar when plot window changes (via translator method setPlotWindowPosition()
298 299
//    connect(activePlot, SIGNAL(windowPositionChanged(quint64)), this, SLOT(setPlotWindowPosition(quint64)));
    connect(activePlot, SIGNAL(curveRemoved(QString)), this, SLOT(removeCurve(QString)));
300 301 302 303 304

    // Update plot when scrollbar is moved (via translator method setPlotWindowPosition()
    connect(this, SIGNAL(plotWindowPositionUpdated(quint64)), activePlot, SLOT(setWindowPosition(quint64)));

    // Set scaling
305 306 307
    connect(scalingLogButton, SIGNAL(toggled(bool)), this, SLOT(toggleLogarithmicScaling(bool)));
}

308 309 310 311 312
void LinechartWidget::timeScaleChanged(int index)
{
    activePlot->setPlotInterval(timeScaleCmb->itemData(index).toInt()*1000);
}

313 314 315 316 317 318
void LinechartWidget::toggleLogarithmicScaling(bool checked)
{
    if(checked)
        activePlot->setLogarithmicScaling();
    else
        activePlot->setLinearScaling();
pixhawk's avatar
pixhawk committed
319 320
}

321
void LinechartWidget::appendData(int uasId, const QString& curve, const QString& unit, qint8 value, quint64 usec)
322
{
323 324
    appendData(uasId, curve, unit, static_cast<qint64>(value), usec);
}
325

326 327 328 329
void LinechartWidget::appendData(int uasId, const QString& curve, const QString& unit, quint8 value, quint64 usec)
{
    appendData(uasId, curve, unit, static_cast<quint64>(value), usec);
}
330

331 332 333
void LinechartWidget::appendData(int uasId, const QString& curve, const QString& unit, qint16 value, quint64 usec)
{
    appendData(uasId, curve, unit, static_cast<qint64>(value), usec);
334 335
}

336 337 338 339
void LinechartWidget::appendData(int uasId, const QString& curve, const QString& unit, quint16 value, quint64 usec)
{
    appendData(uasId, curve, unit, static_cast<quint64>(value), usec);
}
340

341 342 343 344 345 346 347 348 349 350 351
void LinechartWidget::appendData(int uasId, const QString& curve, const QString& unit, qint32 value, quint64 usec)
{
    appendData(uasId, curve, unit, static_cast<qint64>(value), usec);
}

void LinechartWidget::appendData(int uasId, const QString& curve, const QString& unit, quint32 value, quint64 usec)
{
    appendData(uasId, curve, unit, static_cast<quint64>(value), usec);
}

void LinechartWidget::appendData(int uasId, const QString& curve, const QString& unit, qint64 value, quint64 usec)
352
{
lm's avatar
lm committed
353 354
    if ((selectedMAV == -1 && isVisible()) || (selectedMAV == uasId && isVisible()))
    {
355 356 357 358 359
        // Order matters here, first append to plot, then update curve list
        activePlot->appendData(curve+unit, usec, value);
        // Store data
        QLabel* label = curveLabels->value(curve+unit, NULL);
        // Make sure the curve will be created if it does not yet exist
lm's avatar
lm committed
360 361
        if(!label)
        {
362
            intData.insert(curve+unit, 0);
363 364
            addCurve(curve, unit);
        }
365 366 367

        // Add int data
        intData.insert(curve+unit, value);
368 369
    }

370 371 372 373 374
    if (lastTimestamp == 0 && usec != 0)
    {
        lastTimestamp = usec;
    } else if (usec != 0) {
        // Difference larger than 5 secs, enforce ground time
375
        if (((qint64)usec - (qint64)lastTimestamp) > 5000)
376 377 378 379 380 381
        {
            autoGroundTimeSet = true;
            if (activePlot) activePlot->groundTime();
        }
    }

382
    // Log data
lm's avatar
lm committed
383 384 385 386
    if (logging)
    {
        if (activePlot->isVisible(curve+unit))
        {
387
            if (usec == 0) usec = QGC::groundTimeMilliseconds();
388 389 390 391
            if (logStartTime == 0) logStartTime = usec;
            qint64 time = usec - logStartTime;
            if (time < 0) time = 0;

392
            logFile->write(QString(QString::number(time) + "\t" + QString::number(uasId) + "\t" + curve + "\t" + QString::number(value) + "\n").toLatin1());
393 394 395 396
        }
    }
}

397
void LinechartWidget::appendData(int uasId, const QString& curve, const QString& unit, quint64 value, quint64 usec)
lm's avatar
lm committed
398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415
{
    if ((selectedMAV == -1 && isVisible()) || (selectedMAV == uasId && isVisible()))
    {
        // Order matters here, first append to plot, then update curve list
        activePlot->appendData(curve+unit, usec, value);
        // Store data
        QLabel* label = curveLabels->value(curve+unit, NULL);
        // Make sure the curve will be created if it does not yet exist
        if(!label)
        {
            intData.insert(curve+unit, 0);
            addCurve(curve, unit);
        }

        // Add int data
        intData.insert(curve+unit, value);
    }

416 417 418 419 420 421 422 423 424 425 426 427
    if (lastTimestamp == 0 && usec != 0)
    {
        lastTimestamp = usec;
    } else if (usec != 0) {
        // Difference larger than 5 secs, enforce ground time
        if (abs((int)((qint64)usec - (quint64)lastTimestamp)) > 5000)
        {
            autoGroundTimeSet = true;
            if (activePlot) activePlot->groundTime();
        }
    }

lm's avatar
lm committed
428 429 430 431 432
    // Log data
    if (logging)
    {
        if (activePlot->isVisible(curve+unit))
        {
433
            if (usec == 0) usec = QGC::groundTimeMilliseconds();
lm's avatar
lm committed
434 435 436 437 438 439 440 441 442 443
            if (logStartTime == 0) logStartTime = usec;
            qint64 time = usec - logStartTime;
            if (time < 0) time = 0;

            logFile->write(QString(QString::number(time) + "\t" + QString::number(uasId) + "\t" + curve + "\t" + QString::number(value) + "\n").toLatin1());
            logFile->flush();
        }
    }
}

444
void LinechartWidget::appendData(int uasId, const QString& curve, const QString& unit, double value, quint64 usec)
lm's avatar
lm committed
445 446 447
{
    if ((selectedMAV == -1 && isVisible()) || (selectedMAV == uasId && isVisible()))
    {
448
        // Order matters here, first append to plot, then update curve list
449
        activePlot->appendData(curve+unit, usec, value);
450
        // Store data
451
        QLabel* label = curveLabels->value(curve+unit, NULL);
452
        // Make sure the curve will be created if it does not yet exist
lm's avatar
lm committed
453 454
        if(!label)
        {
455
            addCurve(curve, unit);
456
        }
pixhawk's avatar
pixhawk committed
457 458
    }

459 460 461 462 463 464 465 466 467 468 469 470
    if (lastTimestamp == 0 && usec != 0)
    {
        lastTimestamp = usec;
    } else if (usec != 0) {
        // Difference larger than 1 sec, enforce ground time
        if (abs((int)((qint64)usec - (quint64)lastTimestamp)) > 1000)
        {
            autoGroundTimeSet = true;
            if (activePlot) activePlot->groundTime();
        }
    }

pixhawk's avatar
pixhawk committed
471
    // Log data
lm's avatar
lm committed
472 473 474 475
    if (logging)
    {
        if (activePlot->isVisible(curve+unit))
        {
476
            if (usec == 0) usec = QGC::groundTimeMilliseconds();
477 478 479
            if (logStartTime == 0) logStartTime = usec;
            qint64 time = usec - logStartTime;
            if (time < 0) time = 0;
lm's avatar
lm committed
480

481
            logFile->write(QString(QString::number(time) + "\t" + QString::number(uasId) + "\t" + curve + "\t" + QString::number(value,'g',18) + "\n").toLatin1());
pixhawk's avatar
pixhawk committed
482 483 484 485 486
            logFile->flush();
        }
    }
}

487 488
void LinechartWidget::refresh()
{
LM's avatar
LM committed
489
    setUpdatesEnabled(false);
490
    QString str;
491
    // Value
492
    QMap<QString, QLabel*>::iterator i;
493 494
    for (i = curveLabels->begin(); i != curveLabels->end(); ++i) {
        if (intData.contains(i.key())) {
lm's avatar
lm committed
495
            str.sprintf("% 11i", intData.value(i.key()));
496
        } else {
lm's avatar
lm committed
497 498
            double val = activePlot->getCurrentValue(i.key());
            int intval = static_cast<int>(val);
499
            if (intval >= 100000 || intval <= -100000) {
lm's avatar
lm committed
500
                str.sprintf("% 11i", intval);
501
            } else if (intval >= 10000 || intval <= -10000) {
lm's avatar
lm committed
502
                str.sprintf("% 11.2f", val);
503
            } else if (intval >= 1000 || intval <= -1000) {
lm's avatar
lm committed
504
                str.sprintf("% 11.4f", val);
505
            } else {
lm's avatar
lm committed
506 507
                str.sprintf("% 11.6f", val);
            }
508
        }
509 510 511 512 513
        // Value
        i.value()->setText(str);
    }
    // Mean
    QMap<QString, QLabel*>::iterator j;
514
    for (j = curveMeans->begin(); j != curveMeans->end(); ++j) {
515
        double val = activePlot->getMean(j.key());
lm's avatar
lm committed
516
        int intval = static_cast<int>(val);
517
        if (intval >= 100000 || intval <= -100000) {
lm's avatar
lm committed
518
            str.sprintf("% 11i", intval);
519
        } else if (intval >= 10000 || intval <= -10000) {
520
            str.sprintf("% 11.2f", val);
521
        } else if (intval >= 1000 || intval <= -1000) {
lm's avatar
lm committed
522
            str.sprintf("% 11.4f", val);
523
        } else {
524 525
            str.sprintf("% 11.6f", val);
        }
526 527
        j.value()->setText(str);
    }
528 529 530 531 532 533 534
//    QMap<QString, QLabel*>::iterator k;
//    for (k = curveMedians->begin(); k != curveMedians->end(); ++k)
//    {
//        // Median
//        str.sprintf("%+.2f", activePlot->getMedian(k.key()));
//        k.value()->setText(str);
//    }
535
    QMap<QString, QLabel*>::iterator l;
536 537 538 539 540
    for (l = curveVariances->begin(); l != curveVariances->end(); ++l) {
        // Variance
        str.sprintf("% 8.3e", activePlot->getVariance(l.key()));
        l.value()->setText(str);
    }
LM's avatar
LM committed
541
    setUpdatesEnabled(true);
542 543
}

pixhawk's avatar
pixhawk committed
544 545 546 547

void LinechartWidget::startLogging()
{
    // Store reference to file
548 549
    // Append correct file ending if needed
    bool abort = false;
lm's avatar
lm committed
550 551

    // Check if any curve is enabled
552
    if (!activePlot->anyCurveVisible()) {
lm's avatar
lm committed
553 554 555 556 557 558 559 560 561 562 563
        QMessageBox msgBox;
        msgBox.setIcon(QMessageBox::Critical);
        msgBox.setText("No curves selected for logging.");
        msgBox.setInformativeText("Please check all curves you want to log. Currently no data would be logged, aborting the logging.");
        msgBox.setStandardButtons(QMessageBox::Ok);
        msgBox.setDefaultButton(QMessageBox::Ok);
        msgBox.exec();
        return;
    }

    // Let user select the log file name
564
    //QDate date(QDate::currentDate());
lm's avatar
lm committed
565
    // QString("./pixhawk-log-" + date.toString("yyyy-MM-dd") + "-" + QString::number(logindex) + ".log")
Lorenz Meier's avatar
Lorenz Meier committed
566
    QString fileName = QFileDialog::getSaveFileName(this, tr("Specify log file name"), QDesktopServices::storageLocation(QDesktopServices::DesktopLocation), tr("Logfile (*.log);;"));
lm's avatar
lm committed
567

Lorenz Meier's avatar
Lorenz Meier committed
568
    while (!(fileName.endsWith(".log")) && !abort && fileName != "") {
569 570 571
        QMessageBox msgBox;
        msgBox.setIcon(QMessageBox::Critical);
        msgBox.setText("Unsuitable file extension for logfile");
Lorenz Meier's avatar
Lorenz Meier committed
572
        msgBox.setInformativeText("Please choose .log as file extension. Click OK to change the file extension, cancel to not start logging.");
573 574
        msgBox.setStandardButtons(QMessageBox::Ok | QMessageBox::Cancel);
        msgBox.setDefaultButton(QMessageBox::Ok);
575 576
        if(msgBox.exec() != QMessageBox::Ok)
        {
577 578 579
            abort = true;
            break;
        }
Lorenz Meier's avatar
Lorenz Meier committed
580
        fileName = QFileDialog::getSaveFileName(this, tr("Specify log file name"), QDesktopServices::storageLocation(QDesktopServices::DesktopLocation), tr("Logfile (*.log);;"));
pixhawk's avatar
pixhawk committed
581
    }
582

583 584
    qDebug() << "SAVE FILE" << fileName;

585
    // Check if the user did not abort the file save dialog
586
    if (!abort && fileName != "") {
587
        logFile = new QFile(fileName);
Lorenz Meier's avatar
Lorenz Meier committed
588
        if (logFile->open(QIODevice::Truncate | QIODevice::WriteOnly | QIODevice::Text)) {
589
            logging = true;
590 591
            logStartTime = 0;
            curvesWidget->setEnabled(false);
592 593 594 595 596
            logindex++;
            logButton->setText(tr("Stop logging"));
            disconnect(logButton, SIGNAL(clicked()), this, SLOT(startLogging()));
            connect(logButton, SIGNAL(clicked()), this, SLOT(stopLogging()));
        }
pixhawk's avatar
pixhawk committed
597 598 599 600 601 602
    }
}

void LinechartWidget::stopLogging()
{
    logging = false;
603
    curvesWidget->setEnabled(true);
604
    if (logFile->isOpen()) {
pixhawk's avatar
pixhawk committed
605 606 607
        logFile->flush();
        logFile->close();
        // Postprocess log file
608
        compressor = new LogCompressor(logFile->fileName(), logFile->fileName());
609
        connect(compressor, SIGNAL(finishedFile(QString)), this, SIGNAL(logfileWritten(QString)));
lm's avatar
lm committed
610
        connect(compressor, SIGNAL(logProcessingStatusChanged(QString)), MainWindow::instance(), SLOT(showStatusMessage(QString)));
611 612 613 614

        QMessageBox msgBox;
        msgBox.setIcon(QMessageBox::Question);
        msgBox.setText(tr("Starting Log Compression"));
615
        msgBox.setInformativeText(tr("Should empty fields (e.g. due to packet drops) be filled with the previous value of the same variable (zero order hold)?"));
616 617 618 619 620 621 622 623 624 625 626 627 628 629
        msgBox.setStandardButtons(QMessageBox::Yes | QMessageBox::No);
        msgBox.setDefaultButton(QMessageBox::No);
        int ret = msgBox.exec();
        bool fill;
        if (ret == QMessageBox::Yes)
        {
            fill = true;
        }
        else
        {
            fill = false;
        }

        compressor->startCompression(fill);
pixhawk's avatar
pixhawk committed
630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656
    }
    logButton->setText(tr("Start logging"));
    disconnect(logButton, SIGNAL(clicked()), this, SLOT(stopLogging()));
    connect(logButton, SIGNAL(clicked()), this, SLOT(startLogging()));
}

/**
 * The average window size defines the width of the sliding average
 * filter. It also defines the width of the sliding median filter.
 *
 * @param windowSize with (in values) of the sliding average/median filter. Minimum is 2
 */
void LinechartWidget::setAverageWindow(int windowSize)
{
    if (windowSize > 1) activePlot->setAverageWindow(windowSize);
}

void LinechartWidget::createActions()
{
}

/**
 * @brief Add a curve to the curve list
 *
 * @param curve The id-string of the curve
 * @see removeCurve()
 **/
657
void LinechartWidget::addCurve(const QString& curve, const QString& unit)
pixhawk's avatar
pixhawk committed
658
{
659
    LinechartPlot* plot = activePlot;
660
//    QHBoxLayout *horizontalLayout;
pixhawk's avatar
pixhawk committed
661 662 663
    QCheckBox *checkBox;
    QLabel* label;
    QLabel* value;
664
    QLabel* unitLabel;
pixhawk's avatar
pixhawk committed
665
    QLabel* mean;
666
    QLabel* variance;
pixhawk's avatar
pixhawk committed
667

668 669
    curveNames.insert(curve+unit, curve);

670 671
    int labelRow = curvesWidgetLayout->rowCount();

672
    // Checkbox
673
    checkBox = new QCheckBox(this);
pixhawk's avatar
pixhawk committed
674
    checkBox->setCheckable(true);
675
    checkBox->setObjectName(curve+unit);
676 677
    checkBox->setToolTip(tr("Enable the curve in the graph window"));
    checkBox->setWhatsThis(tr("Enable the curve in the graph window"));
678
    curvesWidgetLayout->addWidget(checkBox, labelRow, 0);
pixhawk's avatar
pixhawk committed
679

680
    // Icon
681
    QWidget* colorIcon = new QWidget(this);
682
    colorIcons.insert(curve+unit, colorIcon);
pixhawk's avatar
pixhawk committed
683 684
    colorIcon->setMinimumSize(QSize(5, 14));
    colorIcon->setMaximumSize(4, 14);
685
    curvesWidgetLayout->addWidget(colorIcon, labelRow, 1);
pixhawk's avatar
pixhawk committed
686

687
    // Label
688
    label = new QLabel(this);
689
    label->setText(getCurveName(curve+unit, ui.shortNameCheckBox->isChecked()));
690
    curveNameLabels.insert(curve+unit, label);
691
    curvesWidgetLayout->addWidget(label, labelRow, 2);
692

pixhawk's avatar
pixhawk committed
693
    // Value
694
    value = new QLabel(this);
pixhawk's avatar
pixhawk committed
695
    value->setNum(0.00);
696
    value->setStyleSheet(QString("QLabel {font-family:\"Courier\"; font-weight: bold;}"));
697 698
    value->setToolTip(tr("Current value of %1 in %2 units").arg(curve, unit));
    value->setWhatsThis(tr("Current value of %1 in %2 units").arg(curve, unit));
699
    curveLabels->insert(curve+unit, value);
700
    curvesWidgetLayout->addWidget(value, labelRow, 3);
pixhawk's avatar
pixhawk committed
701

702 703 704 705 706 707
    // Unit
    unitLabel = new QLabel(this);
    unitLabel->setText(unit);
    unitLabel->setToolTip(tr("Unit of ") + curve);
    unitLabel->setWhatsThis(tr("Unit of ") + curve);
    curvesWidgetLayout->addWidget(unitLabel, labelRow, 4);
708 709
    unitLabel->setVisible(ui.showUnitsCheckBox->isChecked());
    connect(ui.showUnitsCheckBox, SIGNAL(clicked(bool)), unitLabel, SLOT(setVisible(bool)));
710

pixhawk's avatar
pixhawk committed
711
    // Mean
712
    mean = new QLabel(this);
pixhawk's avatar
pixhawk committed
713
    mean->setNum(0.00);
lm's avatar
lm committed
714
    mean->setStyleSheet(QString("QLabel {font-family:\"Courier\"; font-weight: bold;}"));
715 716
    mean->setToolTip(tr("Arithmetic mean of %1 in %2 units").arg(curve, unit));
    mean->setWhatsThis(tr("Arithmetic mean of %1 in %2 units").arg(curve, unit));
717 718
    curveMeans->insert(curve+unit, mean);
    curvesWidgetLayout->addWidget(mean, labelRow, 5);
pixhawk's avatar
pixhawk committed
719

720 721 722 723 724
//    // Median
//    median = new QLabel(form);
//    value->setNum(0.00);
//    curveMedians->insert(curve, median);
//    horizontalLayout->addWidget(median);
pixhawk's avatar
pixhawk committed
725

726
    // Variance
727
    variance = new QLabel(this);
728
    variance->setNum(0.00);
lm's avatar
lm committed
729
    variance->setStyleSheet(QString("QLabel {font-family:\"Courier\"; font-weight: bold;}"));
730 731
    variance->setToolTip(tr("Variance of %1 in (%2)^2 units").arg(curve, unit));
    variance->setWhatsThis(tr("Variance of %1 in (%2)^2 units").arg(curve, unit));
732 733
    curveVariances->insert(curve+unit, variance);
    curvesWidgetLayout->addWidget(variance, labelRow, 6);
734

pixhawk's avatar
pixhawk committed
735 736 737 738 739 740 741 742 743 744 745
    /* Color picker
    QColor color = QColorDialog::getColor(Qt::green, this);
         if (color.isValid()) {
             colorLabel->setText(color.name());
             colorLabel->setPalette(QPalette(color));
             colorLabel->setAutoFillBackground(true);
         }
        */

    // Set stretch factors so that the label gets the whole space

746 747 748 749

    // Load visibility settings
    // TODO

pixhawk's avatar
pixhawk committed
750
    // Connect actions
751
    connect(selectAllCheckBox, SIGNAL(clicked(bool)), checkBox, SLOT(setChecked(bool)));
pixhawk's avatar
pixhawk committed
752 753 754 755 756
    QObject::connect(checkBox, SIGNAL(clicked(bool)), this, SLOT(takeButtonClick(bool)));
    QObject::connect(this, SIGNAL(curveVisible(QString, bool)), plot, SLOT(setVisible(QString, bool)));

    // Set UI components to initial state
    checkBox->setChecked(false);
757
    plot->setVisible(curve+unit, false);
pixhawk's avatar
pixhawk committed
758 759 760 761 762 763 764 765
}

/**
 * @brief Remove the curve from the curve list.
 *
 * @param curve The curve to remove
 * @see addCurve()
 **/
766
void LinechartWidget::removeCurve(QString curve)
pixhawk's avatar
pixhawk committed
767
{
768
    Q_UNUSED(curve)
769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790

    QWidget* widget = NULL;
    widget = curveLabels->take(curve);
    curvesWidgetLayout->removeWidget(widget);
    widget->deleteLater();
    widget = curveMeans->take(curve);
    curvesWidgetLayout->removeWidget(widget);
    widget->deleteLater();
    widget = curveMedians->take(curve);
    curvesWidgetLayout->removeWidget(widget);
    widget->deleteLater();
    widget = curveVariances->take(curve);
    curvesWidgetLayout->removeWidget(widget);
    widget->deleteLater();
//    widget = colorIcons->take(curve);
//    curvesWidgetLayout->removeWidget(colorIcons->take(curve));
    widget->deleteLater();
//    intData->remove(curve);
}

void LinechartWidget::recolor()
{
791
    activePlot->styleChanged(MainWindow::instance()->getStyle());
792 793 794
    foreach (QString key, colorIcons.keys())
    {
        QWidget* colorIcon = colorIcons.value(key, 0);
795
        if (colorIcon && !colorIcon->styleSheet().isEmpty())
796
        {
797 798 799
            QString colorstyle;
            QColor color = activePlot->getColorForCurve(key);
            colorstyle = colorstyle.sprintf("QWidget { background-color: #%02X%02X%02X; }", color.red(), color.green(), color.blue());
800 801 802 803 804
            colorIcon->setStyleSheet(colorstyle);
        }
    }
}

805
QString LinechartWidget::getCurveName(const QString& key, bool shortEnabled)
806
{
807
    if (shortEnabled)
808 809
    {
        QString name;
810 811
        QStringList parts = curveNames.value(key).split(".");
        if (parts.length() > 1)
812
        {
813 814 815 816 817 818
            name = parts.at(1);
        }
        else
        {
            name = parts.at(0);
        }
819

820
        const int sizeLimit = 20;
821

822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843
        // Replace known words with abbreviations
        if (name.length() > sizeLimit)
        {
            name.replace("gyroscope", "gyro");
            name.replace("accelerometer", "acc");
            name.replace("magnetometer", "mag");
            name.replace("distance", "dist");
            name.replace("ailerons", "ail");
            name.replace("altitude", "alt");
            name.replace("waypoint", "wp");
            name.replace("throttle", "thr");
            name.replace("elevator", "elev");
            name.replace("rudder", "rud");
            name.replace("error", "err");
            name.replace("version", "ver");
            name.replace("message", "msg");
            name.replace("count", "cnt");
            name.replace("value", "val");
            name.replace("source", "src");
            name.replace("index", "idx");
            name.replace("type", "typ");
            name.replace("mode", "mod");
844
        }
845 846 847

        // Check if sub-part is still exceeding N chars
        if (name.length() > sizeLimit)
848
        {
849 850 851 852 853
            name.replace("a", "");
            name.replace("e", "");
            name.replace("i", "");
            name.replace("o", "");
            name.replace("u", "");
854
        }
855 856 857 858 859 860 861 862 863 864 865 866 867 868

        return name;
    }
    else
    {
        return curveNames.value(key);
    }
}

void LinechartWidget::setShortNames(bool enable)
{
    foreach (QString key, curveNames.keys())
    {
        curveNameLabels.value(key)->setText(getCurveName(key, enable));
869
    }
870
}
pixhawk's avatar
pixhawk committed
871

872 873 874
void LinechartWidget::showEvent(QShowEvent* event)
{
    Q_UNUSED(event);
875 876 877 878 879 880 881
    setActive(true);
}

void LinechartWidget::hideEvent(QHideEvent* event)
{
    Q_UNUSED(event);
    setActive(false);
882 883
}

884 885
void LinechartWidget::setActive(bool active)
{
886
    if (activePlot) {
887 888
        activePlot->setActive(active);
    }
889
    if (active) {
890
        updateTimer->start(updateInterval);
891
    } else {
892
        updateTimer->stop();
pixhawk's avatar
pixhawk committed
893 894 895 896 897 898 899 900 901 902 903
    }
}

/**
 * @brief Set the position of the plot window.
 * The plot covers only a portion of the complete time series. The scrollbar
 * allows to select a window of the time series. The right edge of the window is
 * defined proportional to the position of the scrollbar.
 *
 * @param scrollBarValue The value of the scrollbar, in the range from MIN_TIME_SCROLLBAR_VALUE to MAX_TIME_SCROLLBAR_VALUE
 **/
904 905
void LinechartWidget::setPlotWindowPosition(int scrollBarValue)
{
pixhawk's avatar
pixhawk committed
906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950
    plotWindowLock.lockForWrite();
    // Disable automatic scrolling immediately
    int scrollBarRange = (MAX_TIME_SCROLLBAR_VALUE - MIN_TIME_SCROLLBAR_VALUE);
    double position = (static_cast<double>(scrollBarValue) - MIN_TIME_SCROLLBAR_VALUE) / scrollBarRange;
    quint64 scrollInterval;

    // Activate automatic scrolling if scrollbar is at the right edge
    if(scrollBarValue > MAX_TIME_SCROLLBAR_VALUE - (MAX_TIME_SCROLLBAR_VALUE - MIN_TIME_SCROLLBAR_VALUE) * 0.01f) {
        activePlot->setAutoScroll(true);
    } else {
        activePlot->setAutoScroll(false);
        quint64 rightPosition;
        /* If the data exceeds the plot window, choose the position according to the scrollbar position */
        if(activePlot->getDataInterval() > activePlot->getPlotInterval()) {
            scrollInterval = activePlot->getDataInterval() - activePlot->getPlotInterval();
            rightPosition = activePlot->getMinTime() + activePlot->getPlotInterval() + (scrollInterval * position);
        } else {
            /* If the data interval is smaller as the plot interval, clamp the scrollbar to the right */
            rightPosition = activePlot->getMinTime() + activePlot->getPlotInterval();
        }
        emit plotWindowPositionUpdated(rightPosition);
    }


    // The slider position must be mapped onto an interval of datainterval - plotinterval,
    // because the slider position defines the right edge of the plot window. The leftmost
    // slider position must therefore map to the start of the data interval + plot interval
    // to ensure that the plot is not empty

    //  start> |-- plot interval --||-- (data interval - plotinterval) --| <end

    //@TODO Add notification of scrollbar here
    //plot->setWindowPosition(rightPosition);

    plotWindowLock.unlock();
}

/**
 * @brief Receive an updated plot window position.
 * The plot window can be changed by the arrival of new data or by
 * other user interaction. The scrollbar and other UI components
 * can be notified by calling this method.
 *
 * @param position The absolute position of the right edge of the plot window, in milliseconds
 **/
951 952
void LinechartWidget::setPlotWindowPosition(quint64 position)
{
pixhawk's avatar
pixhawk committed
953 954 955 956 957 958 959
    plotWindowLock.lockForWrite();
    // Calculate the relative position
    double pos;

    // A relative position makes only sense if the plot is filled
    if(activePlot->getDataInterval() > activePlot->getPlotInterval()) {
        //TODO @todo Implement the scrollbar enabling in a more elegant way
960
        //scrollbar->setDisabled(false);
pixhawk's avatar
pixhawk committed
961 962 963 964 965 966
        quint64 scrollInterval = position - activePlot->getMinTime() - activePlot->getPlotInterval();



        pos = (static_cast<double>(scrollInterval) / (activePlot->getDataInterval() - activePlot->getPlotInterval()));
    } else {
967
        //scrollbar->setDisabled(true);
pixhawk's avatar
pixhawk committed
968 969 970 971 972 973 974 975 976 977 978 979 980 981 982
        pos = 1;
    }
    plotWindowLock.unlock();

    emit plotWindowPositionUpdated(static_cast<int>(pos * (MAX_TIME_SCROLLBAR_VALUE - MIN_TIME_SCROLLBAR_VALUE)));
}

/**
 * @brief Set the time interval the plot displays.
 * The time interval of the plot can be adjusted by this method. If the
 * data covers less time than the interval, the plot will be filled from
 * the right to left
 *
 * @param interval The time interval to plot
 **/
983 984
void LinechartWidget::setPlotInterval(quint64 interval)
{
pixhawk's avatar
pixhawk committed
985 986 987 988 989
    activePlot->setPlotInterval(interval);
}

/**
 * @brief Take the click of a curve activation / deactivation button.
990 991 992
 * This method allows to map a button to a plot curve. The text of the
 * button must equal the curve name to activate / deactivate. If the checkbox
 * was clicked, show the curve color, otherwise clear the coloring.
pixhawk's avatar
pixhawk committed
993 994 995
 *
 * @param checked The visibility of the curve: true to display the curve, false otherwise
 **/
996 997
void LinechartWidget::takeButtonClick(bool checked)
{
pixhawk's avatar
pixhawk committed
998 999 1000

    QCheckBox* button = qobject_cast<QCheckBox*>(QObject::sender());

1001 1002
    if(button != NULL)
    {
pixhawk's avatar
pixhawk committed
1003
        activePlot->setVisible(button->objectName(), checked);
1004 1005
        QWidget* colorIcon = colorIcons.value(button->objectName(), 0);
        if (colorIcon)
1006
        {
1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017
            if (checked)
            {
                QColor color = activePlot->getColorForCurve(button->objectName());
                if (color.isValid())
                {
                    QString colorstyle;
                    colorstyle = colorstyle.sprintf("QWidget { background-color: #%02X%02X%02X; }", color.red(), color.green(), color.blue());
                    colorIcon->setStyleSheet(colorstyle);
                }
            }
            else
1018
            {
1019
                colorIcon->setStyleSheet("");
1020 1021
            }
        }
pixhawk's avatar
pixhawk committed
1022 1023 1024 1025 1026 1027 1028 1029 1030 1031
    }
}

/**
 * @brief Factory method to create a new button.
 *
 * @param imagename The name of the image (should be placed at the standard icon location)
 * @param text The button text
 * @param parent The parent object (to ensure that the memory is freed after the deletion of the button)
 **/
1032 1033
QToolButton* LinechartWidget::createButton(QWidget* parent)
{
pixhawk's avatar
pixhawk committed
1034 1035 1036 1037 1038 1039
    QToolButton* button = new QToolButton(parent);
    button->setMinimumSize(QSize(20, 20));
    button->setMaximumSize(60, 20);
    button->setGeometry(button->x(), button->y(), 20, 20);
    return button;
}