render.py 45 KB

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