render.py 46 KB

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