implement updateLegend() in HistogramItem

qgis · Feb 22, 2014 · 4c1e866 · 4c1e866
1 parent 673f7ba
commit 4c1e866
Showing 1 changed file with 62 additions and 0 deletions.
diff --git a/src/gui/raster/qwt5_histogram_item.h b/src/gui/raster/qwt5_histogram_item.h
@@ -38,6 +38,8 @@ class HistogramItem: public QwtPlotItem
     virtual void draw(QPainter *, const QwtScaleMap &xMap, 
         const QwtScaleMap &yMap, const QRect &) const;
 
+    virtual void updateLegend(QwtLegend *) const;
+
     void setBaseline(double reference);
     double baseline() const;
 
@@ -67,6 +69,7 @@ class HistogramItem: public QwtPlotItem
 #include <qwt_interval_data.h>
 #include <qwt_painter.h>
 #include <qwt_scale_map.h>
+#include <qwt_legend_item.h>
 
 class HistogramItem::PrivateData
 {
@@ -344,4 +347,63 @@ void HistogramItem::drawBar(QPainter *painter,
    painter->restore();
 }
 
+//!  Update the widget that represents the curve on the legend
+// this was adapted from QwtPlotCurve::updateLegend()
+void HistogramItem::updateLegend(QwtLegend *legend) const
+{
+    if ( !legend )
+        return;
+
+    QwtPlotItem::updateLegend(legend);
+
+    QWidget *widget = legend->find(this);
+    if ( !widget || !widget->inherits("QwtLegendItem") )
+        return;
+
+    QwtLegendItem *legendItem = (QwtLegendItem *)widget;
+
+#if QT_VERSION < 0x040000
+    const bool doUpdate = legendItem->isUpdatesEnabled();
+#else
+    const bool doUpdate = legendItem->updatesEnabled();
+#endif
+    legendItem->setUpdatesEnabled(false);
+
+    const int policy = legend->displayPolicy();
+
+    if (policy == QwtLegend::FixedIdentifier)
+    {
+        int mode = legend->identifierMode();
+
+        legendItem->setCurvePen(QPen(color()));
+
+        if (mode & QwtLegendItem::ShowText)
+            legendItem->setText(title());
+        else
+            legendItem->setText(QwtText());
+
+        legendItem->setIdentifierMode(mode);
+    }
+    else if (policy == QwtLegend::AutoIdentifier)
+    {
+        int mode = 0;
+
+        legendItem->setCurvePen(QPen(color()));
+        mode |= QwtLegendItem::ShowLine;
+        if ( !title().isEmpty() )
+        {
+            legendItem->setText(title());
+            mode |= QwtLegendItem::ShowText;
+        }
+        else
+        {
+            legendItem->setText(QwtText());
+        }
+        legendItem->setIdentifierMode(mode);
+    }
+
+    legendItem->setUpdatesEnabled(doUpdate);
+    legendItem->update();
+}
+
 #endif