frequency_bands.cpp.hpp

Go to the documentation of this file.

#include "fl/audio/detector/frequency_bands.h"
#include "fl/audio/audio_context.h"
#include "fl/audio/fft/fft.h"
#include "fl/math/math.h"
#include "fl/stl/noexcept.h"
 
namespace fl {
namespace audio {
namespace detector {
 
static int sFrequencyBandsFFTCount = 0;
int FrequencyBands::getPrivateFFTCount() { return sFrequencyBandsFFTCount; }
void FrequencyBands::resetPrivateFFTCount() { sFrequencyBandsFFTCount = 0; }
 
namespace {
// Number of fft::FFT bins — higher count gives better frequency resolution
// and cleaner band separation. With 64 bins, the CQ kernel provides
// good frequency isolation for all three bands (bass, mid, treble).
const int kNumBands = 64;
 
// fft::FFT frequency range covering bass through treble.
// Range [100, 10000] covers the bass (20-250), mid (250-4000), and treble
// (4000-20000) ranges with good CQ kernel resolution. The ratio of 100x
// keeps the highest-frequency kernel window large enough for the 512-sample
// fft::FFT (N_window = 512 / (10000/100) = 5 samples minimum).
const float kFFTMinFreq = 100.0f;
const float kFFTMaxFreq = 10000.0f;
 
} // namespace
 
FrequencyBands::FrequencyBands()
    : mBass(0.0f)
    , mMid(0.0f)
    , mTreble(0.0f)
    , mBassMin(20.0f)
    , mBassMax(250.0f)
    , mMidMin(250.0f)
    , mMidMax(4000.0f)
    , mTrebleMin(4000.0f)
    , mTrebleMax(20000.0f)
{}
 
FrequencyBands::~FrequencyBands() FL_NOEXCEPT = default;
 
void FrequencyBands::update(shared_ptr<Context> context) {
    // Use sample rate from context if available
    mSampleRate = context->getSampleRate();
 
    // Use shared master fft::FFT via context (downsampled to our config)
    mRetainedFFT = context->getFFT(kNumBands, kFFTMinFreq, kFFTMaxFreq);
    const fft::Bins& fftBins = *mRetainedFFT;
    sFrequencyBandsFFTCount++;  // Diagnostic counter (no private fft::FFT anymore)
 
    span<const i16> pcm = context->getPCM();
 
    // Calculate energy for each band using fractional bin overlap
    float bassEnergy = calculateBandEnergy(fftBins, mBassMin, mBassMax, kFFTMinFreq, kFFTMaxFreq);
    float midEnergy = calculateBandEnergy(fftBins, mMidMin, mMidMax, kFFTMinFreq, kFFTMaxFreq);
    float trebleEnergy = calculateBandEnergy(fftBins, mTrebleMin, mTrebleMax, kFFTMinFreq, kFFTMaxFreq);
 
    // Compute dt from actual audio buffer duration: pcmSize / sampleRate
    const float dt = computeAudioDt(pcm.size(), mSampleRate);
    mBass = mBassSmoother.update(bassEnergy, dt);
    mMid = mMidSmoother.update(midEnergy, dt);
    mTreble = mTrebleSmoother.update(trebleEnergy, dt);
 
    // Per-band normalization — mirrors EnergyAnalyzer pattern
    auto normalizeBand = [](float val, AttackDecayFilter<float>& filter,
                            float frameDt) -> float {
        float runningMax = filter.update(val, frameDt);
        if (runningMax < 0.001f) runningMax = 0.001f;
        return fl::min(1.0f, val / runningMax);
    };
    mBassNorm = normalizeBand(mBass, mBassMaxFilter, dt);
    mMidNorm = normalizeBand(mMid, mMidMaxFilter, dt);
    mTrebleNorm = normalizeBand(mTreble, mTrebleMaxFilter, dt);
}
 
void FrequencyBands::fireCallbacks() {
    if (onLevelsUpdate) {
        onLevelsUpdate(mBass, mMid, mTreble);
    }
    if (onBassLevel) {
        onBassLevel(mBass);
    }
    if (onMidLevel) {
        onMidLevel(mMid);
    }
    if (onTrebleLevel) {
        onTrebleLevel(mTreble);
    }
}
 
void FrequencyBands::reset() {
    mBass = 0.0f;
    mMid = 0.0f;
    mTreble = 0.0f;
    mBassSmoother.reset();
    mMidSmoother.reset();
    mTrebleSmoother.reset();
    mBassMaxFilter.reset(0.0f);
    mMidMaxFilter.reset(0.0f);
    mTrebleMaxFilter.reset(0.0f);
    mBassNorm = 0.0f;
    mMidNorm = 0.0f;
    mTrebleNorm = 0.0f;
}
 
float FrequencyBands::calculateBandEnergy(const fft::Bins& fft, float minFreq, float maxFreq,
                                           float fftMinFreq, float fftMaxFreq) {
    const int numBins = static_cast<int>(fft.raw().size());
    if (numBins <= 1) {
        return 0.0f;
    }
 
    float totalEnergy = 0.0f;
    float totalWeight = 0.0f;
 
    for (int i = 0; i < numBins; i++) {
        // Compute the frequency range for this CQ bin using log-spaced
        // boundaries that match the CQ kernel's actual frequency layout.
        float binLow;
        float binHigh;
        if (i == 0) {
            binLow = fftMinFreq;
        } else {
            binLow = fft.binBoundary(i - 1);
        }
        if (i == numBins - 1) {
            binHigh = fftMaxFreq;
        } else {
            binHigh = fft.binBoundary(i);
        }
 
        // Calculate fractional overlap between this bin and the target band
        float overlapMin = fl::max(binLow, minFreq);
        float overlapMax = fl::min(binHigh, maxFreq);
 
        if (overlapMax <= overlapMin) {
            continue; // No overlap
        }
 
        float binWidth = binHigh - binLow;
        float overlapFraction = (overlapMax - overlapMin) / binWidth;
 
        totalEnergy += fft.raw()[i] * overlapFraction;
        totalWeight += overlapFraction;
    }
 
    // Normalize by total fractional weight to make bands comparable
    return (totalWeight > 0.0f) ? totalEnergy / totalWeight : 0.0f;
}
 
} // namespace detector
} // namespace audio
} // namespace fl