Source code for ezmsg.sigproc.math.invert

from ezmsg.util.messages.axisarray import AxisArray
from ezmsg.util.messages.util import replace

from ..base import BaseTransformer, BaseTransformerUnit



[docs]
class InvertTransformer(BaseTransformer[None, AxisArray, AxisArray]):
    def _process(self, message: AxisArray) -> AxisArray:
        return replace(message, data=1 / message.data)




[docs]
class Invert(

    BaseTransformerUnit[None, AxisArray, AxisArray, InvertTransformer]
): ...  # SETTINGS = None



[docs]
def invert() -> InvertTransformer:
    """
    Take the inverse of the data.

    Returns: :obj:`InvertTransformer`.
    """
    return InvertTransformer()