workspace.py 67 KB

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