render.py 53 KB

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