tone_est.py 文件源码

def find_min_phase(sdata,a,f,sr,phase):
    rms1 = 0
    rms2 = 0
    rms3 = 0
    samples = len(sdata)
    for i in xrange(samples):
        diff1 = (sdata[i] - a*cos(2*pi*i*f/sr + phase[0]))
        rms1 += diff1*diff1
        diff2 = (sdata[i] - a*cos(2*pi*i*f/sr + phase[1]))
        rms2 += diff2*diff2
        diff3 = (sdata[i] - a*cos(2*pi*i*f/sr + phase[2]))
        rms3 += diff3*diff3
    rms = numpy.zeros(3)
    rms[0] = rms1
    rms[1] = rms2
    rms[2] = rms3
    i = numpy.argmin(rms)
    p = phase[i]
    return i,p