workspace.py 68 KB

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