diff --git a/wa/instruments/energy_measurement.py b/wa/instruments/energy_measurement.py index fcda3a28..e205180f 100644 --- a/wa/instruments/energy_measurement.py +++ b/wa/instruments/energy_measurement.py @@ -144,7 +144,13 @@ class DAQBackend(EnergyInstrumentBackend): connector on the DAQ (varies between DAQ models). The default assumes DAQ 6363 and similar with AI channels on connectors 0-7 and 16-23. - """) + """), + Parameter('keep_raw', kind=bool, default=False, + description=""" + If set to ``True``, this will prevent the raw files obtained + from the device before processing from being deleted + (this is maily used for debugging). + """), ] instrument = DaqInstrument @@ -189,6 +195,12 @@ class EnergyProbeBackend(EnergyInstrumentBackend): description=""" Path to /dev entry for the energy probe (it should be /dev/ttyACMx) """), + Parameter('keep_raw', kind=bool, default=False, + description=""" + If set to ``True``, this will prevent the raw files obtained + from the device before processing from being deleted + (this is maily used for debugging). + """), ] instrument = EnergyProbeInstrument @@ -224,6 +236,12 @@ class ArmEnergyProbeBackend(EnergyInstrumentBackend): description=""" Path to config file of the AEP """), + Parameter('keep_raw', kind=bool, default=False, + description=""" + If set to ``True``, this will prevent the raw files obtained + from the device before processing from being deleted + (this is maily used for debugging). + """), ] instrument = ArmEnergyProbeInstrument @@ -282,6 +300,12 @@ class AcmeCapeBackend(EnergyInstrumentBackend): description=""" Size of the capture buffer (in KB). """), + Parameter('keep_raw', kind=bool, default=False, + description=""" + If set to ``True``, this will prevent the raw files obtained + from the device before processing from being deleted + (this is maily used for debugging). + """), ] # pylint: disable=arguments-differ @@ -307,7 +331,7 @@ class AcmeCapeBackend(EnergyInstrumentBackend): for iio_device in iio_devices: ret[iio_device] = AcmeCapeInstrument( target, iio_capture=iio_capture, host=host, - iio_device=iio_device, buffer_size=buffer_size) + iio_device=iio_device, buffer_size=buffer_size, keep_raw=keep_raw) return ret