"""
Module dealing with observed lightcurves
"""
from .spectrum import BaseSpectrum
import numpy as np
from taurex.model.lightcurve.lightcurvedata import LightCurveData
[docs]class ObservedLightCurve(BaseSpectrum):
"""
Loads an observed lightcurve from a pickle file.
Parameters
----------
filename : str
Path to pickle file containing lightcurve data
"""
def __init__(self, filename=None):
super().__init__('observed_lightcurve')
import pickle
with open(filename, 'rb') as f:
lc_data = pickle.load(f, encoding='latin1')
# new version
self.obs_spectrum = \
np.empty(shape=(len(lc_data['obs_spectrum'][:, 0]), 4))
# new version
self.obs_spectrum[:, 0] = lc_data['obs_spectrum'][:, 0]
self.obs_spectrum[:, 1] = lc_data['obs_spectrum'][:, 1]
self.obs_spectrum[:, 2] = lc_data['obs_spectrum'][:, 2]
self.obs_spectrum[:, 3] = lc_data['obs_spectrum'][:, 3]
self._spec, self._std = self._load_data_file(lc_data)
[docs] def create_binner(self):
from taurex.binning.lightcurvebinner import LightcurveBinner
"""
Creates the appropriate binning object
"""
return LightcurveBinner()
def _load_data_file(self, lc_data):
"""load data from different instruments."""
raw_data = []
data_std = []
wngrid_min = []
for i in LightCurveData.availableInstruments:
# new version
# raw data includes data and datastd.
if i in lc_data:
wngrid_min.append(lc_data[i]['wl_grid'].min())
raw_data.append(lc_data[i]['data'][:, :, 0])
data_std.append(lc_data[i]['data'][:, :, 1])
wngrid_min, raw_data, data_std = \
(list(t) for t in zip(*sorted(zip(wngrid_min, raw_data, data_std),
key=lambda x: x[0], reverse=True)))
return np.concatenate(raw_data), np.concatenate(data_std)
@property
def spectrum(self):
"""
Returns Light curve spectrum.
The lightcurve spectrum comes in the form of multiple lightcurves
stuck together into
one long spectrum. The number of lightcurves is equal to the number of
bins in :func:`wavelengthGrid`.
Returns
-------
spectrum : :obj:`array`
"""
return self._spec
@property
def rawData(self):
"""
Raw lightcurve data read from file
Returns
-------
lc_data : :obj:`array`
"""
self.obs_spectrum
@property
def wavelengthGrid(self):
"""
Returns wavelength grid in microns
Returns
-------
wlgrid : :obj:`array`
"""
return self.obs_spectrum[:, 0]
@property
def binEdges(self):
"""
Returns bin edges for wavelength grid
Returns
-------
out : :obj:`array`
"""
return self.obs_spectrum[:, 3]
@property
def binWidths(self):
"""
Widths for each bin in wavelength grid
Returns
-------
out : :obj:`array`
"""
return np.zeros(2)
@property
def errorBar(self):
"""
Like :func:`spectrum` except its the error at each point in the
lightcurve spectrum
Returns
-------
err : :obj:`array`
Error at each point in lightcurve spectrum
"""
return self._std
[docs] def write(self, output):
output.write_array('wlgrid', self.wavelengthGrid)
output.write_array('spectrum', self.obs_spectrum[:, 1])
output.write_array('lightcurve', self.spectrum)
output.write_array('binedges', self.binEdges)
output.write_array('binwidths', self.binWidths)
output.write_array('errorbars', self.obs_spectrum[:, 2])
output.write_array('lightcurve_errorbars', self.errorBar)
return output