use crate::{Error, Result};
#[derive(Debug, Clone, Copy, PartialEq, Default)]
pub struct Complex {
pub re: f64,
pub im: f64,
}
impl Complex {
#[inline]
#[must_use]
pub fn new(re: f64, im: f64) -> Self {
Complex { re, im }
}
#[inline]
#[must_use]
pub fn conj(self) -> Self {
Complex {
re: self.re,
im: -self.im,
}
}
#[inline]
#[must_use]
pub fn norm_sqr(self) -> f64 {
self.re * self.re + self.im * self.im
}
}
impl core::ops::Add for Complex {
type Output = Complex;
#[inline]
fn add(self, rhs: Complex) -> Complex {
Complex::new(self.re + rhs.re, self.im + rhs.im)
}
}
impl core::ops::Sub for Complex {
type Output = Complex;
#[inline]
fn sub(self, rhs: Complex) -> Complex {
Complex::new(self.re - rhs.re, self.im - rhs.im)
}
}
impl core::ops::Mul for Complex {
type Output = Complex;
#[inline]
fn mul(self, rhs: Complex) -> Complex {
Complex::new(
self.re * rhs.re - self.im * rhs.im,
self.re * rhs.im + self.im * rhs.re,
)
}
}
impl core::ops::Mul<f64> for Complex {
type Output = Complex;
#[inline]
fn mul(self, rhs: f64) -> Complex {
Complex::new(self.re * rhs, self.im * rhs)
}
}
impl core::ops::AddAssign for Complex {
#[inline]
fn add_assign(&mut self, rhs: Complex) {
self.re += rhs.re;
self.im += rhs.im;
}
}
#[rustfmt::skip]
pub const QMF_WINDOW: [f64; 640] = [
0.0000000000, -0.0005525286, -0.0005617692, -0.0004947518,
-0.0004875227, -0.0004893791, -0.0005040714, -0.0005226564,
-0.0005466565, -0.0005677802, -0.0005870930, -0.0006132747,
-0.0006312493, -0.0006540333, -0.0006777690, -0.0006941614,
-0.0007157736, -0.0007255043, -0.0007440941, -0.0007490598,
-0.0007681371, -0.0007724848, -0.0007834332, -0.0007779869,
-0.0007803664, -0.0007801449, -0.0007757977, -0.0007630793,
-0.0007530001, -0.0007319357, -0.0007215391, -0.0006917937,
-0.0006650415, -0.0006341594, -0.0005946118, -0.0005564576,
-0.0005145572, -0.0004606325, -0.0004095121, -0.0003501175,
-0.0002896981, -0.0002098337, -0.0001446380, -0.0000617334,
0.0000134949, 0.0001094383, 0.0002043017, 0.0002949531,
0.0004026540, 0.0005107388, 0.0006239376, 0.0007458025,
0.0008608443, 0.0009885988, 0.0011250155, 0.0012577884,
0.0013902494, 0.0015443219, 0.0016868083, 0.0018348265,
0.0019841140, 0.0021461583, 0.0023017254, 0.0024625616,
0.0026201758, 0.0027870464, 0.0029469447, 0.0031125420,
0.0032739613, 0.0034418874, 0.0036008268, 0.0037603922,
0.0039207432, 0.0040819753, 0.0042264269, 0.0043730719,
0.0045209852, 0.0046606460, 0.0047932560, 0.0049137603,
0.0050393022, 0.0051407353, 0.0052461166, 0.0053471681,
0.0054196775, 0.0054876040, 0.0055475714, 0.0055938023,
0.0056220643, 0.0056455196, 0.0056389199, 0.0056266114,
0.0055917128, 0.0055404363, 0.0054753783, 0.0053838975,
0.0052715758, 0.0051382275, 0.0049839687, 0.0048109469,
0.0046039530, 0.0043801861, 0.0041251642, 0.0038456408,
0.0035401246, 0.0032091885, 0.0028446757, 0.0024508540,
0.0020274176, 0.0015784682, 0.0010902329, 0.0005832264,
0.0000276045, -0.0005464280, -0.0011568135, -0.0018039472,
-0.0024826723, -0.0031933778, -0.0039401124, -0.0047222596,
-0.0055337211, -0.0063792293, -0.0072615816, -0.0081798233,
-0.0091325329, -0.0101150215, -0.0111315548, -0.0121849995,
0.0132718220, 0.0143904666, 0.0155405553, 0.0167324712,
0.0179433381, 0.0191872431, 0.0204531793, 0.0217467550,
0.0230680169, 0.0244160992, 0.0257875847, 0.0271859429,
0.0286072173, 0.0300502657, 0.0315017608, 0.0329754081,
0.0344620948, 0.0359697560, 0.0374812850, 0.0390053679,
0.0405349170, 0.0420649094, 0.0436097542, 0.0451488405,
0.0466843027, 0.0482165720, 0.0497385755, 0.0512556155,
0.0527630746, 0.0542452768, 0.0557173648, 0.0571616450,
0.0585915683, 0.0599837480, 0.0613455171, 0.0626857808,
0.0639715898, 0.0652247106, 0.0664367512, 0.0676075985,
0.0687043828, 0.0697630244, 0.0707628710, 0.0717002673,
0.0725682583, 0.0733620255, 0.0741003642, 0.0747452558,
0.0753137336, 0.0758008358, 0.0761992479, 0.0764992170,
0.0767093490, 0.0768173975, 0.0768230011, 0.0767204924,
0.0765050718, 0.0761748321, 0.0757305756, 0.0751576255,
0.0744664394, 0.0736406005, 0.0726774642, 0.0715826364,
0.0703533073, 0.0689664013, 0.0674525021, 0.0657690668,
0.0639444805, 0.0619602779, 0.0598166570, 0.0575152691,
0.0550460034, 0.0524093821, 0.0495978676, 0.0466303305,
0.0434768782, 0.0401458278, 0.0366418116, 0.0329583930,
0.0290824006, 0.0250307561, 0.0207997072, 0.0163701258,
0.0117623832, 0.0069636862, 0.0019765601, -0.0032086896,
-0.0085711749, -0.0141288827, -0.0198834129, -0.0258227288,
-0.0319531274, -0.0382776572, -0.0447806821, -0.0514804176,
-0.0583705326, -0.0654409853, -0.0726943300, -0.0801372934,
-0.0877547536, -0.0955533352, -0.1035329531, -0.1116826931,
-0.1200077984, -0.1285002850, -0.1371551761, -0.1459766491,
-0.1549607071, -0.1640958855, -0.1733808172, -0.1828172548,
-0.1923966745, -0.2021250176, -0.2119735853, -0.2219652696,
-0.2320690870, -0.2423016884, -0.2526480309, -0.2631053299,
-0.2736634040, -0.2843214189, -0.2950716717, -0.3059098575,
-0.3168278913, -0.3278113727, -0.3388722693, -0.3499914122,
0.3611589903, 0.3723795546, 0.3836350013, 0.3949211761,
0.4062317676, 0.4175696896, 0.4289119920, 0.4402553754,
0.4515996535, 0.4629308085, 0.4742453214, 0.4855253091,
0.4967708254, 0.5079817500, 0.5191234970, 0.5302240895,
0.5412553448, 0.5522051258, 0.5630789140, 0.5738524131,
0.5845403235, 0.5951123086, 0.6055783538, 0.6159109932,
0.6261242695, 0.6361980107, 0.6461269695, 0.6559016302,
0.6655139880, 0.6749663190, 0.6842353293, 0.6933282376,
0.7022388719, 0.7109410426, 0.7194462634, 0.7277448900,
0.7358211758, 0.7436827863, 0.7513137456, 0.7587080760,
0.7658674865, 0.7727780881, 0.7794287519, 0.7858353120,
0.7919735841, 0.7978466413, 0.8034485751, 0.8087695004,
0.8138191270, 0.8185776004, 0.8230419890, 0.8272275347,
0.8311038457, 0.8346937361, 0.8379717337, 0.8409541392,
0.8436238281, 0.8459818469, 0.8480315777, 0.8497805198,
0.8511971524, 0.8523047035, 0.8531020949, 0.8535720573,
0.8537385600, 0.8535720573, 0.8531020949, 0.8523047035,
0.8511971524, 0.8497805198, 0.8480315777, 0.8459818469,
0.8436238281, 0.8409541392, 0.8379717337, 0.8346937361,
0.8311038457, 0.8272275347, 0.8230419890, 0.8185776004,
0.8138191270, 0.8087695004, 0.8034485751, 0.7978466413,
0.7919735841, 0.7858353120, 0.7794287519, 0.7727780881,
0.7658674865, 0.7587080760, 0.7513137456, 0.7436827863,
0.7358211758, 0.7277448900, 0.7194462634, 0.7109410426,
0.7022388719, 0.6933282376, 0.6842353293, 0.6749663190,
0.6655139880, 0.6559016302, 0.6461269695, 0.6361980107,
0.6261242695, 0.6159109932, 0.6055783538, 0.5951123086,
0.5845403235, 0.5738524131, 0.5630789140, 0.5522051258,
0.5412553448, 0.5302240895, 0.5191234970, 0.5079817500,
0.4967708254, 0.4855253091, 0.4742453214, 0.4629308085,
0.4515996535, 0.4402553754, 0.4289119920, 0.4175696896,
0.4062317676, 0.3949211761, 0.3836350013, 0.3723795546,
-0.3611589903, -0.3499914122, -0.3388722693, -0.3278113727,
-0.3168278913, -0.3059098575, -0.2950716717, -0.2843214189,
-0.2736634040, -0.2631053299, -0.2526480309, -0.2423016884,
-0.2320690870, -0.2219652696, -0.2119735853, -0.2021250176,
-0.1923966745, -0.1828172548, -0.1733808172, -0.1640958855,
-0.1549607071, -0.1459766491, -0.1371551761, -0.1285002850,
-0.1200077984, -0.1116826931, -0.1035329531, -0.0955533352,
-0.0877547536, -0.0801372934, -0.0726943300, -0.0654409853,
-0.0583705326, -0.0514804176, -0.0447806821, -0.0382776572,
-0.0319531274, -0.0258227288, -0.0198834129, -0.0141288827,
-0.0085711749, -0.0032086896, 0.0019765601, 0.0069636862,
0.0117623832, 0.0163701258, 0.0207997072, 0.0250307561,
0.0290824006, 0.0329583930, 0.0366418116, 0.0401458278,
0.0434768782, 0.0466303305, 0.0495978676, 0.0524093821,
0.0550460034, 0.0575152691, 0.0598166570, 0.0619602779,
0.0639444805, 0.0657690668, 0.0674525021, 0.0689664013,
0.0703533073, 0.0715826364, 0.0726774642, 0.0736406005,
0.0744664394, 0.0751576255, 0.0757305756, 0.0761748321,
0.0765050718, 0.0767204924, 0.0768230011, 0.0768173975,
0.0767093490, 0.0764992170, 0.0761992479, 0.0758008358,
0.0753137336, 0.0747452558, 0.0741003642, 0.0733620255,
0.0725682583, 0.0717002673, 0.0707628710, 0.0697630244,
0.0687043828, 0.0676075985, 0.0664367512, 0.0652247106,
0.0639715898, 0.0626857808, 0.0613455171, 0.0599837480,
0.0585915683, 0.0571616450, 0.0557173648, 0.0542452768,
0.0527630746, 0.0512556155, 0.0497385755, 0.0482165720,
0.0466843027, 0.0451488405, 0.0436097542, 0.0420649094,
0.0405349170, 0.0390053679, 0.0374812850, 0.0359697560,
0.0344620948, 0.0329754081, 0.0315017608, 0.0300502657,
0.0286072173, 0.0271859429, 0.0257875847, 0.0244160992,
0.0230680169, 0.0217467550, 0.0204531793, 0.0191872431,
0.0179433381, 0.0167324712, 0.0155405553, 0.0143904666,
-0.0132718220, -0.0121849995, -0.0111315548, -0.0101150215,
-0.0091325329, -0.0081798233, -0.0072615816, -0.0063792293,
-0.0055337211, -0.0047222596, -0.0039401124, -0.0031933778,
-0.0024826723, -0.0018039472, -0.0011568135, -0.0005464280,
0.0000276045, 0.0005832264, 0.0010902329, 0.0015784682,
0.0020274176, 0.0024508540, 0.0028446757, 0.0032091885,
0.0035401246, 0.0038456408, 0.0041251642, 0.0043801861,
0.0046039530, 0.0048109469, 0.0049839687, 0.0051382275,
0.0052715758, 0.0053838975, 0.0054753783, 0.0055404363,
0.0055917128, 0.0056266114, 0.0056389199, 0.0056455196,
0.0056220643, 0.0055938023, 0.0055475714, 0.0054876040,
0.0054196775, 0.0053471681, 0.0052461166, 0.0051407353,
0.0050393022, 0.0049137603, 0.0047932560, 0.0046606460,
0.0045209852, 0.0043730719, 0.0042264269, 0.0040819753,
0.0039207432, 0.0037603922, 0.0036008268, 0.0034418874,
0.0032739613, 0.0031125420, 0.0029469447, 0.0027870464,
0.0026201758, 0.0024625616, 0.0023017254, 0.0021461583,
0.0019841140, 0.0018348265, 0.0016868083, 0.0015443219,
0.0013902494, 0.0012577884, 0.0011250155, 0.0009885988,
0.0008608443, 0.0007458025, 0.0006239376, 0.0005107388,
0.0004026540, 0.0002949531, 0.0002043017, 0.0001094383,
0.0000134949, -0.0000617334, -0.0001446380, -0.0002098337,
-0.0002896981, -0.0003501175, -0.0004095121, -0.0004606325,
-0.0005145572, -0.0005564576, -0.0005946118, -0.0006341594,
-0.0006650415, -0.0006917937, -0.0007215391, -0.0007319357,
-0.0007530001, -0.0007630793, -0.0007757977, -0.0007801449,
-0.0007803664, -0.0007779869, -0.0007834332, -0.0007724848,
-0.0007681371, -0.0007490598, -0.0007440941, -0.0007255043,
-0.0007157736, -0.0006941614, -0.0006777690, -0.0006540333,
-0.0006312493, -0.0006132747, -0.0005870930, -0.0005677802,
-0.0005466565, -0.0005226564, -0.0005040714, -0.0004893791,
-0.0004875227, -0.0004947518, -0.0005617692, -0.000552528,
];
#[derive(Debug, Clone)]
pub struct AnalysisQmf {
x: Vec<f64>,
m: Vec<Complex>,
}
impl Default for AnalysisQmf {
fn default() -> Self {
Self::new()
}
}
impl AnalysisQmf {
#[must_use]
pub fn new() -> Self {
let mut m = Vec::with_capacity(32 * 64);
for k in 0..32 {
for n in 0..64 {
let arg = core::f64::consts::PI / 64.0 * (k as f64 + 0.5) * (2.0 * n as f64 - 0.5);
m.push(Complex::new(2.0 * arg.cos(), 2.0 * arg.sin()));
}
}
AnalysisQmf {
x: vec![0.0; 320],
m,
}
}
pub fn push_slot(&mut self, samples: &[f64]) -> Result<[Complex; 32]> {
if samples.len() != 32 {
return Err(Error::SbrQmfInvalid);
}
self.x.copy_within(0..288, 32);
for (n, s) in samples.iter().enumerate() {
self.x[31 - n] = *s;
}
let mut u = [0.0f64; 64];
for (n, un) in u.iter_mut().enumerate() {
let mut acc = 0.0;
for j in 0..5 {
let idx = n + j * 64;
acc += self.x[idx] * QMF_WINDOW[2 * idx];
}
*un = acc;
}
let mut w = [Complex::default(); 32];
for (k, wk) in w.iter_mut().enumerate() {
let row = &self.m[k * 64..(k + 1) * 64];
let mut acc = Complex::default();
for (n, cell) in row.iter().enumerate() {
acc += *cell * u[n];
}
*wk = acc;
}
Ok(w)
}
}
#[derive(Debug, Clone)]
pub struct EncoderAnalysisQmf {
x: Vec<f64>,
m: Vec<Complex>,
}
impl Default for EncoderAnalysisQmf {
fn default() -> Self {
Self::new()
}
}
impl EncoderAnalysisQmf {
#[must_use]
pub fn new() -> Self {
let mut m = Vec::with_capacity(64 * 128);
for k in 0..64 {
for n in 0..128 {
let arg = core::f64::consts::PI / 128.0 * (k as f64 + 0.5) * (2.0 * n as f64 + 1.0);
m.push(Complex::new(arg.cos(), arg.sin()));
}
}
EncoderAnalysisQmf {
x: vec![0.0; 640],
m,
}
}
pub fn push_slot(&mut self, samples: &[f64]) -> Result<[Complex; 64]> {
if samples.len() != 64 {
return Err(Error::SbrQmfInvalid);
}
self.x.copy_within(0..576, 64);
for (n, s) in samples.iter().enumerate() {
self.x[63 - n] = *s;
}
let mut u = [0.0f64; 128];
for (n, un) in u.iter_mut().enumerate() {
let mut acc = 0.0;
for j in 0..5 {
let idx = n + j * 128;
acc += self.x[idx] * QMF_WINDOW[idx];
}
*un = acc;
}
let mut out = [Complex::default(); 64];
for (k, xk) in out.iter_mut().enumerate() {
let row = &self.m[k * 128..(k + 1) * 128];
let mut acc = Complex::default();
for (n, cell) in row.iter().enumerate() {
acc += *cell * u[n];
}
*xk = acc;
}
Ok(out)
}
}
#[derive(Debug, Clone)]
pub struct SynthesisQmf {
v: Vec<f64>,
n_mat: Vec<Complex>,
}
impl Default for SynthesisQmf {
fn default() -> Self {
Self::new()
}
}
impl SynthesisQmf {
#[must_use]
pub fn new() -> Self {
let mut n_mat = Vec::with_capacity(128 * 64);
for n in 0..128 {
for k in 0..64 {
let arg =
core::f64::consts::PI / 128.0 * (k as f64 + 0.5) * (2.0 * n as f64 - 255.0);
n_mat.push(Complex::new(arg.cos() / 64.0, arg.sin() / 64.0));
}
}
SynthesisQmf {
v: vec![0.0; 1280],
n_mat,
}
}
pub fn push_slot(&mut self, bands: &[Complex]) -> Result<[f64; 64]> {
if bands.len() != 64 {
return Err(Error::SbrQmfInvalid);
}
self.v.copy_within(0..1152, 128);
for n in 0..128 {
let row = &self.n_mat[n * 64..(n + 1) * 64];
let mut acc = 0.0;
for (k, cell) in row.iter().enumerate() {
let x = bands[k];
acc += x.re * cell.re - x.im * cell.im;
}
self.v[n] = acc;
}
let mut out = [0.0f64; 64];
for (k, o) in out.iter_mut().enumerate() {
let mut acc = 0.0;
for n in 0..5 {
acc += self.v[256 * n + k] * QMF_WINDOW[128 * n + k];
acc += self.v[256 * n + 192 + k] * QMF_WINDOW[128 * n + 64 + k];
}
*o = acc;
}
Ok(out)
}
}
#[derive(Debug, Clone)]
pub struct DownsampledSynthesisQmf {
v: Vec<f64>,
n_mat: Vec<Complex>,
}
impl Default for DownsampledSynthesisQmf {
fn default() -> Self {
Self::new()
}
}
impl DownsampledSynthesisQmf {
#[must_use]
pub fn new() -> Self {
let mut n_mat = Vec::with_capacity(64 * 32);
for n in 0..64 {
for k in 0..32 {
let arg =
core::f64::consts::PI / 64.0 * (k as f64 + 0.5) * (2.0 * n as f64 - 127.5);
n_mat.push(Complex::new(arg.cos() / 64.0, arg.sin() / 64.0));
}
}
DownsampledSynthesisQmf {
v: vec![0.0; 640],
n_mat,
}
}
pub fn push_slot(&mut self, bands: &[Complex]) -> Result<[f64; 32]> {
if bands.len() != 32 {
return Err(Error::SbrQmfInvalid);
}
self.v.copy_within(0..576, 64);
for n in 0..64 {
let row = &self.n_mat[n * 32..(n + 1) * 32];
let mut acc = 0.0;
for (k, cell) in row.iter().enumerate() {
let x = bands[k];
acc += x.re * cell.re - x.im * cell.im;
}
self.v[n] = acc;
}
let mut out = [0.0f64; 32];
for (k, o) in out.iter_mut().enumerate() {
let mut acc = 0.0;
for n in 0..5 {
acc += self.v[128 * n + k] * QMF_WINDOW[2 * (64 * n + k)];
acc += self.v[128 * n + 96 + k] * QMF_WINDOW[2 * (64 * n + 32 + k)];
}
*o = acc;
}
Ok(out)
}
}
#[derive(Debug, Clone)]
pub struct RealAnalysisQmf {
x: Vec<f64>,
m: Vec<f64>,
}
impl Default for RealAnalysisQmf {
fn default() -> Self {
Self::new()
}
}
impl RealAnalysisQmf {
#[must_use]
pub fn new() -> Self {
let mut m = Vec::with_capacity(32 * 64);
for k in 0..32 {
for n in 0..64 {
let arg = core::f64::consts::PI / 64.0 * (k as f64 + 0.5) * (2.0 * n as f64 - 96.0);
m.push(2.0 * arg.cos());
}
}
RealAnalysisQmf {
x: vec![0.0; 320],
m,
}
}
pub fn push_slot(&mut self, samples: &[f64]) -> Result<[f64; 32]> {
if samples.len() != 32 {
return Err(Error::SbrQmfInvalid);
}
self.x.copy_within(0..288, 32);
for (n, s) in samples.iter().enumerate() {
self.x[31 - n] = *s;
}
let mut u = [0.0f64; 64];
for (n, un) in u.iter_mut().enumerate() {
let mut acc = 0.0;
for j in 0..5 {
let idx = n + j * 64;
acc += self.x[idx] * QMF_WINDOW[2 * idx];
}
*un = acc;
}
let mut w = [0.0f64; 32];
for (k, wk) in w.iter_mut().enumerate() {
let row = &self.m[k * 64..(k + 1) * 64];
let mut acc = 0.0;
for (n, cell) in row.iter().enumerate() {
acc += *cell * u[n];
}
*wk = acc;
}
Ok(w)
}
}
#[derive(Debug, Clone)]
pub struct RealSynthesisQmf {
v: Vec<f64>,
n_mat: Vec<f64>,
}
impl Default for RealSynthesisQmf {
fn default() -> Self {
Self::new()
}
}
impl RealSynthesisQmf {
#[must_use]
pub fn new() -> Self {
let mut n_mat = Vec::with_capacity(128 * 64);
for n in 0..128 {
for k in 0..64 {
let arg =
core::f64::consts::PI / 128.0 * (k as f64 + 0.5) * (2.0 * n as f64 - 64.0);
n_mat.push(arg.cos() / 32.0);
}
}
RealSynthesisQmf {
v: vec![0.0; 1280],
n_mat,
}
}
pub fn push_slot(&mut self, bands: &[f64]) -> Result<[f64; 64]> {
if bands.len() != 64 {
return Err(Error::SbrQmfInvalid);
}
self.v.copy_within(0..1152, 128);
for n in 0..128 {
let row = &self.n_mat[n * 64..(n + 1) * 64];
let mut acc = 0.0;
for (k, cell) in row.iter().enumerate() {
acc += bands[k] * *cell;
}
self.v[n] = acc;
}
let mut out = [0.0f64; 64];
for (k, o) in out.iter_mut().enumerate() {
let mut acc = 0.0;
for n in 0..5 {
acc += self.v[256 * n + k] * QMF_WINDOW[128 * n + k];
acc += self.v[256 * n + 192 + k] * QMF_WINDOW[128 * n + 64 + k];
}
*o = acc;
}
Ok(out)
}
}
#[derive(Debug, Clone)]
pub struct RealDownsampledSynthesisQmf {
v: Vec<f64>,
n_mat: Vec<f64>,
}
impl Default for RealDownsampledSynthesisQmf {
fn default() -> Self {
Self::new()
}
}
impl RealDownsampledSynthesisQmf {
#[must_use]
pub fn new() -> Self {
let mut n_mat = Vec::with_capacity(64 * 32);
for n in 0..64 {
for k in 0..32 {
let arg = core::f64::consts::PI / 64.0 * (k as f64 + 0.5) * (2.0 * n as f64 - 32.0);
n_mat.push(arg.cos() / 32.0);
}
}
RealDownsampledSynthesisQmf {
v: vec![0.0; 640],
n_mat,
}
}
pub fn push_slot(&mut self, bands: &[f64]) -> Result<[f64; 32]> {
if bands.len() != 32 {
return Err(Error::SbrQmfInvalid);
}
self.v.copy_within(0..576, 64);
for n in 0..64 {
let row = &self.n_mat[n * 32..(n + 1) * 32];
let mut acc = 0.0;
for (k, cell) in row.iter().enumerate() {
acc += bands[k] * *cell;
}
self.v[n] = acc;
}
let mut out = [0.0f64; 32];
for (k, o) in out.iter_mut().enumerate() {
let mut acc = 0.0;
for n in 0..5 {
acc += self.v[128 * n + k] * QMF_WINDOW[2 * (64 * n + k)];
acc += self.v[128 * n + 96 + k] * QMF_WINDOW[2 * (64 * n + 32 + k)];
}
*o = acc;
}
Ok(out)
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn window_spot_values() {
assert_eq!(QMF_WINDOW[0], 0.0);
assert_eq!(QMF_WINDOW[1], -0.0005525286);
assert_eq!(QMF_WINDOW[128], 0.0132718220);
assert_eq!(QMF_WINDOW[320], 0.8537385600);
assert_eq!(QMF_WINDOW[512], -0.0132718220);
assert_eq!(QMF_WINDOW[639], -0.000552528);
}
#[test]
fn window_mirror_structure() {
for i in 1..320usize {
let a = QMF_WINDOW[i].abs();
let b = QMF_WINDOW[640 - i].abs();
assert!((a - b).abs() < 1e-9, "mirror mismatch at {i}: {a} vs {b}");
}
}
#[test]
fn analysis_silence_and_shape() {
let mut a = AnalysisQmf::new();
assert!(matches!(a.push_slot(&[0.0; 16]), Err(Error::SbrQmfInvalid)));
for _ in 0..4 {
let w = a.push_slot(&[0.0; 32]).unwrap();
assert!(w.iter().all(|c| c.re == 0.0 && c.im == 0.0));
}
let mut s = SynthesisQmf::new();
assert!(matches!(
s.push_slot(&[Complex::default(); 32]),
Err(Error::SbrQmfInvalid)
));
let out = s.push_slot(&[Complex::default(); 64]).unwrap();
assert!(out.iter().all(|&x| x == 0.0));
}
#[test]
fn analysis_synthesis_upsamples_a_sine() {
let mut a = AnalysisQmf::new();
let mut s = SynthesisQmf::new();
let freq = 0.03; let slots = 96;
let mut output = Vec::new();
for slot in 0..slots {
let mut input = [0.0f64; 32];
for (n, v) in input.iter_mut().enumerate() {
let t = (slot * 32 + n) as f64;
*v = (2.0 * core::f64::consts::PI * freq * t).sin();
}
let w = a.push_slot(&input).unwrap();
let mut x = [Complex::default(); 64];
x[..32].copy_from_slice(&w);
output.extend_from_slice(&s.push_slot(&x).unwrap());
}
let ideal =
|t: f64, delay: f64| (2.0 * core::f64::consts::PI * freq * (t - delay) / 2.0).sin();
let mut best = (f64::INFINITY, 0usize);
for delay in 0..1200usize {
let mut err = 0.0;
let mut sig = 0.0;
for (t, &out) in output.iter().enumerate().skip(1400) {
let e = out - ideal(t as f64, delay as f64);
err += e * e;
sig += out * out;
}
let ratio = err / sig.max(1e-30);
if ratio < best.0 {
best = (ratio, delay);
}
}
assert!(
best.0 < 1e-4,
"reconstruction error ratio {} at delay {}",
best.0,
best.1
);
}
#[test]
fn analysis_downsampled_synthesis_is_identity() {
let mut a = AnalysisQmf::new();
let mut s = DownsampledSynthesisQmf::new();
let freq = 0.04;
let slots = 96;
let mut input_all = Vec::new();
let mut output = Vec::new();
for slot in 0..slots {
let mut input = [0.0f64; 32];
for (n, v) in input.iter_mut().enumerate() {
let t = (slot * 32 + n) as f64;
*v = (2.0 * core::f64::consts::PI * freq * t).sin()
+ 0.5 * (2.0 * core::f64::consts::PI * 2.3 * freq * t).cos();
}
input_all.extend_from_slice(&input);
let w = a.push_slot(&input).unwrap();
output.extend_from_slice(&s.push_slot(&w).unwrap());
}
let mut best = (f64::INFINITY, 0usize);
for delay in 0..640usize {
let mut err = 0.0;
let mut sig = 0.0;
for t in 800..output.len() {
if t < delay {
continue;
}
let e = output[t] - input_all[t - delay];
err += e * e;
sig += output[t] * output[t];
}
let ratio = err / sig.max(1e-30);
if ratio < best.0 {
best = (ratio, delay);
}
}
assert!(
best.0 < 1e-4,
"identity error ratio {} at delay {}",
best.0,
best.1
);
}
#[test]
fn analysis_is_linear() {
let mut qa = AnalysisQmf::new();
let mut qb = AnalysisQmf::new();
let mut qs = AnalysisQmf::new();
for slot in 0..8 {
let mut a = [0.0f64; 32];
let mut b = [0.0f64; 32];
let mut sum = [0.0f64; 32];
for n in 0..32 {
let t = (slot * 32 + n) as f64;
a[n] = (0.11 * t).sin();
b[n] = (0.031 * t + 1.0).cos();
sum[n] = a[n] + b[n];
}
let wa = qa.push_slot(&a).unwrap();
let wb = qb.push_slot(&b).unwrap();
let ws = qs.push_slot(&sum).unwrap();
for k in 0..32 {
let d = ws[k] - (wa[k] + wb[k]);
assert!(d.norm_sqr() < 1e-18);
}
}
}
#[test]
fn real_analysis_downsampled_synthesis_is_identity() {
let mut a = RealAnalysisQmf::new();
let mut s = RealDownsampledSynthesisQmf::new();
let freq = 0.037;
let slots = 96;
let mut input_all = Vec::new();
let mut output = Vec::new();
for slot in 0..slots {
let mut input = [0.0f64; 32];
for (n, v) in input.iter_mut().enumerate() {
let t = (slot * 32 + n) as f64;
*v = (2.0 * core::f64::consts::PI * freq * t).sin()
+ 0.5 * (2.0 * core::f64::consts::PI * 2.9 * freq * t).cos();
}
input_all.extend_from_slice(&input);
let w = a.push_slot(&input).unwrap();
output.extend_from_slice(&s.push_slot(&w).unwrap());
}
let mut best = (f64::INFINITY, 0usize);
for delay in 0..640usize {
let mut err = 0.0;
let mut sig = 0.0;
for (t, &o) in output.iter().enumerate().skip(900) {
if t < delay {
continue;
}
let e = o - input_all[t - delay];
err += e * e;
sig += o * o;
}
let ratio = err / sig.max(1e-30);
if ratio < best.0 {
best = (ratio, delay);
}
}
assert!(
best.0 < 1e-4,
"identity error ratio {} at delay {}",
best.0,
best.1
);
}
#[test]
fn real_analysis_synthesis_upsamples_a_sine() {
let mut a = RealAnalysisQmf::new();
let mut s = RealSynthesisQmf::new();
let freq = 0.043;
let slots = 96;
let mut output = Vec::new();
for slot in 0..slots {
let mut input = [0.0f64; 32];
for (n, v) in input.iter_mut().enumerate() {
let t = (slot * 32 + n) as f64;
*v = (2.0 * core::f64::consts::PI * freq * t).sin();
}
let w = a.push_slot(&input).unwrap();
let mut x = [0.0f64; 64];
x[..32].copy_from_slice(&w);
output.extend_from_slice(&s.push_slot(&x).unwrap());
}
let ideal =
|t: f64, delay: f64| (2.0 * core::f64::consts::PI * freq * (t - delay) / 2.0).sin();
let mut best = (f64::INFINITY, 0usize);
for delay in 0..1200usize {
let mut err = 0.0;
let mut sig = 0.0;
for (t, &out) in output.iter().enumerate().skip(1600) {
let e = out - ideal(t as f64, delay as f64);
err += e * e;
sig += out * out;
}
let ratio = err / sig.max(1e-30);
if ratio < best.0 {
best = (ratio, delay);
}
}
assert!(
best.0 < 1e-4,
"reconstruction error ratio {} at delay {}",
best.0,
best.1
);
}
#[test]
fn real_banks_silence_shape_linearity() {
let mut a = RealAnalysisQmf::new();
assert!(matches!(a.push_slot(&[0.0; 16]), Err(Error::SbrQmfInvalid)));
for _ in 0..4 {
let w = a.push_slot(&[0.0; 32]).unwrap();
assert!(w.iter().all(|&c| c == 0.0));
}
let mut s = RealSynthesisQmf::new();
assert!(matches!(s.push_slot(&[0.0; 32]), Err(Error::SbrQmfInvalid)));
assert!(s.push_slot(&[0.0; 64]).unwrap().iter().all(|&x| x == 0.0));
let mut d = RealDownsampledSynthesisQmf::new();
assert!(matches!(d.push_slot(&[0.0; 64]), Err(Error::SbrQmfInvalid)));
assert!(d.push_slot(&[0.0; 32]).unwrap().iter().all(|&x| x == 0.0));
let mut qa = RealAnalysisQmf::new();
let mut qb = RealAnalysisQmf::new();
let mut qs = RealAnalysisQmf::new();
for slot in 0..8 {
let mut va = [0.0f64; 32];
let mut vb = [0.0f64; 32];
let mut sum = [0.0f64; 32];
for n in 0..32 {
let t = (slot * 32 + n) as f64;
va[n] = (0.13 * t).sin();
vb[n] = (0.029 * t + 0.4).cos();
sum[n] = va[n] + vb[n];
}
let wa = qa.push_slot(&va).unwrap();
let wb = qb.push_slot(&vb).unwrap();
let ws = qs.push_slot(&sum).unwrap();
for k in 0..32 {
assert!((ws[k] - (wa[k] + wb[k])).abs() < 1e-9);
}
}
}
}
#[cfg(test)]
mod encoder_analysis_tests {
use super::*;
#[test]
fn encoder_bank_matches_decoder_bank_scale_and_band() {
let mut enc = EncoderAnalysisQmf::new();
let mut dec = AnalysisQmf::new();
let f_full = 3.5 / 128.0; let n_slots = 40;
let mut e_mag = 0.0f64;
let mut d_mag = 0.0f64;
let mut e_other = 0.0f64;
for l in 0..n_slots {
let full: Vec<f64> = (0..64)
.map(|i| {
let t = (l * 64 + i) as f64;
1000.0 * (2.0 * core::f64::consts::PI * f_full * t).sin()
})
.collect();
let half: Vec<f64> = full.iter().step_by(2).copied().collect();
let x = enc.push_slot(&full).unwrap();
let w = dec.push_slot(&half).unwrap();
if l >= 20 {
e_mag += x[3].norm_sqr();
d_mag += w[3].norm_sqr();
for (k, c) in x.iter().enumerate() {
if k != 3 && k != 2 && k != 4 {
e_other += c.norm_sqr();
}
}
}
}
assert!(e_mag > 0.0 && d_mag > 0.0);
let ratio = e_mag / d_mag;
assert!(
(ratio - 1.0).abs() < 0.02,
"energy scale mismatch between the banks: {ratio}"
);
assert!(e_other < e_mag * 1e-3, "leakage {e_other} vs {e_mag}");
}
#[test]
fn encoder_bank_resolves_high_half_bands() {
let mut enc = EncoderAnalysisQmf::new();
let f_full = 45.5 / 128.0;
let mut peak_band = 0usize;
let mut peak = 0.0;
for l in 0..30 {
let full: Vec<f64> = (0..64)
.map(|i| {
let t = (l * 64 + i) as f64;
(2.0 * core::f64::consts::PI * f_full * t).cos()
})
.collect();
let x = enc.push_slot(&full).unwrap();
if l == 29 {
for (k, c) in x.iter().enumerate() {
if c.norm_sqr() > peak {
peak = c.norm_sqr();
peak_band = k;
}
}
}
}
assert_eq!(peak_band, 45);
assert!(matches!(
enc.push_slot(&[0.0; 32]),
Err(Error::SbrQmfInvalid)
));
}
}