psemek/libs/audio/source/effect/echo.cpp
2023-10-04 14:34:17 +03:00

67 lines
1.3 KiB
C++

#include <psemek/audio/effect/echo.hpp>
#include <vector>
namespace psemek::audio
{
namespace
{
struct echo_impl
: stream
{
echo_impl(stream_ptr stream, duration delay, float gain)
: stream_(std::move(stream))
, buffer_(2 * std::max(1l, delay.samples()), 0.f)
, gain_(gain)
{}
std::optional<std::size_t> length() const override
{
return std::nullopt;
}
std::size_t read(util::span<float> samples) override
{
auto count = stream_->read(samples);
for (std::size_t i = 0; i < count; ++i)
{
float output = buffer_[buffer_pos_];
buffer_[buffer_pos_] = samples[i] + buffer_[buffer_pos_] * gain_;
samples[i] = output;
++buffer_pos_;
buffer_pos_ %= buffer_.size();
}
for (std::size_t i = count; i < samples.size(); ++i)
{
samples[i] = buffer_[buffer_pos_];
buffer_[buffer_pos_] = buffer_[buffer_pos_] * gain_;
++buffer_pos_;
buffer_pos_ %= buffer_.size();
}
return samples.size();
}
std::size_t played() const override
{
return stream_->played();
}
private:
stream_ptr stream_;
std::vector<float> buffer_;
std::size_t buffer_pos_ = 0;
float gain_;
};
}
stream_ptr echo(stream_ptr stream, duration delay, float gain)
{
return std::make_shared<echo_impl>(std::move(stream), delay, gain);
}
}