workspace.py 30 KB

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