def compareFrequencies(): times = generateTimes(sampleFreq, numSamples) signal = (80.0, 0.1) coherent = (60.0, 1.0) incoherent = (60.1, 1.0) highFNoise = (500.0, 0.01) timeData = generateTimeDomain(times, [signal, coherent, highFNoise]) timeData2 = generateTimeDomain(times, [signal, incoherent, highFNoise]) #timeData3 = generateTimeDomain(times, [signal, highFNoise]) #timeData = generateTimeDomain(times, [(60.0, 1.0)]) #timeData2 = generateTimeDomain(times, [(61.0, 1.0)]) roi = (0, 20) freqData = map(toDb, map(dtype, map(absolute, fourier(timeData))))[roi[0]:roi[1]] freqData2 = map(toDb, map(dtype, map(absolute, fourier(timeData2))))[roi[0]:roi[1]] #freqData3 = map(toDb, map(dtype, map(absolute, fourier(timeData3))))[roi[0]:roi[1]] frequencies = generateFFTFrequencies(sampleFreq, numSamples)[roi[0]:roi[1]] #pylab.subplot(111) pylab.plot(frequencies, freqData) #pylab.subplot(112) pylab.plot(frequencies, freqData2) #pylab.plot(frequencies, freqData3) pylab.grid(True) pylab.show()
def test1(): times = generateTimes(sampleFreq, numSamples) print times[-1] timeData = generateTimeDomain(times, [(60.0, 1.0)]) compression = 0.05 plot(times, timeData) freqData = map(dtype, map(absolute, fourier(timeData, compression = compression))) #freqData = map(toDb, freqData) frequencies = generateFFTFrequencies(sampleFreq, numSamples, compression = compression) print 'bin size: %s' % frequencies[1] #plot(frequencies, freqData, False) plot(frequencies, freqData[:len(frequencies)], False) open('frequencies.txt', 'w').write('\n'.join([str(x) + '\t' + str(y) for x, y in zip(frequencies, freqData[:len(frequencies)])]))