esbo_ds/ESBO-ETC
esbo_ds
/
ESBO-ETC
11
0
Fork 1
Code Issues 1 Pull Requests Releases 1 Wiki Activity

File reader moved to SpectraQty

This commit is contained in:
Lukas Klass 2020-04-14 12:49:51 +02:00
parent be00750f84
commit 5181bab22d
1 changed files with 4 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
esbo_etc/classes/target/FileTarget.py
View File

@ -22,22 +22,7 @@ class FileTarget(ATarget):
`astropy.io.ascii.read(). If the file doesn't provide units via astropy's enhanced CSV format, the units will
be read from the column headers or otherwise assumed to be *nm* and *W / m^2 / nm*.
"""
# Read the file
data = ascii.read(file)
# Check if units are given
if data[data.colnames[0]].unit is None:
# Convert values to float
data[data.colnames[0]] = list(map(float, data[data.colnames[0]]))
data[data.colnames[1]] = list(map(float, data[data.colnames[1]]))
# Check if units are given in column headers
if all([re.search("\\[.+\\]", x) for x in data.colnames]):
# Extract units from headers and apply them on the columns
units = [u.Unit(re.findall("(?<=\\[).+(?=\\])", x)[0]) for x in data.colnames]
data[data.colnames[0]].unit = units[0]
data[data.colnames[1]].unit = units[1]
# Use default units
else:
data[data.colnames[0]].unit = u.nm
data[data.colnames[1]].unit = u.W / (u.m ** 2 * u.nm)
# Create a spectral quantity from the data and initialize the super class
super().__init__(SpectralQty(data[data.colnames[0]].quantity, data[data.colnames[1]].quantity))
# Create spectral quantity from file
sfd = SpectralQty.fromFile(file, u.nm, u.W / (u.m ** 2 * u.nm))
# Initialize the super class
super().__init__(sfd)