workspace.py 55 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312
  1. """!
  2. @package core.workspace
  3. @brief Open/save workspace definition file
  4. Classes:
  5. - workspace::ProcessWorkspaceFile
  6. - workspace::WriteWorkspaceFile
  7. - workspace::ProcessGrcFile
  8. (C) 2007-2011 by the GRASS Development Team
  9. This program is free software under the GNU General Public License
  10. (>=v2). Read the file COPYING that comes with GRASS for details.
  11. @author Martin Landa <landa.martin gmail.com>
  12. """
  13. import os
  14. import wx
  15. from core.utils import normalize_whitespace
  16. from core.settings import UserSettings
  17. from nviz.main import NvizSettings
  18. class ProcessWorkspaceFile:
  19. def __init__(self, tree):
  20. """!A ElementTree handler for the GXW XML file, as defined in
  21. grass-gxw.dtd.
  22. """
  23. self.tree = tree
  24. self.root = self.tree.getroot()
  25. #
  26. # layer manager properties
  27. #
  28. self.layerManager = {}
  29. self.layerManager['pos'] = None # window position
  30. self.layerManager['size'] = None # window size
  31. #
  32. # list of mapdisplays
  33. #
  34. self.displays = []
  35. #
  36. # list of map layers
  37. #
  38. self.layers = []
  39. #
  40. # nviz state
  41. #
  42. self.nviz_state = {}
  43. self.displayIndex = -1 # first display has index '0'
  44. self.__processFile()
  45. if NvizSettings:
  46. self.nvizDefault = NvizSettings()
  47. else:
  48. self.nvizDefault = None
  49. def __filterValue(self, value):
  50. """!Filter value
  51. @param value
  52. """
  53. value = value.replace('&lt;', '<')
  54. value = value.replace('&gt;', '>')
  55. return value
  56. def __getNodeText(self, node, tag, default = ''):
  57. """!Get node text"""
  58. p = node.find(tag)
  59. if p is not None:
  60. return normalize_whitespace(p.text)
  61. return default
  62. def __processFile(self):
  63. """!Process workspace file"""
  64. #
  65. # layer manager
  66. #
  67. node_lm = self.root.find('layer_manager')
  68. if node_lm is not None:
  69. posAttr = node_lm.get('dim', '')
  70. if posAttr:
  71. posVal = map(int, posAttr.split(','))
  72. try:
  73. self.layerManager['pos'] = (posVal[0], posVal[1])
  74. self.layerManager['size'] = (posVal[2], posVal[3])
  75. except:
  76. pass
  77. #
  78. # displays
  79. #
  80. for display in self.root.findall('display'):
  81. self.displayIndex += 1
  82. # window position and size
  83. posAttr = display.get('dim', '')
  84. if posAttr:
  85. posVal = map(int, posAttr.split(','))
  86. try:
  87. pos = (posVal[0], posVal[1])
  88. size = (posVal[2], posVal[3])
  89. except:
  90. pos = None
  91. size = None
  92. else:
  93. pos = None
  94. size = None
  95. extentAttr = display.get('extent', '')
  96. if extentAttr:
  97. # w, s, e, n
  98. extent = map(float, extentAttr.split(','))
  99. else:
  100. extent = None
  101. # projection
  102. node_projection = display.find('projection')
  103. if node_projection is not None:
  104. projection = { 'enabled' : True,
  105. 'epsg' : node_projection.get('epsg', ''),
  106. 'proj' : self.__getNodeText(node_projection, 'value') }
  107. else:
  108. projection = { 'enabled' : False }
  109. self.displays.append( {
  110. "name" : display.get('name'),
  111. "render" : bool(int(display.get('render', "0"))),
  112. "mode" : int(display.get('mode', 0)),
  113. "showCompExtent" : bool(int(display.get('showCompExtent', "0"))),
  114. "pos" : pos,
  115. "size" : size,
  116. "extent" : extent,
  117. "alignExtent" : bool(int(display.get('alignExtent', "0"))),
  118. "constrainRes" : bool(int(display.get('constrainRes', "0"))),
  119. "projection" : projection,
  120. "viewMode" : display.get('viewMode', '2d')} )
  121. # process all layers/groups in the display
  122. self.__processLayers(display)
  123. # process nviz_state
  124. self.__processNvizState(display)
  125. def __processLayers(self, node, inGroup = -1):
  126. """!Process layers/groups of selected display
  127. @param node display tree node
  128. @param inGroup in group -> index of group item otherwise -1
  129. """
  130. for item in node.getchildren():
  131. if item.tag == 'group':
  132. # -> group
  133. self.layers.append( {
  134. "type" : 'group',
  135. "name" : item.get('name', ''),
  136. "checked" : bool(int(item.get('checked', "0"))),
  137. "opacity" : None,
  138. "cmd" : None,
  139. "group" : inGroup,
  140. "display" : self.displayIndex,
  141. "vdigit" : None,
  142. "nviz" : None})
  143. self.__processLayers(item, inGroup = len(self.layers) - 1) # process items in group
  144. elif item.tag == 'layer':
  145. cmd, selected, vdigit, nviz = self.__processLayer(item)
  146. lname = item.get('name', None)
  147. if lname and '\\n' in lname:
  148. lname = lname.replace('\\n', os.linesep)
  149. self.layers.append( {
  150. "type" : item.get('type', None),
  151. "name" : lname,
  152. "checked" : bool(int(item.get('checked', "0"))),
  153. "opacity" : float(item.get('opacity', '1.0')),
  154. "cmd" : cmd,
  155. "group" : inGroup,
  156. "display" : self.displayIndex,
  157. "selected" : selected,
  158. "vdigit" : vdigit,
  159. "nviz" : nviz } )
  160. def __processLayer(self, layer):
  161. """!Process layer item
  162. @param layer tree node
  163. """
  164. cmd = list()
  165. #
  166. # layer attributes (task) - 2D settings
  167. #
  168. node_task = layer.find('task')
  169. cmd.append(node_task.get('name', "unknown"))
  170. # flags
  171. for p in node_task.findall('flag'):
  172. flag = p.get('name', '')
  173. if len(flag) > 1:
  174. cmd.append('--' + flag)
  175. else:
  176. cmd.append('-' + flag)
  177. # parameters
  178. for p in node_task.findall('parameter'):
  179. cmd.append('%s=%s' % (p.get('name', ''),
  180. self.__filterValue(self.__getNodeText(p, 'value'))))
  181. if layer.find('selected') is not None:
  182. selected = True
  183. else:
  184. selected = False
  185. #
  186. # Vector digitizer settings
  187. #
  188. node_vdigit = layer.find('vdigit')
  189. if node_vdigit is not None:
  190. vdigit = self.__processLayerVdigit(node_vdigit)
  191. else:
  192. vdigit = None
  193. #
  194. # Nviz (3D settings)
  195. #
  196. node_nviz = layer.find('nviz')
  197. if node_nviz is not None:
  198. nviz = self.__processLayerNviz(node_nviz)
  199. else:
  200. nviz = None
  201. return (cmd, selected, vdigit, nviz)
  202. def __processLayerVdigit(self, node_vdigit):
  203. """!Process vector digitizer layer settings
  204. @param node_vdigit vdigit node
  205. """
  206. # init nviz layer properties
  207. vdigit = dict()
  208. for node in node_vdigit.findall('geometryAttribute'):
  209. if 'geomAttr' not in vdigit:
  210. vdigit['geomAttr'] = dict()
  211. type = node.get('type')
  212. vdigit['geomAttr'][type] = dict()
  213. vdigit['geomAttr'][type]['column'] = node.get('column') # required
  214. # default map units
  215. vdigit['geomAttr'][type]['units'] = node.get('units', 'mu')
  216. return vdigit
  217. def __processLayerNviz(self, node_nviz):
  218. """!Process 3D layer settings
  219. @param node_nviz nviz node
  220. """
  221. # init nviz layer properties
  222. nviz = {}
  223. if node_nviz.find('surface') is not None: # -> raster
  224. nviz['surface'] = {}
  225. for sec in ('attribute', 'draw', 'position'):
  226. nviz['surface'][sec] = {}
  227. elif node_nviz.find('vlines') is not None or \
  228. node_nviz.find('vpoints') is not None: # -> vector
  229. nviz['vector'] = {}
  230. for sec in ('lines', 'points'):
  231. nviz['vector'][sec] = {}
  232. if 'surface' in nviz:
  233. node_surface = node_nviz.find('surface')
  234. # attributes
  235. for attrb in node_surface.findall('attribute'):
  236. tagName = str(attrb.tag)
  237. attrbName = attrb.get('name', '')
  238. dc = nviz['surface'][tagName][attrbName] = {}
  239. if attrb.get('map', '0') == '0':
  240. dc['map'] = False
  241. else:
  242. dc['map'] = True
  243. value = self.__getNodeText(attrb, 'value')
  244. try:
  245. dc['value'] = int(value)
  246. except ValueError:
  247. try:
  248. dc['value'] = float(value)
  249. except ValueError:
  250. dc['value'] = str(value)
  251. # draw
  252. node_draw = node_surface.find('draw')
  253. if node_draw is not None:
  254. tagName = str(node_draw.tag)
  255. nviz['surface'][tagName]['all'] = False
  256. nviz['surface'][tagName]['mode'] = {}
  257. nviz['surface'][tagName]['mode']['value'] = -1 # to be calculated
  258. nviz['surface'][tagName]['mode']['desc'] = {}
  259. nviz['surface'][tagName]['mode']['desc']['shading'] = \
  260. str(node_draw.get('shading', ''))
  261. nviz['surface'][tagName]['mode']['desc']['style'] = \
  262. str(node_draw.get('style', ''))
  263. nviz['surface'][tagName]['mode']['desc']['mode'] = \
  264. str(node_draw.get('mode', ''))
  265. # resolution
  266. for node_res in node_draw.findall('resolution'):
  267. resType = str(node_res.get('type', ''))
  268. if 'resolution' not in nviz['surface']['draw']:
  269. nviz['surface']['draw']['resolution'] = {}
  270. value = int(self.__getNodeText(node_res, 'value'))
  271. nviz['surface']['draw']['resolution'][resType] = value
  272. # wire-color
  273. node_wire_color = node_draw.find('wire_color')
  274. if node_wire_color is not None:
  275. nviz['surface']['draw']['wire-color'] = {}
  276. value = str(self.__getNodeText(node_wire_color, 'value'))
  277. nviz['surface']['draw']['wire-color']['value'] = value
  278. # position
  279. node_pos = node_surface.find('position')
  280. if node_pos is not None:
  281. dc = nviz['surface']['position'] = {}
  282. for coor in ['x', 'y', 'z']:
  283. node = node_pos.find(coor)
  284. if node is None:
  285. continue
  286. value = int(self.__getNodeText(node_pos, coor))
  287. dc[coor] = value
  288. elif 'vector' in nviz:
  289. # vpoints
  290. node_vpoints = node_nviz.find('vpoints')
  291. if node_vpoints is not None:
  292. marker = str(node_vpoints.get('marker', ''))
  293. markerId = list(UserSettings.Get(group='nviz', key='vector',
  294. subkey=['points', 'marker'], internal=True)).index(marker)
  295. nviz['vector']['points']['marker'] = { 'value' : markerId }
  296. node_mode = node_vpoints.find('mode')
  297. if node_mode is not None:
  298. nviz['vector']['points']['mode'] = {}
  299. nviz['vector']['points']['mode']['type'] = str(node_mode.get('type', 'surface'))
  300. nviz['vector']['points']['mode']['surface'] = {}
  301. nviz['vector']['points']['mode']['surface']['value'] = []
  302. nviz['vector']['points']['mode']['surface']['show'] = []
  303. # map
  304. for node_map in node_mode.findall('map'):
  305. nviz['vector']['points']['mode']['surface']['value'].append(
  306. self.__processLayerNvizNode(node_map, 'name', str))
  307. nviz['vector']['points']['mode']['surface']['show'].append(bool(
  308. self.__processLayerNvizNode(node_map, 'checked', int)))
  309. # color
  310. self.__processLayerNvizNode(node_vpoints, 'color', str,
  311. nviz['vector']['points'])
  312. # width
  313. self.__processLayerNvizNode(node_vpoints, 'width', int,
  314. nviz['vector']['points'])
  315. # height
  316. self.__processLayerNvizNode(node_vpoints, 'height', int,
  317. nviz['vector']['points'])
  318. # height
  319. self.__processLayerNvizNode(node_vpoints, 'size', int,
  320. nviz['vector']['points'])
  321. # thematic
  322. node_thematic = node_vpoints.find('thematic')
  323. thematic = nviz['vector']['points']['thematic'] = {}
  324. thematic['rgbcolumn'] = self.__processLayerNvizNode(node_thematic, 'rgbcolumn', str)
  325. thematic['sizecolumn'] = self.__processLayerNvizNode(node_thematic, 'sizecolumn', str)
  326. for col in ('rgbcolumn', 'sizecolumn'):
  327. if thematic[col] == 'None':
  328. thematic[col] = None
  329. thematic['layer'] = self.__processLayerNvizNode(node_thematic, 'layer', int)
  330. for use in ('usecolor', 'usesize', 'usewidth'):
  331. if node_thematic.get(use, ''):
  332. thematic[use] = int(node_thematic.get(use, '0'))
  333. # vlines
  334. node_vlines = node_nviz.find('vlines')
  335. if node_vlines is not None:
  336. node_mode = node_vlines.find('mode')
  337. if node_mode is not None:
  338. nviz['vector']['lines']['mode'] = {}
  339. nviz['vector']['lines']['mode']['type'] = str(node_mode.get('type', ''))
  340. nviz['vector']['lines']['mode']['surface'] = {}
  341. nviz['vector']['lines']['mode']['surface']['value'] = []
  342. nviz['vector']['lines']['mode']['surface']['show'] = []
  343. # map
  344. for node_map in node_mode.findall('map'):
  345. nviz['vector']['lines']['mode']['surface']['value'].append(
  346. self.__processLayerNvizNode(node_map, 'name', str))
  347. nviz['vector']['lines']['mode']['surface']['show'].append(bool(
  348. self.__processLayerNvizNode(node_map, 'checked', int)))
  349. # color
  350. self.__processLayerNvizNode(node_vlines, 'color', str,
  351. nviz['vector']['lines'])
  352. # width
  353. self.__processLayerNvizNode(node_vlines, 'width', int,
  354. nviz['vector']['lines'])
  355. # height
  356. self.__processLayerNvizNode(node_vlines, 'height', int,
  357. nviz['vector']['lines'])
  358. # thematic
  359. node_thematic = node_vlines.find('thematic')
  360. thematic = nviz['vector']['lines']['thematic'] = {}
  361. thematic['rgbcolumn'] = self.__processLayerNvizNode(node_thematic, 'rgbcolumn', str)
  362. thematic['sizecolumn'] = self.__processLayerNvizNode(node_thematic, 'sizecolumn', str)
  363. for col in ('rgbcolumn', 'sizecolumn'):
  364. if thematic[col] == 'None':
  365. thematic[col] = None
  366. thematic['layer'] = self.__processLayerNvizNode(node_thematic, 'layer', int)
  367. for use in ('usecolor', 'usesize', 'usewidth'):
  368. if node_thematic.get(use, ''):
  369. thematic[use] = int(node_thematic.get(use, '0'))
  370. return nviz
  371. def __processLayerNvizNode(self, node, tag, cast, dc = None):
  372. """!Process given tag nviz/vector"""
  373. node_tag = node.find(tag)
  374. if node_tag is not None:
  375. if node_tag.find('value') is not None:
  376. value = cast(self.__getNodeText(node_tag, 'value'))
  377. else:
  378. try:
  379. value = cast(node_tag.text)
  380. except ValueError:
  381. if cast == str:
  382. value = ''
  383. else:
  384. value = None
  385. if dc:
  386. dc[tag] = dict()
  387. dc[tag]['value'] = value
  388. else:
  389. return value
  390. def __processNvizState(self, node):
  391. """!Process tag nviz_state"""
  392. node_state = node.find('nviz_state')
  393. if node_state is None:
  394. return
  395. self.nviz_state['display'] = self.displayIndex
  396. #
  397. # view
  398. #
  399. node_view = node_state.find('view')
  400. view = {}
  401. iview = {}
  402. node_position = node_view.find('v_position')
  403. view['position'] = {}
  404. view['position']['x'] = self.__processLayerNvizNode(node_position, 'x', float)
  405. view['position']['y'] = self.__processLayerNvizNode(node_position, 'y', float)
  406. node_persp = node_view.find('persp')
  407. view['persp'] = {}
  408. iview['persp'] = {}
  409. view['persp']['value'] = self.__processLayerNvizNode(node_persp, 'value', int)
  410. view['persp']['step'] = self.__processLayerNvizNode(node_persp, 'step', int)
  411. iview['persp']['min'] = self.__processLayerNvizNode(node_persp, 'min', int)
  412. iview['persp']['max'] = self.__processLayerNvizNode(node_persp, 'max', int)
  413. node_height = node_view.find('v_height')
  414. iview['height'] = {}
  415. iview['height']['value'] = self.__processLayerNvizNode(node_height, 'value', int)
  416. iview['height']['min'] = self.__processLayerNvizNode(node_height, 'min', int)
  417. iview['height']['max'] = self.__processLayerNvizNode(node_height, 'max', int)
  418. node_twist = node_view.find('twist')
  419. view['twist'] = {}
  420. iview['twist'] = {}
  421. view['twist']['value'] = self.__processLayerNvizNode(node_twist, 'value', int)
  422. iview['twist']['min'] = self.__processLayerNvizNode(node_twist, 'min', int)
  423. iview['twist']['max'] = self.__processLayerNvizNode(node_twist, 'max', int)
  424. node_zexag = node_view.find('z-exag')
  425. view['z-exag'] = {}
  426. iview['z-exag'] = {}
  427. view['z-exag']['value'] = self.__processLayerNvizNode(node_zexag, 'value', float)
  428. view['z-exag']['min'] = self.__processLayerNvizNode(node_zexag, 'min', int)
  429. view['z-exag']['max'] = self.__processLayerNvizNode(node_zexag, 'max', int)
  430. iview['z-exag']['llRatio'] = self.__processLayerNvizNode(node_zexag, 'llRatio', float)
  431. node_focus = node_view.find('focus')
  432. iview['focus'] = {}
  433. iview['focus']['x'] = self.__processLayerNvizNode(node_focus, 'x', int)
  434. iview['focus']['y'] = self.__processLayerNvizNode(node_focus, 'y', int)
  435. iview['focus']['z'] = self.__processLayerNvizNode(node_focus, 'z', int)
  436. node_dir = node_view.find('dir')
  437. if node_dir:
  438. iview['dir'] = {}
  439. iview['dir']['x'] = self.__processLayerNvizNode(node_dir, 'x', int)
  440. iview['dir']['y'] = self.__processLayerNvizNode(node_dir, 'y', int)
  441. iview['dir']['z'] = self.__processLayerNvizNode(node_dir, 'z', int)
  442. iview['dir']['use'] = True
  443. else:
  444. iview['dir'] = {}
  445. iview['dir']['x'] = -1
  446. iview['dir']['y'] = -1
  447. iview['dir']['z'] = -1
  448. iview['dir']['use'] = False
  449. view['background'] = {}
  450. color = self.__processLayerNvizNode(node_view, 'background_color', str)
  451. view['background']['color'] = tuple(map(int, color.split(':')))
  452. self.nviz_state['view'] = view
  453. self.nviz_state['iview'] = iview
  454. #
  455. # light
  456. #
  457. node_light = node_state.find('light')
  458. light = {}
  459. node_position = node_light.find('l_position')
  460. light['position'] = {}
  461. light['position']['x'] = self.__processLayerNvizNode(node_position, 'x', float)
  462. light['position']['y'] = self.__processLayerNvizNode(node_position, 'y', float)
  463. light['position']['z'] = self.__processLayerNvizNode(node_position, 'z', int)
  464. light['bright'] = self.__processLayerNvizNode(node_light, 'bright', int)
  465. light['ambient'] = self.__processLayerNvizNode(node_light, 'ambient', int)
  466. color = self.__processLayerNvizNode(node_light, 'color', str)
  467. light['color'] = tuple(map(int, color.split(':')))
  468. self.nviz_state['light'] = light
  469. node_constants = node_state.find('constant_planes')
  470. constants = []
  471. if node_constants:
  472. for i, node_plane in enumerate(node_constants.findall('plane')):
  473. plane = {}
  474. plane['color'] = self.__processLayerNvizNode(node_plane, 'color', str)
  475. plane['resolution'] = self.__processLayerNvizNode(node_plane, 'fine_resolution', int)
  476. plane['value'] = self.__processLayerNvizNode(node_plane, 'height', int)
  477. plane['object'] = {}
  478. constants.append({'constant': plane})
  479. self.nviz_state['constants'] = constants
  480. class WriteWorkspaceFile(object):
  481. """!Generic class for writing workspace file"""
  482. def __init__(self, lmgr, file):
  483. self.file = file
  484. self.lmgr = lmgr
  485. self.indent = 0
  486. # write header
  487. self.file.write('<?xml version="1.0" encoding="UTF-8"?>\n')
  488. self.file.write('<!DOCTYPE gxw SYSTEM "grass-gxw.dtd">\n')
  489. self.file.write('%s<gxw>\n' % (' ' * self.indent))
  490. self.indent =+ 4
  491. # layer manager
  492. windowPos = self.lmgr.GetPosition()
  493. windowSize = self.lmgr.GetSize()
  494. file.write('%s<layer_manager dim="%d,%d,%d,%d">\n' % (' ' * self.indent,
  495. windowPos[0],
  496. windowPos[1],
  497. windowSize[0],
  498. windowSize[1]
  499. ))
  500. file.write('%s</layer_manager>\n' % (' ' * self.indent))
  501. # list of displays
  502. for page in range(0, self.lmgr.GetLayerNotebook().GetPageCount()):
  503. dispName = self.lmgr.GetLayerNotebook().GetPageText(page)
  504. mapTree = self.lmgr.GetLayerNotebook().GetPage(page).maptree
  505. region = mapTree.GetMap().GetCurrentRegion()
  506. mapdisp = mapTree.GetMapDisplay()
  507. displayPos = mapdisp.GetPosition()
  508. displaySize = mapdisp.GetSize()
  509. if mapdisp.toolbars['map'].combo.GetSelection() == 1:
  510. viewmode = '3d'
  511. else:
  512. viewmode = '2d'
  513. file.write('%s<display '
  514. 'name="%s" render="%d" '
  515. 'mode="%d" showCompExtent="%d" '
  516. 'alignExtent="%d" '
  517. 'constrainRes="%d" '
  518. 'dim="%d,%d,%d,%d" '
  519. 'extent="%f,%f,%f,%f" '
  520. 'viewMode="%s" >\n' % (' ' * self.indent,
  521. dispName.encode('utf8'),
  522. int(mapdisp.GetProperty('render')),
  523. mapdisp.statusbarManager.GetMode(),
  524. int(mapdisp.GetProperty('region')),
  525. int(mapdisp.GetProperty('alignExtent')),
  526. int(mapdisp.GetProperty('resolution')),
  527. displayPos[0],
  528. displayPos[1],
  529. displaySize[0],
  530. displaySize[1],
  531. region['w'],
  532. region['s'],
  533. region['e'],
  534. region['n'],
  535. viewmode
  536. ))
  537. # projection statusbar info
  538. if mapdisp.GetProperty('projection') and \
  539. UserSettings.Get(group='display', key='projection', subkey='proj4'):
  540. self.indent += 4
  541. file.write('%s<projection' % (' ' * self.indent))
  542. epsg = UserSettings.Get(group='display', key='projection', subkey='epsg')
  543. if epsg:
  544. file.write(' epsg="%s"' % epsg)
  545. file.write('>\n')
  546. proj = UserSettings.Get(group='display', key='projection', subkey='proj4')
  547. self.indent += 4
  548. file.write('%s<value>%s</value>\n' % (' ' * self.indent, proj))
  549. self.indent -= 4
  550. file.write('%s</projection>\n' % (' ' * self.indent))
  551. self.indent -= 4
  552. # list of layers
  553. item = mapTree.GetFirstChild(mapTree.root)[0]
  554. self.__writeLayer(mapTree, item)
  555. if mapdisp.MapWindow3D is not None:
  556. nvizDisp = mapdisp.MapWindow3D
  557. self.__writeNvizState(view = nvizDisp.view, iview = nvizDisp.iview,
  558. light = nvizDisp.light, constants = nvizDisp.constants)
  559. file.write('%s</display>\n' % (' ' * self.indent))
  560. self.indent =- 4
  561. file.write('%s</gxw>\n' % (' ' * self.indent))
  562. def __filterValue(self, value):
  563. """!Make value XML-valid"""
  564. value = value.replace('<', '&lt;')
  565. value = value.replace('>', '&gt;')
  566. return value
  567. def __writeLayer(self, mapTree, item):
  568. """!Write bunch of layers to GRASS Workspace XML file"""
  569. self.indent += 4
  570. itemSelected = mapTree.GetSelections()
  571. while item and item.IsOk():
  572. type = mapTree.GetPyData(item)[0]['type']
  573. if type != 'group':
  574. maplayer = mapTree.GetPyData(item)[0]['maplayer']
  575. else:
  576. maplayer = None
  577. checked = int(item.IsChecked())
  578. if type == 'command':
  579. cmd = mapTree.GetPyData(item)[0]['maplayer'].GetCmd(string=True)
  580. self.file.write('%s<layer type="%s" name="%s" checked="%d">\n' % \
  581. (' ' * self.indent, type, cmd, checked));
  582. self.file.write('%s</layer>\n' % (' ' * self.indent));
  583. elif type == 'group':
  584. name = mapTree.GetItemText(item)
  585. self.file.write('%s<group name="%s" checked="%d">\n' % \
  586. (' ' * self.indent, name.encode('utf8'), checked));
  587. self.indent += 4
  588. subItem = mapTree.GetFirstChild(item)[0]
  589. self.__writeLayer(mapTree, subItem)
  590. self.indent -= 4
  591. self.file.write('%s</group>\n' % (' ' * self.indent));
  592. else:
  593. cmd = mapTree.GetPyData(item)[0]['maplayer'].GetCmd(string = False)
  594. name = mapTree.GetItemText(item).replace(os.linesep, '\\n')
  595. opacity = maplayer.GetOpacity(float = True)
  596. # remove 'opacity' part
  597. if opacity < 1:
  598. name = name.split('(', -1)[0].strip()
  599. self.file.write('%s<layer type="%s" name="%s" checked="%d" opacity="%f">\n' % \
  600. (' ' * self.indent, type, name.encode('utf8'), checked, opacity));
  601. self.indent += 4
  602. # selected ?
  603. if item in itemSelected:
  604. self.file.write('%s<selected />\n' % (' ' * self.indent))
  605. # layer properties
  606. self.file.write('%s<task name="%s">\n' % (' ' * self.indent, cmd[0]))
  607. self.indent += 4
  608. for key, val in cmd[1].iteritems():
  609. if key == 'flags':
  610. for f in val:
  611. self.file.write('%s<flag name="%s" />\n' %
  612. (' ' * self.indent, f))
  613. elif val in (True, False):
  614. self.file.write('%s<flag name="%s" />\n' %
  615. (' ' * self.indent, key))
  616. else: # parameter
  617. self.file.write('%s<parameter name="%s">\n' %
  618. (' ' * self.indent, key))
  619. self.indent += 4
  620. self.file.write('%s<value>%s</value>\n' %
  621. (' ' * self.indent, self.__filterValue(val)))
  622. self.indent -= 4
  623. self.file.write('%s</parameter>\n' % (' ' * self.indent));
  624. self.indent -= 4
  625. self.file.write('%s</task>\n' % (' ' * self.indent));
  626. # vector digitizer
  627. vdigit = mapTree.GetPyData(item)[0]['vdigit']
  628. if vdigit:
  629. self.file.write('%s<vdigit>\n' % (' ' * self.indent))
  630. if 'geomAttr' in vdigit:
  631. self.indent += 4
  632. for type, val in vdigit['geomAttr'].iteritems():
  633. units = ''
  634. if val['units'] != 'mu':
  635. units = ' units="%s"' % val['units']
  636. self.file.write('%s<geometryAttribute type="%s" column="%s"%s />\n' % \
  637. (' ' * self.indent, type, val['column'], units))
  638. self.indent -= 4
  639. self.file.write('%s</vdigit>\n' % (' ' * self.indent))
  640. # nviz
  641. nviz = mapTree.GetPyData(item)[0]['nviz']
  642. if nviz:
  643. self.file.write('%s<nviz>\n' % (' ' * self.indent))
  644. if maplayer.type == 'raster':
  645. self.__writeNvizSurface(nviz['surface'])
  646. elif maplayer.type == 'vector':
  647. self.__writeNvizVector(nviz['vector'])
  648. self.file.write('%s</nviz>\n' % (' ' * self.indent))
  649. self.indent -= 4
  650. self.file.write('%s</layer>\n' % (' ' * self.indent))
  651. item = mapTree.GetNextSibling(item)
  652. self.indent -= 4
  653. def __writeNvizSurface(self, data):
  654. """!Save Nviz raster layer properties to workspace
  655. @param data Nviz layer properties
  656. """
  657. if 'object' not in data: # skip disabled
  658. return
  659. self.indent += 4
  660. self.file.write('%s<surface>\n' % (' ' * self.indent))
  661. self.indent += 4
  662. for attrb in data.iterkeys():
  663. if len(data[attrb]) < 1: # skip empty attributes
  664. continue
  665. if attrb == 'object':
  666. continue
  667. for name in data[attrb].iterkeys():
  668. # surface attribute
  669. if attrb == 'attribute':
  670. if data[attrb][name]['map'] is None:
  671. continue
  672. self.file.write('%s<%s name="%s" map="%d">\n' % \
  673. (' ' * self.indent, attrb, name, data[attrb][name]['map']))
  674. self.indent += 4
  675. self.file.write('%s<value>%s</value>\n' % (' ' * self.indent, data[attrb][name]['value']))
  676. self.indent -= 4
  677. # end tag
  678. self.file.write('%s</%s>\n' % (' ' * self.indent, attrb))
  679. # draw mode
  680. if attrb == 'draw':
  681. self.file.write('%s<%s' %(' ' * self.indent, attrb))
  682. if 'mode' in data[attrb]:
  683. for tag, value in data[attrb]['mode']['desc'].iteritems():
  684. self.file.write(' %s="%s"' % (tag, value))
  685. self.file.write('>\n') # <draw ...>
  686. if 'resolution' in data[attrb]:
  687. self.indent += 4
  688. for type in ('coarse', 'fine'):
  689. self.file.write('%s<resolution type="%s">\n' % (' ' * self.indent, type))
  690. self.indent += 4
  691. self.file.write('%s<value>%d</value>\n' % (' ' * self.indent,
  692. data[attrb]['resolution'][type]))
  693. self.indent -= 4
  694. self.file.write('%s</resolution>\n' % (' ' * self.indent))
  695. if 'wire-color' in data[attrb]:
  696. self.file.write('%s<wire_color>\n' % (' ' * self.indent))
  697. self.indent += 4
  698. self.file.write('%s<value>%s</value>\n' % (' ' * self.indent,
  699. data[attrb]['wire-color']['value']))
  700. self.indent -= 4
  701. self.file.write('%s</wire_color>\n' % (' ' * self.indent))
  702. self.indent -= 4
  703. # position
  704. elif attrb == 'position':
  705. self.file.write('%s<%s>\n' %(' ' * self.indent, attrb))
  706. i = 0
  707. for tag in ('x', 'y', 'z'):
  708. self.indent += 4
  709. self.file.write('%s<%s>%d</%s>\n' % (' ' * self.indent, tag,
  710. data[attrb][tag], tag))
  711. i += 1
  712. self.indent -= 4
  713. if attrb != 'attribute':
  714. # end tag
  715. self.file.write('%s</%s>\n' % (' ' * self.indent, attrb))
  716. self.indent -= 4
  717. self.file.write('%s</surface>\n' % (' ' * self.indent))
  718. self.indent -= 4
  719. def __writeNvizVector(self, data):
  720. """!Save Nviz vector layer properties (lines/points) to workspace
  721. @param data Nviz layer properties
  722. """
  723. self.indent += 4
  724. for attrb in data.iterkeys():
  725. if len(data[attrb]) < 1: # skip empty attributes
  726. continue
  727. if 'object' not in data[attrb]: # skip disabled
  728. continue
  729. if attrb == 'lines':
  730. self.file.write('%s<v%s>\n' % (' ' * self.indent, attrb))
  731. elif attrb == 'points':
  732. markerId = data[attrb]['marker']['value']
  733. marker = UserSettings.Get(group = 'nviz', key = 'vector',
  734. subkey = ['points', 'marker'], internal = True)[markerId]
  735. self.file.write('%s<v%s marker="%s">\n' % (' ' * self.indent,
  736. attrb,
  737. marker))
  738. self.indent += 4
  739. for name in data[attrb].iterkeys():
  740. if name in ('object', 'marker'):
  741. continue
  742. if name == 'mode':
  743. self.file.write('%s<%s type="%s">\n' % (' ' * self.indent, name,
  744. data[attrb][name]['type']))
  745. if data[attrb][name]['type'] == 'surface':
  746. self.indent += 4
  747. for idx, surface in enumerate(data[attrb][name]['surface']['value']):
  748. checked = data[attrb][name]['surface']['show'][idx]
  749. self.file.write('%s<map>\n' % (' ' * self.indent))
  750. self.indent += 4
  751. self.file.write('%s<name>%s</name>\n' % (' ' * self.indent, surface))
  752. self.file.write('%s<checked>%s</checked>\n' % (' ' * self.indent, int(checked)))
  753. self.indent -= 4
  754. self.file.write('%s</map>\n' % (' ' * self.indent))
  755. self.indent -= 4
  756. self.file.write('%s</%s>\n' % ((' ' * self.indent, name)))
  757. elif name == 'thematic':
  758. self.file.write('%s<%s ' % (' ' * self.indent, name))
  759. for key in data[attrb][name].iterkeys():
  760. if key.startswith('use'):
  761. self.file.write('%s="%s" ' % (key, int(data[attrb][name][key])))
  762. self.file.write('>\n')
  763. self.indent += 4
  764. for key, value in data[attrb][name].iteritems():
  765. if key.startswith('use'):
  766. continue
  767. if value is None:
  768. value = ''
  769. self.file.write('%s<%s>%s</%s>\n' % (' ' * self.indent, key, value, key))
  770. self.indent -= 4
  771. self.file.write('%s</%s>\n' % (' ' * self.indent, name))
  772. else:
  773. self.file.write('%s<%s>\n' % (' ' * self.indent, name))
  774. self.indent += 4
  775. self.file.write('%s<value>%s</value>\n' % (' ' * self.indent, data[attrb][name]['value']))
  776. self.indent -= 4
  777. self.file.write('%s</%s>\n' % (' ' * self.indent, name))
  778. self.indent -= 4
  779. self.file.write('%s</v%s>\n' % (' ' * self.indent, attrb))
  780. self.indent -= 4
  781. def __writeNvizState(self, view, iview, light, constants):
  782. """"!Save Nviz properties (view, light) to workspace
  783. @param view Nviz view properties
  784. @param iview Nviz internal view properties
  785. @param light Nviz light properties
  786. """
  787. self.indent += 4
  788. self.file.write('%s<nviz_state>\n' % (' ' * self.indent))
  789. #
  790. # view
  791. #
  792. self.indent += 4
  793. self.file.write('%s<view>\n' % (' ' * self.indent))
  794. self.indent += 4
  795. # position
  796. self.file.write('%s<v_position>\n' % (' ' * self.indent))
  797. self.indent += 4
  798. self.file.write('%s<x>%.2f</x>\n' % (' ' * self.indent, view['position']['x']))
  799. self.file.write('%s<y>%.2f</y>\n' % (' ' * self.indent, view['position']['y']))
  800. self.indent -= 4
  801. self.file.write('%s</v_position>\n' % (' ' * self.indent))
  802. # perspective
  803. self.file.write('%s<persp>\n' % (' ' * self.indent))
  804. self.indent += 4
  805. self.file.write('%s<value>%d</value>\n' % (' ' * self.indent, view['persp']['value']))
  806. self.file.write('%s<step>%d</step>\n' % (' ' * self.indent, view['persp']['step']))
  807. self.file.write('%s<min>%d</min>\n' % (' ' * self.indent, iview['persp']['min']))
  808. self.file.write('%s<max>%d</max>\n' % (' ' * self.indent, iview['persp']['max']))
  809. self.indent -= 4
  810. self.file.write('%s</persp>\n' % (' ' * self.indent))
  811. # height
  812. self.file.write('%s<v_height>\n' % (' ' * self.indent))
  813. self.indent += 4
  814. self.file.write('%s<value>%d</value>\n' % (' ' * self.indent, iview['height']['value']))
  815. self.file.write('%s<min>%d</min>\n' % (' ' * self.indent, iview['height']['min']))
  816. self.file.write('%s<max>%d</max>\n' % (' ' * self.indent, iview['height']['max']))
  817. self.indent -= 4
  818. self.file.write('%s</v_height>\n' % (' ' * self.indent))
  819. # twist
  820. self.file.write('%s<twist>\n' % (' ' * self.indent))
  821. self.indent += 4
  822. self.file.write('%s<value>%d</value>\n' % (' ' * self.indent, view['twist']['value']))
  823. self.file.write('%s<min>%d</min>\n' % (' ' * self.indent, iview['twist']['min']))
  824. self.file.write('%s<max>%d</max>\n' % (' ' * self.indent, iview['twist']['max']))
  825. self.indent -= 4
  826. self.file.write('%s</twist>\n' % (' ' * self.indent))
  827. # z-exag
  828. self.file.write('%s<z-exag>\n' % (' ' * self.indent))
  829. self.indent += 4
  830. self.file.write('%s<value>%.2f</value>\n' % (' ' * self.indent, view['z-exag']['value']))
  831. self.file.write('%s<min>%d</min>\n' % (' ' * self.indent, view['z-exag']['min']))
  832. self.file.write('%s<max>%d</max>\n' % (' ' * self.indent, view['z-exag']['max']))
  833. self.file.write('%s<llRatio>%.2f</llRatio>\n' % (' ' * self.indent, iview['z-exag']['llRatio']))
  834. self.indent -= 4
  835. self.file.write('%s</z-exag>\n' % (' ' * self.indent))
  836. # focus (look here)
  837. self.file.write('%s<focus>\n' % (' ' * self.indent))
  838. self.indent += 4
  839. self.file.write('%s<x>%d</x>\n' % (' ' * self.indent, iview['focus']['x']))
  840. self.file.write('%s<y>%d</y>\n' % (' ' * self.indent, iview['focus']['y']))
  841. self.file.write('%s<z>%d</z>\n' % (' ' * self.indent, iview['focus']['z']))
  842. self.indent -= 4
  843. self.file.write('%s</focus>\n' % (' ' * self.indent))
  844. # background
  845. self.__writeTagWithValue('background_color', view['background']['color'][:3], format = 'd:%d:%d')
  846. self.indent -= 4
  847. self.file.write('%s</view>\n' % (' ' * self.indent))
  848. #
  849. # light
  850. #
  851. self.file.write('%s<light>\n' % (' ' * self.indent))
  852. self.indent += 4
  853. # position
  854. self.file.write('%s<l_position>\n' % (' ' * self.indent))
  855. self.indent += 4
  856. self.file.write('%s<x>%.2f</x>\n' % (' ' * self.indent, light['position']['x']))
  857. self.file.write('%s<y>%.2f</y>\n' % (' ' * self.indent, light['position']['y']))
  858. self.file.write('%s<z>%d</z>\n' % (' ' * self.indent, light['position']['z']))
  859. self.indent -= 4
  860. self.file.write('%s</l_position>\n' % (' ' * self.indent))
  861. # bright
  862. self.__writeTagWithValue('bright', light['bright'])
  863. # ambient
  864. self.__writeTagWithValue('ambient', light['ambient'])
  865. # color
  866. self.__writeTagWithValue('color', light['color'][:3], format = 'd:%d:%d')
  867. self.indent -= 4
  868. self.file.write('%s</light>\n' % (' ' * self.indent))
  869. #
  870. # constant planes
  871. #
  872. if constants:
  873. self.file.write('%s<constant_planes>\n' % (' ' * self.indent))
  874. self.indent += 4
  875. for idx, plane in enumerate(constants):
  876. self.file.write('%s<plane>\n' % (' ' * self.indent))
  877. self.indent += 4
  878. self.__writeTagWithValue('height', constants[idx]['constant']['value'])
  879. self.__writeTagWithValue('fine_resolution', constants[idx]['constant']['resolution'])
  880. self.__writeTagWithValue('color', constants[idx]['constant']['color'], format = 's')
  881. self.indent -= 4
  882. self.file.write('%s</plane>\n' % (' ' * self.indent))
  883. self.indent -= 4
  884. self.file.write('%s</constant_planes>\n' % (' ' * self.indent))
  885. self.indent -= 4
  886. self.file.write('%s</nviz_state>\n' % (' ' * self.indent))
  887. self.indent -= 4
  888. def __writeTagWithValue(self, tag, data, format = 'd'):
  889. """!Helper function for writing pair tag
  890. @param tag written tag
  891. @param data written data
  892. @param format conversion type
  893. """
  894. self.file.write('%s<%s>\n' % (' ' * self.indent, tag))
  895. self.indent += 4
  896. self.file.write('%s' % (' ' * self.indent))
  897. self.file.write(('<value>%' + format + '</value>\n') % data)
  898. self.indent -= 4
  899. self.file.write('%s</%s>\n' % (' ' * self.indent, tag))
  900. class ProcessGrcFile(object):
  901. def __init__(self, filename):
  902. """!Process GRC file"""
  903. self.filename = filename
  904. # elements
  905. self.inGroup = False
  906. self.inRaster = False
  907. self.inVector = False
  908. # list of layers
  909. self.layers = []
  910. # error message
  911. self.error = ''
  912. self.num_error = 0
  913. def read(self, parent):
  914. """!Read GRC file
  915. @param parent parent window
  916. @return list of map layers
  917. """
  918. try:
  919. file = open(self.filename, "r")
  920. except IOError:
  921. wx.MessageBox(parent=parent,
  922. message=_("Unable to open file <%s> for reading.") % self.filename,
  923. caption=_("Error"), style=wx.OK | wx.ICON_ERROR)
  924. return []
  925. line_id = 1
  926. for line in file.readlines():
  927. self.process_line(line.rstrip('\n'), line_id)
  928. line_id +=1
  929. file.close()
  930. if self.num_error > 0:
  931. wx.MessageBox(parent=parent,
  932. message=_("Some lines were skipped when reading settings "
  933. "from file <%(file)s>.\nSee 'Command output' window for details.\n\n"
  934. "Number of skipped lines: %(line)d") % \
  935. { 'file' : self.filename, 'line' : self.num_error },
  936. caption=_("Warning"), style=wx.OK | wx.ICON_EXCLAMATION)
  937. parent.goutput.WriteLog('Map layers loaded from GRC file <%s>' % self.filename)
  938. parent.goutput.WriteLog('Skipped lines:\n%s' % self.error)
  939. return self.layers
  940. def process_line(self, line, line_id):
  941. """!Process line definition"""
  942. element = self._get_element(line)
  943. if element == 'Group':
  944. self.groupName = self._get_value(line)
  945. self.layers.append({
  946. "type" : 'group',
  947. "name" : self.groupName,
  948. "checked" : None,
  949. "opacity" : None,
  950. "cmd" : None,
  951. "group" : self.inGroup,
  952. "display" : 0 })
  953. self.inGroup = True
  954. elif element == '_check':
  955. if int(self._get_value(line)) == 1:
  956. self.layers[-1]['checked'] = True
  957. else:
  958. self.layers[-1]['checked'] = False
  959. elif element == 'End':
  960. if self.inRaster:
  961. self.inRaster = False
  962. elif self.inVector:
  963. self.inVector = False
  964. elif self.inGroup:
  965. self.inGroup = False
  966. elif self.inGridline:
  967. self.inGridline = False
  968. elif element == 'opacity':
  969. self.layers[-1]['opacity'] = float(self._get_value(line))
  970. # raster
  971. elif element == 'Raster':
  972. self.inRaster = True
  973. self.layers.append({
  974. "type" : 'raster',
  975. "name" : self._get_value(line),
  976. "checked" : None,
  977. "opacity" : None,
  978. "cmd" : ['d.rast'],
  979. "group" : self.inGroup,
  980. "display" : 0})
  981. elif element == 'map' and self.inRaster:
  982. self.layers[-1]['cmd'].append('map=%s' % self._get_value(line))
  983. elif element == 'overlay' and self.inRaster:
  984. if int(self._get_value(line)) == 1:
  985. self.layers[-1]['cmd'].append('-o')
  986. elif element == 'rastquery' and self.inRaster:
  987. value = self._get_value(line)
  988. if value != '':
  989. self.layers[-1]['cmd'].append('catlist=%s' % value)
  990. elif element == 'bkcolor' and self.inRaster:
  991. value = self._get_value(line)
  992. if value != '':
  993. self.layers[-1]['cmd'].append('bg=%s' % value)
  994. # vector
  995. elif element == 'Vector':
  996. self.inVector = True
  997. self.layers.append({
  998. "type" : 'vector',
  999. "name" : self._get_value(line),
  1000. "checked" : None,
  1001. "opacity" : None,
  1002. "cmd" : ['d.vect'],
  1003. "group" : self.inGroup,
  1004. "display" : 0})
  1005. elif element == 'vect' and self.inVector:
  1006. self.layers[-1]['cmd'].append('map=%s' % self._get_value(line))
  1007. elif element in ('display_shape',
  1008. 'display_cat',
  1009. 'display_topo',
  1010. 'display_dir',
  1011. 'display_attr',
  1012. 'type_point',
  1013. 'type_line',
  1014. 'type_boundary',
  1015. 'type_centroid',
  1016. 'type_area',
  1017. 'type_face') and self.inVector:
  1018. if int(self._get_value(line)) == 1:
  1019. name = element.split('_')[0]
  1020. type = element.split('_')[1]
  1021. paramId = self._get_cmd_param_index(self.layers[-1]['cmd'], name)
  1022. if paramId == -1:
  1023. self.layers[-1]['cmd'].append('%s=%s' % (name, type))
  1024. else:
  1025. self.layers[-1]['cmd'][paramId] += ',%s' % type
  1026. elif element in ('color',
  1027. 'fcolor',
  1028. 'lcolor') and self.inVector:
  1029. value = self._get_value(line)
  1030. if value != '':
  1031. self.layers[-1]['cmd'].append('%s=%s' % (element,
  1032. self._color_name_to_rgb(value)))
  1033. elif element == 'rdmcolor' and self.inVector:
  1034. if int(self._get_value(line)) == 1:
  1035. self.layers[-1]['cmd'].append('-c')
  1036. elif element == 'sqlcolor' and self.inVector:
  1037. if int(self._get_value(line)) == 1:
  1038. self.layers[-1]['cmd'].append('-a')
  1039. elif element in ('icon',
  1040. 'size',
  1041. 'layer',
  1042. 'xref',
  1043. 'yref',
  1044. 'lsize',
  1045. 'where',
  1046. 'minreg',
  1047. 'maxreg') and self.inVector:
  1048. value = self._get_value(line)
  1049. if value != '':
  1050. self.layers[-1]['cmd'].append('%s=%s' % (element,
  1051. value))
  1052. elif element == 'lwidth':
  1053. value = self._get_value(line)
  1054. if value != '':
  1055. self.layers[-1]['cmd'].append('width=%s' % value)
  1056. elif element == 'lfield':
  1057. value = self._get_value(line)
  1058. if value != '':
  1059. self.layers[-1]['cmd'].append('llayer=%s' % value)
  1060. elif element == 'attribute':
  1061. value = self._get_value(line)
  1062. if value != '':
  1063. self.layers[-1]['cmd'].append('attrcol=%s' % value)
  1064. elif element == 'cat':
  1065. value = self._get_value(line)
  1066. if value != '':
  1067. self.layers[-1]['cmd'].append('cats=%s' % value)
  1068. # gridline
  1069. elif element == 'gridline':
  1070. self.inGridline = True
  1071. self.layers.append({
  1072. "type" : 'grid',
  1073. "name" : self._get_value(line),
  1074. "checked" : None,
  1075. "opacity" : None,
  1076. "cmd" : ['d.grid'],
  1077. "group" : self.inGroup,
  1078. "display" : 0})
  1079. elif element == 'gridcolor':
  1080. value = self._get_value(line)
  1081. if value != '':
  1082. self.layers[-1]['cmd'].append('color=%s' % self._color_name_to_rgb(value))
  1083. elif element == 'gridborder':
  1084. value = self._get_value(line)
  1085. if value != '':
  1086. self.layers[-1]['cmd'].append('bordercolor=%s' % self._color_name_to_rgb(value))
  1087. elif element == 'textcolor':
  1088. value = self._get_value(line)
  1089. if value != '':
  1090. self.layers[-1]['cmd'].append('textcolor=%s' % self._color_name_to_rgb(value))
  1091. elif element in ('gridsize',
  1092. 'gridorigin'):
  1093. value = self._get_value(line)
  1094. if value != '':
  1095. self.layers[-1]['cmd'].append('%s=%s' % (element[4:], value))
  1096. elif element in 'fontsize':
  1097. value = self._get_value(line)
  1098. if value != '':
  1099. self.layers[-1]['cmd'].append('%s=%s' % (element, value))
  1100. elif element == 'griddraw':
  1101. value = self._get_value(line)
  1102. if value == '0':
  1103. self.layers[-1]['cmd'].append('-n')
  1104. elif element == 'gridgeo':
  1105. value = self._get_value(line)
  1106. if value == '1':
  1107. self.layers[-1]['cmd'].append('-g')
  1108. elif element == 'borderdraw':
  1109. value = self._get_value(line)
  1110. if value == '0':
  1111. self.layers[-1]['cmd'].append('-b')
  1112. elif element == 'textdraw':
  1113. value = self._get_value(line)
  1114. if value == '0':
  1115. self.layers[-1]['cmd'].append('-t')
  1116. else:
  1117. self.error += _(' row %d:') % line_id + line + os.linesep
  1118. self.num_error += 1
  1119. def _get_value(self, line):
  1120. """!Get value of element"""
  1121. try:
  1122. return line.strip(' ').split(' ')[1].strip(' ')
  1123. except:
  1124. return ''
  1125. def _get_element(self, line):
  1126. """!Get element tag"""
  1127. return line.strip(' ').split(' ')[0].strip(' ')
  1128. def _get_cmd_param_index(self, cmd, name):
  1129. """!Get index of parameter in cmd list
  1130. @param cmd cmd list
  1131. @param name parameter name
  1132. @return index
  1133. @return -1 if not found
  1134. """
  1135. i = 0
  1136. for param in cmd:
  1137. if '=' not in param:
  1138. i += 1
  1139. continue
  1140. if param.split('=')[0] == name:
  1141. return i
  1142. i += 1
  1143. return -1
  1144. def _color_name_to_rgb(self, value):
  1145. """!Convert color name (#) to rgb values"""
  1146. col = wx.NamedColour(value)
  1147. return str(col.Red()) + ':' + \
  1148. str(col.Green()) + ':' + \
  1149. str(col.Blue())