Prechádzať zdrojové kódy

wxGUI rdigit: update list of available raster map layers, when map layers tree changed (#707)

Tomas Zigo 4 rokov pred
rodič
commit
e321f7589c
1 zmenil súbory, kde vykonal 12 pridanie a 2 odobranie
  1. 12 2
      gui/wxpython/lmgr/layertree.py

+ 12 - 2
gui/wxpython/lmgr/layertree.py

@@ -2155,12 +2155,22 @@ class LayerTree(treemixin.DragAndDrop, CT.CustomTreeCtrl):
 
         self.SetLayerInfo(item, key='maplayer', value=maplayer)
 
-        # if digitization tool enabled -> update list of available vector map
-        # layers
+        # if vector digitization tool enabled -> update list of
+        # available vector map layers
         if self.mapdisplay.GetToolbar('vdigit'):
             self.mapdisplay.GetToolbar(
                 'vdigit').UpdateListOfLayers(updateTool=True)
 
+        # if raster digitization tool enabled -> update list of
+        # available raster map layers
+        if self.mapdisplay.GetToolbar('rdigit'):
+            rasters = self.GetMap().GetListOfLayers(
+                ltype='raster', mapset=grass.gisenv()['MAPSET'])
+            self.mapdisplay.GetToolbar(
+                'rdigit').UpdateRasterLayers(rasters)
+            self.mapdisplay.GetToolbar(
+                'rdigit').SelectDefault()
+
         self.Map.SetLayers(self.GetVisibleLayers())
 
         # redraw map if auto-rendering is enabled