Bases: TSTransform
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
|