workspace.py 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823
  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]['mode'] = {}
  160. self.layerNviz['surface'][tagName]['mode']['all'] = False
  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']['all'] = False
  232. self.layerNviz['surface']['draw']['resolution'][self.resolutionType] = int(self.value)
  233. del self.resolutionType
  234. elif name == 'wire_color':
  235. if self.inTag['nviz'] and self.inTag['surface']:
  236. self.layerNviz['surface']['draw']['wire-color'] = {}
  237. self.layerNviz['surface']['draw']['wire-color']['all'] = False
  238. self.layerNviz['surface']['draw']['wire-color']['value'] = str(self.value)
  239. elif name == 'x':
  240. if self.inTag['nviz'] and self.inTag['surface']:
  241. self.layerNviz['surface']['position']['x'] = int(self.value)
  242. elif name == 'y':
  243. if self.inTag['nviz'] and self.inTag['surface']:
  244. self.layerNviz['surface']['position']['y'] = int(self.value)
  245. elif name == 'z':
  246. if self.inTag['nviz'] and self.inTag['surface']:
  247. self.layerNviz['surface']['position']['z'] = int(self.value)
  248. elif name == 'color':
  249. if self.inTag['nviz']:
  250. if self.inTag['vlines']:
  251. self.layerNviz['vector']['lines']['color'] = str(self.value)
  252. elif self.inTag['vpoints']:
  253. self.layerNviz['vector']['points']['color'] = str(self.value)
  254. elif name == 'width':
  255. if self.inTag['nviz']:
  256. if self.inTag['vlines']:
  257. self.layerNviz['vector']['lines']['width'] = int(self.value)
  258. elif self.inTag['vpoints']:
  259. self.layerNviz['vector']['points']['width'] = int(self.value)
  260. elif name == 'height':
  261. if self.inTag['nviz']:
  262. if self.inTag['vlines']:
  263. self.layerNviz['vector']['lines']['height'] = int(self.value)
  264. elif self.inTag['vpoints']:
  265. self.layerNviz['vector']['points']['height'] = int(self.value)
  266. elif name == 'size':
  267. if self.inTag['nviz'] and self.inTag['vpoints']:
  268. self.layerNviz['vector']['points']['size'] = int(self.value)
  269. elif name == 'map':
  270. if self.inTag['nviz']:
  271. if self.inTag['vlines']:
  272. self.layerNviz['vector']['lines']['mode']['surface'] = str(self.value)
  273. elif self.inTag['vpoints']:
  274. self.layerNviz['vector']['points']['mode']['surface'] = str(self.value)
  275. self.inTag[name] = False
  276. def characters(self, ch):
  277. self.my_characters(ch)
  278. def my_characters(self, ch):
  279. if self.inTag['value'] or \
  280. self.inTag['x'] or \
  281. self.inTag['y'] or \
  282. self.inTag['z'] or \
  283. self.inTag['map']:
  284. self.value += ch
  285. class WriteWorkspaceFile(object):
  286. """Generic class for writing workspace file"""
  287. def __init__(self, lmgr, file):
  288. self.file = file
  289. self.lmgr = lmgr
  290. self.indent = 0
  291. # write header
  292. self.file.write('<?xml version="1.0" encoding="UTF-8"?>\n')
  293. self.file.write('<!DOCTYPE gxw SYSTEM "grass-gxw.dtd">\n')
  294. self.file.write('%s<gxw>\n' % (' ' * self.indent))
  295. self.indent =+ 4
  296. # layer manager
  297. windowPos = self.lmgr.GetPosition()
  298. windowSize = self.lmgr.GetSize()
  299. file.write('%s<layer_manager dim="%d,%d,%d,%d">\n' % (' ' * self.indent,
  300. windowPos[0],
  301. windowPos[1],
  302. windowSize[0],
  303. windowSize[1]
  304. ))
  305. file.write('%s</layer_manager>\n' % (' ' * self.indent))
  306. # list of displays
  307. for page in range(0, self.lmgr.gm_cb.GetPageCount()):
  308. mapTree = self.lmgr.gm_cb.GetPage(page).maptree
  309. region = mapTree.Map.region
  310. displayPos = mapTree.mapdisplay.GetPosition()
  311. displaySize = mapTree.mapdisplay.GetSize()
  312. file.write('%s<display render="%d" '
  313. 'mode="%d" showCompExtent="%d" '
  314. 'dim="%d,%d,%d,%d" '
  315. 'extent="%f,%f,%f,%f">\n' % (' ' * self.indent,
  316. int(mapTree.mapdisplay.autoRender.IsChecked()),
  317. mapTree.mapdisplay.toggleStatus.GetSelection(),
  318. int(mapTree.mapdisplay.showRegion.IsChecked()),
  319. displayPos[0],
  320. displayPos[1],
  321. displaySize[0],
  322. displaySize[1],
  323. region['w'],
  324. region['s'],
  325. region['e'],
  326. region['n']
  327. ))
  328. # list of layers
  329. item = mapTree.GetFirstChild(mapTree.root)[0]
  330. self.__writeLayer(mapTree, item)
  331. file.write('%s</display>\n' % (' ' * self.indent))
  332. self.indent =- 4
  333. file.write('%s</gxw>\n' % (' ' * self.indent))
  334. def __writeLayer(self, mapTree, item):
  335. """Write bunch of layers to GRASS Workspace XML file"""
  336. self.indent += 4
  337. while item and item.IsOk():
  338. type = mapTree.GetPyData(item)[0]['type']
  339. if type != 'group':
  340. maplayer = mapTree.GetPyData(item)[0]['maplayer']
  341. else:
  342. maplayer = None
  343. checked = int(item.IsChecked())
  344. cmd = mapTree.GetPyData(item)[0]['cmd']
  345. if type == 'command':
  346. self.file.write('%s<layer type="%s" name="%s" checked="%d">\n' % \
  347. (' ' * self.indent, type, ' '.join(cmd), checked));
  348. self.file.write('%s</layer>\n' % (' ' * self.indent));
  349. elif type == 'group':
  350. name = mapTree.GetItemText(item)
  351. self.file.write('%s<group name="%s" checked="%d">\n' % \
  352. (' ' * self.indent, name, checked));
  353. self.indent += 4
  354. subItem = mapTree.GetFirstChild(item)[0]
  355. self.__writeLayer(mapTree, subItem)
  356. self.indent -= 4
  357. self.file.write('%s</group>\n' % (' ' * self.indent));
  358. else:
  359. name = mapTree.GetItemText(item)
  360. opacity = maplayer.GetOpacity(float=True)
  361. self.file.write('%s<layer type="%s" name="%s" checked="%d" opacity="%f">\n' % \
  362. (' ' * self.indent, type, name, checked, opacity));
  363. self.indent += 4
  364. # selected ?
  365. if item == mapTree.layer_selected:
  366. self.file.write('%s<selected />\n' % (' ' * self.indent))
  367. # layer properties
  368. self.file.write('%s<task name="%s">\n' % (' ' * self.indent, cmd[0]))
  369. self.indent += 4
  370. for option in cmd[1:]:
  371. if option[0] == '-': # flag
  372. self.file.write('%s<flag name="%s" />\n' %
  373. (' ' * self.indent, option[1]))
  374. else: # parameter
  375. key, value = option.split('=', 1)
  376. self.file.write('%s<parameter name="%s">\n' %
  377. (' ' * self.indent, key))
  378. self.indent += 4
  379. self.file.write('%s<value>%s</value>\n' %
  380. (' ' * self.indent, value))
  381. self.indent -= 4
  382. self.file.write('%s</parameter>\n' % (' ' * self.indent));
  383. self.indent -= 4
  384. self.file.write('%s</task>\n' % (' ' * self.indent));
  385. nviz = mapTree.GetPyData(item)[0]['nviz']
  386. if nviz:
  387. self.file.write('%s<nviz>\n' % (' ' * self.indent));
  388. if maplayer.type == 'raster':
  389. self.__writeNvizSurface(nviz['surface'])
  390. elif maplayer.type == 'vector':
  391. self.__writeNvizVector(nviz['vector'])
  392. self.file.write('%s</nviz>\n' % (' ' * self.indent));
  393. self.indent -= 4
  394. self.file.write('%s</layer>\n' % (' ' * self.indent));
  395. item = mapTree.GetNextSibling(item)
  396. self.indent -= 4
  397. def __writeNvizSurface(self, data):
  398. """Save Nviz raster layer properties to workspace
  399. @param data Nviz layer properties
  400. """
  401. if not data.has_key('object'): # skip disabled
  402. return
  403. self.indent += 4
  404. self.file.write('%s<surface>\n' % (' ' * self.indent))
  405. self.indent += 4
  406. for attrb in data.iterkeys():
  407. if len(data[attrb]) < 1: # skip empty attributes
  408. continue
  409. if attrb == 'object':
  410. continue
  411. for name in data[attrb].iterkeys():
  412. # surface attribute
  413. if attrb == 'attribute':
  414. self.file.write('%s<%s name="%s" map="%d">\n' % \
  415. (' ' * self.indent, attrb, name, data[attrb][name]['map']))
  416. self.indent += 4
  417. self.file.write('%s<value>%s</value>\n' % (' ' * self.indent, data[attrb][name]['value']))
  418. self.indent -= 4
  419. # end tag
  420. self.file.write('%s</%s>\n' % (' ' * self.indent, attrb))
  421. # draw mode
  422. if attrb == 'draw':
  423. self.file.write('%s<%s' %(' ' * self.indent, attrb))
  424. if data[attrb].has_key('mode'):
  425. for tag, value in data[attrb]['mode']['desc'].iteritems():
  426. self.file.write(' %s="%s"' % (tag, value))
  427. self.file.write('>\n') # <draw ...>
  428. if data[attrb].has_key('resolution'):
  429. self.indent += 4
  430. for type in ('coarse', 'fine'):
  431. self.file.write('%s<resolution type="%s">\n' % (' ' * self.indent, type))
  432. self.indent += 4
  433. self.file.write('%s<value>%d</value>\n' % (' ' * self.indent,
  434. data[attrb]['resolution'][type]))
  435. self.indent -= 4
  436. self.file.write('%s</resolution>\n' % (' ' * self.indent))
  437. if data[attrb].has_key('wire-color'):
  438. self.file.write('%s<wire_color>\n' % (' ' * self.indent))
  439. self.indent += 4
  440. self.file.write('%s<value>%s</value>\n' % (' ' * self.indent,
  441. data[attrb]['wire-color']['value']))
  442. self.indent -= 4
  443. self.file.write('%s</wire_color>\n' % (' ' * self.indent))
  444. self.indent -= 4
  445. # position
  446. elif attrb == 'position':
  447. self.file.write('%s<%s>\n' %(' ' * self.indent, attrb))
  448. i = 0
  449. for tag in ('x', 'y', 'z'):
  450. self.indent += 4
  451. self.file.write('%s<%s>%d</%s>\n' % (' ' * self.indent, tag,
  452. data[attrb][tag], tag))
  453. i += 1
  454. self.indent -= 4
  455. if attrb != 'attribute':
  456. # end tag
  457. self.file.write('%s</%s>\n' % (' ' * self.indent, attrb))
  458. self.indent -= 4
  459. self.file.write('%s</surface>\n' % (' ' * self.indent))
  460. self.indent -= 4
  461. def __writeNvizVector(self, data):
  462. """Save Nviz vector layer properties (lines/points) to workspace
  463. @param data Nviz layer properties
  464. """
  465. self.indent += 4
  466. for attrb in data.iterkeys():
  467. if len(data[attrb]) < 1: # skip empty attributes
  468. continue
  469. if not data[attrb].has_key('object'): # skip disabled
  470. continue
  471. if attrb == 'lines':
  472. self.file.write('%s<v%s>\n' % (' ' * self.indent, attrb))
  473. elif attrb == 'points':
  474. markerId = data[attrb]['marker']
  475. marker = UserSettings.Get(group='nviz', key='vector',
  476. subkey=['points', 'marker'], internal=True)[markerId]
  477. self.file.write('%s<v%s marker="%s">\n' % (' ' * self.indent,
  478. attrb,
  479. marker))
  480. self.indent += 4
  481. for name in data[attrb].iterkeys():
  482. if name in ('object', 'marker'):
  483. continue
  484. if name == 'mode':
  485. self.file.write('%s<%s type="%s">\n' % (' ' * self.indent, name,
  486. data[attrb][name]['type']))
  487. if data[attrb][name]['type'] == 'surface':
  488. self.indent += 4
  489. self.file.write('%s<map>%s</map>\n' % (' ' * self.indent,
  490. data[attrb][name]['surface']))
  491. self.indent -= 4
  492. self.file.write('%s</%s>\n' % ((' ' * self.indent, name)))
  493. else:
  494. self.file.write('%s<%s>\n' % (' ' * self.indent, name))
  495. self.indent += 4
  496. self.file.write('%s<value>%s</value>\n' % (' ' * self.indent, data[attrb][name]))
  497. self.indent -= 4
  498. self.file.write('%s</%s>\n' % (' ' * self.indent, name))
  499. self.indent -= 4
  500. self.file.write('%s</v%s>\n' % (' ' * self.indent, attrb))
  501. self.indent -= 4
  502. class ProcessGrcFile(object):
  503. def __init__(self, filename):
  504. """Process GRC file"""
  505. self.filename = filename
  506. # elements
  507. self.inGroup = False
  508. self.inRaster = False
  509. self.inVector = False
  510. # list of layers
  511. self.layers = []
  512. # error message
  513. self.error = ''
  514. self.num_error = 0
  515. def read(self, parent):
  516. """Read GRC file
  517. @param parent parent window
  518. @return list of map layers
  519. """
  520. try:
  521. file = open(self.filename, "r")
  522. except IOError:
  523. wx.MessageBox(parent=parent,
  524. message=_("Unable to open file <%s> for reading.") % self.filename,
  525. caption=_("Error"), style=wx.OK | wx.ICON_ERROR)
  526. return []
  527. line_id = 1
  528. for line in file.readlines():
  529. self.process_line(line.rstrip('\n'), line_id)
  530. line_id +=1
  531. file.close()
  532. if self.num_error > 0:
  533. wx.MessageBox(parent=parent,
  534. message=_("Some lines were skipped when reading settings "
  535. "from file <%(file)s>.\nSee 'Command output' window for details.\n\n"
  536. "Number of skipped lines: %(line)d") % \
  537. { 'file' : self.filename, 'line' : self.num_error },
  538. caption=_("Warning"), style=wx.OK | wx.ICON_EXCLAMATION)
  539. parent.goutput.WriteLog('Map layers loaded from GRC file <%s>' % self.filename)
  540. parent.goutput.WriteLog('Skipped lines:\n%s' % self.error)
  541. return self.layers
  542. def process_line(self, line, line_id):
  543. """Process line definition"""
  544. element = self._get_element(line)
  545. if element == 'Group':
  546. self.groupName = self._get_value(line)
  547. self.layers.append({
  548. "type" : 'group',
  549. "name" : self.groupName,
  550. "checked" : None,
  551. "opacity" : None,
  552. "cmd" : None,
  553. "group" : self.inGroup,
  554. "display" : 0 })
  555. self.inGroup = True
  556. elif element == '_check':
  557. if int(self._get_value(line)) == 1:
  558. self.layers[-1]['checked'] = True
  559. else:
  560. self.layers[-1]['checked'] = False
  561. elif element == 'End':
  562. if self.inRaster:
  563. self.inRaster = False
  564. elif self.inVector:
  565. self.inVector = False
  566. elif self.inGroup:
  567. self.inGroup = False
  568. elif element == 'opacity':
  569. self.layers[-1]['opacity'] = float(self._get_value(line))
  570. # raster
  571. elif element == 'Raster':
  572. self.inRaster = True
  573. self.layers.append({
  574. "type" : 'raster',
  575. "name" : self._get_value(line),
  576. "checked" : None,
  577. "opacity" : None,
  578. "cmd" : ['d.rast'],
  579. "group" : self.inGroup,
  580. "display" : 0})
  581. elif element == 'map' and self.inRaster:
  582. self.layers[-1]['cmd'].append('map=%s' % self._get_value(line))
  583. elif element == 'overlay' and self.inRaster:
  584. if int(self._get_value(line)) == 1:
  585. self.layers[-1]['cmd'].append('-o')
  586. elif element == 'rastquery' and self.inRaster:
  587. value = self._get_value(line)
  588. if value != '':
  589. self.layers[-1]['cmd'].append('catlist=%s' % value)
  590. elif element == 'bkcolor' and self.inRaster:
  591. value = self._get_value(line)
  592. if value != '':
  593. self.layers[-1]['cmd'].append('bg=%s' % value)
  594. # vector
  595. elif element == 'Vector':
  596. self.inVector = True
  597. self.layers.append({
  598. "type" : 'vector',
  599. "name" : self._get_value(line),
  600. "checked" : None,
  601. "opacity" : None,
  602. "cmd" : ['d.vect'],
  603. "group" : self.inGroup,
  604. "display" : 0})
  605. elif element == 'vect' and self.inVector:
  606. self.layers[-1]['cmd'].append('map=%s' % self._get_value(line))
  607. elif element in ('display_shape',
  608. 'display_cat',
  609. 'display_topo',
  610. 'display_dir',
  611. 'display_attr',
  612. 'type_point',
  613. 'type_line',
  614. 'type_boundary',
  615. 'type_centroid',
  616. 'type_area',
  617. 'type_face') and self.inVector:
  618. if int(self._get_value(line)) == 1:
  619. name = element.split('_')[0]
  620. type = element.split('_')[1]
  621. paramId = self._get_cmd_param_index(self.layers[-1]['cmd'], name)
  622. if paramId == -1:
  623. self.layers[-1]['cmd'].append('%s=%s' % (name, type))
  624. else:
  625. self.layers[-1]['cmd'][paramId] += ',%s' % type
  626. elif element in ('color',
  627. 'fcolor',
  628. 'lcolor') and self.inVector:
  629. value = self._get_value(line)
  630. if value != '':
  631. self.layers[-1]['cmd'].append('%s=%s' % (element,
  632. self._color_name_to_rgb(value)))
  633. elif element == 'rdmcolor' and self.inVector:
  634. if int(self._get_value(line)) == 1:
  635. self.layers[-1]['cmd'].append('-c')
  636. elif element == 'sqlcolor' and self.inVector:
  637. if int(self._get_value(line)) == 1:
  638. self.layers[-1]['cmd'].append('-a')
  639. elif element in ('icon',
  640. 'size',
  641. 'layer',
  642. 'xref',
  643. 'yref',
  644. 'lsize',
  645. 'where',
  646. 'minreg',
  647. 'maxreg') and self.inVector:
  648. value = self._get_value(line)
  649. if value != '':
  650. self.layers[-1]['cmd'].append('%s=%s' % (element,
  651. value))
  652. elif element == 'lwidth':
  653. value = self._get_value(line)
  654. if value != '':
  655. self.layers[-1]['cmd'].append('width=%s' % value)
  656. elif element == 'lfield':
  657. value = self._get_value(line)
  658. if value != '':
  659. self.layers[-1]['cmd'].append('llayer=%s' % value)
  660. elif element == 'attribute':
  661. value = self._get_value(line)
  662. if value != '':
  663. self.layers[-1]['cmd'].append('attrcol=%s' % value)
  664. elif element == 'cat':
  665. value = self._get_value(line)
  666. if value != '':
  667. self.layers[-1]['cmd'].append('cats=%s' % value)
  668. else:
  669. self.error += _(' row %d:') % line_id + line + os.linesep
  670. self.num_error += 1
  671. def _get_value(self, line):
  672. """Get value of element"""
  673. try:
  674. return line.strip(' ').split(' ')[1].strip(' ')
  675. except:
  676. return ''
  677. def _get_element(self, line):
  678. """Get element tag"""
  679. return line.strip(' ').split(' ')[0].strip(' ')
  680. def _get_cmd_param_index(self, cmd, name):
  681. """Get index of parameter in cmd list
  682. @param cmd cmd list
  683. @param name parameter name
  684. @return index
  685. @return -1 if not found
  686. """
  687. i = 0
  688. for param in cmd:
  689. if '=' not in param:
  690. i += 1
  691. continue
  692. if param.split('=')[0] == name:
  693. return i
  694. i += 1
  695. return -1
  696. def _color_name_to_rgb(self, value):
  697. """Convert color name (#) to rgb values"""
  698. col = wx.NamedColour(value)
  699. return str(col.Red()) + ':' + \
  700. str(col.Green()) + ':' + \
  701. str(col.Blue())