pitch_track.hpp

Go to the documentation of this file.

#pragma once

 
#include "audio_metrics.hpp"  // kSilenceFloorDb, FrequencyEstimate
#include "audio_spectrum.hpp" // Window, kDefaultKaiserBeta, fit_tone
 
#include <pulp/audio/buffer.hpp>
 
#include <span>
#include <vector>
 
namespace pulp::test::audio {

double cents_between(double hz, double reference_hz);

struct PitchEstimate {
    double hz = 0.0;
    double confidence = 0.0;
    bool voiced = false;
    double coarse_hz = 0.0;
    double harmonic_energy_ratio = 0.0;
};

struct PitchOptions {
    int fft_length = 16384;
    double min_hz = 20.0;
    double max_hz = 0.0;
    Window window = Window::blackman;
    double kaiser_beta = kDefaultKaiserBeta;
    double min_confidence = 0.5;
    int harmonic_count = 0;
    double silence_rms = 1.0e-4;
    int channel = 0;
    int analysis_offset = 0;
    int analysis_length = 0;
};

PitchEstimate estimate_pitch(const pulp::audio::BufferView<const float>& signal,
                             double sample_rate,
                             const PitchOptions& options = {});

PitchEstimate estimate_pitch(std::span<const float> samples, double sample_rate,
                             const PitchOptions& options = {});

struct PitchTrackPoint {
    double time_s = 0.0;      
    double hz = 0.0;          
    double confidence = 0.0;  
    bool voiced = false;      
};

struct PitchTrackOptions {
    int window_length = 8192;
    int hop_length = 2048;
    PitchOptions pitch{};
    int channel = 0;
};

struct PitchTrack {
    double sample_rate = 0.0;
    int window_length = 0;
    int hop_length = 0;
    std::vector<PitchTrackPoint> points;

    std::vector<double> voiced_times_s() const;
    std::vector<double> voiced_hz() const;
};

PitchTrack track_pitch(std::span<const float> samples, double sample_rate,
                       const PitchTrackOptions& options = {});
 
} // namespace pulp::test::audio