use std::time::Duration;
use super::SeekError;
use crate::common::{ChannelCount, SampleRate};
use crate::{Float, Source};
pub(crate) fn distortion<I>(input: I, gain: Float, threshold: Float) -> Distortion<I>
where
I: Source,
{
Distortion {
input,
gain,
threshold,
}
}
#[derive(Clone, Debug)]
pub struct Distortion<I> {
input: I,
gain: Float,
threshold: Float,
}
impl<I> Distortion<I> {
#[inline]
pub fn set_gain(&mut self, gain: Float) {
self.gain = gain;
}
#[inline]
pub fn set_threshold(&mut self, threshold: Float) {
self.threshold = threshold;
}
#[inline]
pub fn inner(&self) -> &I {
&self.input
}
#[inline]
pub fn inner_mut(&mut self) -> &mut I {
&mut self.input
}
#[inline]
pub fn into_inner(self) -> I {
self.input
}
}
impl<I> Iterator for Distortion<I>
where
I: Source,
{
type Item = I::Item;
#[inline]
fn next(&mut self) -> Option<Self::Item> {
self.input.next().map(|value| {
let v = value * self.gain;
let t = self.threshold;
v.clamp(-t, t)
})
}
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
self.input.size_hint()
}
}
impl<I> ExactSizeIterator for Distortion<I> where I: Source + ExactSizeIterator {}
impl<I> Source for Distortion<I>
where
I: Source,
{
#[inline]
fn current_span_len(&self) -> Option<usize> {
self.input.current_span_len()
}
#[inline]
fn channels(&self) -> ChannelCount {
self.input.channels()
}
#[inline]
fn sample_rate(&self) -> SampleRate {
self.input.sample_rate()
}
#[inline]
fn total_duration(&self) -> Option<Duration> {
self.input.total_duration()
}
#[inline]
fn try_seek(&mut self, pos: Duration) -> Result<(), SeekError> {
self.input.try_seek(pos)
}
}