alex/nih-plug
1
0
Fork 0
Code Activity

Use unchecked access instead of zip

Browse source
This commit is contained in:
Robbert van der Helm 2022-07-25 16:06:55 +02:00
parent dd49bd9e7f
commit 6da8756e31
1 changed files with 25 additions and 28 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
plugins/spectral_compressor/src/compressor_bank.rs
View file

@ -590,6 +590,11 @@ impl CompressorBank {
/// Actually do the thing. [`Self::update_envelopes()`] must have been called before calling /// Actually do the thing. [`Self::update_envelopes()`] must have been called before calling
/// this. /// this.
///
/// # Panics
///
/// Panics if the buffer does not have the same length as the one that was passed to the last
/// `resize()` call.
fn compress( fn compress(
&self, &self,
buffer: &mut [Complex32], buffer: &mut [Complex32],
@ -607,40 +612,30 @@ impl CompressorBank {
let upwards_knee_scaling_factor = let upwards_knee_scaling_factor =
compute_knee_scaling_factor(params.compressors.upwards.knee_width_db.value).sqrt(); compute_knee_scaling_factor(params.compressors.upwards.knee_width_db.value).sqrt();
// Is this what they mean by zip and and ship it? assert!(self.downwards_thresholds.len() == buffer.len());
let downwards_knees = self assert!(self.downwards_ratio_recips.len() == buffer.len());
.downwards_knee_starts assert!(self.downwards_knee_starts.len() == buffer.len());
.iter() assert!(self.downwards_knee_ends.len() == buffer.len());
.zip(self.downwards_knee_ends.iter()); assert!(self.upwards_thresholds.len() == buffer.len());
let downwards_values = self assert!(self.upwards_ratio_recips.len() == buffer.len());
.downwards_thresholds assert!(self.upwards_knee_starts.len() == buffer.len());
.iter() assert!(self.upwards_knee_ends.len() == buffer.len());
.zip(self.downwards_ratio_recips.iter()) for (bin_idx, (bin, envelope)) in buffer
.zip(downwards_knees);
let upwards_knees = self
.upwards_knee_starts
.iter()
.zip(self.upwards_knee_ends.iter());
let upwards_values = self
.upwards_thresholds
.iter()
.zip(self.upwards_ratio_recips.iter())
.zip(upwards_knees);
for (((bin, envelope), downwards_values), upwards_values) in buffer
.iter_mut() .iter_mut()
.zip(self.envelopes[channel_idx].iter()) .zip(self.envelopes[channel_idx].iter())
.zip(downwards_values) .enumerate()
.zip(upwards_values)
.skip(skip_bins_below) .skip(skip_bins_below)
{ {
// This works by computing a scaling factor, and then scaling the bin magnitudes by that. // This works by computing a scaling factor, and then scaling the bin magnitudes by that.
let mut scale = 1.0; let mut scale = 1.0;
// All compression happens in the linear domain to save a logarithm // All compression happens in the linear domain to save a logarithm
let ( // SAFETY: These sizes were asserted above
(downwards_threshold, downwards_ratio_recip), let downwards_threshold = unsafe { self.downwards_thresholds.get_unchecked(bin_idx) };
(downwards_knee_start, downwards_knee_end), let downwards_ratio_recip =
) = downwards_values; unsafe { self.downwards_ratio_recips.get_unchecked(bin_idx) };
let downwards_knee_start = unsafe { self.downwards_knee_starts.get_unchecked(bin_idx) };
let downwards_knee_end = unsafe { self.downwards_knee_ends.get_unchecked(bin_idx) };
if *downwards_ratio_recip != 1.0 { if *downwards_ratio_recip != 1.0 {
scale *= compress_downwards( scale *= compress_downwards(
*envelope, *envelope,
@ -654,8 +649,10 @@ impl CompressorBank {
// Upwards compression should not happen when the signal is _too_ quiet as we'd only be // Upwards compression should not happen when the signal is _too_ quiet as we'd only be
// amplifying noise // amplifying noise
let ((upwards_threshold, upwards_ratio_recip), (upwards_knee_start, upwards_knee_end)) = let upwards_threshold = unsafe { self.upwards_thresholds.get_unchecked(bin_idx) };
upwards_values; let upwards_ratio_recip = unsafe { self.upwards_ratio_recips.get_unchecked(bin_idx) };
let upwards_knee_start = unsafe { self.upwards_knee_starts.get_unchecked(bin_idx) };
let upwards_knee_end = unsafe { self.upwards_knee_ends.get_unchecked(bin_idx) };
if *upwards_ratio_recip != 1.0 && *envelope > 1e-6 { if *upwards_ratio_recip != 1.0 && *envelope > 1e-6 {
scale *= compress_upwards( scale *= compress_upwards(
*envelope, *envelope,