workspace.py 63 KB

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