12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562 |
- """
- @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 sys
- 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 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=_("GRASS GIS - 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")
- # 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)
- if globalvar.wxPython3:
- 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}'. Details:\n{error}").format(command=' '.join(cmd), error=error)))
- def GetMapWindow(self):
- return self.MapWindow
- def SetTitleNumber(self, displayId=1):
- """Set map display title"""
- try:
- grassVersion = grass.version()['version']
- except KeyError:
- sys.stderr.write(_("Unable to get GRASS version\n"))
- grassVersion = "?"
- gisenv = grass.gisenv()
- title = _("GRASS GIS %(version)s Map Display: %(id)s - Location: %(loc)s@%(mapset)s") % {
- 'version': grassVersion,
- 'id': str(displayId),
- 'loc': gisenv["LOCATION_NAME"],
- 'mapset': gisenv["MAPSET"]}
- self.SetTitle(title)
- 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.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 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()
- 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()
- 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:]:
- p, val = param.split('=')
- 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 = wx.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 OnCloseWindow(self, event):
- """Window closed.
- Also close associated layer tree page
- """
- Debug.msg(2, "MapFrame.OnCloseWindow(): function starts")
- pgnum = None
- 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 not self._layerManager:
- self.Destroy()
- elif self.page:
- pgnum = self.layerbook.GetPageIndex(self.page)
- if pgnum > -1:
- self.layerbook.RemovePage(pgnum)
- Debug.msg(2, "MapFrame.OnCloseWindow(): function ends")
- self._mgr.UnInit()
- 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)
- 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'])
- 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, 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, 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, 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
- """
- 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 = wx.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 properies of map display window"""
- self.mapWindowProperties.autoRender = render
- self.statusbarManager.SetMode(mode)
- self.StatusbarUpdate()
- self.mapWindowProperties.showRegion = showCompExtent
- self.mapWindowProperties.alignExtent = alignExtent
- self.mapWindowProperties.resolution = constrainRes
- self.SetProperty('projection', projection)
- 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"
- "depreciated, 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"
- "depreciated, 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.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, 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."""
- 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.RemoveToolbar('rdigit', destroy=True)
- self.rdigit = None
|