flamo 0.2.9__py3-none-any.whl → 0.2.10__py3-none-any.whl

flamo/processor/dsp.py

@@ -3520,3 +3520,80 @@ class GainDelay(DSP):
         delay_samples = delay_samples.to(gain.real.dtype)
         phase = torch.einsum("fo, omn -> fmn", self.omega, delay_samples.unsqueeze(0))
         return gain.unsqueeze(0) * (self.gamma ** delay_samples) * torch.exp(-1j * phase)
+
+
+class parallelGainDelay(GainDelay):
+    """
+    Parallel counterpart of the :class:`GainDelay` class.
+    For information about **attributes** and **methods** see :class:`GainDelay`.
+
+    Shape:
+        - input: :math:`(B, M, N, ...)`
+        - param: :math:`(2, N)`
+        - output: :math:`(B, M, N, ...)`
+
+    where :math:`B` is the batch size, :math:`M` is the number of frequency bins,
+    and :math:`N` is the number of input channels.
+    Ellipsis :math:`(...)` represents additional dimensions.
+    """
+
+    def __init__(
+        self,
+        size: tuple = (1,),
+        max_len: int = 2000,
+        isint: bool = False,
+        unit: int = 100,
+        nfft: int = 2**11,
+        fs: int = 48000,
+        map_gain: Optional[callable] = None,
+        map_delay: Optional[callable] = None,
+        requires_grad: bool = False,
+        alias_decay_db: float = 0.0,
+        device: Optional[str] = None,
+        dtype: torch.dtype = torch.float32,
+    ):
+        super().__init__(
+            size=size,
+            max_len=max_len,
+            isint=isint,
+            unit=unit,
+            nfft=nfft,
+            fs=fs,
+            map_gain=map_gain,
+            map_delay=map_delay,
+            requires_grad=requires_grad,
+            alias_decay_db=alias_decay_db,
+            device=device,
+            dtype=dtype,
+        )
+
+    def check_param_shape(self):
+        """
+        Checks if the shape of the gain-delay parameters is valid.
+        """
+        assert (
+            len(self.size) == 2 and self.size[0] == 2
+        ), "parallelGainDelay parameters must have shape (2, N), for MIMO use GainDelay module."
+
+    def get_freq_convolve(self):
+        """
+        Computes the frequency convolution function for parallel gain-delay processing.
+        """
+        self.freq_convolve = lambda x, param: torch.einsum(
+            "fn,bfn...->bfn...", self.freq_response(param), x
+        )
+
+    def get_io(self):
+        """
+        Computes the number of input and output channels based on the size parameter.
+        """
+        self.input_channels = self.size[-1]
+        self.output_channels = self.size[-1]
+
+    def _combine_gain_delay(self, gain: torch.Tensor, delay_samples: torch.Tensor):
+        """
+        Combines gain and delay for parallel processing (element-wise).
+        """
+        delay_samples = delay_samples.to(gain.real.dtype)
+        phase = torch.einsum("fo, on -> fn", self.omega, delay_samples.unsqueeze(0))
+        return gain.unsqueeze(0) * (self.gamma ** delay_samples) * torch.exp(-1j * phase)

{flamo-0.2.9.dist-info → flamo-0.2.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flamo
-Version: 0.2.9
+Version: 0.2.10
 Summary: An Open-Source Library for Frequency-Domain Differentiable Audio Processing
 Project-URL: Homepage, https://github.com/gdalsanto/flamo
 Project-URL: Issues, https://github.com/gdalsanto/flamo/issues
@@ -50,7 +50,7 @@ Description-Content-Type: text/markdown
 
 Open-source library for frequency-domain differentiable audio processing.
 
-It contains differentiable implementation of common LTI audio systems modules with learnable parameters.
+It contains differentiable implementation of common LTI audio system modules with learnable parameters.
 
 ---
 
@@ -58,7 +58,7 @@ It contains differentiable implementation of common LTI audio systems modules wi
 
 Available differentiable audio signal processors - in `flamo.processor.dsp`: 
 - **Gains** : Gains, Matrices, Householder Matrices
-- **Filters** : Biquads, State Variable Filters (SVF), Graphic Equalizers (GEQ), Parametric Equiliers (PEQ - not released yet)
+- **Filters** : Biquads, State Variable Filters (SVF), Graphic Equalizers (GEQ), Parametric Equalizers (PEQ - not released yet)
 - **Delays** : Integer Delays, Fractional Delays 
 
 Transforms  - in `flamo.processor.dsp`: 
@@ -77,7 +77,7 @@ Optimization - in `flamo.optimize`:
 --- 
 
 ### 🛠️ Installation
-To install it via pip, on a new python virtual environment `flamo-env` 
+To install it via pip, on a new Python virtual environment `flamo-env` 
 ```shell
 python3.10 -m venv .flamo-env
 source .flamo-env/bin/activate
@@ -91,7 +91,7 @@ pip install flamo
 conda install -c conda-forge libsndfile
 ```
 
-For local installation: clone and install dependencies on a new pyton virtual environment `flamo-env` 
+For local installation: clone and install dependencies on a new Python virtual environment `flamo-env` 
 ```shell
 git clone https://github.com/gdalsanto/flamo
 cd flamo
@@ -160,7 +160,7 @@ model = system.Shell(core=filt, input_layer=input_layer, output_layer=output_lay
 estimation_init = model.get_freq_response()
 ````
 
-Set up optimization framework and launch it. The `Trainer` class is used to contain the model, training parameters, and training/valid steps in one class. 
+Set up the optimization framework and launch it. The `Trainer` class is used to contain the model, training parameters, and training/valid steps in one class. 
 
 ```python
 input = signal_gallery(1, n_samples=nfft, n=in_ch, signal_type='impulse', fs=fs)

{flamo-0.2.9.dist-info → flamo-0.2.10.dist-info}/RECORD

@@ -16,9 +16,9 @@ flamo/optimize/surface.py,sha256=sWy1ImwxUh_QLoY6S68LXBa82_HdWJGplFg2ObtpNGc,266
 flamo/optimize/trainer.py,sha256=LITPVS87mI6bnq4J6GIXqGb4wW7TKWVXeCu4UQ-csxM,12155
 flamo/optimize/utils.py,sha256=R5-KoZagRho3eykY88pC3UB2mc5SsE4Yv9X-ogskXdA,1610
 flamo/processor/__init__.py,sha256=paGdxGVZgA2VAs0tBwRd0bobzGxeyK79DS7ZGO8drkI,41
-flamo/processor/dsp.py,sha256=Znp_9qjHRb7V0DBaqPWxa9oOlU84CVTjy6h3DcAh-TU,144811
+flamo/processor/dsp.py,sha256=N9IZNO7taV9zwx67g1rVoQEedF4EVHXqGp23Z0BqohA,147358
 flamo/processor/system.py,sha256=Hct-o6IgF5NQ2xYbX-1j3st94hMoM8dOgAzle2gjDqU,43145
-flamo-0.2.9.dist-info/METADATA,sha256=qS-4mXx4gdV_e4xtUIuQ3oGKknKqHtPXLJZL8MCeLpg,7825
-flamo-0.2.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-flamo-0.2.9.dist-info/licenses/LICENSE,sha256=smMocRH7xdPT5RvFNqSLtbSNzohXJM5G_rX1Qaej6vg,1120
-flamo-0.2.9.dist-info/RECORD,,
+flamo-0.2.10.dist-info/METADATA,sha256=E3OE6I1j_xJH8IM-BGZZhkzPfu5beXI2jpYQDz98Cko,7831
+flamo-0.2.10.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+flamo-0.2.10.dist-info/licenses/LICENSE,sha256=smMocRH7xdPT5RvFNqSLtbSNzohXJM5G_rX1Qaej6vg,1120
+flamo-0.2.10.dist-info/RECORD,,

{flamo-0.2.9.dist-info → flamo-0.2.10.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.27.0
+Generator: hatchling 1.28.0
 Root-Is-Purelib: true
 Tag: py3-none-any