[96fb8ad] | 1 | %module aubiowrapper |
---|
| 2 | |
---|
| 3 | %{ |
---|
| 4 | #include "aubio.h" |
---|
[205da86] | 5 | #include "aubioext.h" |
---|
[96fb8ad] | 6 | %} |
---|
| 7 | |
---|
| 8 | #include "aubio.h" |
---|
[205da86] | 9 | #include "aubioext.h" |
---|
[96fb8ad] | 10 | |
---|
| 11 | /* type aliases */ |
---|
| 12 | typedef unsigned int uint_t; |
---|
| 13 | typedef int sint_t; |
---|
| 14 | typedef float smpl_t; |
---|
| 15 | |
---|
| 16 | /* fvec */ |
---|
| 17 | extern fvec_t * new_fvec(uint_t length, uint_t channels); |
---|
| 18 | extern void del_fvec(fvec_t *s); |
---|
| 19 | smpl_t fvec_read_sample(fvec_t *s, uint_t channel, uint_t position); |
---|
| 20 | void fvec_write_sample(fvec_t *s, smpl_t data, uint_t channel, uint_t position); |
---|
| 21 | smpl_t * fvec_get_channel(fvec_t *s, uint_t channel); |
---|
| 22 | void fvec_put_channel(fvec_t *s, smpl_t * data, uint_t channel); |
---|
| 23 | smpl_t ** fvec_get_data(fvec_t *s); |
---|
| 24 | |
---|
| 25 | /* another way, passing -c++ option to swig */ |
---|
| 26 | /* |
---|
| 27 | class fvec_t{ |
---|
| 28 | public: |
---|
| 29 | %extend { |
---|
| 30 | fvec_t(uint_t length, uint_t channels){ |
---|
| 31 | return new_fvec(length, channels); |
---|
| 32 | } |
---|
| 33 | ~fvec_t() { |
---|
| 34 | del_fvec(self); |
---|
| 35 | } |
---|
| 36 | smpl_t get( uint_t channel, uint_t position) { |
---|
| 37 | return fvec_read_sample(self,channel,position); |
---|
| 38 | } |
---|
| 39 | void set( smpl_t data, uint_t channel, uint_t position) { |
---|
| 40 | fvec_write_sample(self, data, channel, position); |
---|
| 41 | } |
---|
| 42 | #smpl_t * fvec_get_channel(fvec_t *s, uint_t channel); |
---|
| 43 | #void fvec_put_channel(fvec_t *s, smpl_t * data, uint_t channel); |
---|
| 44 | } |
---|
| 45 | }; |
---|
| 46 | */ |
---|
| 47 | |
---|
| 48 | /* cvec */ |
---|
| 49 | extern cvec_t * new_cvec(uint_t length, uint_t channels); |
---|
| 50 | extern void del_cvec(cvec_t *s); |
---|
[7c206df] | 51 | extern void cvec_write_norm(cvec_t *s, smpl_t data, uint_t channel, uint_t position); |
---|
| 52 | extern void cvec_write_phas(cvec_t *s, smpl_t data, uint_t channel, uint_t position); |
---|
| 53 | extern smpl_t cvec_read_norm(cvec_t *s, uint_t channel, uint_t position); |
---|
| 54 | extern smpl_t cvec_read_phas(cvec_t *s, uint_t channel, uint_t position); |
---|
| 55 | extern void cvec_put_norm_channel(cvec_t *s, smpl_t * data, uint_t channel); |
---|
| 56 | extern void cvec_put_phas_channel(cvec_t *s, smpl_t * data, uint_t channel); |
---|
| 57 | extern smpl_t * cvec_get_norm_channel(cvec_t *s, uint_t channel); |
---|
| 58 | extern smpl_t * cvec_get_phas_channel(cvec_t *s, uint_t channel); |
---|
| 59 | extern smpl_t ** cvec_get_norm(cvec_t *s); |
---|
| 60 | extern smpl_t ** cvec_get_phas(cvec_t *s); |
---|
[96fb8ad] | 61 | |
---|
| 62 | |
---|
| 63 | /* sndfile */ |
---|
[eb089fb] | 64 | %#if HAVE_SNDFILE |
---|
[5e9c68a] | 65 | extern aubio_sndfile_t * new_aubio_sndfile_ro (const char * inputfile); |
---|
| 66 | extern aubio_sndfile_t * new_aubio_sndfile_wo(aubio_sndfile_t * existingfile, const char * outputname); |
---|
| 67 | extern void aubio_sndfile_info(aubio_sndfile_t * file); |
---|
| 68 | extern int aubio_sndfile_write(aubio_sndfile_t * file, int frames, fvec_t * write); |
---|
| 69 | extern int aubio_sndfile_read(aubio_sndfile_t * file, int frames, fvec_t * read); |
---|
| 70 | extern int del_aubio_sndfile(aubio_sndfile_t * file); |
---|
| 71 | extern uint_t aubio_sndfile_channels(aubio_sndfile_t * file); |
---|
| 72 | extern uint_t aubio_sndfile_samplerate(aubio_sndfile_t * file); |
---|
[eb089fb] | 73 | %#endif /* HAVE_SNDFILE */ |
---|
[96fb8ad] | 74 | |
---|
| 75 | /* fft */ |
---|
[49407f3] | 76 | extern aubio_fft_t * new_aubio_fft(uint_t size, uint_t channels); |
---|
| 77 | extern void del_aubio_fft(aubio_fft_t * s); |
---|
| 78 | extern void aubio_fft_do (aubio_fft_t *s, fvec_t * input, cvec_t * spectrum); |
---|
| 79 | extern void aubio_fft_rdo (aubio_fft_t *s, cvec_t * spectrum, fvec_t * output); |
---|
| 80 | extern void aubio_fft_do_complex (aubio_fft_t *s, fvec_t * input, fvec_t * compspec); |
---|
| 81 | extern void aubio_fft_rdo_complex (aubio_fft_t *s, fvec_t * compspec, fvec_t * output); |
---|
| 82 | extern void aubio_fft_get_spectrum(fvec_t * compspec, cvec_t * spectrum); |
---|
| 83 | extern void aubio_fft_get_realimag(cvec_t * spectrum, fvec_t * compspec); |
---|
| 84 | extern void aubio_fft_get_phas(fvec_t * compspec, cvec_t * spectrum); |
---|
| 85 | extern void aubio_fft_get_imag(cvec_t * spectrum, fvec_t * compspec); |
---|
| 86 | extern void aubio_fft_get_norm(fvec_t * compspec, cvec_t * spectrum); |
---|
| 87 | extern void aubio_fft_get_real(cvec_t * spectrum, fvec_t * compspec); |
---|
[ec10780] | 88 | |
---|
[96fb8ad] | 89 | /* filter */ |
---|
[a7667ce] | 90 | extern aubio_filter_t * new_aubio_filter(uint_t samplerate, uint_t order, uint_t channels); |
---|
[96fb8ad] | 91 | extern void aubio_filter_do(aubio_filter_t * b, fvec_t * in); |
---|
| 92 | extern void aubio_filter_do_outplace(aubio_filter_t * b, fvec_t * in, fvec_t * out); |
---|
| 93 | extern void aubio_filter_do_filtfilt(aubio_filter_t * b, fvec_t * in, fvec_t * tmp); |
---|
[4123f16] | 94 | extern void del_aubio_filter(aubio_filter_t * b); |
---|
| 95 | |
---|
[bc4e175] | 96 | extern aubio_filter_t * new_aubio_filter_a_weighting (uint_t samplerate, uint_t channels); |
---|
[eebb0e1] | 97 | extern uint_t aubio_filter_set_a_weighting (aubio_filter_t * b); |
---|
[4123f16] | 98 | |
---|
[bc4e175] | 99 | extern aubio_filter_t * new_aubio_filter_c_weighting (uint_t samplerate, uint_t channels); |
---|
[eebb0e1] | 100 | extern uint_t aubio_filter_set_c_weighting (aubio_filter_t * b); |
---|
[96fb8ad] | 101 | |
---|
| 102 | /* biquad */ |
---|
| 103 | extern aubio_biquad_t * new_aubio_biquad(lsmp_t b1, lsmp_t b2, lsmp_t b3, lsmp_t a2, lsmp_t a3); |
---|
| 104 | extern void aubio_biquad_do(aubio_biquad_t * b, fvec_t * in); |
---|
| 105 | extern void aubio_biquad_do_filtfilt(aubio_biquad_t * b, fvec_t * in, fvec_t * tmp); |
---|
[055d674] | 106 | extern void del_aubio_biquad(aubio_biquad_t * b); |
---|
[96fb8ad] | 107 | |
---|
| 108 | /* hist */ |
---|
| 109 | extern aubio_hist_t * new_aubio_hist(smpl_t flow, smpl_t fhig, uint_t nelems, uint_t channels); |
---|
| 110 | extern void del_aubio_hist(aubio_hist_t *s); |
---|
| 111 | extern void aubio_hist_do(aubio_hist_t *s, fvec_t * input); |
---|
| 112 | extern void aubio_hist_do_notnull(aubio_hist_t *s, fvec_t * input); |
---|
[49407f3] | 113 | extern void aubio_hist_dyn_notnull(aubio_hist_t *s, fvec_t *input); |
---|
| 114 | extern void aubio_hist_weight(aubio_hist_t *s); |
---|
| 115 | extern smpl_t aubio_hist_mean(aubio_hist_t *s); |
---|
[96fb8ad] | 116 | |
---|
| 117 | /* mathutils */ |
---|
| 118 | typedef enum { |
---|
[b4b0324] | 119 | aubio_win_rectangle, |
---|
| 120 | aubio_win_hamming, |
---|
| 121 | aubio_win_hanning, |
---|
| 122 | aubio_win_hanningz, |
---|
| 123 | aubio_win_blackman, |
---|
| 124 | aubio_win_blackman_harris, |
---|
| 125 | aubio_win_gaussian, |
---|
| 126 | aubio_win_welch, |
---|
| 127 | aubio_win_parzen |
---|
[28d8c4a] | 128 | } aubio_window_type; |
---|
[96fb8ad] | 129 | |
---|
[d84d19e] | 130 | fvec_t * new_aubio_window(uint_t size, aubio_window_type wintype); |
---|
[28d8c4a] | 131 | smpl_t aubio_unwrap2pi (smpl_t phase); |
---|
[8b28524] | 132 | smpl_t fvec_mean(fvec_t *s); |
---|
[1e2c82f] | 133 | smpl_t fvec_max(fvec_t *s); |
---|
[2f64b0e] | 134 | smpl_t fvec_min(fvec_t *s); |
---|
| 135 | uint_t fvec_min_elem(fvec_t *s); |
---|
[1e2c82f] | 136 | uint_t fvec_max_elem(fvec_t *s); |
---|
[5c4ec3c] | 137 | void fvec_shift(fvec_t *s); |
---|
| 138 | smpl_t fvec_sum(fvec_t *s); |
---|
| 139 | smpl_t fvec_local_energy(fvec_t * f); |
---|
| 140 | smpl_t fvec_local_hfc(fvec_t * f); |
---|
| 141 | smpl_t fvec_alpha_norm(fvec_t * DF, smpl_t alpha); |
---|
[c0b295c] | 142 | void fvec_min_removal(fvec_t * mag); |
---|
[5c4ec3c] | 143 | void fvec_alpha_normalise(fvec_t * mag, uint_t alpha); |
---|
| 144 | void fvec_add(fvec_t * mag, smpl_t threshold); |
---|
| 145 | void fvec_adapt_thres(fvec_t * vec, fvec_t * tmp, uint_t post, uint_t pre); |
---|
| 146 | smpl_t fvec_moving_thres(fvec_t * vec, fvec_t * tmp, uint_t post, uint_t pre, uint_t pos); |
---|
| 147 | smpl_t fvec_median(fvec_t * input); |
---|
| 148 | smpl_t fvec_quadint(fvec_t * x,uint_t pos, uint_t span); |
---|
[28d8c4a] | 149 | smpl_t aubio_quadfrac(smpl_t s0, smpl_t s1, smpl_t s2, smpl_t pf); |
---|
[5c4ec3c] | 150 | uint_t fvec_peakpick(fvec_t * input, uint_t pos); |
---|
[28d8c4a] | 151 | smpl_t aubio_bintomidi(smpl_t bin, smpl_t samplerate, smpl_t fftsize); |
---|
[79c2e52] | 152 | smpl_t aubio_miditobin(smpl_t midi, smpl_t samplerate, smpl_t fftsize); |
---|
[28d8c4a] | 153 | smpl_t aubio_bintofreq(smpl_t bin, smpl_t samplerate, smpl_t fftsize); |
---|
[79c2e52] | 154 | smpl_t aubio_freqtobin(smpl_t freq, smpl_t samplerate, smpl_t fftsize); |
---|
[28d8c4a] | 155 | smpl_t aubio_freqtomidi(smpl_t freq); |
---|
[79c2e52] | 156 | smpl_t aubio_miditofreq(smpl_t midi); |
---|
[96fb8ad] | 157 | uint_t aubio_silence_detection(fvec_t * ibuf, smpl_t threshold); |
---|
| 158 | smpl_t aubio_level_detection(fvec_t * ibuf, smpl_t threshold); |
---|
[fe4f78a] | 159 | void aubio_autocorr(fvec_t * in, fvec_t * acf); |
---|
[5507e9d] | 160 | smpl_t aubio_zero_crossing_rate(fvec_t * input); |
---|
[c8b80cd] | 161 | smpl_t aubio_spectral_centroid(cvec_t * spectrum, smpl_t samplerate); |
---|
[96fb8ad] | 162 | |
---|
[bc4ba75] | 163 | /* filterbank */ |
---|
[dc73a4d0] | 164 | aubio_filterbank_t * new_aubio_filterbank(uint_t win_s, uint_t channels); |
---|
[8766cb6] | 165 | void aubio_filterbank_set_triangle_bands (aubio_filterbank_t *fb, uint_t samplerate, fvec_t *freqs); |
---|
[1e37ade] | 166 | void aubio_filterbank_set_mel_coeffs_slaney(aubio_filterbank_t *fb, uint_t samplerate); |
---|
[dc73a4d0] | 167 | void del_aubio_filterbank(aubio_filterbank_t * fb); |
---|
| 168 | void aubio_filterbank_do(aubio_filterbank_t * fb, cvec_t * in, fvec_t *out); |
---|
[06cae6c] | 169 | fvec_t * aubio_filterbank_get_coeffs(aubio_filterbank_t * fb); |
---|
[bc4ba75] | 170 | |
---|
| 171 | /* mfcc */ |
---|
[adcf405] | 172 | aubio_mfcc_t * new_aubio_mfcc (uint_t win_s, uint_t samplerate, uint_t n_filters, uint_t n_coefs); |
---|
[bc4ba75] | 173 | void del_aubio_mfcc(aubio_mfcc_t *mf); |
---|
| 174 | void aubio_mfcc_do(aubio_mfcc_t *mf, cvec_t *in, fvec_t *out); |
---|
| 175 | |
---|
[96fb8ad] | 176 | /* scale */ |
---|
[b061ef8] | 177 | extern aubio_scale_t * new_aubio_scale(smpl_t flow, smpl_t fhig, smpl_t ilow, smpl_t ihig); |
---|
[0ce97483] | 178 | extern void aubio_scale_set_limits (aubio_scale_t *s, smpl_t ilow, smpl_t ihig, smpl_t olow, smpl_t ohig); |
---|
[96fb8ad] | 179 | extern void aubio_scale_do(aubio_scale_t *s, fvec_t * input); |
---|
| 180 | extern void del_aubio_scale(aubio_scale_t *s); |
---|
| 181 | |
---|
| 182 | /* resampling */ |
---|
[eb089fb] | 183 | %#if HAVE_SAMPLERATE |
---|
[96fb8ad] | 184 | extern aubio_resampler_t * new_aubio_resampler(float ratio, uint_t type); |
---|
[64c3658] | 185 | extern void aubio_resampler_do (aubio_resampler_t *s, fvec_t * input, fvec_t * output); |
---|
[96fb8ad] | 186 | extern void del_aubio_resampler(aubio_resampler_t *s); |
---|
[eb089fb] | 187 | %#endif /* HAVE_SAMPLERATE */ |
---|
[96fb8ad] | 188 | |
---|
| 189 | /* onset detection */ |
---|
[b4f5967] | 190 | aubio_onsetdetection_t * new_aubio_onsetdetection(char * onset_mode, uint_t size, uint_t channels); |
---|
[01b8fcc] | 191 | void aubio_onsetdetection_do (aubio_onsetdetection_t *o, cvec_t * fftgrain, fvec_t * onset); |
---|
[2a00568] | 192 | void del_aubio_onsetdetection(aubio_onsetdetection_t *o); |
---|
[96fb8ad] | 193 | |
---|
| 194 | /* pvoc */ |
---|
| 195 | aubio_pvoc_t * new_aubio_pvoc (uint_t win_s, uint_t hop_s, uint_t channels); |
---|
| 196 | void del_aubio_pvoc(aubio_pvoc_t *pv); |
---|
| 197 | void aubio_pvoc_do(aubio_pvoc_t *pv, fvec_t *in, cvec_t * fftgrain); |
---|
| 198 | void aubio_pvoc_rdo(aubio_pvoc_t *pv, cvec_t * fftgrain, fvec_t *out); |
---|
| 199 | |
---|
| 200 | /* pitch detection */ |
---|
[fe163ad] | 201 | aubio_pitchdetection_t *new_aubio_pitchdetection (char *pitch_mode, |
---|
| 202 | uint_t bufsize, uint_t hopsize, uint_t channels, uint_t samplerate); |
---|
[05773a2c] | 203 | void aubio_pitchdetection_do (aubio_pitchdetection_t * p, fvec_t * ibuf, fvec_t * obuf); |
---|
| 204 | void aubio_pitchdetection_set_tolerance(aubio_pitchdetection_t *p, smpl_t thres); |
---|
[fe163ad] | 205 | void aubio_pitchdetection_set_unit(aubio_pitchdetection_t *p, char * pitch_unit); |
---|
[8e8bc50] | 206 | void del_aubio_pitchdetection(aubio_pitchdetection_t * p); |
---|
| 207 | |
---|
[96fb8ad] | 208 | |
---|
| 209 | /* pitch mcomb */ |
---|
[05773a2c] | 210 | aubio_pitchmcomb_t * new_aubio_pitchmcomb(uint_t bufsize, uint_t hopsize, uint_t channels); |
---|
| 211 | void aubio_pitchmcomb_do (aubio_pitchmcomb_t * p, cvec_t * fftgrain, fvec_t * out); |
---|
[7a04950] | 212 | void del_aubio_pitchmcomb (aubio_pitchmcomb_t *p); |
---|
[96fb8ad] | 213 | |
---|
| 214 | /* pitch yin */ |
---|
[05773a2c] | 215 | aubio_pitchyin_t * new_aubio_pitchyin(uint_t bufsize); |
---|
| 216 | void aubio_pitchyin_do (aubio_pitchyin_t *o, fvec_t * in, fvec_t *out); |
---|
| 217 | void del_aubio_pitchyin (aubio_pitchyin_t *p); |
---|
[96fb8ad] | 218 | |
---|
[7a04950] | 219 | /* pitch schmitt */ |
---|
[05773a2c] | 220 | aubio_pitchschmitt_t * new_aubio_pitchschmitt (uint_t size); |
---|
| 221 | void aubio_pitchschmitt_do (aubio_pitchschmitt_t *p, fvec_t * input, fvec_t *out); |
---|
[7a04950] | 222 | void del_aubio_pitchschmitt (aubio_pitchschmitt_t *p); |
---|
| 223 | |
---|
| 224 | /* pitch fcomb */ |
---|
[4afa096d] | 225 | aubio_pitchfcomb_t * new_aubio_pitchfcomb (uint_t size, uint_t hopsize, uint_t samplerate); |
---|
[05773a2c] | 226 | void aubio_pitchfcomb_do (aubio_pitchfcomb_t *p, fvec_t * input, fvec_t *out); |
---|
[7a04950] | 227 | void del_aubio_pitchfcomb (aubio_pitchfcomb_t *p); |
---|
| 228 | |
---|
[96fb8ad] | 229 | /* peakpicker */ |
---|
[8766cb6] | 230 | aubio_peakpicker_t * new_aubio_peakpicker(smpl_t threshold); |
---|
| 231 | smpl_t aubio_peakpicker_do(aubio_peakpicker_t * p, fvec_t * df); |
---|
| 232 | smpl_t aubio_peakpicker_get_thresholded_input(aubio_peakpicker_t* p); |
---|
| 233 | void del_aubio_peakpicker(aubio_peakpicker_t * p); |
---|
| 234 | void aubio_peakpicker_set_threshold(aubio_peakpicker_t * p, smpl_t threshold); |
---|
| 235 | smpl_t aubio_peakpicker_get_threshold(aubio_peakpicker_t * p); |
---|
[96fb8ad] | 236 | |
---|
| 237 | /* transient/steady state separation */ |
---|
| 238 | aubio_tss_t * new_aubio_tss(smpl_t thrs, smpl_t alfa, smpl_t beta, |
---|
| 239 | uint_t size, uint_t overlap,uint_t channels); |
---|
| 240 | void del_aubio_tss(aubio_tss_t *s); |
---|
| 241 | void aubio_tss_do(aubio_tss_t *s, cvec_t * input, cvec_t * trans, cvec_t * stead); |
---|
| 242 | |
---|
[de92b1c] | 243 | /* beattracking */ |
---|
| 244 | aubio_beattracking_t * new_aubio_beattracking(uint_t winlen, uint_t channels); |
---|
| 245 | void aubio_beattracking_do(aubio_beattracking_t * bt, fvec_t * dfframes, fvec_t * out); |
---|
| 246 | void del_aubio_beattracking(aubio_beattracking_t * p); |
---|
[e34b010] | 247 | smpl_t aubio_beattracking_get_bpm(aubio_beattracking_t * p); |
---|
| 248 | smpl_t aubio_beattracking_get_confidence(aubio_beattracking_t * p); |
---|
[de92b1c] | 249 | |
---|