nviz_mapdisp.py 39 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066
  1. """!
  2. @package nviz_mapdisp.py
  3. @brief wxGUI 3D view mode (map canvas)
  4. This module implements 3D visualization mode for map display.
  5. List of classes:
  6. - NvizThread
  7. - GLWindow
  8. (C) 2008-2010 by the GRASS Development Team
  9. This program is free software under the GNU General Public
  10. License (>=v2). Read the file COPYING that comes with GRASS
  11. for details.
  12. @author Martin Landa <landa.martin gmail.com> (Google SoC 2008/2010)
  13. """
  14. import os
  15. import sys
  16. import time
  17. import copy
  18. from threading import Thread
  19. import wx
  20. import wx.lib.scrolledpanel as scrolled
  21. from wx.lib.newevent import NewEvent
  22. from wx import glcanvas
  23. import gcmd
  24. import globalvar
  25. from debug import Debug as Debug
  26. from mapdisp_window import MapWindow
  27. from goutput import wxCmdOutput
  28. from preferences import globalSettings as UserSettings
  29. from workspace import Nviz as NvizDefault
  30. import wxnviz
  31. # sys.path.append(os.path.join(globalvar.ETCWXDIR, "nviz"))
  32. # import grass7_wxnviz as wxnviz
  33. wxUpdateProperties, EVT_UPDATE_PROP = NewEvent()
  34. wxUpdateView, EVT_UPDATE_VIEW = NewEvent()
  35. class NvizThread(Thread):
  36. def __init__(self, log, progressbar, window):
  37. Thread.__init__(self)
  38. self.log = log
  39. self.progressbar = progressbar
  40. self.window = window
  41. self.nvizClass = None
  42. self.setDaemon(True)
  43. def run(self):
  44. self.nvizClass = wxnviz.Nviz(self.log)
  45. class GLWindow(MapWindow, glcanvas.GLCanvas):
  46. """!OpenGL canvas for Map Display Window"""
  47. def __init__(self, parent, id,
  48. pos = wx.DefaultPosition,
  49. size = wx.DefaultSize,
  50. style = wx.NO_FULL_REPAINT_ON_RESIZE,
  51. Map = None, tree = None, lmgr = None):
  52. self.parent = parent # MapFrame
  53. self.Map = Map
  54. self.tree = tree
  55. self._layermanager = lmgr
  56. glcanvas.GLCanvas.__init__(self, parent, id)
  57. MapWindow.__init__(self, parent, id, pos, size, style,
  58. Map, tree, self._layermanager)
  59. self.Hide()
  60. self.init = False
  61. self.initView = False
  62. # render mode
  63. self.render = { 'quick' : False,
  64. # do not render vector lines in quick mode
  65. 'vlines' : False,
  66. 'vpoints' : False }
  67. # list of loaded map layers (layer tree items)
  68. self.layers = []
  69. #
  70. # use display region instead of computational
  71. #
  72. os.environ['GRASS_REGION'] = self.Map.SetRegion()
  73. #
  74. # create nviz instance
  75. #
  76. if self._layermanager:
  77. logerr = self._layermanager.goutput.cmd_stderr
  78. logmsg = self._layermanager.goutput.cmd_output
  79. else:
  80. logerr = logmsg = None
  81. self.nvizThread = NvizThread(logerr,
  82. self.parent.statusbarWin['progress'],
  83. logmsg)
  84. self.nvizThread.start()
  85. time.sleep(.1)
  86. self.nvizClass = self.nvizThread.nvizClass
  87. # GRASS_REGION needed only for initialization
  88. del os.environ['GRASS_REGION']
  89. self.img = wx.Image(self.Map.mapfile, wx.BITMAP_TYPE_ANY)
  90. #
  91. # default values
  92. #
  93. self.view = copy.deepcopy(UserSettings.Get(group = 'nviz', key = 'view')) # copy
  94. self.iview = UserSettings.Get(group = 'nviz', key = 'view', internal = True)
  95. self.nvizDefault = NvizDefault()
  96. self.size = None
  97. self.Bind(wx.EVT_ERASE_BACKGROUND, self.OnEraseBackground)
  98. self.Bind(wx.EVT_SIZE, self.OnSize)
  99. self.Bind(wx.EVT_PAINT, self.OnPaint)
  100. self.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown)
  101. self.Bind(wx.EVT_LEFT_UP, self.OnLeftUp)
  102. self.Bind(wx.EVT_MOTION, self.OnMouseAction)
  103. self.Bind(wx.EVT_MOUSE_EVENTS, self.OnMouseAction)
  104. self.Bind(EVT_UPDATE_PROP, self.UpdateMapObjProperties)
  105. self.Bind(EVT_UPDATE_VIEW, self.UpdateView)
  106. self.Bind(wx.EVT_CLOSE, self.OnClose)
  107. def OnClose(self, event):
  108. # cleanup when window actually closes (on quit) and not just is hidden
  109. self.Reset()
  110. def OnEraseBackground(self, event):
  111. pass # do nothing, to avoid flashing on MSW
  112. def OnSize(self, event):
  113. self.size = self.parent.GetClientSize()
  114. if self.GetContext():
  115. Debug.msg(3, "GLCanvas.OnSize(): w = %d, h = %d" % \
  116. (self.size.width, self.size.height))
  117. self.SetCurrent()
  118. self.nvizClass.ResizeWindow(self.size.width,
  119. self.size.height)
  120. event.Skip()
  121. def OnPaint(self, event):
  122. Debug.msg(3, "GLCanvas.OnPaint()")
  123. dc = wx.PaintDC(self)
  124. self.SetCurrent()
  125. if not self.initView:
  126. self.nvizClass.InitView()
  127. self.initView = True
  128. self.LoadDataLayers()
  129. self.UnloadDataLayers()
  130. if not self.init:
  131. self.ResetView()
  132. if hasattr(self.parent, "nvizToolWin"):
  133. self.parent.nvizToolWin.UpdatePage('view')
  134. layer = self.GetSelectedLayer()
  135. if layer:
  136. if layer.type == 'raster':
  137. self.parent.nvizToolWin.UpdatePage('surface')
  138. elif layer.type == 'vector':
  139. self.parent.nvizToolWin.UpdatePage('vector')
  140. self.parent.nvizToolWin.UpdateSettings()
  141. # update widgets
  142. win = self.parent.nvizToolWin.FindWindowById( \
  143. self.parent.nvizToolWin.win['vector']['lines']['surface'])
  144. win.SetItems(self.GetLayerNames('raster'))
  145. self.init = True
  146. self.UpdateMap()
  147. def OnMouseAction(self, event):
  148. # change position
  149. if event.Dragging() and event.LeftIsDown():
  150. ### self.lastX = self.lastY = self.x = self.y
  151. ### self.x, self.y = event.GetPosition()
  152. ### self.Refresh(False)
  153. pass
  154. # change perspective with mouse wheel
  155. wheel = event.GetWheelRotation()
  156. if wheel != 0:
  157. current = event.GetPositionTuple()[:]
  158. Debug.msg (5, "GLWindow.OnMouseMotion(): wheel = %d" % wheel)
  159. prev_value = self.view['persp']['value']
  160. if wheel > 0:
  161. value = -1 * self.view['persp']['step']
  162. else:
  163. value = self.view['persp']['step']
  164. self.view['persp']['value'] += value
  165. if self.view['persp']['value'] < 1:
  166. self.view['persp']['value'] = 1
  167. elif self.view['persp']['value'] > 100:
  168. self.view['persp']['value'] = 100
  169. if prev_value != self.view['persp']['value']:
  170. if hasattr(self.parent, "nvizToolWin"):
  171. self.parent.nvizToolWin.UpdateSettings()
  172. self.nvizClass.SetView(self.view['pos']['x'], self.view['pos']['y'],
  173. self.iview['height']['value'],
  174. self.view['persp']['value'],
  175. self.view['twist']['value'])
  176. # redraw map
  177. self.OnPaint(None)
  178. # update statusbar
  179. ### self.parent.StatusbarUpdate()
  180. def OnLeftDown(self, event):
  181. self.CaptureMouse()
  182. def OnLeftUp(self, event):
  183. self.ReleaseMouse()
  184. def UpdateView(self, event):
  185. """!Change view settings"""
  186. self.nvizClass.SetView(self.view['pos']['x'], self.view['pos']['y'],
  187. self.iview['height']['value'],
  188. self.view['persp']['value'],
  189. self.view['twist']['value'])
  190. if event and event.zExag:
  191. self.nvizClass.SetZExag(self.view['z-exag']['value'])
  192. if event: event.Skip()
  193. def UpdateMap(self, render = True):
  194. """!Updates the canvas anytime there is a change to the
  195. underlaying images or to the geometry of the canvas.
  196. @param render re-render map composition
  197. """
  198. start = time.clock()
  199. self.resize = False
  200. if self.render['quick'] is False:
  201. self.parent.statusbarWin['progress'].Show()
  202. self.parent.statusbarWin['progress'].SetRange(2)
  203. self.parent.statusbarWin['progress'].SetValue(0)
  204. if self.render['quick'] is False:
  205. self.parent.statusbarWin['progress'].SetValue(1)
  206. self.nvizClass.Draw(False, -1)
  207. elif self.render['quick'] is True:
  208. # quick
  209. mode = wxnviz.DRAW_QUICK_SURFACE | wxnviz.DRAW_QUICK_VOLUME
  210. if self.render['vlines']:
  211. mode |= wxnviz.DRAW_QUICK_VLINES
  212. if self.render['vpoints']:
  213. mode |= wxnviz.DRAW_QUICK_VPOINTS
  214. self.nvizClass.Draw(True, mode)
  215. else: # None -> reuse last rendered image
  216. pass # TODO
  217. self.SwapBuffers()
  218. stop = time.clock()
  219. if self.render['quick'] is False:
  220. self.parent.statusbarWin['progress'].SetValue(2)
  221. # hide process bar
  222. self.parent.statusbarWin['progress'].Hide()
  223. #
  224. # update statusbar
  225. #
  226. # self.parent.StatusbarUpdate()
  227. Debug.msg(3, "GLWindow.UpdateMap(): quick = %d, -> time = %g" % \
  228. (self.render['quick'], (stop-start)))
  229. def EraseMap(self):
  230. """!Erase the canvas
  231. """
  232. self.nvizClass.EraseMap()
  233. self.SwapBuffers()
  234. def IsLoaded(self, item):
  235. """!Check if layer (item) is already loaded
  236. @param item layer item
  237. """
  238. layer = self.tree.GetPyData(item)[0]['maplayer']
  239. data = self.tree.GetPyData(item)[0]['nviz']
  240. if not data:
  241. return 0
  242. if layer.type == 'raster':
  243. if not data['surface'].has_key('object'):
  244. return 0
  245. elif layer.type == 'vector':
  246. if not data['vlines'].has_key('object') and \
  247. not data['points'].has_key('object'):
  248. return 0
  249. return 1
  250. def _GetDataLayers(self, item, litems):
  251. """!Return get list of enabled map layers"""
  252. # load raster & vector maps
  253. while item and item.IsOk():
  254. type = self.tree.GetPyData(item)[0]['type']
  255. if type == 'group':
  256. subItem = self.tree.GetFirstChild(item)[0]
  257. self._GetDataLayers(subItem, litems)
  258. item = self.tree.GetNextSibling(item)
  259. if not item.IsChecked() or \
  260. type not in ('raster', 'vector', '3d-raster'):
  261. item = self.tree.GetNextSibling(item)
  262. continue
  263. litems.append(item)
  264. item = self.tree.GetNextSibling(item)
  265. def LoadDataLayers(self):
  266. """!Load raster/vector from current layer tree
  267. @todo volumes
  268. """
  269. if not self.tree:
  270. return
  271. listOfItems = []
  272. item = self.tree.GetFirstChild(self.tree.root)[0]
  273. self._GetDataLayers(item, listOfItems)
  274. start = time.time()
  275. while(len(listOfItems) > 0):
  276. item = listOfItems.pop()
  277. type = self.tree.GetPyData(item)[0]['type']
  278. if item not in self.layers:
  279. try:
  280. if type == 'raster':
  281. self.LoadRaster(item)
  282. elif type == '3d-raster':
  283. self.LoadRaster3d(item)
  284. except gcmd.NvizError, e:
  285. print >> sys.stderr, "Nviz:" + e.message
  286. try:
  287. if type == 'vector':
  288. data = self.tree.GetPyData(item)[0]['nviz']
  289. vecType = []
  290. if data and data.has_key('vector'):
  291. for v in ('lines', 'points'):
  292. if data['vector'][v]:
  293. vecType.append(v)
  294. self.LoadVector(item, vecType)
  295. except gcmd.NvizError, e:
  296. print >> sys.stderr, "Nviz:" + e.message
  297. self.init = False
  298. stop = time.time()
  299. Debug.msg(3, "GLWindow.LoadDataLayers(): time = %f" % (stop-start))
  300. def UnloadDataLayers(self):
  301. """!Unload any layers that have been deleted from layer tree"""
  302. if not self.tree:
  303. return
  304. listOfItems = []
  305. item = self.tree.GetFirstChild(self.tree.root)[0]
  306. self._GetDataLayers(item, listOfItems)
  307. start = time.time()
  308. for layer in self.layers:
  309. if layer not in listOfItems:
  310. ltype = self.tree.GetPyData(layer)[0]['type']
  311. try:
  312. if ltype == 'raster':
  313. self.UnloadRaster(layer)
  314. elif ltype == '3d-raster':
  315. self.UnloadRaster3d(layer)
  316. elif ltype == 'vector':
  317. data = self.tree.GetPyData(layer)[0]['nviz']
  318. vecType = []
  319. if data and data.has_key('vector'):
  320. for v in ('lines', 'points'):
  321. if data['vector'][v]:
  322. vecType.append(v)
  323. self.UnloadVector(layer, vecType)
  324. self.UpdateView(None)
  325. except gcmd.NvizError, e:
  326. print >> sys.stderr, "Nviz:" + e.message
  327. self.parent.nvizToolWin.UpdateSettings()
  328. stop = time.time()
  329. Debug.msg(3, "GLWindow.UnloadDataLayers(): time = %f" % (stop-start))
  330. def SetMapObjProperties(self, item, id, nvizType):
  331. """!Set map object properties
  332. Properties must be afterwards updated by
  333. UpdateMapObjProperties().
  334. @param item layer item
  335. @param id nviz layer id (or -1)
  336. @param nvizType nviz data type (surface, points, vector)
  337. """
  338. type = self.tree.GetPyData(item)[0]['maplayer'].type
  339. # reference to original layer properties (can be None)
  340. data = self.tree.GetPyData(item)[0]['nviz']
  341. if data is None:
  342. # init data structure
  343. self.tree.GetPyData(item)[0]['nviz'] = {}
  344. data = self.tree.GetPyData(item)[0]['nviz']
  345. if type == 'raster':
  346. # reset to default properties
  347. data[nvizType] = self.nvizDefault.SetSurfaceDefaultProp()
  348. elif type == 'vector':
  349. # reset to default properties (lines/points)
  350. data['vector'] = self.nvizDefault.SetVectorDefaultProp()
  351. elif type == '3d-raster':
  352. # reset to default properties
  353. data[nvizType] = self.nvizDefault.SetVolumeDefaultProp()
  354. else:
  355. # complete data (use default values)
  356. if type == 'raster':
  357. data['surface'] = self.nvizDefault.SetSurfaceDefaultProp()
  358. if type == 'vector':
  359. if not data['vector']['lines']:
  360. self.nvizDefault.SetVectorLinesDefaultProp(data['vector']['lines'])
  361. if not data['vector']['points']:
  362. self.nvizDefault.SetVectorPointsDefaultProp(data['vector']['points'])
  363. # set updates
  364. for sec in data.keys():
  365. for sec1 in data[sec].keys():
  366. for sec2 in data[sec][sec1].keys():
  367. if sec2 != 'all':
  368. data[sec][sec1][sec2]['update'] = None
  369. event = wxUpdateProperties(data = data)
  370. wx.PostEvent(self, event)
  371. # set id
  372. if id > 0:
  373. if type in ('raster', '3d-raster'):
  374. data[nvizType]['object'] = { 'id' : id,
  375. 'init' : False }
  376. elif type == 'vector':
  377. data['vector'][nvizType]['object'] = { 'id' : id,
  378. 'init' : False }
  379. return data
  380. def LoadRaster(self, item):
  381. """!Load 2d raster map and set surface attributes
  382. @param layer item
  383. """
  384. return self._loadRaster(item)
  385. def LoadRaster3d(self, item):
  386. """!Load 3d raster map and set surface attributes
  387. @param layer item
  388. """
  389. return self._loadRaster(item)
  390. def _loadRaster(self, item):
  391. """!Load 2d/3d raster map and set its attributes
  392. @param layer item
  393. """
  394. layer = self.tree.GetPyData(item)[0]['maplayer']
  395. if layer.type not in ('raster', '3d-raster'):
  396. return
  397. if layer.type == 'raster':
  398. id = self.nvizClass.LoadSurface(str(layer.name), None, None)
  399. nvizType = 'surface'
  400. errorMsg = _("Loading raster map")
  401. elif layer.type == '3d-raster':
  402. id = self.nvizClass.LoadVolume(str(layer.name), None, None)
  403. nvizType = 'volume'
  404. errorMsg = _("Loading 3d raster map")
  405. else:
  406. id = -1
  407. if id < 0:
  408. if layer.type in ('raster', '3d-raster'):
  409. print >> sys.stderr, "Nviz:" + "%s <%s> %s" % (errorMsg, layer.name, _("failed"))
  410. else:
  411. print >> sys.stderr, "Nviz:" + _("Unsupported layer type '%s'") % layer.type
  412. self.layers.append(item)
  413. # set default/workspace layer properties
  414. data = self.SetMapObjProperties(item, id, nvizType)
  415. # update properties
  416. event = wxUpdateProperties(data = data)
  417. wx.PostEvent(self, event)
  418. # update tools window
  419. if hasattr(self.parent, "nvizToolWin") and \
  420. item == self.GetSelectedLayer(type = 'item'):
  421. toolWin = self.parent.nvizToolWin
  422. if layer.type == 'raster':
  423. win = toolWin.FindWindowById( \
  424. toolWin.win['vector']['lines']['surface'])
  425. win.SetItems(self.GetLayerNames(layer.type))
  426. #toolWin.UpdatePage(nvizType)
  427. #toolWin.SetPage(nvizType)
  428. return id
  429. def UnloadRaster(self, item):
  430. """!Unload 2d raster map
  431. @param layer item
  432. """
  433. return self._unloadRaster(item)
  434. def UnloadRaster3d(self, item):
  435. """!Unload 3d raster map
  436. @param layer item
  437. """
  438. return self._unloadRaster(item)
  439. def _unloadRaster(self, item):
  440. """!Unload 2d/3d raster map
  441. @param item layer item
  442. """
  443. layer = self.tree.GetPyData(item)[0]['maplayer']
  444. if layer.type not in ('raster', '3d-raster'):
  445. return
  446. data = self.tree.GetPyData(item)[0]['nviz']
  447. if layer.type == 'raster':
  448. nvizType = 'surface'
  449. unloadFn = self.nvizClass.UnloadSurface
  450. errorMsg = _("Unable to unload raster map")
  451. successMsg = _("Raster map")
  452. else:
  453. nvizType = 'volume'
  454. unloadFn = self.nvizClass.UnloadVolume
  455. errorMsg = _("Unable to unload 3d raster map")
  456. successMsg = _("3d raster map")
  457. id = data[nvizType]['object']['id']
  458. if unloadFn(id) == 0:
  459. print >> sys.stderr, "Nviz:" + "%s <%s>" % (errorMsg, layer.name)
  460. else:
  461. print "Nviz:" + "%s <%s> %s" % (successMsg, layer.name, _("unloaded successfully"))
  462. data[nvizType].pop('object')
  463. self.layers.remove(item)
  464. # update tools window
  465. if hasattr(self.parent, "nvizToolWin") and \
  466. layer.type == 'raster':
  467. toolWin = self.parent.nvizToolWin
  468. win = toolWin.FindWindowById( \
  469. toolWin.win['vector']['lines']['surface'])
  470. win.SetItems(self.GetLayerNames(layer.type))
  471. # remove surface page
  472. if toolWin.notebook.GetSelection() == toolWin.page[nvizType]['id']:
  473. toolWin.notebook.RemovePage(toolWin.page[nvizType]['id'])
  474. toolWin.page[nvizType]['id'] = -1
  475. toolWin.page['settings']['id'] = 1
  476. def LoadVector(self, item, vecType = None):
  477. """!Load 2D or 3D vector map overlay
  478. @param item layer item
  479. @param vecType vector type (lines / points)
  480. """
  481. layer = self.tree.GetPyData(item)[0]['maplayer']
  482. if layer.type != 'vector':
  483. return
  484. if vecType is None:
  485. # load data type by default
  486. vecType = []
  487. for v in ('lines', 'points'):
  488. if UserSettings.Get(group = 'nviz', key = 'vector',
  489. subkey = [v, 'show']):
  490. vecType.append(v)
  491. # set default properties
  492. self.SetMapObjProperties(item, -1, 'lines')
  493. self.SetMapObjProperties(item, -1, 'points')
  494. id = -1
  495. for type in vecType:
  496. if type == 'lines':
  497. id = self.nvizClass.LoadVector(str(layer.name), False)
  498. else:
  499. id = self.nvizClass.LoadVector(str(layer.name), True)
  500. if id < 0:
  501. print >> sys.stderr, "Nviz:" + _("Loading vector map <%(name)s> (%(type)s) failed") % \
  502. { 'name' : layer.name, 'type' : type }
  503. continue
  504. # update layer properties
  505. self.SetMapObjProperties(item, id, type)
  506. self.layers.append(item)
  507. # update properties
  508. data = self.tree.GetPyData(item)[0]['nviz']
  509. event = wxUpdateProperties(data = data)
  510. wx.PostEvent(self, event)
  511. # update tools window
  512. if hasattr(self.parent, "nvizToolWin") and \
  513. item == self.GetSelectedLayer(type = 'item'):
  514. toolWin = self.parent.nvizToolWin
  515. toolWin.UpdatePage('vector')
  516. toolWin.SetPage('vector')
  517. return id
  518. def UnloadVector(self, item, vecType = None):
  519. """!Unload vector map overlay
  520. @param item layer item
  521. @param vecType vector type (lines, points)
  522. """
  523. layer = self.tree.GetPyData(item)[0]['maplayer']
  524. data = self.tree.GetPyData(item)[0]['nviz']['vector']
  525. if vecType is None:
  526. vecType = []
  527. for v in ('lines', 'points'):
  528. if UserSettings.Get(group = 'nviz', key = 'vector',
  529. subkey = [v, 'show']):
  530. vecType.append(v)
  531. for vtype in vecType:
  532. if not data[vtype].has_key('object'):
  533. continue
  534. id = data[vtype]['object']['id']
  535. if vtype == 'lines':
  536. ret = self.nvizClass.UnloadVector(id, False)
  537. else:
  538. ret = self.nvizClass.UnloadVector(id, True)
  539. if ret == 0:
  540. print >> sys.stderr, "Nviz:" + _("Unable to unload vector map <%(name)s> (%(type)s)") % \
  541. { 'name': layer.name, 'type' : vtype }
  542. else:
  543. print "Nviz:" + _("Vector map <%(name)s> (%(type)s) unloaded successfully") % \
  544. { 'name' : layer.name, 'type' : vtype }
  545. data[vtype].pop('object')
  546. self.layers.remove(id)
  547. # update tools window
  548. if hasattr(self.parent, "nvizToolWin") and \
  549. vecType is None:
  550. toolWin = self.parent.nvizToolWin
  551. # remove surface page
  552. if toolWin.notebook.GetSelection() == toolWin.page['surface']['id']:
  553. toolWin.notebook.RemovePage(toolWin.page['surface']['id'])
  554. toolWin.page['surface']['id'] = -1
  555. toolWin.page['settings']['id'] = 1
  556. def Reset(self):
  557. """!Reset (unload data)"""
  558. for item in self.layers:
  559. type = self.tree.GetPyData(item)[0]['maplayer'].type
  560. if type == 'raster':
  561. self.UnloadRaster(item)
  562. elif type == '3d-raster':
  563. self.UnloadRaster3d(item)
  564. elif type == 'vector':
  565. self.UnloadVector(item)
  566. self.init = False
  567. def OnZoomToMap(self, event):
  568. """!Set display extents to match selected raster or vector
  569. map or volume.
  570. @todo vector, volume
  571. """
  572. layer = self.GetSelectedLayer()
  573. if layer is None:
  574. return
  575. Debug.msg (3, "GLWindow.OnZoomToMap(): layer = %s, type = %s" % \
  576. (layer.name, layer.type))
  577. self.nvizClass.SetViewportDefault()
  578. def ResetView(self):
  579. """!Reset to default view"""
  580. self.view['z-exag']['value'], \
  581. self.iview['height']['value'], \
  582. self.iview['height']['min'], \
  583. self.iview['height']['max'] = self.nvizClass.SetViewDefault()
  584. self.view['pos']['x'] = UserSettings.Get(group = 'nviz', key = 'view',
  585. subkey = ('pos', 'x'))
  586. self.view['pos']['y'] = UserSettings.Get(group = 'nviz', key = 'view',
  587. subkey = ('pos', 'x'))
  588. self.view['persp']['value'] = UserSettings.Get(group = 'nviz', key = 'view',
  589. subkey = ('persp', 'value'))
  590. self.view['twist']['value'] = UserSettings.Get(group = 'nviz', key = 'view',
  591. subkey = ('twist', 'value'))
  592. event = wxUpdateView(zExag = False)
  593. wx.PostEvent(self, event)
  594. def UpdateMapObjProperties(self, event):
  595. """!Generic method to update data layer properties"""
  596. data = event.data
  597. if data.has_key('surface'):
  598. id = data['surface']['object']['id']
  599. self.UpdateSurfaceProperties(id, data['surface'])
  600. # -> initialized
  601. data['surface']['object']['init'] = True
  602. elif data.has_key('volume'):
  603. id = data['volume']['object']['id']
  604. self.UpdateVolumeProperties(id, data['volume'])
  605. # -> initialized
  606. data['volume']['object']['init'] = True
  607. elif data.has_key('vector'):
  608. for type in ('lines', 'points'):
  609. if data['vector'][type].has_key('object'):
  610. id = data['vector'][type]['object']['id']
  611. self.UpdateVectorProperties(id, data['vector'], type)
  612. # -> initialized
  613. data['vector'][type]['object']['init'] = True
  614. def UpdateSurfaceProperties(self, id, data):
  615. """!Update surface map object properties"""
  616. # surface attributes
  617. for attrb in ('topo', 'color', 'mask',
  618. 'transp', 'shine', 'emit'):
  619. if not data['attribute'].has_key(attrb) or \
  620. not data['attribute'][attrb].has_key('update'):
  621. continue
  622. map = data['attribute'][attrb]['map']
  623. value = data['attribute'][attrb]['value']
  624. if map is None: # unset
  625. # only optional attributes
  626. if attrb == 'mask':
  627. # TODO: invert mask
  628. # TODO: broken in NVIZ
  629. self.nvizClass.UnsetSurfaceMask(id)
  630. elif attrb == 'transp':
  631. self.nvizClass.UnsetSurfaceTransp(id)
  632. elif attrb == 'emit':
  633. self.nvizClass.UnsetSurfaceEmit(id)
  634. else:
  635. if type(value) == type('') and \
  636. len(value) <= 0: # ignore empty values (TODO: warning)
  637. continue
  638. if attrb == 'topo':
  639. self.nvizClass.SetSurfaceTopo(id, map, str(value))
  640. elif attrb == 'color':
  641. self.nvizClass.SetSurfaceColor(id, map, str(value))
  642. elif attrb == 'mask':
  643. # TODO: invert mask
  644. # TODO: broken in NVIZ
  645. self.nvizClass.SetSurfaceMask(id, False, str(value))
  646. elif attrb == 'transp':
  647. self.nvizClass.SetSurfaceTransp(id, map, str(value))
  648. elif attrb == 'shine':
  649. self.nvizClass.SetSurfaceShine(id, map, str(value))
  650. elif attrb == 'emit':
  651. self.nvizClass.SetSurfaceEmit(id, map, str(value))
  652. data['attribute'][attrb].pop('update')
  653. # draw res
  654. if data['draw']['resolution'].has_key('update'):
  655. coarse = data['draw']['resolution']['coarse']
  656. fine = data['draw']['resolution']['fine']
  657. if data['draw']['all']:
  658. self.nvizClass.SetSurfaceRes(-1, fine, coarse)
  659. else:
  660. self.nvizClass.SetSurfaceRes(id, fine, coarse)
  661. data['draw']['resolution'].pop('update')
  662. # draw style
  663. if data['draw']['mode'].has_key('update'):
  664. if data['draw']['mode']['value'] < 0: # need to calculate
  665. data['draw']['mode']['value'] = \
  666. self.nvizDefault.GetDrawMode(mode = data['draw']['mode']['desc']['mode'],
  667. style = data['draw']['mode']['desc']['style'],
  668. shade = data['draw']['mode']['desc']['shading'],
  669. string = True)
  670. style = data['draw']['mode']['value']
  671. if data['draw']['all']:
  672. self.nvizClass.SetSurfaceStyle(-1, style)
  673. else:
  674. self.nvizClass.SetSurfaceStyle(id, style)
  675. data['draw']['mode'].pop('update')
  676. # wire color
  677. if data['draw']['wire-color'].has_key('update'):
  678. color = data['draw']['wire-color']['value']
  679. if data['draw']['all']:
  680. self.nvizClass.SetWireColor(-1, str(color))
  681. else:
  682. self.nvizClass.SetWireColor(id, str(color))
  683. data['draw']['wire-color'].pop('update')
  684. # position
  685. if data['position'].has_key('update'):
  686. x = data['position']['x']
  687. y = data['position']['y']
  688. z = data['position']['z']
  689. self.nvizClass.SetSurfacePosition(id, x, y, z)
  690. data['position'].pop('update')
  691. def UpdateVolumeProperties(self, id, data, isosurfId = None):
  692. """!Update volume (isosurface/slice) map object properties"""
  693. if data['draw']['resolution'].has_key('update'):
  694. self.nvizClass.SetIsosurfaceRes(id, data['draw']['resolution']['value'])
  695. data['draw']['resolution'].pop('update')
  696. if data['draw']['shading'].has_key('update'):
  697. if data['draw']['shading']['value'] < 0: # need to calculate
  698. data['draw']['shading']['value'] = \
  699. self.nvizDefault.GetDrawMode(shade = data['draw']['shading'],
  700. string = False)
  701. data['draw']['shading'].pop('update')
  702. #
  703. # isosurface attributes
  704. #
  705. isosurfId = 0
  706. for isosurf in data['isosurface']:
  707. for attrb in ('color', 'mask',
  708. 'transp', 'shine', 'emit'):
  709. if not isosurf.has_key(attrb) or \
  710. not isosurf[attrb].has_key('update'):
  711. continue
  712. map = isosurf[attrb]['map']
  713. value = isosurf[attrb]['value']
  714. if map is None: # unset
  715. # only optional attributes
  716. if attrb == 'mask':
  717. # TODO: invert mask
  718. # TODO: broken in NVIZ
  719. self.nvizClass.UnsetIsosurfaceMask(id, isosurfId)
  720. elif attrb == 'transp':
  721. self.nvizClass.UnsetIsosurfaceTransp(id, isosurfId)
  722. elif attrb == 'emit':
  723. self.nvizClass.UnsetIsosurfaceEmit(id, isosurfId)
  724. else:
  725. if type(value) == type('') and \
  726. len(value) <= 0: # ignore empty values (TODO: warning)
  727. continue
  728. elif attrb == 'color':
  729. self.nvizClass.SetIsosurfaceColor(id, isosurfId, map, str(value))
  730. elif attrb == 'mask':
  731. # TODO: invert mask
  732. # TODO: broken in NVIZ
  733. self.nvizClass.SetIsosurfaceMask(id, isosurfId, False, str(value))
  734. elif attrb == 'transp':
  735. self.nvizClass.SetIsosurfaceTransp(id, isosurfId, map, str(value))
  736. elif attrb == 'shine':
  737. self.nvizClass.SetIsosurfaceShine(id, isosurfId, map, str(value))
  738. elif attrb == 'emit':
  739. self.nvizClass.SetIsosurfaceEmit(id, isosurfId, map, str(value))
  740. isosurf[attrb].pop('update')
  741. isosurfId += 1
  742. def UpdateVectorProperties(self, id, data, type):
  743. """!Update vector layer properties
  744. @param id layer id
  745. @param data properties
  746. @param type lines/points
  747. """
  748. if type == 'points':
  749. self.UpdateVectorPointsProperties(id, data[type])
  750. else:
  751. self.UpdateVectorLinesProperties(id, data[type])
  752. def UpdateVectorLinesProperties(self, id, data):
  753. """!Update vector line map object properties"""
  754. # mode
  755. if data['color'].has_key('update') or \
  756. data['width'].has_key('update') or \
  757. data['mode'].has_key('update'):
  758. width = data['width']['value']
  759. color = data['color']['value']
  760. if data['mode']['type'] == 'flat':
  761. flat = True
  762. if data.has_key('surface'):
  763. data.pop('surface')
  764. else:
  765. flat = False
  766. self.nvizClass.SetVectorLineMode(id, color,
  767. width, flat)
  768. if data['color'].has_key('update'):
  769. data['color'].pop('update')
  770. if data['width'].has_key('update'):
  771. data['width'].pop('update')
  772. if data['mode'].has_key('update'):
  773. data['mode'].pop('update')
  774. # height
  775. if data['height'].has_key('update'):
  776. self.nvizClass.SetVectorLineHeight(id,
  777. data['height']['value'])
  778. data['height'].pop('update')
  779. # surface
  780. if data['mode'].has_key('update'):
  781. sid = self.GetLayerId(type = 'raster', name = data['mode']['surface'])
  782. if sid > -1:
  783. self.nvizClass.SetVectorLineSurface(id, sid)
  784. data['mode'].pop('update')
  785. def UpdateVectorPointsProperties(self, id, data):
  786. """!Update vector point map object properties"""
  787. if data['size'].has_key('update') or \
  788. data['width'].has_key('update') or \
  789. data['marker'].has_key('update') or \
  790. data['color'].has_key('update'):
  791. ret = self.nvizClass.SetVectorPointMode(id, data['color']['value'],
  792. data['width']['value'], float(data['size']['value']),
  793. data['marker']['value'] + 1)
  794. error = None
  795. if ret == -1:
  796. error = _("Vector point layer not found (id = %d)") % id
  797. elif ret == -2:
  798. error = _("Unable to set data layer properties (id = %d)") % id
  799. if error:
  800. raise gcmd.NvizError(parent = self.parent,
  801. message = _("Setting data layer properties failed.\n\n%s") % error)
  802. for prop in ('size', 'width', 'marker', 'color'):
  803. if data[prop].has_key('update'):
  804. data[prop].pop('update')
  805. # height
  806. if data['height'].has_key('update'):
  807. self.nvizClass.SetVectorPointHeight(id,
  808. data['height']['value'])
  809. data['height'].pop('update')
  810. # surface
  811. if data['mode'].has_key('update'):
  812. sid = self.GetLayerId(type = 'raster', name = data['mode']['surface'])
  813. if sid > -1:
  814. self.nvizClass.SetVectorPointSurface(id, sid)
  815. data['mode'].pop('update')
  816. def GetLayerNames(self, type):
  817. """!Return list of map layer names of given type"""
  818. layerName = []
  819. for item in self.layers:
  820. mapLayer = self.tree.GetPyData(item)[0]['maplayer']
  821. if type != mapLayer.GetType():
  822. continue
  823. layerName.append(mapLayer.GetName())
  824. return layerName
  825. def GetLayerData(self, type, name):
  826. """!Return layer item data
  827. @return {} if no layer item found
  828. """
  829. for item in self.layers:
  830. mapLayer = self.tree.GetPyData(item)[0]['maplayer'].GetName()
  831. if mapLayer == name:
  832. return self.tree.GetPyData(item)[0]['nviz']
  833. return {}
  834. def GetLayerId(self, type, name):
  835. """!Get layer object id or -1"""
  836. if len(name) < 1:
  837. return -1
  838. for item in self.layers:
  839. mapLayer = self.tree.GetPyData(item)[0]['maplayer']
  840. if type != mapLayer.GetType() or \
  841. name != mapLayer.GetName():
  842. continue
  843. data = self.tree.GetPyData(item)[0]['nviz']
  844. if type == 'raster':
  845. return data['surface']['object']['id']
  846. elif type == 'vpoint':
  847. return data['vector']['points']['object']['id']
  848. elif type == 'vline':
  849. return data['vector']['lines']['object']['id']
  850. elif type == '3d-raster':
  851. return data['volume']['object']['id']
  852. return -1