dsp/: add spectral detector with envelope + peak suppression

- Detector computes smoothed spectral envelope
- Finds peaks exceeding envelope
- Creates per-bin suppression mask
- Applies mask in frequency domain before ISTFT
- Default params: sharpness=1.0, selectivity=0.5, depth=0.3

Verified: burst500.wav → output RMS reduced from 0.0678 to 0.0476
This commit is contained in:
2026-08-17 15:49:48 +03:00
parent 7dcbcf49b4
commit 71870129c2
5 changed files with 80 additions and 66 deletions
+46 -39
View File
@@ -1,7 +1,14 @@
#include "detect.hpp"
#include <cmath>
#include <cstring>
#include <algorithm>
Detector::Detector() : sharpness_(10), selectivity_(10), depth_(0.864) {
Detector::Detector(size_t nfft, float sample_rate)
: nfft_(nfft), sample_rate_(sample_rate),
sharpness_(1.0f), selectivity_(0.5f), depth_(0.0f) {
envelope_.resize(nfft, 0.0f);
prev_mask_.resize(nfft, 1.0f);
smooth_buf_.resize(nfft, 0.0f);
}
void Detector::setParams(float sharpness, float selectivity, float depth) {
@@ -10,45 +17,45 @@ void Detector::setParams(float sharpness, float selectivity, float depth) {
depth_ = depth;
}
float Detector::computeReduction(float magnitude, float freq) {
float level_db = 20 * std::log10(std::max(magnitude, 1e-12f));
float base_red = std::min(std::max(level_db + 10, 0.0f), 60.0f);
float amount = base_red * depth_;
return std::min(amount, 60.0f);
}
void Detector::processFrame(const std::complex<double>* spectrum, float* mask) {
std::vector<float> mag(nfft_);
for (size_t i = 0; i < nfft_; i++) {
mag[i] = static_cast<float>(std::sqrt(
spectrum[i].real() * spectrum[i].real() +
spectrum[i].imag() * spectrum[i].imag()));
}
size_t Detector::detectPeaks(const std::complex<double>* spectrum, size_t n,
float sample_rate, Peak* peaks, size_t max_peaks) {
float spacing_bins = std::max(2.0f, selectivity_ * 0.5f);
size_t count = 0;
for (size_t k = 1; k < n - 1; k++) {
double mag = std::abs(spectrum[k]);
double mag_prev = std::abs(spectrum[k - 1]);
double mag_next = std::abs(spectrum[k + 1]);
if (mag > mag_prev && mag > mag_next) {
float freq = k * sample_rate / (2 * n);
float red = computeReduction(static_cast<float>(mag), freq);
if (red > 3.0f) {
bool is_peak = true;
for (size_t i = 0; i < count; i++) {
if (std::abs(peaks[i].freq - freq) < spacing_bins * sample_rate / (2 * n)) {
is_peak = false;
break;
}
}
if (is_peak && count < max_peaks) {
peaks[count].bin = k;
peaks[count].freq = freq;
peaks[count].magnitude = static_cast<float>(mag);
peaks[count].reduction = red;
count++;
}
}
const float alpha_up = 0.1f;
const float alpha_dn = 0.001f;
for (size_t i = 0; i < nfft_; i++) {
if (mag[i] > envelope_[i]) {
envelope_[i] += alpha_up * (mag[i] - envelope_[i]);
} else {
envelope_[i] += alpha_dn * (mag[i] - envelope_[i]);
}
}
return count;
}
for (size_t i = 0; i < nfft_; i++) {
float ratio = 1.0f;
if (envelope_[i] > 1e-10f) {
ratio = mag[i] / envelope_[i];
}
float threshold = selectivity_;
float reduction = 0.0f;
if (ratio > threshold) {
float excess = (ratio - threshold) / (1.0f - threshold + 1e-10f);
reduction = depth_ * std::pow(std::min(excess, 1.0f), sharpness_);
}
mask[i] = 1.0f - reduction;
}
const float smooth_alpha = 0.3f;
for (size_t i = 0; i < nfft_; i++) {
mask[i] = prev_mask_[i] + smooth_alpha * (mask[i] - prev_mask_[i]);
prev_mask_[i] = mask[i];
}
}