render.py 46 KB

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