nviz_mapdisp.py 40 KB

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