wavGen.py

import numpy as np
from scipy.io.wavfile import write
import sys
import math

def wavGen(name,func,length=60):
    rate = 96000
    if (name != ""):
        if (func == "sine"):
            frequency = 440
            # length = 60
            # rate = 44100
            data = 10*np.sin(np.arange(int(length * rate)) * float(frequency) * (math.pi * 2) / rate) # sine # np.random.uniform(-1,1,44100) # 44100 random samples between -1 and 1
        else:
            data = np.random.uniform(-1,1,int(length * rate)) # 44100 random samples between -1 and 1
    else:
    	print "Please provide a filename and a function name (sine or rand)"
    scaled = np.int16(data/np.max(np.abs(data)) * 32767)
    write(name, rate, scaled)