workspace.py 69 KB

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