Source code for mosfit.modules.arrays.resttimes

"""Definitions for the `RestTimes` class."""
from collections import OrderedDict

import numpy as np
from mosfit.modules.arrays.array import Array


# Important: Only define one ``Module`` class per file.


[docs]class RestTimes(Array): """This class converts the observed times to rest-frame times."""
[docs] def process(self, **kwargs): """Process module.""" self._times = kwargs['all_times'] self._t_explosion = kwargs[self.key('texplosion')] self._z = kwargs[self.key('redshift')] outputs = OrderedDict() outputs['rest_times'] = np.array([ x / (1.0 + self._z) for x in self._times ]) outputs[self.key('resttexplosion')] = self._t_explosion / ( 1.0 + self._z) return outputs