render.py 43 KB

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