dsignal.py

from postprocessor.core.processes.base import ParametersABC, ProcessABC
from postprocessor.core.functions.tracks import clean_tracks, merge_tracks, join_tracks


class dSignalParameters(ParametersABC):
    """
    :window: Number of timepoints to consider for signal.
    """

    def __init__(self, window):
        super().__init__()

    @classmethod
    def default(cls):
        return cls.from_dict({"window": 3})


class dSignal(ProcessABC):
    """
    Calculate the change in a signal depending on a window
    """

    def __init__(self, parameters: ParametersTemplate):
        super().__init__(parameters)

    def run(self, signal: pd.DataFrame):
        return signal.rolling(window=self.parameters.window, axis=1).mean().diff(axis=1)