diff --git a/ext/bscacquisition.py b/ext/bscacquisition.py index 28efa25..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,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'