Skip to content

sgnts.transforms.align

Align dataclass

Bases: TSTransform


              flowchart TD
              sgnts.transforms.align.Align[Align]
              sgnts.base.base.TSTransform[TSTransform]
              sgnts.base.base.TimeSeriesMixin[TimeSeriesMixin]

                              sgnts.base.base.TSTransform --> sgnts.transforms.align.Align
                                sgnts.base.base.TimeSeriesMixin --> sgnts.base.base.TSTransform
                



              click sgnts.transforms.align.Align href "" "sgnts.transforms.align.Align"
              click sgnts.base.base.TSTransform href "" "sgnts.base.base.TSTransform"
              click sgnts.base.base.TimeSeriesMixin href "" "sgnts.base.base.TimeSeriesMixin"
            

Align frames from multiple sink pads.

Source code in sgnts/transforms/align.py
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
@dataclass
class Align(TSTransform):
    """Align frames from multiple sink pads."""

    def __post_init__(self):
        assert set(self.source_pad_names) == set(self.sink_pad_names), (
            f"Source and sink pad names must match. "
            f"Source: {self.source_pad_names}, Sink: {self.sink_pad_names}"
        )
        super().__post_init__()
        self.pad_map = {
            p: self.sink_pad_dict["%s:snk:%s" % (self.name, self.rsrcs[p])]
            for p in self.source_pads
        }

    def new(self, pad: SourcePad) -> TSFrame:
        out = self.preparedframes[self.pad_map[pad]]
        self.preparedframes[self.pad_map[pad]] = None
        return out