123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749 |
- """
- @package mapdisp.frame
- @brief Map display with toolbar for various display management
- functions, and additional toolbars (vector digitizer, 3d view).
- Can be used either from Layer Manager or as d.mon backend.
- Classes:
- - mapdisp::MapFrame
- (C) 2006-2016 by the GRASS Development Team
- This program is free software under the GNU General Public License
- (>=v2). Read the file COPYING that comes with GRASS for details.
- @author Michael Barton
- @author Jachym Cepicky
- @author Martin Landa <landa.martin gmail.com>
- @author Vaclav Petras <wenzeslaus gmail.com> (SingleMapFrame, handlers support)
- @author Anna Kratochvilova <kratochanna gmail.com> (SingleMapFrame)
- @author Stepan Turek <stepan.turek seznam.cz> (handlers support)
- """
- import os
- import copy
- from core import globalvar
- import wx
- import wx.aui
- from mapdisp.toolbars import MapToolbar, NvizIcons
- from mapdisp.gprint import PrintOptions
- from core.gcmd import GError, GMessage, RunCommand
- from core.utils import ListOfCatsToRange, GetLayerNameFromCmd
- from gui_core.dialogs import GetImageHandlers, ImageSizeDialog
- from core.debug import Debug
- from core.settings import UserSettings
- from gui_core.mapdisp import SingleMapFrame
- from mapwin.base import MapWindowProperties
- from gui_core.query import QueryDialog, PrepareQueryResults
- from mapwin.buffered import BufferedMapWindow
- from mapwin.decorations import (
- LegendController,
- BarscaleController,
- ArrowController,
- DtextController,
- LegendVectController,
- )
- from mapwin.analysis import (
- ProfileController,
- MeasureDistanceController,
- MeasureAreaController,
- )
- from gui_core.forms import GUI
- from core.giface import Notification
- from gui_core.vselect import VectorSelectBase, VectorSelectHighlighter
- from gui_core.wrap import Menu
- from mapdisp import statusbar as sb
- import grass.script as grass
- from grass.pydispatch.signal import Signal
- class MapFrame(SingleMapFrame):
- """Main frame for map display window. Drawing takes place in
- child double buffered drawing window.
- """
- def __init__(
- self,
- parent,
- giface,
- title=_("Map Display"),
- toolbars=["map"],
- statusbar=True,
- tree=None,
- notebook=None,
- lmgr=None,
- page=None,
- Map=None,
- auimgr=None,
- name="MapWindow",
- **kwargs,
- ):
- """Main map display window with toolbars, statusbar and
- 2D map window, 3D map window and digitizer.
- :param toolbars: array of activated toolbars, e.g. ['map', 'digit']
- :param statusbar: True to add statusbar
- :param tree: reference to layer tree
- :param notebook: control book ID in Layer Manager
- :param lmgr: Layer Manager
- :param page: notebook page with layer tree
- :param map: instance of render.Map
- :param auimgr: AUI manager
- :param name: frame name
- :param kwargs: wx.Frame attributes
- """
- SingleMapFrame.__init__(
- self,
- parent=parent,
- title=title,
- Map=Map,
- auimgr=auimgr,
- name=name,
- **kwargs,
- )
- self._giface = giface
- # Layer Manager object
- # need by GLWindow (a lot), VDigitWindow (a little bit)
- self._layerManager = lmgr
- # Layer Manager layer tree object
- # used for VDigit toolbar and window and GLWindow
- self.tree = tree
- # Notebook page holding the layer tree
- # used only in OnCloseWindow
- self.page = page
- # Layer Manager layer tree notebook
- # used only in OnCloseWindow
- self.layerbook = notebook
- # Emitted when starting (switching to) 3D mode.
- # Parameter firstTime specifies if 3D was already actived.
- self.starting3dMode = Signal("MapFrame.starting3dMode")
- # Emitted when ending (switching from) 3D mode.
- self.ending3dMode = Signal("MapFrame.ending3dMode")
- # Emitted when closing display by closing its window.
- self.closingDisplay = Signal("MapFrame.closingDisplay")
- # Emitted when closing display by closing its window.
- self.closingVNETDialog = Signal("MapFrame.closingVNETDialog")
- # properties are shared in other objects, so defining here
- self.mapWindowProperties = MapWindowProperties()
- self.mapWindowProperties.setValuesFromUserSettings()
- #
- # Add toolbars
- #
- for toolb in toolbars:
- self.AddToolbar(toolb)
- #
- # Add statusbar
- #
- self.statusbarManager = None
- if statusbar:
- self.CreateStatusbar()
- # init decoration objects
- self.decorations = {}
- self._decorationWindows = {}
- self.mapWindowProperties.autoRenderChanged.connect(
- lambda value: self.OnRender(None) if value else None
- )
- #
- # Init map display (buffered DC & set default cursor)
- #
- self.MapWindow2D = BufferedMapWindow(
- self,
- giface=self._giface,
- Map=self.Map,
- properties=self.mapWindowProperties,
- overlays=self.decorations,
- )
- self.MapWindow2D.mapQueried.connect(self.Query)
- self.MapWindow2D.overlayActivated.connect(self._activateOverlay)
- self.MapWindow2D.overlayRemoved.connect(self.RemoveOverlay)
- self._setUpMapWindow(self.MapWindow2D)
- self.MapWindow2D.mouseHandlerUnregistered.connect(self.ResetPointer)
- self.MapWindow2D.InitZoomHistory()
- self.MapWindow2D.zoomChanged.connect(self.StatusbarUpdate)
- self._giface.updateMap.connect(self.MapWindow2D.UpdateMap)
- # default is 2D display mode
- self.MapWindow = self.MapWindow2D
- self.MapWindow.SetNamedCursor("default")
- # used by vector digitizer
- self.MapWindowVDigit = None
- # used by Nviz (3D display mode)
- self.MapWindow3D = None
- if "map" in self.toolbars:
- self.toolbars["map"].SelectDefault()
- #
- # Bind various events
- #
- self.Bind(wx.EVT_CLOSE, self.OnCloseWindow)
- self.Bind(wx.EVT_SIZE, self.OnSize)
- #
- # Update fancy gui style
- #
- self._mgr.AddPane(
- self.MapWindow,
- wx.aui.AuiPaneInfo()
- .CentrePane()
- .Dockable(False)
- .BestSize((-1, -1))
- .Name("2d")
- .CloseButton(False)
- .DestroyOnClose(True)
- .Layer(0),
- )
- self._mgr.Update()
- #
- # Init print module and classes
- #
- self.printopt = PrintOptions(self, self.MapWindow)
- #
- # Re-use dialogs
- #
- self.dialogs = {}
- self.dialogs["attributes"] = None
- self.dialogs["category"] = None
- self.dialogs["vnet"] = None
- self.dialogs["query"] = None
- self.dialogs["vselect"] = None
- # initialize layers to query (d.what.vect/rast)
- self._vectQueryLayers = []
- self._rastQueryLayers = []
- # initialize highlighter for vector features
- self._highlighter_layer = None
- self.measureController = None
- self._resize()
- def CreateStatusbar(self):
- if self.statusbarManager:
- return
- # items for choice
- self.statusbarItems = [
- sb.SbCoordinates,
- sb.SbRegionExtent,
- sb.SbCompRegionExtent,
- sb.SbShowRegion,
- sb.SbAlignExtent,
- sb.SbResolution,
- sb.SbDisplayGeometry,
- sb.SbMapScale,
- sb.SbGoTo,
- sb.SbProjection,
- ]
- self.statusbarItemsHiddenInNviz = (
- sb.SbAlignExtent,
- sb.SbDisplayGeometry,
- sb.SbShowRegion,
- sb.SbResolution,
- sb.SbMapScale,
- )
- # create statusbar and its manager
- statusbar = self.CreateStatusBar(number=4, style=0)
- statusbar.SetMinHeight(24)
- statusbar.SetStatusWidths([-5, -2, -1, -1])
- self.statusbarManager = sb.SbManager(mapframe=self, statusbar=statusbar)
- # fill statusbar manager
- self.statusbarManager.AddStatusbarItemsByClass(
- self.statusbarItems, mapframe=self, statusbar=statusbar
- )
- self.statusbarManager.AddStatusbarItem(
- sb.SbMask(self, statusbar=statusbar, position=2)
- )
- sbRender = sb.SbRender(self, statusbar=statusbar, position=3)
- self.statusbarManager.AddStatusbarItem(sbRender)
- self.statusbarManager.Update()
- #
- self.Map.GetRenderMgr().updateProgress.connect(
- self.statusbarManager.SetProgress
- )
- self.Map.GetRenderMgr().renderingFailed.connect(
- lambda cmd, error: self._giface.WriteError(
- _("Failed to run command '%(command)s'. Details:\n%(error)s")
- % dict(command=" ".join(cmd), error=error)
- )
- )
- def GetMapWindow(self):
- return self.MapWindow
- def _addToolbarVDigit(self):
- """Add vector digitizer toolbar"""
- from vdigit.main import haveVDigit, VDigit
- from vdigit.toolbars import VDigitToolbar
- if not haveVDigit:
- from vdigit import errorMsg
- self.toolbars["map"].combo.SetValue(_("2D view"))
- GError(
- _("Unable to start wxGUI vector digitizer.\n" "Details: %s") % errorMsg,
- parent=self,
- )
- return
- if not self.MapWindowVDigit:
- from vdigit.mapwindow import VDigitWindow
- self.MapWindowVDigit = VDigitWindow(
- parent=self,
- giface=self._giface,
- properties=self.mapWindowProperties,
- Map=self.Map,
- tree=self.tree,
- lmgr=self._layerManager,
- overlays=self.decorations,
- )
- self._setUpMapWindow(self.MapWindowVDigit)
- self.MapWindowVDigit.digitizingInfo.connect(
- lambda text: self.statusbarManager.statusbarItems[
- "coordinates"
- ].SetAdditionalInfo(text)
- )
- self.MapWindowVDigit.digitizingInfoUnavailable.connect(
- lambda: self.statusbarManager.statusbarItems[
- "coordinates"
- ].SetAdditionalInfo(None)
- )
- self.MapWindowVDigit.Show()
- self._mgr.AddPane(
- self.MapWindowVDigit,
- wx.aui.AuiPaneInfo()
- .CentrePane()
- .Dockable(False)
- .BestSize((-1, -1))
- .Name("vdigit")
- .CloseButton(False)
- .DestroyOnClose(True)
- .Layer(0),
- )
- self._switchMapWindow(self.MapWindowVDigit)
- if self._mgr.GetPane("2d").IsShown():
- self._mgr.GetPane("2d").Hide()
- elif self._mgr.GetPane("3d").IsShown():
- self._mgr.GetPane("3d").Hide()
- self._mgr.GetPane("vdigit").Show()
- if "vdigit" not in self.toolbars:
- self.toolbars["vdigit"] = VDigitToolbar(
- parent=self,
- toolSwitcher=self._toolSwitcher,
- MapWindow=self.MapWindow,
- digitClass=VDigit,
- giface=self._giface,
- )
- self.toolbars["vdigit"].quitDigitizer.connect(self.QuitVDigit)
- self.Map.layerAdded.connect(self._updateVDigitLayers)
- self.MapWindowVDigit.SetToolbar(self.toolbars["vdigit"])
- self._mgr.AddPane(
- self.toolbars["vdigit"],
- wx.aui.AuiPaneInfo()
- .Name("vdigittoolbar")
- .Caption(_("Vector Digitizer Toolbar"))
- .ToolbarPane()
- .Top()
- .Row(1)
- .LeftDockable(False)
- .RightDockable(False)
- .BottomDockable(False)
- .TopDockable(True)
- .CloseButton(False)
- .Layer(2)
- .BestSize((self.toolbars["vdigit"].GetBestSize())),
- )
- # change mouse to draw digitized line
- self.MapWindow.mouse["box"] = "point"
- self.MapWindow.zoomtype = 0
- self.MapWindow.pen = wx.Pen(colour="red", width=2, style=wx.SOLID)
- self.MapWindow.polypen = wx.Pen(colour="green", width=2, style=wx.SOLID)
- def _updateVDigitLayers(self, layer):
- """Update vdigit layers"""
- if "vdigit" in self.toolbars:
- self.toolbars["vdigit"].UpdateListOfLayers(updateTool=True)
- def AddNviz(self):
- """Add 3D view mode window"""
- from nviz.main import haveNviz, GLWindow, errorMsg
- # check for GLCanvas and OpenGL
- if not haveNviz:
- self.toolbars["map"].combo.SetValue(_("2D view"))
- GError(
- parent=self,
- message=_(
- "Unable to switch to 3D display mode.\nThe Nviz python extension "
- "was not found or loaded properly.\n"
- "Switching back to 2D display mode.\n\nDetails: %s" % errorMsg
- ),
- )
- return
- # here was disabling 3D for other displays, now done on starting3dMode
- self.toolbars["map"].Enable2D(False)
- # add rotate tool to map toolbar
- self.toolbars["map"].InsertTool(
- (("rotate", NvizIcons["rotate"], self.OnRotate, wx.ITEM_CHECK, 7),)
- ) # 7 is position
- self._toolSwitcher.AddToolToGroup(
- group="mouseUse",
- toolbar=self.toolbars["map"],
- tool=self.toolbars["map"].rotate,
- )
- self.toolbars["map"].InsertTool(
- (
- (
- "flyThrough",
- NvizIcons["flyThrough"],
- self.OnFlyThrough,
- wx.ITEM_CHECK,
- 8,
- ),
- )
- )
- self._toolSwitcher.AddToolToGroup(
- group="mouseUse",
- toolbar=self.toolbars["map"],
- tool=self.toolbars["map"].flyThrough,
- )
- # update status bar
- self.statusbarManager.HideStatusbarChoiceItemsByClass(
- self.statusbarItemsHiddenInNviz
- )
- self.statusbarManager.SetMode(0)
- # erase map window
- self.MapWindow.EraseMap()
- self._giface.WriteCmdLog(
- _("Starting 3D view mode..."), notification=Notification.HIGHLIGHT
- )
- self.SetStatusText(_("Please wait, loading data..."), 0)
- # create GL window
- if not self.MapWindow3D:
- self.MapWindow3D = GLWindow(
- self,
- giface=self._giface,
- id=wx.ID_ANY,
- frame=self,
- Map=self.Map,
- tree=self.tree,
- lmgr=self._layerManager,
- )
- self._setUpMapWindow(self.MapWindow3D)
- self.MapWindow3D.mapQueried.connect(self.Query)
- self._switchMapWindow(self.MapWindow3D)
- self.MapWindow.SetNamedCursor("default")
- # here was AddNvizTools in lmgr
- self.starting3dMode.emit(firstTime=True)
- # switch from MapWindow to MapWindowGL
- self._mgr.GetPane("2d").Hide()
- self._mgr.AddPane(
- self.MapWindow3D,
- wx.aui.AuiPaneInfo()
- .CentrePane()
- .Dockable(False)
- .BestSize((-1, -1))
- .Name("3d")
- .CloseButton(False)
- .DestroyOnClose(True)
- .Layer(0),
- )
- self.MapWindow3D.Show()
- self.MapWindow3D.ResetViewHistory()
- self.MapWindow3D.UpdateView(None)
- self.MapWindow3D.overlayActivated.connect(self._activateOverlay)
- self.MapWindow3D.overlayRemoved.connect(self.RemoveOverlay)
- else:
- self._switchMapWindow(self.MapWindow3D)
- os.environ["GRASS_REGION"] = self.Map.SetRegion(windres=True, windres3=True)
- self.MapWindow3D.GetDisplay().Init()
- self.MapWindow3D.LoadDataLayers()
- del os.environ["GRASS_REGION"]
- # switch from MapWindow to MapWindowGL
- self._mgr.GetPane("2d").Hide()
- self._mgr.GetPane("3d").Show()
- # here was AddNvizTools in lmgr and updating of pages
- self.starting3dMode.emit(firstTime=False)
- self.MapWindow3D.ResetViewHistory()
- # connect signals for updating overlays
- for overlay in self.decorations.values():
- overlay.overlayChanged.connect(self.MapWindow3D.UpdateOverlays)
- self.Map.GetRenderMgr().renderDone.connect(self.MapWindow3D._onUpdateOverlays)
- self._giface.updateMap.disconnect(self.MapWindow2D.UpdateMap)
- self._giface.updateMap.connect(self.MapWindow3D.UpdateMap)
- self.MapWindow3D.overlays = self.MapWindow2D.overlays
- # update overlays needs to be called after because getClientSize
- # is called during update and it must give reasonable values
- wx.CallAfter(self.MapWindow3D.UpdateOverlays)
- self.SetStatusText("", 0)
- self._mgr.Update()
- def Disable3dMode(self):
- """Disables 3D mode (NVIZ) in user interface."""
- # TODO: this is broken since item is removed but switch is drived by
- # index
- if "3D" in self.toolbars["map"].combo.GetString(1):
- self.toolbars["map"].combo.Delete(1)
- def RemoveNviz(self):
- """Restore 2D view"""
- try:
- self.toolbars["map"].RemoveTool(self.toolbars["map"].rotate)
- self.toolbars["map"].RemoveTool(self.toolbars["map"].flyThrough)
- except AttributeError:
- pass
- # update status bar
- self.statusbarManager.ShowStatusbarChoiceItemsByClass(
- self.statusbarItemsHiddenInNviz
- )
- self.statusbarManager.SetMode(
- UserSettings.Get(group="display", key="statusbarMode", subkey="selection")
- )
- self.SetStatusText(_("Please wait, unloading data..."), 0)
- # unloading messages from library cause highlight anyway
- self._giface.WriteCmdLog(
- _("Switching back to 2D view mode..."),
- notification=Notification.NO_NOTIFICATION,
- )
- if self.MapWindow3D:
- self.MapWindow3D.OnClose(event=None)
- # switch from MapWindowGL to MapWindow
- self._mgr.GetPane("2d").Show()
- self._mgr.GetPane("3d").Hide()
- self._switchMapWindow(self.MapWindow2D)
- # here was RemoveNvizTools form lmgr
- self.ending3dMode.emit()
- try:
- self.MapWindow2D.overlays = self.MapWindow3D.overlays
- except AttributeError:
- pass
- # TODO: here we end because self.MapWindow3D is None for a while
- self._giface.updateMap.disconnect(self.MapWindow3D.UpdateMap)
- self._giface.updateMap.connect(self.MapWindow2D.UpdateMap)
- # disconnect overlays
- for overlay in self.decorations.values():
- overlay.overlayChanged.disconnect(self.MapWindow3D.UpdateOverlays)
- self.Map.GetRenderMgr().renderDone.disconnect(
- self.MapWindow3D._onUpdateOverlays
- )
- self.MapWindow3D.ClearTextures()
- self.MapWindow.UpdateMap()
- self._mgr.Update()
- self.GetMapToolbar().SelectDefault()
- def AddToolbar(self, name, fixed=False):
- """Add defined toolbar to the window
- Currently recognized toolbars are:
- - 'map' - basic map toolbar
- - 'vdigit' - vector digitizer
- :param name: toolbar to add
- :param fixed: fixed toolbar
- """
- # default toolbar
- if name == "map":
- if "map" not in self.toolbars:
- self.toolbars["map"] = MapToolbar(self, toolSwitcher=self._toolSwitcher)
- self._mgr.AddPane(
- self.toolbars["map"],
- wx.aui.AuiPaneInfo()
- .Name("maptoolbar")
- .Caption(_("Map Toolbar"))
- .ToolbarPane()
- .Top()
- .Name("mapToolbar")
- .LeftDockable(False)
- .RightDockable(False)
- .BottomDockable(False)
- .TopDockable(True)
- .CloseButton(False)
- .Layer(2)
- .BestSize((self.toolbars["map"].GetBestSize())),
- )
- # vector digitizer
- elif name == "vdigit":
- self.toolbars["map"].combo.SetValue(_("Vector digitizer"))
- self._addToolbarVDigit()
- # raster digitizer
- elif name == "rdigit":
- self.toolbars["map"].combo.SetValue(_("Raster digitizer"))
- self.AddRDigit()
- if fixed:
- self.toolbars["map"].combo.Disable()
- self._mgr.Update()
- def RemoveToolbar(self, name, destroy=False):
- """Removes defined toolbar from the window
- :param name toolbar to remove
- :param destroy True to destroy otherwise toolbar is only hidden
- """
- self._mgr.DetachPane(self.toolbars[name])
- if destroy:
- self._toolSwitcher.RemoveToolbarFromGroup("mouseUse", self.toolbars[name])
- self.toolbars[name].Destroy()
- self.toolbars.pop(name)
- else:
- self.toolbars[name].Hide()
- if name == "vdigit":
- self._mgr.GetPane("vdigit").Hide()
- self._mgr.GetPane("2d").Show()
- self._switchMapWindow(self.MapWindow2D)
- self.toolbars["map"].Enable2D(True)
- self._mgr.Update()
- def IsPaneShown(self, name):
- """Check if pane (toolbar, mapWindow ...) of given name is currently shown"""
- if self._mgr.GetPane(name).IsOk():
- return self._mgr.GetPane(name).IsShown()
- return False
- def RemoveQueryLayer(self):
- """Removes temporary map layers (queries)"""
- qlayer = self.GetMap().GetListOfLayers(name=globalvar.QUERYLAYER)
- for layer in qlayer:
- self.GetMap().DeleteLayer(layer)
- def OnRender(self, event):
- """Re-render map composition (each map layer)"""
- self.RemoveQueryLayer()
- # deselect features in vdigit
- if self.GetToolbar("vdigit"):
- if self.MapWindow.digit:
- self.MapWindow.digit.GetDisplay().SetSelected([])
- self.MapWindow.UpdateMap(render=True, renderVector=True)
- else:
- self.MapWindow.UpdateMap(render=True)
- # reset dialog with selected features
- if self.dialogs["vselect"]:
- self.dialogs["vselect"].Reset()
- # update statusbar
- self.StatusbarUpdate()
- def OnPointer(self, event):
- """Pointer button clicked"""
- self.MapWindow.SetModePointer()
- if self.GetToolbar("vdigit"):
- self.toolbars["vdigit"].action["id"] = -1
- self.toolbars["vdigit"].action["desc"] = ""
- def OnSelect(self, event):
- """Vector feature selection button clicked"""
- layerList = self._giface.GetLayerList()
- layerSelected = layerList.GetSelectedLayer()
- if not self.dialogs["vselect"]:
- if layerSelected is None:
- GMessage(_("No map layer selected. Operation canceled."))
- return
- self.dialogs["vselect"] = VectorSelectBase(self.parent, self._giface)
- self.dialogs["vselect"].CreateDialog(createButton=True)
- self.dialogs["vselect"].onCloseDialog.connect(
- self._onCloseVectorSelectDialog
- )
- def _onCloseVectorSelectDialog(self):
- self.dialogs["vselect"] = None
- def OnRotate(self, event):
- """Rotate 3D view"""
- self.MapWindow.mouse["use"] = "rotate"
- # change the cursor
- self.MapWindow.SetNamedCursor("hand")
- def OnFlyThrough(self, event):
- """Fly-through mode"""
- self.MapWindow.mouse["use"] = "fly"
- # change the cursor
- self.MapWindow.SetNamedCursor("hand")
- self.MapWindow.SetFocus()
- def SaveToFile(self, event):
- """Save map to image"""
- filetype, ltype = self._prepareSaveToFile()
- if not ltype:
- return
- # get size
- dlg = ImageSizeDialog(self)
- dlg.CentreOnParent()
- if dlg.ShowModal() != wx.ID_OK:
- dlg.Destroy()
- return
- width, height = dlg.GetValues()
- dlg.Destroy()
- # get filename
- dlg = wx.FileDialog(
- parent=self,
- message=_(
- "Choose a file name to save the image " "(no need to add extension)"
- ),
- wildcard=filetype,
- style=wx.FD_SAVE | wx.FD_OVERWRITE_PROMPT,
- )
- if dlg.ShowModal() == wx.ID_OK:
- path = dlg.GetPath()
- if not path:
- dlg.Destroy()
- return
- base, ext = os.path.splitext(path)
- fileType = ltype[dlg.GetFilterIndex()]["type"]
- extType = ltype[dlg.GetFilterIndex()]["ext"]
- if ext != extType:
- path = base + "." + extType
- self.MapWindow.SaveToFile(path, fileType, width, height)
- dlg.Destroy()
- def DOutFile(self, command, callback=None):
- """Saves map to image by running d.out.file from gui or d.mon.
- Command is expected to be validated by parser.
- """
- filetype, ltype = self._prepareSaveToFile()
- if not ltype:
- return
- width, height = self.MapWindow.GetClientSize()
- for param in command[1:]:
- try:
- p, val = param.split("=")
- except ValueError:
- # --overwrite
- continue
- if p == "format": # must be there
- if self.IsPaneShown("3d"):
- extType = "ppm"
- else:
- extType = val
- if p == "output": # must be there
- name = val
- elif p == "size":
- width, height = val.split(",")
- base, ext = os.path.splitext(name)
- if not ext:
- name = base + "." + extType
- elif ext[1:] != extType:
- extType = ext[1:]
- if self.IsPaneShown("3d"):
- bitmapType = "ppm"
- else:
- bitmapType = wx.BITMAP_TYPE_PNG # default type
- for each in ltype:
- if each["ext"] == extType:
- bitmapType = each["type"]
- break
- self.MapWindow.SaveToFile(name, bitmapType, int(width), int(height), callback)
- def DOutFileOptData(self, dcmd, layer, params, propwin):
- """Dummy function which is called when d.out.file is called
- and returns parsed and validated command which is then passed
- to DOutFile method."""
- if not dcmd:
- return
- self.DOutFile(dcmd)
- def DToRast(self, command):
- """Saves currently loaded composition of layers as a raster map."""
- def _DToRastDone():
- # import back as red, green, blue rasters
- returncode, messages = RunCommand(
- "r.in.gdal",
- flags="o",
- input=pngFile,
- output=tmpName,
- quiet=True,
- overwrite=overwrite,
- getErrorMsg=True,
- )
- if not returncode == 0:
- self._giface.WriteError(_("Failed to run d.to.rast:\n") + messages)
- return
- # set region for composite
- grass.use_temp_region()
- returncode, messages = RunCommand(
- "g.region", raster=tmpName + ".red", quiet=True, getErrorMsg=True
- )
- if not returncode == 0:
- grass.del_temp_region()
- self._giface.WriteError(_("Failed to run d.to.rast:\n") + messages)
- return
- # composite
- returncode, messages = RunCommand(
- "r.composite",
- red=tmpName + ".red",
- green=tmpName + ".green",
- blue=tmpName + ".blue",
- output=outputRaster,
- quiet=True,
- overwrite=overwrite,
- getErrorMsg=True,
- )
- grass.del_temp_region()
- RunCommand(
- "g.remove",
- type="raster",
- flags="f",
- quiet=True,
- name=[tmpName + ".red", tmpName + ".green", tmpName + ".blue"],
- )
- if not returncode == 0:
- self._giface.WriteError(_("Failed to run d.to.rast:\n") + messages)
- grass.try_remove(pngFile)
- return
- # alignExtent changes only region variable
- oldRegion = self.GetMap().GetCurrentRegion().copy()
- self.GetMap().AlignExtentFromDisplay()
- region = self.GetMap().GetCurrentRegion().copy()
- self.GetMap().region.update(oldRegion)
- RunCommand(
- "r.region",
- map=outputRaster,
- n=region["n"],
- s=region["s"],
- e=region["e"],
- w=region["w"],
- quiet=True,
- )
- grass.try_remove(pngFile)
- if self.IsPaneShown("3d"):
- self._giface.WriteError(_("d.to.rast can be used only in 2D mode."))
- return
- outputRaster = None
- overwrite = False
- for param in command[1:]:
- try:
- p, val = param.split("=")
- if p == "output":
- outputRaster = val
- except ValueError:
- if param.startswith("--overwrite"):
- overwrite = True
- if not outputRaster:
- return
- # output file as PNG
- tmpName = "d_to_rast_tmp"
- pngFile = grass.tempfile(create=False) + ".png"
- dOutFileCmd = ["d.out.file", "output=" + pngFile, "format=png"]
- self.DOutFile(dOutFileCmd, callback=_DToRastDone)
- def DToRastOptData(self, dcmd, layer, params, propwin):
- """Dummy function which is called when d.to.rast is called
- and returns parsed and validated command which is then passed
- to DToRast method."""
- if not dcmd:
- return
- self.DToRast(dcmd)
- def _prepareSaveToFile(self):
- """Get wildcards and format extensions."""
- if self.IsPaneShown("3d"):
- filetype = "TIF file (*.tif)|*.tif|PPM file (*.ppm)|*.ppm"
- ltype = [{"ext": "tif", "type": "tif"}, {"ext": "ppm", "type": "ppm"}]
- else:
- img = self.MapWindow.img
- if not img:
- GMessage(
- parent=self,
- message=_("Nothing to render (empty map). Operation canceled."),
- )
- return None, None
- filetype, ltype = GetImageHandlers(img)
- return filetype, ltype
- def PrintMenu(self, event):
- """
- Print options and output menu for map display
- """
- printmenu = Menu()
- # Add items to the menu
- setup = wx.MenuItem(printmenu, wx.ID_ANY, _("Page setup"))
- printmenu.AppendItem(setup)
- self.Bind(wx.EVT_MENU, self.printopt.OnPageSetup, setup)
- preview = wx.MenuItem(printmenu, wx.ID_ANY, _("Print preview"))
- printmenu.AppendItem(preview)
- self.Bind(wx.EVT_MENU, self.printopt.OnPrintPreview, preview)
- doprint = wx.MenuItem(printmenu, wx.ID_ANY, _("Print display"))
- printmenu.AppendItem(doprint)
- self.Bind(wx.EVT_MENU, self.printopt.OnDoPrint, doprint)
- # Popup the menu. If an item is selected then its handler
- # will be called before PopupMenu returns.
- self.PopupMenu(printmenu)
- printmenu.Destroy()
- def CleanUp(self):
- """Clean up before closing map display.
- End digitizer/nviz."""
- Debug.msg(2, "MapFrame.CleanUp()")
- self.Map.Clean()
- # close edited map and 3D tools properly
- if self.GetToolbar("vdigit"):
- maplayer = self.toolbars["vdigit"].GetLayer()
- if maplayer:
- self.toolbars["vdigit"].OnExit()
- if self.IsPaneShown("3d"):
- self.RemoveNviz()
- if hasattr(self, "rdigit") and self.rdigit:
- self.rdigit.CleanUp()
- if self.dialogs["vnet"]:
- self.closingVNETDialog.emit()
- self._mgr.UnInit()
- def OnCloseWindow(self, event, askIfSaveWorkspace=True):
- """Window closed.
- Also close associated layer tree page
- """
- Debug.msg(2, "MapFrame.OnCloseWindow()")
- if self._layerManager:
- pgnum = self.layerbook.GetPageIndex(self.page)
- name = self.layerbook.GetPageText(pgnum)
- caption = _("Close Map Display {}").format(name)
- if not askIfSaveWorkspace or (
- askIfSaveWorkspace
- and self._layerManager.workspace_manager.CanClosePage(caption)
- ):
- self.CleanUp()
- if pgnum > -1:
- self.closingDisplay.emit(page_index=pgnum)
- # Destroy is called when notebook page is deleted
- else:
- self.CleanUp()
- self.Destroy()
- def Query(self, x, y):
- """Query selected layers.
- :param x,y: coordinates
- """
- if self._vectQueryLayers or self._rastQueryLayers:
- rast = self._rastQueryLayers
- vect = self._vectQueryLayers
- else:
- layers = self._giface.GetLayerList().GetSelectedLayers(checkedOnly=False)
- rast = []
- vect = []
- for layer in layers:
- if layer.type == "command":
- continue
- name, found = GetLayerNameFromCmd(layer.cmd)
- if not found:
- continue
- ltype = layer.maplayer.GetType()
- if ltype == "raster":
- rast.append(name)
- elif ltype in ("rgb", "his"):
- for iname in name.split("\n"):
- rast.append(iname)
- elif ltype in ("vector", "thememap", "themechart"):
- vect.append(name)
- if vect:
- # check for vector maps open to be edited
- digitToolbar = self.GetToolbar("vdigit")
- if digitToolbar:
- lmap = digitToolbar.GetLayer().GetName()
- for name in vect:
- if lmap == name:
- self._giface.WriteWarning(
- _("Vector map <%s> " "opened for editing - skipped.")
- % lmap
- )
- vect.remove(name)
- if not (rast + vect):
- GMessage(
- parent=self,
- message=_("No raster or vector map layer selected for querying."),
- )
- return
- # set query snap distance for v.what at map unit equivalent of 10
- # pixels
- qdist = 10.0 * ((self.Map.region["e"] - self.Map.region["w"]) / self.Map.width)
- # TODO: replace returning None by exception or so
- try:
- east, north = self.MapWindow.Pixel2Cell((x, y))
- except TypeError:
- return
- if not self.IsPaneShown("3d"):
- self.QueryMap(east, north, qdist, rast, vect)
- else:
- if rast:
- self.MapWindow.QuerySurface(x, y)
- if vect:
- self.QueryMap(east, north, qdist, rast=[], vect=vect)
- def SetQueryLayersAndActivate(self, ltype, maps):
- """Activate query mode and set layers to query.
- This method is used for querying in d.mon using d.what.rast/vect"""
- self.toolbars["map"].SelectTool(self.toolbars["map"].query)
- if ltype == "vector":
- self._vectQueryLayers = maps
- elif ltype == "raster":
- self._rastQueryLayers = maps
- def QueryMap(self, east, north, qdist, rast, vect):
- """Query raster or vector map layers by r/v.what
- :param east,north: coordinates
- :param qdist: query distance
- :param rast: raster map names
- :param vect: vector map names
- """
- Debug.msg(
- 1, "QueryMap(): raster=%s vector=%s" % (",".join(rast), ",".join(vect))
- )
- if self._highlighter_layer is None:
- self._highlighter_layer = VectorSelectHighlighter(
- mapdisp=self._giface.GetMapDisplay(), giface=self._giface
- )
- # use display region settings instead of computation region settings
- self.tmpreg = os.getenv("GRASS_REGION")
- os.environ["GRASS_REGION"] = self.Map.SetRegion(windres=False)
- rastQuery = []
- vectQuery = []
- if rast:
- rastQuery = grass.raster_what(map=rast, coord=(east, north), localized=True)
- if vect:
- encoding = UserSettings.Get(group="atm", key="encoding", subkey="value")
- try:
- vectQuery = grass.vector_what(
- map=vect,
- coord=(east, north),
- distance=qdist,
- encoding=encoding,
- multiple=True,
- )
- except grass.ScriptError:
- GError(
- parent=self,
- message=_(
- "Failed to query vector map(s) <{maps}>. "
- "Check database settings and topology."
- ).format(maps=",".join(vect)),
- )
- self._QueryMapDone()
- self._highlighter_layer.Clear()
- if vectQuery and "Category" in vectQuery[0]:
- self._queryHighlight(vectQuery)
- result = rastQuery + vectQuery
- result = PrepareQueryResults(coordinates=(east, north), result=result)
- if self.dialogs["query"]:
- self.dialogs["query"].Raise()
- self.dialogs["query"].SetData(result)
- else:
- self.dialogs["query"] = QueryDialog(parent=self, data=result)
- self.dialogs["query"].Bind(wx.EVT_CLOSE, self._oncloseQueryDialog)
- self.dialogs["query"].redirectOutput.connect(self._onRedirectQueryOutput)
- self.dialogs["query"].Show()
- def _oncloseQueryDialog(self, event):
- self.dialogs["query"] = None
- self._vectQueryLayers = []
- self._rastQueryLayers = []
- self._highlighter_layer.Clear()
- self._highlighter_layer = None
- event.Skip()
- def _onRedirectQueryOutput(self, output, style="log"):
- """Writes query output into console"""
- if style == "log":
- self._giface.WriteLog(output, notification=Notification.MAKE_VISIBLE)
- elif style == "cmd":
- self._giface.WriteCmdLog(output)
- def _queryHighlight(self, vectQuery):
- """Highlight category from query."""
- if len(vectQuery) > 0:
- self._highlighter_layer.SetLayer(vectQuery[0]["Layer"])
- self._highlighter_layer.SetMap(
- vectQuery[0]["Map"] + "@" + vectQuery[0]["Mapset"]
- )
- tmp = list()
- for i in vectQuery:
- tmp.append(i["Category"])
- self._highlighter_layer.SetCats(tmp)
- self._highlighter_layer.DrawSelected()
- def _QueryMapDone(self):
- """Restore settings after querying (restore GRASS_REGION)"""
- if hasattr(self, "tmpreg"):
- if self.tmpreg:
- os.environ["GRASS_REGION"] = self.tmpreg
- elif "GRASS_REGION" in os.environ:
- del os.environ["GRASS_REGION"]
- elif "GRASS_REGION" in os.environ:
- del os.environ["GRASS_REGION"]
- if hasattr(self, "tmpreg"):
- del self.tmpreg
- def OnQuery(self, event):
- """Query tools menu"""
- self.MapWindow.mouse["use"] = "query"
- self.MapWindow.mouse["box"] = "point"
- self.MapWindow.zoomtype = 0
- # change the cursor
- self.MapWindow.SetNamedCursor("cross")
- def AddTmpVectorMapLayer(self, name, cats, useId=False, addLayer=True):
- """Add temporal vector map layer to map composition
- :param name: name of map layer
- :param useId: use feature id instead of category
- """
- # color settings from ATM
- color = UserSettings.Get(group="atm", key="highlight", subkey="color")
- colorStr = str(color[0]) + ":" + str(color[1]) + ":" + str(color[2])
- # icon used in vector display and its size
- icon = ""
- size = 0
- # here we know that there is one selected layer and it is vector
- layerSelected = self._giface.GetLayerList().GetSelectedLayer()
- if not layerSelected:
- return None
- vparam = layerSelected.cmd
- for p in vparam:
- if "=" in p:
- parg, pval = p.split("=", 1)
- if parg == "icon":
- icon = pval
- elif parg == "size":
- size = float(pval)
- pattern = [
- "d.vect",
- "map=%s" % name,
- "color=%s" % colorStr,
- "fill_color=%s" % colorStr,
- "width=%d" % UserSettings.Get(group="atm", key="highlight", subkey="width"),
- ]
- if icon != "":
- pattern.append("icon=%s" % icon)
- if size > 0:
- pattern.append("size=%i" % size)
- if useId:
- cmd = pattern
- cmd.append("-i")
- cmd.append("cats=%s" % str(cats))
- else:
- cmd = []
- for layer in cats.keys():
- cmd.append(copy.copy(pattern))
- lcats = cats[layer]
- cmd[-1].append("layer=%d" % layer)
- cmd[-1].append("cats=%s" % ListOfCatsToRange(lcats))
- if addLayer:
- args = {}
- if useId:
- args["ltype"] = "vector"
- else:
- args["ltype"] = "command"
- return self.Map.AddLayer(
- name=globalvar.QUERYLAYER,
- command=cmd,
- active=True,
- hidden=True,
- opacity=1.0,
- render=True,
- **args,
- )
- else:
- return cmd
- def OnMeasureDistance(self, event):
- self._onMeasure(MeasureDistanceController)
- def OnMeasureArea(self, event):
- self._onMeasure(MeasureAreaController)
- def _onMeasure(self, controller):
- """Starts measurement mode.
- :param controller: measurement class (MeasureDistanceController, MeasureAreaController)
- """
- self.measureController = controller(self._giface, mapWindow=self.GetMapWindow())
- # assure that the mode is ended and lines are cleared whenever other
- # tool is selected
- self._toolSwitcher.toggleToolChanged.connect(
- lambda: self.measureController.Stop()
- )
- self.measureController.Start()
- def OnProfile(self, event):
- """Launch profile tool"""
- rasters = []
- layers = self._giface.GetLayerList().GetSelectedLayers()
- for layer in layers:
- if layer.type == "raster":
- rasters.append(layer.maplayer.name)
- self.Profile(rasters=rasters)
- def Profile(self, rasters=None):
- """Launch profile tool"""
- from wxplot.profile import ProfileFrame
- self.profileController = ProfileController(
- self._giface, mapWindow=self.GetMapWindow()
- )
- win = ProfileFrame(
- parent=self,
- giface=self._giface,
- rasterList=rasters,
- units=self.Map.projinfo["units"],
- controller=self.profileController,
- )
- win.Show()
- # Open raster select dialog to make sure that a raster (and
- # the desired raster) is selected to be profiled
- win.OnSelectRaster(None)
- def OnHistogramPyPlot(self, event):
- """Init PyPlot histogram display canvas and tools"""
- raster = []
- for layer in self._giface.GetLayerList().GetSelectedLayers():
- if layer.maplayer.GetType() == "raster":
- raster.append(layer.maplayer.GetName())
- from wxplot.histogram import HistogramPlotFrame
- win = HistogramPlotFrame(parent=self, giface=self._giface, rasterList=raster)
- win.CentreOnParent()
- win.Show()
- def OnScatterplot(self, event):
- """Init PyPlot scatterplot display canvas and tools"""
- raster = []
- for layer in self._giface.GetLayerList().GetSelectedLayers():
- if layer.maplayer.GetType() == "raster":
- raster.append(layer.maplayer.GetName())
- from wxplot.scatter import ScatterFrame
- win = ScatterFrame(parent=self, giface=self._giface, rasterList=raster)
- win.CentreOnParent()
- win.Show()
- # Open raster select dialog to make sure that at least 2 rasters (and the desired rasters)
- # are selected to be plotted
- win.OnSelectRaster(None)
- def OnHistogram(self, event):
- """Init histogram display canvas and tools"""
- from modules.histogram import HistogramFrame
- win = HistogramFrame(self, giface=self._giface)
- win.CentreOnParent()
- win.Show()
- win.Refresh()
- win.Update()
- def _activateOverlay(self, overlayId):
- """Launch decoration dialog according to overlay id.
- :param overlayId: id of overlay
- """
- if overlayId in self.decorations:
- dlg = self.decorations[overlayId].dialog
- if dlg.IsShown():
- dlg.SetFocus()
- dlg.Raise()
- else:
- dlg.Show()
- def RemoveOverlay(self, overlayId):
- """Hide overlay.
- :param overlayId: id of overlay
- """
- del self._decorationWindows[self.decorations[overlayId].dialog]
- self.decorations[overlayId].Remove()
- del self.decorations[overlayId]
- def AddBarscale(self, cmd=None):
- """Handler for scale bar map decoration menu selection."""
- if self.IsPaneShown("3d"):
- self.MapWindow3D.SetDrawScalebar((70, 70))
- return
- if cmd:
- show = False
- else:
- show = True
- cmd = ["d.barscale"]
- # Decoration overlay control dialog
- GUI(parent=self, giface=self._giface, show=show, modal=False).ParseCommand(
- cmd, completed=(self.GetOptData, None, None)
- )
- self.MapWindow.mouse["use"] = "pointer"
- def AddLegendRast(self, cmd=None):
- """Handler for legend raster map decoration menu selection."""
- if cmd:
- show = False
- else:
- show = True
- cmd = ["d.legend"]
- layers = self._giface.GetLayerList().GetSelectedLayers()
- for layer in layers:
- if layer.type == "raster":
- cmd.append("raster={rast}".format(rast=layer.maplayer.name))
- break
- GUI(parent=self, giface=self._giface, show=show, modal=False).ParseCommand(
- cmd, completed=(self.GetOptData, None, None)
- )
- self.MapWindow.mouse["use"] = "pointer"
- def AddLegendVect(self, cmd=None, showDialog=None):
- """Handler for legend vector map decoration menu selection."""
- if cmd:
- show = False
- else:
- show = True
- cmd = ["d.legend.vect"]
- GUI(parent=self, giface=self._giface, show=show, modal=False).ParseCommand(
- cmd, completed=(self.GetOptData, None, None)
- )
- self.MapWindow.mouse["use"] = "pointer"
- def AddArrow(self, cmd=None):
- """Handler for north arrow menu selection."""
- if self.IsPaneShown("3d"):
- # here was opening of appearance page of nviz notebook
- # but now moved to MapWindow3D where are other problematic nviz
- # calls
- self.MapWindow3D.SetDrawArrow((70, 70))
- return
- if cmd:
- show = False
- else:
- show = True
- cmd = ["d.northarrow"]
- # Decoration overlay control dialog
- GUI(parent=self, giface=self._giface, show=show, modal=False).ParseCommand(
- cmd, completed=(self.GetOptData, None, None)
- )
- self.MapWindow.mouse["use"] = "pointer"
- def AddDtext(self, cmd=None):
- """Handler for d.text menu selection."""
- if cmd:
- show = False
- else:
- show = True
- cmd = ["d.text"]
- # Decoration overlay control dialog
- GUI(parent=self, giface=self._giface, show=show, modal=False).ParseCommand(
- cmd, completed=(self.GetOptData, None, None)
- )
- self.MapWindow.mouse["use"] = "pointer"
- def GetOptData(self, dcmd, layer, params, propwin):
- """Called after options are set through module dialog.
- :param dcmd: resulting command
- :param layer: not used
- :param params: module parameters (not used)
- :param propwin: dialog window
- """
- if not dcmd:
- return
- if propwin in self._decorationWindows:
- overlay = self._decorationWindows[propwin]
- else:
- cmd = dcmd[0]
- if cmd == "d.northarrow":
- overlay = ArrowController(self.Map, self._giface)
- elif cmd == "d.barscale":
- overlay = BarscaleController(self.Map, self._giface)
- elif cmd == "d.legend":
- overlay = LegendController(self.Map, self._giface)
- elif cmd == "d.legend.vect":
- overlay = LegendVectController(self.Map, self._giface)
- elif cmd == "d.text":
- overlay = DtextController(self.Map, self._giface)
- self.decorations[overlay.id] = overlay
- overlay.overlayChanged.connect(
- lambda: self.MapWindow2D.UpdateMap(render=False, renderVector=False)
- )
- if self.IsPaneShown("3d"):
- overlay.overlayChanged.connect(self.MapWindow3D.UpdateOverlays)
- overlay.dialog = propwin
- self._decorationWindows[propwin] = overlay
- overlay.cmd = dcmd
- overlay.Show()
- def OnZoomToMap(self, event):
- """Set display extents to match selected raster (including
- NULLs) or vector map.
- """
- Debug.msg(3, "MapFrame.OnZoomToMap()")
- layers = None
- if self.IsStandalone():
- layers = self.MapWindow.GetMap().GetListOfLayers(active=False)
- self.MapWindow.ZoomToMap(layers=layers)
- def OnZoomToRaster(self, event):
- """Set display extents to match selected raster map (ignore NULLs)"""
- self.MapWindow.ZoomToMap(ignoreNulls=True)
- def OnZoomToSaved(self, event):
- """Set display geometry to match extents in
- saved region file
- """
- self.MapWindow.SetRegion(zoomOnly=True)
- def OnSetDisplayToWind(self, event):
- """Set computational region (WIND file) to match display
- extents
- """
- self.MapWindow.DisplayToWind()
- def OnSetWindToRegion(self, event):
- """Set computational region (WIND file) from named region
- file
- """
- self.MapWindow.SetRegion(zoomOnly=False)
- def OnSetExtentToWind(self, event):
- """Set compulational region extent interactively"""
- self.MapWindow.SetModeDrawRegion()
- def OnSaveDisplayRegion(self, event):
- """Save display extents to named region file."""
- self.MapWindow.SaveRegion(display=True)
- def OnSaveWindRegion(self, event):
- """Save computational region to named region file."""
- self.MapWindow.SaveRegion(display=False)
- def OnZoomMenu(self, event):
- """Popup Zoom menu"""
- zoommenu = Menu()
- for label, handler in (
- (_("Zoom to default region"), self.OnZoomToDefault),
- (_("Zoom to saved region"), self.OnZoomToSaved),
- (None, None),
- (
- _("Set computational region extent from display"),
- self.OnSetDisplayToWind,
- ),
- (
- _("Set computational region extent interactively"),
- self.OnSetExtentToWind,
- ),
- (_("Set computational region from named region"), self.OnSetWindToRegion),
- (None, None),
- (_("Save display geometry to named region"), self.OnSaveDisplayRegion),
- (_("Save computational region to named region"), self.OnSaveWindRegion),
- ):
- if label:
- mid = wx.MenuItem(zoommenu, wx.ID_ANY, label)
- zoommenu.AppendItem(mid)
- self.Bind(wx.EVT_MENU, handler, mid)
- else:
- zoommenu.AppendSeparator()
- # Popup the menu. If an item is selected then its handler will
- # be called before PopupMenu returns.
- self.PopupMenu(zoommenu)
- zoommenu.Destroy()
- def SetProperties(
- self,
- render=False,
- mode=0,
- showCompExtent=False,
- constrainRes=False,
- projection=False,
- alignExtent=True,
- ):
- """Set properties of map display window"""
- self.mapWindowProperties.autoRender = render
- if self.statusbarManager:
- self.statusbarManager.SetMode(mode)
- self.StatusbarUpdate()
- self.SetProperty("projection", projection)
- self.mapWindowProperties.showRegion = showCompExtent
- self.mapWindowProperties.alignExtent = alignExtent
- self.mapWindowProperties.resolution = constrainRes
- def IsStandalone(self):
- """Check if Map display is standalone
- .. deprecated:: 7.0
- """
- # TODO: once it is removed from 2 places in vdigit it can be deleted
- # here and also in base class and other classes in the tree (hopefully)
- # and one place here still uses IsStandalone
- Debug.msg(
- 1,
- "MapFrame.IsStandalone(): Method IsStandalone is"
- "deprecated, use some general approach instead such as"
- " Signals or giface",
- )
- if self._layerManager:
- return False
- return True
- def GetLayerManager(self):
- """Get reference to Layer Manager
- :return: window reference
- :return: None (if standalone)
- .. deprecated:: 7.0
- """
- Debug.msg(
- 1,
- "MapFrame.GetLayerManager(): Method GetLayerManager is"
- "deprecated, use some general approach instead such as"
- " Signals or giface",
- )
- return self._layerManager
- def GetMapToolbar(self):
- """Returns toolbar with zooming tools"""
- return self.toolbars["map"] if "map" in self.toolbars else None
- def GetToolbarNames(self):
- """Return toolbar names"""
- return self.toolbars.keys()
- def GetDialog(self, name):
- """Get selected dialog if exist"""
- return self.dialogs.get(name, None)
- def OnVNet(self, event):
- """Dialog for v.net* modules"""
- if self.dialogs["vnet"]:
- self.dialogs["vnet"].Raise()
- return
- from vnet.dialogs import VNETDialog
- self.dialogs["vnet"] = VNETDialog(parent=self, giface=self._giface)
- self.closingVNETDialog.connect(self.dialogs["vnet"].OnCloseDialog)
- self.dialogs["vnet"].CenterOnScreen()
- self.dialogs["vnet"].Show()
- def ResetPointer(self):
- """Sets pointer mode.
- Sets pointer and toggles it (e.g. after unregistration of mouse
- handler).
- """
- self.GetMapToolbar().SelectDefault()
- def _switchMapWindow(self, map_win):
- """Notifies activated and disactivated map_wins."""
- self.MapWindow.DisactivateWin()
- map_win.ActivateWin()
- self.MapWindow = map_win
- def AddRDigit(self):
- """Adds raster digitizer: creates toolbar and digitizer controller,
- binds events and signals."""
- from rdigit.controller import RDigitController, EVT_UPDATE_PROGRESS
- from rdigit.toolbars import RDigitToolbar
- self.rdigit = RDigitController(self._giface, mapWindow=self.GetMapWindow())
- self.toolbars["rdigit"] = RDigitToolbar(
- parent=self,
- giface=self._giface,
- controller=self.rdigit,
- toolSwitcher=self._toolSwitcher,
- )
- # connect signals
- self.rdigit.newRasterCreated.connect(self.toolbars["rdigit"].NewRasterAdded)
- self.rdigit.newRasterCreated.connect(
- lambda name: self._giface.mapCreated.emit(name=name, ltype="raster")
- )
- self.rdigit.newFeatureCreated.connect(self.toolbars["rdigit"].UpdateCellValues)
- self.rdigit.uploadMapCategories.connect(
- self.toolbars["rdigit"].UpdateCellValues
- )
- self.rdigit.showNotification.connect(lambda text: self.SetStatusText(text, 0))
- self.rdigit.quitDigitizer.connect(self.QuitRDigit)
- self.rdigit.Bind(
- EVT_UPDATE_PROGRESS,
- lambda evt: self.statusbarManager.SetProgress(
- evt.range, evt.value, evt.text
- ),
- )
- rasters = self.GetMap().GetListOfLayers(
- ltype="raster", mapset=grass.gisenv()["MAPSET"]
- )
- self.toolbars["rdigit"].UpdateRasterLayers(rasters)
- self.toolbars["rdigit"].SelectDefault()
- self.GetMap().layerAdded.connect(self._updateRDigitLayers)
- self.GetMap().layerRemoved.connect(self._updateRDigitLayers)
- self.GetMap().layerChanged.connect(self._updateRDigitLayers)
- self._mgr.AddPane(
- self.toolbars["rdigit"],
- wx.aui.AuiPaneInfo()
- .Name("rdigit toolbar")
- .Caption(_("Raster Digitizer Toolbar"))
- .ToolbarPane()
- .Top()
- .Row(1)
- .LeftDockable(False)
- .RightDockable(False)
- .BottomDockable(False)
- .TopDockable(True)
- .Floatable()
- .CloseButton(False)
- .Layer(2)
- .DestroyOnClose()
- .BestSize((self.toolbars["rdigit"].GetBestSize())),
- )
- self._mgr.Update()
- self.rdigit.Start()
- def _updateRDigitLayers(self, layer):
- mapset = grass.gisenv()["MAPSET"]
- self.toolbars["rdigit"].UpdateRasterLayers(
- rasters=self.GetMap().GetListOfLayers(ltype="raster", mapset=mapset)
- )
- def QuitRDigit(self):
- """Calls digitizer cleanup, removes digitizer object and disconnects
- signals from Map."""
- if not self.IsStandalone():
- self.rdigit.CleanUp()
- # disconnect updating layers
- self.GetMap().layerAdded.disconnect(self._updateRDigitLayers)
- self.GetMap().layerRemoved.disconnect(self._updateRDigitLayers)
- self.GetMap().layerChanged.disconnect(self._updateRDigitLayers)
- self._toolSwitcher.toggleToolChanged.disconnect(
- self.toolbars["rdigit"].CheckSelectedTool,
- )
- self.RemoveToolbar("rdigit", destroy=True)
- self.rdigit = None
- else:
- self.Close()
- def QuitVDigit(self):
- """Quit VDigit"""
- if not self.IsStandalone():
- # disable the toolbar
- self.RemoveToolbar("vdigit", destroy=True)
- else:
- self.Close()
|