Files
amodem/scripts/plot.py
2015-03-27 09:57:13 +03:00

32 lines
661 B
Python
Executable File

#!/usr/bin/env python
import pylab
import numpy as np
from amodem import common
from amodem.config import Configuration
import sys
def spectrogram(t, x, Fs, NFFT=256):
ax1 = pylab.subplot(211)
pylab.plot(t, x)
pylab.subplot(212, sharex=ax1)
pylab.specgram(x, NFFT=NFFT, Fs=Fs, noverlap=NFFT/2,
cmap=pylab.cm.gist_heat)
def main():
config = Configuration()
for fname in sys.argv[1:]:
x = common.load(open(fname, 'rb'))
t = np.arange(len(x)) * config.Ts
pylab.figure()
pylab.title(fname)
spectrogram(t, x, config.Fs)
pylab.show()
if __name__ == '__main__':
main()