1 | /* |
---|
2 | Copyright (C) 2006-2009 Paul Brossier <piem@aubio.org> |
---|
3 | |
---|
4 | This file is part of aubio. |
---|
5 | |
---|
6 | aubio is free software: you can redistribute it and/or modify |
---|
7 | it under the terms of the GNU General Public License as published by |
---|
8 | the Free Software Foundation, either version 3 of the License, or |
---|
9 | (at your option) any later version. |
---|
10 | |
---|
11 | aubio is distributed in the hope that it will be useful, |
---|
12 | but WITHOUT ANY WARRANTY; without even the implied warranty of |
---|
13 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
---|
14 | GNU General Public License for more details. |
---|
15 | |
---|
16 | You should have received a copy of the GNU General Public License |
---|
17 | along with aubio. If not, see <http://www.gnu.org/licenses/>. |
---|
18 | |
---|
19 | */ |
---|
20 | |
---|
21 | #include "aubio_priv.h" |
---|
22 | #include "fvec.h" |
---|
23 | #include "cvec.h" |
---|
24 | #include "spectral/specdesc.h" |
---|
25 | #include "tempo/beattracking.h" |
---|
26 | #include "spectral/phasevoc.h" |
---|
27 | #include "onset/peakpicker.h" |
---|
28 | #include "mathutils.h" |
---|
29 | #include "tempo/tempo.h" |
---|
30 | |
---|
31 | // TODO implement get/set_delay |
---|
32 | |
---|
33 | /** set current delay |
---|
34 | |
---|
35 | \param o beat tracking object |
---|
36 | |
---|
37 | \return current delay, in samples |
---|
38 | |
---|
39 | */ |
---|
40 | uint_t aubio_tempo_get_delay(aubio_tempo_t * o); |
---|
41 | |
---|
42 | /** set current delay |
---|
43 | |
---|
44 | \param o beat tracking object |
---|
45 | \param delay delay to set tempo to, in samples |
---|
46 | |
---|
47 | \return `0` if successful, non-zero otherwise |
---|
48 | |
---|
49 | */ |
---|
50 | uint_t aubio_tempo_set_delay(aubio_tempo_t * o, uint_t delay); |
---|
51 | |
---|
52 | /* structure to store object state */ |
---|
53 | struct _aubio_tempo_t { |
---|
54 | aubio_specdesc_t * od; /** onset detection */ |
---|
55 | aubio_pvoc_t * pv; /** phase vocoder */ |
---|
56 | aubio_peakpicker_t * pp; /** peak picker */ |
---|
57 | aubio_beattracking_t * bt; /** beat tracking */ |
---|
58 | cvec_t * fftgrain; /** spectral frame */ |
---|
59 | fvec_t * of; /** onset detection function value */ |
---|
60 | fvec_t * dfframe; /** peak picked detection function buffer */ |
---|
61 | fvec_t * out; /** beat tactus candidates */ |
---|
62 | fvec_t * onset; /** onset results */ |
---|
63 | smpl_t silence; /** silence parameter */ |
---|
64 | smpl_t threshold; /** peak picking threshold */ |
---|
65 | sint_t blockpos; /** current position in dfframe */ |
---|
66 | uint_t winlen; /** dfframe bufsize */ |
---|
67 | uint_t step; /** dfframe hopsize */ |
---|
68 | uint_t samplerate; /** sampling rate of the signal */ |
---|
69 | uint_t hop_size; /** get hop_size */ |
---|
70 | uint_t total_frames; /** total frames since beginning */ |
---|
71 | uint_t last_beat; /** time of latest detected beat, in samples */ |
---|
72 | uint_t delay; /** delay to remove to last beat, in samples */ |
---|
73 | }; |
---|
74 | |
---|
75 | /* execute tempo detection function on iput buffer */ |
---|
76 | void aubio_tempo_do(aubio_tempo_t *o, fvec_t * input, fvec_t * tempo) |
---|
77 | { |
---|
78 | uint_t i; |
---|
79 | uint_t winlen = o->winlen; |
---|
80 | uint_t step = o->step; |
---|
81 | fvec_t * thresholded; |
---|
82 | aubio_pvoc_do (o->pv, input, o->fftgrain); |
---|
83 | aubio_specdesc_do (o->od, o->fftgrain, o->of); |
---|
84 | /*if (usedoubled) { |
---|
85 | aubio_specdesc_do(o2,fftgrain, onset2); |
---|
86 | onset->data[0] *= onset2->data[0]; |
---|
87 | }*/ |
---|
88 | /* execute every overlap_size*step */ |
---|
89 | if (o->blockpos == (signed)step -1 ) { |
---|
90 | /* check dfframe */ |
---|
91 | aubio_beattracking_do(o->bt,o->dfframe,o->out); |
---|
92 | /* rotate dfframe */ |
---|
93 | for (i = 0 ; i < winlen - step; i++ ) |
---|
94 | o->dfframe->data[i] = o->dfframe->data[i+step]; |
---|
95 | for (i = winlen - step ; i < winlen; i++ ) |
---|
96 | o->dfframe->data[i] = 0.; |
---|
97 | o->blockpos = -1; |
---|
98 | } |
---|
99 | o->blockpos++; |
---|
100 | aubio_peakpicker_do (o->pp, o->of, o->onset); |
---|
101 | tempo->data[1] = o->onset->data[0]; |
---|
102 | thresholded = aubio_peakpicker_get_thresholded_input(o->pp); |
---|
103 | o->dfframe->data[winlen - step + o->blockpos] = thresholded->data[0]; |
---|
104 | /* end of second level loop */ |
---|
105 | tempo->data[0] = 0; /* reset tactus */ |
---|
106 | i=0; |
---|
107 | for (i = 1; i < o->out->data[0]; i++ ) { |
---|
108 | /* if current frame is a predicted tactus */ |
---|
109 | if (o->blockpos == FLOOR(o->out->data[i])) { |
---|
110 | tempo->data[0] = o->out->data[i] - FLOOR(o->out->data[i]); /* set tactus */ |
---|
111 | /* test for silence */ |
---|
112 | /* |
---|
113 | if (aubio_silence_detection(input, o->silence)==1) { |
---|
114 | tempo->data[0] = 0; // unset beat if silent |
---|
115 | } |
---|
116 | */ |
---|
117 | o->last_beat = o->total_frames + (uint_t)ROUND(tempo->data[0] * o->hop_size); |
---|
118 | } |
---|
119 | } |
---|
120 | o->total_frames += o->hop_size; |
---|
121 | return; |
---|
122 | } |
---|
123 | |
---|
124 | uint_t aubio_tempo_get_last (aubio_tempo_t *o) |
---|
125 | { |
---|
126 | return o->last_beat - o->delay; |
---|
127 | } |
---|
128 | |
---|
129 | smpl_t aubio_tempo_get_last_s (aubio_tempo_t *o) |
---|
130 | { |
---|
131 | return aubio_tempo_get_last (o) / (smpl_t) (o->samplerate); |
---|
132 | } |
---|
133 | |
---|
134 | smpl_t aubio_tempo_get_last_ms (aubio_tempo_t *o) |
---|
135 | { |
---|
136 | return aubio_tempo_get_last_s (o) * 1000.; |
---|
137 | } |
---|
138 | |
---|
139 | uint_t aubio_tempo_set_delay(aubio_tempo_t * o, uint_t delay) { |
---|
140 | o->delay = delay; |
---|
141 | return AUBIO_OK; |
---|
142 | } |
---|
143 | |
---|
144 | uint_t aubio_tempo_get_delay(aubio_tempo_t * o) { |
---|
145 | return o->delay; |
---|
146 | } |
---|
147 | |
---|
148 | uint_t aubio_tempo_set_silence(aubio_tempo_t * o, smpl_t silence) { |
---|
149 | o->silence = silence; |
---|
150 | return AUBIO_OK; |
---|
151 | } |
---|
152 | |
---|
153 | uint_t aubio_tempo_set_threshold(aubio_tempo_t * o, smpl_t threshold) { |
---|
154 | o->threshold = threshold; |
---|
155 | aubio_peakpicker_set_threshold(o->pp, o->threshold); |
---|
156 | return AUBIO_OK; |
---|
157 | } |
---|
158 | |
---|
159 | /* Allocate memory for an tempo detection */ |
---|
160 | aubio_tempo_t * new_aubio_tempo (char_t * tempo_mode, |
---|
161 | uint_t buf_size, uint_t hop_size, uint_t samplerate) |
---|
162 | { |
---|
163 | aubio_tempo_t * o = AUBIO_NEW(aubio_tempo_t); |
---|
164 | char_t specdesc_func[20]; |
---|
165 | o->samplerate = samplerate; |
---|
166 | // check parameters are valid |
---|
167 | if ((sint_t)hop_size < 1) { |
---|
168 | AUBIO_ERR("tempo: got hop size %d, but can not be < 1\n", hop_size); |
---|
169 | goto beach; |
---|
170 | } else if ((sint_t)buf_size < 1) { |
---|
171 | AUBIO_ERR("tempo: got window size %d, but can not be < 1\n", buf_size); |
---|
172 | goto beach; |
---|
173 | } else if (buf_size < hop_size) { |
---|
174 | AUBIO_ERR("tempo: hop size (%d) is larger than window size (%d)\n", buf_size, hop_size); |
---|
175 | goto beach; |
---|
176 | } else if ((sint_t)samplerate < 1) { |
---|
177 | AUBIO_ERR("tempo: samplerate (%d) can not be < 1\n", samplerate); |
---|
178 | goto beach; |
---|
179 | } |
---|
180 | |
---|
181 | /* length of observations, worth about 6 seconds */ |
---|
182 | o->winlen = aubio_next_power_of_two(5.8 * samplerate / hop_size); |
---|
183 | if (o->winlen < 4) o->winlen = 4; |
---|
184 | o->step = o->winlen/4; |
---|
185 | o->blockpos = 0; |
---|
186 | o->threshold = 0.3; |
---|
187 | o->silence = -90.; |
---|
188 | o->total_frames = 0; |
---|
189 | o->last_beat = 0; |
---|
190 | o->delay = 0; |
---|
191 | o->hop_size = hop_size; |
---|
192 | o->dfframe = new_fvec(o->winlen); |
---|
193 | o->fftgrain = new_cvec(buf_size); |
---|
194 | o->out = new_fvec(o->step); |
---|
195 | o->pv = new_aubio_pvoc(buf_size, hop_size); |
---|
196 | o->pp = new_aubio_peakpicker(); |
---|
197 | aubio_peakpicker_set_threshold (o->pp, o->threshold); |
---|
198 | if ( strcmp(tempo_mode, "default") == 0 ) { |
---|
199 | strcpy(specdesc_func, "specflux"); |
---|
200 | } else { |
---|
201 | strcpy(specdesc_func, tempo_mode); |
---|
202 | } |
---|
203 | o->od = new_aubio_specdesc(specdesc_func,buf_size); |
---|
204 | o->of = new_fvec(1); |
---|
205 | o->bt = new_aubio_beattracking(o->winlen, o->hop_size, o->samplerate); |
---|
206 | o->onset = new_fvec(1); |
---|
207 | /*if (usedoubled) { |
---|
208 | o2 = new_aubio_specdesc(type_onset2,buffer_size); |
---|
209 | onset2 = new_fvec(1); |
---|
210 | }*/ |
---|
211 | return o; |
---|
212 | |
---|
213 | beach: |
---|
214 | AUBIO_FREE(o); |
---|
215 | return NULL; |
---|
216 | } |
---|
217 | |
---|
218 | smpl_t aubio_tempo_get_bpm(aubio_tempo_t *o) { |
---|
219 | return aubio_beattracking_get_bpm(o->bt); |
---|
220 | } |
---|
221 | |
---|
222 | smpl_t aubio_tempo_get_confidence(aubio_tempo_t *o) { |
---|
223 | return aubio_beattracking_get_confidence(o->bt); |
---|
224 | } |
---|
225 | |
---|
226 | void del_aubio_tempo (aubio_tempo_t *o) |
---|
227 | { |
---|
228 | del_aubio_specdesc(o->od); |
---|
229 | del_aubio_beattracking(o->bt); |
---|
230 | del_aubio_peakpicker(o->pp); |
---|
231 | del_aubio_pvoc(o->pv); |
---|
232 | del_fvec(o->out); |
---|
233 | del_fvec(o->of); |
---|
234 | del_cvec(o->fftgrain); |
---|
235 | del_fvec(o->dfframe); |
---|
236 | del_fvec(o->onset); |
---|
237 | AUBIO_FREE(o); |
---|
238 | return; |
---|
239 | } |
---|