ground/openpilotgcs/src/plugins/uavobjectwidgetutils/mixercurvewidget.cpp

   1 /**
   2  ******************************************************************************
   3  *
   4  * @file       mixercurvewidget.cpp
   5  * @author     The OpenPilot Team, http://www.openpilot.org Copyright (C) 2010.
   6  * @addtogroup GCSPlugins GCS Plugins
   7  * @{
   8  * @addtogroup UAVObjectWidgetUtils Plugin
   9  * @{
  10  * @brief Utility plugin for UAVObject to Widget relation management
  11  *****************************************************************************/
  12 /*
  13  * This program is free software; you can redistribute it and/or modify
  14  * it under the terms of the GNU General Public License as published by
  15  * the Free Software Foundation; either version 3 of the License, or
  16  * (at your option) any later version.
  17  *
  18  * This program is distributed in the hope that it will be useful, but
  19  * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  20  * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
  21  * for more details.
  22  *
  23  * You should have received a copy of the GNU General Public License along
  24  * with this program; if not, write to the Free Software Foundation, Inc.,
  25  * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  26  */
  27
  28 #include "mixercurvewidget.h"
  29 #include "mixercurveline.h"
  30 #include "mixercurvepoint.h"
  31
  32 #include <QObject>
  33 #include <QtGui>
  34 #include <QDebug>
  35
  36 /*
  37  * Initialize the widget
  38  */
  39 MixerCurveWidget::MixerCurveWidget(QWidget *parent) :
  40     QGraphicsView(parent), m_xAxisTextItem(0), m_yAxisTextItem(0)
  41 {
  42     // Create a layout, add a QGraphicsView and put the SVG inside.
  43     // The Mixer Curve widget looks like this:
  44     // |--------------------|
  45     // |                    |
  46     // |                    |
  47     // |     Graph  |
  48     // |                    |
  49     // |                    |
  50     // |                    |
  51     // |--------------------|
  52
  53
  54     setHorizontalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
  55     setVerticalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
  56     setRenderHint(QPainter::Antialiasing);
  57
  58     m_curveMin = 0.0;
  59     m_curveMax = 1.0;
  60
  61     setFrameStyle(QFrame::NoFrame);
  62     setStyleSheet("background:transparent");
  63     setRenderHint(QPainter::HighQualityAntialiasing, true);
  64     QGraphicsScene *scene  = new QGraphicsScene(this);
  65     QSvgRenderer *renderer = new QSvgRenderer();
  66     m_plot = new QGraphicsSvgItem();
  67     renderer->load(QString(":/uavobjectwidgetutils/images/curve-bg.svg"));
  68     m_plot->setSharedRenderer(renderer);
  69
  70     scene->addItem(m_plot);
  71     m_plot->setZValue(-1);
  72
  73     scene->setSceneRect(m_plot->boundingRect());
  74     setScene(scene);
  75
  76     setupXAxisLabel();
  77     setupYAxisLabel();
  78     initNodes(MixerCurveWidget::NODE_NUMELEM);
  79 }
  80
  81 MixerCurveWidget::~MixerCurveWidget()
  82 {
  83     while (!m_nodeList.isEmpty()) {
  84         delete m_nodeList.takeFirst();
  85     }
  86
  87     while (!m_edgeList.isEmpty()) {
  88         delete m_edgeList.takeFirst();
  89     }
  90     if (m_xAxisTextItem) {
  91         delete m_xAxisTextItem;
  92         m_xAxisTextItem = NULL;
  93     }
  94     if (m_yAxisTextItem) {
  95         delete m_yAxisTextItem;
  96         m_yAxisTextItem = NULL;
  97     }
  98 }
  99
 100 void MixerCurveWidget::setPositiveColor(QString color)
 101 {
 102     for (int i = 0; i < m_nodeList.count(); i++) {
 103         MixerNode *node = m_nodeList.at(i);
 104         node->setPositiveColor(color);
 105     }
 106 }
 107
 108 void MixerCurveWidget::setNegativeColor(QString color)
 109 {
 110     for (int i = 0; i < m_nodeList.count(); i++) {
 111         MixerNode *node = m_nodeList.at(i);
 112         node->setNegativeColor(color);
 113     }
 114 }
 115
 116 /**
 117    Init curve: create a (flat) curve with a specified number of points.
 118
 119    If a curve exists already, resets it.
 120    Points should be between 0 and 1.
 121  */
 122 void MixerCurveWidget::initCurve(const QList<double> *points)
 123 {
 124     if (points->length() < 2) {
 125         return; // We need at least 2 points on a curve!
 126     }
 127     // finally, set node positions
 128     setCurve(points);
 129 }
 130
 131 void MixerCurveWidget::initNodes(int numPoints)
 132 {
 133     // First of all, clear any existing list
 134     if (m_nodeList.count()) {
 135         foreach(MixerNode * node, m_nodeList) {
 136             foreach(Edge * edge, node->edges()) {
 137                 if (edge->sourceNode() == node) {
 138                     scene()->removeItem(edge);
 139                     delete edge;
 140                 }
 141             }
 142             scene()->removeItem(node);
 143             delete node;
 144         }
 145
 146         m_nodeList.clear();
 147     }
 148
 149     // Create the nodes and edges
 150     MixerNode *prevNode = 0;
 151     for (int i = 0; i < numPoints; i++) {
 152         MixerNode *node = new MixerNode(this, m_plot);
 153
 154         m_nodeList.append(node);
 155         scene()->addItem(node);
 156
 157         node->setPos(0, 0);
 158
 159         if (prevNode) {
 160             scene()->addItem(new Edge(prevNode, node));
 161         }
 162
 163         prevNode = node;
 164     }
 165 }
 166
 167 void MixerCurveWidget::setupXAxisLabel()
 168 {
 169     if (!m_xAxisString.isEmpty()) {
 170         if (m_xAxisTextItem) {
 171             m_xAxisTextItem->setPlainText(m_xAxisString);
 172         } else {
 173             m_xAxisTextItem = new QGraphicsTextItem(m_xAxisString, m_plot);
 174             scene()->addItem(m_xAxisTextItem);
 175         }
 176     }
 177 }
 178
 179 void MixerCurveWidget::setupYAxisLabel()
 180 {
 181     if (!m_yAxisString.isEmpty()) {
 182         if (m_yAxisTextItem) {
 183             m_yAxisTextItem->setPlainText(m_yAxisString);
 184         } else {
 185             m_yAxisTextItem = new QGraphicsTextItem(m_yAxisString, m_plot);
 186             m_yAxisTextItem->setRotation(270);
 187             scene()->addItem(m_yAxisTextItem);
 188         }
 189     }
 190 }
 191
 192 /**
 193    Returns the current curve settings
 194  */
 195 QList<double> MixerCurveWidget::getCurve()
 196 {
 197     QList<double> list;
 198
 199     foreach(MixerNode * node, m_nodeList) {
 200         list.append(node->value());
 201     }
 202
 203     return list;
 204 }
 205 /**
 206    Sets a linear graph
 207  */
 208 void MixerCurveWidget::initLinearCurve(int numPoints, double maxValue, double minValue)
 209 {
 210     double range = maxValue - minValue; // setRange(minValue, maxValue);
 211
 212     QList<double> points;
 213     for (double i = 0; i < (double)numPoints; i++) {
 214         double val = (range * (i / (double)(numPoints - 1))) + minValue;
 215         points.append(val);
 216     }
 217     initCurve(&points);
 218 }
 219 /**
 220    Setd the current curve settings
 221  */
 222 void MixerCurveWidget::setCurve(const QList<double> *points)
 223 {
 224     m_curveUpdating = true;
 225
 226     int ptCnt = points->count();
 227     if (m_nodeList.count() != ptCnt) {
 228         initNodes(ptCnt);
 229     }
 230
 231     double range = m_curveMax - m_curveMin;
 232
 233     qreal w = m_plot->boundingRect().width() / (ptCnt - 1);
 234     qreal h = m_plot->boundingRect().height();
 235     for (int i = 0; i < ptCnt; i++) {
 236         double val = (points->at(i) < m_curveMin) ? m_curveMin : (points->at(i) > m_curveMax) ? m_curveMax : points->at(i);
 237
 238         val += range;
 239         val -= (m_curveMin + range);
 240         val /= range;
 241
 242         MixerNode *node = m_nodeList.at(i);
 243         node->setPos(w * i, h - (val * h));
 244         node->verticalMove(true);
 245
 246         node->update();
 247     }
 248     m_curveUpdating = false;
 249
 250     update();
 251
 252     emit curveUpdated();
 253 }
 254
 255 void MixerCurveWidget::showEvent(QShowEvent *event)
 256 {
 257     Q_UNUSED(event)
 258     positionAxisLabels();
 259     setSceneRect(scene()->itemsBoundingRect());
 260     fitInView(scene()->itemsBoundingRect(), Qt::KeepAspectRatio);
 261 }
 262
 263 void MixerCurveWidget::resizeEvent(QResizeEvent *event)
 264 {
 265     Q_UNUSED(event);
 266     positionAxisLabels();
 267     setSceneRect(scene()->itemsBoundingRect());
 268     fitInView(scene()->itemsBoundingRect(), Qt::KeepAspectRatio);
 269 }
 270
 271 void MixerCurveWidget::changeEvent(QEvent *event)
 272 {
 273     QGraphicsView::changeEvent(event);
 274
 275     if (event->type() == QEvent::EnabledChange) {
 276         foreach(MixerNode * node, m_nodeList) {
 277             node->update();
 278         }
 279     }
 280 }
 281
 282 void MixerCurveWidget::positionAxisLabels()
 283 {
 284     QRectF rect = m_plot->boundingRect();
 285
 286     if (m_xAxisTextItem) {
 287         m_xAxisTextItem->setPos(rect.right() -
 288                                 m_xAxisTextItem->boundingRect().width(), rect.bottom() - 4);
 289     }
 290
 291     if (m_yAxisTextItem) {
 292         m_yAxisTextItem->setPos(rect.left() -
 293                                 m_yAxisTextItem->boundingRect().height(), m_yAxisTextItem->boundingRect().width());
 294     }
 295 }
 296
 297 void MixerCurveWidget::itemMoved(double itemValue)
 298 {
 299     Q_UNUSED(itemValue);
 300
 301     if (!m_curveUpdating) {
 302         emit curveUpdated();
 303     }
 304 }
 305
 306 void MixerCurveWidget::setMin(double value)
 307 {
 308     if (m_curveMin != value) {
 309         emit curveMinChanged(value);
 310     }
 311
 312     m_curveMin = value;
 313 }
 314
 315 void MixerCurveWidget::setMax(double value)
 316 {
 317     if (m_curveMax != value) {
 318         emit curveMaxChanged(value);
 319     }
 320
 321     m_curveMax = value;
 322 }
 323
 324 double MixerCurveWidget::getMin()
 325 {
 326     return m_curveMin;
 327 }
 328 double MixerCurveWidget::getMax()
 329 {
 330     return m_curveMax;
 331 }
 332 double MixerCurveWidget::setRange(double min, double max)
 333 {
 334     m_curveMin = min;
 335     m_curveMax = max;
 336     return m_curveMax - m_curveMin;
 337 }
 338
 339 void MixerCurveWidget::setXAxisLabel(QString label)
 340 {
 341     m_xAxisString = label;
 342     setupXAxisLabel();
 343 }
 344
 345 void MixerCurveWidget::setYAxisLabel(QString label)
 346 {
 347     m_yAxisString = label;
 348     setupYAxisLabel();
 349 }