PytchControl/pytchcontrol.py at master · WyattWismer/PytchControl · 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
import pyaudio
import wave
import sys
import struct
import math
import numpy as np
import bat

#--------CONSTANTS----------
TOLERANCE = 50
THRESH = 0.1
N = 10


#---------COMMANDS----------

# n = neutral, u = up, d = down
commands = {'uu':'web.bat','dd':'notepad.bat','nn':'','nu':'','nd':'','un':'','ud':'','dn':'','du':''}


def rms( data ):
    count = len(data)/2
    format = "%dh"%(count)
    shorts = struct.unpack( format, data )
    sum_squares = 0.0
    for sample in shorts:
        n = sample * (1.0/32768)
        sum_squares += n*n
    return math.sqrt( sum_squares / count )

def listen():
        CHUNK = 2048
        FORMAT = pyaudio.paInt16
        CHANNELS = 2
        RATE = 44100
        RECORD_SECONDS = 100

        window = np.blackman(CHUNK)
        swidth = 2

        past = [-1]*N
        command = []
        active = False

        p = pyaudio.PyAudio()

        stream = p.open(format=FORMAT,
                channels=CHANNELS,
                rate=RATE,
                input=True,
                frames_per_buffer=CHUNK)

        print "* listening for command"


        for i in range(0, int(RATE / CHUNK * RECORD_SECONDS)):
                data = stream.read(CHUNK/2)
                indata = np.array(wave.struct.unpack("%dh"%(len(data)/swidth),\
                                         data))*window

                # Take fft and square each value
                fftData=abs(np.fft.rfft(indata))**2
                # find the max
                which = fftData[1:].argmax() + 1

                if which != len(fftData)-1:
                    y0,y1,y2 = np.log(fftData[which-1:which+2:])
                    x1 = (y2 - y0) * .5 / (2 * y1 - y2 - y0)
                    # find the frequency and output it
                    thefreq = (which+x1)*RATE/CHUNK
                else:
                    thefreq = which*RATE/CHUNK

                #update past
                past.pop(0)
                past.append(thefreq)
                #check for part of command
                if rms(data)>THRESH:
                    if not active and max(past)-min(past)<TOLERANCE:
                        command.append(np.mean(past))
                        active = True
                        print command
                else:
                    active = False
                #check for command completion
                if len(command)==3:
                    #convert command to string
                    comstr = ''
                    for i in range(len(command)-1):
                        t = command[i+1]-command[i]
                        if abs(t)<TOLERANCE:
                            comstr += 'n'
                        elif t > 0:
                            comstr += 'u'
                        else:
                            comstr += 'd'
                    if comstr in commands and not commands[comstr]=='':
                        bat.launch(commands[comstr])
                        break

                    command = []

        print "* done"

        stream.stop_stream()
        stream.close()
        p.terminate()


listen()