workspace.py 67 KB

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