ext/bscacquisition.py

@@ -8,10 +8,11 @@ from slic.core.acquisition.acquisition import Acquisition
 
 class BSCAcquisition(Acquisition):
 
-    def __init__(self, bscache, *args, **kwargs):
+    def __init__(self, bscache, *args, use_channels=None, **kwargs):
         self.bscache = bscache
+        self.use_channels = use_channels
         self.grp = 0
-        super().__init__(*args, **kwargs)
+        super().__init__(*args, default_channels=bscache.channels.keys(), **kwargs)
 
     def setGroup(self,idx):
         self.grp = idx
@@ -25,12 +26,13 @@ class BSCAcquisition(Acquisition):
             #print('Received %i' % i)
 
         # write out the data file
+        use_channels = self.use_channels or self.bscache.channels.keys()
         with h5py.File(filename,'a') as hid:
             # save the pulse ID
             singledata = [ele['pid'] for ele in data]
             pidname = 'pulse_id/group%d' % self.grp
             hid.create_dataset(pidname, data=singledata)
-            for chn in channels:
+            for chn in use_channels:
                 singledata = [ele[chn] for ele in data]
                 if not chn == 'pid':
                     dname = chn.replace(':','/')+'/data'