workspace.py 68 KB

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