147 lines
4.3 KiB
Python
147 lines
4.3 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
"""!
|
|
Author: J.A. de Jong - ASCEE
|
|
|
|
Description:
|
|
|
|
Data Acquistiion (DAQ) device descriptors, and the DAQ devices themselves
|
|
|
|
"""
|
|
from dataclasses import dataclass, field
|
|
import numpy as np
|
|
|
|
|
|
@dataclass
|
|
class DeviceInfo:
|
|
index: int
|
|
probed: bool
|
|
name: str
|
|
outputchannels: int
|
|
inputchannels: int
|
|
duplexchannels: int
|
|
samplerates: list
|
|
sampleformats: list
|
|
prefsamplerate: int
|
|
|
|
|
|
from ..lasp_common import lasp_shelve
|
|
|
|
@dataclass
|
|
class DAQInputChannel:
|
|
channel_enabled: bool
|
|
channel_name: str
|
|
sensitivity: float
|
|
|
|
|
|
|
|
|
|
@dataclass
|
|
class DAQConfiguration:
|
|
"""
|
|
Initialize a device descriptor
|
|
|
|
Args:
|
|
duplex_mode: Set device to duplex mode, if possible
|
|
monitor_gen: If set to true, add monitor channel to recording.
|
|
input_device_name: ASCII name with which to open the device when connected
|
|
outut_device_name: ASCII name with which to open the device when connected
|
|
|
|
==============================
|
|
en_format: index of the format in the list of sample formats
|
|
en_input_rate: index of enabled input sampling frequency [Hz]
|
|
in the list of frequencies.
|
|
input_channel_configs: list of channel indices which are used to
|
|
acquire data from.
|
|
input_sensitivity: List of sensitivity values, in units of [Pa^-1]
|
|
input_gain_settings: If a DAQ supports it, list of indices which
|
|
corresponds to a position in the possible input
|
|
gains for each channel. Should only be not equal
|
|
to None when the hardware supports changing the
|
|
input gain.
|
|
en_output_rate: index in the list of possible output sampling
|
|
frequencies.
|
|
en_output_channels: list of enabled output channels
|
|
===============================
|
|
|
|
|
|
"""
|
|
duplex_mode: bool
|
|
|
|
input_device_name: str
|
|
output_device_name: str
|
|
|
|
en_input_sample_format: str
|
|
en_output_sample_format: str
|
|
en_input_rate: int
|
|
en_output_rate: int
|
|
|
|
input_channel_configs: list = None
|
|
monitor_gen: bool = False
|
|
|
|
|
|
def __post_init__(self):
|
|
"""
|
|
We do a check here to see whether the list of enabled channels is
|
|
contiguous. Non-contiguous is not yet implemented in RtAudio backend.
|
|
"""
|
|
en_input = self.input_channel_configs
|
|
first_ch_enabled_found = False
|
|
ch_disabled_found_after = False
|
|
print(en_input)
|
|
for ch in en_input:
|
|
if ch.channel_enabled:
|
|
first_ch_enabled_found = True
|
|
if ch_disabled_found_after:
|
|
raise ValueError('Error: non-contiguous array of channels'
|
|
' found. This is not yet implemented in'
|
|
' backend')
|
|
else:
|
|
if first_ch_enabled_found:
|
|
ch_disabled_found_after = True
|
|
|
|
def firstEnabledInputChannelNumber(self):
|
|
"""
|
|
Returns the channel number of the first enabled channel. Returns -1 if
|
|
no channels are enabled.
|
|
"""
|
|
for i, ch in enumerate(self.input_channel_configs):
|
|
if ch.channel_enabled:
|
|
return i
|
|
return -1
|
|
|
|
|
|
def getEnabledChannels(self):
|
|
en_channels = []
|
|
for chan in self.input_channel_configs:
|
|
if chan.channel_enabled:
|
|
en_channels.append(chan)
|
|
return en_channels
|
|
|
|
def getEnabledChannelSensitivities(self):
|
|
return np.array(
|
|
[float(channel.sensitivity) for channel in
|
|
self.getEnabledChannels()])
|
|
|
|
@staticmethod
|
|
def loadConfigs():
|
|
"""
|
|
Returns a list of currently available configurations
|
|
"""
|
|
with lasp_shelve() as sh:
|
|
return sh.load('daqconfigs', {})
|
|
|
|
def saveConfig(self, name):
|
|
with lasp_shelve() as sh:
|
|
cur_configs = self.loadConfigs()
|
|
cur_configs[name] = self
|
|
sh.store('daqconfigs', cur_configs)
|
|
|
|
@staticmethod
|
|
def deleteConfig(name):
|
|
with lasp_shelve() as sh:
|
|
cur_configs = DAQConfiguration.loadConfigs()
|
|
del cur_configs[name]
|
|
sh.store('daqconfigs', cur_configs)
|
|
|