source: src/onset/onset.c @ 32d6958

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

src/: more moving and splitting

  • Property mode set to 100644
File size: 3.6 KB
Line 
1/*
2   Copyright (C) 2006 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.
17
18*/
19
20#include "aubio_priv.h"
21#include "sample.h"
22#include "onset/onsetdetection.h"
23#include "spectral/phasevoc.h"
24#include "onset/peakpick.h"
25#include "mathutils.h"
26#include "onset/onset.h"
27
28/** structure to store object state */
29struct _aubio_onset_t {
30  aubio_pvoc_t * pv;            /**< phase vocoder */
31  aubio_onsetdetection_t * od;  /**< onset detection */ 
32  aubio_pickpeak_t * pp;        /**< peak picker */
33  cvec_t * fftgrain;            /**< phase vocoder output */
34  fvec_t * of;                  /**< onset detection function */
35  smpl_t threshold;             /**< onset peak picking threshold */
36  smpl_t silence;               /**< silence threhsold */
37  uint_t minioi;                /**< minimum inter onset interval */
38  uint_t wasonset;              /**< number of frames since last onset */
39};
40
41/* execute onset detection function on iput buffer */
42void aubio_onset(aubio_onset_t *o, fvec_t * input, fvec_t * onset)
43{
44  uint_t isonset = 0;
45  uint_t wasonset = o->wasonset;
46  aubio_pvoc_do (o->pv,input, o->fftgrain);
47  aubio_onsetdetection(o->od,o->fftgrain, o->of);
48  /*if (usedoubled) {
49    aubio_onsetdetection(o2,fftgrain, onset2);
50    onset->data[0][0] *= onset2->data[0][0];
51  }*/
52  isonset = aubio_peakpick_pimrt(o->of,o->pp);
53  if (isonset) {
54    if (aubio_silence_detection(input, o->silence)==1) {
55      isonset  = 0;
56      wasonset++;
57    } else {
58      if (wasonset > o->minioi) {
59        wasonset = 0;
60      } else {
61        isonset  = 0;
62        wasonset++;
63      }
64    }
65  } else {
66    wasonset++;
67  }
68  o->wasonset = wasonset;
69  onset->data[0][0] = isonset;
70  return;
71}
72
73void aubio_onset_set_silence(aubio_onset_t * o, smpl_t silence) {
74  o->silence = silence;
75  return;
76}
77
78void aubio_onset_set_threshold(aubio_onset_t * o, smpl_t threshold) {
79  o->threshold = threshold;
80  aubio_peakpicker_set_threshold(o->pp, o->threshold);
81  return;
82}
83
84void aubio_onset_set_minioi(aubio_onset_t * o, uint_t minioi) {
85  o->minioi = minioi;
86  return;
87}
88
89/* Allocate memory for an onset detection */
90aubio_onset_t * new_aubio_onset (aubio_onsetdetection_type type_onset, 
91    uint_t buf_size, uint_t hop_size, uint_t channels)
92{
93  aubio_onset_t * o = AUBIO_NEW(aubio_onset_t);
94  /** set some default parameter */
95  o->threshold = 0.3;
96  o->minioi    = 4;
97  o->silence   = -70;
98  o->wasonset  = 0;
99  o->pv = new_aubio_pvoc(buf_size, hop_size, channels);
100  o->pp = new_aubio_peakpicker(o->threshold);
101  o->od = new_aubio_onsetdetection(type_onset,buf_size,channels);
102  o->fftgrain = new_cvec(buf_size,channels);
103  o->of = new_fvec(1, channels);
104  /*if (usedoubled)    {
105    o2 = new_aubio_onsetdetection(type_onset2,buffer_size,channels);
106    onset2 = new_fvec(1 , channels);
107  }*/
108  return o;
109}
110
111void del_aubio_onset (aubio_onset_t *o)
112{
113  del_aubio_onsetdetection(o->od);
114  del_aubio_peakpicker(o->pp);
115  del_aubio_pvoc(o->pv);
116  del_fvec(o->of);
117  del_cvec(o->fftgrain);
118  AUBIO_FREE(o);
119}
Note: See TracBrowser for help on using the repository browser.