Merge pull request #2163 from carolinux/master

nyalldawson · nyalldawson · commit 7e8041cd9857 · 2015-06-24T10:05:56.000+10:00
Added setBand capability to qgssinglebandpseudocolorrenderer API
diff --git a/python/core/raster/qgssinglebandpseudocolorrenderer.sip b/python/core/raster/qgssinglebandpseudocolorrenderer.sip
@@ -29,6 +29,12 @@ class QgsSingleBandPseudoColorRenderer: QgsRasterRenderer
      */
     int band() const;
 
+     /** Sets the band used by the renderer.
+     * @see band
+     * @note added in QGIS 2.10
+     */
+    void setBand(int bandNo);
+
     double classificationMin() const;
     double classificationMax() const;
     void setClassificationMin( double min );
diff --git a/src/core/raster/qgssinglebandpseudocolorrenderer.cpp b/src/core/raster/qgssinglebandpseudocolorrenderer.cpp
@@ -38,6 +38,15 @@ QgsSingleBandPseudoColorRenderer::~QgsSingleBandPseudoColorRenderer()
   delete mShader;
 }
 
+void QgsSingleBandPseudoColorRenderer::setBand(int bandNo)
+{
+  if ( bandNo > mInput->bandCount() || bandNo <= 0 )
+  {
+    return;
+  }
+  mBand = bandNo;
+}
+
 QgsRasterInterface * QgsSingleBandPseudoColorRenderer::clone() const
 {
   QgsRasterShader *shader = 0;
diff --git a/src/core/raster/qgssinglebandpseudocolorrenderer.h b/src/core/raster/qgssinglebandpseudocolorrenderer.h
@@ -54,6 +54,12 @@ class CORE_EXPORT QgsSingleBandPseudoColorRenderer: public QgsRasterRenderer
      */
     int band() const { return mBand; }
 
+     /** Sets the band used by the renderer.
+     * @see band
+     * @note added in QGIS 2.10
+     */
+    void setBand( int bandNo );
+
     double classificationMin() const { return mClassificationMin; }
     double classificationMax() const { return mClassificationMax; }
     void setClassificationMin( double min ) { mClassificationMin = min; }
