lasprs 0.2.1

Library for Acoustic Signal Processing (Rust edition, with optional Python bindings via pyo3)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62

#!/usr/bin/env python3
import numpy as np
from lasprs.filter import Biquad, SeriesBiquad, BiquadBank


def test_biquad1():
    """
    Test if manual unit biquad matches response for manual input biquad.
    """
    input_ = np.array([1.0, 0, 0, 0, 0, 0, 0])

    b = Biquad(np.array([1., 0, 0, 1, 0, 0]))
    b2 = Biquad.unit()
    out1 = b.filter(input_)
    out2 = b2.filter(input_)

    expected_output = input_
    assert(np.linalg.norm(out1 - out2) == 0.)
    assert(np.linalg.norm(out1 - expected_output) == 0.)

def test_seriesbiquad():
    """
    Test if manual unit biquad matches response for manual input biquad.
    """
    input_ = np.array([1.0, 0, 0, 0, 0, 0, 0])

    f1 = [1., 0, 0, 1, 0, 0]
    f2 = [1., 0, 0, 1, 0, 0]
    f = f1+f2

    # Two biquads in series
    b = SeriesBiquad(np.array(f))

    # Single one
    b2 = SeriesBiquad.unit()

    out1 = b.filter(input_)
    out2 = b2.filter(input_)

    expected_output = input_
    assert(np.linalg.norm(out1 - out2) == 0.)
    assert(np.linalg.norm(out1 - expected_output) == 0.)

def test_biquadbank():
    """
    See if two filters with half gain produce the output=input
    """
    input_ = np.array([1.0, 0, 0, 0, 0, 0, 0])
    f1 = [1., 0, 0, 1, 0, 0]
    f2 = [1., 0, 0, 1, 0, 0]
    filters = np.array([f1, f2]).T
    bank = BiquadBank(filters)
    bank.set_gains([0.5, 0.5])
    filtered = bank.filter(input_)
    assert(bank.len() == 2)
    assert(np.linalg.norm(input_ - filtered) == 0)


if __name__ == '__main__':
    test_biquad1()
    test_seriesbiquad()
    test_biquadbank()