Showing the changes to python (1 of 1 file). Show all
Unified · Split
| @@ -0,0 +1,78 @@ | |||
| 1 | #! /usr/bin/env python | ||
| 2 | |||
| 3 | """ A simple demo using aubio and pyaudio to play beats in real time | ||
| 4 | |||
| 5 | Note you will need to have pyaudio installed: `pip install pyaudio`. | ||
| 6 | |||
| 7 | Examples: | ||
| 8 | ./demo_tapthebeat.py ~/Music/track1.ogg | ||
| 9 | |||
| 10 | When compiled with ffmpeg/libav, you should be able to open remote streams. For | ||
| 11 | instance using youtube-dl (`pip install youtube-dl`): | ||
| 12 | |||
| 13 | ./demo_tapthebeat.py `youtube-dl -xg https://youtu.be/zZbM9n9j3_g` | ||
| 14 | |||
| 15 | """ | ||
| 16 | |||
| 17 | import sys | ||
| 18 | import time | ||
| 19 | import pyaudio | ||
| 20 | import aubio | ||
| 21 | import numpy as np | ||
| 22 | |||
| 23 | win_s = 1024 # fft size | ||
| 24 | hop_s = win_s // 2 # hop size | ||
| 25 | |||
| 26 | # parse command line arguments | ||
| 27 | if len(sys.argv) < 2: | ||
| 28 | print("Usage: %s <filename> [samplerate]" % sys.argv[0]) | ||
| 29 | sys.exit(1) | ||
| 30 | |||
| 31 | filename = sys.argv[1] | ||
| 32 | |||
| 33 | samplerate = 0 | ||
| 34 | if len( sys.argv ) > 2: samplerate = int(sys.argv[2]) | ||
| 35 | |||
| 36 | # create aubio source | ||
| 37 | a_source = aubio.source(filename, samplerate, hop_s) | ||
| 38 | samplerate = a_source.samplerate | ||
| 39 | |||
| 40 | # create aubio tempo detection | ||
| 41 | a_tempo = aubio.tempo("default", win_s, hop_s, samplerate) | ||
| 42 | |||
| 43 | # create a simple click sound | ||
| 44 | click = 0.7 * np.sin(2. * np.pi * np.arange(hop_s) / hop_s * samplerate / 3000.) | ||
| 45 | |||
| 46 | # pyaudio callback | ||
| 47 | def pyaudio_callback(in_data, frame_count, time_info, status): | ||
| 48 | samples, read = a_source() | ||
| 49 | is_beat = a_tempo(samples) | ||
| 50 | if is_beat: | ||
| 51 | samples += click | ||
| 52 | #print ('tick') # avoid print in audio callback | ||
| 53 | audiobuf = samples.tobytes() | ||
| 54 | if read < hop_s: | ||
| 55 | return (audiobuf, pyaudio.paComplete) | ||
| 56 | return (audiobuf, pyaudio.paContinue) | ||
| 57 | |||
| 58 | # create pyaudio stream with frames_per_buffer=hop_s and format=paFloat32 | ||
| 59 | p = pyaudio.PyAudio() | ||
| 60 | pyaudio_format = pyaudio.paFloat32 | ||
| 61 | frames_per_buffer = hop_s | ||
| 62 | n_channels = 1 | ||
| 63 | stream = p.open(format=pyaudio_format, channels=n_channels, rate=samplerate, | ||
| 64 | output=True, frames_per_buffer=frames_per_buffer, | ||
| 65 | stream_callback=pyaudio_callback) | ||
| 66 | |||
| 67 | # start pyaudio stream | ||
| 68 | stream.start_stream() | ||
| 69 | |||
| 70 | # wait for stream to finish | ||
| 71 | while stream.is_active(): | ||
| 72 | time.sleep(0.1) | ||
| 73 | |||
| 74 | # stop pyaudio stream | ||
| 75 | stream.stop_stream() | ||
| 76 | stream.close() | ||
| 77 | # close pyaudio | ||
| 78 | p.terminate() | ||