Changeset f36ecea for python/lib

Timestamp:

Jan 12, 2014, 5:54:42 AM (11 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:

4320679

Parents:

d945976

Message:

python/lib/aubio/slicing.py: use start and end stamps, make sure read > 0, improve tests

File:

• python/lib/aubio/slicing.py (modified) (4 diffs)

python/lib/aubio/slicing.py

@@ -1 +1 @@
 from aubio import source, sink
 import os
+
+max_timestamp = 1e120
 
 def slice_source_at_stamps(source_file, timestamps, timestamps_end = None,
@@ -10 +12 @@
         raise ValueError ("no timestamps given")
 
+    if timestamps[0] != 0:
+        timestamps = [0] + timestamps
+
     if timestamps_end != None and len(timestamps_end) != len(timestamps):
         raise ValueError ("len(timestamps_end) != len(timestamps)")
+    else:
+        timestamps_end = [t - 1 for t in timestamps[1:] ] + [ max_timestamp ]
 
     source_base_name, source_ext = os.path.splitext(os.path.basename(source_file))
@@ -19 +26 @@
         source_base_name = os.path.join(output_dir, source_base_name)
 
-    def new_sink_name(source_base_name, timestamp):
-        return source_base_name + '_%02.3f' % (timestamp) + '.wav'
+    def new_sink_name(source_base_name, timestamp, samplerate):
+        timestamp_seconds = timestamp / float(samplerate)
+        #print source_base_name + '_%02.3f' % (timestamp_seconds) + '.wav'
+        return source_base_name + '_%02.3f' % (timestamp_seconds) + '.wav'
 
     # reopen source file
     s = source(source_file, samplerate, hopsize)
     if samplerate == 0: samplerate = s.get_samplerate()
-    # create first sink at 0
-    g = sink(new_sink_name(source_base_name, 0.), samplerate)
     total_frames = 0
     # get next region
-    next_stamp = int(timestamps.pop(0))
+    start_stamp = int(timestamps.pop(0))
+    end_stamp = int(timestamps_end.pop(0))
+
+    # create first sink
+    new_sink_path = new_sink_name(source_base_name, start_stamp, samplerate)
+    #print "new slice", total_frames, "+", remaining, "=", end_stamp
+    g = sink(new_sink_path, samplerate)
 
     while True:
         # get hopsize new samples from source
         vec, read = s()
-        remaining = next_stamp - total_frames
+        # number of samples until end of region
+        remaining = end_stamp - total_frames
         # not enough frames remaining, time to split
         if remaining < read:
@@ -42 +56 @@
             # close this file
             del g
+            # get the next region
+            start_stamp = int(timestamps.pop(0))
+            end_stamp = int(timestamps_end.pop(0))
             # create a new file for the new region
-            new_sink_path = new_sink_name(source_base_name, next_stamp / float(samplerate))
-            #print "new slice", total_frames, "+", remaining, "=", next_stamp
+            new_sink_path = new_sink_name(source_base_name, start_stamp, samplerate)
+            #print "new slice", total_frames, "+", remaining, "=", end_stamp
             g = sink(new_sink_path, samplerate)
             # write the remaining samples in the new file
             g(vec[remaining:read], read - remaining)
-            if len(timestamps):
-                next_stamp = int(timestamps.pop(0))
-            else:
-                next_stamp = 1e120
-        else:
+        elif read > 0:
+            # write all the samples
             g(vec[0:read], read)
         total_frames += read