mathutils.c: add aubio_spectral_centroid

diff --git a/src/mathutils.c b/src/mathutils.c
index e91f3e9fcabfc608700e084ca910dbb4270c0e79..d561fced4ea6f38df8749d3153b1cbb5440e511e 100644 (file)
--- a/src/mathutils.c
+++ b/src/mathutils.c
@@ -457,6 +457,19 @@ smpl_t aubio_zero_crossing_rate(fvec_t * input) {
   return zcr/(smpl_t)input->length;
 }
 
+smpl_t aubio_spectral_centroid(cvec_t * spectrum, smpl_t samplerate) {
+  uint_t i=0, j;
+  smpl_t sum = 0., sc = 0.;
+  for ( j = 0; j < spectrum->length; j++ ) {
+    sum += spectrum->norm[i][j];
+  }
+  if (sum == 0.) return 0.;
+  for ( j = 0; j < spectrum->length; j++ ) {
+    sc += (smpl_t)j * spectrum->norm[i][j];
+  }
+  return sc / sum * samplerate / (smpl_t)(spectrum->length);
+}
+
 void aubio_autocorr(fvec_t * input, fvec_t * output){
         uint_t i = 0, j = 0, length = input->length;
         smpl_t * data = input->data[0];

diff --git a/src/mathutils.h b/src/mathutils.h
index 9bf0f5836660bd1cbe752dc65bcd25799dbb2771..75fc5d7ef70c8c47abf0a41b72516e3f0e299852 100644 (file)
--- a/src/mathutils.h
+++ b/src/mathutils.h
@@ -207,6 +207,10 @@ void aubio_autocorr(fvec_t * input, fvec_t * output);
  * zero-crossing rate (number of zero cross per sample)
  */
 smpl_t aubio_zero_crossing_rate(fvec_t * input);
+/**
+ * spectrum centroid computed on a cvec
+ */
+smpl_t aubio_spectral_centroid(cvec_t * input, smpl_t samplerate);
 /**
  * clean up cached memory at the end of program
  *