render.py 50 KB

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