render.py 40 KB

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