frame.py 55 KB

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