68 lines
2.1 KiB
Python
Executable File
68 lines
2.1 KiB
Python
Executable File
#!/usr/bin/python3.8
|
|
import sys, logging, os, argparse
|
|
parser = argparse.ArgumentParser(
|
|
description='Acquire data and store to a measurement file.'
|
|
)
|
|
parser.add_argument('filename', type=str,
|
|
help='File name to record to.'
|
|
' Extension is automatically added.')
|
|
parser.add_argument('--duration', '-d', type=float,
|
|
help='The recording duration in [s]')
|
|
|
|
device_help = 'DAQ Device to record from'
|
|
parser.add_argument('--input-daq', '-i', help=device_help, type=str,
|
|
default='Default')
|
|
parser.add_argument('--log', '-l',
|
|
help='Specify log level [info, debug, warning, ...]',
|
|
type=str, default='info')
|
|
|
|
|
|
args = parser.parse_args()
|
|
numeric_level = getattr(logging, args.log.upper(), None)
|
|
if not isinstance(numeric_level, int):
|
|
raise ValueError('Invalid log level: %s' % args.loglevel)
|
|
|
|
FORMAT = "[%(filename)s:%(lineno)s - %(funcName)20s() ] %(message)s"
|
|
logging.basicConfig(format=FORMAT, level=numeric_level)
|
|
|
|
import multiprocessing
|
|
from lasp import StreamMgr, Recording, DaqConfiguration
|
|
|
|
def main(args):
|
|
try:
|
|
streammgr = StreamMgr.getInstance()
|
|
configs = DaqConfigurations.loadAllConfigs()
|
|
|
|
config_keys = [key for key in configs.keys()]
|
|
for i, key in enumerate(config_keys):
|
|
print(f'{i:2} : {key}')
|
|
|
|
choosen_index = input('Number of configuration to use: ')
|
|
try:
|
|
daqindex = int(choosen_index)
|
|
except:
|
|
print('Invalid configuration number. Exiting.')
|
|
sys.exit(0)
|
|
|
|
choosen_key = config_keys[daqindex]
|
|
config = configs[choosen_key].input_config
|
|
|
|
print(f'Choosen configuration: {choosen_key}')
|
|
|
|
streammgr.startStream(AvType.audio_input, config, wait=True)
|
|
rec = Recording(args.filename, streammgr, args.duration)
|
|
|
|
streammgr.stopStream(AvType.audio_output)
|
|
finally:
|
|
try:
|
|
streammgr.cleanup()
|
|
del stream
|
|
except NameError:
|
|
pass
|
|
|
|
if __name__ == '__main__':
|
|
|
|
multiprocessing.set_start_method('forkserver', force=True)
|
|
|
|
main(args)
|