render.py 46 KB

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