nviz_mapdisp.py 40 KB

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