nviz_mapdisp.py 43 KB

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