render.py 48 KB

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