workspace.py 35 KB

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