render.py 46 KB

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