Changeset cfa46b9

Timestamp:

Apr 25, 2016, 6:05:02 PM (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:

07867cd

Parents:

6e765ed

Message:

python/ext/py-source.c: added duration

Location:

python

Files:

• ext/py-source.c (modified) (3 diffs)
• tests/test_source.py (modified) (1 diff)

python/ext/py-source.c

@@ -9 +9 @@
   uint_t channels;
   uint_t hop_size;
+  uint_t duration;
   fvec_t *read_to;
   fmat_t *mread_to;
@@ -147 +148 @@
     self->channels = aubio_source_get_channels ( self->o );
   }
+  self->duration = aubio_source_get_duration ( self->o );
 
   self->read_to = new_fvec(self->hop_size);
@@ -227 +229 @@
   {"hop_size", T_INT, offsetof (Py_source, hop_size), READONLY,
     "number of consecutive frames that will be read at each do or do_multi call"},
+  {"duration", T_INT, offsetof (Py_source, duration), READONLY,
+    "total number of frames in the source (estimated)"},
   { NULL } // sentinel
 };

python/tests/test_source.py

@@ -100 +100 @@
             assert a == b + c
 
+    def test_duration(self):
+        for p in list_of_sounds:
+            total_frames = 0
+            f = source(p)
+            duration = f.duration
+            while True:
+                vec, read = f()
+                total_frames += read
+                if read < f.hop_size: break
+            self.assertEqual(duration, total_frames)
+
 class aubio_source_readmulti_test_case(aubio_source_read_test_case):