render.py 48 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475
  1. """
  2. @package core.render
  3. @brief Rendering map layers and overlays into map composition image.
  4. Classes:
  5. - render::Layer
  6. - render::MapLayer
  7. - render::Overlay
  8. - render::Map
  9. - render::RenderLayerMgr
  10. - render::RenderMapMgr
  11. (C) 2006-2015 by the GRASS Development Team
  12. This program is free software under the GNU General Public License
  13. (>=v2). Read the file COPYING that comes with GRASS for details.
  14. @author Michael Barton
  15. @author Jachym Cepicky
  16. @author Martin Landa <landa.martin gmail.com>
  17. """
  18. import os
  19. import sys
  20. import glob
  21. import math
  22. import copy
  23. import tempfile
  24. import types
  25. import time
  26. import wx
  27. from grass.script import core as grass
  28. from grass.script.utils import try_remove
  29. from grass.script.task import cmdlist_to_tuple, cmdtuple_to_list
  30. from grass.pydispatch.signal import Signal
  31. from grass.exceptions import CalledModuleError
  32. from core import utils
  33. from core.utils import _
  34. from core.ws import RenderWMSMgr
  35. from core.gcmd import GException, GError, RunCommand
  36. from core.debug import Debug
  37. from core.settings import UserSettings
  38. from core.gthread import gThread
  39. class Layer(object):
  40. """Virtual class which stores information about layers (map layers and
  41. overlays) of the map composition.
  42. - For map layer use MapLayer class.
  43. - For overlays use Overlay class.
  44. """
  45. def __init__(self, ltype, cmd, Map, name=None,
  46. active=True, hidden=False, opacity=1.0, mapfile=None, render=True):
  47. """Create new instance
  48. .. todo::
  49. pass cmd as tuple instead of list
  50. :param ltype: layer type ('raster', 'vector', 'overlay', 'command', etc.)
  51. :param cmd: GRASS command to render layer, given as list,
  52. e.g. ['d.rast', 'map=elevation@PERMANENT']
  53. :param map: render.Map instance
  54. :param name: layer name, e.g. 'elevation@PERMANENT' (for layer tree)
  55. :param active: layer is active, will be rendered only if True
  56. :param hidden: layer is hidden, won't be listed in Layer Manager if True
  57. :param float opacity: layer opacity <0;1>
  58. :param mapfile full path to output file or None
  59. """
  60. if mapfile:
  61. self.mapfile = mapfile
  62. else:
  63. if ltype == 'overlay':
  64. tempfile_sfx = ".png"
  65. else:
  66. tempfile_sfx = ".ppm"
  67. mapfile = tempfile.NamedTemporaryFile(suffix=tempfile_sfx, delete=False)
  68. # we don't want it open, we just need the name
  69. self.mapfile = mapfile.name
  70. mapfile.close()
  71. os.remove(self.mapfile) # remove empty file
  72. self.maskfile = self.mapfile.rsplit(".",1)[0] + ".pgm"
  73. # stores class which manages rendering instead of simple command - e.g. WMS
  74. self.renderMgr = None
  75. self.Map = Map
  76. self.type = None
  77. self.SetType(ltype)
  78. self.name = name
  79. if self.type == 'command':
  80. self.cmd = list()
  81. for c in cmd:
  82. self.cmd.append(cmdlist_to_tuple(c))
  83. else:
  84. self.cmd = cmdlist_to_tuple(cmd)
  85. self.active = active
  86. self.hidden = hidden
  87. self.opacity = opacity
  88. self.forceRender = render
  89. Debug.msg (3, "Layer.__init__(): type=%s, cmd='%s', name=%s, " \
  90. "active=%d, opacity=%d, hidden=%d" % \
  91. (self.type, self.GetCmd(string=True), self.name,
  92. self.active, self.opacity, self.hidden))
  93. def __del__(self):
  94. Debug.msg (3, "Layer.__del__(): layer=%s, cmd='%s'" %
  95. (self.name, self.GetCmd(string = True)))
  96. def __str__(self):
  97. return self.GetCmd(string=True)
  98. def __repr__(self):
  99. return self.__str__()
  100. def Render(self, env=None):
  101. """Render layer to image
  102. :return: rendered image filename
  103. :return: None on error or if cmdfile is defined
  104. """
  105. if not self.cmd:
  106. return None
  107. # ignore in 2D
  108. if self.type == '3d-raster':
  109. return None
  110. Debug.msg (3, "Layer.Render(): type=%s, name=%s, file=%s" % \
  111. (self.type, self.name, self.mapfile))
  112. # prepare command for each layer
  113. layertypes = utils.command2ltype.values() + ['overlay', 'command']
  114. if self.type not in layertypes:
  115. raise GException(_("<%(name)s>: layer type <%(type)s> is not supported") % \
  116. {'type' : self.type, 'name' : self.name})
  117. if not env:
  118. env = os.environ.copy()
  119. # render layers
  120. try:
  121. if self.type == 'command':
  122. first = True
  123. for c in self.cmd:
  124. self.renderMgr.Render(c, env)
  125. if first:
  126. env["GRASS_RENDER_FILE_READ"] = "TRUE"
  127. first = False
  128. else:
  129. self.renderMgr.Render(self.cmd, env)
  130. except GException:
  131. sys.stderr.write(_("Command '%s' failed\n") % self.GetCmd(string = True))
  132. sys.stderr.write(_("Details: %s\n") % e)
  133. # clean up after problems
  134. for f in [self.mapfile, self.maskfile]:
  135. if not f:
  136. continue
  137. try_remove(f)
  138. f = None
  139. return self.mapfile
  140. def GetCmd(self, string = False):
  141. """Get GRASS command as list of string.
  142. :param string: get command as string if True otherwise as list
  143. :return: command list/string
  144. """
  145. if string:
  146. if self.type == 'command':
  147. scmd = []
  148. for c in self.cmd:
  149. scmd.append(utils.GetCmdString(c))
  150. return ';'.join(scmd)
  151. else:
  152. return utils.GetCmdString(self.cmd)
  153. else:
  154. return self.cmd
  155. def GetType(self):
  156. """Get map layer type"""
  157. return self.type
  158. def GetElement(self):
  159. """Get map element type"""
  160. if self.type == 'raster':
  161. return 'cell'
  162. return self.type
  163. def GetOpacity(self):
  164. """
  165. Get layer opacity level
  166. :return: opacity level (<0, 1>)
  167. """
  168. return self.opacity
  169. def GetName(self, fullyQualified = True):
  170. """Get map layer name
  171. :param bool fullyQualified: True to return fully qualified name
  172. as a string 'name@mapset' otherwise
  173. directory { 'name', 'mapset' } is
  174. returned
  175. :return: string / directory
  176. """
  177. if fullyQualified:
  178. return self.name
  179. else:
  180. if '@' in self.name:
  181. return { 'name' : self.name.split('@')[0],
  182. 'mapset' : self.name.split('@')[1] }
  183. else:
  184. return { 'name' : self.name,
  185. 'mapset' : '' }
  186. def IsActive(self):
  187. """Check if layer is activated for rendering"""
  188. return self.active
  189. def IsHidden(self):
  190. """Check if layer is hidden"""
  191. return self.hidden
  192. def IsRendered(self):
  193. """!Check if layer was rendered (if the image file exists)"""
  194. if os.path.exists(self.mapfile):
  195. return True
  196. return False
  197. def SetType(self, ltype):
  198. """Set layer type"""
  199. if ltype not in utils.command2ltype.values() + ['overlay', 'command']:
  200. raise GException(_("Unsupported map layer type '%s'") % ltype)
  201. if not self.renderMgr:
  202. env = {}
  203. if ltype == 'wms':
  204. renderMgr = RenderWMSMgr
  205. else:
  206. renderMgr = RenderLayerMgr
  207. env['GRASS_RENDER_FILE'] = self.mapfile
  208. if ltype == 'overlay':
  209. env['GRASS_RENDER_FILE_READ'] = 'FALSE'
  210. env['GRASS_RENDER_TRANSPARENT'] = 'TRUE'
  211. self.renderMgr = renderMgr(self, env)
  212. self.type = ltype
  213. def SetName(self, name):
  214. """Set layer name"""
  215. self.name = name
  216. def SetActive(self, enable = True):
  217. """Active or deactive layer"""
  218. self.active = bool(enable)
  219. def SetHidden(self, enable = False):
  220. """Hide or show map layer in Layer Manager"""
  221. self.hidden = bool(enable)
  222. def SetOpacity(self, value):
  223. """Set opacity value"""
  224. if value < 0:
  225. value = 0.
  226. elif value > 1:
  227. value = 1.
  228. self.opacity = float(value)
  229. def SetCmd(self, cmd):
  230. """Set new command for layer"""
  231. if self.type == 'command':
  232. self.cmd = []
  233. for c in cmd:
  234. self.cmd.append(cmdlist_to_tuple(c))
  235. else:
  236. self.cmd = cmdlist_to_tuple(cmd)
  237. Debug.msg(3, "Layer.SetCmd(): cmd='%s'" % self.GetCmd(string = True))
  238. # for re-rendering
  239. self.forceRender = True
  240. def IsDownloading(self):
  241. """Is data downloading from web server e. g. wms"""
  242. return self.renderMgr.IsDownloading()
  243. def AbortThread(self):
  244. """Abort running thread e. g. downloading data"""
  245. self.renderMgr.Abort()
  246. def GetRenderMgr(self):
  247. """Get render manager """
  248. return self.renderMgr
  249. class MapLayer(Layer):
  250. def __init__(self, *args, **kwargs):
  251. """Represents map layer in the map canvas
  252. """
  253. Layer.__init__(self, *args, **kwargs)
  254. def GetMapset(self):
  255. """Get mapset of map layer
  256. :return: mapset name
  257. :return: '' on error (no name given)
  258. """
  259. if not self.name:
  260. return ''
  261. try:
  262. return self.name.split('@')[1]
  263. except IndexError:
  264. return self.name
  265. class Overlay(Layer):
  266. def __init__(self, id, *args, **kwargs):
  267. """Represents overlay displayed in map canvas
  268. :param id: overlay id (for PseudoDC)
  269. """
  270. Layer.__init__(self, ltype='overlay', *args, **kwargs)
  271. self.id = id
  272. class RenderLayerMgr(wx.EvtHandler):
  273. def __init__(self, layer, env):
  274. """Render layer into image
  275. :param layer: Layer to be rendered
  276. """
  277. self.layer = layer
  278. wx.EvtHandler.__init__(self)
  279. self.thread = gThread()
  280. self.updateProgress = Signal('RenderLayerMgr.updateProgress')
  281. self._startTime = None
  282. self._render_env = env
  283. def UpdateRenderEnv(self, env):
  284. self._render_env.update(env)
  285. def Render(self, cmd, env):
  286. """Render layer
  287. :param cmd: display command given as tuple
  288. :param env: environmental variables used for rendering
  289. """
  290. Debug.msg(1, "RenderLayerMgr.Render(%s): force=%d img=%s" % \
  291. (self.layer, self.layer.forceRender, self.layer.mapfile))
  292. env_cmd = env.copy()
  293. env_cmd.update(self._render_env)
  294. env_cmd['GRASS_RENDER_FILE'] = self.layer.mapfile
  295. cmd_render = copy.deepcopy(cmd)
  296. cmd_render[1]['quiet'] = True # be quiet
  297. # for k, v in env_cmd.iteritems():
  298. # if 'GRASS_' in k:
  299. # print 'export %s=%s' % (k, v)
  300. self._startTime = time.time()
  301. self.thread.Run(callable=self._render, cmd=cmd_render, env=env_cmd,
  302. ondone=self.OnRenderDone)
  303. self.layer.forceRender = False
  304. def _render(self, cmd, env):
  305. try:
  306. return grass.run_command(cmd[0], env=env, **cmd[1])
  307. except CalledModuleError as e:
  308. return 1
  309. def Abort(self):
  310. """Abort rendering process"""
  311. Debug.msg(1, "RenderLayerMgr({}).Abort()".format(self.layer))
  312. self.thread.Terminate()
  313. # force rendering layer next time
  314. self.layer.forceRender = True
  315. self.thread.Terminate(False)
  316. def IsDownloading(self):
  317. """Is downloading
  318. :return: always False
  319. """
  320. return False
  321. def OnRenderDone(self, event):
  322. """Rendering done
  323. Emits updateProcess
  324. """
  325. Debug.msg(1, "RenderLayerMgr.OnRenderDone(%s): ret=%d time=%f" % \
  326. (self.layer, event.ret, time.time() - self._startTime))
  327. if event.ret != 0:
  328. try:
  329. os.remove(self.layer.mapfile)
  330. except:
  331. pass
  332. self.updateProgress.emit(layer=self.layer)
  333. class RenderMapMgr(wx.EvtHandler):
  334. def __init__(self, Map):
  335. """Render map layers as image composition
  336. :param Map: Map object to be rendered
  337. """
  338. wx.EvtHandler.__init__(self)
  339. self.Map = Map
  340. self.updateMap = Signal('RenderMapMgr.updateMap')
  341. self.updateProgress = Signal('RenderMapMgr.updateProgress')
  342. self.renderDone = Signal('RenderMapMgr.renderDone')
  343. self.renderDone.connect(self.OnRenderDone)
  344. # GRASS environment variable (for rendering)
  345. self._render_env = {"GRASS_RENDER_BACKGROUNDCOLOR" : "000000",
  346. "GRASS_RENDER_FILE_COMPRESSION" : "0",
  347. "GRASS_RENDER_TRUECOLOR" : "TRUE",
  348. "GRASS_RENDER_TRANSPARENT" : "TRUE" }
  349. self._init()
  350. self._rendering = False
  351. def _init(self, env=None):
  352. """Init render manager
  353. :param env: environmental variables or None
  354. """
  355. self._startTime = time.time()
  356. self.progressInfo = None
  357. self._env = env
  358. self.layers = []
  359. # re-render from scratch
  360. if os.path.exists(self.Map.mapfile):
  361. os.remove(self.Map.mapfile)
  362. def _renderLayers(self, env, force = False, overlaysOnly = False):
  363. """Render all map layers into files
  364. :param dict env: environmental variables to be used for rendering process
  365. :param bool force: True to force rendering
  366. :param bool overlaysOnly: True to render only overlays
  367. :return: number of layers to be rendered
  368. """
  369. self.layers = self.Map.GetListOfLayers(ltype='overlay', active=True)
  370. if not overlaysOnly:
  371. self.layers += self.Map.GetListOfLayers(active=True,
  372. ltype='3d-raster',
  373. except_ltype=True)
  374. # reset progress
  375. self.ReportProgress()
  376. # render map layers if forced
  377. nlayers = 0
  378. for layer in self.layers:
  379. if force or layer.forceRender:
  380. nlayers += 1
  381. layer.Render(env)
  382. else:
  383. layer.GetRenderMgr().updateProgress.emit(layer=layer)
  384. Debug.msg(1, "RenderMapMgr.Render(): %d layers to be rendered "
  385. "(force=%d, all active layers -> %d)" % (nlayers, force,
  386. len(self.layers)))
  387. return nlayers
  388. def GetRenderEnv(self, windres=False):
  389. env = os.environ.copy()
  390. env.update(self._render_env)
  391. # use external gisrc if defined
  392. if self.Map.gisrc:
  393. env['GISRC'] = self.Map.gisrc
  394. env['GRASS_REGION'] = self.Map.SetRegion(windres)
  395. env['GRASS_RENDER_WIDTH'] = str(self.Map.width)
  396. env['GRASS_RENDER_HEIGHT'] = str(self.Map.height)
  397. if UserSettings.Get(group = 'display', key = 'driver', subkey = 'type') == 'png':
  398. env['GRASS_RENDER_IMMEDIATE'] = 'png'
  399. else:
  400. env['GRASS_RENDER_IMMEDIATE'] = 'cairo'
  401. return env
  402. def Render(self, force = False, windres = False):
  403. """Render map composition
  404. :param bool force: force rendering all map layers in the composition
  405. :param windres: True for region resolution instead for map resolution
  406. """
  407. if self._rendering:
  408. Debug.msg(1, "RenderMapMgr().Render(): cancelled (already rendering)")
  409. return
  410. wx.BeginBusyCursor()
  411. self._rendering = True
  412. env = self.GetRenderEnv(windres)
  413. self._init(env)
  414. if self._renderLayers(env, force, windres) == 0:
  415. self.renderDone.emit()
  416. def OnRenderDone(self):
  417. """Rendering process done
  418. Make image composiotion, emits updateMap event.
  419. """
  420. stopTime = time.time()
  421. maps = list()
  422. masks = list()
  423. opacities = list()
  424. for layer in self.layers:
  425. if layer.GetType() == 'overlay':
  426. continue
  427. if os.path.isfile(layer.mapfile):
  428. maps.append(layer.mapfile)
  429. masks.append(layer.maskfile)
  430. opacities.append(str(layer.opacity))
  431. # run g.pngcomp to get composite image
  432. bgcolor = ':'.join(map(str, UserSettings.Get(group = 'display', key = 'bgcolor',
  433. subkey = 'color')))
  434. startCompTime = time.time()
  435. if maps:
  436. ret, msg = RunCommand('g.pnmcomp',
  437. getErrorMsg = True,
  438. overwrite = True,
  439. input = '%s' % ",".join(maps),
  440. mask = '%s' % ",".join(masks),
  441. opacity = '%s' % ",".join(opacities),
  442. bgcolor = bgcolor,
  443. width = self.Map.width,
  444. height = self.Map.height,
  445. output = self.Map.mapfile,
  446. env=self._env)
  447. if ret != 0:
  448. self._rendering = False
  449. if wx.IsBusy():
  450. wx.EndBusyCursor()
  451. raise GException(_("Rendering failed: %s" % msg))
  452. stop = time.time()
  453. Debug.msg (1, "RenderMapMgr.OnRenderDone() time=%f sec (comp: %f)" % \
  454. (stop - self._startTime, stop - startCompTime))
  455. self._rendering = False
  456. if wx.IsBusy():
  457. wx.EndBusyCursor()
  458. self.updateMap.emit()
  459. def Abort(self):
  460. """Abort all rendering processes"""
  461. Debug.msg(1, "RenderMapMgr.Abort()")
  462. for layer in self.layers:
  463. layer.GetRenderMgr().Abort()
  464. self._init()
  465. if wx.IsBusy():
  466. wx.EndBusyCursor()
  467. self.updateProgress.emit(range=0, value=0, text=_("Rendering aborted"))
  468. def ReportProgress(self, layer=None):
  469. """Calculates progress in rendering/downloading
  470. and emits signal to inform progress bar about progress.
  471. Emits renderDone event when progressVal is equal to range.
  472. :param layer: Layer to be processed or None to reset
  473. """
  474. if self.progressInfo is None or layer is None:
  475. self.progressInfo = {'progresVal' : 0, # current progress value
  476. 'downloading' : [], # layers, which are downloading data
  477. 'rendered' : [], # already rendered layers
  478. 'range' : len(self.layers)}
  479. else:
  480. if layer not in self.progressInfo['rendered']:
  481. self.progressInfo['rendered'].append(layer)
  482. if layer.IsDownloading() and \
  483. layer not in self.progressInfo['downloading']:
  484. self.progressInfo['downloading'].append(layer)
  485. else:
  486. self.progressInfo['progresVal'] += 1
  487. if layer in self.progressInfo['downloading']:
  488. self.progressInfo['downloading'].remove(layer)
  489. # for updating statusbar text
  490. stText = ''
  491. first = True
  492. for layer in self.progressInfo['downloading']:
  493. if first:
  494. stText += _("Downloading data ")
  495. first = False
  496. else:
  497. stText += ', '
  498. stText += '<%s>' % layer.GetName()
  499. if stText:
  500. stText += '...'
  501. if self.progressInfo['range'] != len(self.progressInfo['rendered']):
  502. if stText:
  503. stText = _('Rendering & ') + stText
  504. else:
  505. stText = _('Rendering...')
  506. self.updateProgress.emit(range=self.progressInfo['range'],
  507. value=self.progressInfo['progresVal'],
  508. text=stText)
  509. if layer and self.progressInfo['progresVal'] == self.progressInfo['range']:
  510. self.renderDone.emit()
  511. class Map(object):
  512. def __init__(self, gisrc = None):
  513. """Map composition (stack of map layers and overlays)
  514. :param gisrc: alternative gisrc (used eg. by georectifier)
  515. """
  516. Debug.msg (1, "Map.__init__(): gisrc=%s" % gisrc)
  517. # region/extent settigns
  518. self.wind = dict() # WIND settings (wind file)
  519. self.region = dict() # region settings (g.region)
  520. self.width = 640 # map width
  521. self.height = 480 # map height
  522. # list of layers
  523. self.layers = list() # stack of available GRASS layer
  524. self.overlays = list() # stack of available overlays
  525. self.ovlookup = dict() # lookup dictionary for overlay items and overlays
  526. # path to external gisrc
  527. self.gisrc = gisrc
  528. # generated file for g.pnmcomp output for rendering the map
  529. self.mapfile = grass.tempfile(create = False) + '.ppm'
  530. # setting some initial env. variables
  531. if not self.GetWindow():
  532. sys.stderr.write(_("Trying to recover from default region..."))
  533. RunCommand('g.region', flags='d')
  534. # projection info
  535. self.projinfo = self._projInfo()
  536. self.layerChanged = Signal('Map.layerChanged')
  537. self.layerRemoved = Signal('Map:layerRemoved')
  538. self.layerAdded = Signal('Map:layerAdded')
  539. self.renderMgr = RenderMapMgr(self)
  540. def GetRenderMgr(self):
  541. """Get render manager """
  542. return self.renderMgr
  543. def GetProjInfo(self):
  544. """Get projection info"""
  545. return self.projinfo
  546. def _projInfo(self):
  547. """Return region projection and map units information
  548. """
  549. projinfo = dict()
  550. if not grass.find_program('g.proj', '--help'):
  551. sys.exit(_("GRASS module '%s' not found. Unable to start map "
  552. "display window.") % 'g.proj')
  553. env = os.environ.copy()
  554. if self.gisrc:
  555. env['GISRC'] = self.gisrc
  556. ret = RunCommand(prog='g.proj', read=True, flags='p', env=env)
  557. if not ret:
  558. return projinfo
  559. for line in ret.splitlines():
  560. if ':' in line:
  561. key, val = map(lambda x: x.strip(), line.split(':'))
  562. if key in ['units']:
  563. val = val.lower()
  564. projinfo[key] = val
  565. elif "XY location (unprojected)" in line:
  566. projinfo['proj'] = 'xy'
  567. projinfo['units'] = ''
  568. break
  569. return projinfo
  570. def GetWindow(self):
  571. """Read WIND file and set up self.wind dictionary"""
  572. # FIXME: duplicated region WIND == g.region (at least some values)
  573. env = grass.gisenv()
  574. filename = os.path.join (env['GISDBASE'],
  575. env['LOCATION_NAME'],
  576. env['MAPSET'],
  577. "WIND")
  578. try:
  579. windfile = open (filename, "r")
  580. except IOError as e:
  581. sys.exit(_("Error: Unable to open '%(file)s'. Reason: %(ret)s. wxGUI exited.\n") % \
  582. { 'file' : filename, 'ret' : e})
  583. for line in windfile.readlines():
  584. line = line.strip()
  585. try:
  586. key, value = line.split(":", 1)
  587. except ValueError as e:
  588. sys.stderr.write(_("\nERROR: Unable to read WIND file: %s\n") % e)
  589. return None
  590. self.wind[key.strip()] = value.strip()
  591. windfile.close()
  592. return self.wind
  593. def AdjustRegion(self):
  594. """Adjusts display resolution to match monitor size in
  595. pixels. Maintains constant display resolution, not related to
  596. computational region. Do NOT use the display resolution to set
  597. computational resolution. Set computational resolution through
  598. g.region.
  599. """
  600. mapwidth = abs(self.region["e"] - self.region["w"])
  601. mapheight = abs(self.region['n'] - self.region['s'])
  602. self.region["nsres"] = mapheight / self.height
  603. self.region["ewres"] = mapwidth / self.width
  604. self.region['rows'] = round(mapheight / self.region["nsres"])
  605. self.region['cols'] = round(mapwidth / self.region["ewres"])
  606. self.region['cells'] = self.region['rows'] * self.region['cols']
  607. Debug.msg (3, "Map.AdjustRegion(): %s" % self.region)
  608. return self.region
  609. def AlignResolution(self):
  610. """Sets display extents to even multiple of current
  611. resolution defined in WIND file from SW corner. This must be
  612. done manually as using the -a flag can produce incorrect
  613. extents.
  614. """
  615. # new values to use for saving to region file
  616. new = {}
  617. n = s = e = w = 0.0
  618. nsres = ewres = 0.0
  619. # Get current values for region and display
  620. reg = self.GetRegion()
  621. nsres = reg['nsres']
  622. ewres = reg['ewres']
  623. n = float(self.region['n'])
  624. s = float(self.region['s'])
  625. e = float(self.region['e'])
  626. w = float(self.region['w'])
  627. # Calculate rows, columns, and extents
  628. new['rows'] = math.fabs(round((n-s)/nsres))
  629. new['cols'] = math.fabs(round((e-w)/ewres))
  630. # Calculate new extents
  631. new['s'] = nsres * round(s / nsres)
  632. new['w'] = ewres * round(w / ewres)
  633. new['n'] = new['s'] + (new['rows'] * nsres)
  634. new['e'] = new['w'] + (new['cols'] * ewres)
  635. return new
  636. def AlignExtentFromDisplay(self):
  637. """Align region extent based on display size from center
  638. point"""
  639. # calculate new bounding box based on center of display
  640. if self.region["ewres"] > self.region["nsres"]:
  641. res = self.region["ewres"]
  642. else:
  643. res = self.region["nsres"]
  644. Debug.msg(3, "Map.AlignExtentFromDisplay(): width=%d, height=%d, res=%f, center=%f,%f" % \
  645. (self.width, self.height, res, self.region['center_easting'],
  646. self.region['center_northing']))
  647. ew = (self.width / 2) * res
  648. ns = (self.height / 2) * res
  649. self.region['n'] = self.region['center_northing'] + ns
  650. self.region['s'] = self.region['center_northing'] - ns
  651. self.region['e'] = self.region['center_easting'] + ew
  652. self.region['w'] = self.region['center_easting'] - ew
  653. # LL locations
  654. if self.projinfo['proj'] == 'll':
  655. self.region['n'] = min(self.region['n'], 90.0)
  656. self.region['s'] = max(self.region['s'], -90.0)
  657. def ChangeMapSize(self, size):
  658. """Change size of rendered map.
  659. :param size: map size given as tuple
  660. """
  661. try:
  662. self.width = int(size[0])
  663. self.height = int(size[1])
  664. if self.width < 1 or self.height < 1:
  665. sys.stderr.write(_("Invalid map size %d,%d\n") % (self.width, self.height))
  666. raise ValueError
  667. except ValueError:
  668. self.width = 640
  669. self.height = 480
  670. Debug.msg(2, "Map.ChangeMapSize(): width=%d, height=%d" % \
  671. (self.width, self.height))
  672. def GetRegion(self, rast=None, zoom=False, vect=None, rast3d=None, regionName=None,
  673. n=None, s=None, e=None, w=None, default=False,
  674. update=False, add3d=False):
  675. """Get region settings (g.region -upgc)
  676. Optionally extent, raster or vector map layer can be given.
  677. :param rast: list of raster maps
  678. :param zoom: zoom to raster map (ignore NULLs)
  679. :param vect: list of vector maps
  680. :param rast3d: 3d raster map (not list, no support of multiple 3d rasters in g.region)
  681. :param regionName: named region or None
  682. :param n,s,e,w: force extent
  683. :param default: force default region settings
  684. :param update: if True update current display region settings
  685. :param add3d: add 3d region settings
  686. :return: region settings as dictionary, e.g. {
  687. 'n':'4928010', 's':'4913700', 'w':'589980',...}
  688. :func:`GetCurrentRegion()`
  689. """
  690. region = {}
  691. env = os.environ.copy()
  692. if self.gisrc:
  693. env['GISRC'] = self.gisrc
  694. # do not update & shell style output
  695. cmd = {}
  696. cmd['flags'] = 'ugpc'
  697. if default:
  698. cmd['flags'] += 'd'
  699. if add3d:
  700. cmd['flags'] += '3'
  701. if regionName:
  702. cmd['region'] = regionName
  703. if n:
  704. cmd['n'] = n
  705. if s:
  706. cmd['s'] = s
  707. if e:
  708. cmd['e'] = e
  709. if w:
  710. cmd['w'] = w
  711. if rast:
  712. if zoom:
  713. cmd['zoom'] = rast[0]
  714. else:
  715. cmd['raster'] = ','.join(rast)
  716. if vect:
  717. cmd['vector'] = ','.join(vect)
  718. if rast3d:
  719. cmd['raster_3d'] = rast3d
  720. ret, reg, msg = RunCommand('g.region',
  721. read = True,
  722. getErrorMsg = True,
  723. env=env,
  724. **cmd)
  725. if ret != 0:
  726. if rast:
  727. message = _("Unable to zoom to raster map <%s>.") % rast[0] + \
  728. "\n\n" + _("Details:") + " %s" % msg
  729. elif vect:
  730. message = _("Unable to zoom to vector map <%s>.") % vect[0] + \
  731. "\n\n" + _("Details:") + " %s" % msg
  732. elif rast3d:
  733. message = _("Unable to zoom to 3d raster map <%s>.") % rast3d + \
  734. "\n\n" + _("Details:") + " %s" % msg
  735. else:
  736. message = _("Unable to get current geographic extent. "
  737. "Force quiting wxGUI. Please manually run g.region to "
  738. "fix the problem.")
  739. GError(message)
  740. return self.region
  741. for r in reg.splitlines():
  742. key, val = r.split("=", 1)
  743. try:
  744. region[key] = float(val)
  745. except ValueError:
  746. region[key] = val
  747. Debug.msg (3, "Map.GetRegion(): %s" % region)
  748. if update:
  749. self.region = region
  750. return region
  751. def GetCurrentRegion(self):
  752. """Get current display region settings
  753. :func:`GetRegion()`
  754. """
  755. return self.region
  756. def SetRegion(self, windres = False, windres3 = False):
  757. """Render string for GRASS_REGION env. variable, so that the
  758. images will be rendered from desired zoom level.
  759. :param windres: uses resolution from WIND file rather than
  760. display (for modules that require set resolution
  761. like d.rast.num)
  762. :return: String usable for GRASS_REGION variable or None
  763. """
  764. grass_region = ""
  765. if windres:
  766. compRegion = self.GetRegion(add3d = windres3)
  767. region = copy.copy(self.region)
  768. for key in ('nsres', 'ewres', 'cells'):
  769. region[key] = compRegion[key]
  770. if windres3:
  771. for key in ('nsres3', 'ewres3', 'tbres', 'cells3',
  772. 'cols3', 'rows3', 'depths'):
  773. if key in compRegion:
  774. region[key] = compRegion[key]
  775. else:
  776. # adjust region settings to match monitor
  777. region = self.AdjustRegion()
  778. # read values from wind file
  779. try:
  780. for key in self.wind.keys():
  781. if key == 'north':
  782. grass_region += "north: %s; " % \
  783. (region['n'])
  784. continue
  785. elif key == "south":
  786. grass_region += "south: %s; " % \
  787. (region['s'])
  788. continue
  789. elif key == "east":
  790. grass_region += "east: %s; " % \
  791. (region['e'])
  792. continue
  793. elif key == "west":
  794. grass_region += "west: %s; " % \
  795. (region['w'])
  796. continue
  797. elif key == "e-w resol":
  798. grass_region += "e-w resol: %f; " % \
  799. (region['ewres'])
  800. continue
  801. elif key == "n-s resol":
  802. grass_region += "n-s resol: %f; " % \
  803. (region['nsres'])
  804. continue
  805. elif key == "cols":
  806. if windres:
  807. continue
  808. grass_region += 'cols: %d; ' % \
  809. region['cols']
  810. continue
  811. elif key == "rows":
  812. if windres:
  813. continue
  814. grass_region += 'rows: %d; ' % \
  815. region['rows']
  816. continue
  817. elif key == "n-s resol3" and windres3:
  818. grass_region += "n-s resol3: %f; " % \
  819. (region['nsres3'])
  820. elif key == "e-w resol3" and windres3:
  821. grass_region += "e-w resol3: %f; " % \
  822. (region['ewres3'])
  823. elif key == "t-b resol" and windres3:
  824. grass_region += "t-b resol: %f; " % \
  825. (region['tbres'])
  826. elif key == "cols3" and windres3:
  827. grass_region += "cols3: %d; " % \
  828. (region['cols3'])
  829. elif key == "rows3" and windres3:
  830. grass_region += "rows3: %d; " % \
  831. (region['rows3'])
  832. elif key == "depths" and windres3:
  833. grass_region += "depths: %d; " % \
  834. (region['depths'])
  835. else:
  836. grass_region += key + ": " + self.wind[key] + "; "
  837. Debug.msg (3, "Map.SetRegion(): %s" % grass_region)
  838. return grass_region
  839. except:
  840. return None
  841. def GetListOfLayers(self, ltype = None, mapset = None, name = None,
  842. active = None, hidden = None, except_ltype=False):
  843. """Returns list of layers of selected properties or list of
  844. all layers.
  845. :param ltype: layer type, e.g. raster/vector/wms/overlay (value or tuple of values)
  846. :param mapset: all layers from given mapset (only for maplayers)
  847. :param name: all layers with given name
  848. :param active: only layers with 'active' attribute set to True or False
  849. :param hidden: only layers with 'hidden' attribute set to True or False
  850. :param except_ltype: True to return all layers with type not in ltype
  851. :return: list of selected layers
  852. """
  853. selected = []
  854. if type(ltype) == types.StringType:
  855. one_type = True
  856. else:
  857. one_type = False
  858. if one_type and ltype == 'overlay':
  859. llist = self.overlays
  860. else:
  861. llist = self.layers
  862. # ["raster", "vector", "wms", ... ]
  863. for layer in llist:
  864. # specified type only
  865. if ltype != None:
  866. if one_type:
  867. if (not except_ltype and layer.type != ltype) or \
  868. (except_ltype and layer.type == ltype):
  869. continue
  870. elif not one_type:
  871. if (not except_ltype and layer.type not in ltype) or \
  872. (except_ltype and layer.type in ltype):
  873. continue
  874. # mapset
  875. if (mapset != None and ltype != 'overlay') and \
  876. layer.GetMapset() != mapset:
  877. continue
  878. # name
  879. if name != None and layer.name != name:
  880. continue
  881. # hidden and active layers
  882. if active != None and \
  883. hidden != None:
  884. if layer.active == active and \
  885. layer.hidden == hidden:
  886. selected.append(layer)
  887. # active layers
  888. elif active != None:
  889. if layer.active == active:
  890. selected.append(layer)
  891. # hidden layers
  892. elif hidden != None:
  893. if layer.hidden == hidden:
  894. selected.append(layer)
  895. # all layers
  896. else:
  897. selected.append(layer)
  898. Debug.msg (3, "Map.GetListOfLayers(ltype=%s): -> %d" % (ltype, len(selected)))
  899. return selected
  900. def Render(self, force = False, windres = False):
  901. """Creates final image composite
  902. This function can conditionaly use high-level tools, which
  903. should be avaliable in wxPython library
  904. :param force: force rendering
  905. :param windres: use region resolution (True) otherwise display
  906. resolution
  907. """
  908. self.renderMgr.Render(force, windres)
  909. def _addLayer(self, layer, pos=-1):
  910. if layer.type == 'overlay':
  911. llist = self.overlays
  912. else:
  913. llist = self.layers
  914. # add maplayer to the list of layers
  915. if pos > -1:
  916. llist.insert(pos, layer)
  917. else:
  918. llist.append(layer)
  919. Debug.msg (3, "Map._addLayer(): layer=%s type=%s" % (layer.name, layer.type))
  920. return layer
  921. def AddLayer(self, ltype, command, name = None,
  922. active = True, hidden = False, opacity = 1.0, render = False,
  923. pos = -1):
  924. """Adds generic map layer to list of layers
  925. :param ltype: layer type ('raster', 'vector', etc.)
  926. :param command: GRASS command given as list
  927. :param name: layer name
  928. :param active: layer render only if True
  929. :param hidden: layer not displayed in layer tree if True
  930. :param opacity: opacity level range from 0(transparent) - 1(not transparent)
  931. :param render: render an image if True
  932. :param pos: position in layer list (-1 for append)
  933. :return: new layer on success
  934. :return: None on failure
  935. """
  936. # opacity must be <0;1>
  937. if opacity < 0:
  938. opacity = 0
  939. elif opacity > 1:
  940. opacity = 1
  941. layer = MapLayer(ltype = ltype, name = name, cmd = command, Map = self,
  942. active = active, hidden = hidden, opacity = opacity)
  943. self._addLayer(layer, pos)
  944. renderMgr = layer.GetRenderMgr()
  945. Debug.msg(1, "Map.AddLayer(): ltype={}, command={}".format(ltype, layer.GetCmd(string=True)))
  946. if renderMgr:
  947. if layer.type == 'wms':
  948. renderMgr.dataFetched.connect(self.renderMgr.ReportProgress)
  949. renderMgr.updateProgress.connect(self.renderMgr.ReportProgress)
  950. layer.forceRender = render
  951. self.layerAdded.emit(layer=layer)
  952. return layer
  953. def DeleteAllLayers(self, overlay = False):
  954. """Delete all layers
  955. :param overlay: True to delete also overlayes
  956. """
  957. self.layers = []
  958. if overlay:
  959. self.overlays = []
  960. def DeleteLayer(self, layer, overlay = False):
  961. """Removes layer from list of layers
  962. :param layer: layer instance in layer tree
  963. :param overlay: delete overlay (use self.DeleteOverlay() instead)
  964. :return: removed layer on success or None
  965. """
  966. Debug.msg (3, "Map.DeleteLayer(): name=%s" % layer.name)
  967. if overlay:
  968. list = self.overlays
  969. else:
  970. list = self.layers
  971. if layer in list:
  972. if layer.mapfile:
  973. base = os.path.split(layer.mapfile)[0]
  974. mapfile = os.path.split(layer.mapfile)[1]
  975. tempbase = mapfile.split('.')[0]
  976. if base == '' or tempbase == '':
  977. return None
  978. basefile = os.path.join(base, tempbase) + r'.*'
  979. for f in glob.glob(basefile):
  980. os.remove(f)
  981. list.remove(layer)
  982. self.layerRemoved.emit(layer=layer)
  983. return layer
  984. return None
  985. def SetLayers(self, layers):
  986. self.layers = layers
  987. Debug.msg(5, "Map.SetLayers(): layers=%s" % (layers))
  988. def ChangeLayer(self, layer, render = False, **kargs):
  989. """Change map layer properties
  990. :param layer: map layer instance
  991. :param ltype: layer type ('raster', 'vector', etc.)
  992. :param command: GRASS command given as list
  993. :param name: layer name
  994. :param active: layer render only if True
  995. :param hidden: layer not displayed in layer tree if True
  996. :param opacity: opacity level range from 0(transparent) - 1(not transparent)
  997. :param render: render an image if True
  998. """
  999. Debug.msg (3, "Map.ChangeLayer(): layer=%s" % layer.name)
  1000. if 'ltype' in kargs:
  1001. layer.SetType(kargs['ltype']) # check type
  1002. if 'command' in kargs:
  1003. layer.SetCmd(kargs['command'])
  1004. if 'name' in kargs:
  1005. layer.SetName(kargs['name'])
  1006. if 'active' in kargs:
  1007. layer.SetActive(kargs['active'])
  1008. if 'hidden' in kargs:
  1009. layer.SetHidden(kargs['hidden'])
  1010. if 'opacity' in kargs:
  1011. layer.SetOpacity(kargs['opacity'])
  1012. self.forceRender = render
  1013. # not needed since there is self.forceRender
  1014. ### self.layerChanged(layer=layer)
  1015. return layer
  1016. def ChangeOpacity(self, layer, opacity):
  1017. """Changes opacity value of map layer
  1018. :param layer: layer instance in layer tree
  1019. :param opacity: opacity level <0;1>
  1020. """
  1021. # opacity must be <0;1>
  1022. if opacity < 0: opacity = 0
  1023. elif opacity > 1: opacity = 1
  1024. layer.opacity = opacity
  1025. Debug.msg (3, "Map.ChangeOpacity(): layer=%s, opacity=%f" % \
  1026. (layer.name, layer.opacity))
  1027. def ChangeLayerActive(self, layer, active):
  1028. """Enable or disable map layer
  1029. :param layer: layer instance in layer tree
  1030. :param active: to be rendered (True)
  1031. """
  1032. layer.active = active
  1033. if active:
  1034. layer.forceRender = True
  1035. Debug.msg (3, "Map.ChangeLayerActive(): name='%s' -> active=%d" % \
  1036. (layer.name, layer.active))
  1037. def ChangeLayerName (self, layer, name):
  1038. """Change name of the layer
  1039. :param layer: layer instance in layer tree
  1040. :param name: layer name to set up
  1041. """
  1042. Debug.msg (3, "Map.ChangeLayerName(): from=%s to=%s" % \
  1043. (layer.name, name))
  1044. layer.name = name
  1045. def RemoveLayer(self, name = None, id = None):
  1046. """Removes layer from layer list
  1047. Layer is defined by name@mapset or id.
  1048. :param name: layer name (must be unique)
  1049. :param id: layer index in layer list def __init__(self,
  1050. targetFile, region, bandsNum, gdalDriver,
  1051. fillValue = None):
  1052. :return: removed layer on success
  1053. :return: None on failure
  1054. """
  1055. # delete by name
  1056. if name:
  1057. retlayer = None
  1058. for layer in self.layers:
  1059. if layer.name == name:
  1060. retlayer = layer
  1061. os.remove(layer.mapfile)
  1062. os.remove(layer.maskfile)
  1063. self.layers.remove(layer)
  1064. return retlayer
  1065. # del by id
  1066. elif id != None:
  1067. return self.layers.pop(id)
  1068. return None
  1069. def GetLayerIndex(self, layer, overlay = False):
  1070. """Get index of layer in layer list.
  1071. :param layer: layer instace in layer tree
  1072. :param overlay: use list of overlays instead
  1073. :return: layer index
  1074. :return: -1 if layer not found
  1075. """
  1076. if overlay:
  1077. list = self.overlays
  1078. else:
  1079. list = self.layers
  1080. if layer in list:
  1081. return list.index(layer)
  1082. return -1
  1083. def AddOverlay(self, id, ltype, command,
  1084. active = True, hidden = True, opacity = 1.0, render = False):
  1085. """Adds overlay (grid, barscale, legend, etc.) to list of
  1086. overlays
  1087. :param id: overlay id (PseudoDC)
  1088. :param ltype: overlay type (barscale, legend)
  1089. :param command: GRASS command to render overlay
  1090. :param active: overlay activated (True) or disabled (False)
  1091. :param hidden: overlay is not shown in layer tree (if True)
  1092. :param render: render an image (if True)
  1093. :return: new layer on success
  1094. :return: None on failure
  1095. """
  1096. Debug.msg (1, "Map.AddOverlay(): cmd={}".format(command))
  1097. overlay = Overlay(id = id, name = ltype, cmd = command, Map = self,
  1098. active = active, hidden = hidden, opacity = opacity)
  1099. self._addLayer(overlay, render)
  1100. renderMgr = overlay.GetRenderMgr()
  1101. if renderMgr:
  1102. renderMgr.updateProgress.connect(self.renderMgr.ReportProgress)
  1103. return overlay
  1104. def ChangeOverlay(self, id, render = False, **kargs):
  1105. """Change overlay properities
  1106. Add new overlay if overlay with 'id' doesn't exist.
  1107. :param id: overlay id (PseudoDC)
  1108. :param ltype: overlay ltype (barscale, legend)
  1109. :param command: GRASS command to render overlay
  1110. :param active: overlay activated (True) or disabled (False)
  1111. :param hidden: overlay is not shown in layer tree (if True)
  1112. :param render: render an image (if True)
  1113. :return: new layer on success
  1114. """
  1115. overlay = self.GetOverlay(id, list = False)
  1116. if overlay is None:
  1117. overlay = Overlay(id, ltype = None, cmd = None)
  1118. if 'ltype' in kargs:
  1119. overlay.SetName(kargs['ltype']) # ltype -> overlay
  1120. if 'command' in kargs:
  1121. overlay.SetCmd(kargs['command'])
  1122. if 'active' in kargs:
  1123. overlay.SetActive(kargs['active'])
  1124. if 'hidden' in kargs:
  1125. overlay.SetHidden(kargs['hidden'])
  1126. if 'opacity' in kargs:
  1127. overlay.SetOpacity(kargs['opacity'])
  1128. overlay.forceRender = render
  1129. return overlay
  1130. def GetOverlay(self, id, list=False):
  1131. """Return overlay(s) with 'id'
  1132. :param id: overlay id
  1133. :param list: return list of overlays of True
  1134. otherwise suppose 'id' to be unique
  1135. :return: list of overlays (list=True)
  1136. :return: overlay (list=False)
  1137. :return: None (list=False) if no overlay or more overlays found
  1138. """
  1139. ovl = []
  1140. for overlay in self.overlays:
  1141. if overlay.id == id:
  1142. ovl.append(overlay)
  1143. if not list:
  1144. if len(ovl) != 1:
  1145. return None
  1146. else:
  1147. return ovl[0]
  1148. return ovl
  1149. def DeleteOverlay(self, overlay):
  1150. """Delete overlay
  1151. :param overlay: overlay layer
  1152. :return: removed overlay on success or None
  1153. """
  1154. return self.DeleteLayer(overlay, overlay = True)
  1155. def _clean(self, llist):
  1156. for layer in llist:
  1157. if layer.maskfile:
  1158. try_remove(layer.maskfile)
  1159. if layer.mapfile:
  1160. try_remove(layer.mapfile)
  1161. llist.remove(layer)
  1162. def Clean(self):
  1163. """Clean layer stack - go trough all layers and remove them
  1164. from layer list.
  1165. Removes also mapfile and maskfile.
  1166. """
  1167. self._clean(self.layers)
  1168. self._clean(self.overlays)
  1169. def ReverseListOfLayers(self):
  1170. """Reverse list of layers"""
  1171. return self.layers.reverse()
  1172. def RenderOverlays(self, force):
  1173. """Render overlays only (for nviz)"""
  1174. for layer in self.overlays:
  1175. if force or layer.forceRender:
  1176. layer.Render()
  1177. def AbortAllThreads(self):
  1178. """Abort all layers threads e. g. donwloading data"""
  1179. self.renderMgr.Abort()