nviz_mapdisp.py 43 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150
  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(1, "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['position']['x'], self.view['position']['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['position']['x'], data['position']['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['position']['x'], y = data['position']['y'],
  239. z = data['position']['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.GException, e:
  331. GError(parent = self,
  332. message = e)
  333. try:
  334. if type == 'vector':
  335. data = self.tree.GetPyData(item)[0]['nviz']
  336. vecType = []
  337. if data and data.has_key('vector'):
  338. for v in ('lines', 'points'):
  339. if data['vector'][v]:
  340. vecType.append(v)
  341. self.LoadVector(item, vecType)
  342. except gcmd.GException, e:
  343. GError(parent = self,
  344. message = e)
  345. self.init = False
  346. stop = time.time()
  347. Debug.msg(3, "GLWindow.LoadDataLayers(): time = %f" % (stop-start))
  348. def UnloadDataLayers(self):
  349. """!Unload any layers that have been deleted from layer tree"""
  350. if not self.tree:
  351. return
  352. listOfItems = []
  353. item = self.tree.GetFirstChild(self.tree.root)[0]
  354. self._GetDataLayers(item, listOfItems)
  355. start = time.time()
  356. for layer in self.layers:
  357. if layer not in listOfItems:
  358. ltype = self.tree.GetPyData(layer)[0]['type']
  359. try:
  360. if ltype == 'raster':
  361. self.UnloadRaster(layer)
  362. elif ltype == '3d-raster':
  363. self.UnloadRaster3d(layer)
  364. elif ltype == 'vector':
  365. data = self.tree.GetPyData(layer)[0]['nviz']
  366. vecType = []
  367. if data and data.has_key('vector'):
  368. for v in ('lines', 'points'):
  369. if data['vector'][v]:
  370. vecType.append(v)
  371. self.UnloadVector(layer, vecType)
  372. self.UpdateView(None)
  373. except gcmd.GException, e:
  374. gcmd.GError(parent = self,
  375. message = e)
  376. self.lmgr.nviz.UpdateSettings()
  377. stop = time.time()
  378. Debug.msg(3, "GLWindow.UnloadDataLayers(): time = %f" % (stop-start))
  379. def SetMapObjProperties(self, item, id, nvizType):
  380. """!Set map object properties
  381. Properties must be afterwards updated by
  382. UpdateMapObjProperties().
  383. @param item layer item
  384. @param id nviz layer id (or -1)
  385. @param nvizType nviz data type (surface, points, vector)
  386. """
  387. type = self.tree.GetPyData(item)[0]['maplayer'].type
  388. # reference to original layer properties (can be None)
  389. data = self.tree.GetPyData(item)[0]['nviz']
  390. if data is None:
  391. # init data structure
  392. self.tree.GetPyData(item)[0]['nviz'] = {}
  393. data = self.tree.GetPyData(item)[0]['nviz']
  394. if type == 'raster':
  395. # reset to default properties
  396. data[nvizType] = self.nvizDefault.SetSurfaceDefaultProp()
  397. elif type == 'vector':
  398. # reset to default properties (lines/points)
  399. data['vector'] = self.nvizDefault.SetVectorDefaultProp()
  400. elif type == '3d-raster':
  401. # reset to default properties
  402. data[nvizType] = self.nvizDefault.SetVolumeDefaultProp()
  403. else:
  404. # complete data (use default values)
  405. if type == 'raster':
  406. data['surface'] = self.nvizDefault.SetSurfaceDefaultProp()
  407. if type == 'vector':
  408. if not data['vector']['lines']:
  409. self.nvizDefault.SetVectorLinesDefaultProp(data['vector']['lines'])
  410. if not data['vector']['points']:
  411. self.nvizDefault.SetVectorPointsDefaultProp(data['vector']['points'])
  412. # set updates
  413. for sec in data.keys():
  414. for sec1 in data[sec].keys():
  415. for sec2 in data[sec][sec1].keys():
  416. if sec2 != 'all':
  417. data[sec][sec1][sec2]['update'] = None
  418. event = wxUpdateProperties(data = data)
  419. wx.PostEvent(self, event)
  420. # set id
  421. if id > 0:
  422. if type in ('raster', '3d-raster'):
  423. data[nvizType]['object'] = { 'id' : id,
  424. 'init' : False }
  425. elif type == 'vector':
  426. data['vector'][nvizType]['object'] = { 'id' : id,
  427. 'init' : False }
  428. return data
  429. def LoadRaster(self, item):
  430. """!Load 2d raster map and set surface attributes
  431. @param layer item
  432. """
  433. return self._loadRaster(item)
  434. def LoadRaster3d(self, item):
  435. """!Load 3d raster map and set surface attributes
  436. @param layer item
  437. """
  438. return self._loadRaster(item)
  439. def _loadRaster(self, item):
  440. """!Load 2d/3d raster map and set its attributes
  441. @param layer item
  442. """
  443. layer = self.tree.GetPyData(item)[0]['maplayer']
  444. if layer.type not in ('raster', '3d-raster'):
  445. return
  446. if layer.type == 'raster':
  447. id = self._display.LoadSurface(str(layer.name), None, None)
  448. nvizType = 'surface'
  449. errorMsg = _("Loading raster map")
  450. elif layer.type == '3d-raster':
  451. id = self._display.LoadVolume(str(layer.name), None, None)
  452. nvizType = 'volume'
  453. errorMsg = _("Loading 3d raster map")
  454. else:
  455. id = -1
  456. if id < 0:
  457. if layer.type in ('raster', '3d-raster'):
  458. print >> sys.stderr, "Nviz:" + "%s <%s> %s" % (errorMsg, layer.name, _("failed"))
  459. else:
  460. print >> sys.stderr, "Nviz:" + _("Unsupported layer type '%s'") % layer.type
  461. self.layers.append(item)
  462. # set default/workspace layer properties
  463. data = self.SetMapObjProperties(item, id, nvizType)
  464. # update properties
  465. event = wxUpdateProperties(data = data)
  466. wx.PostEvent(self, event)
  467. # update tools window
  468. if hasattr(self.lmgr, "nviz") and \
  469. item == self.GetSelectedLayer(type = 'item'):
  470. toolWin = self.lmgr.nviz
  471. if layer.type == 'raster':
  472. win = toolWin.FindWindowById( \
  473. toolWin.win['vector']['lines']['surface'])
  474. win.SetItems(self.GetLayerNames(layer.type))
  475. #toolWin.UpdatePage(nvizType)
  476. #toolWin.SetPage(nvizType)
  477. return id
  478. def UnloadRaster(self, item):
  479. """!Unload 2d raster map
  480. @param layer item
  481. """
  482. return self._unloadRaster(item)
  483. def UnloadRaster3d(self, item):
  484. """!Unload 3d raster map
  485. @param layer item
  486. """
  487. return self._unloadRaster(item)
  488. def _unloadRaster(self, item):
  489. """!Unload 2d/3d raster map
  490. @param item layer item
  491. """
  492. layer = self.tree.GetPyData(item)[0]['maplayer']
  493. if layer.type not in ('raster', '3d-raster'):
  494. return
  495. data = self.tree.GetPyData(item)[0]['nviz']
  496. if layer.type == 'raster':
  497. nvizType = 'surface'
  498. unloadFn = self._display.UnloadSurface
  499. errorMsg = _("Unable to unload raster map")
  500. successMsg = _("Raster map")
  501. else:
  502. nvizType = 'volume'
  503. unloadFn = self._display.UnloadVolume
  504. errorMsg = _("Unable to unload 3d raster map")
  505. successMsg = _("3d raster map")
  506. id = data[nvizType]['object']['id']
  507. if unloadFn(id) == 0:
  508. print >> sys.stderr, "Nviz:" + "%s <%s>" % (errorMsg, layer.name)
  509. else:
  510. print "Nviz:" + "%s <%s> %s" % (successMsg, layer.name, _("unloaded successfully"))
  511. data[nvizType].pop('object')
  512. self.layers.remove(item)
  513. # update tools window
  514. if hasattr(self.lmgr, "nviz") and \
  515. layer.type == 'raster':
  516. toolWin = self.lmgr.nviz
  517. win = toolWin.FindWindowById( \
  518. toolWin.win['vector']['lines']['surface'])
  519. win.SetItems(self.GetLayerNames(layer.type))
  520. # remove surface page
  521. if toolWin.GetSelection() == toolWin.page[nvizType]['id']:
  522. toolWin.RemovePage(toolWin.page[nvizType]['id'])
  523. toolWin.page[nvizType]['id'] = -1
  524. toolWin.page['settings']['id'] = 1
  525. def LoadVector(self, item, vecType = None):
  526. """!Load 2D or 3D vector map overlay
  527. @param item layer item
  528. @param vecType vector type (lines / points)
  529. """
  530. layer = self.tree.GetPyData(item)[0]['maplayer']
  531. if layer.type != 'vector':
  532. return
  533. if vecType is None:
  534. # load data type by default
  535. vecType = []
  536. for v in ('lines', 'points'):
  537. if UserSettings.Get(group = 'nviz', key = 'vector',
  538. subkey = [v, 'show']):
  539. vecType.append(v)
  540. # set default properties
  541. self.SetMapObjProperties(item, -1, 'lines')
  542. self.SetMapObjProperties(item, -1, 'points')
  543. id = -1
  544. for type in vecType:
  545. if type == 'lines':
  546. id = self._display.LoadVector(str(layer.name), False)
  547. else:
  548. id = self._display.LoadVector(str(layer.name), True)
  549. if id < 0:
  550. print >> sys.stderr, "Nviz:" + _("Loading vector map <%(name)s> (%(type)s) failed") % \
  551. { 'name' : layer.name, 'type' : type }
  552. continue
  553. # update layer properties
  554. self.SetMapObjProperties(item, id, type)
  555. self.layers.append(item)
  556. # update properties
  557. data = self.tree.GetPyData(item)[0]['nviz']
  558. event = wxUpdateProperties(data = data)
  559. wx.PostEvent(self, event)
  560. # update tools window
  561. if hasattr(self.lmgr, "nviz") and \
  562. item == self.GetSelectedLayer(type = 'item'):
  563. toolWin = self.lmgr.nviz
  564. toolWin.UpdatePage('vector')
  565. toolWin.SetPage('vector')
  566. return id
  567. def UnloadVector(self, item, vecType = None):
  568. """!Unload vector map overlay
  569. @param item layer item
  570. @param vecType vector type (lines, points)
  571. """
  572. layer = self.tree.GetPyData(item)[0]['maplayer']
  573. data = self.tree.GetPyData(item)[0]['nviz']['vector']
  574. if vecType is None:
  575. vecType = []
  576. for v in ('lines', 'points'):
  577. if UserSettings.Get(group = 'nviz', key = 'vector',
  578. subkey = [v, 'show']):
  579. vecType.append(v)
  580. for vtype in vecType:
  581. if not data[vtype].has_key('object'):
  582. continue
  583. id = data[vtype]['object']['id']
  584. if vtype == 'lines':
  585. ret = self._display.UnloadVector(id, False)
  586. else:
  587. ret = self._display.UnloadVector(id, True)
  588. if ret == 0:
  589. print >> sys.stderr, "Nviz:" + _("Unable to unload vector map <%(name)s> (%(type)s)") % \
  590. { 'name': layer.name, 'type' : vtype }
  591. else:
  592. print "Nviz:" + _("Vector map <%(name)s> (%(type)s) unloaded successfully") % \
  593. { 'name' : layer.name, 'type' : vtype }
  594. data[vtype].pop('object')
  595. ### self.layers.remove(id)
  596. # update tools window
  597. if hasattr(self.lmgr, "nviz") and \
  598. vecType is None:
  599. toolWin = self.lmgr.nviz
  600. # remove surface page
  601. if toolWin.GetSelection() == toolWin.page['surface']['id']:
  602. toolWin.RemovePage(toolWin.page['surface']['id'])
  603. toolWin.page['surface']['id'] = -1
  604. toolWin.page['settings']['id'] = 1
  605. def Reset(self):
  606. """!Reset (unload data)"""
  607. for item in self.layers:
  608. type = self.tree.GetPyData(item)[0]['maplayer'].type
  609. if type == 'raster':
  610. self.UnloadRaster(item)
  611. elif type == '3d-raster':
  612. self.UnloadRaster3d(item)
  613. elif type == 'vector':
  614. self.UnloadVector(item)
  615. self.init = False
  616. def OnZoomToMap(self, event):
  617. """!Set display extents to match selected raster or vector
  618. map or volume.
  619. @todo vector, volume
  620. """
  621. layer = self.GetSelectedLayer()
  622. if layer is None:
  623. return
  624. Debug.msg (3, "GLWindow.OnZoomToMap(): layer = %s, type = %s" % \
  625. (layer.name, layer.type))
  626. self._display.SetViewportDefault()
  627. def ResetView(self):
  628. """!Reset to default view"""
  629. self.view['z-exag']['value'], \
  630. self.iview['height']['value'], \
  631. self.iview['height']['min'], \
  632. self.iview['height']['max'] = self._display.SetViewDefault()
  633. self.view['z-exag']['min'] = 0
  634. self.view['z-exag']['max'] = self.view['z-exag']['value'] * 10
  635. self.view['position']['x'] = UserSettings.Get(group = 'nviz', key = 'view',
  636. subkey = ('position', 'x'))
  637. self.view['position']['y'] = UserSettings.Get(group = 'nviz', key = 'view',
  638. subkey = ('position', 'y'))
  639. self.view['persp']['value'] = UserSettings.Get(group = 'nviz', key = 'view',
  640. subkey = ('persp', 'value'))
  641. self.view['twist']['value'] = UserSettings.Get(group = 'nviz', key = 'view',
  642. subkey = ('twist', 'value'))
  643. event = wxUpdateView(zExag = False)
  644. wx.PostEvent(self, event)
  645. def UpdateMapObjProperties(self, event):
  646. """!Generic method to update data layer properties"""
  647. data = event.data
  648. if data.has_key('surface'):
  649. id = data['surface']['object']['id']
  650. self.UpdateSurfaceProperties(id, data['surface'])
  651. # -> initialized
  652. data['surface']['object']['init'] = True
  653. elif data.has_key('volume'):
  654. id = data['volume']['object']['id']
  655. self.UpdateVolumeProperties(id, data['volume'])
  656. # -> initialized
  657. data['volume']['object']['init'] = True
  658. elif data.has_key('vector'):
  659. for type in ('lines', 'points'):
  660. if data['vector'][type].has_key('object'):
  661. id = data['vector'][type]['object']['id']
  662. self.UpdateVectorProperties(id, data['vector'], type)
  663. # -> initialized
  664. data['vector'][type]['object']['init'] = True
  665. def UpdateSurfaceProperties(self, id, data):
  666. """!Update surface map object properties"""
  667. # surface attributes
  668. for attrb in ('topo', 'color', 'mask',
  669. 'transp', 'shine', 'emit'):
  670. if not data['attribute'].has_key(attrb) or \
  671. not data['attribute'][attrb].has_key('update'):
  672. continue
  673. map = data['attribute'][attrb]['map']
  674. value = data['attribute'][attrb]['value']
  675. if map is None: # unset
  676. # only optional attributes
  677. if attrb == 'mask':
  678. # TODO: invert mask
  679. # TODO: broken in NVIZ
  680. self._display.UnsetSurfaceMask(id)
  681. elif attrb == 'transp':
  682. self._display.UnsetSurfaceTransp(id)
  683. elif attrb == 'emit':
  684. self._display.UnsetSurfaceEmit(id)
  685. else:
  686. if type(value) == type('') and \
  687. len(value) <= 0: # ignore empty values (TODO: warning)
  688. continue
  689. if attrb == 'topo':
  690. self._display.SetSurfaceTopo(id, map, str(value))
  691. elif attrb == 'color':
  692. self._display.SetSurfaceColor(id, map, str(value))
  693. elif attrb == 'mask':
  694. # TODO: invert mask
  695. # TODO: broken in NVIZ
  696. self._display.SetSurfaceMask(id, False, str(value))
  697. elif attrb == 'transp':
  698. self._display.SetSurfaceTransp(id, map, str(value))
  699. elif attrb == 'shine':
  700. self._display.SetSurfaceShine(id, map, str(value))
  701. elif attrb == 'emit':
  702. self._display.SetSurfaceEmit(id, map, str(value))
  703. data['attribute'][attrb].pop('update')
  704. # draw res
  705. if data['draw']['resolution'].has_key('update'):
  706. coarse = data['draw']['resolution']['coarse']
  707. fine = data['draw']['resolution']['fine']
  708. if data['draw']['all']:
  709. self._display.SetSurfaceRes(-1, fine, coarse)
  710. else:
  711. self._display.SetSurfaceRes(id, fine, coarse)
  712. data['draw']['resolution'].pop('update')
  713. # draw style
  714. if data['draw']['mode'].has_key('update'):
  715. if data['draw']['mode']['value'] < 0: # need to calculate
  716. data['draw']['mode']['value'] = \
  717. self.nvizDefault.GetDrawMode(mode = data['draw']['mode']['desc']['mode'],
  718. style = data['draw']['mode']['desc']['style'],
  719. shade = data['draw']['mode']['desc']['shading'],
  720. string = True)
  721. style = data['draw']['mode']['value']
  722. if data['draw']['all']:
  723. self._display.SetSurfaceStyle(-1, style)
  724. else:
  725. self._display.SetSurfaceStyle(id, style)
  726. data['draw']['mode'].pop('update')
  727. # wire color
  728. if data['draw']['wire-color'].has_key('update'):
  729. color = data['draw']['wire-color']['value']
  730. if data['draw']['all']:
  731. self._display.SetWireColor(-1, str(color))
  732. else:
  733. self._display.SetWireColor(id, str(color))
  734. data['draw']['wire-color'].pop('update')
  735. # position
  736. if data['position'].has_key('update'):
  737. x = data['position']['x']
  738. y = data['position']['y']
  739. z = data['position']['z']
  740. self._display.SetSurfacePosition(id, x, y, z)
  741. data['position'].pop('update')
  742. def UpdateVolumeProperties(self, id, data, isosurfId = None):
  743. """!Update volume (isosurface/slice) map object properties"""
  744. if data['draw']['resolution'].has_key('update'):
  745. self._display.SetIsosurfaceRes(id, data['draw']['resolution']['value'])
  746. data['draw']['resolution'].pop('update')
  747. if data['draw']['shading'].has_key('update'):
  748. if data['draw']['shading']['value'] < 0: # need to calculate
  749. data['draw']['shading']['value'] = \
  750. self.nvizDefault.GetDrawMode(shade = data['draw']['shading'],
  751. string = False)
  752. data['draw']['shading'].pop('update')
  753. #
  754. # isosurface attributes
  755. #
  756. isosurfId = 0
  757. for isosurf in data['isosurface']:
  758. for attrb in ('color', 'mask',
  759. 'transp', 'shine', 'emit'):
  760. if not isosurf.has_key(attrb) or \
  761. not isosurf[attrb].has_key('update'):
  762. continue
  763. map = isosurf[attrb]['map']
  764. value = isosurf[attrb]['value']
  765. if map is None: # unset
  766. # only optional attributes
  767. if attrb == 'mask':
  768. # TODO: invert mask
  769. # TODO: broken in NVIZ
  770. self._display.UnsetIsosurfaceMask(id, isosurfId)
  771. elif attrb == 'transp':
  772. self._display.UnsetIsosurfaceTransp(id, isosurfId)
  773. elif attrb == 'emit':
  774. self._display.UnsetIsosurfaceEmit(id, isosurfId)
  775. else:
  776. if type(value) == type('') and \
  777. len(value) <= 0: # ignore empty values (TODO: warning)
  778. continue
  779. elif attrb == 'color':
  780. self._display.SetIsosurfaceColor(id, isosurfId, map, str(value))
  781. elif attrb == 'mask':
  782. # TODO: invert mask
  783. # TODO: broken in NVIZ
  784. self._display.SetIsosurfaceMask(id, isosurfId, False, str(value))
  785. elif attrb == 'transp':
  786. self._display.SetIsosurfaceTransp(id, isosurfId, map, str(value))
  787. elif attrb == 'shine':
  788. self._display.SetIsosurfaceShine(id, isosurfId, map, str(value))
  789. elif attrb == 'emit':
  790. self._display.SetIsosurfaceEmit(id, isosurfId, map, str(value))
  791. isosurf[attrb].pop('update')
  792. isosurfId += 1
  793. def UpdateVectorProperties(self, id, data, type):
  794. """!Update vector layer properties
  795. @param id layer id
  796. @param data properties
  797. @param type lines/points
  798. """
  799. if type == 'points':
  800. self.UpdateVectorPointsProperties(id, data[type])
  801. else:
  802. self.UpdateVectorLinesProperties(id, data[type])
  803. def UpdateVectorLinesProperties(self, id, data):
  804. """!Update vector line map object properties"""
  805. # mode
  806. if data['color'].has_key('update') or \
  807. data['width'].has_key('update') or \
  808. data['mode'].has_key('update'):
  809. width = data['width']['value']
  810. color = data['color']['value']
  811. if data['mode']['type'] == 'flat':
  812. flat = True
  813. if data.has_key('surface'):
  814. data.pop('surface')
  815. else:
  816. flat = False
  817. self._display.SetVectorLineMode(id, color,
  818. width, flat)
  819. if data['color'].has_key('update'):
  820. data['color'].pop('update')
  821. if data['width'].has_key('update'):
  822. data['width'].pop('update')
  823. if data['mode'].has_key('update'):
  824. data['mode'].pop('update')
  825. # height
  826. if data['height'].has_key('update'):
  827. self._display.SetVectorLineHeight(id,
  828. data['height']['value'])
  829. data['height'].pop('update')
  830. # surface
  831. if data['mode'].has_key('update'):
  832. sid = self.GetLayerId(type = 'raster', name = data['mode']['surface'])
  833. if sid > -1:
  834. self._display.SetVectorLineSurface(id, sid)
  835. data['mode'].pop('update')
  836. def UpdateVectorPointsProperties(self, id, data):
  837. """!Update vector point map object properties"""
  838. if data['size'].has_key('update') or \
  839. data['width'].has_key('update') or \
  840. data['marker'].has_key('update') or \
  841. data['color'].has_key('update'):
  842. ret = self._display.SetVectorPointMode(id, data['color']['value'],
  843. data['width']['value'], float(data['size']['value']),
  844. data['marker']['value'] + 1)
  845. error = None
  846. if ret == -1:
  847. error = _("Vector point layer not found (id = %d)") % id
  848. elif ret == -2:
  849. error = _("Unable to set data layer properties (id = %d)") % id
  850. if error:
  851. raise gcmd.GException(_("Setting data layer properties failed.\n\n%s") % error)
  852. for prop in ('size', 'width', 'marker', 'color'):
  853. if data[prop].has_key('update'):
  854. data[prop].pop('update')
  855. # height
  856. if data['height'].has_key('update'):
  857. self._display.SetVectorPointHeight(id,
  858. data['height']['value'])
  859. data['height'].pop('update')
  860. # surface
  861. if data['mode'].has_key('update'):
  862. sid = self.GetLayerId(type = 'raster', name = data['mode']['surface'])
  863. if sid > -1:
  864. self._display.SetVectorPointSurface(id, sid)
  865. data['mode'].pop('update')
  866. def GetLayerNames(self, type):
  867. """!Return list of map layer names of given type"""
  868. layerName = []
  869. for item in self.layers:
  870. mapLayer = self.tree.GetPyData(item)[0]['maplayer']
  871. if type != mapLayer.GetType():
  872. continue
  873. layerName.append(mapLayer.GetName())
  874. return layerName
  875. def GetLayerData(self, type, name):
  876. """!Return layer item data
  877. @return {} if no layer item found
  878. """
  879. for item in self.layers:
  880. mapLayer = self.tree.GetPyData(item)[0]['maplayer'].GetName()
  881. if mapLayer == name:
  882. return self.tree.GetPyData(item)[0]['nviz']
  883. return {}
  884. def GetLayerId(self, type, name):
  885. """!Get layer object id or -1"""
  886. if len(name) < 1:
  887. return -1
  888. for item in self.layers:
  889. mapLayer = self.tree.GetPyData(item)[0]['maplayer']
  890. if type != mapLayer.GetType() or \
  891. name != mapLayer.GetName():
  892. continue
  893. data = self.tree.GetPyData(item)[0]['nviz']
  894. if type == 'raster':
  895. return data['surface']['object']['id']
  896. elif type == 'vpoint':
  897. return data['vector']['points']['object']['id']
  898. elif type == 'vline':
  899. return data['vector']['lines']['object']['id']
  900. elif type == '3d-raster':
  901. return data['volume']['object']['id']
  902. return -1
  903. def SaveToFile(self, FileName, FileType, width, height):
  904. """!This draws the DC to a buffer that can be saved to a file.
  905. @todo fix BufferedPaintDC
  906. @param FileName file name
  907. @param FileType type of bitmap
  908. @param width image width
  909. @param height image height
  910. """
  911. self._display.SaveToFile(FileName, width, height)
  912. # pbuffer = wx.EmptyBitmap(max(1, self.Map.width), max(1, self.Map.height))
  913. # dc = wx.BufferedPaintDC(self, pbuffer)
  914. # dc.Clear()
  915. # self.SetCurrent()
  916. # self._display.Draw(False, -1)
  917. # pbuffer.SaveFile(FileName, FileType)
  918. # self.SwapBuffers()
  919. def GetDisplay(self):
  920. """!Get display instance"""
  921. return self._display
  922. def ZoomToMap(self):
  923. """!Reset view
  924. """
  925. self.lmgr.nviz.OnResetView(None)