acoular 25.4__py3-none-any.whl → 25.10__py3-none-any.whl

acoular/spectra.py

@@ -3,6 +3,12 @@
 # ------------------------------------------------------------------------------
 """Estimation of power spectra and related tools.
 
+.. inheritance-diagram::
+                acoular.spectra
+    :top-classes:
+                acoular.spectra.BaseSpectra
+    :parts: 1
+
 .. autosummary::
     :toctree: generated/
 
@@ -13,26 +19,7 @@
 
 from abc import abstractmethod
 
-from numpy import (
-    arange,
-    array,
-    bartlett,
-    blackman,
-    dot,
-    empty,
-    fill_diagonal,
-    hamming,
-    hanning,
-    imag,
-    linalg,
-    ndarray,
-    newaxis,
-    ones,
-    real,
-    searchsorted,
-    sum,  # noqa A004
-    zeros,
-)
+import numpy as np
 from scipy import fft
 from traits.api import (
     ABCHasStrictTraits,
@@ -53,7 +40,6 @@ from traits.api import (
 # acoular imports
 from .base import SamplesGenerator
 from .configuration import config
-from .deprecation import deprecated_alias
 from .fastFuncs import calcCSM
 from .h5cache import H5cache
 from .h5files import H5CacheFileBase
@@ -61,8 +47,6 @@ from .internal import digest
 from .tools.utils import find_basename
 
 
-@deprecated_alias({'numchannels': 'num_channels'}, read_only=True)
-@deprecated_alias({'time_data': 'source'}, read_only=False)
 class BaseSpectra(ABCHasStrictTraits):
     """
     Base class for handling spectral data in Acoular.
@@ -93,7 +77,13 @@ class BaseSpectra(ABCHasStrictTraits):
     #:
     #: - ``'Blackman'``
     window = Map(
-        {'Rectangular': ones, 'Hanning': hanning, 'Hamming': hamming, 'Bartlett': bartlett, 'Blackman': blackman},
+        {
+            'Rectangular': np.ones,
+            'Hanning': np.hanning,
+            'Hamming': np.hamming,
+            'Bartlett': np.bartlett,
+            'Blackman': np.blackman,
+        },
         default_value='Rectangular',
         desc='type of window for FFT',
     )
@@ -180,7 +170,7 @@ class BaseSpectra(ABCHasStrictTraits):
     # generator that yields the time data blocks for every channel (with optional overlap)
     def _get_source_data(self):
         bs = self.block_size
-        temp = empty((2 * bs, self.num_channels))
+        temp = np.empty((2 * bs, self.num_channels))
         pos = bs
         posinc = bs / self.overlap_
         for data_block in self.source.result(bs):
@@ -292,7 +282,7 @@ class PowerSpectra(BaseSpectra):
         fftfreq = self.fftfreq()
         if fftfreq is not None:
             if self._ind_high is None:
-                return array([fftfreq[self.ind_low], None])
+                return np.array([fftfreq[self.ind_low], None])
             return fftfreq[[self.ind_low, self.ind_high]]
         return None
 
@@ -308,7 +298,7 @@ class PowerSpectra(BaseSpectra):
         if fftfreq is not None:
             if self._index_set_last:
                 return min(self._ind_low, fftfreq.shape[0] - 1)
-            return searchsorted(fftfreq[:-1], self._freqlc)
+            return np.searchsorted(fftfreq[:-1], self._freqlc)
         return 0
 
     @property_depends_on(['source.sample_freq', 'block_size', '_ind_high', '_freqhc'])
@@ -321,7 +311,7 @@ class PowerSpectra(BaseSpectra):
                 return min(self._ind_high, fftfreq.shape[0] - 1)
             if self._freqhc is None:
                 return None
-            return searchsorted(fftfreq[:-1], self._freqhc)
+            return np.searchsorted(fftfreq[:-1], self._freqhc)
         return None
 
     def _set_ind_high(self, ind_high):  # by setting this the user sets the lower index
@@ -337,7 +327,7 @@ class PowerSpectra(BaseSpectra):
         fftfreq = self.fftfreq()
         if fftfreq is not None:
             try:
-                indices = arange(fftfreq.shape[0], dtype=int)
+                indices = np.arange(fftfreq.shape[0], dtype=int)
                 if self.ind_high is None:
                     return indices[self.ind_low :]
                 return indices[self.ind_low : self.ind_high]
@@ -384,18 +374,18 @@ class PowerSpectra(BaseSpectra):
         """
         t = self.source
         wind = self.window_(self.block_size)
-        weight = dot(wind, wind)
-        wind = wind[newaxis, :].swapaxes(0, 1)
+        weight = np.dot(wind, wind)
+        wind = wind[np.newaxis, :].swapaxes(0, 1)
         numfreq = int(self.block_size / 2 + 1)
         csm_shape = (numfreq, t.num_channels, t.num_channels)
-        csm_upper = zeros(csm_shape, dtype=self.precision)
+        csm_upper = np.zeros(csm_shape, dtype=self.precision)
         # get time data blockwise
         for data in self._get_source_data():
             ft = fft.rfft(data * wind, None, 0).astype(self.precision)
             calcCSM(csm_upper, ft)  # only upper triangular part of matrix is calculated (for speed reasons)
         # create the full csm matrix via transposing and complex conj.
         csm_lower = csm_upper.conj().transpose(0, 2, 1)
-        [fill_diagonal(csm_lower[cntFreq, :, :], 0) for cntFreq in range(csm_lower.shape[0])]
+        [np.fill_diagonal(csm_lower[cntFreq, :, :], 0) for cntFreq in range(csm_lower.shape[0])]
         csm = csm_lower + csm_upper
         # onesided spectrum: multiplication by 2.0=sqrt(2)^2
         return csm * (2.0 / self.block_size / weight / self.num_blocks)
@@ -444,10 +434,10 @@ class PowerSpectra(BaseSpectra):
             eva_dtype = 'float32'
         #        csm = self.csm #trigger calculation
         csm_shape = self.csm.shape
-        eva = empty(csm_shape[0:2], dtype=eva_dtype)
-        eve = empty(csm_shape, dtype=self.precision)
+        eva = np.empty(csm_shape[0:2], dtype=eva_dtype)
+        eve = np.empty(csm_shape, dtype=self.precision)
         for i in range(csm_shape[0]):
-            (eva[i], eve[i]) = linalg.eigh(self.csm[i])
+            (eva[i], eve[i]) = np.linalg.eigh(self.csm[i])
         return (eva, eve)
 
     def calc_eva(self):
@@ -604,12 +594,12 @@ class PowerSpectra(BaseSpectra):
         f = self.fftfreq()
         if num == 0:
             # single frequency line
-            return self.eva[searchsorted(f, freq)]
-        f1 = searchsorted(f, freq * 2.0 ** (-0.5 / num))
-        f2 = searchsorted(f, freq * 2.0 ** (0.5 / num))
+            return self.eva[np.searchsorted(f, freq)]
+        f1 = np.searchsorted(f, freq * 2.0 ** (-0.5 / num))
+        f2 = np.searchsorted(f, freq * 2.0 ** (0.5 / num))
         if f1 == f2:
             return self.eva[f1]
-        return sum(self.eva[f1:f2], 0)
+        return np.sum(self.eva[f1:f2], 0)
 
 
 class PowerSpectraImport(PowerSpectra):
@@ -695,7 +685,7 @@ class PowerSpectraImport(PowerSpectra):
                 (number of frequencies, num_channels, num_channels)!'
                 raise ValueError(msg)
             self._csm = csm
-            self._csmsum = real(self._csm).sum() + (imag(self._csm) ** 2).sum()  # to trigger new digest creation
+            self._csmsum = np.real(self._csm).sum() + (np.imag(self._csm) ** 2).sum()  # to trigger new digest creation
 
     @property_depends_on(['digest'])
     def _get_eva(self):
@@ -718,7 +708,7 @@ class PowerSpectraImport(PowerSpectra):
             Array containing the frequencies.
         """
         if isinstance(self.frequencies, float):
-            return array([self.frequencies])
-        if isinstance(self.frequencies, ndarray):
+            return np.array([self.frequencies])
+        if isinstance(self.frequencies, np.ndarray):
             return self.frequencies
         return self.frequencies