frame.py 52 KB

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