nviz_mapdisp.py 44 KB

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