Updated linspaces to use LASP_NUMPY_FLOAT_TYPE.

This commit is contained in:
Anne de Jong 2024-07-03 11:56:40 +02:00
parent 6b0442fe90
commit 67bd7e6c9d
6 changed files with 37 additions and 26 deletions

View File

@ -15,6 +15,7 @@ import warnings
import numpy as np import numpy as np
# For designing second-order sections # For designing second-order sections
from scipy.signal import butter from scipy.signal import butter
from ..lasp_config import LASP_NUMPY_FLOAT_TYPE
from .fir_design import bandpass_fir_design from .fir_design import bandpass_fir_design
from .fir_design import freqResponse as firFreqResponse from .fir_design import freqResponse as firFreqResponse
@ -254,9 +255,9 @@ class FilterBankDesigner:
fuu = self.fu(xu) fuu = self.fu(xu)
if scale == 'lin': if scale == 'lin':
freq = np.linspace(fll, fuu, npoints) freq = np.linspace(fll, fuu, npoints, dtype=LASP_NUMPY_FLOAT_TYPE)
elif scale == 'log': elif scale == 'log':
freq = np.logspace(np.log10(fll), np.log10(fuu), npoints) freq = np.logspace(np.log10(fll), np.log10(fuu), npoints, dtype=LASP_NUMPY_FLOAT_TYPE)
else: else:
raise ValueError(f'Invalid scale parameter: {scale}') raise ValueError(f'Invalid scale parameter: {scale}')

View File

@ -7,6 +7,7 @@ from dataclasses import dataclass
from dataclasses_json import dataclass_json from dataclasses_json import dataclass_json
from enum import Enum, unique, auto from enum import Enum, unique, auto
from .lasp_cpp import DaqChannel from .lasp_cpp import DaqChannel
from .lasp_config import LASP_NUMPY_FLOAT_TYPE
""" """
Common definitions used throughout the code. Common definitions used throughout the code.
@ -395,7 +396,9 @@ def getTime(fs, N, start=0):
start: Optional start ofset in number of samples start: Optional start ofset in number of samples
""" """
assert N > 0 and fs > 0 assert N > 0 and fs > 0
return np.linspace(start, start + N/fs, N, endpoint=False) return np.linspace(
start, start + N / fs, N, endpoint=False, dtype=LASP_NUMPY_FLOAT_TYPE
)
def getFreq(fs, nfft): def getFreq(fs, nfft):
@ -408,4 +411,4 @@ def getFreq(fs, nfft):
""" """
df = fs / nfft # frequency resolution df = fs / nfft # frequency resolution
K = nfft // 2 + 1 # number of frequency bins K = nfft // 2 + 1 # number of frequency bins
return np.linspace(0, (K-1)*df, K) return np.linspace(0, (K - 1) * df, K, dtype=LASP_NUMPY_FLOAT_TYPE)

View File

@ -5,9 +5,12 @@ Author: J.A. de Jong - ASCEE
Description: LASP configuration Description: LASP configuration
""" """
import numpy as np import numpy as np
from .lasp_cpp import LASP_DOUBLE_PRECISION from .lasp_cpp import LASP_DOUBLE_PRECISION
__all__ = ["zeros", "ones", "empty", "LASP_NUMPY_FLOAT_TYPE", "LASP_NUMPY_COMPLEX_TYPE"]
if LASP_DOUBLE_PRECISION: if LASP_DOUBLE_PRECISION:
LASP_NUMPY_FLOAT_TYPE = np.float64 LASP_NUMPY_FLOAT_TYPE = np.float64
LASP_NUMPY_COMPLEX_TYPE = np.complex128 LASP_NUMPY_COMPLEX_TYPE = np.complex128
@ -16,28 +19,28 @@ else:
LASP_NUMPY_COMPLEX_TYPE = np.float64 LASP_NUMPY_COMPLEX_TYPE = np.float64
def zeros(shape, dtype=float, order='F'): def zeros(shape, dtype=float, order="F"):
if dtype == float: if dtype is float:
return np.zeros(shape, dtype=LASP_NUMPY_FLOAT_TYPE, order=order) return np.zeros(shape, dtype=LASP_NUMPY_FLOAT_TYPE, order=order)
elif dtype == complex: elif dtype is complex:
return np.zeros(shape, dtype=LASP_NUMPY_COMPLEX_TYPE, order=order) return np.zeros(shape, dtype=LASP_NUMPY_COMPLEX_TYPE, order=order)
else: else:
raise RuntimeError(f"Unknown dtype: {dtype}") raise RuntimeError(f"Unknown dtype: {dtype}")
def ones(shape, dtype=float, order='F'): def ones(shape, dtype=float, order="F"):
if dtype == float: if dtype is float:
return np.ones(shape, dtype=LASP_NUMPY_FLOAT_TYPE, order=order) return np.ones(shape, dtype=LASP_NUMPY_FLOAT_TYPE, order=order)
elif dtype == complex: elif dtype is complex:
return np.ones(shape, dtype=LASP_NUMPY_COMPLEX_TYPE, order=order) return np.ones(shape, dtype=LASP_NUMPY_COMPLEX_TYPE, order=order)
else: else:
raise RuntimeError(f"Unknown dtype: {dtype}") raise RuntimeError(f"Unknown dtype: {dtype}")
def empty(shape, dtype=float, order='F'):
if dtype == float: def empty(shape, dtype=float, order="F"):
if dtype is float:
return np.empty(shape, dtype=LASP_NUMPY_FLOAT_TYPE, order=order) return np.empty(shape, dtype=LASP_NUMPY_FLOAT_TYPE, order=order)
elif dtype == complex: elif dtype is complex:
return np.empty(shape, dtype=LASP_NUMPY_COMPLEX_TYPE, order=order) return np.empty(shape, dtype=LASP_NUMPY_COMPLEX_TYPE, order=order)
else: else:
raise RuntimeError(f"Unknown dtype: {dtype}") raise RuntimeError(f"Unknown dtype: {dtype}")

View File

@ -11,6 +11,7 @@ __all__ = ["OverallFilterBank", "SosOctaveFilterBank", "SosThirdOctaveFilterBank
from .filter.filterbank_design import OctaveBankDesigner, ThirdOctaveBankDesigner from .filter.filterbank_design import OctaveBankDesigner, ThirdOctaveBankDesigner
from .lasp_cpp import BiquadBank from .lasp_cpp import BiquadBank
from .lasp_config import empty, LASP_NUMPY_FLOAT_TYPE
import numpy as np import numpy as np
@ -46,7 +47,7 @@ class OverallFilterBank:
Ncur = data.shape[0] Ncur = data.shape[0]
tend = tstart + Ncur / self.fs tend = tstart + Ncur / self.fs
t = np.linspace(tstart, tend, Ncur, endpoint=False) t = np.linspace(tstart, tend, Ncur, endpoint=False, dtype=LASP_NUMPY_FLOAT_TYPE)
self.N += Ncur self.N += Ncur
output["Overall"] = {"t": t, "data": data, "x": 0} output["Overall"] = {"t": t, "data": data, "x": 0}
@ -114,7 +115,7 @@ class SosFilterBank:
Ncur = data.shape[0] Ncur = data.shape[0]
tend = tstart + Ncur / self.fs tend = tstart + Ncur / self.fs
t = np.linspace(tstart, tend, Ncur, endpoint=False) t = np.linspace(tstart, tend, Ncur, endpoint=False, dtype=LASP_NUMPY_FLOAT_TYPE)
self.N += Ncur self.N += Ncur
for i, x in enumerate(self.xs): for i, x in enumerate(self.xs):

View File

@ -4,14 +4,15 @@
Sound level meter implementation Sound level meter implementation
@author: J.A. de Jong - ASCEE @author: J.A. de Jong - ASCEE
""" """
from .lasp_cpp import cppSLM from .lasp_cpp import cppSLM
from .lasp_config import empty from .lasp_config import empty, LASP_NUMPY_FLOAT_TYPE
import numpy as np import numpy as np
from .lasp_common import (TimeWeighting, FreqWeighting, P_REF) from .lasp_common import TimeWeighting, FreqWeighting, P_REF
from .filter import SPLFilterDesigner from .filter import SPLFilterDesigner
import logging import logging
__all__ = ['SLM', 'Dummy'] __all__ = ["SLM", "Dummy"]
class Dummy: class Dummy:
@ -89,7 +90,7 @@ class SLM:
elif fw == FreqWeighting.Z: elif fw == FreqWeighting.Z:
prefilter = None prefilter = None
else: else:
raise ValueError(f'Not implemented prefilter {fw}') raise ValueError(f"Not implemented prefilter {fw}")
# 'Probe' size of filter coefficients # 'Probe' size of filter coefficients
self.nom_txt = [] self.nom_txt = []
@ -102,7 +103,7 @@ class SLM:
assert fbdesigner.fs == fs assert fbdesigner.fs == fs
sos_firstx = fbdesigner.createSOSFilter(self.xs[0]).flatten() sos_firstx = fbdesigner.createSOSFilter(self.xs[0]).flatten()
self.nom_txt.append(fbdesigner.nominal_txt(self.xs[0])) self.nom_txt.append(fbdesigner.nominal_txt(self.xs[0]))
sos = empty((sos_firstx.size, nfilters), dtype=float, order='C') sos = empty((sos_firstx.size, nfilters), dtype=float, order="C")
sos[:, 0] = sos_firstx sos[:, 0] = sos_firstx
for i, x in enumerate(self.xs[1:]): for i, x in enumerate(self.xs[1:]):

View File

@ -7,6 +7,7 @@ Weighting and calibration filter in one
from .lasp_common import FreqWeighting from .lasp_common import FreqWeighting
from .filter import SPLFilterDesigner from .filter import SPLFilterDesigner
from lasp.lasp_config import ones, empty from lasp.lasp_config import ones, empty
from ..lasp_config import LASP_NUMPY_FLOAT_TYPE
from .wrappers import FilterBank from .wrappers import FilterBank
import numpy as np import numpy as np
@ -39,7 +40,7 @@ class WeighCal:
self.calfile = calfile self.calfile = calfile
# Frequencies used for the filter design # Frequencies used for the filter design
freq_design = np.linspace(0, 17e3, 3000) freq_design = np.linspace(0, 17e3, 3000, dtype=LASP_NUMPY_FLOAT_TYPE)
freq_design[-1] = fs/2 freq_design[-1] = fs/2
# Objective function for the frequency response # Objective function for the frequency response
@ -140,6 +141,7 @@ class WeighCal:
""" """
Returns the frequency response of the designed FIR filter Returns the frequency response of the designed FIR filter
""" """
raise RuntimeError('This code bugs. TODO: Re-implement?')
if freq is None: if freq is None:
freq = np.logspace(1, np.log10(self.fs/2), 500) freq = np.logspace(1, np.log10(self.fs/2), 500)
return (freq, frp(self.fs, freq, self._firs[chan]), return (freq, frp(self.fs, freq, self._firs[chan]),