render.py 53 KB

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