Changeset 6514bb6 for python

Timestamp:

Mar 3, 2013, 10:27:17 PM (12 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:

7a6521d

Parents:

6a50b9e

Message:

aubiomodule.c: add unwrap2pi

Location:

python

Files:

• aubiomodule.c (modified) (2 diffs)
• tests/test_mathutils.py (modified) (2 diffs)

python/aubiomodule.c

@@ -38 +38 @@
 
   return result;
+}
+
+static char Py_unwrap2pi_doc[] = "unwrap phase value to [-pi, pi]";
+
+static PyObject *
+Py_unwrap2pi (PyObject * self, PyObject * args)
+{
+  smpl_t input;
+  smpl_t output;
+
+  if (!PyArg_ParseTuple (args, "|f", &input)) {
+    return NULL;
+  }
+
+  output = aubio_unwrap2pi (input);
+
+  return (PyObject *)PyFloat_FromDouble (output);
 }
 
@@ -209 +226 @@
 
 static PyMethodDef aubio_methods[] = {
+  {"unwrap2pi", Py_unwrap2pi, METH_VARARGS, Py_unwrap2pi_doc},
   {"bintomidi", Py_bintomidi, METH_VARARGS, Py_bintomidi_doc},
   {"miditobin", Py_miditobin, METH_VARARGS, Py_miditobin_doc},

python/tests/test_mathutils.py

@@ -4 +4 @@
 from numpy.testing import assert_equal, assert_almost_equal
 from aubio import bintomidi, miditobin, freqtobin, bintofreq, freqtomidi, miditofreq
+from aubio import unwrap2pi
 
 class aubio_mathutils(TestCase):
+
+    def test_unwrap2pi(self):
+       a = [ x/100. for x in range(-600,600,100) ]
+       b = [ unwrap2pi(x) for x in a ]
+       #print b
 
     def test_miditobin(self):
@@ -25 +31 @@
        freqs = [ miditofreq(a) for a in range(128) ]
        midis = [ freqtomidi(a) for a in freqs ]
-       print midis
 
 if __name__ == '__main__':