Source code for xpsi.Instrument

from __future__ import division, print_function

__all__ = ["Instrument"]

from .global_imports import *
from . import global_imports

from . import make_verbose

from .ParameterSubspace import ParameterSubspace

[docs]class ResponseError(xpsiError): """ Raised if there is a problem with the input response matrix. """
[docs]class EdgesError(xpsiError): """ Raised if there is a problem with the input energy edges. """
[docs]class ChannelError(xpsiError): """ Raised if there is a problem with the input channel numbers. """
[docs]class Instrument(ParameterSubspace): """ Base class for astronomical X-ray instruments on-board space telescopes. The body of the initialiser must not be changed to ensure inter-module compatibility, but can be extended if appropriate using a call to ``super().__init__``. Specialist constructors can be defined in a subclass using the ``@classmethod`` decorator. :param ndarray[p,q] matrix: A :math:`p \\times q` matrix which is the product of a redistribution matrix and effective area vector. The input energy intervals must increase along the columns of :attr:`matrix`, and the output channels must increase along the rows of :attr:`matrix`. The *units* of the elements must be that of an *effective* area (:math:`cm^2`). Generally there will be some available calibration product, and deviations from this nominal response model will be parametrised. So here load some nominal response matrix. :param ndarray[q+1] energy_edges: Energy edges in keV of the instrument energy intervals which must be congruent to the first dimension of the :attr:`matrix`: the number of edges must be :math:`q + 1`. The edges must be monotonically increasing. These edges will correspond to the nominal response matrix and any deviation from this matrix (see above). :param ndarray[p] channels: Instrument channel numbers which must be equal in number to the number of rows of the :attr:`matrix`. The number of channels must therefore be :math:`p`. These channels will correspond to the nominal response matrix and any deviation from this matrix (see above). In common usage patterns, the channel numbers will increase monotonically with row number, and usually increment by one (but this is not necessary). .. note:: That these channel numbers are not used to index the loaded instrument (sub)matrix. The :attr:`xpsi.Data.index_range` property returns bounding row numbers that index the loaded instrument response (sub)matrix in order to operate on an incident signal flux. The channel array contained in :attr:`xpsi.Data.channels` must be a contiguous (ordered) subset of the channel array loaded here. .. note:: The dimensions of the response matrix need not be equal, but it is required that the number of input intervals be greater than or equal to the number of output channels -- i.e., :math:`p \leq q`. If :math:`p < q` then it is implied that subsets of adjacent output channels are effectively grouped together. :param ndarray[p+1] channel_edges: The channel (energy) edges of the instrument, in keV. The array must be congruent to the zeroth dimension of the :attr:`matrix`: the number of edges must be :math:`p + 1`. The edges must be monotonically increasing. These edges will correspond to the nominal response matrix and any deviation from this matrix (see above). :param tuple args: Container of parameter instances. :param dict kwargs: If you want to prefix parameters of an instance of this instrument subspace with an identifier, pass it as keyword argument and it will find its way to the base class. """ def __init__(self, matrix, energy_edges, channels, channel_edges=None, *args, **kwargs): self.matrix = matrix self.energy_edges = energy_edges self.channels = channels if channel_edges is not None: self.channel_edges = channel_edges super(Instrument, self).__init__(*args, **kwargs) @property def matrix(self): """ Get the reference response matrix. In common usage patterns there will be some fiducial or nominal response matrix that either defines fixed instrument operation or is a basis for parametrised deviations. This matrix is usually a calibration product distributed by an instrument calibration team. A matrix of dimension :math:`p \\times q`. Here :math:`p` must be the number of input energy intervals, and :math:`q \geq p` the number of output channels. .. note:: The attribute :attr:`matrix` must be assigned, and it must be a :class:`numpy.ndarray` for use with :func:`numpy.dot` (even if the matrix is sparse to some degree). """ return self._matrix @matrix.setter def matrix(self, matrix): """ Set the matrix. """ try: assert isinstance(matrix, _np.ndarray) assert matrix.ndim == 2 assert matrix.shape[0] <= matrix.shape[1] assert (matrix >= 0.0).all() except AssertionError: raise ResponseError('Input matrix must be a two-dimensional ndarray, with number of channel intervals' '(dimension 0) smaller than the number of energy intervals (dimension 1),' ' and with all matrix elements that are zero or positive.') try: for i in range(matrix.shape[0]): assert matrix[i,:].any() for j in range(matrix.shape[1]): assert matrix[:,j].any() except AssertionError: raise ResponseError('Each row and column of the matrix must contain at least one positive number.' ' Try reducing the \"gap\" between the minimum input energy ' 'and minimum input channel.') self._matrix = matrix
[docs] def construct_matrix(self): """ Construct the response matrix if it is parameterised. If customising, do operations to calculate a matrix, and return it. You can access parameters (free, fixed, and derived) via the container access ``self[<name>]``. If the instrument operation is fixed, you might not need to subclass, because the default behaviour is to return the nominal response you loaded. If for some reason the matrix you loaded is to be modified in some fixed manner, possibly as a function of some custom fixed parameters that you defined, you would also have to subclass and provide the correct implementation of this method. """ return self.matrix
[docs] def __call__(self, signal, irange, orange): """ Register an incident signal. :param ndarray[m,n] signal: An :math:`m \\times n` matrix, where input energy interval increments along rows, and phase increases along columns. The number of rows, :math:`m`, must equal the number of columns of :attr:`matrix`: :math:`m=q`. :param array-like irange: Indexable object with two elements respectively denoting the indices of the first and last *input* intervals. The response matrix :attr:`matrix` must be indexable with these numbers, i.e., they must satisfy :math:`indx < q`. :param array-like orange: Indexable object with two elements respectively denoting the indices of the first and last *output* channels. The response matrix :attr:`matrix` must be indexable with these numbers, i.e., they must satisfy :math:`indx < p`. :return: *ndarray[p,n]* containing the registered signal. .. note:: The product of the most recent operation is stored as the property :attr:`cached_signal`. """ matrix = self.construct_matrix() self._cached_signal = _np.dot(matrix[orange[0]:orange[1], irange[0]:irange[1]], signal) return self._cached_signal
@property def cached_signal(self): """ Get the cached registered signal. """ return self._cached_signal @property def energy_edges(self): """ Get the energy edges of the instrument, in keV. A :class:`numpy.ndarray` of edges of the input energy intervals which map to channels defined in the data space. """ return self._energy_edges @energy_edges.setter def energy_edges(self, energy_edges): """ Set the energy edges in keV. """ if not isinstance(energy_edges, _np.ndarray): try: energy_edges = _np.array(energy_edges) except TypeError: raise EdgesError('Energy edges must be in a one-dimensional array of positive increasing values.') try: assert energy_edges.ndim == 1 assert (energy_edges >= 0.0).all() assert energy_edges.shape[0] == self._matrix.shape[1] + 1 assert not (energy_edges[1:] <= energy_edges[:-1]).any() except AssertionError: raise EdgesError('Energy edges must be in a one-dimensional array of positive increasing values, with a ' 'length equal to number of energy intervals in the matrix + 1.') self._energy_edges = energy_edges @property def channel_edges(self): """ Get the channel (energy) edges of the instrument, in keV. A :class:`numpy.ndarray` of edges of the registered energy intervals labelled as channels defined in the data space. This is relevant when there is a detector-by-detector gain scale applied to event data (such as for NICER instrument calibration products), meaning that the redistribution matrix is effectively shared by detectors and the channels across detectors can share an energy scale definition. An incident photon of given energy then has a registered-energy distribution that generally peaks in the vicinity of the true photon energy. The resdistribution matrix will have some energy resolution (along with other features such as shelves). With thanks to Paul S. Ray for explaining the choice to calibrate in this manner. .. note:: If you made a channel cut that results in a non-contiguous subset of channels, you will need to overwrite the setter method because the checks will fail. """ return self._channel_edges @channel_edges.setter def channel_edges(self, channel_edges): """ Set the channel (energy) edges in keV. """ if not isinstance(channel_edges, _np.ndarray): try: channel_edges = _np.array(channel_edges) except TypeError: raise EdgesError('Channel edges must be in a one-dimensional array of positive increasing values.') try: assert channel_edges.ndim == 1 assert (channel_edges >= 0.0).all() assert channel_edges.shape[0] == self._matrix.shape[0] + 1 assert not (channel_edges[1:] <= channel_edges[:-1]).any() except AssertionError: raise EdgesError('Channel edges must be in a one-dimensional array of positive increasing values, with a ' 'length equal to the number of channel intervals in the matrix + 1.') self._channel_edges = channel_edges @property def channels(self): """ Get the array of channels corresponding to rows of the matrix. The matrix being the loaded instrument response (sub)matrix. """ return self._channels @channels.setter @make_verbose('Setting channels for loaded instrument response (sub)matrix', 'Channels set') def channels(self, channel_array): if not isinstance(channel_array, _np.ndarray): try: channel_array = _np.array(channel_array) except TypeError: raise ChannelError('Channel numbers must be in a one-dimensional array of positive integers' '(including zero).') try: assert channel_array.ndim == 1 assert (channel_array >= 0).all() assert channel_array.shape[0] == self._matrix.shape[0] except AssertionError: raise ChannelError('Channel numbers must be in a one-dimensional array of positive integers' '(including zero).') if (channel_array[1:] - channel_array[:-1] != 1).any(): yield ('Warning: Channel numbers do not uniformly increment by one.\n' ' Please check for correctness.') self._channels = channel_array yield