frame.py 55 KB

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