fix Heatmap plugin unloading, also place button to the Raster toolbar

qgis · Feb 8, 2012 · 093aae5 · 093aae5
1 parent 30e2f32
commit 093aae5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/plugins/heatmap/heatmap.cpp b/src/plugins/heatmap/heatmap.cpp
@@ -78,7 +78,7 @@ void Heatmap::initGui()
   // Connect the action to the run
   connect( mQActionPointer, SIGNAL( triggered() ), this, SLOT( run() ) );
   // Add the icon to the toolbar
-  mQGisIface->addToolBarIcon( mQActionPointer );
+  mQGisIface->addRasterToolBarIcon( mQActionPointer );
   mQGisIface->addPluginToRasterMenu( tr( "&Heatmap" ), mQActionPointer );
 
 }
@@ -108,8 +108,8 @@ void Heatmap::run()
 void Heatmap::unload()
 {
   // remove the GUI
-  mQGisIface->removePluginMenu( "&Heatmap", mQActionPointer );
-  mQGisIface->removeToolBarIcon( mQActionPointer );
+  mQGisIface->removePluginRasterMenu( "&Heatmap", mQActionPointer );
+  mQGisIface->removeRasterToolBarIcon( mQActionPointer );
   delete mQActionPointer;
 }