render.py 47 KB

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