Cirrus Logic - RPI 2 FFT on streaming audio

import numpy as np

import wave

import struct

import os

from struct import unpack

import time

import RPi.GPIO as GPIO

GPIO.setmode(GPIO.BCM)

GPIO.setwarnings(False)

import smbus

import time

bus = smbus.SMBus(1)

address = 0x40

mode = 0x00

def sendbyte(txbyte):

    CLK = 5

    OUT = 6

    NEXT = 25

    GPIO.setup(CLK,GPIO.OUT)

    GPIO.setup(OUT,GPIO.OUT)

    GPIO.setup(NEXT,GPIO.IN)

    number = 1

    for i in range (1,9):                   #for bits 1 to 8, send bit values

        if txbyte &number != 0:             #if txbyte bit = 1, data line = 1

            GPIO.output(OUT, 1)

            #print "1 %s" %(number)

        else:                               #else data line = 0

            GPIO.output(OUT, 0)

            #print "0 %s" %(number)

        while GPIO.input(NEXT) == 0:        #wait for pic to say its ready for data

            time.sleep(0.0000005)                #sleep for 0.5 microsecond

        GPIO.output(CLK, 1)                #raise the clk edge indicating data is valid

        while GPIO.input(NEXT) == 1:        #wait for pic to say it is not ready

            time.sleep(0.0000005)                #sleep for 0.5 usec

        GPIO.output(CLK, 0)

        number = number*2                   #change to the next bit

    GPIO.output(OUT, 0)                     #lower the data line as byte tx has ended

os.system("bash /home/pi/Desktop/Code/Reset_paths.sh")

#os.system("bash /home/pi/Desktop/Code/Record_from_DMIC.sh")

os.system("bash /home/pi/Desktop/Code/Record_from_lineIn.sh")

audiolr = []

index = 0

#frequency bandwidths for display                               

bw = [7,9,13,19,27,37,51,69,99,139,189,249,319,419,519,699,899,1099,1399,1899,2499,3199,4099,3200]

#number of values in each band - divisible factor

band = [7,3,3,6,6,11,14,20,26,37,52,70,193,119,156,193,261,334,409,520,705,929,1186,1528]

#frequency center of bandwidths

freq = [10,24,32,45,62,85,118,163,224,310,430,595,815,1100,1470,1940,2550,3350,4350,5600,7250,9500,12300,15950]

subindex = 0

for i in range(0,16384):

    audiolr.append(0)

freqoutput = []

for i in range(0,24):

    freqoutput.append(0)

while True:

    startt = time.time()

    os.system("aplay -vv /home/pi/Desktop/Code/Song.wav")

    os.system("arecord -f cd -c2 -D hw:sndrpiwsp -d 1 Song.wav")

    wavfile = wave.open("/home/pi/Desktop/Code/Song.wav",'r')

    paramet = wavfile.getparams()

    wavfile.setpos(27100)

    wavedata = wavfile.readframes(16384)

    audio = struct.unpack("<32768h",wavedata)

    i = 0

    index = 0

    for i in range(0,32768,2):

        audiolr[index] = audio[i] +audio[i+1]

        index += 1

    complexfreq = np.fft.fft(audiolr[0:16384])

    print len(audiolr)

    f = [0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0]

    start = 1

    stop = 8

    for index in range(0,24):

        for subindex in range(start,stop):

            f[index] = f[index] + abs(complexfreq[subindex])

        f[index] = (f[index]/32768)/band[index]

        #print "%s, Fc=%sHz,    BW-%sHz,     power %s" %(index+1, freq[index], bw[index], f[index])

        start = stop+1

        stop = stop + band[index]

    for index in range(0,24):

        if f[index]>32:

            freqoutput[index]=255

        elif f[index]>16:

            freqoutput[index]=127

        elif f[index]>8:

            freqoutput[index]=63

        elif f[index]>4:

            freqoutput[index]=31

        elif f[index]>2:

            freqoutput[index]=15

        elif f[index]>1:

            freqoutput[index]=7

        elif f[index]>0.5:

            freqoutput[index]=3

        elif f[index]>0.25:

            freqoutput[index]=1

        else:

            freqoutput[index]=0

        sendbyte(freqoutput[index])

    for index in range(0,24):

        print "%s, Fc=%sHz has a power of %s, Equalizer Level = %s     BW=%sHz" %(index+1, freq[index], f[index],freqoutput[index], bw[index])

    print (time.time()-startt)