workspace.py 73 KB

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