2 Copyright (C) 2003-2009 Paul Brossier <piem@aubio.org>
4 This file is part of aubio.
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.
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.
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/>.
21 /* default values : alpha=4, beta=3, threshold=0.25 */
23 #include "aubio_priv.h"
26 #include "mathutils.h"
27 #include "spectral/tss.h"
43 void aubio_tss_do(aubio_tss_t *o, const cvec_t * input,
44 cvec_t * trans, cvec_t * stead)
48 uint_t nbins = input->length;
49 smpl_t alpha = o->alpha;
50 smpl_t beta = o->beta;
51 smpl_t parm = o->parm;
52 smpl_t * dev = (smpl_t *)o->dev->data;
53 smpl_t * oft1 = (smpl_t *)o->oft1->data;
54 smpl_t * oft2 = (smpl_t *)o->oft2->data;
55 smpl_t * theta1 = (smpl_t *)o->theta1->data;
56 smpl_t * theta2 = (smpl_t *)o->theta2->data;
57 /* second phase derivative */
58 for (j=0;j<nbins; j++){
59 dev[j] = aubio_unwrap2pi(input->phas[j]
60 -2.0*theta1[j]+theta2[j]);
61 theta2[j] = theta1[j];
62 theta1[j] = input->phas[j];
64 /* transient analysis */
65 test = (ABS(dev[j]) > parm*oft1[j]);
66 trans->norm[j] = input->norm[j] * test;
67 trans->phas[j] = input->phas[j] * test;
69 /* steady state analysis */
70 test = (ABS(dev[j]) < parm*oft2[j]);
71 stead->norm[j] = input->norm[j] * test;
72 stead->phas[j] = input->phas[j] * test;
74 /*increase probability for transient */
75 test = (trans->norm[j]==0.);
77 test = (trans->norm[j]>0.);
78 oft1[j] += alpha*test;
79 test = (oft1[j]>1. && trans->norm[j]>0.);
82 /*increase probability for steady states */
83 test = (stead->norm[j]==0.);
85 test = (stead->norm[j]>0.);
86 oft2[j] += alpha*test;
87 test = (oft2[j]>1. && stead->norm[j]>0.);
92 uint_t aubio_tss_set_threshold(aubio_tss_t *o, smpl_t threshold){
93 o->threshold = threshold;
94 o->parm = o->threshold * o->thrsfact;
98 aubio_tss_t * new_aubio_tss(uint_t buf_size, uint_t hop_size)
100 aubio_tss_t * o = AUBIO_NEW(aubio_tss_t);
101 uint_t rsize = buf_size/2+1;
103 o->thrsfact = TWO_PI*hop_size/rsize;
106 o->parm = o->threshold*o->thrsfact;
107 o->theta1 = new_fvec(rsize);
108 o->theta2 = new_fvec(rsize);
109 o->oft1 = new_fvec(rsize);
110 o->oft2 = new_fvec(rsize);
111 o->dev = new_fvec(rsize);
115 void del_aubio_tss(aubio_tss_t *s)
125 uint_t aubio_tss_set_alpha(aubio_tss_t *o, smpl_t alpha){
130 uint_t aubio_tss_set_beta(aubio_tss_t *o, smpl_t beta){