X-Git-Url: https://git.nihav.org/?p=nihav.git;a=blobdiff_plain;f=nihav-mpeg%2Fsrc%2Fcodecs%2Faac%2Fsbr%2Fdsp.rs;fp=nihav-mpeg%2Fsrc%2Fcodecs%2Faac%2Fsbr%2Fdsp.rs;h=a9762b02e4f6a9b077ad7bb40eb8040ee451c750;hp=0000000000000000000000000000000000000000;hb=0e18270ddb438c765a4597404f9053a15ce84704;hpb=9c3cd867ce7645f6f828675bd5d383033d6263f7 diff --git a/nihav-mpeg/src/codecs/aac/sbr/dsp.rs b/nihav-mpeg/src/codecs/aac/sbr/dsp.rs new file mode 100644 index 0000000..a9762b0 --- /dev/null +++ b/nihav-mpeg/src/codecs/aac/sbr/dsp.rs @@ -0,0 +1,328 @@ +pub use nihav_codec_support::dsp::fft::*; +pub use nihav_codec_support::dsp::mdct::*; + +pub struct SBRDSP { + imdct: IMDCT, + tmp0: [f32; 64], + tmp1: [f32; 64], + tmp2: [f32; 64], + tmp3: [f32; 64], +} + +impl SBRDSP { + pub fn new() -> Self { + Self { + imdct: IMDCT::new(128, false), + tmp0: [0.0; 64], + tmp1: [0.0; 64], + tmp2: [0.0; 64], + tmp3: [0.0; 64], + } + } +} + +#[derive(Clone)] +pub struct SBRAnalysis { + hist: [f32; 320], + pos: usize, +} + +impl SBRAnalysis { + pub fn new() -> Self { + Self { + hist: [0.0; 320], + pos: 0, + } + } + pub fn analysis(&mut self, dsp: &mut SBRDSP, samples: &[f32], dst: &mut [FFTComplex; 64]) { + self.pos += self.hist.len() - 32; + if self.pos >= self.hist.len() { + self.pos -= self.hist.len(); + } + for (dst, &src) in self.hist[self.pos..][..32].iter_mut().zip(samples.iter().rev()) { + *dst = src; + } + + let sum = &mut dsp.tmp0; + *sum = [0.0; 64]; + for (n, dst) in sum.iter_mut().enumerate() { + *dst = 0.0; + for j in 0..5 { + *dst += self.hist[(self.pos + n + j * 64) % self.hist.len()] * SBR_WINDOW[(n + j * 64) * 2]; + } + } + + let src = &mut dsp.tmp1; + + src[0] = sum[0]; + src[1] = sum[1]; + for k in (1..31).step_by(2) { + src[k * 2] = -sum[64 - k]; + src[k * 2 + 1] = sum[k + 1]; + src[k * 2 + 2] = -sum[63 - k]; + src[k * 2 + 3] = sum[k + 2]; + } + src[62] = -sum[33]; + src[63] = sum[32]; + + let out = &mut dsp.tmp2; + dsp.imdct.imdct_half(src, out); + + *dst = [FFTC_ZERO; 64]; + for (k, dst) in dst[..32].iter_mut().enumerate() { + dst.re = -out[63 - k]; + dst.im = out[k]; + } + } +} + +#[derive(Clone)] +pub struct SBRSynthesis { + hist: [f32; 1280], + pos: usize, +} + +impl SBRSynthesis { + pub fn new() -> Self { + Self { + hist: [0.0; 1280], + pos: 0, + } + } + pub fn synthesis(&mut self, dsp: &mut SBRDSP, src: &[FFTComplex; 64], dst: &mut [f32]) { + self.pos += self.hist.len() - 128; + if self.pos >= self.hist.len() { + self.pos -= self.hist.len(); + } + let re = &mut dsp.tmp0; + let im = &mut dsp.tmp1; + for ((re, im), src) in re.iter_mut().zip(im.iter_mut()).zip(src.iter()) { + *re = src.re / 32.0; + *im = src.im / 32.0; + } + for pair in im.chunks_mut(2) { + pair[1] = -pair[1]; + } + let out0 = &mut dsp.tmp2; + let out1 = &mut dsp.tmp3; + dsp.imdct.imdct_half(re, out0); + dsp.imdct.imdct_half(im, out1); + + for (k, (&a, &b)) in out1.iter().zip(out0.iter().rev()).enumerate() { + self.hist[self.pos..][k] = a - b; + self.hist[self.pos..][127 - k] = a + b; + } + + for (k, dst) in dst[..64].iter_mut().enumerate() { + *dst = 0.0; + for n in 0..5 { + *dst += self.hist[(self.pos + 256 * n + k) % self.hist.len()] * SBR_WINDOW[128 * n + k]; + *dst += self.hist[(self.pos + 256 * n + k + 192) % self.hist.len()] * SBR_WINDOW[128 * n + k + 64]; + } + } + } +} + +/*#[derive(Clone)] +pub struct SBRSynthesisDownsampled { + hist: [f32; 640], + pos: usize, +} + +impl SBRSynthesisDownsampled { + pub fn new() -> Self { + Self { + hist: [0.0; 640], + pos: 0, + } + } + pub fn synthesis(&mut self, dsp: &mut SBRDSP, src: &[FFTComplex; 64], dst: &mut [f32]) { + self.pos += self.hist.len() - 64; + if self.pos >= self.hist.len() { + self.pos -= self.hist.len(); + } + + let (p0, p1) = dsp.tmp0.split_at_mut(32); + let out = &mut dsp.tmp1; + for (n, (dst0, dst1)) in p0.iter_mut().zip(p1.iter_mut()).enumerate() { + *dst0 = -src[n].re / 32.0; + *dst1 = src[31 - n].im / 32.0; + } + dsp.imdct.imdct_half(&dsp.tmp0, out); + for (k, pair) in out.chunks(2).rev().enumerate() { + self.hist[self.pos..][k] = pair[1]; + self.hist[self.pos..][63 - k] = -pair[0]; + } + + for (k, dst) in dst[..32].iter_mut().enumerate() { + *dst = 0.0; + for n in 0..5 { + *dst += self.hist[(self.pos + 128 * n + k) % self.hist.len()] * SBR_WINDOW[128 * n + k]; + *dst += self.hist[(self.pos + 128 * n + k + 96) % self.hist.len()] * SBR_WINDOW[128 * n + k + 64]; + } + } + } +}*/ + +const SBR_WINDOW: [f32; 640] = [ + 0.0, -0.00055252865047, -0.00056176925738, -0.00049475180896, + -0.00048752279712, -0.00048937912498, -0.00050407143497, -0.00052265642972, + -0.00054665656337, -0.00056778025613, -0.00058709304852, -0.00061327473938, + -0.00063124935319, -0.00065403333621, -0.00067776907764, -0.00069416146273, + -0.00071577364744, -0.00072550431222, -0.00074409418541, -0.00074905980532, + -0.0007681371927, -0.00077248485949, -0.00078343322877, -0.00077798694927, + -0.000780366471, -0.00078014496257, -0.0007757977331, -0.00076307935757, + -0.00075300014201, -0.00073193571525, -0.00072153919876, -0.00069179375372, + -0.00066504150893, -0.00063415949025, -0.0005946118933, -0.00055645763906, + -0.00051455722108, -0.00046063254803, -0.00040951214522, -0.00035011758756, + -0.00028969811748, -0.0002098337344, -0.00014463809349, -0.00006173344072, + 0.00001349497418, 0.00010943831274, 0.00020430170688, 0.00029495311041, + 0.0004026540216, 0.00051073884952, 0.00062393761391, 0.00074580258865, + 0.00086084433262, 0.00098859883015, 0.00112501551307, 0.00125778846475, + 0.00139024948272, 0.00154432198471, 0.00168680832531, 0.00183482654224, + 0.00198411407369, 0.00214615835557, 0.00230172547746, 0.00246256169126, + 0.00262017586902, 0.00278704643465, 0.00294694477165, 0.00311254206525, + 0.00327396134847, 0.00344188741828, 0.00360082681231, 0.00376039229104, + 0.00392074323703, 0.00408197531935, 0.0042264269227, 0.00437307196781, + 0.00452098527825, 0.00466064606118, 0.00479325608498, 0.00491376035745, + 0.00503930226013, 0.00514073539032, 0.00524611661324, 0.00534716811982, + 0.00541967759307, 0.00548760401507, 0.00554757145088, 0.00559380230045, + 0.00562206432097, 0.00564551969164, 0.00563891995151, 0.00562661141932, + 0.0055917128663, 0.005540436394, 0.0054753783077, 0.0053838975897, + 0.00527157587272, 0.00513822754514, 0.00498396877629, 0.004810946906, + 0.00460395301471, 0.00438018617447, 0.0041251642327, 0.00384564081246, + 0.00354012465507, 0.00320918858098, 0.00284467578623, 0.00245085400321, + 0.0020274176185, 0.00157846825768, 0.00109023290512, 0.0005832264248, + 0.00002760451905, -0.00054642808664, -0.00115681355227, -0.00180394725893, + -0.00248267236449, -0.003193377839, -0.00394011240522, -0.004722259624, + -0.00553372111088, -0.00637922932685, -0.00726158168517, -0.00817982333726, + -0.00913253296085, -0.01011502154986, -0.01113155480321, -0.01218499959508, + 0.01327182200351, 0.01439046660792, 0.01554055533423, 0.01673247129989, + 0.01794333813443, 0.01918724313698, 0.02045317933555, 0.02174675502535, + 0.02306801692862, 0.02441609920285, 0.02578758475467, 0.02718594296329, + 0.02860721736385, 0.03005026574279, 0.03150176087389, 0.03297540810337, + 0.03446209487686, 0.03596975605542, 0.03748128504252, 0.03900536794745, + 0.04053491705584, 0.04206490946367, 0.04360975421304, 0.04514884056413, + 0.04668430272642, 0.04821657200672, 0.04973857556014, 0.05125561555216, + 0.05276307465207, 0.05424527683589, 0.05571736482138, 0.05716164501299, + 0.0585915683626, 0.05998374801761, 0.06134551717207, 0.06268578081172, + 0.06397158980681, 0.0652247106438, 0.06643675122104, 0.06760759851228, + 0.06870438283512, 0.06976302447127, 0.07076287107266, 0.07170026731102, + 0.07256825833083, 0.07336202550803, 0.07410036424342, 0.07474525581194, + 0.07531373362019, 0.07580083586584, 0.07619924793396, 0.07649921704119, + 0.07670934904245, 0.07681739756964, 0.07682300113923, 0.07672049241746, + 0.07650507183194, 0.07617483218536, 0.07573057565061, 0.0751576255287, + 0.07446643947564, 0.0736406005762, 0.07267746427299, 0.07158263647903, + 0.07035330735093, 0.06896640131951, 0.06745250215166, 0.06576906686508, + 0.06394448059633, 0.06196027790387, 0.0598166570809, 0.05751526919867, + 0.05504600343009, 0.05240938217366, 0.04959786763445, 0.04663033051701, + 0.04347687821958, 0.04014582784127, 0.03664181168133, 0.03295839306691, + 0.02908240060125, 0.02503075618909, 0.02079970728622, 0.01637012582228, + 0.01176238327857, 0.00696368621617, 0.00197656014503, -0.00320868968304, + -0.00857117491366, -0.01412888273558, -0.01988341292573, -0.02582272888064, + -0.03195312745332, -0.03827765720822, -0.04478068215856, -0.05148041767934, + -0.05837053268336, -0.06544098531359, -0.07269433008129, -0.08013729344279, + -0.08775475365593, -0.09555333528914, -0.10353295311463, -0.1116826931773, + -0.120007798468, -0.12850028503878, -0.13715517611934, -0.1459766491187, + -0.15496070710605, -0.16409588556669, -0.17338081721706, -0.18281725485142, + -0.19239667457267, -0.20212501768103, -0.21197358538056, -0.22196526964149, + -0.23206908706791, -0.24230168845974, -0.25264803095722, -0.26310532994603, + -0.27366340405625, -0.28432141891085, -0.29507167170646, -0.30590985751916, + -0.31682789136456, -0.32781137272105, -0.33887226938665, -0.3499914122931, + 0.36115899031355, 0.37237955463061, 0.38363500139043, 0.39492117615675, + 0.40623176767625, 0.41756968968409, 0.42891199207373, 0.44025537543665, + 0.45159965356824, 0.46293080852757, 0.47424532146115, 0.48552530911099, + 0.49677082545707, 0.50798175000434, 0.51912349702391, 0.53022408956855, + 0.54125534487322, 0.55220512585061, 0.5630789140137, 0.57385241316923, + 0.58454032354679, 0.59511230862496, 0.6055783538918, 0.61591099320291, + 0.62612426956055, 0.63619801077286, 0.64612696959461, 0.65590163024671, + 0.66551398801627, 0.67496631901712, 0.68423532934598, 0.69332823767032, + 0.70223887193539, 0.71094104263095, 0.71944626349561, 0.72774489002994, + 0.73582117582769, 0.74368278636488, 0.75131374561237, 0.75870807608242, + 0.76586748650939, 0.77277808813327, 0.77942875190216, 0.7858353120392, + 0.79197358416424, 0.797846641377, 0.80344857518505, 0.80876950044491, + 0.81381912706217, 0.81857760046468, 0.82304198905409, 0.8272275347336, + 0.8311038457152, 0.83469373618402, 0.83797173378865, 0.84095413924722, + 0.84362382812005, 0.84598184698206, 0.84803157770763, 0.84978051984268, + 0.85119715249343, 0.85230470352147, 0.85310209497017, 0.85357205739107, + 0.85373856005937, 0.85357205739107, 0.85310209497017, 0.85230470352147, + 0.85119715249343, 0.84978051984268, 0.84803157770763, 0.84598184698206, + 0.84362382812005, 0.84095413924722, 0.83797173378865, 0.83469373618402, + 0.8311038457152, 0.8272275347336, 0.82304198905409, 0.81857760046468, + 0.81381912706217, 0.80876950044491, 0.80344857518505, 0.797846641377, + 0.79197358416424, 0.7858353120392, 0.77942875190216, 0.77277808813327, + 0.76586748650939, 0.75870807608242, 0.75131374561237, 0.74368278636488, + 0.73582117582769, 0.72774489002994, 0.71944626349561, 0.71094104263095, + 0.70223887193539, 0.69332823767032, 0.68423532934598, 0.67496631901712, + 0.66551398801627, 0.65590163024671, 0.64612696959461, 0.63619801077286, + 0.62612426956055, 0.61591099320291, 0.6055783538918, 0.59511230862496, + 0.58454032354679, 0.57385241316923, 0.5630789140137, 0.55220512585061, + 0.54125534487322, 0.53022408956855, 0.51912349702391, 0.50798175000434, + 0.49677082545707, 0.48552530911099, 0.47424532146115, 0.46293080852757, + 0.45159965356824, 0.44025537543665, 0.42891199207373, 0.41756968968409, + 0.40623176767625, 0.39492117615675, 0.38363500139043, 0.37237955463061, + -0.36115899031355, -0.3499914122931, -0.33887226938665, -0.32781137272105, + -0.31682789136456, -0.30590985751916, -0.29507167170646, -0.28432141891085, + -0.27366340405625, -0.26310532994603, -0.25264803095722, -0.24230168845974, + -0.23206908706791, -0.22196526964149, -0.21197358538056, -0.20212501768103, + -0.19239667457267, -0.18281725485142, -0.17338081721706, -0.16409588556669, + -0.15496070710605, -0.1459766491187, -0.13715517611934, -0.12850028503878, + -0.120007798468, -0.1116826931773, -0.10353295311463, -0.09555333528914, + -0.08775475365593, -0.08013729344279, -0.07269433008129, -0.06544098531359, + -0.05837053268336, -0.05148041767934, -0.04478068215856, -0.03827765720822, + -0.03195312745332, -0.02582272888064, -0.01988341292573, -0.01412888273558, + -0.00857117491366, -0.00320868968304, 0.00197656014503, 0.00696368621617, + 0.01176238327857, 0.01637012582228, 0.02079970728622, 0.02503075618909, + 0.02908240060125, 0.03295839306691, 0.03664181168133, 0.04014582784127, + 0.04347687821958, 0.04663033051701, 0.04959786763445, 0.05240938217366, + 0.05504600343009, 0.05751526919867, 0.0598166570809, 0.06196027790387, + 0.06394448059633, 0.06576906686508, 0.06745250215166, 0.06896640131951, + 0.07035330735093, 0.07158263647903, 0.07267746427299, 0.0736406005762, + 0.07446643947564, 0.0751576255287, 0.07573057565061, 0.07617483218536, + 0.07650507183194, 0.07672049241746, 0.07682300113923, 0.07681739756964, + 0.07670934904245, 0.07649921704119, 0.07619924793396, 0.07580083586584, + 0.07531373362019, 0.07474525581194, 0.07410036424342, 0.07336202550803, + 0.07256825833083, 0.07170026731102, 0.07076287107266, 0.06976302447127, + 0.06870438283512, 0.06760759851228, 0.06643675122104, 0.0652247106438, + 0.06397158980681, 0.06268578081172, 0.06134551717207, 0.05998374801761, + 0.0585915683626, 0.05716164501299, 0.05571736482138, 0.05424527683589, + 0.05276307465207, 0.05125561555216, 0.04973857556014, 0.04821657200672, + 0.04668430272642, 0.04514884056413, 0.04360975421304, 0.04206490946367, + 0.04053491705584, 0.03900536794745, 0.03748128504252, 0.03596975605542, + 0.03446209487686, 0.03297540810337, 0.03150176087389, 0.03005026574279, + 0.02860721736385, 0.02718594296329, 0.02578758475467, 0.02441609920285, + 0.02306801692862, 0.02174675502535, 0.02045317933555, 0.01918724313698, + 0.01794333813443, 0.01673247129989, 0.01554055533423, 0.01439046660792, + -0.01327182200351, -0.01218499959508, -0.01113155480321, -0.01011502154986, + -0.00913253296085, -0.00817982333726, -0.00726158168517, -0.00637922932685, + -0.00553372111088, -0.004722259624, -0.00394011240522, -0.003193377839, + -0.00248267236449, -0.00180394725893, -0.00115681355227, -0.00054642808664, + 0.00002760451905, 0.0005832264248, 0.00109023290512, 0.00157846825768, + 0.0020274176185, 0.00245085400321, 0.00284467578623, 0.00320918858098, + 0.00354012465507, 0.00384564081246, 0.0041251642327, 0.00438018617447, + 0.00460395301471, 0.004810946906, 0.00498396877629, 0.00513822754514, + 0.00527157587272, 0.0053838975897, 0.0054753783077, 0.005540436394, + 0.0055917128663, 0.00562661141932, 0.00563891995151, 0.00564551969164, + 0.00562206432097, 0.00559380230045, 0.00554757145088, 0.00548760401507, + 0.00541967759307, 0.00534716811982, 0.00524611661324, 0.00514073539032, + 0.00503930226013, 0.00491376035745, 0.00479325608498, 0.00466064606118, + 0.00452098527825, 0.00437307196781, 0.0042264269227, 0.00408197531935, + 0.00392074323703, 0.00376039229104, 0.00360082681231, 0.00344188741828, + 0.00327396134847, 0.00311254206525, 0.00294694477165, 0.00278704643465, + 0.00262017586902, 0.00246256169126, 0.00230172547746, 0.00214615835557, + 0.00198411407369, 0.00183482654224, 0.00168680832531, 0.00154432198471, + 0.00139024948272, 0.00125778846475, 0.00112501551307, 0.00098859883015, + 0.00086084433262, 0.00074580258865, 0.00062393761391, 0.00051073884952, + 0.0004026540216, 0.00029495311041, 0.00020430170688, 0.00010943831274, + 0.00001349497418, -0.00006173344072, -0.00014463809349, -0.0002098337344, + -0.00028969811748, -0.00035011758756, -0.00040951214522, -0.00046063254803, + -0.00051455722108, -0.00055645763906, -0.0005946118933, -0.00063415949025, + -0.00066504150893, -0.00069179375372, -0.00072153919876, -0.00073193571525, + -0.00075300014201, -0.00076307935757, -0.0007757977331, -0.00078014496257, + -0.000780366471, -0.00077798694927, -0.00078343322877, -0.00077248485949, + -0.0007681371927, -0.00074905980532, -0.00074409418541, -0.00072550431222, + -0.00071577364744, -0.00069416146273, -0.00067776907764, -0.00065403333621, + -0.00063124935319, -0.00061327473938, -0.00058709304852, -0.00056778025613, + -0.00054665656337, -0.00052265642972, -0.00050407143497, -0.00048937912498, + -0.00048752279712, -0.00049475180896, -0.00056176925738, -0.00055252865047 +];