render.py 46 KB

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