render.py 37 KB

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