render.py 46 KB

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