Changeset 9c8c8a6

Timestamp:

Jul 10, 2015, 2:34:48 AM (9 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/autosink, feature/cnn, feature/cnn_org, feature/constantq, feature/crepe, feature/crepe_org, feature/pitchshift, feature/pydocstrings, feature/timestretch, fix/ffmpeg5, master, pitchshift, sampler, timestretch, yinfft+

Children:

0e362b5

Parents:

31a09d2

Message:

ext/py-musicutils.c: add level_detection (closes #21)

Location:

python

Files:

• ext/aubiomodule.c (modified) (1 diff)
• ext/py-musicutils.c (modified) (1 diff)
• ext/py-musicutils.h (modified) (1 diff)
• tests/test_musicutils.py (modified) (2 diffs)

python/ext/aubiomodule.c

@@ -244 +244 @@
   {"db_spl", Py_aubio_db_spl, METH_VARARGS, Py_aubio_db_spl_doc},
   {"silence_detection", Py_aubio_silence_detection, METH_VARARGS, Py_aubio_silence_detection_doc},
+  {"level_detection", Py_aubio_level_detection, METH_VARARGS, Py_aubio_level_detection_doc},
   {"window", Py_aubio_window, METH_VARARGS, Py_aubio_window_doc},
   {NULL, NULL} /* Sentinel */

python/ext/py-musicutils.c

@@ -112 +112 @@
   return silence_detection;
 }
+
+PyObject *
+Py_aubio_level_detection(PyObject *self, PyObject *args)
+{
+  PyObject *input;
+  fvec_t *vec;
+  PyObject *level_detection;
+  smpl_t threshold;
+
+  if (!PyArg_ParseTuple (args, "Of:level_detection", &input, &threshold)) {
+    PyErr_SetString (PyExc_ValueError, "failed parsing arguments");
+    return NULL;
+  }
+
+  if (input == NULL) {
+    return NULL;
+  }
+
+  vec = PyAubio_ArrayToCFvec (input);
+  if (vec == NULL) {
+    return NULL;
+  }
+
+  level_detection = Py_BuildValue("f", aubio_level_detection(vec, threshold));
+  if (level_detection == NULL) {
+    PyErr_SetString (PyExc_ValueError, "failed computing level_detection");
+    return NULL;
+  }
+
+  return level_detection;
+}

python/ext/py-musicutils.h

@@ -60 +60 @@
 PyObject * Py_aubio_silence_detection(PyObject *self, PyObject *args);
 
+static char Py_aubio_level_detection_doc[] = ""
+"Get buffer level in dB SPL if over a given threshold, 1. otherwise.\n"
+"\n"
+"Example\n"
+"-------\n"
+"\n"
+">>> import numpy\n"""
+">>> level_detection(0.7*numpy.ones(1024, dtype=\"float32\"), -80)\n"
+"0";
+
+PyObject * Py_aubio_level_detection(PyObject *self, PyObject *args);
+
 #endif /* _PY_AUBIO_MUSICUTILS_H_ */

python/tests/test_musicutils.py

@@ -6 +6 @@
 from math import pi
 
-from aubio import window, level_lin, db_spl, silence_detection
+from aubio import window, level_lin, db_spl, silence_detection, level_detection
 
 from aubio import fvec
@@ -96 +96 @@
         assert silence_detection(ones(1024, dtype="float32"), -70) == 0
 
+class aubio_level_detection(TestCase):
+    def test_accept_fvec(self):
+        level_detection(fvec(1024), -70.)
+
+    def test_fail_not_fvec(self):
+        try:
+            level_detection("default", -70)
+        except ValueError, e:
+            pass
+        else:
+            self.fail('non-number input phase does not raise a TypeError')
+
+    def test_zeros_is_one(self):
+        from math import isinf
+        assert level_detection(fvec(1024), -70) == 1
+
+    def test_minus_ones_is_zero(self):
+        from numpy import ones
+        assert level_detection(ones(1024, dtype="float32"), -70) == 0
+
 if __name__ == '__main__':
     from unittest import main