IncrementalPlot.cc 11 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
/*=====================================================================

QGroundControl Open Source Ground Control Station

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

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 IncrementalPlot
 *   @author Lorenz Meier <mavteam@student.ethz.ch>
 *
 */

31 32 33 34 35 36 37 38 39 40 41 42
#include <qwt_plot.h>
#include <qwt_plot_canvas.h>
#include <qwt_plot_curve.h>
#include <qwt_symbol.h>
#include <qwt_plot_layout.h>
#include <qwt_plot_grid.h>
#include <qwt_scale_engine.h>
#include "IncrementalPlot.h"
#include <Scrollbar.h>
#include <ScrollZoomer.h>
#include <float.h>
#include <qpaintengine.h>
43 44

#include <QDebug>
45 46

CurveData::CurveData():
47
    d_count(0)
48 49 50 51 52 53
{
}

void CurveData::append(double *x, double *y, int count)
{
    int newSize = ( (d_count + count) / 1000 + 1 ) * 1000;
54
    if ( newSize > size() ) {
55 56 57 58
        d_x.resize(newSize);
        d_y.resize(newSize);
    }

59
    for ( register int i = 0; i < count; i++ ) {
60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75
        d_x[d_count + i] = x[i];
        d_y[d_count + i] = y[i];
    }
    d_count += count;
}

int CurveData::count() const
{
    return d_count;
}

int CurveData::size() const
{
    return d_x.size();
}

76
const double* CurveData::x() const
77 78 79 80
{
    return d_x.data();
}

81
const double* CurveData::y() const
82 83 84 85 86
{
    return d_y.data();
}

IncrementalPlot::IncrementalPlot(QWidget *parent):
87
    ChartPlot(parent),
88
    symmetric(false)
89
{
90
    setStyleText("solid crosses");
91 92 93 94 95 96 97 98 99 100 101

    plotLayout()->setAlignCanvasToScales(true);

    QwtLinearScaleEngine* yScaleEngine = new QwtLinearScaleEngine();
    setAxisScaleEngine(QwtPlot::yLeft, yScaleEngine);

    setAxisAutoScale(xBottom);
    setAxisAutoScale(yLeft);

    resetScaling();

102
    legend = NULL;
103

104
    connect(this, SIGNAL(legendChecked(QwtPlotItem*,bool)), this, SLOT(handleLegendClick(QwtPlotItem*,bool)));
105 106 107 108 109 110 111
}

IncrementalPlot::~IncrementalPlot()
{

}

112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128
/**
 * @param symmetric true will enforce that both axes have the same interval,
 *        centered around the data plot. A circle will thus remain a circle if true,
 *        if set to false it might become an ellipse because of axis scaling.
 */
void IncrementalPlot::setSymmetric(bool symmetric)
{
    this->symmetric = symmetric;
    updateScale(); // Updates the scaling at replots
}

void IncrementalPlot::handleLegendClick(QwtPlotItem* item, bool on)
{
    item->setVisible(!on);
    replot();
}

129 130
void IncrementalPlot::showLegend(bool show)
{
131 132
    if (show) {
        if (legend == NULL) {
133 134
            legend = new QwtLegend;
            legend->setFrameStyle(QFrame::Box);
135
            legend->setItemMode(QwtLegend::CheckableItem);
136 137
        }
        insertLegend(legend, QwtPlot::RightLegend);
138
    } else {
139 140 141
        delete legend;
        legend = NULL;
    }
142
    updateScale(); // Updates the scaling at replots
143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159
}

/**
 * Set datapoint and line style. This interface is intented
 * to be directly connected to the UI and allows to parse
 * human-readable, textual descriptions into plot specs.
 *
 * Data points: Either "circles", "crosses" or the default "dots"
 * Lines: Either "dotted", ("solid"/"line") or no lines if not used
 *
 * No special formatting is needed, as long as the keywords are contained
 * in the string. Lower/uppercase is ignored as well.
 *
 * @param style Formatting string for line/data point style
 */
void IncrementalPlot::setStyleText(QString style)
{
160
    foreach (QwtPlotCurve* curve, curves) {
161
        // Style of datapoints
162
        if (style.toLower().contains("circles")) {
163
            curve->setSymbol(QwtSymbol(QwtSymbol::Ellipse,
164
                                       Qt::NoBrush, QPen(QBrush(curve->symbol().pen().color()), symbolWidth), QSize(6, 6)) );
165
        } else if (style.toLower().contains("crosses")) {
166
            curve->setSymbol(QwtSymbol(QwtSymbol::XCross,
167
                                       Qt::NoBrush, QPen(QBrush(curve->symbol().pen().color()), symbolWidth), QSize(5, 5)) );
168
        } else if (style.toLower().contains("rect")) {
169
            curve->setSymbol(QwtSymbol(QwtSymbol::Rect,
170
                                       Qt::NoBrush, QPen(QBrush(curve->symbol().pen().color()), symbolWidth), QSize(6, 6)) );
171
        } else if (style.toLower().contains("line")) { // Show no symbol
172 173
            curve->setSymbol(QwtSymbol(QwtSymbol::NoSymbol,
                                       Qt::NoBrush, QPen(QBrush(curve->symbol().pen().color()), symbolWidth), QSize(6, 6)) );
174 175 176
        }

        // Style of lines
177
        if (style.toLower().contains("dotted")) {
178 179 180
            curve->setPen(QPen(QBrush(curve->symbol().pen().color()), curveWidth, Qt::DotLine));
        } else if (style.toLower().contains("dashed")) {
            curve->setPen(QPen(QBrush(curve->symbol().pen().color()), curveWidth, Qt::DashLine));
181
        } else if (style.toLower().contains("line") || style.toLower().contains("solid")) {
182
            curve->setPen(QPen(QBrush(curve->symbol().pen().color()), curveWidth, Qt::SolidLine));
183
        } else {
184
            curve->setPen(QPen(QBrush(curve->symbol().pen().color()), curveWidth, Qt::NoPen));
185
        }
186
        curve->setStyle(QwtPlotCurve::Lines);
187

188
    }
189
    replot();
190 191
}

192 193 194 195
void IncrementalPlot::resetScaling()
{
    xmin = 0;
    xmax = 500;
196 197
    ymin = xmin;
    ymax = xmax;
198 199 200 201 202 203 204 205 206 207 208 209 210

    setAxisScale(xBottom, xmin+xmin*0.05, xmax+xmax*0.05);
    setAxisScale(yLeft, ymin+ymin*0.05, ymax+ymax*0.05);

    replot();

    // Make sure the first data access hits these
    xmin = DBL_MAX;
    xmax = DBL_MIN;
    ymin = DBL_MAX;
    ymax = DBL_MIN;
}

211 212 213 214 215 216
/**
 * Updates the scale calculation and re-plots the whole plot
 */
void IncrementalPlot::updateScale()
{
    const double margin = 0.05;
LM's avatar
LM committed
217 218 219 220
    double xMinRange = xmin-(qAbs(xmin*margin));
    double xMaxRange = xmax+(qAbs(xmax*margin));
    double yMinRange = ymin-(qAbs(ymin*margin));
    double yMaxRange = ymax+(qAbs(ymax*margin));
221
    if (symmetric) {
222 223 224 225 226 227 228 229 230 231
        double xRange = xMaxRange - xMinRange;
        double yRange = yMaxRange - yMinRange;

        // Get the aspect ratio of the plot
        float xSize = width();
        if (legend != NULL) xSize -= legend->width();
        float ySize = height();

        float aspectRatio = xSize / ySize;

232
        if (xRange > yRange) {
233
            double yCenter = yMinRange + yRange/2.0;
234
            double xCenter = xMinRange + xRange/2.0;
235 236
            yMinRange = yCenter - xRange/2.0;
            yMaxRange = yCenter + xRange/2.0;
237 238
            xMinRange = xCenter - (xRange*aspectRatio)/2.0;
            xMaxRange = xCenter + (xRange*aspectRatio)/2.0;
239
        } else {
240
            double xCenter = xMinRange + xRange/2.0;
241 242
            xMinRange = xCenter - (yRange*aspectRatio)/2.0;
            xMaxRange = xCenter + (yRange*aspectRatio)/2.0;
243 244 245 246 247 248 249
        }
    }
    setAxisScale(xBottom, xMinRange, xMaxRange);
    setAxisScale(yLeft, yMinRange, yMaxRange);
    zoomer->setZoomBase(true);
}

250 251 252 253 254 255 256 257 258
void IncrementalPlot::appendData(QString key, double x, double y)
{
    appendData(key, &x, &y, 1);
}

void IncrementalPlot::appendData(QString key, double *x, double *y, int size)
{
    CurveData* data;
    QwtPlotCurve* curve;
259
    if (!d_data.contains(key)) {
260 261
        data = new CurveData;
        d_data.insert(key, data);
262
    } else {
263 264 265
        data = d_data.value(key);
    }

266
    if (!curves.contains(key)) {
267
        curve = new QwtPlotCurve(key);
268
        curves.insert(key, curve);
269 270 271 272 273
        curve->setStyle(QwtPlotCurve::NoCurve);
        curve->setPaintAttribute(QwtPlotCurve::PaintFiltered);

        const QColor &c = getNextColor();
        curve->setSymbol(QwtSymbol(QwtSymbol::XCross,
274
                                   QBrush(c), QPen(c, symbolWidth), QSize(5, 5)) );
275 276

        curve->attach(this);
277
    } else {
278
        curve = curves.value(key);
279 280 281 282 283 284 285 286
    }

    data->append(x, y, size);
    curve->setRawData(data->x(), data->y(), data->count());

    bool scaleChanged = false;

    // Update scales
287 288
    for (int i = 0; i<size; i++) {
        if (x[i] < xmin) {
289 290 291
            xmin = x[i];
            scaleChanged = true;
        }
292
        if (x[i] > xmax) {
293 294 295 296
            xmax = x[i];
            scaleChanged = true;
        }

297
        if (y[i] < ymin) {
298 299 300
            ymin = y[i];
            scaleChanged = true;
        }
301
        if (y[i] > ymax) {
302 303 304 305 306 307 308 309 310 311 312 313 314
            ymax = y[i];
            scaleChanged = true;
        }
    }
    //    setAxisScale(xBottom, xmin+xmin*0.05, xmax+xmax*0.05);
    //    setAxisScale(yLeft, ymin+ymin*0.05, ymax+ymax*0.05);

    //#ifdef __GNUC__
    //#warning better use QwtData
    //#endif

    //replot();

315
    if(scaleChanged) {
316
        updateScale();
317
    } else {
318 319

        const bool cacheMode =
320
            canvas()->testPaintAttribute(QwtPlotCanvas::PaintCached);
321 322 323 324 325 326 327 328 329 330

#if QT_VERSION >= 0x040000 && defined(Q_WS_X11)
        // Even if not recommended by TrollTech, Qt::WA_PaintOutsidePaintEvent
        // works on X11. This has an tremendous effect on the performance..

        canvas()->setAttribute(Qt::WA_PaintOutsidePaintEvent, true);
#endif

        canvas()->setPaintAttribute(QwtPlotCanvas::PaintCached, false);
        // FIXME Check if here all curves should be drawn
331
        //        QwtPlotCurve* plotCurve;
332
        //        foreach(plotCurve, curves)
333 334 335
        //        {
        //            plotCurve->draw(0, curve->dataSize()-1);
        //        }
336 337 338 339 340 341 342

        curve->draw(curve->dataSize() - size, curve->dataSize() - 1);
        canvas()->setPaintAttribute(QwtPlotCanvas::PaintCached, cacheMode);

#if QT_VERSION >= 0x040000 && defined(Q_WS_X11)
        canvas()->setAttribute(Qt::WA_PaintOutsidePaintEvent, false);
#endif
343 344
    }
}
345

346 347 348 349 350 351
/**
 * @return Number of copied data points, 0 on failure
 */
int IncrementalPlot::data(QString key, double* r_x, double* r_y, int maxSize)
{
    int result = 0;
352
    if (d_data.contains(key)) {
353
        CurveData* d = d_data.value(key);
354
        if (maxSize >= d->count()) {
355 356 357
            result = d->count();
            memcpy(r_x, d->x(), sizeof(double) * d->count());
            memcpy(r_y, d->y(), sizeof(double) * d->count());
358
        } else {
359 360
            result = 0;
        }
361
    }
362 363 364 365 366 367 368 369 370 371 372
    return result;
}

/**
 * @param show true to show the grid, false else
 */
void IncrementalPlot::showGrid(bool show)
{
    grid->setVisible(show);
    replot();
}
373

374 375 376
bool IncrementalPlot::gridEnabled()
{
    return grid->isVisible();
377 378 379 380
}

void IncrementalPlot::removeData()
{
381
    foreach (QwtPlotCurve* curve, curves) {
pixhawk's avatar
pixhawk committed
382 383
        delete curve;
    }
384
    curves.clear();
pixhawk's avatar
pixhawk committed
385

386
    foreach (CurveData* data, d_data) {
pixhawk's avatar
pixhawk committed
387 388
        delete data;
    }
389 390 391 392
    d_data.clear();
    resetScaling();
    replot();
}