source: src/spectral/tss.c @ 2763582

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

src/spectral/: add const qualifiers

  • Property mode set to 100644
File size: 3.4 KB
Line 
1/*
2  Copyright (C) 2003-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/* default values : alpha=4, beta=3, threshold=0.25 */
22
23#include "aubio_priv.h"
24#include "fvec.h"
25#include "cvec.h"
26#include "mathutils.h"
27#include "spectral/tss.h"
28
29struct _aubio_tss_t
30{
31  smpl_t threshold;
32  smpl_t alpha;
33  smpl_t beta;
34  smpl_t parm;
35  smpl_t thrsfact;
36  fvec_t *theta1;
37  fvec_t *theta2;
38  fvec_t *oft1;
39  fvec_t *oft2;
40  fvec_t *dev;
41};
42
43void aubio_tss_do(aubio_tss_t *o, const cvec_t * input,
44    cvec_t * trans, cvec_t * stead)
45{
46  uint_t j;
47  uint_t test;
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];
63
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;
68
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;
73
74    /*increase probability for transient */
75    test = (trans->norm[j]==0.);
76    oft1[j]  = test;
77    test = (trans->norm[j]>0.);
78    oft1[j] += alpha*test;
79    test = (oft1[j]>1. && trans->norm[j]>0.);
80    oft1[j] += beta*test;
81
82    /*increase probability for steady states */
83    test = (stead->norm[j]==0.);
84    oft2[j]  = test;
85    test = (stead->norm[j]>0.);
86    oft2[j] += alpha*test;
87    test = (oft2[j]>1. && stead->norm[j]>0.);
88    oft2[j] += beta*test;
89  }
90}
91
92uint_t aubio_tss_set_threshold(aubio_tss_t *o, smpl_t threshold){
93  o->threshold = threshold;
94  o->parm = o->threshold * o->thrsfact;
95  return AUBIO_OK;
96}
97
98aubio_tss_t * new_aubio_tss(uint_t buf_size, uint_t hop_size)
99{
100  aubio_tss_t * o = AUBIO_NEW(aubio_tss_t);
101  uint_t rsize = buf_size/2+1;
102  o->threshold = 0.25;
103  o->thrsfact = TWO_PI*hop_size/rsize;
104  o->alpha = 3.;
105  o->beta = 4.;
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);
112  return o;
113}
114
115void del_aubio_tss(aubio_tss_t *s)
116{
117  del_fvec(s->theta1);
118  del_fvec(s->theta2);
119  del_fvec(s->oft1);
120  del_fvec(s->oft2);
121  del_fvec(s->dev);
122  AUBIO_FREE(s);
123}
124
125uint_t aubio_tss_set_alpha(aubio_tss_t *o, smpl_t alpha){
126  o->alpha = alpha;
127  return AUBIO_OK;
128}
129
130uint_t aubio_tss_set_beta(aubio_tss_t *o, smpl_t beta){
131  o->beta = beta;
132  return AUBIO_OK;
133}
134
Note: See TracBrowser for help on using the repository browser.