render.py 45 KB

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