diff --git a/pafsim/processor/generator.py b/pafsim/processor/generator.py index 76ed71fb675a3f927001a36b7ccad79a38f03971..2669c73b905d030802819a86ec2fe2c4146cf4c4 100644 --- a/pafsim/processor/generator.py +++ b/pafsim/processor/generator.py @@ -52,6 +52,8 @@ class FrontEndSimACMReader(h5.File): def npol(self) -> int: return len(set(self["array/element_ids"])) + def PafLayout(self): + pass def exists(self, path) -> bool: if path in self: @@ -114,14 +116,14 @@ class Generator(Processor): self.duration = self.conf.get("duration", 0.000001) self.fs = self.conf.get("fs", -1) self.mode = self.conf.get("mode", "wideband") + self.reader = FrontEndSimACMReader(self.dataset) def wideband(self): - reader = FrontEndSimACMReader(self.dataset) - signals = reader.signal(self.signal) + signals = self.reader.signal(self.signal) # noise = reader.noise(self.noise) # rfi = reader.rfi(self.rfi) - freq = reader.signal(self.signal, "frequencies")[0]*1e9 + freq = self.reader.signal(self.signal, "frequencies")[0]*1e9 # Nyquist sample if self.fs == -1: bw = freq[-1] - freq[0] @@ -138,6 +140,9 @@ class Generator(Processor): def plot(self): print(self.output.shape) + def plotPafLayout(self): + self.reader.plotPaf() + @property def shape(self): return self.output.shape