render.py 44 KB

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