[96fb8ad] | 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. |
---|
[6d4ec49] | 17 | */ |
---|
[96fb8ad] | 18 | |
---|
| 19 | #include "aubio_priv.h" |
---|
| 20 | #include "sample.h" |
---|
| 21 | #include "phasevoc.h" |
---|
| 22 | #include "mathutils.h" |
---|
[f8a38c5] | 23 | #include "filter.h" |
---|
[96fb8ad] | 24 | #include "pitchmcomb.h" |
---|
[c078336] | 25 | #include "pitchyin.h" |
---|
| 26 | #include "pitchfcomb.h" |
---|
| 27 | #include "pitchschmitt.h" |
---|
[650e39b] | 28 | #include "pitchyinfft.h" |
---|
[96fb8ad] | 29 | #include "pitchdetection.h" |
---|
| 30 | |
---|
[6d4ec49] | 31 | typedef smpl_t (*aubio_pitchdetection_func_t) |
---|
| 32 | (aubio_pitchdetection_t *p, fvec_t * ibuf); |
---|
| 33 | typedef smpl_t (*aubio_pitchdetection_conv_t) |
---|
| 34 | (smpl_t value, uint_t srate, uint_t bufsize); |
---|
| 35 | |
---|
[651b97e] | 36 | void aubio_pitchdetection_slideblock(aubio_pitchdetection_t *p, fvec_t *ibuf); |
---|
[c078336] | 37 | |
---|
[6d4ec49] | 38 | smpl_t aubio_pitchdetection_mcomb (aubio_pitchdetection_t *p, fvec_t *ibuf); |
---|
| 39 | smpl_t aubio_pitchdetection_yin (aubio_pitchdetection_t *p, fvec_t *ibuf); |
---|
| 40 | smpl_t aubio_pitchdetection_schmitt (aubio_pitchdetection_t *p, fvec_t *ibuf); |
---|
| 41 | smpl_t aubio_pitchdetection_fcomb (aubio_pitchdetection_t *p, fvec_t *ibuf); |
---|
| 42 | smpl_t aubio_pitchdetection_yinfft (aubio_pitchdetection_t *p, fvec_t *ibuf); |
---|
[f44b111] | 43 | |
---|
[475da2f] | 44 | /** generic pitch detection structure */ |
---|
[96fb8ad] | 45 | struct _aubio_pitchdetection_t { |
---|
[475da2f] | 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 */ |
---|
[96fb8ad] | 62 | }; |
---|
| 63 | |
---|
[3ec9d9c] | 64 | /* convenience wrapper function for frequency unit conversions |
---|
| 65 | * should probably be rewritten with #defines */ |
---|
| 66 | smpl_t freqconvbin(smpl_t f,uint_t srate,uint_t bufsize); |
---|
| 67 | smpl_t freqconvbin(smpl_t f,uint_t srate,uint_t bufsize){ |
---|
[6d4ec49] | 68 | return aubio_freqtobin(f,srate,bufsize); |
---|
[3ec9d9c] | 69 | } |
---|
| 70 | |
---|
| 71 | smpl_t freqconvmidi(smpl_t f,uint_t srate,uint_t bufsize); |
---|
[b377d04] | 72 | smpl_t freqconvmidi(smpl_t f,uint_t srate UNUSED,uint_t bufsize UNUSED){ |
---|
[6d4ec49] | 73 | return aubio_freqtomidi(f); |
---|
[3ec9d9c] | 74 | } |
---|
| 75 | |
---|
| 76 | smpl_t freqconvpass(smpl_t f,uint_t srate,uint_t bufsize); |
---|
[b377d04] | 77 | smpl_t freqconvpass(smpl_t f,uint_t srate UNUSED,uint_t bufsize UNUSED){ |
---|
[6d4ec49] | 78 | return f; |
---|
[3ec9d9c] | 79 | } |
---|
| 80 | |
---|
[96fb8ad] | 81 | aubio_pitchdetection_t * new_aubio_pitchdetection(uint_t bufsize, |
---|
[6d4ec49] | 82 | uint_t hopsize, |
---|
| 83 | uint_t channels, |
---|
| 84 | uint_t samplerate, |
---|
| 85 | aubio_pitchdetection_type type, |
---|
| 86 | aubio_pitchdetection_mode mode) |
---|
[96fb8ad] | 87 | { |
---|
[6d4ec49] | 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; |
---|
[96fb8ad] | 144 | } |
---|
| 145 | |
---|
| 146 | void del_aubio_pitchdetection(aubio_pitchdetection_t * p) { |
---|
[6d4ec49] | 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); |
---|
[96fb8ad] | 173 | } |
---|
| 174 | |
---|
[651b97e] | 175 | void aubio_pitchdetection_slideblock(aubio_pitchdetection_t *p, fvec_t *ibuf){ |
---|
[6d4ec49] | 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 | } |
---|
[651b97e] | 188 | } |
---|
| 189 | |
---|
[f8a38c5] | 190 | void aubio_pitchdetection_set_yinthresh(aubio_pitchdetection_t *p, smpl_t thres) { |
---|
[6d4ec49] | 191 | p->yinthres = thres; |
---|
[f8a38c5] | 192 | } |
---|
| 193 | |
---|
[96fb8ad] | 194 | smpl_t aubio_pitchdetection(aubio_pitchdetection_t *p, fvec_t * ibuf) { |
---|
[6d4ec49] | 195 | return p->freqconv(p->callback(p,ibuf),p->srate,p->bufsize); |
---|
[c078336] | 196 | } |
---|
| 197 | |
---|
| 198 | smpl_t aubio_pitchdetection_mcomb(aubio_pitchdetection_t *p, fvec_t *ibuf) { |
---|
[6d4ec49] | 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; |
---|
[c078336] | 210 | } |
---|
| 211 | |
---|
| 212 | smpl_t aubio_pitchdetection_yin(aubio_pitchdetection_t *p, fvec_t *ibuf) { |
---|
[6d4ec49] | 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; |
---|
[c078336] | 222 | } |
---|
| 223 | |
---|
| 224 | |
---|
[650e39b] | 225 | smpl_t aubio_pitchdetection_yinfft(aubio_pitchdetection_t *p, fvec_t *ibuf){ |
---|
[6d4ec49] | 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; |
---|
[650e39b] | 235 | } |
---|
| 236 | |
---|
[c078336] | 237 | smpl_t aubio_pitchdetection_fcomb(aubio_pitchdetection_t *p, fvec_t *ibuf){ |
---|
[6d4ec49] | 238 | aubio_pitchdetection_slideblock(p,ibuf); |
---|
| 239 | return aubio_pitchfcomb_detect(p->fcomb,p->buf); |
---|
[c078336] | 240 | } |
---|
| 241 | |
---|
| 242 | smpl_t aubio_pitchdetection_schmitt(aubio_pitchdetection_t *p, fvec_t *ibuf){ |
---|
[6d4ec49] | 243 | aubio_pitchdetection_slideblock(p,ibuf); |
---|
| 244 | return aubio_pitchschmitt_detect(p->schmitt,p->buf); |
---|
[96fb8ad] | 245 | } |
---|