diff --git a/ext/bscacquisition.py b/ext/bscacquisition.py index da3ad3a..5c789d8 100644 --- a/ext/bscacquisition.py +++ b/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,7 +26,7 @@ class BSCAcquisition(Acquisition): #print('Received %i' % i) # write out the data file - use_channels = channels or self.bscache.channels + 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]