Updated linspaces to use LASP_NUMPY_FLOAT_TYPE.

python_src/lasp/filter/filterbank_design.py

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

python_src/lasp/lasp_common.py

@@ -7,6 +7,7 @@ from dataclasses import dataclass
 from dataclasses_json import dataclass_json
 from enum import Enum, unique, auto
 from .lasp_cpp import DaqChannel
+from .lasp_config import LASP_NUMPY_FLOAT_TYPE
 
 """
 Common definitions used throughout the code.
@@ -395,7 +396,9 @@ def getTime(fs, N, start=0):
         start: Optional start ofset in number of samples
     """
     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):
@@ -406,6 +409,6 @@ def getFreq(fs, nfft):
         fs: Sampling frequency [Hz]
         nfft: Fft length (int)
     """
-    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)

python_src/lasp/lasp_config.py

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

python_src/lasp/lasp_octavefilter.py

@@ -11,6 +11,7 @@ __all__ = ["OverallFilterBank", "SosOctaveFilterBank", "SosThirdOctaveFilterBank
 
 from .filter.filterbank_design import OctaveBankDesigner, ThirdOctaveBankDesigner
 from .lasp_cpp import BiquadBank
+from .lasp_config import empty, LASP_NUMPY_FLOAT_TYPE
 import numpy as np
 
 
@@ -46,7 +47,7 @@ class OverallFilterBank:
         Ncur = data.shape[0]
         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
 
         output["Overall"] = {"t": t, "data": data, "x": 0}
@@ -114,7 +115,7 @@ class SosFilterBank:
         Ncur = data.shape[0]
         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
 
         for i, x in enumerate(self.xs):

python_src/lasp/lasp_slm.py

@@ -4,14 +4,15 @@
 Sound level meter implementation
 @author: J.A. de Jong - ASCEE
 """
+
 from .lasp_cpp import cppSLM
-from .lasp_config import empty
+from .lasp_config import empty, LASP_NUMPY_FLOAT_TYPE
 import numpy as np
-from .lasp_common import (TimeWeighting, FreqWeighting, P_REF)
+from .lasp_common import TimeWeighting, FreqWeighting, P_REF
 from .filter import SPLFilterDesigner
 import logging
 
-__all__ = ['SLM', 'Dummy']
+__all__ = ["SLM", "Dummy"]
 
 
 class Dummy:
@@ -89,24 +90,24 @@ class SLM:
         elif fw == FreqWeighting.Z:
             prefilter = None
         else:
-            raise ValueError(f'Not implemented prefilter {fw}')
+            raise ValueError(f"Not implemented prefilter {fw}")
 
         # 'Probe' size of filter coefficients
         self.nom_txt = []
 
         # This is a bit of a hack, as the 5 is hard-encoded here, but should in
         # fact be coming from somewhere else..
-        sos_overall = np.array([1, 0, 0, 1, 0, 0]*5, dtype=float)
+        sos_overall = np.array([1, 0, 0, 1, 0, 0] * 5, dtype=float)
 
         if fbdesigner is not None:
             assert fbdesigner.fs == fs
             sos_firstx = fbdesigner.createSOSFilter(self.xs[0]).flatten()
             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
 
             for i, x in enumerate(self.xs[1:]):
-                sos[:, i+1] = fbdesigner.createSOSFilter(x).flatten()
+                sos[:, i + 1] = fbdesigner.createSOSFilter(x).flatten()
                 self.nom_txt.append(fbdesigner.nominal_txt(x))
 
             if include_overall:

python_src/lasp/lasp_weighcal.py

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