Turn on polygon smoothing, in an attempt to get even smoother in the edges.
[pitch] / pitchdetector.h
1 #ifndef _PITCHDETECTOR_H
2 #define _PITCHDETECTOR_H 1
3
4 #include <fftw3.h>
5
6 #include <complex>
7 #include <vector>
8 #include <utility>
9
10 class PitchDetector {
11 public:
12         PitchDetector(unsigned sample_rate, unsigned fft_length, unsigned pad_factor, unsigned overlap);
13         ~PitchDetector();
14         std::pair<double, double> detect_pitch(short *buf);
15
16 private:
17         // parameters
18         unsigned sample_rate;
19         unsigned fft_length;
20         unsigned pad_factor;
21         unsigned overlap;
22
23         // buffers
24         double *in, *in_windowed;
25         std::complex<double> *out;
26         double *bins;
27         fftw_plan plan;
28         double *window_data;
29
30         // utility functions
31         void apply_window(double *in, double *out, unsigned num_samples);
32         void find_peak_magnitudes(std::complex<double> *in, double *out, unsigned num_samples);
33         std::pair<double, double> find_peak(double *in, unsigned num_samples);
34         std::pair<double, double> adjust_for_overtones(std::pair<double, double> base, double *in, unsigned num_samples);
35         double bin_to_freq(double bin, unsigned num_samples);
36         double freq_to_bin(double freq, unsigned num_samples);
37         std::pair<double, double> interpolate_peak(double ym1, double y0, double y1);
38 };
39
40 #endif /* !defined(_PITCHDETECTOR_H) */