Added setBand capability to qgssinglebandpseudocolorrenderer API

qgis · Jun 23, 2015 · 45681e6 · 45681e6
1 parent da39ab3
commit 45681e6
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/python/core/raster/qgssinglebandpseudocolorrenderer.sip b/python/core/raster/qgssinglebandpseudocolorrenderer.sip
@@ -28,6 +28,7 @@ class QgsSingleBandPseudoColorRenderer: QgsRasterRenderer
      * @note added in QGIS 2.7
      */
     int band() const;
+    void setBand(int bandNo);
 
     double classificationMin() const;
     double classificationMax() const;

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
@@ -53,6 +53,7 @@ class CORE_EXPORT QgsSingleBandPseudoColorRenderer: public QgsRasterRenderer
      * @note added in QGIS 2.7
      */
     int band() const { return mBand; }
+    void setBand( int bandNo );
 
     double classificationMin() const { return mClassificationMin; }
     double classificationMax() const { return mClassificationMax; }