123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638 |
- """!
- @package workspace.py
- @brief Open/save workspace definition file
- Classes:
- - ProcessWorkspaceFile
- - Nviz
- - WriteWorkspaceFile
- - ProcessGrcFile
- (C) 2007-2011 by the GRASS Development Team
- This program is free software under the GNU General Public
- License (>=v2). Read the file COPYING that comes with GRASS
- for details.
- @author Martin Landa <landa.martin gmail.com>
- @author Anna Kratochvilova <kratochanna gmail.com> (wxNviz / Google SoC 2011)
- """
- import os
- import sys
- import copy
- import types
- import wx
- ### for gxw (workspace file) parsering
- # xmlproc not available on Mac OS
- # from xml.parsers.xmlproc import xmlproc
- # from xml.parsers.xmlproc import xmlval
- # from xml.parsers.xmlproc import xmldtd
- try:
- import xml.etree.ElementTree as etree
- except ImportError:
- import elementtree.ElementTree as etree # Python <= 2.4
- import utils
- import globalvar
- from preferences import globalSettings as UserSettings
- try:
- import wxnviz
- except ImportError:
- wxnviz = None
- class ProcessWorkspaceFile:
- def __init__(self, tree):
- """!A ElementTree handler for the GXW XML file, as defined in
- grass-gxw.dtd.
- """
- self.tree = tree
- self.root = self.tree.getroot()
-
- #
- # layer manager properties
- #
- self.layerManager = {}
- self.layerManager['pos'] = None # window position
- self.layerManager['size'] = None # window size
-
- #
- # list of mapdisplays
- #
- self.displays = []
- #
- # list of map layers
- #
- self.layers = []
- #
- # nviz state
- #
- self.nviz_state = {}
-
- self.displayIndex = -1 # first display has index '0'
-
- self.__processFile()
-
- self.nvizDefault = Nviz()
-
- def __filterValue(self, value):
- """!Filter value
-
- @param value
- """
- value = value.replace('<', '<')
- value = value.replace('>', '>')
-
- return value
- def __getNodeText(self, node, tag, default = ''):
- """!Get node text"""
- p = node.find(tag)
- if p is not None:
- return utils.normalize_whitespace(p.text)
-
- return default
-
- def __processFile(self):
- """!Process workspace file"""
- #
- # layer manager
- #
- node_lm = self.root.find('layer_manager')
- if node_lm is not None:
- posAttr = node_lm.get('dim', '')
- if posAttr:
- posVal = map(int, posAttr.split(','))
- try:
- self.layerManager['pos'] = (posVal[0], posVal[1])
- self.layerManager['size'] = (posVal[2], posVal[3])
- except:
- pass
-
- #
- # displays
- #
- for display in self.root.findall('display'):
- self.displayIndex += 1
-
- # window position and size
- posAttr = display.get('dim', '')
- if posAttr:
- posVal = map(int, posAttr.split(','))
- try:
- pos = (posVal[0], posVal[1])
- size = (posVal[2], posVal[3])
- except:
- pos = None
- size = None
- else:
- pos = None
- size = None
-
- extentAttr = display.get('extent', '')
- if extentAttr:
- # w, s, e, n
- extent = map(float, extentAttr.split(','))
- else:
- extent = None
-
- # projection
- node_projection = display.find('projection')
- if node_projection is not None:
- projection = { 'enabled' : True,
- 'epsg' : node_projection.get('epsg', ''),
- 'proj' : self.__getNodeText(node_projection, 'value') }
- else:
- projection = { 'enabled' : False }
-
- self.displays.append( {
- "render" : bool(int(display.get('render', "0"))),
- "mode" : int(display.get('mode', 0)),
- "showCompExtent" : bool(int(display.get('showCompExtent', "0"))),
- "pos" : pos,
- "size" : size,
- "extent" : extent,
- "alignExtent" : bool(int(display.get('alignExtent', "0"))),
- "constrainRes" : bool(int(display.get('constrainRes', "0"))),
- "projection" : projection,
- "viewMode" : display.get('viewMode', '2d')} )
-
- # process all layers/groups in the display
- self.__processLayers(display)
- # process nviz_state
- self.__processNvizState(display)
- def __processLayers(self, node, inGroup = -1):
- """!Process layers/groups of selected display
-
- @param node display tree node
- @param inGroup in group -> index of group item otherwise -1
- """
- for item in node.getchildren():
- if item.tag == 'group':
- # -> group
- self.layers.append( {
- "type" : 'group',
- "name" : item.get('name', ''),
- "checked" : bool(int(item.get('checked', "0"))),
- "opacity" : None,
- "cmd" : None,
- "group" : inGroup,
- "display" : self.displayIndex,
- "vdigit" : None,
- "nviz" : None})
-
- self.__processLayers(item, inGroup = len(self.layers) - 1) # process items in group
-
- elif item.tag == 'layer':
- cmd, selected, vdigit, nviz = self.__processLayer(item)
-
- self.layers.append( {
- "type" : item.get('type', None),
- "name" : item.get('name', None),
- "checked" : bool(int(item.get('checked', "0"))),
- "opacity" : float(item.get('opacity', '1.0')),
- "cmd" : cmd,
- "group" : inGroup,
- "display" : self.displayIndex,
- "selected" : selected,
- "vdigit" : vdigit,
- "nviz" : nviz } )
-
- def __processLayer(self, layer):
- """!Process layer item
- @param layer tree node
- """
- cmd = list()
-
- #
- # layer attributes (task) - 2D settings
- #
- node_task = layer.find('task')
- cmd.append(node_task.get('name', "unknown"))
-
- # flags
- for p in node_task.findall('flag'):
- flag = p.get('name', '')
- if len(flag) > 1:
- cmd.append('--' + flag)
- else:
- cmd.append('-' + flag)
-
- # parameters
- for p in node_task.findall('parameter'):
- cmd.append('%s=%s' % (p.get('name', ''),
- self.__filterValue(self.__getNodeText(p, 'value'))))
-
- if layer.find('selected') is not None:
- selected = True
- else:
- selected = False
-
- #
- # Vector digitizer settings
- #
- node_vdigit = layer.find('vdigit')
- if node_vdigit is not None:
- vdigit = self.__processLayerVdigit(node_vdigit)
- else:
- vdigit = None
-
- #
- # Nviz (3D settings)
- #
- node_nviz = layer.find('nviz')
- if node_nviz is not None:
- nviz = self.__processLayerNviz(node_nviz)
- else:
- nviz = None
-
- return (cmd, selected, vdigit, nviz)
- def __processLayerVdigit(self, node_vdigit):
- """!Process vector digitizer layer settings
- @param node_vdigit vdigit node
- """
- # init nviz layer properties
- vdigit = dict()
- for node in node_vdigit.findall('geometryAttribute'):
- if 'geomAttr' not in vdigit:
- vdigit['geomAttr'] = dict()
- type = node.get('type')
- vdigit['geomAttr'][type] = dict()
- vdigit['geomAttr'][type]['column'] = node.get('column') # required
- # default map units
- vdigit['geomAttr'][type]['units'] = node.get('units', 'mu')
-
- return vdigit
-
- def __processLayerNviz(self, node_nviz):
- """!Process 3D layer settings
- @param node_nviz nviz node
- """
- # init nviz layer properties
- nviz = {}
- if node_nviz.find('surface') is not None: # -> raster
- nviz['surface'] = {}
- for sec in ('attribute', 'draw', 'mask', 'position'):
- nviz['surface'][sec] = {}
- elif node_nviz.find('vlines') is not None or \
- node_nviz.find('vpoints') is not None: # -> vector
- nviz['vector'] = {}
- for sec in ('lines', 'points'):
- nviz['vector'][sec] = {}
-
- if 'surface' in nviz:
- node_surface = node_nviz.find('surface')
- # attributes
- for attrb in node_surface.findall('attribute'):
- tagName = str(attrb.tag)
- attrbName = attrb.get('name', '')
- dc = nviz['surface'][tagName][attrbName] = {}
- if attrb.get('map', '0') == '0':
- dc['map'] = False
- else:
- dc['map'] = True
- value = self.__getNodeText(attrb, 'value')
- try:
- dc['value'] = int(value)
- except ValueError:
- try:
- dc['value'] = float(value)
- except ValueError:
- dc['value'] = str(value)
-
- # draw
- node_draw = node_surface.find('draw')
- if node_draw is not None:
- tagName = str(node_draw.tag)
- nviz['surface'][tagName]['all'] = False
- nviz['surface'][tagName]['mode'] = {}
- nviz['surface'][tagName]['mode']['value'] = -1 # to be calculated
- nviz['surface'][tagName]['mode']['desc'] = {}
- nviz['surface'][tagName]['mode']['desc']['shading'] = \
- str(node_draw.get('shading', ''))
- nviz['surface'][tagName]['mode']['desc']['style'] = \
- str(node_draw.get('style', ''))
- nviz['surface'][tagName]['mode']['desc']['mode'] = \
- str(node_draw.get('mode', ''))
-
- # resolution
- for node_res in node_draw.findall('resolution'):
- resType = str(node_res.get('type', ''))
- if 'resolution' not in nviz['surface']['draw']:
- nviz['surface']['draw']['resolution'] = {}
- value = int(self.__getNodeText(node_res, 'value'))
- nviz['surface']['draw']['resolution'][resType] = value
-
- # wire-color
- node_wire_color = node_draw.find('wire_color')
- if node_wire_color is not None:
- nviz['surface']['draw']['wire-color'] = {}
- value = str(self.__getNodeText(node_wire_color, 'value'))
- nviz['surface']['draw']['wire-color']['value'] = value
-
- # position
- node_pos = node_surface.find('position')
- if node_pos is not None:
- dc = nviz['surface']['position'] = {}
- for coor in ['x', 'y', 'z']:
- node = node_pos.find(coor)
- if node is None:
- continue
- value = int(self.__getNodeText(node_pos, coor))
- dc[coor] = value
-
- elif 'vector' in nviz:
- # vpoints
- node_vpoints = node_nviz.find('vpoints')
- if node_vpoints is not None:
- marker = str(node_vpoints.get('marker', ''))
- markerId = list(UserSettings.Get(group='nviz', key='vector',
- subkey=['points', 'marker'], internal=True)).index(marker)
- nviz['vector']['points']['marker'] = { 'value' : markerId }
-
- node_mode = node_vpoints.find('mode')
- if node_mode is not None:
- nviz['vector']['points']['mode'] = {}
- nviz['vector']['points']['mode']['type'] = str(node_mode.get('type', 'surface'))
- nviz['vector']['points']['mode']['surface'] = {}
- nviz['vector']['points']['mode']['surface']['value'] = []
- nviz['vector']['points']['mode']['surface']['show'] = []
-
- # map
- for node_map in node_mode.findall('map'):
- nviz['vector']['points']['mode']['surface']['value'].append(
- self.__processLayerNvizNode(node_map, 'name', str))
- nviz['vector']['points']['mode']['surface']['show'].append(bool(
- self.__processLayerNvizNode(node_map, 'checked', int)))
-
- # color
- self.__processLayerNvizNode(node_vpoints, 'color', str,
- nviz['vector']['points'])
-
- # width
- self.__processLayerNvizNode(node_vpoints, 'width', int,
- nviz['vector']['points'])
-
- # height
- self.__processLayerNvizNode(node_vpoints, 'height', int,
- nviz['vector']['points'])
-
- # height
- self.__processLayerNvizNode(node_vpoints, 'size', int,
- nviz['vector']['points'])
-
- # thematic
- node_thematic = node_vpoints.find('thematic')
- thematic = nviz['vector']['points']['thematic'] = {}
- thematic['rgbcolumn'] = self.__processLayerNvizNode(node_thematic, 'rgbcolumn', str)
- thematic['sizecolumn'] = self.__processLayerNvizNode(node_thematic, 'sizecolumn', str)
- for col in ('rgbcolumn', 'sizecolumn'):
- if thematic[col] == 'None':
- thematic[col] = None
- thematic['layer'] = self.__processLayerNvizNode(node_thematic, 'layer', int)
- for use in ('usecolor', 'usesize', 'usewidth'):
- if node_thematic.get(use, ''):
- thematic[use] = int(node_thematic.get(use, '0'))
-
- # vlines
- node_vlines = node_nviz.find('vlines')
- if node_vlines is not None:
- node_mode = node_vlines.find('mode')
- if node_mode is not None:
- nviz['vector']['lines']['mode'] = {}
- nviz['vector']['lines']['mode']['type'] = str(node_mode.get('type', ''))
- nviz['vector']['lines']['mode']['surface'] = {}
- nviz['vector']['lines']['mode']['surface']['value'] = []
- nviz['vector']['lines']['mode']['surface']['show'] = []
-
- # map
- for node_map in node_mode.findall('map'):
- nviz['vector']['lines']['mode']['surface']['value'].append(
- self.__processLayerNvizNode(node_map, 'name', str))
- nviz['vector']['lines']['mode']['surface']['show'].append(bool(
- self.__processLayerNvizNode(node_map, 'checked', int)))
-
- # color
- self.__processLayerNvizNode(node_vlines, 'color', str,
- nviz['vector']['lines'])
-
- # width
- self.__processLayerNvizNode(node_vlines, 'width', int,
- nviz['vector']['lines'])
-
- # height
- self.__processLayerNvizNode(node_vlines, 'height', int,
- nviz['vector']['lines'])
-
- # thematic
- node_thematic = node_vlines.find('thematic')
- thematic = nviz['vector']['lines']['thematic'] = {}
- thematic['rgbcolumn'] = self.__processLayerNvizNode(node_thematic, 'rgbcolumn', str)
- thematic['sizecolumn'] = self.__processLayerNvizNode(node_thematic, 'sizecolumn', str)
- for col in ('rgbcolumn', 'sizecolumn'):
- if thematic[col] == 'None':
- thematic[col] = None
- thematic['layer'] = self.__processLayerNvizNode(node_thematic, 'layer', int)
- for use in ('usecolor', 'usesize', 'usewidth'):
- if node_thematic.get(use, ''):
- thematic[use] = int(node_thematic.get(use, '0'))
-
- return nviz
-
- def __processLayerNvizNode(self, node, tag, cast, dc = None):
- """!Process given tag nviz/vector"""
- node_tag = node.find(tag)
- if node_tag is not None:
- if node_tag.find('value') is not None:
- value = cast(self.__getNodeText(node_tag, 'value'))
- else:
- try:
- value = cast(node_tag.text)
- except ValueError:
- if cast == str:
- value = ''
- else:
- value = None
- if dc:
- dc[tag] = dict()
- dc[tag]['value'] = value
- else:
- return value
-
- def __processNvizState(self, node):
- """!Process tag nviz_state"""
- node_state = node.find('nviz_state')
- if node_state is None:
- return
- self.nviz_state['display'] = self.displayIndex
- #
- # view
- #
- node_view = node_state.find('view')
- view = {}
- iview = {}
-
- node_position = node_view.find('v_position')
- view['position'] = {}
- view['position']['x'] = self.__processLayerNvizNode(node_position, 'x', float)
- view['position']['y'] = self.__processLayerNvizNode(node_position, 'y', float)
- node_persp = node_view.find('persp')
- view['persp'] = {}
- iview['persp'] = {}
- view['persp']['value'] = self.__processLayerNvizNode(node_persp, 'value', int)
- view['persp']['step'] = self.__processLayerNvizNode(node_persp, 'step', int)
- iview['persp']['min'] = self.__processLayerNvizNode(node_persp, 'min', int)
- iview['persp']['max'] = self.__processLayerNvizNode(node_persp, 'max', int)
- node_height = node_view.find('v_height')
- iview['height'] = {}
- iview['height']['value'] = self.__processLayerNvizNode(node_height, 'value', int)
- iview['height']['min'] = self.__processLayerNvizNode(node_height, 'min', int)
- iview['height']['max'] = self.__processLayerNvizNode(node_height, 'max', int)
- node_twist = node_view.find('twist')
- view['twist'] = {}
- iview['twist'] = {}
- view['twist']['value'] = self.__processLayerNvizNode(node_twist, 'value', int)
- iview['twist']['min'] = self.__processLayerNvizNode(node_twist, 'min', int)
- iview['twist']['max'] = self.__processLayerNvizNode(node_twist, 'max', int)
- node_zexag = node_view.find('z-exag')
- view['z-exag'] = {}
- iview['z-exag'] = {}
- view['z-exag']['value'] = self.__processLayerNvizNode(node_zexag, 'value', int)
- view['z-exag']['min'] = self.__processLayerNvizNode(node_zexag, 'min', int)
- view['z-exag']['max'] = self.__processLayerNvizNode(node_zexag, 'max', int)
- iview['z-exag']['original'] = self.__processLayerNvizNode(node_zexag, 'original', float)
- node_focus = node_view.find('focus')
- iview['focus'] = {}
- iview['focus']['x'] = self.__processLayerNvizNode(node_focus, 'x', int)
- iview['focus']['y'] = self.__processLayerNvizNode(node_focus, 'y', int)
- iview['focus']['z'] = self.__processLayerNvizNode(node_focus, 'z', int)
- node_dir = node_view.find('dir')
- if node_dir:
- iview['dir'] = {}
- iview['dir']['x'] = self.__processLayerNvizNode(node_dir, 'x', int)
- iview['dir']['y'] = self.__processLayerNvizNode(node_dir, 'y', int)
- iview['dir']['z'] = self.__processLayerNvizNode(node_dir, 'z', int)
- iview['dir']['use'] = True
- else:
- iview['dir'] = {}
- iview['dir']['x'] = -1
- iview['dir']['y'] = -1
- iview['dir']['z'] = -1
- iview['dir']['use'] = False
-
- view['background'] = {}
- color = self.__processLayerNvizNode(node_view, 'background_color', str)
- view['background']['color'] = tuple(map(int, color.split(':')))
-
- self.nviz_state['view'] = view
- self.nviz_state['iview'] = iview
- #
- # light
- #
- node_light = node_state.find('light')
- light = {}
-
- node_position = node_light.find('l_position')
- light['position'] = {}
- light['position']['x'] = self.__processLayerNvizNode(node_position, 'x', float)
- light['position']['y'] = self.__processLayerNvizNode(node_position, 'y', float)
- light['position']['z'] = self.__processLayerNvizNode(node_position, 'z', int)
-
- light['bright'] = self.__processLayerNvizNode(node_light, 'bright', int)
- light['ambient'] = self.__processLayerNvizNode(node_light, 'ambient', int)
- color = self.__processLayerNvizNode(node_light, 'color', str)
- light['color'] = tuple(map(int, color.split(':')))
-
- self.nviz_state['light'] = light
-
- node_constants = node_state.find('constant_planes')
- constants = []
- if node_constants:
- for i, node_plane in enumerate(node_constants.findall('plane')):
- plane = {}
- plane['color'] = self.__processLayerNvizNode(node_plane, 'color', str)
- plane['resolution'] = self.__processLayerNvizNode(node_plane, 'fine_resolution', int)
- plane['value'] = self.__processLayerNvizNode(node_plane, 'height', int)
- plane['object'] = {}
- constants.append({'constant': plane})
- self.nviz_state['constants'] = constants
-
- class Nviz:
- def __init__(self):
- """Default 3D settings"""
- UserSettings.Reset('nviz')
- UserSettings.ReadSettingsFile()
-
- def SetConstantDefaultProp(self):
- """Set default constant data properties"""
- data = dict()
- for key, value in UserSettings.Get(group='nviz', key='constant').iteritems():
- data[key] = value
- color = str(data['color'][0]) + ':' + str(data['color'][1]) + ':' + str(data['color'][2])
- data['color'] = color
- return data
-
- def SetSurfaceDefaultProp(self, data = None):
- """Set default surface data properties"""
- if not data:
- data = dict()
- for sec in ('attribute', 'draw', 'mask', 'position'):
- data[sec] = {}
-
- #
- # attributes
- #
- for attrb in ('shine', ):
- data['attribute'][attrb] = {}
- for key, value in UserSettings.Get(group='nviz', key='surface',
- subkey=attrb).iteritems():
- data['attribute'][attrb][key] = value
- data['attribute'][attrb]['update'] = None
-
- #
- # draw
- #
- data['draw']['all'] = False # apply only for current surface
- for control, value in UserSettings.Get(group='nviz', key='surface', subkey='draw').iteritems():
- if control[:3] == 'res':
- if 'resolution' not in data['draw']:
- data['draw']['resolution'] = {}
- if 'update' not in data['draw']['resolution']:
- data['draw']['resolution']['update'] = None
- data['draw']['resolution'][control[4:]] = value
- continue
-
- if control == 'wire-color':
- value = str(value[0]) + ':' + str(value[1]) + ':' + str(value[2])
- elif control in ('mode', 'style', 'shading'):
- if 'mode' not in data['draw']:
- data['draw']['mode'] = {}
- continue
- data['draw'][control] = { 'value' : value }
- data['draw'][control]['update'] = None
-
- value, desc = self.GetDrawMode(UserSettings.Get(group='nviz', key='surface', subkey=['draw', 'mode']),
- UserSettings.Get(group='nviz', key='surface', subkey=['draw', 'style']),
- UserSettings.Get(group='nviz', key='surface', subkey=['draw', 'shading']))
-
- data['draw']['mode'] = { 'value' : value,
- 'desc' : desc,
- 'update': None }
- # position
- for coord in ('x', 'y', 'z'):
- data['position'][coord] = UserSettings.Get(group='nviz', key='surface', subkey=['position', coord])
- data['position']['update'] = None
-
- return data
-
- def SetVolumeDefaultProp(self):
- """Set default volume data properties"""
- data = dict()
- for sec in ('attribute', 'draw', 'position'):
- data[sec] = dict()
- for sec in ('isosurface', 'slice'):
- data[sec] = list()
-
- #
- # draw
- #
- for control, value in UserSettings.Get(group='nviz', key='volume', subkey='draw').iteritems():
- if control == 'shading':
- sel = UserSettings.Get(group='nviz', key='volume', subkey=['draw', 'shading'])
- value, desc = self.GetDrawMode(shade=sel, string=False)
-
- data['draw']['shading'] = {}
- data['draw']['shading']['isosurface'] = { 'value' : value,
- 'desc' : desc['shading'] }
- data['draw']['shading']['slice'] = { 'value' : value,
- 'desc' : desc['shading'] }
- elif control == 'mode':
- sel = UserSettings.Get(group='nviz', key='volume', subkey=['draw', 'mode'])
- if sel == 0:
- desc = 'isosurface'
- else:
- desc = 'slice'
- data['draw']['mode'] = { 'value' : sel,
- 'desc' : desc, }
- else:
- data['draw'][control] = {}
- data['draw'][control]['isosurface'] = { 'value' : value }
- data['draw'][control]['slice'] = { 'value' : value }
- if 'update' not in data['draw'][control]:
- data['draw'][control]['update'] = None
-
- #
- # isosurface attributes
- #
- for attrb in ('shine', ):
- data['attribute'][attrb] = {}
- for key, value in UserSettings.Get(group='nviz', key='volume',
- subkey=attrb).iteritems():
- data['attribute'][attrb][key] = value
-
- return data
-
- def SetIsosurfaceDefaultProp(self):
- """!Set default isosurface properties"""
- data = dict()
- for attr in ('shine', 'topo', 'transp', 'color'):
- data[attr] = {}
- for key, value in UserSettings.Get(group = 'nviz', key = 'volume',
- subkey = attr).iteritems():
- data[attr][key] = value
- data[attr]['update'] = None
- return data
-
- def SetSliceDefaultProp(self):
- """!Set default slice properties"""
- data = dict()
- data['position'] = copy.deepcopy(UserSettings.Get(group = 'nviz', key = 'volume',
- subkey = 'slice_position'))
- data['position']['update'] = None
-
- data['transp'] = copy.deepcopy(UserSettings.Get(group = 'nviz', key = 'volume',
- subkey = 'transp'))
- return data
-
- def SetVectorDefaultProp(self, data = None):
- """Set default vector data properties"""
- if not data:
- data = dict()
- for sec in ('lines', 'points'):
- data[sec] = {}
-
- self.SetVectorLinesDefaultProp(data['lines'])
- self.SetVectorPointsDefaultProp(data['points'])
- return data
-
- def SetVectorLinesDefaultProp(self, data):
- """Set default vector properties -- lines"""
- # width
- data['width'] = {'value' : UserSettings.Get(group='nviz', key='vector',
- subkey=['lines', 'width']) }
-
- # color
- value = UserSettings.Get(group='nviz', key='vector',
- subkey=['lines', 'color'])
- color = str(value[0]) + ':' + str(value[1]) + ':' + str(value[2])
- data['color'] = { 'value' : color }
- # mode
- if UserSettings.Get(group='nviz', key='vector',
- subkey=['lines', 'flat']):
- type = 'flat'
-
- else:
- type = 'surface'
-
- data['mode'] = {}
- data['mode']['type'] = type
- data['mode']['update'] = None
-
- # height
- data['height'] = { 'value' : UserSettings.Get(group='nviz', key='vector',
- subkey=['lines', 'height']) }
- # thematic
- data['thematic'] = {'rgbcolumn' : UserSettings.Get(group='nviz', key='vector',
- subkey=['lines', 'rgbcolumn']),
- 'sizecolumn' : UserSettings.Get(group='nviz', key='vector',
- subkey=['lines', 'sizecolumn']),
- 'layer': 1,
- 'usecolor' : False,
- 'usewidth' : False}
- if 'object' in data:
- for attrb in ('color', 'width', 'mode', 'height', 'thematic'):
- data[attrb]['update'] = None
-
- def SetVectorPointsDefaultProp(self, data):
- """Set default vector properties -- points"""
- # size
- data['size'] = { 'value' : UserSettings.Get(group='nviz', key='vector',
- subkey=['points', 'size']) }
- # width
- data['width'] = { 'value' : UserSettings.Get(group='nviz', key='vector',
- subkey=['points', 'width']) }
- # marker
- data['marker'] = { 'value' : UserSettings.Get(group='nviz', key='vector',
- subkey=['points', 'marker']) }
- # color
- value = UserSettings.Get(group='nviz', key='vector',
- subkey=['points', 'color'])
- color = str(value[0]) + ':' + str(value[1]) + ':' + str(value[2])
- data['color'] = { 'value' : color }
- # mode
- data['mode'] = { 'type' : 'surface'}
- ## 'surface' : '', }
-
- # height
- data['height'] = { 'value' : UserSettings.Get(group='nviz', key='vector',
- subkey=['points', 'height']) }
-
- data['thematic'] = {'rgbcolumn' : UserSettings.Get(group='nviz', key='vector',
- subkey=['points', 'rgbcolumn']),
- 'sizecolumn' : UserSettings.Get(group='nviz', key='vector',
- subkey=['points', 'sizecolumn']),
- 'layer': 1,
- 'usecolor' : False,
- 'usesize' : False}
- if 'object' in data:
- for attrb in ('size', 'width', 'marker',
- 'color', 'height', 'thematic'):
- data[attrb]['update'] = None
-
- def GetDrawMode(self, mode=None, style=None, shade=None, string=False):
- """Get surface draw mode (value) from description/selection
- @param mode,style,shade modes
- @param string if True input parameters are strings otherwise
- selections
- """
- if not wxnviz:
- return None
-
- value = 0
- desc = {}
- if string:
- if mode is not None:
- if mode == 'coarse':
- value |= wxnviz.DM_WIRE
- elif mode == 'fine':
- value |= wxnviz.DM_POLY
- else: # both
- value |= wxnviz.DM_WIRE_POLY
- if style is not None:
- if style == 'wire':
- value |= wxnviz.DM_GRID_WIRE
- else: # surface
- value |= wxnviz.DM_GRID_SURF
-
- if shade is not None:
- if shade == 'flat':
- value |= wxnviz.DM_FLAT
- else: # surface
- value |= wxnviz.DM_GOURAUD
- return value
- # -> string is False
- if mode is not None:
- if mode == 0: # coarse
- value |= wxnviz.DM_WIRE
- desc['mode'] = 'coarse'
- elif mode == 1: # fine
- value |= wxnviz.DM_POLY
- desc['mode'] = 'fine'
- else: # both
- value |= wxnviz.DM_WIRE_POLY
- desc['mode'] = 'both'
- if style is not None:
- if style == 0: # wire
- value |= wxnviz.DM_GRID_WIRE
- desc['style'] = 'wire'
- else: # surface
- value |= wxnviz.DM_GRID_SURF
- desc['style'] = 'surface'
- if shade is not None:
- if shade == 0:
- value |= wxnviz.DM_FLAT
- desc['shading'] = 'flat'
- else: # surface
- value |= wxnviz.DM_GOURAUD
- desc['shading'] = 'gouraud'
-
- return (value, desc)
-
- def SetDecorDefaultProp(self, type):
- """!Set default arrow properties
- """
- data = {}
-
- # arrow
- if type == 'arrow':
- data['arrow'] = UserSettings.Get(group = 'nviz', key = 'arrow')
- data['arrow']['color'] = "%d:%d:%d" % (
- UserSettings.Get(group = 'nviz', key = 'arrow', subkey = 'color')[:3])
- data['arrow'].update(UserSettings.Get(group = 'nviz', key = 'arrow', internal = True))
- data['arrow']['show'] = False
-
- # arrow
- if type == 'scalebar':
- data['scalebar'] = copy.deepcopy(UserSettings.Get(group = 'nviz', key = 'scalebar'))
- data['scalebar']['color'] = "%d:%d:%d" % (
- UserSettings.Get(group = 'nviz', key = 'scalebar', subkey = 'color')[:3])
- data['scalebar'].update(copy.deepcopy(UserSettings.Get(group = 'nviz', key = 'scalebar', internal = True)))
- data['scalebar']['id'] = 0
- return data
-
- class WriteWorkspaceFile(object):
- """!Generic class for writing workspace file"""
- def __init__(self, lmgr, file):
- self.file = file
- self.lmgr = lmgr
- self.indent = 0
-
- # write header
- self.file.write('<?xml version="1.0" encoding="UTF-8"?>\n')
- self.file.write('<!DOCTYPE gxw SYSTEM "grass-gxw.dtd">\n')
- self.file.write('%s<gxw>\n' % (' ' * self.indent))
-
- self.indent =+ 4
-
- # layer manager
- windowPos = self.lmgr.GetPosition()
- windowSize = self.lmgr.GetSize()
- file.write('%s<layer_manager dim="%d,%d,%d,%d">\n' % (' ' * self.indent,
- windowPos[0],
- windowPos[1],
- windowSize[0],
- windowSize[1]
- ))
-
- file.write('%s</layer_manager>\n' % (' ' * self.indent))
-
- # list of displays
- for page in range(0, self.lmgr.gm_cb.GetPageCount()):
- mapTree = self.lmgr.gm_cb.GetPage(page).maptree
- region = mapTree.Map.region
-
- displayPos = mapTree.mapdisplay.GetPosition()
- displaySize = mapTree.mapdisplay.GetSize()
- if mapTree.mapdisplay.toolbars['map'].combo.GetSelection() == 1:
- viewmode = '3d'
- else:
- viewmode = '2d'
-
- file.write('%s<display render="%d" '
- 'mode="%d" showCompExtent="%d" '
- 'alignExtent="%d" '
- 'constrainRes="%d" '
- 'dim="%d,%d,%d,%d" '
- 'extent="%f,%f,%f,%f" '
- 'viewMode="%s" >\n' % (' ' * self.indent,
- int(mapTree.mapdisplay.GetProperty('render')),
- mapTree.mapdisplay.statusbarManager.GetMode(),
- int(mapTree.mapdisplay.GetProperty('region')),
- int(mapTree.mapdisplay.GetProperty('alignExtent')),
- int(mapTree.mapdisplay.GetProperty('resolution')),
- displayPos[0],
- displayPos[1],
- displaySize[0],
- displaySize[1],
- region['w'],
- region['s'],
- region['e'],
- region['n'],
- viewmode
- ))
- # projection statusbar info
- if mapTree.mapdisplay.GetProperty('projection') and \
- UserSettings.Get(group='display', key='projection', subkey='proj4'):
- self.indent += 4
- file.write('%s<projection' % (' ' * self.indent))
- epsg = UserSettings.Get(group='display', key='projection', subkey='epsg')
- if epsg:
- file.write(' epsg="%s"' % epsg)
- file.write('>\n')
- proj = UserSettings.Get(group='display', key='projection', subkey='proj4')
- self.indent += 4
- file.write('%s<value>%s</value>\n' % (' ' * self.indent, proj))
- self.indent -= 4
- file.write('%s</projection>\n' % (' ' * self.indent))
- self.indent -= 4
-
- # list of layers
- item = mapTree.GetFirstChild(mapTree.root)[0]
- self.__writeLayer(mapTree, item)
-
- if mapTree.mapdisplay.MapWindow3D is not None:
- nvizDisp = mapTree.mapdisplay.MapWindow3D
- self.__writeNvizState(view = nvizDisp.view, iview = nvizDisp.iview,
- light = nvizDisp.light, constants = nvizDisp.constants)
-
- file.write('%s</display>\n' % (' ' * self.indent))
-
- self.indent =- 4
- file.write('%s</gxw>\n' % (' ' * self.indent))
- def __filterValue(self, value):
- """!Make value XML-valid"""
- value = value.replace('<', '<')
- value = value.replace('>', '>')
-
- return value
-
- def __writeLayer(self, mapTree, item):
- """!Write bunch of layers to GRASS Workspace XML file"""
- self.indent += 4
- itemSelected = mapTree.GetSelections()
- while item and item.IsOk():
- type = mapTree.GetPyData(item)[0]['type']
- if type != 'group':
- maplayer = mapTree.GetPyData(item)[0]['maplayer']
- else:
- maplayer = None
- checked = int(item.IsChecked())
- if type == 'command':
- cmd = mapTree.GetPyData(item)[0]['maplayer'].GetCmd(string=True)
- self.file.write('%s<layer type="%s" name="%s" checked="%d">\n' % \
- (' ' * self.indent, type, cmd, checked));
- self.file.write('%s</layer>\n' % (' ' * self.indent));
- elif type == 'group':
- name = mapTree.GetItemText(item)
- self.file.write('%s<group name="%s" checked="%d">\n' % \
- (' ' * self.indent, name.encode('utf8'), checked));
- self.indent += 4
- subItem = mapTree.GetFirstChild(item)[0]
- self.__writeLayer(mapTree, subItem)
- self.indent -= 4
- self.file.write('%s</group>\n' % (' ' * self.indent));
- else:
- cmd = mapTree.GetPyData(item)[0]['maplayer'].GetCmd(string=False)
- name = mapTree.GetItemText(item)
- opacity = maplayer.GetOpacity(float = True)
- # remove 'opacity' part
- if opacity < 1:
- name = name.split('(', -1)[0].strip()
- self.file.write('%s<layer type="%s" name="%s" checked="%d" opacity="%f">\n' % \
- (' ' * self.indent, type, name.encode('utf8'), checked, opacity));
-
- self.indent += 4
- # selected ?
- if item in itemSelected:
- self.file.write('%s<selected />\n' % (' ' * self.indent))
- # layer properties
- self.file.write('%s<task name="%s">\n' % (' ' * self.indent, cmd[0]))
- self.indent += 4
- for key, val in cmd[1].iteritems():
- if key == 'flags':
- for f in val:
- self.file.write('%s<flag name="%s" />\n' %
- (' ' * self.indent, f))
- elif val in (True, False):
- self.file.write('%s<flag name="%s" />\n' %
- (' ' * self.indent, key))
- else: # parameter
- self.file.write('%s<parameter name="%s">\n' %
- (' ' * self.indent, key))
- self.indent += 4
- self.file.write('%s<value>%s</value>\n' %
- (' ' * self.indent, self.__filterValue(val)))
- self.indent -= 4
- self.file.write('%s</parameter>\n' % (' ' * self.indent));
- self.indent -= 4
- self.file.write('%s</task>\n' % (' ' * self.indent));
- # vector digitizer
- vdigit = mapTree.GetPyData(item)[0]['vdigit']
- if vdigit:
- self.file.write('%s<vdigit>\n' % (' ' * self.indent))
- if 'geomAttr' in vdigit:
- self.indent += 4
- for type, val in vdigit['geomAttr'].iteritems():
- units = ''
- if val['units'] != 'mu':
- units = ' units="%s"' % val['units']
- self.file.write('%s<geometryAttribute type="%s" column="%s"%s />\n' % \
- (' ' * self.indent, type, val['column'], units))
- self.indent -= 4
- self.file.write('%s</vdigit>\n' % (' ' * self.indent))
- # nviz
- nviz = mapTree.GetPyData(item)[0]['nviz']
- if nviz:
- self.file.write('%s<nviz>\n' % (' ' * self.indent))
- if maplayer.type == 'raster':
- self.__writeNvizSurface(nviz['surface'])
- elif maplayer.type == 'vector':
- self.__writeNvizVector(nviz['vector'])
- self.file.write('%s</nviz>\n' % (' ' * self.indent))
- self.indent -= 4
- self.file.write('%s</layer>\n' % (' ' * self.indent))
- item = mapTree.GetNextSibling(item)
- self.indent -= 4
-
- def __writeNvizSurface(self, data):
- """!Save Nviz raster layer properties to workspace
- @param data Nviz layer properties
- """
- if 'object' not in data: # skip disabled
- return
- self.indent += 4
- self.file.write('%s<surface>\n' % (' ' * self.indent))
- self.indent += 4
- for attrb in data.iterkeys():
- if len(data[attrb]) < 1: # skip empty attributes
- continue
- if attrb == 'object':
- continue
-
- for name in data[attrb].iterkeys():
- # surface attribute
- if attrb == 'attribute':
- self.file.write('%s<%s name="%s" map="%d">\n' % \
- (' ' * self.indent, attrb, name, data[attrb][name]['map']))
- self.indent += 4
- self.file.write('%s<value>%s</value>\n' % (' ' * self.indent, data[attrb][name]['value']))
- self.indent -= 4
- # end tag
- self.file.write('%s</%s>\n' % (' ' * self.indent, attrb))
- # draw mode
- if attrb == 'draw':
- self.file.write('%s<%s' %(' ' * self.indent, attrb))
- if 'mode' in data[attrb]:
- for tag, value in data[attrb]['mode']['desc'].iteritems():
- self.file.write(' %s="%s"' % (tag, value))
- self.file.write('>\n') # <draw ...>
- if 'resolution' in data[attrb]:
- self.indent += 4
- for type in ('coarse', 'fine'):
- self.file.write('%s<resolution type="%s">\n' % (' ' * self.indent, type))
- self.indent += 4
- self.file.write('%s<value>%d</value>\n' % (' ' * self.indent,
- data[attrb]['resolution'][type]))
- self.indent -= 4
- self.file.write('%s</resolution>\n' % (' ' * self.indent))
- if 'wire-color' in data[attrb]:
- self.file.write('%s<wire_color>\n' % (' ' * self.indent))
- self.indent += 4
- self.file.write('%s<value>%s</value>\n' % (' ' * self.indent,
- data[attrb]['wire-color']['value']))
- self.indent -= 4
- self.file.write('%s</wire_color>\n' % (' ' * self.indent))
- self.indent -= 4
-
- # position
- elif attrb == 'position':
- self.file.write('%s<%s>\n' %(' ' * self.indent, attrb))
- i = 0
- for tag in ('x', 'y', 'z'):
- self.indent += 4
- self.file.write('%s<%s>%d</%s>\n' % (' ' * self.indent, tag,
- data[attrb][tag], tag))
- i += 1
- self.indent -= 4
- if attrb != 'attribute':
- # end tag
- self.file.write('%s</%s>\n' % (' ' * self.indent, attrb))
- self.indent -= 4
- self.file.write('%s</surface>\n' % (' ' * self.indent))
- self.indent -= 4
- def __writeNvizVector(self, data):
- """!Save Nviz vector layer properties (lines/points) to workspace
- @param data Nviz layer properties
- """
- self.indent += 4
- for attrb in data.iterkeys():
- if len(data[attrb]) < 1: # skip empty attributes
- continue
- if 'object' not in data[attrb]: # skip disabled
- continue
- if attrb == 'lines':
- self.file.write('%s<v%s>\n' % (' ' * self.indent, attrb))
- elif attrb == 'points':
- markerId = data[attrb]['marker']['value']
- marker = UserSettings.Get(group = 'nviz', key = 'vector',
- subkey = ['points', 'marker'], internal = True)[markerId]
- self.file.write('%s<v%s marker="%s">\n' % (' ' * self.indent,
- attrb,
- marker))
- self.indent += 4
- for name in data[attrb].iterkeys():
- if name in ('object', 'marker'):
- continue
- if name == 'mode':
- self.file.write('%s<%s type="%s">\n' % (' ' * self.indent, name,
- data[attrb][name]['type']))
- if data[attrb][name]['type'] == 'surface':
- self.indent += 4
- for idx, surface in enumerate(data[attrb][name]['surface']['value']):
- checked = data[attrb][name]['surface']['show'][idx]
- self.file.write('%s<map>\n' % (' ' * self.indent))
- self.indent += 4
- self.file.write('%s<name>%s</name>\n' % (' ' * self.indent, surface))
- self.file.write('%s<checked>%s</checked>\n' % (' ' * self.indent, int(checked)))
- self.indent -= 4
- self.file.write('%s</map>\n' % (' ' * self.indent))
- self.indent -= 4
- self.file.write('%s</%s>\n' % ((' ' * self.indent, name)))
- elif name == 'thematic':
- self.file.write('%s<%s ' % (' ' * self.indent, name))
- for key in data[attrb][name].iterkeys():
- if key.startswith('use'):
- self.file.write('%s="%s" ' % (key, int(data[attrb][name][key])))
- self.file.write('>\n')
- self.indent += 4
- for key, value in data[attrb][name].iteritems():
- if key.startswith('use'):
- continue
- if value is None:
- value = ''
- self.file.write('%s<%s>%s</%s>\n' % (' ' * self.indent, key, value, key))
- self.indent -= 4
- self.file.write('%s</%s>\n' % (' ' * self.indent, name))
- else:
- self.file.write('%s<%s>\n' % (' ' * self.indent, name))
- self.indent += 4
- self.file.write('%s<value>%s</value>\n' % (' ' * self.indent, data[attrb][name]['value']))
- self.indent -= 4
- self.file.write('%s</%s>\n' % (' ' * self.indent, name))
- self.indent -= 4
- self.file.write('%s</v%s>\n' % (' ' * self.indent, attrb))
- self.indent -= 4
- def __writeNvizState(self, view, iview, light, constants):
- """"!Save Nviz properties (view, light) to workspace
- @param view Nviz view properties
- @param iview Nviz internal view properties
- @param light Nviz light properties
- """
- self.indent += 4
- self.file.write('%s<nviz_state>\n' % (' ' * self.indent))
- #
- # view
- #
- self.indent += 4
- self.file.write('%s<view>\n' % (' ' * self.indent))
- self.indent += 4
- # position
- self.file.write('%s<v_position>\n' % (' ' * self.indent))
- self.indent += 4
- self.file.write('%s<x>%.2f</x>\n' % (' ' * self.indent, view['position']['x']))
- self.file.write('%s<y>%.2f</y>\n' % (' ' * self.indent, view['position']['y']))
- self.indent -= 4
- self.file.write('%s</v_position>\n' % (' ' * self.indent))
- # perspective
- self.file.write('%s<persp>\n' % (' ' * self.indent))
- self.indent += 4
- self.file.write('%s<value>%d</value>\n' % (' ' * self.indent, view['persp']['value']))
- self.file.write('%s<step>%d</step>\n' % (' ' * self.indent, view['persp']['step']))
- self.file.write('%s<min>%d</min>\n' % (' ' * self.indent, iview['persp']['min']))
- self.file.write('%s<max>%d</max>\n' % (' ' * self.indent, iview['persp']['max']))
- self.indent -= 4
- self.file.write('%s</persp>\n' % (' ' * self.indent))
- # height
- self.file.write('%s<v_height>\n' % (' ' * self.indent))
- self.indent += 4
- self.file.write('%s<value>%d</value>\n' % (' ' * self.indent, iview['height']['value']))
- self.file.write('%s<min>%d</min>\n' % (' ' * self.indent, iview['height']['min']))
- self.file.write('%s<max>%d</max>\n' % (' ' * self.indent, iview['height']['max']))
- self.indent -= 4
- self.file.write('%s</v_height>\n' % (' ' * self.indent))
- # twist
- self.file.write('%s<twist>\n' % (' ' * self.indent))
- self.indent += 4
- self.file.write('%s<value>%d</value>\n' % (' ' * self.indent, view['twist']['value']))
- self.file.write('%s<min>%d</min>\n' % (' ' * self.indent, iview['twist']['min']))
- self.file.write('%s<max>%d</max>\n' % (' ' * self.indent, iview['twist']['max']))
- self.indent -= 4
- self.file.write('%s</twist>\n' % (' ' * self.indent))
- # z-exag
- self.file.write('%s<z-exag>\n' % (' ' * self.indent))
- self.indent += 4
- self.file.write('%s<value>%d</value>\n' % (' ' * self.indent, view['z-exag']['value']))
- self.file.write('%s<min>%d</min>\n' % (' ' * self.indent, view['z-exag']['min']))
- self.file.write('%s<max>%d</max>\n' % (' ' * self.indent, view['z-exag']['max']))
- self.file.write('%s<original>%d</original>\n' % (' ' * self.indent, iview['z-exag']['original']))
- self.indent -= 4
- self.file.write('%s</z-exag>\n' % (' ' * self.indent))
- # focus (look here)
- self.file.write('%s<focus>\n' % (' ' * self.indent))
- self.indent += 4
- self.file.write('%s<x>%d</x>\n' % (' ' * self.indent, iview['focus']['x']))
- self.file.write('%s<y>%d</y>\n' % (' ' * self.indent, iview['focus']['y']))
- self.file.write('%s<z>%d</z>\n' % (' ' * self.indent, iview['focus']['z']))
- self.indent -= 4
- self.file.write('%s</focus>\n' % (' ' * self.indent))
- # background
- self.__writeTagWithValue('background_color', view['background']['color'][:3], format = 'd:%d:%d')
-
- self.indent -= 4
- self.file.write('%s</view>\n' % (' ' * self.indent))
- #
- # light
- #
- self.file.write('%s<light>\n' % (' ' * self.indent))
- self.indent += 4
- # position
- self.file.write('%s<l_position>\n' % (' ' * self.indent))
- self.indent += 4
- self.file.write('%s<x>%.2f</x>\n' % (' ' * self.indent, light['position']['x']))
- self.file.write('%s<y>%.2f</y>\n' % (' ' * self.indent, light['position']['y']))
- self.file.write('%s<z>%d</z>\n' % (' ' * self.indent, light['position']['z']))
- self.indent -= 4
- self.file.write('%s</l_position>\n' % (' ' * self.indent))
- # bright
- self.__writeTagWithValue('bright', light['bright'])
- # ambient
- self.__writeTagWithValue('ambient', light['ambient'])
- # color
- self.__writeTagWithValue('color', light['color'][:3], format = 'd:%d:%d')
-
- self.indent -= 4
- self.file.write('%s</light>\n' % (' ' * self.indent))
- #
- # constant planes
- #
- if constants:
- self.file.write('%s<constant_planes>\n' % (' ' * self.indent))
- self.indent += 4
- for idx, plane in enumerate(constants):
- self.file.write('%s<plane>\n' % (' ' * self.indent))
- self.indent += 4
- self.__writeTagWithValue('height', constants[idx]['constant']['value'])
- self.__writeTagWithValue('fine_resolution', constants[idx]['constant']['resolution'])
- self.__writeTagWithValue('color', constants[idx]['constant']['color'], format = 's')
- self.indent -= 4
- self.file.write('%s</plane>\n' % (' ' * self.indent))
- self.indent -= 4
- self.file.write('%s</constant_planes>\n' % (' ' * self.indent))
- self.indent -= 4
-
- self.file.write('%s</nviz_state>\n' % (' ' * self.indent))
- self.indent -= 4
-
- def __writeTagWithValue(self, tag, data, format = 'd'):
- """!Helper function for writing pair tag
-
- @param tag written tag
- @param data written data
- @param format conversion type
- """
- self.file.write('%s<%s>\n' % (' ' * self.indent, tag))
- self.indent += 4
- self.file.write('%s' % (' ' * self.indent))
- self.file.write(('<value>%' + format + '</value>\n') % data)
- self.indent -= 4
- self.file.write('%s</%s>\n' % (' ' * self.indent, tag))
-
- class ProcessGrcFile(object):
- def __init__(self, filename):
- """!Process GRC file"""
- self.filename = filename
- # elements
- self.inGroup = False
- self.inRaster = False
- self.inVector = False
- # list of layers
- self.layers = []
- # error message
- self.error = ''
- self.num_error = 0
- def read(self, parent):
- """!Read GRC file
- @param parent parent window
- @return list of map layers
- """
- try:
- file = open(self.filename, "r")
- except IOError:
- wx.MessageBox(parent=parent,
- message=_("Unable to open file <%s> for reading.") % self.filename,
- caption=_("Error"), style=wx.OK | wx.ICON_ERROR)
- return []
- line_id = 1
- for line in file.readlines():
- self.process_line(line.rstrip('\n'), line_id)
- line_id +=1
- file.close()
- if self.num_error > 0:
- wx.MessageBox(parent=parent,
- message=_("Some lines were skipped when reading settings "
- "from file <%(file)s>.\nSee 'Command output' window for details.\n\n"
- "Number of skipped lines: %(line)d") % \
- { 'file' : self.filename, 'line' : self.num_error },
- caption=_("Warning"), style=wx.OK | wx.ICON_EXCLAMATION)
- parent.goutput.WriteLog('Map layers loaded from GRC file <%s>' % self.filename)
- parent.goutput.WriteLog('Skipped lines:\n%s' % self.error)
- return self.layers
- def process_line(self, line, line_id):
- """!Process line definition"""
- element = self._get_element(line)
- if element == 'Group':
- self.groupName = self._get_value(line)
- self.layers.append({
- "type" : 'group',
- "name" : self.groupName,
- "checked" : None,
- "opacity" : None,
- "cmd" : None,
- "group" : self.inGroup,
- "display" : 0 })
- self.inGroup = True
- elif element == '_check':
- if int(self._get_value(line)) == 1:
- self.layers[-1]['checked'] = True
- else:
- self.layers[-1]['checked'] = False
-
- elif element == 'End':
- if self.inRaster:
- self.inRaster = False
- elif self.inVector:
- self.inVector = False
- elif self.inGroup:
- self.inGroup = False
- elif self.inGridline:
- self.inGridline = False
-
- elif element == 'opacity':
- self.layers[-1]['opacity'] = float(self._get_value(line))
- # raster
- elif element == 'Raster':
- self.inRaster = True
- self.layers.append({
- "type" : 'raster',
- "name" : self._get_value(line),
- "checked" : None,
- "opacity" : None,
- "cmd" : ['d.rast'],
- "group" : self.inGroup,
- "display" : 0})
- elif element == 'map' and self.inRaster:
- self.layers[-1]['cmd'].append('map=%s' % self._get_value(line))
-
- elif element == 'overlay' and self.inRaster:
- if int(self._get_value(line)) == 1:
- self.layers[-1]['cmd'].append('-o')
-
- elif element == 'rastquery' and self.inRaster:
- value = self._get_value(line)
- if value != '':
- self.layers[-1]['cmd'].append('catlist=%s' % value)
-
- elif element == 'bkcolor' and self.inRaster:
- value = self._get_value(line)
- if value != '':
- self.layers[-1]['cmd'].append('bg=%s' % value)
- # vector
- elif element == 'Vector':
- self.inVector = True
- self.layers.append({
- "type" : 'vector',
- "name" : self._get_value(line),
- "checked" : None,
- "opacity" : None,
- "cmd" : ['d.vect'],
- "group" : self.inGroup,
- "display" : 0})
- elif element == 'vect' and self.inVector:
- self.layers[-1]['cmd'].append('map=%s' % self._get_value(line))
-
- elif element in ('display_shape',
- 'display_cat',
- 'display_topo',
- 'display_dir',
- 'display_attr',
- 'type_point',
- 'type_line',
- 'type_boundary',
- 'type_centroid',
- 'type_area',
- 'type_face') and self.inVector:
-
- if int(self._get_value(line)) == 1:
- name = element.split('_')[0]
- type = element.split('_')[1]
- paramId = self._get_cmd_param_index(self.layers[-1]['cmd'], name)
- if paramId == -1:
- self.layers[-1]['cmd'].append('%s=%s' % (name, type))
- else:
- self.layers[-1]['cmd'][paramId] += ',%s' % type
- elif element in ('color',
- 'fcolor',
- 'lcolor') and self.inVector:
- value = self._get_value(line)
- if value != '':
- self.layers[-1]['cmd'].append('%s=%s' % (element,
- self._color_name_to_rgb(value)))
- elif element == 'rdmcolor' and self.inVector:
- if int(self._get_value(line)) == 1:
- self.layers[-1]['cmd'].append('-c')
- elif element == 'sqlcolor' and self.inVector:
- if int(self._get_value(line)) == 1:
- self.layers[-1]['cmd'].append('-a')
- elif element in ('icon',
- 'size',
- 'layer',
- 'xref',
- 'yref',
- 'lsize',
- 'where',
- 'minreg',
- 'maxreg') and self.inVector:
- value = self._get_value(line)
- if value != '':
- self.layers[-1]['cmd'].append('%s=%s' % (element,
- value))
-
- elif element == 'lwidth':
- value = self._get_value(line)
- if value != '':
- self.layers[-1]['cmd'].append('width=%s' % value)
- elif element == 'lfield':
- value = self._get_value(line)
- if value != '':
- self.layers[-1]['cmd'].append('llayer=%s' % value)
-
- elif element == 'attribute':
- value = self._get_value(line)
- if value != '':
- self.layers[-1]['cmd'].append('attrcol=%s' % value)
- elif element == 'cat':
- value = self._get_value(line)
- if value != '':
- self.layers[-1]['cmd'].append('cats=%s' % value)
- # gridline
- elif element == 'gridline':
- self.inGridline = True
- self.layers.append({
- "type" : 'grid',
- "name" : self._get_value(line),
- "checked" : None,
- "opacity" : None,
- "cmd" : ['d.grid'],
- "group" : self.inGroup,
- "display" : 0})
- elif element == 'gridcolor':
- value = self._get_value(line)
- if value != '':
- self.layers[-1]['cmd'].append('color=%s' % self._color_name_to_rgb(value))
- elif element == 'gridborder':
- value = self._get_value(line)
- if value != '':
- self.layers[-1]['cmd'].append('bordercolor=%s' % self._color_name_to_rgb(value))
- elif element == 'textcolor':
- value = self._get_value(line)
- if value != '':
- self.layers[-1]['cmd'].append('textcolor=%s' % self._color_name_to_rgb(value))
- elif element in ('gridsize',
- 'gridorigin'):
- value = self._get_value(line)
- if value != '':
- self.layers[-1]['cmd'].append('%s=%s' % (element[4:], value))
- elif element in 'fontsize':
- value = self._get_value(line)
- if value != '':
- self.layers[-1]['cmd'].append('%s=%s' % (element, value))
-
- elif element == 'griddraw':
- value = self._get_value(line)
- if value == '0':
- self.layers[-1]['cmd'].append('-n')
-
- elif element == 'gridgeo':
- value = self._get_value(line)
- if value == '1':
- self.layers[-1]['cmd'].append('-g')
-
- elif element == 'borderdraw':
- value = self._get_value(line)
- if value == '0':
- self.layers[-1]['cmd'].append('-b')
- elif element == 'textdraw':
- value = self._get_value(line)
- if value == '0':
- self.layers[-1]['cmd'].append('-t')
-
- else:
- self.error += _(' row %d:') % line_id + line + os.linesep
- self.num_error += 1
- def _get_value(self, line):
- """!Get value of element"""
- try:
- return line.strip(' ').split(' ')[1].strip(' ')
- except:
- return ''
- def _get_element(self, line):
- """!Get element tag"""
- return line.strip(' ').split(' ')[0].strip(' ')
- def _get_cmd_param_index(self, cmd, name):
- """!Get index of parameter in cmd list
- @param cmd cmd list
- @param name parameter name
- @return index
- @return -1 if not found
- """
- i = 0
- for param in cmd:
- if '=' not in param:
- i += 1
- continue
- if param.split('=')[0] == name:
- return i
- i += 1
- return -1
- def _color_name_to_rgb(self, value):
- """!Convert color name (#) to rgb values"""
- col = wx.NamedColour(value)
- return str(col.Red()) + ':' + \
- str(col.Green()) + ':' + \
- str(col.Blue())
|