Source code for PyMca5.PyMcaIO.RTXMap

#!/usr/bin/env python
#/*##########################################################################
#
# The PyMca X-Ray Fluorescence Toolkit
#
# Copyright (c) 2004-2015 European Synchrotron Radiation Facility
#
# This file is part of the PyMca X-ray Fluorescence Toolkit developed at
# the ESRF by the Software group.
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
#
#############################################################################*/
__author__ = "V.A. Sole - ESRF Data Analysis"
__contact__ = "sole@esrf.fr"
__license__ = "MIT"
__copyright__ = "European Synchrotron Radiation Facility, Grenoble, France"
import sys
import os
import numpy
import types

#spx file format is based on XML
import xml.etree.ElementTree as ElementTree
from PyMca5.PyMcaCore import DataObject
SOURCE_TYPE = "EdfFileStack"

[docs]def myFloat(x): try: return float(x) except ValueError: if ',' in x: try: return float(x.replace(',','.')) except: return float(x) elif '.' in x: try: return float(x.replace('.',',')) except: return float(x) else: raise
[docs]class RTXMap(DataObject.DataObject): ''' Class to read ARTAX .rtx files ''' def __init__(self, filename): ''' Parameters: ----------- filename : str Name of the .rtx file. ''' DataObject.DataObject.__init__(self) if not os.path.exists(filename): raise IOError("File %s does not exists" % filename) f = ElementTree.parse(filename) root = f.getroot() #print("Tag = ", root.tag) #print("Tag = ", root.attrib) info = {} if 1: # this works for key in ["XFirst", "YFirst", "ZFirst", "XLast", "YLast", "ZLast", "MeasNo"]: element = root.find(".//ClassInstance/%s" % key) if element is not None: info[key] = myFloat(element.text) if "MeasNo" in info: nSpectra = int(info["MeasNo"]) else: nSpectra = None if 1: # This works i = 0 axesList = root.findall('.//ClassInstance/AxesParameter') nRows = None nColumns = 0 y = None done = False for axes in axesList: if not done: for axis in axes: if axis.attrib['AxisName'] == 'y': if i == 0: y = myFloat(axis.attrib['AxisPosition']) if y == myFloat(axis.attrib['AxisPosition']): nColumns += 1 else: done = True #print("#U%d %s %f %s" % (i, # axis.attrib['AxisName'], # myFloat(axis.attrib['AxisPosition']), # axis.attrib['AxisUnit'])) i += 1 #print("Found %d positions" % i) #print("nColumns = %d" % nColumns) if nColumns == 0: nColumns = 1 nRows = i //nColumns #print("nRows = %d" % nRows) if nSpectra is None: nSpectra = nRows * nColumns else: if nSpectra != nRows * nColumns: print("WARNING Header info does not match") data = None if 1: # This works for key in ["Channels"]: keyToSearch = './/ClassInstance/%s' % key i = 0 for content in root.findall(keyToSearch): if content is not None: spectrum = numpy.array([myFloat(x) for x in content.text.split(',')]) nChannels = len(spectrum) if data is None: data = numpy.zeros((nSpectra, nChannels)) if i < nSpectra: data[i] = spectrum i += 1 print("Found %d spectra" % i) #print nSpectra #print nRows * nColumns data.shape = nRows, nColumns, nChannels self.data = data self.info = info
[docs]def test(filename): a = RTXMap(filename) print("info = ", a.info) print("Data shape = ", a.data.shape)
if __name__ == "__main__": test(sys.argv[1])