FreeDV
/
phasenn
mirror of https://github.com/drowe67/phasenn.git
1
0
Fork 0
Code Issues Releases Wiki Activity

added better visualisation of phase and amplitude together in PNGs

master
David 2019-12-06 12:29:12 +10:30
parent f0ee4a7684
commit 6c8a2052ff
1 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
plot_n0.py
View File

@ -35,6 +35,7 @@ import matplotlib.pyplot as plt
from scipy import signal from scipy import signal
import codec2_model import codec2_model
import argparse import argparse
import os
# constants # constants
@ -46,6 +47,7 @@ parser = argparse.ArgumentParser(description='Plot phase spectra and synthesised
parser.add_argument('modelfile', help='Codec 2 model file') parser.add_argument('modelfile', help='Codec 2 model file')
parser.add_argument('--n0file', help='text file of n0 estimates') parser.add_argument('--n0file', help='text file of n0 estimates')
parser.add_argument('--start', type=int, default=30, help=' start frame') parser.add_argument('--start', type=int, default=30, help=' start frame')
parser.add_argument('--png', action='store_true')
args = parser.parse_args() args = parser.parse_args()
# read in model file records # read in model file records
@ -144,3 +146,22 @@ plt.show(block=False)
print("Click on last figure to finish....") print("Click on last figure to finish....")
plt.waitforbuttonpress(0) plt.waitforbuttonpress(0)
plt.close() plt.close()
# optionally generate some PNGs to flip though
if args.png:
png_basename = os.path.basename(args.modelfile)
png_basename = os.path.splitext(png_basename)[0]
for r in range(nb_plots):
fig, ax1 = plt.subplots()
f = frame[r];
freq = range(1,L[f])*Wo[f]*Fs/(2*np.pi)
ax1.plot(freq, amp[f,1:L[f]],'g')
ax1.set_ylabel('amp', color='g')
ax2 = ax1.twinx()
ax2.plot(freq,phase[f,1:L[f]]*180/np.pi, 'r')
ax2.set_ylabel('phase', color='r')
png_name = "%s_%02d.png" % (png_basename,f)
print("Writing %s" % (png_name))
plt.savefig(png_name)