Skip to content

sgnts.transforms.amplify

Amplify dataclass

Bases: TSTransform


              flowchart TD
              sgnts.transforms.amplify.Amplify[Amplify]
              sgnts.base.base.TSTransform[TSTransform]
              sgnts.base.base.TimeSeriesMixin[TimeSeriesMixin]

                              sgnts.base.base.TSTransform --> sgnts.transforms.amplify.Amplify
                                sgnts.base.base.TimeSeriesMixin --> sgnts.base.base.TSTransform
                



              click sgnts.transforms.amplify.Amplify href "" "sgnts.transforms.amplify.Amplify"
              click sgnts.base.base.TSTransform href "" "sgnts.base.base.TSTransform"
              click sgnts.base.base.TimeSeriesMixin href "" "sgnts.base.base.TimeSeriesMixin"

Amplify data by a factor.

Parameters:

Name Type Description Default
factor float

float, the factor to multiply the data with

 1
Source code in sgnts/transforms/amplify.py 
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
@dataclass
class Amplify(TSTransform):
    """Amplify data by a factor.

    Args:
        factor:
            float, the factor to multiply the data with
    """

    factor: float = 1

    def __post_init__(self):
        super().__post_init__()
        assert len(self.sink_pads) == 1 and len(self.source_pads) == 1, (
            f"Amplify requires exactly one sink pad and one source pad, "
            f"got {len(self.sink_pads)} sink pads and "
            f"{len(self.source_pads)} source pads"
        )
        self.sink_pad = self.sink_pads[0]

    def new(self, pad: SourcePad) -> TSFrame:
        outbufs = []
        # loop over the input data, only amplify non-gap data
        frame = self.preparedframes[self.sink_pad]
        for inbuf in frame:
            if inbuf.is_gap:
                data = None
            else:
                data = inbuf.data * self.factor

            outbuf = SeriesBuffer(
                offset=inbuf.offset,
                sample_rate=inbuf.sample_rate,
                data=data,
                shape=inbuf.shape,
            )
            outbufs.append(outbuf)

        return TSFrame(buffers=outbufs, EOS=frame.EOS, metadata=frame.metadata)