render.py 46 KB

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