render.py 47 KB

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