draft

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