bpm-finder/null.py at master · yoavrozner/bpm-finder · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
import wave
import random
import struct
import sys
import pyaudio
import matplotlib.pyplot as plt
import numpy as np
from pydub import AudioSegment

SAMPLE_LEN = 230000
CHUNK = 1024


def hear_the_wave():
    sys.argv.append('noise.wav')
    if len(sys.argv) < 2:
        print("Plays a wave file.\n\nUsage: %s filename.wav" % sys.argv[0])
        sys.exit(-1)

    wf = wave.open(sys.argv[1], 'rb')

    # instantiate PyAudio (1)
    p = pyaudio.PyAudio()

    # open stream (2)
    stream = p.open(format=p.get_format_from_width(wf.getsampwidth()),
                    channels=wf.getnchannels(),
                    rate=wf.getframerate(),
                    output=True)

    # read data
    data = wf.readframes(CHUNK)

    # play stream (3)
    while len(data) > 0:
        stream.write(data)
        data = wf.readframes(CHUNK)

    # stop stream (4)
    stream.stop_stream()
    stream.close()

    # close PyAudio (5)
    p.terminate()


def slice_wav_file():
    t1 = 30
    t2 = 35
    t1 *= 1000  # Works in milliseconds
    t2 *= 1000
    new_audio = AudioSegment.from_wav("try.wav")
    new_audio = new_audio[t1:t2]
    new_audio.export('newSong.wav', format="wav")  # Exports to a wav file in the current path.


def wav_to_floats(wave_file="newSong.wav"):
    w = wave.open(wave_file)
    astr = w.readframes(w.getnframes())
    # convert binary chunks to short
    a = struct.unpack("%ih" % (w.getnframes()*w.getnchannels()), astr)
    a = [float(val) / pow(2, 15) for val in a]
    return a


def print_float():
    signal = wav_to_floats()
    print "read " + str(len(signal)) + " frames"
    print "in the range " + str(min(signal)) + " to " + str(min(signal))


def find_max(signal):
    for j in range(9):
        i = (j - 1) * len(signal) / 9
        max_num = 0
        max_num_i = 0
        while i < int(j * len(signal) / 9):
            if signal[i] > max_num:
                max_num = signal[i]
                max_num_i = i
            i += 1
        print max_num, max_num_i


def main():
    slice_wav_file()
    print_float()
    spf = wave.open('newSong.wav', 'r')

    #Extract Raw Audio from Wav File
    signal = spf.readframes(-1)
    signal = np.fromstring(signal, 'Int16')

    find_max(signal)
    # plt.figure(1)
    # plt.title('Signal Wave...')
    # plt.plot(signal)
    # plt.show()


    # arr = []
    # arr_i = [0]
    # print signal
    # max_num = 0
    # max_num_i = 0
    # i = 0
    # while i < len(signal):
    #     if signal[i] > max_num:
    #         max_num = signal[i]
    #         max_num_i = i
    #     i += 1
    # print max_num, max_num_i
    # i = 0
    # while i < len(signal):
    #     if not signal[i] + 3000 >= max_num or arr_i[len(arr_i) - 1] + 7000 >= i:
    #         signal[i] = 0
    #     else:
    #         arr.append(signal[i])
    #         arr_i.append(i)
    #     i += 1
    # print arr
    # print "BPM: " + str(len(arr) * 12)
    #If Stereo
    # if spf.getnchannels() == 2:
    #     print 'Just mono files'
    #     sys.exit(0)
    plt.figure(1)
    plt.title('Signal Wave...')
    plt.plot(signal)
    plt.show()


if __name__ == '__main__':
    main()