2 Copyright (C) 2004, 2005 Mario Lang <mlang@delysid.org>
4 This program is free software; you can redistribute it and/or modify
5 it under the terms of the GNU General Public License as published by
6 the Free Software Foundation; either version 2 of the License, or
7 (at your option) any later version.
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
14 You should have received a copy of the GNU General Public License
15 along with this program; if not, write to the Free Software
16 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20 #include "aubio_priv.h"
23 #include "mathutils.h"
24 #include "spectral/fft.h"
25 #include "pitch/pitchfcomb.h"
34 struct _aubio_pitchfcomb_t {
41 fvec_t * fftLastPhase;
45 aubio_pitchfcomb_t * new_aubio_pitchfcomb (uint_t bufsize, uint_t hopsize, uint_t channels)
47 aubio_pitchfcomb_t * p = AUBIO_NEW(aubio_pitchfcomb_t);
49 p->stepSize = hopsize;
50 p->winput = new_fvec(bufsize,1);
51 p->fftOut = new_cvec(bufsize,1);
52 p->fftLastPhase = new_fvec(bufsize, channels);
53 p->fft = new_aubio_fft(bufsize, 1);
54 p->win = new_aubio_window("hanning", bufsize);
58 /* input must be stepsize long */
59 void aubio_pitchfcomb_do (aubio_pitchfcomb_t * p, fvec_t * input, fvec_t * output)
61 uint_t i, k, l, maxharm = 0;
62 smpl_t phaseDifference = TWO_PI*(smpl_t)p->stepSize/(smpl_t)p->fftSize;
63 aubio_fpeak_t peaks[MAX_PEAKS];
65 for (i = 0; i < input->channels; i++) {
67 for (k=0; k<MAX_PEAKS; k++) {
72 for (k=0; k < input->length; k++){
73 p->winput->data[0][k] = p->win->data[0][k] * input->data[i][k];
75 aubio_fft_do(p->fft,p->winput,p->fftOut);
77 for (k=0; k<=p->fftSize/2; k++) {
79 magnitude = 20.*LOG10(2.*p->fftOut->norm[0][k]/(smpl_t)p->fftSize),
80 phase = p->fftOut->phas[0][k],
83 /* compute phase difference */
84 tmp = phase - p->fftLastPhase->data[i][k];
85 p->fftLastPhase->data[i][k] = phase;
87 /* subtract expected phase difference */
88 tmp -= (smpl_t)k*phaseDifference;
90 /* map delta phase into +/- Pi interval */
91 tmp = aubio_unwrap2pi(tmp);
93 /* get deviation from bin frequency from the +/- Pi interval */
94 tmp = p->fftSize/(smpl_t)p->stepSize*tmp/(TWO_PI);
96 /* compute the k-th partials' true bin */
97 bin = (smpl_t)k + tmp;
99 if (bin > 0.0 && magnitude > peaks[0].db) { // && magnitude < 0) {
100 memmove(peaks+1, peaks, sizeof(aubio_fpeak_t)*(MAX_PEAKS-1));
102 peaks[0].db = magnitude;
107 for (l=1; l<MAX_PEAKS && peaks[l].bin > 0.0; l++) {
109 for (harmonic=5; harmonic>1; harmonic--) {
110 if (peaks[0].bin / peaks[l].bin < harmonic+.02 &&
111 peaks[0].bin / peaks[l].bin > harmonic-.02) {
112 if (harmonic > (sint_t)maxharm &&
113 peaks[0].db < peaks[l].db/2) {
120 output->data[i][0] = peaks[k].bin;
121 /* quick hack to clean output a bit */
122 if (peaks[k].bin > 5000.) output->data[i][0] = 0.;
126 void del_aubio_pitchfcomb (aubio_pitchfcomb_t * p)
129 del_fvec(p->fftLastPhase);
132 del_aubio_fft(p->fft);