render.py 48 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424
  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-2011 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, cmdfile = None, mapfile = None, envfile = None, monitor = None):
  269. """!Map composition (stack of map layers and overlays)
  270. @param gisrc alternative gisrc (used eg. by georectifier)
  271. @param cmdline full path to the cmd file (defined by d.mon)
  272. @param mapfile full path to the map file (defined by d.mon)
  273. @param envfile full path to the env file (defined by d.mon)
  274. @param monitor name of monitor (defined by d.mon)
  275. """
  276. # region/extent settigns
  277. self.wind = dict() # WIND settings (wind file)
  278. self.region = dict() # region settings (g.region)
  279. self.width = 640 # map width
  280. self.height = 480 # map height
  281. # list of layers
  282. self.layers = list() # stack of available GRASS layer
  283. self.overlays = list() # stack of available overlays
  284. self.ovlookup = dict() # lookup dictionary for overlay items and overlays
  285. # environment settings
  286. self.env = dict()
  287. # path to external gisrc
  288. self.gisrc = gisrc
  289. self.cmdfile = cmdfile
  290. self.envfile = envfile
  291. self.monitor = monitor
  292. if mapfile:
  293. self.mapfileCmd = mapfile
  294. self.maskfileCmd = os.path.splitext(mapfile)[0] + '.pgm'
  295. # generated file for g.pnmcomp output for rendering the map
  296. self.mapfile = grass.tempfile(create = False) + '.ppm'
  297. # setting some initial env. variables
  298. self._initGisEnv() # g.gisenv
  299. self.GetWindow()
  300. # GRASS environment variable (for rendering)
  301. env = {"GRASS_BACKGROUNDCOLOR" : "FFFFFF",
  302. "GRASS_COMPRESSION" : "0",
  303. "GRASS_TRUECOLOR" : "TRUE",
  304. "GRASS_TRANSPARENT" : "TRUE",
  305. "GRASS_PNG_READ" : "FALSE",
  306. }
  307. self._writeEnvFile(env)
  308. self._writeEnvFile({"GRASS_PNG_READ" : "TRUE"})
  309. for k, v in env.iteritems():
  310. os.environ[k] = v
  311. # projection info
  312. self.projinfo = self._projInfo()
  313. def _runCommand(self, cmd, **kwargs):
  314. """!Run command in environment defined by self.gisrc if
  315. defined"""
  316. # use external gisrc if defined
  317. gisrc_orig = os.getenv("GISRC")
  318. if self.gisrc:
  319. os.environ["GISRC"] = self.gisrc
  320. ret = cmd(**kwargs)
  321. # back to original gisrc
  322. if self.gisrc:
  323. os.environ["GISRC"] = gisrc_orig
  324. return ret
  325. def _initGisEnv(self):
  326. """!Stores GRASS variables (g.gisenv) to self.env variable
  327. """
  328. if not os.getenv("GISBASE"):
  329. sys.exit(_("GISBASE not set. You must be in GRASS GIS to run this program."))
  330. self.env = self._runCommand(grass.gisenv)
  331. def GetProjInfo(self):
  332. """!Get projection info"""
  333. return self.projinfo
  334. def _projInfo(self):
  335. """!Return region projection and map units information
  336. """
  337. projinfo = dict()
  338. if not grass.find_program('g.proj', ['--help']):
  339. sys.exit(_("GRASS module '%s' not found. Unable to start map "
  340. "display window.") % 'g.proj')
  341. ret = self._runCommand(RunCommand, prog = 'g.proj',
  342. read = True, flags = 'p')
  343. if not ret:
  344. return projinfo
  345. for line in ret.splitlines():
  346. if ':' in line:
  347. key, val = map(lambda x: x.strip(), line.split(':'))
  348. if key in ['units']:
  349. val = val.lower()
  350. projinfo[key] = val
  351. elif "XY location (unprojected)" in line:
  352. projinfo['proj'] = 'xy'
  353. projinfo['units'] = ''
  354. break
  355. return projinfo
  356. def GetWindow(self):
  357. """!Read WIND file and set up self.wind dictionary"""
  358. # FIXME: duplicated region WIND == g.region (at least some values)
  359. filename = os.path.join (self.env['GISDBASE'],
  360. self.env['LOCATION_NAME'],
  361. self.env['MAPSET'],
  362. "WIND")
  363. try:
  364. windfile = open (filename, "r")
  365. except IOError, e:
  366. sys.exit(_("Error: Unable to open '%(file)s'. Reason: %(ret)s. wxGUI exited.\n") % \
  367. { 'file' : filename, 'ret' : e})
  368. for line in windfile.readlines():
  369. line = line.strip()
  370. key, value = line.split(":", 1)
  371. self.wind[key.strip()] = value.strip()
  372. windfile.close()
  373. return self.wind
  374. def AdjustRegion(self):
  375. """!Adjusts display resolution to match monitor size in
  376. pixels. Maintains constant display resolution, not related to
  377. computational region. Do NOT use the display resolution to set
  378. computational resolution. Set computational resolution through
  379. g.region.
  380. """
  381. mapwidth = abs(self.region["e"] - self.region["w"])
  382. mapheight = abs(self.region['n'] - self.region['s'])
  383. self.region["nsres"] = mapheight / self.height
  384. self.region["ewres"] = mapwidth / self.width
  385. self.region['rows'] = round(mapheight / self.region["nsres"])
  386. self.region['cols'] = round(mapwidth / self.region["ewres"])
  387. self.region['cells'] = self.region['rows'] * self.region['cols']
  388. Debug.msg (3, "Map.AdjustRegion(): %s" % self.region)
  389. return self.region
  390. def AlignResolution(self):
  391. """!Sets display extents to even multiple of current
  392. resolution defined in WIND file from SW corner. This must be
  393. done manually as using the -a flag can produce incorrect
  394. extents.
  395. """
  396. # new values to use for saving to region file
  397. new = {}
  398. n = s = e = w = 0.0
  399. nwres = ewres = 0.0
  400. # Get current values for region and display
  401. reg = self.GetRegion()
  402. nsres = reg['nsres']
  403. ewres = reg['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
  419. point"""
  420. # calculate new bounding box based on center of display
  421. if self.region["ewres"] > self.region["nsres"]:
  422. res = self.region["ewres"]
  423. else:
  424. res = self.region["nsres"]
  425. Debug.msg(3, "Map.AlignExtentFromDisplay(): width=%d, height=%d, res=%f, center=%f,%f" % \
  426. (self.width, self.height, res, self.region['center_easting'],
  427. self.region['center_northing']))
  428. ew = (self.width / 2) * res
  429. ns = (self.height / 2) * res
  430. self.region['n'] = self.region['center_northing'] + ns
  431. self.region['s'] = self.region['center_northing'] - ns
  432. self.region['e'] = self.region['center_easting'] + ew
  433. self.region['w'] = self.region['center_easting'] - ew
  434. # LL locations
  435. if self.projinfo['proj'] == 'll':
  436. self.region['n'] = min(self.region['n'], 90.0)
  437. self.region['s'] = max(self.region['s'], -90.0)
  438. def _writeEnvFile(self, data):
  439. """!Write display-related variable to the file (used for
  440. standalone app)
  441. """
  442. if not self.envfile:
  443. return
  444. try:
  445. fd = open(self.envfile, "r")
  446. for line in fd.readlines():
  447. key, value = line.split('=')
  448. if key not in data.keys():
  449. data[key] = value
  450. fd.close()
  451. fd = open(self.envfile, "w")
  452. for k, v in data.iteritems():
  453. fd.write('%s=%s\n' % (k.strip(), str(v).strip()))
  454. except IOError, e:
  455. grass.warning(_("Unable to open file '%(file)s' for writting. Details: %(det)s") % \
  456. { 'cmd' : self.envfile, 'det' : e })
  457. return
  458. fd.close()
  459. def ChangeMapSize(self, (width, height)):
  460. """!Change size of rendered map.
  461. @param width,height map size
  462. """
  463. try:
  464. self.width = int(width)
  465. self.height = int(height)
  466. except:
  467. self.width = 640
  468. self.height = 480
  469. Debug.msg(2, "Map.ChangeMapSize(): width=%d, height=%d" % \
  470. (self.width, self.height))
  471. self._writeEnvFile({'GRASS_WIDTH' : self.width,
  472. 'GRASS_HEIGHT' : self.height})
  473. def GetRegion(self, rast = [], zoom = False, vect = [], regionName = None,
  474. n = None, s = None, e = None, w = None, default = False,
  475. update = False):
  476. """!Get region settings (g.region -upgc)
  477. Optionally extent, raster or vector map layer can be given.
  478. @param rast list of raster maps
  479. @param zoom zoom to raster map (ignore NULLs)
  480. @param vect list of vector maps
  481. @param regionName named region or None
  482. @param n,s,e,w force extent
  483. @param default force default region settings
  484. @param update if True update current display region settings
  485. @return region settings as directory, e.g. {
  486. 'n':'4928010', 's':'4913700', 'w':'589980',...}
  487. @see GetCurrentRegion()
  488. """
  489. region = {}
  490. tmpreg = os.getenv("GRASS_REGION")
  491. if tmpreg:
  492. del os.environ["GRASS_REGION"]
  493. # use external gisrc if defined
  494. gisrc_orig = os.getenv("GISRC")
  495. if self.gisrc:
  496. os.environ["GISRC"] = self.gisrc
  497. # do not update & shell style output
  498. cmd = {}
  499. cmd['flags'] = 'ugpc'
  500. if default:
  501. cmd['flags'] += 'd'
  502. if regionName:
  503. cmd['region'] = regionName
  504. if n:
  505. cmd['n'] = n
  506. if s:
  507. cmd['s'] = s
  508. if e:
  509. cmd['e'] = e
  510. if w:
  511. cmd['w'] = w
  512. if rast:
  513. if zoom:
  514. cmd['zoom'] = rast[0]
  515. else:
  516. cmd['rast'] = ','.join(rast)
  517. if vect:
  518. cmd['vect'] = ','.join(vect)
  519. ret, reg, msg = RunCommand('g.region',
  520. read = True,
  521. getErrorMsg = True,
  522. **cmd)
  523. if ret != 0:
  524. if rast:
  525. message = _("Unable to zoom to raster map <%s>.") % rast[0] + \
  526. "\n\n" + _("Details:") + " %s" % msg
  527. elif vect:
  528. message = _("Unable to zoom to vector map <%s>.") % vect[0] + \
  529. "\n\n" + _("Details:") + " %s" % msg
  530. else:
  531. message = _("Unable to get current geographic extent. "
  532. "Force quiting wxGUI. Please manually run g.region to "
  533. "fix the problem.")
  534. GError(message)
  535. return self.region
  536. for r in reg.splitlines():
  537. key, val = r.split("=", 1)
  538. try:
  539. region[key] = float(val)
  540. except ValueError:
  541. region[key] = val
  542. # back to original gisrc
  543. if self.gisrc:
  544. os.environ["GISRC"] = gisrc_orig
  545. # restore region
  546. if tmpreg:
  547. os.environ["GRASS_REGION"] = tmpreg
  548. Debug.msg (3, "Map.GetRegion(): %s" % region)
  549. if update:
  550. self.region = region
  551. return region
  552. def GetCurrentRegion(self):
  553. """!Get current display region settings
  554. @see GetRegion()
  555. """
  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', 'cells'):
  570. region[key] = compRegion[key]
  571. else:
  572. # adjust region settings to match monitor
  573. region = self.AdjustRegion()
  574. # read values from wind file
  575. try:
  576. for key in self.wind.keys():
  577. if key == 'north':
  578. grass_region += "north: %s; " % \
  579. (region['n'])
  580. continue
  581. elif key == "south":
  582. grass_region += "south: %s; " % \
  583. (region['s'])
  584. continue
  585. elif key == "east":
  586. grass_region += "east: %s; " % \
  587. (region['e'])
  588. continue
  589. elif key == "west":
  590. grass_region += "west: %s; " % \
  591. (region['w'])
  592. continue
  593. elif key == "e-w resol":
  594. grass_region += "e-w resol: %f; " % \
  595. (region['ewres'])
  596. continue
  597. elif key == "n-s resol":
  598. grass_region += "n-s resol: %f; " % \
  599. (region['nsres'])
  600. continue
  601. elif key == "cols":
  602. if windres:
  603. continue
  604. grass_region += 'cols: %d; ' % \
  605. region['cols']
  606. continue
  607. elif key == "rows":
  608. if windres:
  609. continue
  610. grass_region += 'rows: %d; ' % \
  611. region['rows']
  612. continue
  613. else:
  614. grass_region += key + ": " + self.wind[key] + "; "
  615. Debug.msg (3, "Map.SetRegion(): %s" % grass_region)
  616. return grass_region
  617. except:
  618. return None
  619. def GetListOfLayers(self, l_type = None, l_mapset = None, l_name = None,
  620. l_active = None, l_hidden = None):
  621. """!Returns list of layers of selected properties or list of
  622. all layers.
  623. @param l_type layer type, e.g. raster/vector/wms/overlay (value or tuple of values)
  624. @param l_mapset all layers from given mapset (only for maplayers)
  625. @param l_name all layers with given name
  626. @param l_active only layers with 'active' attribute set to True or False
  627. @param l_hidden only layers with 'hidden' attribute set to True or False
  628. @return list of selected layers
  629. """
  630. selected = []
  631. if type(l_type) == types.StringType:
  632. one_type = True
  633. else:
  634. one_type = False
  635. if one_type and l_type == 'overlay':
  636. llist = self.overlays
  637. else:
  638. llist = self.layers
  639. # ["raster", "vector", "wms", ... ]
  640. for layer in llist:
  641. # specified type only
  642. if l_type != None:
  643. if one_type and layer.type != l_type:
  644. continue
  645. elif not one_type and layer.type not in l_type:
  646. continue
  647. # mapset
  648. if (l_mapset != None and l_type != 'overlay') and \
  649. layer.GetMapset() != l_mapset:
  650. continue
  651. # name
  652. if l_name != None and layer.name != l_name:
  653. continue
  654. # hidden and active layers
  655. if l_active != None and \
  656. l_hidden != None:
  657. if layer.active == l_active and \
  658. layer.hidden == l_hidden:
  659. selected.append(layer)
  660. # active layers
  661. elif l_active != None:
  662. if layer.active == l_active:
  663. selected.append(layer)
  664. # hidden layers
  665. elif l_hidden != None:
  666. if layer.hidden == l_hidden:
  667. selected.append(layer)
  668. # all layers
  669. else:
  670. selected.append(layer)
  671. Debug.msg (3, "Map.GetListOfLayers(): numberof=%d" % len(selected))
  672. return selected
  673. def _renderLayers(self, force = False, mapWindow = None, overlaysOnly = False):
  674. maps = list()
  675. masks = list()
  676. opacities = list()
  677. # render map layers
  678. ilayer = 1
  679. if overlaysOnly:
  680. layers = self.overlays
  681. else:
  682. layers = self.layers + self.overlays
  683. for layer in layers:
  684. # skip non-active map layers
  685. if not layer or not layer.active:
  686. continue
  687. # render
  688. if force or layer.force_render:
  689. if not layer.Render():
  690. continue
  691. if mapWindow:
  692. # update progress bar
  693. ### wx.SafeYield(mapWindow)
  694. event = wxUpdateProgressBar(value = ilayer)
  695. wx.PostEvent(mapWindow, event)
  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 GetLayersFromCmdFile(self):
  705. """!Get list of map layers from cmdfile
  706. """
  707. if not self.cmdfile:
  708. return
  709. nlayers = 0
  710. try:
  711. fd = open(self.cmdfile, 'r')
  712. for line in fd.readlines():
  713. cmd = utils.split(line.strip())
  714. ltype = None
  715. if cmd[0] == 'd.rast':
  716. ltype = 'raster'
  717. elif cmd[0] == 'd.vect':
  718. ltype = 'vector'
  719. name = utils.GetLayerNameFromCmd(cmd, fullyQualified = True,
  720. layerType = ltype)[0]
  721. self.AddLayer(type = ltype, command = cmd, l_active = False, name = name)
  722. nlayers += 1
  723. except IOError, e:
  724. grass.warning(_("Unable to read cmdfile '%(cmd)s'. Details: %(det)s") % \
  725. { 'cmd' : self.cmdfile, 'det' : e })
  726. return
  727. fd.close()
  728. Debug.msg(1, "Map.GetLayersFromCmdFile(): cmdfile=%s" % self.cmdfile)
  729. Debug.msg(1, " nlayers=%d" % nlayers)
  730. def _parseCmdFile(self):
  731. """!Parse cmd file for standalone application
  732. """
  733. nlayers = 0
  734. try:
  735. fd = open(self.cmdfile, 'r')
  736. grass.try_remove(self.mapfile)
  737. cmdLines = fd.readlines()
  738. RunCommand('g.gisenv',
  739. set = 'MONITOR_%s_CMDFILE=' % self.monitor)
  740. for cmd in cmdLines:
  741. cmdStr = utils.split(cmd.strip())
  742. cmd = utils.CmdToTuple(cmdStr)
  743. RunCommand(cmd[0], **cmd[1])
  744. nlayers += 1
  745. RunCommand('g.gisenv',
  746. set = 'MONITOR_%s_CMDFILE=%s' % (self.monitor, self.cmdfile))
  747. except IOError, e:
  748. grass.warning(_("Unable to read cmdfile '%(cmd)s'. Details: %(det)s") % \
  749. { 'cmd' : self.cmdfile, 'det' : e })
  750. return
  751. fd.close()
  752. Debug.msg(1, "Map.__parseCmdFile(): cmdfile=%s" % self.cmdfile)
  753. Debug.msg(1, " nlayers=%d" % nlayers)
  754. return nlayers
  755. def _renderCmdFile(self, force, windres):
  756. if not force:
  757. return ([self.mapfileCmd],
  758. [self.maskfileCmd],
  759. ['1.0'])
  760. region = os.environ["GRASS_REGION"] = self.SetRegion(windres)
  761. self._writeEnvFile({'GRASS_REGION' : region})
  762. currMon = grass.gisenv()['MONITOR']
  763. if currMon != self.monitor:
  764. RunCommand('g.gisenv',
  765. set = 'MONITOR=%s' % self.monitor)
  766. grass.try_remove(self.mapfileCmd) # GRASS_PNG_READ is TRUE
  767. nlayers = self._parseCmdFile()
  768. if self.overlays:
  769. RunCommand('g.gisenv',
  770. unset = 'MONITOR') # GRASS_RENDER_IMMEDIATE doesn't like monitors
  771. driver = UserSettings.Get(group = 'display', key = 'driver', subkey = 'type')
  772. if driver == 'png':
  773. os.environ["GRASS_RENDER_IMMEDIATE"] = "png"
  774. else:
  775. os.environ["GRASS_RENDER_IMMEDIATE"] = "cairo"
  776. self._renderLayers(overlaysOnly = True)
  777. del os.environ["GRASS_RENDER_IMMEDIATE"]
  778. RunCommand('g.gisenv',
  779. set = 'MONITOR=%s' % currMon)
  780. if currMon != self.monitor:
  781. RunCommand('g.gisenv',
  782. set = 'MONITOR=%s' % currMon)
  783. if nlayers > 0:
  784. return ([self.mapfileCmd],
  785. [self.maskfileCmd],
  786. ['1.0'])
  787. else:
  788. return ([], [], [])
  789. def Render(self, force = False, mapWindow = None, windres = False):
  790. """!Creates final image composite
  791. This function can conditionaly use high-level tools, which
  792. should be avaliable in wxPython library
  793. @param force force rendering
  794. @param reference for MapFrame instance (for progress bar)
  795. @param windres use region resolution (True) otherwise display resolution
  796. @return name of file with rendered image or None
  797. """
  798. wx.BeginBusyCursor()
  799. # use external gisrc if defined
  800. gisrc_orig = os.getenv("GISRC")
  801. if self.gisrc:
  802. os.environ["GISRC"] = self.gisrc
  803. tmp_region = os.getenv("GRASS_REGION")
  804. os.environ["GRASS_REGION"] = self.SetRegion(windres)
  805. os.environ["GRASS_WIDTH"] = str(self.width)
  806. os.environ["GRASS_HEIGHT"] = str(self.height)
  807. driver = UserSettings.Get(group = 'display', key = 'driver', subkey = 'type')
  808. if driver == 'png':
  809. os.environ["GRASS_RENDER_IMMEDIATE"] = "png"
  810. else:
  811. os.environ["GRASS_RENDER_IMMEDIATE"] = "cairo"
  812. if self.cmdfile:
  813. maps, masks, opacities = self._renderCmdFile(force, windres)
  814. else:
  815. maps, masks, opacities = self._renderLayers(force, mapWindow)
  816. # ugly hack for MSYS
  817. if sys.platform != 'win32':
  818. mapstr = ",".join(maps)
  819. maskstr = ",".join(masks)
  820. mapoutstr = self.mapfile
  821. else:
  822. mapstr = ""
  823. for item in maps:
  824. mapstr += item.replace('\\', '/')
  825. mapstr = mapstr.rstrip(',')
  826. maskstr = ""
  827. for item in masks:
  828. maskstr += item.replace('\\', '/')
  829. maskstr = maskstr.rstrip(',')
  830. mapoutstr = self.mapfile.replace('\\', '/')
  831. # run g.pngcomp to get composite image
  832. bgcolor = ':'.join(map(str, UserSettings.Get(group = 'display', key = 'bgcolor',
  833. subkey = 'color')))
  834. if maps:
  835. ret, msg = RunCommand('g.pnmcomp',
  836. getErrorMsg = True,
  837. overwrite = True,
  838. input = '%s' % ",".join(maps),
  839. mask = '%s' % ",".join(masks),
  840. opacity = '%s' % ",".join(opacities),
  841. bgcolor = bgcolor,
  842. width = self.width,
  843. height = self.height,
  844. output = self.mapfile)
  845. if ret != 0:
  846. print >> sys.stderr, _("ERROR: Rendering failed. Details: %s") % msg
  847. wx.EndBusyCursor()
  848. return None
  849. Debug.msg (3, "Map.Render() force=%s file=%s" % (force, self.mapfile))
  850. # back to original region
  851. if tmp_region:
  852. os.environ["GRASS_REGION"] = tmp_region
  853. else:
  854. del os.environ["GRASS_REGION"]
  855. # back to original gisrc
  856. if self.gisrc:
  857. os.environ["GISRC"] = gisrc_orig
  858. wx.EndBusyCursor()
  859. if not maps:
  860. return None
  861. return self.mapfile
  862. def AddLayer(self, type, command, name = None,
  863. l_active = True, l_hidden = False, l_opacity = 1.0, l_render = False,
  864. pos = -1):
  865. """!Adds generic map layer to list of layers
  866. @param type layer type ('raster', 'vector', etc.)
  867. @param command GRASS command given as list
  868. @param name layer name
  869. @param l_active layer render only if True
  870. @param l_hidden layer not displayed in layer tree if True
  871. @param l_opacity opacity level range from 0(transparent) - 1(not transparent)
  872. @param l_render render an image if True
  873. @param pos position in layer list (-1 for append)
  874. @return new layer on success
  875. @return None on failure
  876. """
  877. wx.BeginBusyCursor()
  878. # l_opacity must be <0;1>
  879. if l_opacity < 0:
  880. l_opacity = 0
  881. elif l_opacity > 1:
  882. l_opacity = 1
  883. layer = MapLayer(type = type, name = name, cmd = command,
  884. active = l_active, hidden = l_hidden, opacity = l_opacity)
  885. # add maplayer to the list of layers
  886. if pos > -1:
  887. self.layers.insert(pos, layer)
  888. else:
  889. self.layers.append(layer)
  890. Debug.msg (3, "Map.AddLayer(): layer=%s" % layer.name)
  891. if l_render:
  892. if not layer.Render():
  893. raise GException(_("Unable to render map layer <%s>.") % name)
  894. wx.EndBusyCursor()
  895. return layer
  896. def DeleteAllLayers(self, overlay = False):
  897. """!Delete all layers
  898. @param overlay True to delete also overlayes
  899. """
  900. self.layers = []
  901. if overlay:
  902. self.overlays = []
  903. def DeleteLayer(self, layer, overlay = False):
  904. """!Removes layer from list of layers
  905. @param layer layer instance in layer tree
  906. @param overlay delete overlay (use self.DeleteOverlay() instead)
  907. @return removed layer on success or None
  908. """
  909. Debug.msg (3, "Map.DeleteLayer(): name=%s" % layer.name)
  910. if overlay:
  911. list = self.overlays
  912. else:
  913. list = self.layers
  914. if layer in list:
  915. if layer.mapfile:
  916. base = os.path.split(layer.mapfile)[0]
  917. mapfile = os.path.split(layer.mapfile)[1]
  918. tempbase = mapfile.split('.')[0]
  919. if base == '' or tempbase == '':
  920. return None
  921. basefile = os.path.join(base, tempbase) + r'.*'
  922. for f in glob.glob(basefile):
  923. os.remove(f)
  924. list.remove(layer)
  925. return layer
  926. return None
  927. def ReorderLayers(self, layerList):
  928. """!Reorder list to match layer tree
  929. @param layerList list of layers
  930. """
  931. self.layers = layerList
  932. layerNameList = ""
  933. for layer in self.layers:
  934. if layer.name:
  935. layerNameList += layer.name + ','
  936. Debug.msg (4, "Map.ReoderLayers(): layers=%s" % \
  937. (layerNameList))
  938. def ChangeLayer(self, layer, render = False, **kargs):
  939. """!Change map layer properties
  940. @param layer map layer instance
  941. @param type layer type ('raster', 'vector', etc.)
  942. @param command GRASS command given as list
  943. @param name layer name
  944. @param active layer render only if True
  945. @param hidden layer not displayed in layer tree if True
  946. @param opacity opacity level range from 0(transparent) - 1(not transparent)
  947. @param render render an image if True
  948. """
  949. Debug.msg (3, "Map.ChangeLayer(): layer=%s" % layer.name)
  950. if 'type' in kargs:
  951. layer.SetType(kargs['type']) # check type
  952. if 'command' in kargs:
  953. layer.SetCmd(kargs['command'])
  954. if 'name' in kargs:
  955. layer.SetName(kargs['name'])
  956. if 'active' in kargs:
  957. layer.SetActive(kargs['active'])
  958. if 'hidden' in kargs:
  959. layer.SetHidden(kargs['hidden'])
  960. if 'opacity' in kargs:
  961. layer.SetOpacity(kargs['opacity'])
  962. if render and not layer.Render():
  963. raise GException(_("Unable to render map layer <%s>.") %
  964. name)
  965. return layer
  966. def ChangeOpacity(self, layer, l_opacity):
  967. """!Changes opacity value of map layer
  968. @param layer layer instance in layer tree
  969. @param l_opacity opacity level <0;1>
  970. """
  971. # l_opacity must be <0;1>
  972. if l_opacity < 0: l_opacity = 0
  973. elif l_opacity > 1: l_opacity = 1
  974. layer.opacity = l_opacity
  975. Debug.msg (3, "Map.ChangeOpacity(): layer=%s, opacity=%f" % \
  976. (layer.name, layer.opacity))
  977. def ChangeLayerActive(self, layer, active):
  978. """!Enable or disable map layer
  979. @param layer layer instance in layer tree
  980. @param active to be rendered (True)
  981. """
  982. layer.active = active
  983. Debug.msg (3, "Map.ChangeLayerActive(): name='%s' -> active=%d" % \
  984. (layer.name, layer.active))
  985. def ChangeLayerName (self, layer, name):
  986. """!Change name of the layer
  987. @param layer layer instance in layer tree
  988. @param name layer name to set up
  989. """
  990. Debug.msg (3, "Map.ChangeLayerName(): from=%s to=%s" % \
  991. (layer.name, name))
  992. layer.name = name
  993. def RemoveLayer(self, name = None, id = None):
  994. """!Removes layer from layer list
  995. Layer is defined by name@mapset or id.
  996. @param name layer name (must be unique)
  997. @param id layer index in layer list
  998. @return removed layer on success
  999. @return None on failure
  1000. """
  1001. # delete by name
  1002. if name:
  1003. retlayer = None
  1004. for layer in self.layers:
  1005. if layer.name == name:
  1006. retlayer = layer
  1007. os.remove(layer.mapfile)
  1008. os.remove(layer.maskfile)
  1009. self.layers.remove(layer)
  1010. return layer
  1011. # del by id
  1012. elif id != None:
  1013. return self.layers.pop(id)
  1014. return None
  1015. def GetLayerIndex(self, layer, overlay = False):
  1016. """!Get index of layer in layer list.
  1017. @param layer layer instace in layer tree
  1018. @param overlay use list of overlays instead
  1019. @return layer index
  1020. @return -1 if layer not found
  1021. """
  1022. if overlay:
  1023. list = self.overlay
  1024. else:
  1025. list = self.layers
  1026. if layer in list:
  1027. return list.index(layer)
  1028. return -1
  1029. def AddOverlay(self, id, type, command,
  1030. l_active = True, l_hidden = True, l_opacity = 1.0, l_render = False):
  1031. """!Adds overlay (grid, barscale, legend, etc.) to list of
  1032. overlays
  1033. @param id overlay id (PseudoDC)
  1034. @param type overlay type (barscale, legend)
  1035. @param command GRASS command to render overlay
  1036. @param l_active overlay activated (True) or disabled (False)
  1037. @param l_hidden overlay is not shown in layer tree (if True)
  1038. @param l_render render an image (if True)
  1039. @return new layer on success
  1040. @retutn None on failure
  1041. """
  1042. Debug.msg (2, "Map.AddOverlay(): cmd=%s, render=%d" % (command, l_render))
  1043. overlay = Overlay(id = id, type = type, cmd = command,
  1044. active = l_active, hidden = l_hidden, opacity = l_opacity)
  1045. # add maplayer to the list of layers
  1046. self.overlays.append(overlay)
  1047. if l_render and command != '' and not overlay.Render():
  1048. raise GException(_("Unable render overlay <%s>.") %
  1049. name)
  1050. return self.overlays[-1]
  1051. def ChangeOverlay(self, id, render = False, **kargs):
  1052. """!Change overlay properities
  1053. Add new overlay if overlay with 'id' doesn't exist.
  1054. @param id overlay id (PseudoDC)
  1055. @param type overlay type (barscale, legend)
  1056. @param command GRASS command to render overlay
  1057. @param l_active overlay activated (True) or disabled (False)
  1058. @param l_hidden overlay is not shown in layer tree (if True)
  1059. @param l_render render an image (if True)
  1060. @return new layer on success
  1061. """
  1062. overlay = self.GetOverlay(id, list = False)
  1063. if overlay is None:
  1064. overlay = Overlay(id, type = None, cmd = None)
  1065. if 'type' in kargs:
  1066. overlay.SetName(kargs['type']) # type -> overlay
  1067. if 'command' in kargs:
  1068. overlay.SetCmd(kargs['command'])
  1069. if 'active' in kargs:
  1070. overlay.SetActive(kargs['active'])
  1071. if 'hidden' in kargs:
  1072. overlay.SetHidden(kargs['hidden'])
  1073. if 'opacity' in kargs:
  1074. overlay.SetOpacity(kargs['opacity'])
  1075. if render and overlay.GetCmd() != [] and not overlay.Render():
  1076. raise GException(_("Unable render overlay <%s>") %
  1077. name)
  1078. return overlay
  1079. def GetOverlay(self, id, list = False):
  1080. """!Return overlay(s) with 'id'
  1081. @param id overlay id
  1082. @param list return list of overlays of True
  1083. otherwise suppose 'id' to be unique
  1084. @return list of overlays (list=True)
  1085. @return overlay (list=False)
  1086. @retur None (list=False) if no overlay or more overlays found
  1087. """
  1088. ovl = []
  1089. for overlay in self.overlays:
  1090. if overlay.id == id:
  1091. ovl.append(overlay)
  1092. if not list:
  1093. if len(ovl) != 1:
  1094. return None
  1095. else:
  1096. return ovl[0]
  1097. return ovl
  1098. def DeleteOverlay(self, overlay):
  1099. """!Delete overlay
  1100. @param overlay overlay layer
  1101. @return removed overlay on success or None
  1102. """
  1103. return self.DeleteLayer(overlay, overlay = True)
  1104. def _clean(self, llist):
  1105. for layer in llist:
  1106. if layer.maskfile:
  1107. grass.try_remove(layer.maskfile)
  1108. if layer.mapfile:
  1109. grass.try_remove(layer.mapfile)
  1110. llist.remove(layer)
  1111. def Clean(self):
  1112. """!Clean layer stack - go trough all layers and remove them
  1113. from layer list.
  1114. Removes also mapfile and maskfile.
  1115. """
  1116. self._clean(self.layers)
  1117. self._clean(self.overlays)
  1118. def ReverseListOfLayers(self):
  1119. """!Reverse list of layers"""
  1120. return self.layers.reverse()
  1121. def RenderOverlays(self, force):
  1122. """!Render overlays only (for nviz)"""
  1123. for layer in self.overlays:
  1124. if force or layer.force_render:
  1125. layer.Render()
  1126. if __name__ == "__main__":
  1127. import gettext
  1128. gettext.install('grasswxpy', os.path.join(os.getenv("GISBASE"), 'locale'), unicode = True)
  1129. Map = Map()
  1130. Map.GetRegion(update = True)
  1131. Map.AddLayer(type = "raster",
  1132. name = "elevation",
  1133. command = ["d.rast", "map=elevation@PERMANENT"],
  1134. l_opacity = .7)
  1135. Map.AddLayer(type = "vector",
  1136. name = "roadsmajor",
  1137. command = ["d.vect", "map=roadsmajor@PERMANENT", "color=red", "width=3", "type=line"])
  1138. image = Map.Render(force = True)
  1139. if image:
  1140. grass.call(["display", image])