workspace.py 32 KB

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