render.py 44 KB

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