workspace.py 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824
  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" : None,
  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. cmd = mapTree.GetPyData(item)[0]['cmd']
  343. if type == 'command':
  344. self.file.write('%s<layer type="%s" name="%s" checked="%d">\n' % \
  345. (' ' * self.indent, type, ' '.join(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. name = mapTree.GetItemText(item)
  358. # remove 'opacity' part
  359. if '(opacity' in name:
  360. name = name.split('(', -1)[0].strip()
  361. opacity = maplayer.GetOpacity(float=True)
  362. self.file.write('%s<layer type="%s" name="%s" checked="%d" opacity="%f">\n' % \
  363. (' ' * self.indent, type, name, checked, opacity));
  364. self.indent += 4
  365. # selected ?
  366. if item == mapTree.layer_selected:
  367. self.file.write('%s<selected />\n' % (' ' * self.indent))
  368. # layer properties
  369. self.file.write('%s<task name="%s">\n' % (' ' * self.indent, cmd[0]))
  370. self.indent += 4
  371. for option in cmd[1:]:
  372. if option[0] == '-': # flag
  373. self.file.write('%s<flag name="%s" />\n' %
  374. (' ' * self.indent, option[1]))
  375. else: # parameter
  376. key, value = option.split('=', 1)
  377. self.file.write('%s<parameter name="%s">\n' %
  378. (' ' * self.indent, key))
  379. self.indent += 4
  380. self.file.write('%s<value>%s</value>\n' %
  381. (' ' * self.indent, value))
  382. self.indent -= 4
  383. self.file.write('%s</parameter>\n' % (' ' * self.indent));
  384. self.indent -= 4
  385. self.file.write('%s</task>\n' % (' ' * self.indent));
  386. nviz = mapTree.GetPyData(item)[0]['nviz']
  387. if nviz:
  388. self.file.write('%s<nviz>\n' % (' ' * self.indent));
  389. if maplayer.type == 'raster':
  390. self.__writeNvizSurface(nviz['surface'])
  391. elif maplayer.type == 'vector':
  392. self.__writeNvizVector(nviz['vector'])
  393. self.file.write('%s</nviz>\n' % (' ' * self.indent));
  394. self.indent -= 4
  395. self.file.write('%s</layer>\n' % (' ' * self.indent));
  396. item = mapTree.GetNextSibling(item)
  397. self.indent -= 4
  398. def __writeNvizSurface(self, data):
  399. """Save Nviz raster layer properties to workspace
  400. @param data Nviz layer properties
  401. """
  402. if not data.has_key('object'): # skip disabled
  403. return
  404. self.indent += 4
  405. self.file.write('%s<surface>\n' % (' ' * self.indent))
  406. self.indent += 4
  407. for attrb in data.iterkeys():
  408. if len(data[attrb]) < 1: # skip empty attributes
  409. continue
  410. if attrb == 'object':
  411. continue
  412. for name in data[attrb].iterkeys():
  413. # surface attribute
  414. if attrb == 'attribute':
  415. self.file.write('%s<%s name="%s" map="%d">\n' % \
  416. (' ' * self.indent, attrb, name, data[attrb][name]['map']))
  417. self.indent += 4
  418. self.file.write('%s<value>%s</value>\n' % (' ' * self.indent, data[attrb][name]['value']))
  419. self.indent -= 4
  420. # end tag
  421. self.file.write('%s</%s>\n' % (' ' * self.indent, attrb))
  422. # draw mode
  423. if attrb == 'draw':
  424. self.file.write('%s<%s' %(' ' * self.indent, attrb))
  425. if data[attrb].has_key('mode'):
  426. for tag, value in data[attrb]['mode']['desc'].iteritems():
  427. self.file.write(' %s="%s"' % (tag, value))
  428. self.file.write('>\n') # <draw ...>
  429. if data[attrb].has_key('resolution'):
  430. self.indent += 4
  431. for type in ('coarse', 'fine'):
  432. self.file.write('%s<resolution type="%s">\n' % (' ' * self.indent, type))
  433. self.indent += 4
  434. self.file.write('%s<value>%d</value>\n' % (' ' * self.indent,
  435. data[attrb]['resolution'][type]))
  436. self.indent -= 4
  437. self.file.write('%s</resolution>\n' % (' ' * self.indent))
  438. if data[attrb].has_key('wire-color'):
  439. self.file.write('%s<wire_color>\n' % (' ' * self.indent))
  440. self.indent += 4
  441. self.file.write('%s<value>%s</value>\n' % (' ' * self.indent,
  442. data[attrb]['wire-color']['value']))
  443. self.indent -= 4
  444. self.file.write('%s</wire_color>\n' % (' ' * self.indent))
  445. self.indent -= 4
  446. # position
  447. elif attrb == 'position':
  448. self.file.write('%s<%s>\n' %(' ' * self.indent, attrb))
  449. i = 0
  450. for tag in ('x', 'y', 'z'):
  451. self.indent += 4
  452. self.file.write('%s<%s>%d</%s>\n' % (' ' * self.indent, tag,
  453. data[attrb][tag], tag))
  454. i += 1
  455. self.indent -= 4
  456. if attrb != 'attribute':
  457. # end tag
  458. self.file.write('%s</%s>\n' % (' ' * self.indent, attrb))
  459. self.indent -= 4
  460. self.file.write('%s</surface>\n' % (' ' * self.indent))
  461. self.indent -= 4
  462. def __writeNvizVector(self, data):
  463. """Save Nviz vector layer properties (lines/points) to workspace
  464. @param data Nviz layer properties
  465. """
  466. self.indent += 4
  467. for attrb in data.iterkeys():
  468. if len(data[attrb]) < 1: # skip empty attributes
  469. continue
  470. if not data[attrb].has_key('object'): # skip disabled
  471. continue
  472. if attrb == 'lines':
  473. self.file.write('%s<v%s>\n' % (' ' * self.indent, attrb))
  474. elif attrb == 'points':
  475. markerId = data[attrb]['marker']
  476. marker = UserSettings.Get(group='nviz', key='vector',
  477. subkey=['points', 'marker'], internal=True)[markerId]
  478. self.file.write('%s<v%s marker="%s">\n' % (' ' * self.indent,
  479. attrb,
  480. marker))
  481. self.indent += 4
  482. for name in data[attrb].iterkeys():
  483. if name in ('object', 'marker'):
  484. continue
  485. if name == 'mode':
  486. self.file.write('%s<%s type="%s">\n' % (' ' * self.indent, name,
  487. data[attrb][name]['type']))
  488. if data[attrb][name]['type'] == 'surface':
  489. self.indent += 4
  490. self.file.write('%s<map>%s</map>\n' % (' ' * self.indent,
  491. data[attrb][name]['surface']))
  492. self.indent -= 4
  493. self.file.write('%s</%s>\n' % ((' ' * self.indent, name)))
  494. else:
  495. self.file.write('%s<%s>\n' % (' ' * self.indent, name))
  496. self.indent += 4
  497. self.file.write('%s<value>%s</value>\n' % (' ' * self.indent, data[attrb][name]))
  498. self.indent -= 4
  499. self.file.write('%s</%s>\n' % (' ' * self.indent, name))
  500. self.indent -= 4
  501. self.file.write('%s</v%s>\n' % (' ' * self.indent, attrb))
  502. self.indent -= 4
  503. class ProcessGrcFile(object):
  504. def __init__(self, filename):
  505. """Process GRC file"""
  506. self.filename = filename
  507. # elements
  508. self.inGroup = False
  509. self.inRaster = False
  510. self.inVector = False
  511. # list of layers
  512. self.layers = []
  513. # error message
  514. self.error = ''
  515. self.num_error = 0
  516. def read(self, parent):
  517. """Read GRC file
  518. @param parent parent window
  519. @return list of map layers
  520. """
  521. try:
  522. file = open(self.filename, "r")
  523. except IOError:
  524. wx.MessageBox(parent=parent,
  525. message=_("Unable to open file <%s> for reading.") % self.filename,
  526. caption=_("Error"), style=wx.OK | wx.ICON_ERROR)
  527. return []
  528. line_id = 1
  529. for line in file.readlines():
  530. self.process_line(line.rstrip('\n'), line_id)
  531. line_id +=1
  532. file.close()
  533. if self.num_error > 0:
  534. wx.MessageBox(parent=parent,
  535. message=_("Some lines were skipped when reading settings "
  536. "from file <%(file)s>.\nSee 'Command output' window for details.\n\n"
  537. "Number of skipped lines: %(line)d") % \
  538. { 'file' : self.filename, 'line' : self.num_error },
  539. caption=_("Warning"), style=wx.OK | wx.ICON_EXCLAMATION)
  540. parent.goutput.WriteLog('Map layers loaded from GRC file <%s>' % self.filename)
  541. parent.goutput.WriteLog('Skipped lines:\n%s' % self.error)
  542. return self.layers
  543. def process_line(self, line, line_id):
  544. """Process line definition"""
  545. element = self._get_element(line)
  546. if element == 'Group':
  547. self.groupName = self._get_value(line)
  548. self.layers.append({
  549. "type" : 'group',
  550. "name" : self.groupName,
  551. "checked" : None,
  552. "opacity" : None,
  553. "cmd" : None,
  554. "group" : self.inGroup,
  555. "display" : 0 })
  556. self.inGroup = True
  557. elif element == '_check':
  558. if int(self._get_value(line)) == 1:
  559. self.layers[-1]['checked'] = True
  560. else:
  561. self.layers[-1]['checked'] = False
  562. elif element == 'End':
  563. if self.inRaster:
  564. self.inRaster = False
  565. elif self.inVector:
  566. self.inVector = False
  567. elif self.inGroup:
  568. self.inGroup = False
  569. elif element == 'opacity':
  570. self.layers[-1]['opacity'] = float(self._get_value(line))
  571. # raster
  572. elif element == 'Raster':
  573. self.inRaster = True
  574. self.layers.append({
  575. "type" : 'raster',
  576. "name" : self._get_value(line),
  577. "checked" : None,
  578. "opacity" : None,
  579. "cmd" : ['d.rast'],
  580. "group" : self.inGroup,
  581. "display" : 0})
  582. elif element == 'map' and self.inRaster:
  583. self.layers[-1]['cmd'].append('map=%s' % self._get_value(line))
  584. elif element == 'overlay' and self.inRaster:
  585. if int(self._get_value(line)) == 1:
  586. self.layers[-1]['cmd'].append('-o')
  587. elif element == 'rastquery' and self.inRaster:
  588. value = self._get_value(line)
  589. if value != '':
  590. self.layers[-1]['cmd'].append('catlist=%s' % value)
  591. elif element == 'bkcolor' and self.inRaster:
  592. value = self._get_value(line)
  593. if value != '':
  594. self.layers[-1]['cmd'].append('bg=%s' % value)
  595. # vector
  596. elif element == 'Vector':
  597. self.inVector = True
  598. self.layers.append({
  599. "type" : 'vector',
  600. "name" : self._get_value(line),
  601. "checked" : None,
  602. "opacity" : None,
  603. "cmd" : ['d.vect'],
  604. "group" : self.inGroup,
  605. "display" : 0})
  606. elif element == 'vect' and self.inVector:
  607. self.layers[-1]['cmd'].append('map=%s' % self._get_value(line))
  608. elif element in ('display_shape',
  609. 'display_cat',
  610. 'display_topo',
  611. 'display_dir',
  612. 'display_attr',
  613. 'type_point',
  614. 'type_line',
  615. 'type_boundary',
  616. 'type_centroid',
  617. 'type_area',
  618. 'type_face') and self.inVector:
  619. if int(self._get_value(line)) == 1:
  620. name = element.split('_')[0]
  621. type = element.split('_')[1]
  622. paramId = self._get_cmd_param_index(self.layers[-1]['cmd'], name)
  623. if paramId == -1:
  624. self.layers[-1]['cmd'].append('%s=%s' % (name, type))
  625. else:
  626. self.layers[-1]['cmd'][paramId] += ',%s' % type
  627. elif element in ('color',
  628. 'fcolor',
  629. 'lcolor') and self.inVector:
  630. value = self._get_value(line)
  631. if value != '':
  632. self.layers[-1]['cmd'].append('%s=%s' % (element,
  633. self._color_name_to_rgb(value)))
  634. elif element == 'rdmcolor' and self.inVector:
  635. if int(self._get_value(line)) == 1:
  636. self.layers[-1]['cmd'].append('-c')
  637. elif element == 'sqlcolor' and self.inVector:
  638. if int(self._get_value(line)) == 1:
  639. self.layers[-1]['cmd'].append('-a')
  640. elif element in ('icon',
  641. 'size',
  642. 'layer',
  643. 'xref',
  644. 'yref',
  645. 'lsize',
  646. 'where',
  647. 'minreg',
  648. 'maxreg') and self.inVector:
  649. value = self._get_value(line)
  650. if value != '':
  651. self.layers[-1]['cmd'].append('%s=%s' % (element,
  652. value))
  653. elif element == 'lwidth':
  654. value = self._get_value(line)
  655. if value != '':
  656. self.layers[-1]['cmd'].append('width=%s' % value)
  657. elif element == 'lfield':
  658. value = self._get_value(line)
  659. if value != '':
  660. self.layers[-1]['cmd'].append('llayer=%s' % value)
  661. elif element == 'attribute':
  662. value = self._get_value(line)
  663. if value != '':
  664. self.layers[-1]['cmd'].append('attrcol=%s' % value)
  665. elif element == 'cat':
  666. value = self._get_value(line)
  667. if value != '':
  668. self.layers[-1]['cmd'].append('cats=%s' % value)
  669. else:
  670. self.error += _(' row %d:') % line_id + line + os.linesep
  671. self.num_error += 1
  672. def _get_value(self, line):
  673. """Get value of element"""
  674. try:
  675. return line.strip(' ').split(' ')[1].strip(' ')
  676. except:
  677. return ''
  678. def _get_element(self, line):
  679. """Get element tag"""
  680. return line.strip(' ').split(' ')[0].strip(' ')
  681. def _get_cmd_param_index(self, cmd, name):
  682. """Get index of parameter in cmd list
  683. @param cmd cmd list
  684. @param name parameter name
  685. @return index
  686. @return -1 if not found
  687. """
  688. i = 0
  689. for param in cmd:
  690. if '=' not in param:
  691. i += 1
  692. continue
  693. if param.split('=')[0] == name:
  694. return i
  695. i += 1
  696. return -1
  697. def _color_name_to_rgb(self, value):
  698. """Convert color name (#) to rgb values"""
  699. col = wx.NamedColour(value)
  700. return str(col.Red()) + ':' + \
  701. str(col.Green()) + ':' + \
  702. str(col.Blue())