frame.py 50 KB

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