workspace.py 32 KB

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