frame.py 52 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358
  1. """!
  2. @package mapdisp.frame
  3. @brief Map display with toolbar for various display management
  4. functions, and additional toolbars (vector digitizer, 3d view).
  5. Can be used either from Layer Manager or as d.mon backend.
  6. Classes:
  7. - mapdisp::MapFrame
  8. (C) 2006-2013 by the GRASS Development Team
  9. This program is free software under the GNU General Public License
  10. (>=v2). Read the file COPYING that comes with GRASS for details.
  11. @author Michael Barton
  12. @author Jachym Cepicky
  13. @author Martin Landa <landa.martin gmail.com>
  14. @author Vaclav Petras <wenzeslaus gmail.com> (SingleMapFrame, handlers support)
  15. @author Anna Kratochvilova <kratochanna gmail.com> (SingleMapFrame)
  16. @author Stepan Turek <stepan.turek seznam.cz> (handlers support)
  17. """
  18. import os
  19. import sys
  20. import math
  21. import copy
  22. from core import globalvar
  23. import wx
  24. import wx.aui
  25. if os.path.join(globalvar.ETCWXDIR, "icons") not in sys.path:
  26. sys.path.append(os.path.join(globalvar.ETCWXDIR, "icons"))
  27. if os.path.join(globalvar.ETCDIR, "python") not in sys.path:
  28. sys.path.append(os.path.join(globalvar.ETCDIR, "python"))
  29. from core import globalvar
  30. from core.render import Map
  31. from vdigit.toolbars import VDigitToolbar
  32. from mapdisp.toolbars import MapToolbar, NvizIcons
  33. from mapdisp.gprint import PrintOptions
  34. from core.gcmd import GError, GMessage
  35. from dbmgr.dialogs import DisplayAttributesDialog
  36. from core.utils import ListOfCatsToRange, GetLayerNameFromCmd, _
  37. from gui_core.dialogs import GetImageHandlers, ImageSizeDialog, DecorationDialog, TextLayerDialog, \
  38. DECOR_DIALOG_LEGEND, DECOR_DIALOG_BARSCALE
  39. from core.debug import Debug
  40. from core.settings import UserSettings
  41. from gui_core.mapdisp import SingleMapFrame
  42. from gui_core.query import QueryDialog, PrepareQueryResults
  43. from mapdisp.mapwindow import BufferedWindow
  44. from mapdisp.overlays import LegendController, BarscaleController
  45. from modules.histogram import HistogramFrame
  46. from wxplot.histogram import HistogramPlotFrame
  47. from wxplot.profile import ProfileFrame
  48. from wxplot.scatter import ScatterFrame
  49. from mapdisp import statusbar as sb
  50. import grass.script as grass
  51. haveCtypes = False
  52. class MapFrame(SingleMapFrame):
  53. """!Main frame for map display window. Drawing takes place in
  54. child double buffered drawing window.
  55. """
  56. def __init__(self, parent, giface, title = _("GRASS GIS - Map display"),
  57. toolbars = ["map"], tree = None, notebook = None, lmgr = None,
  58. page = None, Map = Map(), auimgr = None, name = 'MapWindow', **kwargs):
  59. """!Main map display window with toolbars, statusbar and
  60. BufferedWindow (map canvas)
  61. @param toolbars array of activated toolbars, e.g. ['map', 'digit']
  62. @param tree reference to layer tree
  63. @param notebook control book ID in Layer Manager
  64. @param lmgr Layer Manager
  65. @param page notebook page with layer tree
  66. @param Map instance of render.Map
  67. @param auimgs AUI manager
  68. @param name frame name
  69. @param kwargs wx.Frame attributes
  70. """
  71. SingleMapFrame.__init__(self, parent = parent, title = title,
  72. Map = Map, auimgr = auimgr, name = name, **kwargs)
  73. self._giface = giface
  74. self._layerManager = lmgr # Layer Manager object
  75. self.tree = tree # Layer Manager layer tree object
  76. self.page = page # Notebook page holding the layer tree
  77. self.layerbook = notebook # Layer Manager layer tree notebook
  78. #
  79. # Add toolbars
  80. #
  81. for toolb in toolbars:
  82. self.AddToolbar(toolb)
  83. #
  84. # Add statusbar
  85. #
  86. # items for choice
  87. self.statusbarItems = [sb.SbCoordinates,
  88. sb.SbRegionExtent,
  89. sb.SbCompRegionExtent,
  90. sb.SbShowRegion,
  91. sb.SbAlignExtent,
  92. sb.SbResolution,
  93. sb.SbDisplayGeometry,
  94. sb.SbMapScale,
  95. sb.SbGoTo,
  96. sb.SbProjection]
  97. self.statusbarItemsHiddenInNviz = (sb.SbAlignExtent,
  98. sb.SbDisplayGeometry,
  99. sb.SbShowRegion,
  100. sb.SbResolution,
  101. sb.SbMapScale)
  102. # create statusbar and its manager
  103. statusbar = self.CreateStatusBar(number = 4, style = 0)
  104. statusbar.SetStatusWidths([-5, -2, -1, -1])
  105. self.statusbarManager = sb.SbManager(mapframe = self, statusbar = statusbar)
  106. # fill statusbar manager
  107. self.statusbarManager.AddStatusbarItemsByClass(self.statusbarItems, mapframe = self, statusbar = statusbar)
  108. self.statusbarManager.AddStatusbarItem(sb.SbMask(self, statusbar = statusbar, position = 2))
  109. sbRender = sb.SbRender(self, statusbar = statusbar, position = 3)
  110. sbRender.autoRender.connect(lambda state: self.OnRender(None) if state else None)
  111. self.statusbarManager.AddStatusbarItem(sbRender)
  112. self.statusbarManager.Update()
  113. #
  114. self.Map.updateProgress.connect(self.statusbarManager.SetProgress)
  115. # init decoration objects
  116. self.decorations = {}
  117. self.legend = LegendController(self.Map)
  118. self.barscale = BarscaleController(self.Map)
  119. self.decorations[self.legend.id] = self.legend
  120. self.decorations[self.barscale.id] = self.barscale
  121. #
  122. # Init map display (buffered DC & set default cursor)
  123. #
  124. self.MapWindow2D = BufferedWindow(self, giface = self._giface, id = wx.ID_ANY,
  125. Map = self.Map, frame = self, tree = self.tree,
  126. lmgr = self._layerManager, overlays = self.decorations)
  127. self.MapWindow2D.mapQueried.connect(self.Query)
  128. # enable or disable zoom history tool
  129. self.MapWindow2D.zoomHistoryAvailable.connect(
  130. lambda:
  131. self.GetMapToolbar().Enable('zoomBack', enable=True))
  132. self.MapWindow2D.zoomHistoryUnavailable.connect(
  133. lambda:
  134. self.GetMapToolbar().Enable('zoomBack', enable=False))
  135. # manage the state of toolbars connected to mouse cursor
  136. self.MapWindow2D.mouseHandlerRegistered.connect(
  137. lambda:
  138. self.UpdateTools(None))
  139. self.MapWindow2D.mouseHandlerUnregistered.connect(self.ResetPointer)
  140. self.MapWindow2D.InitZoomHistory()
  141. self.MapWindow2D.zoomChanged.connect(self.StatusbarUpdate)
  142. self._giface.updateMap.connect(self.MapWindow2D.UpdateMap)
  143. # default is 2D display mode
  144. self.MapWindow = self.MapWindow2D
  145. self.MapWindow.SetCursor(self.cursors["default"])
  146. # used by vector digitizer
  147. self.MapWindowVDigit = None
  148. # used by Nviz (3D display mode)
  149. self.MapWindow3D = None
  150. #
  151. # initialize region values
  152. #
  153. self._initMap(Map = self.Map)
  154. #
  155. # Bind various events
  156. #
  157. self.Bind(wx.EVT_ACTIVATE, self.OnFocus)
  158. self.Bind(wx.EVT_CLOSE, self.OnCloseWindow)
  159. self.Bind(wx.EVT_SIZE, self.OnSize)
  160. #
  161. # Update fancy gui style
  162. #
  163. self._mgr.AddPane(self.MapWindow, wx.aui.AuiPaneInfo().CentrePane().
  164. Dockable(False).BestSize((-1,-1)).Name('2d').
  165. CloseButton(False).DestroyOnClose(True).
  166. Layer(0))
  167. self._mgr.Update()
  168. #
  169. # Init print module and classes
  170. #
  171. self.printopt = PrintOptions(self, self.MapWindow)
  172. #
  173. # Re-use dialogs
  174. #
  175. self.dialogs = {}
  176. self.dialogs['attributes'] = None
  177. self.dialogs['category'] = None
  178. self.dialogs['barscale'] = None
  179. self.dialogs['legend'] = None
  180. self.dialogs['vnet'] = None
  181. self.dialogs['query'] = None
  182. self.decorationDialog = None # decoration/overlays
  183. def OnSize(self, event):
  184. """!Adjust statusbar on changing size"""
  185. # reposition checkbox in statusbar
  186. self.StatusbarReposition()
  187. # update statusbar
  188. self.StatusbarUpdate()
  189. def GetMapWindow(self):
  190. return self.MapWindow
  191. def _addToolbarVDigit(self):
  192. """!Add vector digitizer toolbar
  193. """
  194. from vdigit.main import haveVDigit, VDigit
  195. if not haveVDigit:
  196. from vdigit import errorMsg
  197. self.toolbars['map'].combo.SetValue(_("2D view"))
  198. GError(_("Unable to start wxGUI vector digitizer.\n"
  199. "Details: %s") % errorMsg, parent = self)
  200. return
  201. if self._layerManager:
  202. log = self._layerManager.GetLogWindow()
  203. else:
  204. log = None
  205. if not self.MapWindowVDigit:
  206. from vdigit.mapwindow import VDigitWindow
  207. self.MapWindowVDigit = VDigitWindow(parent = self, giface = self._giface,
  208. id = wx.ID_ANY, frame = self,
  209. Map = self.Map, tree = self.tree,
  210. lmgr = self._layerManager)
  211. self.MapWindowVDigit.Show()
  212. self._mgr.AddPane(self.MapWindowVDigit, wx.aui.AuiPaneInfo().CentrePane().
  213. Dockable(False).BestSize((-1,-1)).Name('vdigit').
  214. CloseButton(False).DestroyOnClose(True).
  215. Layer(0))
  216. self.MapWindow = self.MapWindowVDigit
  217. if self._mgr.GetPane('2d').IsShown():
  218. self._mgr.GetPane('2d').Hide()
  219. elif self._mgr.GetPane('3d').IsShown():
  220. self._mgr.GetPane('3d').Hide()
  221. self._mgr.GetPane('vdigit').Show()
  222. self.toolbars['vdigit'] = VDigitToolbar(parent = self, MapWindow = self.MapWindow,
  223. digitClass = VDigit, giface = self._giface,
  224. layerTree = self.tree, log = log)
  225. self.MapWindowVDigit.SetToolbar(self.toolbars['vdigit'])
  226. self._mgr.AddPane(self.toolbars['vdigit'],
  227. wx.aui.AuiPaneInfo().
  228. Name("vdigittoolbar").Caption(_("Vector Digitizer Toolbar")).
  229. ToolbarPane().Top().Row(1).
  230. LeftDockable(False).RightDockable(False).
  231. BottomDockable(False).TopDockable(True).
  232. CloseButton(False).Layer(2).
  233. BestSize((self.toolbars['vdigit'].GetBestSize())))
  234. # change mouse to draw digitized line
  235. self.MapWindow.mouse['box'] = "point"
  236. self.MapWindow.zoomtype = 0
  237. self.MapWindow.pen = wx.Pen(colour = 'red', width = 2, style = wx.SOLID)
  238. self.MapWindow.polypen = wx.Pen(colour = 'green', width = 2, style = wx.SOLID)
  239. def AddNviz(self):
  240. """!Add 3D view mode window
  241. """
  242. from nviz.main import haveNviz, GLWindow, errorMsg
  243. # check for GLCanvas and OpenGL
  244. if not haveNviz:
  245. self.toolbars['map'].combo.SetValue(_("2D view"))
  246. GError(parent = self,
  247. message = _("Unable to switch to 3D display mode.\nThe Nviz python extension "
  248. "was not found or loaded properly.\n"
  249. "Switching back to 2D display mode.\n\nDetails: %s" % errorMsg))
  250. return
  251. # disable 3D mode for other displays
  252. for page in range(0, self._layerManager.GetLayerNotebook().GetPageCount()):
  253. mapdisp = self._layerManager.GetLayerNotebook().GetPage(page).maptree.GetMapDisplay()
  254. if self._layerManager.GetLayerNotebook().GetPage(page) != self._layerManager.currentPage:
  255. if '3D' in mapdisp.toolbars['map'].combo.GetString(1):
  256. mapdisp.toolbars['map'].combo.Delete(1)
  257. self.toolbars['map'].Enable2D(False)
  258. # add rotate tool to map toolbar
  259. self.toolbars['map'].InsertTool((('rotate', NvizIcons['rotate'],
  260. self.OnRotate, wx.ITEM_CHECK, 7),)) # 7 is position
  261. self.toolbars['map'].InsertTool((('flyThrough', NvizIcons['flyThrough'],
  262. self.OnFlyThrough, wx.ITEM_CHECK, 8),))
  263. self.toolbars['map'].ChangeToolsDesc(mode2d = False)
  264. # update status bar
  265. self.statusbarManager.HideStatusbarChoiceItemsByClass(self.statusbarItemsHiddenInNviz)
  266. self.statusbarManager.SetMode(0)
  267. # erase map window
  268. self.MapWindow.EraseMap()
  269. self._giface.WriteCmdLog(_("Starting 3D view mode..."))
  270. self.SetStatusText(_("Please wait, loading data..."), 0)
  271. # create GL window
  272. if not self.MapWindow3D:
  273. self.MapWindow3D = GLWindow(self, giface = self._giface, id = wx.ID_ANY, frame = self,
  274. Map = self.Map, tree = self.tree, lmgr = self._layerManager)
  275. self.MapWindow = self.MapWindow3D
  276. self.MapWindow.SetCursor(self.cursors["default"])
  277. # add Nviz notebookpage
  278. self._layerManager.AddNvizTools()
  279. # switch from MapWindow to MapWindowGL
  280. self._mgr.GetPane('2d').Hide()
  281. self._mgr.AddPane(self.MapWindow3D, wx.aui.AuiPaneInfo().CentrePane().
  282. Dockable(False).BestSize((-1,-1)).Name('3d').
  283. CloseButton(False).DestroyOnClose(True).
  284. Layer(0))
  285. self.MapWindow3D.Show()
  286. self.MapWindow3D.ResetViewHistory()
  287. self.MapWindow3D.UpdateView(None)
  288. else:
  289. self.MapWindow = self.MapWindow3D
  290. os.environ['GRASS_REGION'] = self.Map.SetRegion(windres = True, windres3 = True)
  291. self.MapWindow3D.GetDisplay().Init()
  292. del os.environ['GRASS_REGION']
  293. # switch from MapWindow to MapWindowGL
  294. self._mgr.GetPane('2d').Hide()
  295. self._mgr.GetPane('3d').Show()
  296. # add Nviz notebookpage
  297. self._layerManager.AddNvizTools()
  298. self.MapWindow3D.ResetViewHistory()
  299. for page in ('view', 'light', 'fringe', 'constant', 'cplane', 'animation'):
  300. self._layerManager.nviz.UpdatePage(page)
  301. self.MapWindow3D.overlays = self.MapWindow2D.overlays
  302. self.MapWindow3D.textdict = self.MapWindow2D.textdict
  303. # update overlays needs to be called after because getClientSize
  304. # is called during update and it must give reasonable values
  305. wx.CallAfter(self.MapWindow3D.UpdateOverlays)
  306. self.SetStatusText("", 0)
  307. self._mgr.Update()
  308. def RemoveNviz(self):
  309. """!Restore 2D view"""
  310. try:
  311. self.toolbars['map'].RemoveTool(self.toolbars['map'].rotate)
  312. self.toolbars['map'].RemoveTool(self.toolbars['map'].flyThrough)
  313. except AttributeError:
  314. pass
  315. # update status bar
  316. self.statusbarManager.ShowStatusbarChoiceItemsByClass(self.statusbarItemsHiddenInNviz)
  317. self.statusbarManager.SetMode(UserSettings.Get(group = 'display',
  318. key = 'statusbarMode',
  319. subkey = 'selection'))
  320. self.SetStatusText(_("Please wait, unloading data..."), 0)
  321. self._giface.WriteCmdLog(_("Switching back to 2D view mode..."))
  322. if self.MapWindow3D:
  323. self.MapWindow3D.OnClose(event = None)
  324. # switch from MapWindowGL to MapWindow
  325. self._mgr.GetPane('2d').Show()
  326. self._mgr.GetPane('3d').Hide()
  327. self.MapWindow = self.MapWindow2D
  328. # remove nviz notebook page
  329. self._layerManager.RemoveNvizTools()
  330. try:
  331. self.MapWindow2D.overlays = self.MapWindow3D.overlays
  332. self.MapWindow2D.textdict = self.MapWindow3D.textdict
  333. except AttributeError:
  334. pass
  335. self.MapWindow.UpdateMap()
  336. self._mgr.Update()
  337. def AddToolbar(self, name, fixed = False):
  338. """!Add defined toolbar to the window
  339. Currently recognized toolbars are:
  340. - 'map' - basic map toolbar
  341. - 'vdigit' - vector digitizer
  342. @param name toolbar to add
  343. @param fixed fixed toolbar
  344. """
  345. # default toolbar
  346. if name == "map":
  347. self.toolbars['map'] = MapToolbar(self, self.Map)
  348. self._mgr.AddPane(self.toolbars['map'],
  349. wx.aui.AuiPaneInfo().
  350. Name("maptoolbar").Caption(_("Map Toolbar")).
  351. ToolbarPane().Top().Name('mapToolbar').
  352. LeftDockable(False).RightDockable(False).
  353. BottomDockable(False).TopDockable(True).
  354. CloseButton(False).Layer(2).
  355. BestSize((self.toolbars['map'].GetBestSize())))
  356. # vector digitizer
  357. elif name == "vdigit":
  358. self.toolbars['map'].combo.SetValue(_("Digitize"))
  359. self._addToolbarVDigit()
  360. if fixed:
  361. self.toolbars['map'].combo.Disable()
  362. self._mgr.Update()
  363. def RemoveToolbar (self, name):
  364. """!Removes defined toolbar from the window
  365. @todo Only hide, activate by calling AddToolbar()
  366. """
  367. # cannot hide main toolbar
  368. if name == "map":
  369. return
  370. self._mgr.DetachPane(self.toolbars[name])
  371. self.toolbars[name].Destroy()
  372. self.toolbars.pop(name)
  373. if name == 'vdigit':
  374. self._mgr.GetPane('vdigit').Hide()
  375. self._mgr.GetPane('2d').Show()
  376. self.MapWindow = self.MapWindow2D
  377. self.toolbars['map'].combo.SetValue(_("2D view"))
  378. self.toolbars['map'].Enable2D(True)
  379. self._mgr.Update()
  380. def IsPaneShown(self, name):
  381. """!Check if pane (toolbar, mapWindow ...) of given name is currently shown"""
  382. if self._mgr.GetPane(name).IsOk():
  383. return self._mgr.GetPane(name).IsShown()
  384. return False
  385. def OnFocus(self, event):
  386. """!Change choicebook page to match display.
  387. """
  388. # change bookcontrol page to page associated with display
  389. if self.page:
  390. pgnum = self.layerbook.GetPageIndex(self.page)
  391. if pgnum > -1:
  392. self.layerbook.SetSelection(pgnum)
  393. self._layerManager.currentPage = self.layerbook.GetCurrentPage()
  394. event.Skip()
  395. def RemoveQueryLayer(self):
  396. """!Removes temporary map layers (queries)"""
  397. qlayer = self.GetMap().GetListOfLayers(name = globalvar.QUERYLAYER)
  398. for layer in qlayer:
  399. self.GetMap().DeleteLayer(layer)
  400. def OnRender(self, event):
  401. """!Re-render map composition (each map layer)
  402. """
  403. self.RemoveQueryLayer()
  404. # delete tmp lines
  405. if self.MapWindow.mouse["use"] in ("measure",
  406. "profile"):
  407. self.MapWindow.polycoords = []
  408. self.MapWindow.ClearLines()
  409. # deselect features in vdigit
  410. if self.GetToolbar('vdigit'):
  411. if self.MapWindow.digit:
  412. self.MapWindow.digit.GetDisplay().SetSelected([])
  413. self.MapWindow.UpdateMap(render = True, renderVector = True)
  414. else:
  415. self.MapWindow.UpdateMap(render = True)
  416. # update statusbar
  417. self.StatusbarUpdate()
  418. def OnPointer(self, event):
  419. """!Pointer button clicked
  420. """
  421. if self.GetMapToolbar():
  422. if event:
  423. self.SwitchTool(self.toolbars['map'], event)
  424. self.toolbars['map'].action['desc'] = ''
  425. self.MapWindow.mouse['use'] = "pointer"
  426. self.MapWindow.mouse['box'] = "point"
  427. # change the cursor
  428. if self.GetToolbar('vdigit'):
  429. # digitization tool activated
  430. self.MapWindow.SetCursor(self.cursors["cross"])
  431. # reset mouse['box'] if needed
  432. if self.toolbars['vdigit'].GetAction() in ['addLine']:
  433. if self.toolbars['vdigit'].GetAction('type') in ['point', 'centroid']:
  434. self.MapWindow.mouse['box'] = 'point'
  435. else: # line, boundary
  436. self.MapWindow.mouse['box'] = 'line'
  437. elif self.toolbars['vdigit'].GetAction() in ['addVertex', 'removeVertex', 'splitLine',
  438. 'editLine', 'displayCats', 'queryMap',
  439. 'copyCats']:
  440. self.MapWindow.mouse['box'] = 'point'
  441. else: # moveLine, deleteLine
  442. self.MapWindow.mouse['box'] = 'box'
  443. else:
  444. self.MapWindow.SetCursor(self.cursors["default"])
  445. def OnRotate(self, event):
  446. """!Rotate 3D view
  447. """
  448. if self.GetMapToolbar():
  449. self.SwitchTool(self.toolbars['map'], event)
  450. self.toolbars['map'].action['desc'] = ''
  451. self.MapWindow.mouse['use'] = "rotate"
  452. # change the cursor
  453. self.MapWindow.SetCursor(self.cursors["hand"])
  454. def OnFlyThrough(self, event):
  455. """!Fly-through mode
  456. """
  457. if self.GetMapToolbar():
  458. self.SwitchTool(self.toolbars['map'], event)
  459. self.toolbars['map'].action['desc'] = ''
  460. self.MapWindow.mouse['use'] = "fly"
  461. # change the cursor
  462. self.MapWindow.SetCursor(self.cursors["hand"])
  463. self.MapWindow.SetFocus()
  464. def OnZoomRegion(self, event):
  465. """!Zoom to region
  466. """
  467. self.Map.getRegion()
  468. self.Map.getResolution()
  469. self.UpdateMap()
  470. # event.Skip()
  471. def OnAlignRegion(self, event):
  472. """!Align region
  473. """
  474. if not self.Map.alignRegion:
  475. self.Map.alignRegion = True
  476. else:
  477. self.Map.alignRegion = False
  478. # event.Skip()
  479. def SaveToFile(self, event):
  480. """!Save map to image
  481. """
  482. if self.IsPaneShown('3d'):
  483. filetype = "TIF file (*.tif)|*.tif|PPM file (*.ppm)|*.ppm"
  484. ltype = [{ 'ext' : 'tif', 'type' : 'tif' },
  485. { 'ext' : 'ppm', 'type' : 'ppm' }]
  486. else:
  487. img = self.MapWindow.img
  488. if not img:
  489. GMessage(parent = self,
  490. message = _("Nothing to render (empty map). Operation canceled."))
  491. return
  492. filetype, ltype = GetImageHandlers(img)
  493. # get size
  494. dlg = ImageSizeDialog(self)
  495. dlg.CentreOnParent()
  496. if dlg.ShowModal() != wx.ID_OK:
  497. dlg.Destroy()
  498. return
  499. width, height = dlg.GetValues()
  500. dlg.Destroy()
  501. # get filename
  502. dlg = wx.FileDialog(parent = self,
  503. message = _("Choose a file name to save the image "
  504. "(no need to add extension)"),
  505. wildcard = filetype,
  506. style = wx.SAVE | wx.FD_OVERWRITE_PROMPT)
  507. if dlg.ShowModal() == wx.ID_OK:
  508. path = dlg.GetPath()
  509. if not path:
  510. dlg.Destroy()
  511. return
  512. base, ext = os.path.splitext(path)
  513. fileType = ltype[dlg.GetFilterIndex()]['type']
  514. extType = ltype[dlg.GetFilterIndex()]['ext']
  515. if ext != extType:
  516. path = base + '.' + extType
  517. self.MapWindow.SaveToFile(path, fileType,
  518. width, height)
  519. dlg.Destroy()
  520. def PrintMenu(self, event):
  521. """
  522. Print options and output menu for map display
  523. """
  524. printmenu = wx.Menu()
  525. # Add items to the menu
  526. setup = wx.MenuItem(printmenu, wx.ID_ANY, _('Page setup'))
  527. printmenu.AppendItem(setup)
  528. self.Bind(wx.EVT_MENU, self.printopt.OnPageSetup, setup)
  529. preview = wx.MenuItem(printmenu, wx.ID_ANY, _('Print preview'))
  530. printmenu.AppendItem(preview)
  531. self.Bind(wx.EVT_MENU, self.printopt.OnPrintPreview, preview)
  532. doprint = wx.MenuItem(printmenu, wx.ID_ANY, _('Print display'))
  533. printmenu.AppendItem(doprint)
  534. self.Bind(wx.EVT_MENU, self.printopt.OnDoPrint, doprint)
  535. # Popup the menu. If an item is selected then its handler
  536. # will be called before PopupMenu returns.
  537. self.PopupMenu(printmenu)
  538. printmenu.Destroy()
  539. def OnCloseWindow(self, event):
  540. """!Window closed.
  541. Also close associated layer tree page
  542. """
  543. pgnum = None
  544. self.Map.Clean()
  545. # close edited map and 3D tools properly
  546. if self.GetToolbar('vdigit'):
  547. maplayer = self.toolbars['vdigit'].GetLayer()
  548. if maplayer:
  549. self.toolbars['vdigit'].OnExit()
  550. if self.IsPaneShown('3d'):
  551. self.RemoveNviz()
  552. if not self._layerManager:
  553. self.Destroy()
  554. elif self.page:
  555. pgnum = self.layerbook.GetPageIndex(self.page)
  556. if pgnum > -1:
  557. self.layerbook.DeletePage(pgnum)
  558. def Query(self, x, y):
  559. """!Query selected layers.
  560. @param x,y coordinates
  561. @param layers selected tree item layers
  562. """
  563. layers = self._giface.GetLayerList().GetSelectedLayers(checkedOnly=False)
  564. rast = []
  565. vect = []
  566. for layer in layers:
  567. name, found = GetLayerNameFromCmd(layer.cmd)
  568. if not found:
  569. continue
  570. ltype = layer.maplayer.GetType()
  571. if ltype == 'raster':
  572. rast.append(name)
  573. elif ltype in ('rgb', 'his'):
  574. for iname in name.split('\n'):
  575. rast.append(iname)
  576. elif ltype in ('vector', 'thememap', 'themechart'):
  577. vect.append(name)
  578. if vect:
  579. # check for vector maps open to be edited
  580. digitToolbar = self.GetToolbar('vdigit')
  581. if digitToolbar:
  582. lmap = digitToolbar.GetLayer().GetName()
  583. for name in vect:
  584. if lmap == name:
  585. self._giface.WriteWarning(_("Vector map <%s> "
  586. "opened for editing - skipped.") % lmap)
  587. vect.remove(name)
  588. if not (rast + vect):
  589. GMessage(parent = self,
  590. message = _('No raster or vector map layer selected for querying.'))
  591. return
  592. # set query snap distance for v.what at map unit equivalent of 10 pixels
  593. qdist = 10.0 * ((self.Map.region['e'] - self.Map.region['w']) / self.Map.width)
  594. # TODO: replace returning None by exception or so
  595. try:
  596. east, north = self.MapWindow.Pixel2Cell((x, y))
  597. except TypeError:
  598. return
  599. if not self.IsPaneShown('3d'):
  600. self.QueryMap(east, north, qdist, rast, vect)
  601. else:
  602. if rast:
  603. self.MapWindow.QuerySurface(x, y)
  604. if vect:
  605. self.QueryMap(east, north, qdist, rast = [], vect = vect)
  606. def QueryMap(self, east, north, qdist, rast, vect):
  607. """!Query raster or vector map layers by r/v.what
  608. @param east,north coordinates
  609. @param qdist query distance
  610. @param rast raster map names
  611. @param vect vector map names
  612. """
  613. Debug.msg(1, "QueryMap(): raster=%s vector=%s" % (','.join(rast),
  614. ','.join(vect)))
  615. # use display region settings instead of computation region settings
  616. self.tmpreg = os.getenv("GRASS_REGION")
  617. os.environ["GRASS_REGION"] = self.Map.SetRegion(windres = False)
  618. rastQuery = []
  619. vectQuery = []
  620. if rast:
  621. rastQuery = grass.raster_what(map=rast, coord=(east, north))
  622. if vect:
  623. vectQuery = grass.vector_what(map=vect, coord=(east, north), distance=qdist)
  624. self._QueryMapDone()
  625. if 'Id' in vectQuery:
  626. self._queryHighlight(vectQuery)
  627. result = rastQuery + vectQuery
  628. result = PrepareQueryResults(coordinates = (east, north), result = result)
  629. if self.dialogs['query']:
  630. self.dialogs['query'].Raise()
  631. self.dialogs['query'].SetData(result)
  632. else:
  633. self.dialogs['query'] = QueryDialog(parent = self, data = result)
  634. self.dialogs['query'].Bind(wx.EVT_CLOSE, self._oncloseQueryDialog)
  635. self.dialogs['query'].Show()
  636. def _oncloseQueryDialog(self, event):
  637. self.dialogs['query'] = None
  638. event.Skip()
  639. def _queryHighlight(self, vectQuery):
  640. """!Highlight category from query."""
  641. cats = name = None
  642. for res in vectQuery:
  643. cats = {res['Layer']: [res['Category']]}
  644. name = res['Map']
  645. try:
  646. qlayer = self.Map.GetListOfLayers(name = globalvar.QUERYLAYER)[0]
  647. except IndexError:
  648. qlayer = None
  649. if not (cats and name):
  650. if qlayer:
  651. self.Map.DeleteLayer(qlayer)
  652. self.MapWindow.UpdateMap(render = False, renderVector = False)
  653. return
  654. if not self.IsPaneShown('3d') and self.IsAutoRendered():
  655. # highlight feature & re-draw map
  656. if qlayer:
  657. qlayer.SetCmd(self.AddTmpVectorMapLayer(name, cats,
  658. useId = False,
  659. addLayer = False))
  660. else:
  661. qlayer = self.AddTmpVectorMapLayer(name, cats, useId = False)
  662. # set opacity based on queried layer
  663. # TODO fix
  664. # opacity = layer.maplayer.GetOpacity(float = True)
  665. # qlayer.SetOpacity(opacity)
  666. self.MapWindow.UpdateMap(render = False, renderVector = False)
  667. def _QueryMapDone(self):
  668. """!Restore settings after querying (restore GRASS_REGION)
  669. """
  670. if hasattr(self, "tmpreg"):
  671. if self.tmpreg:
  672. os.environ["GRASS_REGION"] = self.tmpreg
  673. elif 'GRASS_REGION' in os.environ:
  674. del os.environ["GRASS_REGION"]
  675. elif 'GRASS_REGION' in os.environ:
  676. del os.environ["GRASS_REGION"]
  677. if hasattr(self, "tmpreg"):
  678. del self.tmpreg
  679. def OnQuery(self, event):
  680. """!Query tools menu"""
  681. if self.GetMapToolbar():
  682. self.SwitchTool(self.toolbars['map'], event)
  683. self.toolbars['map'].action['desc'] = 'queryMap'
  684. self.MapWindow.mouse['use'] = "query"
  685. self.MapWindow.mouse['box'] = "point"
  686. self.MapWindow.zoomtype = 0
  687. # change the cursor
  688. self.MapWindow.SetCursor(self.cursors["cross"])
  689. def AddTmpVectorMapLayer(self, name, cats, useId = False, addLayer = True):
  690. """!Add temporal vector map layer to map composition
  691. @param name name of map layer
  692. @param useId use feature id instead of category
  693. """
  694. # color settings from ATM
  695. color = UserSettings.Get(group = 'atm', key = 'highlight', subkey = 'color')
  696. colorStr = str(color[0]) + ":" + \
  697. str(color[1]) + ":" + \
  698. str(color[2])
  699. # icon used in vector display and its size
  700. icon = ''
  701. size = 0
  702. # here we know that there is one selected layer and it is vector
  703. vparam = self._giface.GetLayerList().GetSelectedLayers()[0].cmd
  704. for p in vparam:
  705. if '=' in p:
  706. parg,pval = p.split('=', 1)
  707. if parg == 'icon': icon = pval
  708. elif parg == 'size': size = float(pval)
  709. pattern = ["d.vect",
  710. "map=%s" % name,
  711. "color=%s" % colorStr,
  712. "fcolor=%s" % colorStr,
  713. "width=%d" % UserSettings.Get(group = 'atm', key = 'highlight', subkey = 'width')]
  714. if icon != '':
  715. pattern.append('icon=%s' % icon)
  716. if size > 0:
  717. pattern.append('size=%i' % size)
  718. if useId:
  719. cmd = pattern
  720. cmd.append('-i')
  721. cmd.append('cats=%s' % str(cats))
  722. else:
  723. cmd = []
  724. for layer in cats.keys():
  725. cmd.append(copy.copy(pattern))
  726. lcats = cats[layer]
  727. cmd[-1].append("layer=%d" % layer)
  728. cmd[-1].append("cats=%s" % ListOfCatsToRange(lcats))
  729. if addLayer:
  730. if useId:
  731. return self.Map.AddLayer(ltype = 'vector', name = globalvar.QUERYLAYER, command = cmd,
  732. active = True, hidden = True, opacity = 1.0)
  733. else:
  734. return self.Map.AddLayer(ltype = 'command', name = globalvar.QUERYLAYER, command = cmd,
  735. active = True, hidden = True, opacity = 1.0)
  736. else:
  737. return cmd
  738. def OnMeasure(self, event):
  739. """!Init measurement routine that calculates map distance
  740. along transect drawn on map display
  741. """
  742. self.totaldist = 0.0 # total measured distance
  743. self.SwitchTool(self.toolbars['map'], event)
  744. # change mouse to draw line for measurement
  745. self.MapWindow.mouse['use'] = "measure"
  746. self.MapWindow.mouse['box'] = "line"
  747. self.MapWindow.zoomtype = 0
  748. self.MapWindow.pen = wx.Pen(colour = 'red', width = 2, style = wx.SHORT_DASH)
  749. self.MapWindow.polypen = wx.Pen(colour = 'green', width = 2, style = wx.SHORT_DASH)
  750. # change the cursor
  751. self.MapWindow.SetCursor(self.cursors["pencil"])
  752. # initiating output (and write a message)
  753. # e.g., in Layer Manager switch to output console
  754. # TODO: this should be something like: write important message or write tip
  755. # TODO: mixed 'switching' and message? no, measuring handles 'swithing' on its own
  756. self._giface.WriteWarning(_('Click and drag with left mouse button '
  757. 'to measure.%s'
  758. 'Double click with left button to clear.') % \
  759. (os.linesep))
  760. if self.Map.projinfo['proj'] != 'xy':
  761. units = self.Map.projinfo['units']
  762. self._giface.WriteCmdLog(_('Measuring distance') + ' ('
  763. + units + '):')
  764. else:
  765. self._giface.WriteCmdLog(_('Measuring distance:'))
  766. if self.Map.projinfo['proj'] == 'll':
  767. try:
  768. import grass.lib.gis as gislib
  769. global haveCtypes
  770. haveCtypes = True
  771. gislib.G_begin_distance_calculations()
  772. except ImportError, e:
  773. self._giface.WriteWarning(_('Geodesic distance is not yet '
  774. 'supported by this tool.\n'
  775. 'Reason: %s' % e))
  776. def MeasureDist(self, beginpt, endpt):
  777. """!Calculate map distance from screen distance
  778. and print to output window
  779. """
  780. dist, (north, east) = self.MapWindow.Distance(beginpt, endpt)
  781. dist = round(dist, 3)
  782. d, dunits = self.FormatDist(dist)
  783. self.totaldist += dist
  784. td, tdunits = self.FormatDist(self.totaldist)
  785. strdist = str(d)
  786. strtotdist = str(td)
  787. if self.Map.projinfo['proj'] == 'xy' or 'degree' not in self.Map.projinfo['unit']:
  788. angle = int(math.degrees(math.atan2(north,east)) + 0.5)
  789. # uncomment below (or flip order of atan2(y,x) above) to use
  790. # the mathematical theta convention (CCW from +x axis)
  791. #angle = 90 - angle
  792. if angle < 0:
  793. angle = 360 + angle
  794. mstring = '%s = %s %s\n%s = %s %s\n%s = %d %s\n%s' \
  795. % (_('segment'), strdist, dunits,
  796. _('total distance'), strtotdist, tdunits,
  797. _('bearing'), angle, _('degrees (clockwise from grid-north)'),
  798. '-' * 60)
  799. else:
  800. mstring = '%s = %s %s\n%s = %s %s\n%s' \
  801. % (_('segment'), strdist, dunits,
  802. _('total distance'), strtotdist, tdunits,
  803. '-' * 60)
  804. self._giface.WriteLog(mstring, priority=2)
  805. return dist
  806. def OnProfile(self, event):
  807. """!Launch profile tool
  808. """
  809. rasters = []
  810. layers = self._giface.GetLayerList().GetSelectedLayers()
  811. for layer in layers:
  812. if layer.type == 'raster':
  813. rasters.append(layer.maplayer.name)
  814. win = ProfileFrame(parent = self, mapwindow = self.GetMapWindow(), rasterList = rasters)
  815. win.CentreOnParent()
  816. win.Show()
  817. # Open raster select dialog to make sure that a raster (and
  818. # the desired raster) is selected to be profiled
  819. win.OnSelectRaster(None)
  820. def FormatDist(self, dist):
  821. """!Format length numbers and units in a nice way,
  822. as a function of length. From code by Hamish Bowman
  823. Grass Development Team 2006"""
  824. mapunits = self.Map.projinfo['units']
  825. if mapunits == 'metres':
  826. mapunits = 'meters'
  827. outunits = mapunits
  828. dist = float(dist)
  829. divisor = 1.0
  830. # figure out which units to use
  831. if mapunits == 'meters':
  832. if dist > 2500.0:
  833. outunits = 'km'
  834. divisor = 1000.0
  835. else: outunits = 'm'
  836. elif mapunits == 'feet':
  837. # nano-bug: we match any "feet", but US Survey feet is really
  838. # 5279.9894 per statute mile, or 10.6' per 1000 miles. As >1000
  839. # miles the tick markers are rounded to the nearest 10th of a
  840. # mile (528'), the difference in foot flavours is ignored.
  841. if dist > 5280.0:
  842. outunits = 'miles'
  843. divisor = 5280.0
  844. else:
  845. outunits = 'ft'
  846. elif 'degree' in mapunits and \
  847. not haveCtypes:
  848. if dist < 1:
  849. outunits = 'min'
  850. divisor = (1/60.0)
  851. else:
  852. outunits = 'deg'
  853. else:
  854. outunits = 'meters'
  855. # format numbers in a nice way
  856. if (dist/divisor) >= 2500.0:
  857. outdist = round(dist/divisor)
  858. elif (dist/divisor) >= 1000.0:
  859. outdist = round(dist/divisor,1)
  860. elif (dist/divisor) > 0.0:
  861. outdist = round(dist/divisor,int(math.ceil(3-math.log10(dist/divisor))))
  862. else:
  863. outdist = float(dist/divisor)
  864. return (outdist, outunits)
  865. def OnHistogramPyPlot(self, event):
  866. """!Init PyPlot histogram display canvas and tools
  867. """
  868. raster = []
  869. for layer in self._giface.GetLayerList().GetSelectedLayers():
  870. if layer.maplayer.GetType() == 'raster':
  871. raster.append(layer.maplayer.GetName())
  872. win = HistogramPlotFrame(parent = self, rasterList = raster)
  873. win.CentreOnParent()
  874. win.Show()
  875. def OnScatterplot(self, event):
  876. """!Init PyPlot scatterplot display canvas and tools
  877. """
  878. raster = []
  879. for layer in self._giface.GetLayerList().GetSelectedLayers():
  880. if layer.maplayer.GetType() == 'raster':
  881. raster.append(layer.maplayer.GetName())
  882. win = ScatterFrame(parent = self, rasterList = raster)
  883. win.CentreOnParent()
  884. win.Show()
  885. # Open raster select dialog to make sure that at least 2 rasters (and the desired rasters)
  886. # are selected to be plotted
  887. win.OnSelectRaster(None)
  888. def OnHistogram(self, event):
  889. """!Init histogram display canvas and tools
  890. """
  891. win = HistogramFrame(self)
  892. win.CentreOnParent()
  893. win.Show()
  894. win.Refresh()
  895. win.Update()
  896. def AddBarscale(self, cmd = None, showDialog = True):
  897. """!Handler for scale/arrow map decoration menu selection.
  898. """
  899. if cmd:
  900. self.barscale.cmd = cmd
  901. if not showDialog:
  902. self.barscale.Show()
  903. self.MapWindow.UpdateMap()
  904. return
  905. # Decoration overlay control dialog
  906. if self.dialogs['barscale']:
  907. if self.dialogs['barscale'].IsShown():
  908. self.dialogs['barscale'].SetFocus()
  909. else:
  910. self.dialogs['barscale'].Show()
  911. else:
  912. # If location is latlon, only display north arrow (scale won't work)
  913. # proj = self.Map.projinfo['proj']
  914. # if proj == 'll':
  915. # barcmd = 'd.barscale -n'
  916. # else:
  917. # barcmd = 'd.barscale'
  918. # decoration overlay control dialog
  919. self.dialogs['barscale'] = \
  920. DecorationDialog(parent = self, title = _('Scale and North arrow'),
  921. overlayController = self.barscale,
  922. ddstyle = DECOR_DIALOG_BARSCALE,
  923. size = (350, 200),
  924. style = wx.DEFAULT_DIALOG_STYLE | wx.CENTRE)
  925. self.dialogs['barscale'].CentreOnParent()
  926. ### dialog cannot be show as modal - in the result d.barscale is not selectable
  927. ### self.dialogs['barscale'].ShowModal()
  928. self.dialogs['barscale'].Show()
  929. self.MapWindow.mouse['use'] = 'pointer'
  930. def AddLegend(self, cmd = None, showDialog = True):
  931. """!Handler for legend map decoration menu selection.
  932. """
  933. if cmd:
  934. self.legend.cmd = cmd
  935. else:
  936. layers = self._giface.GetLayerList().GetSelectedLayers()
  937. for layer in layers:
  938. if layer.type == 'raster':
  939. self.legend.cmd.append('map=%s' % layer.maplayer.name)
  940. break
  941. if not showDialog:
  942. self.legend.Show()
  943. self.MapWindow.UpdateMap()
  944. return
  945. # Decoration overlay control dialog
  946. if self.dialogs['legend']:
  947. if self.dialogs['legend'].IsShown():
  948. self.dialogs['legend'].SetFocus()
  949. else:
  950. self.dialogs['legend'].Show()
  951. else:
  952. self.dialogs['legend'] = \
  953. DecorationDialog(parent = self, title = _("Legend"),
  954. overlayController = self.legend,
  955. ddstyle = DECOR_DIALOG_LEGEND,
  956. size = (350, 200),
  957. style = wx.DEFAULT_DIALOG_STYLE | wx.CENTRE)
  958. self.dialogs['legend'].CentreOnParent()
  959. ### dialog cannot be show as modal - in the result d.legend is not selectable
  960. ### self.dialogs['legend'].ShowModal()
  961. self.dialogs['legend'].Show()
  962. self.MapWindow.mouse['use'] = 'pointer'
  963. def OnAddText(self, event):
  964. """!Handler for text decoration menu selection.
  965. """
  966. self.SwitchTool(self.toolbars['map'], event)
  967. if self.MapWindow.dragid > -1:
  968. id = self.MapWindow.dragid
  969. self.MapWindow.dragid = -1
  970. else:
  971. # index for overlay layer in render
  972. if len(self.MapWindow.textdict.keys()) > 0:
  973. id = max(self.MapWindow.textdict.keys()) + 1
  974. else:
  975. id = 101
  976. self.dialogs['text'] = TextLayerDialog(parent = self, ovlId = id,
  977. title = _('Add text layer'),
  978. size = (400, 200))
  979. self.dialogs['text'].CenterOnParent()
  980. # If OK button pressed in decoration control dialog
  981. if self.dialogs['text'].ShowModal() == wx.ID_OK:
  982. text = self.dialogs['text'].GetValues()['text']
  983. active = self.dialogs['text'].GetValues()['active']
  984. # delete object if it has no text or is not active
  985. if text == '' or active == False:
  986. try:
  987. self.MapWindow2D.pdc.ClearId(id)
  988. self.MapWindow2D.pdc.RemoveId(id)
  989. del self.MapWindow.textdict[id]
  990. if self.IsPaneShown('3d'):
  991. self.MapWindow3D.UpdateOverlays()
  992. self.MapWindow.UpdateMap()
  993. else:
  994. self.MapWindow2D.UpdateMap(render = False, renderVector = False)
  995. except:
  996. pass
  997. return
  998. self.MapWindow.textdict[id] = self.dialogs['text'].GetValues()
  999. if self.IsPaneShown('3d'):
  1000. self.MapWindow3D.UpdateOverlays()
  1001. self.MapWindow3D.UpdateMap()
  1002. else:
  1003. self.MapWindow2D.pdc.ClearId(id)
  1004. self.MapWindow2D.pdc.SetId(id)
  1005. self.MapWindow2D.UpdateMap(render = False, renderVector = False)
  1006. self.MapWindow.mouse['use'] = 'pointer'
  1007. def OnAddArrow(self, event):
  1008. """!Handler for north arrow menu selection.
  1009. Opens Appearance page of nviz notebook.
  1010. """
  1011. self._layerManager.nviz.SetPage('decoration')
  1012. self.MapWindow3D.SetDrawArrow((70, 70))
  1013. def GetOptData(self, dcmd, type, params, propwin):
  1014. """!Callback method for decoration overlay command generated by
  1015. dialog created in menuform.py
  1016. """
  1017. # Reset comand and rendering options in render.Map. Always render decoration.
  1018. # Showing/hiding handled by PseudoDC
  1019. self.Map.ChangeOverlay(ovltype = type, type = 'overlay', name = '', command = dcmd,
  1020. active = True, render = False)
  1021. self.params[type] = params
  1022. self.propwin[type] = propwin
  1023. def OnZoomToMap(self, event):
  1024. """!Set display extents to match selected raster (including
  1025. NULLs) or vector map.
  1026. """
  1027. layers = None
  1028. if self.IsStandalone():
  1029. layers = self.MapWindow.GetMap().GetListOfLayers(active = False)
  1030. self.MapWindow.ZoomToMap(layers = layers)
  1031. def OnZoomToRaster(self, event):
  1032. """!Set display extents to match selected raster map (ignore NULLs)
  1033. """
  1034. self.MapWindow.ZoomToMap(ignoreNulls = True)
  1035. def OnZoomToSaved(self, event):
  1036. """!Set display geometry to match extents in
  1037. saved region file
  1038. """
  1039. self.MapWindow.ZoomToSaved()
  1040. def OnDisplayToWind(self, event):
  1041. """!Set computational region (WIND file) to match display
  1042. extents
  1043. """
  1044. self.MapWindow.DisplayToWind()
  1045. def OnSaveDisplayRegion(self, event):
  1046. """!Save display extents to named region file.
  1047. """
  1048. self.MapWindow.SaveRegion(display = True)
  1049. def OnSaveWindRegion(self, event):
  1050. """!Save computational region to named region file.
  1051. """
  1052. self.MapWindow.SaveRegion(display = False)
  1053. def OnZoomMenu(self, event):
  1054. """!Popup Zoom menu
  1055. """
  1056. zoommenu = wx.Menu()
  1057. for label, handler in ((_('Zoom to computational region'), self.OnZoomToWind),
  1058. (_('Zoom to default region'), self.OnZoomToDefault),
  1059. (_('Zoom to saved region'), self.OnZoomToSaved),
  1060. (_('Set computational region from display extent'), self.OnDisplayToWind),
  1061. (_('Save display geometry to named region'), self.OnSaveDisplayRegion),
  1062. (_('Save computational region to named region'), self.OnSaveWindRegion)):
  1063. mid = wx.MenuItem(zoommenu, wx.ID_ANY, label)
  1064. zoommenu.AppendItem(mid)
  1065. self.Bind(wx.EVT_MENU, handler, mid)
  1066. # Popup the menu. If an item is selected then its handler will
  1067. # be called before PopupMenu returns.
  1068. self.PopupMenu(zoommenu)
  1069. zoommenu.Destroy()
  1070. def SetProperties(self, render = False, mode = 0, showCompExtent = False,
  1071. constrainRes = False, projection = False, alignExtent = True):
  1072. """!Set properies of map display window"""
  1073. self.SetProperty('render', render)
  1074. self.statusbarManager.SetMode(mode)
  1075. self.StatusbarUpdate()
  1076. self.SetProperty('region', showCompExtent)
  1077. self.SetProperty('alignExtent', alignExtent)
  1078. self.SetProperty('resolution', constrainRes)
  1079. self.SetProperty('projection', projection)
  1080. def IsStandalone(self):
  1081. """!Check if Map display is standalone"""
  1082. if self._layerManager:
  1083. return False
  1084. return True
  1085. def GetLayerManager(self):
  1086. """!Get reference to Layer Manager
  1087. @return window reference
  1088. @return None (if standalone)
  1089. """
  1090. return self._layerManager
  1091. def GetMapToolbar(self):
  1092. """!Returns toolbar with zooming tools"""
  1093. return self.toolbars['map']
  1094. def OnVNet(self, event):
  1095. """!Dialog for v.net* modules
  1096. """
  1097. if self.dialogs['vnet']:
  1098. self.dialogs['vnet'].Raise()
  1099. return
  1100. from vnet.dialogs import VNETDialog
  1101. self.dialogs['vnet'] = VNETDialog(parent=self, giface=self._giface)
  1102. self.dialogs['vnet'].CenterOnScreen()
  1103. self.dialogs['vnet'].Show()
  1104. def SwitchTool(self, toolbar, event):
  1105. """!Calls UpdateTools to manage connected toolbars"""
  1106. self.UpdateTools(event)
  1107. SingleMapFrame.SwitchTool(self, toolbar, event)
  1108. def UpdateTools(self, event):
  1109. """!Method deals with relations of toolbars and other
  1110. elements"""
  1111. # untoggles button in other toolbars
  1112. for toolbar in self.toolbars.itervalues():
  1113. if hasattr(event, 'GetEventObject') == True:
  1114. if event.GetEventObject() == toolbar:
  1115. continue
  1116. toolbar.ToggleTool(toolbar.action['id'], False)
  1117. toolbar.action['id'] = -1
  1118. toolbar.OnTool(None)
  1119. # mouse settings
  1120. self.MapWindow.mouse['box'] = 'point'
  1121. self.MapWindow.mouse['use'] = 'pointer'
  1122. # untoggles button in add legend dialog
  1123. # FIXME: remove this mess
  1124. if self.dialogs['legend']:
  1125. if hasattr(event ,'GetEventObject') and hasattr(event.GetEventObject() ,'GetId'):
  1126. if event.GetEventObject().GetId() == \
  1127. self.dialogs['legend'].resizeBtn.GetId():
  1128. return
  1129. self.dialogs['legend'].resizeBtn.SetValue(0)
  1130. def ResetPointer(self):
  1131. """Sets pointer mode.
  1132. Sets pointer and toggles it (e.g. after unregistration of mouse
  1133. handler).
  1134. Somehow related to UpdateTools.
  1135. """
  1136. # sets pointer mode
  1137. toolbar = self.toolbars['map']
  1138. toolbar.action['id'] = vars(toolbar)["pointer"]
  1139. toolbar.OnTool(None)
  1140. self.OnPointer(event=None)