source: src/pitchdetection.c @ d1ec8cb

feature/autosinkfeature/cnnfeature/cnn_orgfeature/constantqfeature/crepefeature/crepe_orgfeature/pitchshiftfeature/pydocstringsfeature/timestretchfix/ffmpeg5pitchshiftsamplertimestretchyinfft+
Last change on this file since d1ec8cb was 6d4ec49, checked in by Paul Brossier <piem@piem.org>, 16 years ago

pitchdetection.{c,h}: remove tabs

  • Property mode set to 100644
File size: 7.9 KB
Line 
1/*
2   Copyright (C) 2003 Paul Brossier
3
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.
8
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.
13
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.
17   */
18
19#include "aubio_priv.h"
20#include "sample.h"
21#include "phasevoc.h"
22#include "mathutils.h"
23#include "filter.h"
24#include "pitchmcomb.h"
25#include "pitchyin.h"
26#include "pitchfcomb.h"
27#include "pitchschmitt.h"
28#include "pitchyinfft.h"
29#include "pitchdetection.h"
30
31typedef smpl_t (*aubio_pitchdetection_func_t)
32  (aubio_pitchdetection_t *p, fvec_t * ibuf);
33typedef smpl_t (*aubio_pitchdetection_conv_t)
34  (smpl_t value, uint_t srate, uint_t bufsize);
35
36void aubio_pitchdetection_slideblock(aubio_pitchdetection_t *p, fvec_t *ibuf);
37
38smpl_t aubio_pitchdetection_mcomb   (aubio_pitchdetection_t *p, fvec_t *ibuf);
39smpl_t aubio_pitchdetection_yin     (aubio_pitchdetection_t *p, fvec_t *ibuf);
40smpl_t aubio_pitchdetection_schmitt (aubio_pitchdetection_t *p, fvec_t *ibuf);
41smpl_t aubio_pitchdetection_fcomb   (aubio_pitchdetection_t *p, fvec_t *ibuf);
42smpl_t aubio_pitchdetection_yinfft  (aubio_pitchdetection_t *p, fvec_t *ibuf);
43
44/** generic pitch detection structure */
45struct _aubio_pitchdetection_t {
46  aubio_pitchdetection_type type; /**< pitch detection mode */
47  aubio_pitchdetection_mode mode; /**< pitch detection output mode */
48  uint_t srate;                   /**< samplerate */
49  uint_t bufsize;                 /**< buffer size */
50  aubio_pitchmcomb_t * mcomb;     /**< mcomb object */
51  aubio_pitchfcomb_t * fcomb;     /**< fcomb object */
52  aubio_pitchschmitt_t * schmitt; /**< schmitt object */
53  aubio_pitchyinfft_t * yinfft;   /**< yinfft object */
54  aubio_filter_t * filter;        /**< filter */
55  aubio_pvoc_t * pv;              /**< phase vocoder for mcomb */ 
56  cvec_t * fftgrain;              /**< spectral frame for mcomb */
57  fvec_t * buf;                   /**< temporary buffer for yin */
58  fvec_t * yin;                   /**< yin function */
59  smpl_t yinthres;                /**< yin peak picking threshold parameter */
60  aubio_pitchdetection_func_t callback; /**< pointer to current pitch detection method */
61  aubio_pitchdetection_conv_t freqconv; /**< pointer to current pitch conversion method */ 
62};
63
64/* convenience wrapper function for frequency unit conversions
65 * should probably be rewritten with #defines */
66smpl_t freqconvbin(smpl_t f,uint_t srate,uint_t bufsize);
67smpl_t freqconvbin(smpl_t f,uint_t srate,uint_t bufsize){
68  return aubio_freqtobin(f,srate,bufsize);
69}
70
71smpl_t freqconvmidi(smpl_t f,uint_t srate,uint_t bufsize);
72smpl_t freqconvmidi(smpl_t f,uint_t srate UNUSED,uint_t bufsize UNUSED){
73  return aubio_freqtomidi(f);
74}
75
76smpl_t freqconvpass(smpl_t f,uint_t srate,uint_t bufsize);
77smpl_t freqconvpass(smpl_t f,uint_t srate UNUSED,uint_t bufsize UNUSED){
78  return f;
79}
80
81aubio_pitchdetection_t * new_aubio_pitchdetection(uint_t bufsize, 
82    uint_t hopsize, 
83    uint_t channels,
84    uint_t samplerate,
85    aubio_pitchdetection_type type,
86    aubio_pitchdetection_mode mode)
87{
88  aubio_pitchdetection_t *p = AUBIO_NEW(aubio_pitchdetection_t);
89  p->srate = samplerate;
90  p->type = type;
91  p->mode = mode;
92  p->bufsize = bufsize;
93  switch(p->type) {
94    case aubio_pitch_yin:
95      p->buf      = new_fvec(bufsize,channels);
96      p->yin      = new_fvec(bufsize/2,channels);
97      p->callback = aubio_pitchdetection_yin;
98      p->yinthres = 0.15;
99      break;
100    case aubio_pitch_mcomb:
101      p->pv       = new_aubio_pvoc(bufsize, hopsize, channels);
102      p->fftgrain = new_cvec(bufsize, channels);
103      p->mcomb    = new_aubio_pitchmcomb(bufsize,hopsize,channels,samplerate);
104      p->filter   = new_aubio_cdsgn_filter(samplerate);
105      p->callback = aubio_pitchdetection_mcomb;
106      break;
107    case aubio_pitch_fcomb:
108      p->buf      = new_fvec(bufsize,channels);
109      p->fcomb    = new_aubio_pitchfcomb(bufsize,hopsize,samplerate);
110      p->callback = aubio_pitchdetection_fcomb;
111      break;
112    case aubio_pitch_schmitt:
113      p->buf      = new_fvec(bufsize,channels);
114      p->schmitt  = new_aubio_pitchschmitt(bufsize,samplerate);
115      p->callback = aubio_pitchdetection_schmitt;
116      break;
117    case aubio_pitch_yinfft:
118      p->buf      = new_fvec(bufsize,channels);
119      p->yinfft   = new_aubio_pitchyinfft(bufsize);
120      p->callback = aubio_pitchdetection_yinfft;
121      p->yinthres = 0.85;
122      break;
123    default:
124      break;
125  }
126  switch(p->mode) {
127    case aubio_pitchm_freq:
128      p->freqconv = freqconvpass;
129      break;
130    case aubio_pitchm_midi:
131      p->freqconv = freqconvmidi;
132      break;
133    case aubio_pitchm_cent:
134      /* bug: not implemented */
135      p->freqconv = freqconvmidi;
136      break;
137    case aubio_pitchm_bin:
138      p->freqconv = freqconvbin;
139      break;
140    default:
141      break;
142  }
143  return p;
144}
145
146void del_aubio_pitchdetection(aubio_pitchdetection_t * p) {
147  switch(p->type) {
148    case aubio_pitch_yin:
149      del_fvec(p->yin);
150      del_fvec(p->buf);
151      break;
152    case aubio_pitch_mcomb:
153      del_aubio_pvoc(p->pv);
154      del_cvec(p->fftgrain);
155      del_aubio_pitchmcomb(p->mcomb);
156      break;
157    case aubio_pitch_schmitt:
158      del_fvec(p->buf);
159      del_aubio_pitchschmitt(p->schmitt);
160      break;
161    case aubio_pitch_fcomb:
162      del_fvec(p->buf);
163      del_aubio_pitchfcomb(p->fcomb);
164      break;
165    case aubio_pitch_yinfft:
166      del_fvec(p->buf);
167      del_aubio_pitchyinfft(p->yinfft);
168      break;
169    default:
170      break;
171  }
172  AUBIO_FREE(p);
173}
174
175void aubio_pitchdetection_slideblock(aubio_pitchdetection_t *p, fvec_t *ibuf){
176  uint_t i,j = 0, overlap_size = 0;
177  overlap_size = p->buf->length-ibuf->length;
178  for (i=0;i<p->buf->channels;i++){
179    for (j=0;j<overlap_size;j++){
180      p->buf->data[i][j] = p->buf->data[i][j+ibuf->length];
181    }
182  }
183  for (i=0;i<ibuf->channels;i++){
184    for (j=0;j<ibuf->length;j++){
185      p->buf->data[i][j+overlap_size] = ibuf->data[i][j];
186    }
187  }
188}
189
190void aubio_pitchdetection_set_yinthresh(aubio_pitchdetection_t *p, smpl_t thres) {
191  p->yinthres = thres;
192}
193
194smpl_t aubio_pitchdetection(aubio_pitchdetection_t *p, fvec_t * ibuf) {
195  return p->freqconv(p->callback(p,ibuf),p->srate,p->bufsize);
196}
197
198smpl_t aubio_pitchdetection_mcomb(aubio_pitchdetection_t *p, fvec_t *ibuf) {
199  smpl_t pitch = 0.;
200  aubio_filter_do(p->filter,ibuf);
201  aubio_pvoc_do(p->pv,ibuf,p->fftgrain);
202  pitch = aubio_pitchmcomb_detect(p->mcomb,p->fftgrain);
203  /** \bug should move the >0 check within aubio_bintofreq */
204  if (pitch>0.) {
205    pitch = aubio_bintofreq(pitch,p->srate,p->bufsize);
206  } else {
207    pitch = 0.;
208  }
209  return pitch;
210}
211
212smpl_t aubio_pitchdetection_yin(aubio_pitchdetection_t *p, fvec_t *ibuf) {
213  smpl_t pitch = 0.;
214  aubio_pitchdetection_slideblock(p,ibuf);
215  pitch = aubio_pitchyin_getpitchfast(p->buf,p->yin, p->yinthres);
216  if (pitch>0) {
217    pitch = p->srate/(pitch+0.);
218  } else {
219    pitch = 0.;
220  }
221  return pitch;
222}
223
224
225smpl_t aubio_pitchdetection_yinfft(aubio_pitchdetection_t *p, fvec_t *ibuf){
226  smpl_t pitch = 0.;
227  aubio_pitchdetection_slideblock(p,ibuf);
228  pitch = aubio_pitchyinfft_detect(p->yinfft,p->buf,p->yinthres);
229  if (pitch>0) {
230    pitch = p->srate/(pitch+0.);
231  } else {
232    pitch = 0.;
233  }
234  return pitch; 
235}
236
237smpl_t aubio_pitchdetection_fcomb(aubio_pitchdetection_t *p, fvec_t *ibuf){
238  aubio_pitchdetection_slideblock(p,ibuf);
239  return aubio_pitchfcomb_detect(p->fcomb,p->buf);
240}
241
242smpl_t aubio_pitchdetection_schmitt(aubio_pitchdetection_t *p, fvec_t *ibuf){
243  aubio_pitchdetection_slideblock(p,ibuf);
244  return aubio_pitchschmitt_detect(p->schmitt,p->buf);
245}
Note: See TracBrowser for help on using the repository browser.