nviz_mapdisp.py 50 KB

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