src/onset/onset.{c,h}: add method to set adaptive whitening

diff --git a/src/onset/onset.c b/src/onset/onset.c
index 6579d65..fb215fe 100644 (file)
--- a/src/onset/onset.c
+++ b/src/onset/onset.c
@@ -112,6 +112,17 @@ smpl_t aubio_onset_get_last_ms (aubio_onset_t *o)
   return aubio_onset_get_last_s (o) * 1000.;
 }
 
+uint_t aubio_onset_set_adaptive_whitening (aubio_onset_t *o, uint_t apply_adaptive_whitening)
+{
+  o->apply_adaptive_whitening = apply_adaptive_whitening;
+  return AUBIO_OK;
+}
+
+uint_t aubio_onset_get_adaptive_whitening (aubio_onset_t *o)
+{
+  return o->apply_adaptive_whitening;
+}
+
 uint_t aubio_onset_set_silence(aubio_onset_t * o, smpl_t silence) {
   o->silence = silence;
   return AUBIO_OK;
@@ -228,7 +239,7 @@ aubio_onset_t * new_aubio_onset (char_t * onset_mode,
   aubio_onset_set_silence(o, -70.);
 
   o->spectral_whitening = new_aubio_spectral_whitening(buf_size, hop_size, samplerate);
-  o->apply_adaptive_whitening = 1;
+  o->apply_adaptive_whitening = 0;
 
   /* initialize internal variables */
   o->last_onset = 0;

diff --git a/src/onset/onset.h b/src/onset/onset.h
index cda8d9a..984d35c 100644 (file)
--- a/src/onset/onset.h
+++ b/src/onset/onset.h
@@ -117,6 +117,23 @@ smpl_t aubio_onset_get_last_s (aubio_onset_t *o);
 */
 smpl_t aubio_onset_get_last_ms (aubio_onset_t *o);
 
+/** set onset detection adaptive whitening
+
+  \param o onset detection object as returned by new_aubio_onset()
+  \param apply_adaptive_whitening 1 to enable, 0 to disable
+
+*/
+uint_t aubio_onset_set_adaptive_whitening(aubio_onset_t * o, uint_t apply_adaptive_whitening);
+
+/** get onset detection silence threshold
+
+  \param o onset detection object as returned by new_aubio_onset()
+
+  \return adaptive whitening mode, 1 if enabled, 0 otherwise
+
+*/
+uint_t aubio_onset_get_adaptive_whitening(aubio_onset_t * o);
+
 /** set onset detection silence threshold
 
   \param o onset detection object as returned by new_aubio_onset()