workspace.py 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825
  1. """
  2. @package workspace
  3. @brief Open/save workspace definition file
  4. Classes:
  5. - ProcessWorkspaceFile
  6. - WriteWorkspaceFile
  7. - ProcessGrcFile
  8. (C) 2007-2008 by the GRASS Development Team
  9. This program is free software under the GNU General Public
  10. License (>=v2). Read the file COPYING that comes with GRASS
  11. for details.
  12. @author Martin Landa <landa.martin gmail.com>
  13. @date 2007-2008
  14. """
  15. import os
  16. import wx
  17. ### for gxw (workspace file) parsering
  18. # xmlproc not available on Mac OS
  19. # from xml.parsers.xmlproc import xmlproc
  20. # from xml.parsers.xmlproc import xmlval
  21. # from xml.parsers.xmlproc import xmldtd
  22. import xml.sax
  23. import xml.sax.handler
  24. HandlerBase=xml.sax.handler.ContentHandler
  25. from xml.sax import make_parser
  26. from preferences import globalSettings as UserSettings
  27. class ProcessWorkspaceFile(HandlerBase):
  28. """
  29. A SAX handler for the GXW XML file, as
  30. defined in grass-gxw.dtd.
  31. """
  32. def __init__(self):
  33. self.inTag = {}
  34. for tag in ('gxw', 'layer', 'task', 'parameter',
  35. 'flag', 'value', 'group', 'display',
  36. 'layer_manager',
  37. 'nviz',
  38. # surface
  39. 'surface', 'attribute', 'draw', 'resolution',
  40. 'wire_color', 'position', 'x', 'y', 'z',
  41. # vector lines
  42. 'vlines', 'color', 'width', 'mode',
  43. 'map', 'height',
  44. # vector points
  45. 'vpoints', 'size'):
  46. self.inTag[tag] = False
  47. #
  48. # layer manager properties
  49. #
  50. self.layerManager = {}
  51. self.layerManager['pos'] = None # window position
  52. self.layerManager['size'] = None # window size
  53. # list of mapdisplays
  54. self.displays = []
  55. # list of map layers
  56. self.layers = []
  57. self.cmd = []
  58. self.displayIndex = -1 # first display has index '0'
  59. def startElement(self, name, attrs):
  60. if name == 'display':
  61. self.displayIndex += 1
  62. # window position and size
  63. posAttr = attrs.get('dim', '')
  64. if posAttr:
  65. posVal = map(int, posAttr.split(','))
  66. try:
  67. pos = (posVal[0], posVal[1])
  68. size = (posVal[2], posVal[3])
  69. except:
  70. pos = None
  71. size = None
  72. else:
  73. pos = None
  74. size = None
  75. extentAttr = attrs.get('extent', '')
  76. if extentAttr:
  77. # w, s, e, n
  78. extent = map(float, extentAttr.split(','))
  79. else:
  80. extent = None
  81. self.displays.append({
  82. "render" : bool(int(attrs.get('render', "0"))),
  83. "mode" : int(attrs.get('mode', 0)),
  84. "showCompExtent" : bool(int(attrs.get('showCompExtent', "0"))),
  85. "pos" : pos,
  86. "size" : size,
  87. "extent" : extent})
  88. elif name == 'group':
  89. self.groupName = attrs.get('name', None)
  90. self.groupChecked = attrs.get('checked', None)
  91. self.layers.append({
  92. "type" : 'group',
  93. "name" : self.groupName,
  94. "checked" : int(self.groupChecked),
  95. "opacity" : None,
  96. "cmd" : None,
  97. "group" : self.inTag['group'],
  98. "display" : self.displayIndex,
  99. "nviz" : None})
  100. elif name == 'layer':
  101. self.layerType = attrs.get('type', None)
  102. self.layerName = attrs.get('name', None)
  103. self.layerChecked = attrs.get('checked', None)
  104. self.layerOpacity = attrs.get('opacity', None)
  105. self.layerSelected = False
  106. self.layerNviz = None
  107. self.cmd = []
  108. elif name == 'task':
  109. name = attrs.get('name', None)
  110. self.cmd.append(name)
  111. elif name == 'parameter':
  112. self.parameterName = attrs.get('name', None)
  113. elif name in ('value', 'x', 'y', 'z', 'map'):
  114. self.value = ''
  115. elif name == 'flag':
  116. name = attrs.get('name', None)
  117. self.cmd.append('-' + name)
  118. elif name == 'selected':
  119. if self.inTag['layer']:
  120. self.layerSelected = True;
  121. elif name == 'layer_manager':
  122. posAttr = attrs.get('dim', '')
  123. if posAttr:
  124. posVal = map(int, posAttr.split(','))
  125. try:
  126. self.layerManager['pos'] = (posVal[0], posVal[1])
  127. self.layerManager['size'] = (posVal[2], posVal[3])
  128. except:
  129. pass
  130. else:
  131. pass
  132. #
  133. # Nviz section
  134. #
  135. elif name == 'nviz':
  136. # init nviz layer properties
  137. self.layerNviz = {}
  138. if self.layerType == 'raster':
  139. self.layerNviz['surface'] = {}
  140. for sec in ('attribute', 'draw', 'mask', 'position'):
  141. self.layerNviz['surface'][sec] = {}
  142. elif self.layerType == 'vector':
  143. self.layerNviz['vector'] = {}
  144. for sec in ('lines', 'points'):
  145. self.layerNviz['vector'][sec] = {}
  146. elif name == 'attribute':
  147. if self.inTag['nviz'] and self.inTag['surface']:
  148. tagName = str(name)
  149. attrbName = str(attrs.get('name', ''))
  150. self.layerNviz['surface'][tagName][attrbName] = {}
  151. if attrs.get('map', '0') == '0':
  152. self.layerNviz['surface'][tagName][attrbName]['map'] = False
  153. else:
  154. self.layerNviz['surface'][tagName][attrbName]['map'] = True
  155. self.refAttribute = self.layerNviz['surface'][tagName][attrbName]
  156. elif name == 'draw':
  157. if self.inTag['nviz'] and self.inTag['surface']:
  158. tagName = str(name)
  159. self.layerNviz['surface'][tagName]['all'] = False
  160. self.layerNviz['surface'][tagName]['mode'] = {}
  161. self.layerNviz['surface'][tagName]['mode']['value'] = -1 # to be calculated
  162. self.layerNviz['surface'][tagName]['mode']['desc'] = {}
  163. self.layerNviz['surface'][tagName]['mode']['desc']['shading'] = \
  164. str(attrs.get('shading', ''))
  165. self.layerNviz['surface'][tagName]['mode']['desc']['style'] = \
  166. str(attrs.get('style', ''))
  167. self.layerNviz['surface'][tagName]['mode']['desc']['mode'] = \
  168. str(attrs.get('mode', ''))
  169. elif name == 'resolution':
  170. if self.inTag['nviz'] and self.inTag['surface']:
  171. self.resolutionType = str(attrs.get('type', ''))
  172. if not self.layerNviz['surface']['draw'].has_key(str(name)):
  173. self.layerNviz['surface']['draw'][str(name)] = {}
  174. elif name == 'position':
  175. if self.inTag['nviz'] and inTag['surface']:
  176. self.layerNviz['surface']['position'] = {}
  177. elif name == 'mode':
  178. if self.inTag['nviz']:
  179. if self.inTag['vlines']:
  180. self.layerNviz['vector']['lines']['mode'] = {}
  181. self.layerNviz['vector']['lines']['mode']['type'] = str(attrs.get('type', ''))
  182. self.layerNviz['vector']['lines']['mode']['surface'] = ''
  183. elif self.inTag['vpoints']:
  184. self.layerNviz['vector']['points']['mode'] = {}
  185. self.layerNviz['vector']['points']['mode']['type'] = str(attrs.get('type', ''))
  186. self.layerNviz['vector']['points']['mode']['surface'] = ''
  187. elif name == 'vpoints':
  188. if self.inTag['nviz']:
  189. marker = str(attrs.get('marker', ''))
  190. markerId = list(UserSettings.Get(group='nviz', key='vector',
  191. subkey=['points', 'marker'], internal=True)).index(marker)
  192. self.layerNviz['vector']['points']['marker'] = markerId
  193. self.inTag[name] = True
  194. def endElement(self, name):
  195. if name == 'group':
  196. self.groupName = self.groupChecked = None
  197. elif name == 'layer':
  198. self.layers.append({
  199. "type" : self.layerType,
  200. "name" : self.layerName,
  201. "checked" : int(self.layerChecked),
  202. "opacity" : 1.0,
  203. "cmd" : None,
  204. "group" : self.inTag['group'],
  205. "display" : self.displayIndex,
  206. "selected" : self.layerSelected,
  207. "nviz" : self.layerNviz})
  208. if self.layerOpacity:
  209. self.layers[-1]["opacity"] = float(self.layerOpacity)
  210. if self.cmd:
  211. self.layers[-1]["cmd"] = self.cmd
  212. self.layerType = self.layerName = self.Checked = \
  213. self.Opacity = self.cmd = None
  214. elif name == 'parameter':
  215. self.cmd.append('%s=%s' % (self.parameterName, self.value))
  216. self.parameterName = self.value = None
  217. #
  218. # Nviz section
  219. #
  220. elif name == 'attribute':
  221. if self.inTag['nviz'] and self.inTag['surface']:
  222. try:
  223. self.refAttribute['value'] = int(self.value)
  224. except ValueError:
  225. try:
  226. self.refAttribute['value'] = float(self.value)
  227. except ValueError:
  228. self.refAttribute['value'] = str(self.value)
  229. elif name == 'resolution':
  230. if self.inTag['nviz'] and self.inTag['surface']:
  231. self.layerNviz['surface']['draw']['resolution'][self.resolutionType] = int(self.value)
  232. del self.resolutionType
  233. elif name == 'wire_color':
  234. if self.inTag['nviz'] and self.inTag['surface']:
  235. self.layerNviz['surface']['draw']['wire-color'] = {}
  236. self.layerNviz['surface']['draw']['wire-color']['value'] = str(self.value)
  237. elif name == 'x':
  238. if self.inTag['nviz'] and self.inTag['surface']:
  239. self.layerNviz['surface']['position']['x'] = int(self.value)
  240. elif name == 'y':
  241. if self.inTag['nviz'] and self.inTag['surface']:
  242. self.layerNviz['surface']['position']['y'] = int(self.value)
  243. elif name == 'z':
  244. if self.inTag['nviz'] and self.inTag['surface']:
  245. self.layerNviz['surface']['position']['z'] = int(self.value)
  246. elif name == 'color':
  247. if self.inTag['nviz']:
  248. if self.inTag['vlines']:
  249. self.layerNviz['vector']['lines']['color'] = str(self.value)
  250. elif self.inTag['vpoints']:
  251. self.layerNviz['vector']['points']['color'] = str(self.value)
  252. elif name == 'width':
  253. if self.inTag['nviz']:
  254. if self.inTag['vlines']:
  255. self.layerNviz['vector']['lines']['width'] = int(self.value)
  256. elif self.inTag['vpoints']:
  257. self.layerNviz['vector']['points']['width'] = int(self.value)
  258. elif name == 'height':
  259. if self.inTag['nviz']:
  260. if self.inTag['vlines']:
  261. self.layerNviz['vector']['lines']['height'] = int(self.value)
  262. elif self.inTag['vpoints']:
  263. self.layerNviz['vector']['points']['height'] = int(self.value)
  264. elif name == 'size':
  265. if self.inTag['nviz'] and self.inTag['vpoints']:
  266. self.layerNviz['vector']['points']['size'] = int(self.value)
  267. elif name == 'map':
  268. if self.inTag['nviz']:
  269. if self.inTag['vlines']:
  270. self.layerNviz['vector']['lines']['mode']['surface'] = str(self.value)
  271. elif self.inTag['vpoints']:
  272. self.layerNviz['vector']['points']['mode']['surface'] = str(self.value)
  273. self.inTag[name] = False
  274. def characters(self, ch):
  275. self.my_characters(ch)
  276. def my_characters(self, ch):
  277. if self.inTag['value'] or \
  278. self.inTag['x'] or \
  279. self.inTag['y'] or \
  280. self.inTag['z'] or \
  281. self.inTag['map']:
  282. self.value += ch
  283. class WriteWorkspaceFile(object):
  284. """Generic class for writing workspace file"""
  285. def __init__(self, lmgr, file):
  286. self.file = file
  287. self.lmgr = lmgr
  288. self.indent = 0
  289. # write header
  290. self.file.write('<?xml version="1.0" encoding="UTF-8"?>\n')
  291. self.file.write('<!DOCTYPE gxw SYSTEM "grass-gxw.dtd">\n')
  292. self.file.write('%s<gxw>\n' % (' ' * self.indent))
  293. self.indent =+ 4
  294. # layer manager
  295. windowPos = self.lmgr.GetPosition()
  296. windowSize = self.lmgr.GetSize()
  297. file.write('%s<layer_manager dim="%d,%d,%d,%d">\n' % (' ' * self.indent,
  298. windowPos[0],
  299. windowPos[1],
  300. windowSize[0],
  301. windowSize[1]
  302. ))
  303. file.write('%s</layer_manager>\n' % (' ' * self.indent))
  304. # list of displays
  305. for page in range(0, self.lmgr.gm_cb.GetPageCount()):
  306. mapTree = self.lmgr.gm_cb.GetPage(page).maptree
  307. region = mapTree.Map.region
  308. displayPos = mapTree.mapdisplay.GetPosition()
  309. displaySize = mapTree.mapdisplay.GetSize()
  310. file.write('%s<display render="%d" '
  311. 'mode="%d" showCompExtent="%d" '
  312. 'dim="%d,%d,%d,%d" '
  313. 'extent="%f,%f,%f,%f">\n' % (' ' * self.indent,
  314. int(mapTree.mapdisplay.autoRender.IsChecked()),
  315. mapTree.mapdisplay.toggleStatus.GetSelection(),
  316. int(mapTree.mapdisplay.showRegion.IsChecked()),
  317. displayPos[0],
  318. displayPos[1],
  319. displaySize[0],
  320. displaySize[1],
  321. region['w'],
  322. region['s'],
  323. region['e'],
  324. region['n']
  325. ))
  326. # list of layers
  327. item = mapTree.GetFirstChild(mapTree.root)[0]
  328. self.__writeLayer(mapTree, item)
  329. file.write('%s</display>\n' % (' ' * self.indent))
  330. self.indent =- 4
  331. file.write('%s</gxw>\n' % (' ' * self.indent))
  332. def __writeLayer(self, mapTree, item):
  333. """Write bunch of layers to GRASS Workspace XML file"""
  334. self.indent += 4
  335. while item and item.IsOk():
  336. type = mapTree.GetPyData(item)[0]['type']
  337. if type != 'group':
  338. maplayer = mapTree.GetPyData(item)[0]['maplayer']
  339. else:
  340. maplayer = None
  341. checked = int(item.IsChecked())
  342. if type == 'command':
  343. cmd = mapTree.GetPyData(item)[0]['maplayer'].GetCmd(string=True)
  344. self.file.write('%s<layer type="%s" name="%s" checked="%d">\n' % \
  345. (' ' * self.indent, type, cmd, checked));
  346. self.file.write('%s</layer>\n' % (' ' * self.indent));
  347. elif type == 'group':
  348. name = mapTree.GetItemText(item)
  349. self.file.write('%s<group name="%s" checked="%d">\n' % \
  350. (' ' * self.indent, name, checked));
  351. self.indent += 4
  352. subItem = mapTree.GetFirstChild(item)[0]
  353. self.__writeLayer(mapTree, subItem)
  354. self.indent -= 4
  355. self.file.write('%s</group>\n' % (' ' * self.indent));
  356. else:
  357. cmd = mapTree.GetPyData(item)[0]['maplayer'].GetCmd(string=False)
  358. name = mapTree.GetItemText(item)
  359. # remove 'opacity' part
  360. if '(opacity' in name:
  361. name = name.split('(', -1)[0].strip()
  362. opacity = maplayer.GetOpacity(float=True)
  363. self.file.write('%s<layer type="%s" name="%s" checked="%d" opacity="%f">\n' % \
  364. (' ' * self.indent, type, name, checked, opacity));
  365. self.indent += 4
  366. # selected ?
  367. if item == mapTree.layer_selected:
  368. self.file.write('%s<selected />\n' % (' ' * self.indent))
  369. # layer properties
  370. self.file.write('%s<task name="%s">\n' % (' ' * self.indent, cmd[0]))
  371. self.indent += 4
  372. for option in cmd[1:]:
  373. if option[0] == '-': # flag
  374. self.file.write('%s<flag name="%s" />\n' %
  375. (' ' * self.indent, option[1]))
  376. else: # parameter
  377. key, value = option.split('=', 1)
  378. self.file.write('%s<parameter name="%s">\n' %
  379. (' ' * self.indent, key))
  380. self.indent += 4
  381. self.file.write('%s<value>%s</value>\n' %
  382. (' ' * self.indent, value))
  383. self.indent -= 4
  384. self.file.write('%s</parameter>\n' % (' ' * self.indent));
  385. self.indent -= 4
  386. self.file.write('%s</task>\n' % (' ' * self.indent));
  387. nviz = mapTree.GetPyData(item)[0]['nviz']
  388. if nviz:
  389. self.file.write('%s<nviz>\n' % (' ' * self.indent));
  390. if maplayer.type == 'raster':
  391. self.__writeNvizSurface(nviz['surface'])
  392. elif maplayer.type == 'vector':
  393. self.__writeNvizVector(nviz['vector'])
  394. self.file.write('%s</nviz>\n' % (' ' * self.indent));
  395. self.indent -= 4
  396. self.file.write('%s</layer>\n' % (' ' * self.indent));
  397. item = mapTree.GetNextSibling(item)
  398. self.indent -= 4
  399. def __writeNvizSurface(self, data):
  400. """Save Nviz raster layer properties to workspace
  401. @param data Nviz layer properties
  402. """
  403. if not data.has_key('object'): # skip disabled
  404. return
  405. self.indent += 4
  406. self.file.write('%s<surface>\n' % (' ' * self.indent))
  407. self.indent += 4
  408. for attrb in data.iterkeys():
  409. if len(data[attrb]) < 1: # skip empty attributes
  410. continue
  411. if attrb == 'object':
  412. continue
  413. for name in data[attrb].iterkeys():
  414. # surface attribute
  415. if attrb == 'attribute':
  416. self.file.write('%s<%s name="%s" map="%d">\n' % \
  417. (' ' * self.indent, attrb, name, data[attrb][name]['map']))
  418. self.indent += 4
  419. self.file.write('%s<value>%s</value>\n' % (' ' * self.indent, data[attrb][name]['value']))
  420. self.indent -= 4
  421. # end tag
  422. self.file.write('%s</%s>\n' % (' ' * self.indent, attrb))
  423. # draw mode
  424. if attrb == 'draw':
  425. self.file.write('%s<%s' %(' ' * self.indent, attrb))
  426. if data[attrb].has_key('mode'):
  427. for tag, value in data[attrb]['mode']['desc'].iteritems():
  428. self.file.write(' %s="%s"' % (tag, value))
  429. self.file.write('>\n') # <draw ...>
  430. if data[attrb].has_key('resolution'):
  431. self.indent += 4
  432. for type in ('coarse', 'fine'):
  433. self.file.write('%s<resolution type="%s">\n' % (' ' * self.indent, type))
  434. self.indent += 4
  435. self.file.write('%s<value>%d</value>\n' % (' ' * self.indent,
  436. data[attrb]['resolution'][type]))
  437. self.indent -= 4
  438. self.file.write('%s</resolution>\n' % (' ' * self.indent))
  439. if data[attrb].has_key('wire-color'):
  440. self.file.write('%s<wire_color>\n' % (' ' * self.indent))
  441. self.indent += 4
  442. self.file.write('%s<value>%s</value>\n' % (' ' * self.indent,
  443. data[attrb]['wire-color']['value']))
  444. self.indent -= 4
  445. self.file.write('%s</wire_color>\n' % (' ' * self.indent))
  446. self.indent -= 4
  447. # position
  448. elif attrb == 'position':
  449. self.file.write('%s<%s>\n' %(' ' * self.indent, attrb))
  450. i = 0
  451. for tag in ('x', 'y', 'z'):
  452. self.indent += 4
  453. self.file.write('%s<%s>%d</%s>\n' % (' ' * self.indent, tag,
  454. data[attrb][tag], tag))
  455. i += 1
  456. self.indent -= 4
  457. if attrb != 'attribute':
  458. # end tag
  459. self.file.write('%s</%s>\n' % (' ' * self.indent, attrb))
  460. self.indent -= 4
  461. self.file.write('%s</surface>\n' % (' ' * self.indent))
  462. self.indent -= 4
  463. def __writeNvizVector(self, data):
  464. """Save Nviz vector layer properties (lines/points) to workspace
  465. @param data Nviz layer properties
  466. """
  467. self.indent += 4
  468. for attrb in data.iterkeys():
  469. if len(data[attrb]) < 1: # skip empty attributes
  470. continue
  471. if not data[attrb].has_key('object'): # skip disabled
  472. continue
  473. if attrb == 'lines':
  474. self.file.write('%s<v%s>\n' % (' ' * self.indent, attrb))
  475. elif attrb == 'points':
  476. markerId = data[attrb]['marker']
  477. marker = UserSettings.Get(group='nviz', key='vector',
  478. subkey=['points', 'marker'], internal=True)[markerId]
  479. self.file.write('%s<v%s marker="%s">\n' % (' ' * self.indent,
  480. attrb,
  481. marker))
  482. self.indent += 4
  483. for name in data[attrb].iterkeys():
  484. if name in ('object', 'marker'):
  485. continue
  486. if name == 'mode':
  487. self.file.write('%s<%s type="%s">\n' % (' ' * self.indent, name,
  488. data[attrb][name]['type']))
  489. if data[attrb][name]['type'] == 'surface':
  490. self.indent += 4
  491. self.file.write('%s<map>%s</map>\n' % (' ' * self.indent,
  492. data[attrb][name]['surface']))
  493. self.indent -= 4
  494. self.file.write('%s</%s>\n' % ((' ' * self.indent, name)))
  495. else:
  496. self.file.write('%s<%s>\n' % (' ' * self.indent, name))
  497. self.indent += 4
  498. self.file.write('%s<value>%s</value>\n' % (' ' * self.indent, data[attrb][name]))
  499. self.indent -= 4
  500. self.file.write('%s</%s>\n' % (' ' * self.indent, name))
  501. self.indent -= 4
  502. self.file.write('%s</v%s>\n' % (' ' * self.indent, attrb))
  503. self.indent -= 4
  504. class ProcessGrcFile(object):
  505. def __init__(self, filename):
  506. """Process GRC file"""
  507. self.filename = filename
  508. # elements
  509. self.inGroup = False
  510. self.inRaster = False
  511. self.inVector = False
  512. # list of layers
  513. self.layers = []
  514. # error message
  515. self.error = ''
  516. self.num_error = 0
  517. def read(self, parent):
  518. """Read GRC file
  519. @param parent parent window
  520. @return list of map layers
  521. """
  522. try:
  523. file = open(self.filename, "r")
  524. except IOError:
  525. wx.MessageBox(parent=parent,
  526. message=_("Unable to open file <%s> for reading.") % self.filename,
  527. caption=_("Error"), style=wx.OK | wx.ICON_ERROR)
  528. return []
  529. line_id = 1
  530. for line in file.readlines():
  531. self.process_line(line.rstrip('\n'), line_id)
  532. line_id +=1
  533. file.close()
  534. if self.num_error > 0:
  535. wx.MessageBox(parent=parent,
  536. message=_("Some lines were skipped when reading settings "
  537. "from file <%(file)s>.\nSee 'Command output' window for details.\n\n"
  538. "Number of skipped lines: %(line)d") % \
  539. { 'file' : self.filename, 'line' : self.num_error },
  540. caption=_("Warning"), style=wx.OK | wx.ICON_EXCLAMATION)
  541. parent.goutput.WriteLog('Map layers loaded from GRC file <%s>' % self.filename)
  542. parent.goutput.WriteLog('Skipped lines:\n%s' % self.error)
  543. return self.layers
  544. def process_line(self, line, line_id):
  545. """Process line definition"""
  546. element = self._get_element(line)
  547. if element == 'Group':
  548. self.groupName = self._get_value(line)
  549. self.layers.append({
  550. "type" : 'group',
  551. "name" : self.groupName,
  552. "checked" : None,
  553. "opacity" : None,
  554. "cmd" : None,
  555. "group" : self.inGroup,
  556. "display" : 0 })
  557. self.inGroup = True
  558. elif element == '_check':
  559. if int(self._get_value(line)) == 1:
  560. self.layers[-1]['checked'] = True
  561. else:
  562. self.layers[-1]['checked'] = False
  563. elif element == 'End':
  564. if self.inRaster:
  565. self.inRaster = False
  566. elif self.inVector:
  567. self.inVector = False
  568. elif self.inGroup:
  569. self.inGroup = False
  570. elif element == 'opacity':
  571. self.layers[-1]['opacity'] = float(self._get_value(line))
  572. # raster
  573. elif element == 'Raster':
  574. self.inRaster = True
  575. self.layers.append({
  576. "type" : 'raster',
  577. "name" : self._get_value(line),
  578. "checked" : None,
  579. "opacity" : None,
  580. "cmd" : ['d.rast'],
  581. "group" : self.inGroup,
  582. "display" : 0})
  583. elif element == 'map' and self.inRaster:
  584. self.layers[-1]['cmd'].append('map=%s' % self._get_value(line))
  585. elif element == 'overlay' and self.inRaster:
  586. if int(self._get_value(line)) == 1:
  587. self.layers[-1]['cmd'].append('-o')
  588. elif element == 'rastquery' and self.inRaster:
  589. value = self._get_value(line)
  590. if value != '':
  591. self.layers[-1]['cmd'].append('catlist=%s' % value)
  592. elif element == 'bkcolor' and self.inRaster:
  593. value = self._get_value(line)
  594. if value != '':
  595. self.layers[-1]['cmd'].append('bg=%s' % value)
  596. # vector
  597. elif element == 'Vector':
  598. self.inVector = True
  599. self.layers.append({
  600. "type" : 'vector',
  601. "name" : self._get_value(line),
  602. "checked" : None,
  603. "opacity" : None,
  604. "cmd" : ['d.vect'],
  605. "group" : self.inGroup,
  606. "display" : 0})
  607. elif element == 'vect' and self.inVector:
  608. self.layers[-1]['cmd'].append('map=%s' % self._get_value(line))
  609. elif element in ('display_shape',
  610. 'display_cat',
  611. 'display_topo',
  612. 'display_dir',
  613. 'display_attr',
  614. 'type_point',
  615. 'type_line',
  616. 'type_boundary',
  617. 'type_centroid',
  618. 'type_area',
  619. 'type_face') and self.inVector:
  620. if int(self._get_value(line)) == 1:
  621. name = element.split('_')[0]
  622. type = element.split('_')[1]
  623. paramId = self._get_cmd_param_index(self.layers[-1]['cmd'], name)
  624. if paramId == -1:
  625. self.layers[-1]['cmd'].append('%s=%s' % (name, type))
  626. else:
  627. self.layers[-1]['cmd'][paramId] += ',%s' % type
  628. elif element in ('color',
  629. 'fcolor',
  630. 'lcolor') and self.inVector:
  631. value = self._get_value(line)
  632. if value != '':
  633. self.layers[-1]['cmd'].append('%s=%s' % (element,
  634. self._color_name_to_rgb(value)))
  635. elif element == 'rdmcolor' and self.inVector:
  636. if int(self._get_value(line)) == 1:
  637. self.layers[-1]['cmd'].append('-c')
  638. elif element == 'sqlcolor' and self.inVector:
  639. if int(self._get_value(line)) == 1:
  640. self.layers[-1]['cmd'].append('-a')
  641. elif element in ('icon',
  642. 'size',
  643. 'layer',
  644. 'xref',
  645. 'yref',
  646. 'lsize',
  647. 'where',
  648. 'minreg',
  649. 'maxreg') and self.inVector:
  650. value = self._get_value(line)
  651. if value != '':
  652. self.layers[-1]['cmd'].append('%s=%s' % (element,
  653. value))
  654. elif element == 'lwidth':
  655. value = self._get_value(line)
  656. if value != '':
  657. self.layers[-1]['cmd'].append('width=%s' % value)
  658. elif element == 'lfield':
  659. value = self._get_value(line)
  660. if value != '':
  661. self.layers[-1]['cmd'].append('llayer=%s' % value)
  662. elif element == 'attribute':
  663. value = self._get_value(line)
  664. if value != '':
  665. self.layers[-1]['cmd'].append('attrcol=%s' % value)
  666. elif element == 'cat':
  667. value = self._get_value(line)
  668. if value != '':
  669. self.layers[-1]['cmd'].append('cats=%s' % value)
  670. else:
  671. self.error += _(' row %d:') % line_id + line + os.linesep
  672. self.num_error += 1
  673. def _get_value(self, line):
  674. """Get value of element"""
  675. try:
  676. return line.strip(' ').split(' ')[1].strip(' ')
  677. except:
  678. return ''
  679. def _get_element(self, line):
  680. """Get element tag"""
  681. return line.strip(' ').split(' ')[0].strip(' ')
  682. def _get_cmd_param_index(self, cmd, name):
  683. """Get index of parameter in cmd list
  684. @param cmd cmd list
  685. @param name parameter name
  686. @return index
  687. @return -1 if not found
  688. """
  689. i = 0
  690. for param in cmd:
  691. if '=' not in param:
  692. i += 1
  693. continue
  694. if param.split('=')[0] == name:
  695. return i
  696. i += 1
  697. return -1
  698. def _color_name_to_rgb(self, value):
  699. """Convert color name (#) to rgb values"""
  700. col = wx.NamedColour(value)
  701. return str(col.Red()) + ':' + \
  702. str(col.Green()) + ':' + \
  703. str(col.Blue())