nviz_mapdisp.py 38 KB

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