frame.py 58 KB

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