workspace.py 62 KB

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