Changeset bd72039

Timestamp:

Mar 23, 2017, 3:19:19 PM (8 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, sampler

Children:

4077fa1

Parents:

5ab3c4e

Message:

python/lib/aubio/cmd.py: add tempo subcommand to extract overall bpm

File:

• python/lib/aubio/cmd.py (modified) (3 diffs)

python/lib/aubio/cmd.py

@@ -54 +54 @@
     # tempo subcommand
     subparser = subparsers.add_parser('beat',
+            help='get locations of beats')
+    parser_add_input(subparser)
+    parser_add_buf_hop_size(subparser, buf_size=1024, hop_size=512)
+    parser_add_time_format(subparser)
+    parser_add_verbose_help(subparser)
+    subparser.set_defaults(process=process_beat)
+
+    # tempo subcommand
+    subparser = subparsers.add_parser('tempo',
             help='get locations of beats')
     parser_add_input(subparser)
@@ -238 +247 @@
         sys.stdout.write(fmt_out + "%.6f\n" % res[0])
 
-class process_tempo(default_process):
+class process_beat(default_process):
     valid_opts = ['method', 'hop_size', 'buf_size', 'samplerate']
     def __init__(self, args):
         self.options = parse_options(args, self.valid_opts)
         self.tempo = aubio.tempo(**self.options)
-        super(process_tempo, self).__init__(args)
+        super(process_beat, self).__init__(args)
     def __call__(self, block):
         return self.tempo(block)
@@ -250 +259 @@
             outstr = self.time2string(self.tempo.get_last(), samplerate)
             sys.stdout.write(outstr + '\n')
+
+class process_tempo(process_beat):
+    def __init__(self, args):
+        super(process_tempo, self).__init__(args)
+        self.beat_locations = []
+    def repr_res(self, res, frames_read, samplerate):
+        if res[0] != 0:
+            self.beat_locations.append(self.tempo.get_last_s())
+    def flush(self, frames_read, samplerate):
+        import numpy as np
+        bpms = 60./ np.diff(self.beat_locations)
+        median_bpm = np.mean(bpms)
+        sys.stdout.write('%.2f bpm' % median_bpm + '\n')
 
 class process_notes(default_process):