update for subplot support
This commit is contained in:
parent
581a05201e
commit
db9b4842d2
3 changed files with 91 additions and 32 deletions
|
@ -10,10 +10,11 @@ import matplotlib
|
||||||
from matplotlib import rcParams, pyplot as pp
|
from matplotlib import rcParams, pyplot as pp
|
||||||
from cycler import cycler
|
from cycler import cycler
|
||||||
|
|
||||||
rcParams['grid.alpha'] = 0.7
|
rcParams['grid.alpha'] = 0.5
|
||||||
rcParams['grid.linestyle'] = ':'
|
rcParams['grid.linestyle'] = ':'
|
||||||
rcParams['font.family'] = ['serif']
|
rcParams['font.family'] = ['serif']
|
||||||
rcParams['font.size'] = 8.0
|
rcParams['font.size'] = 6.5
|
||||||
|
rcParams['mathtext.fontset'] = 'dejavuserif'
|
||||||
rcParams['mathtext.fontset'] = 'dejavuserif'
|
rcParams['mathtext.fontset'] = 'dejavuserif'
|
||||||
rcParams['mathtext.it'] = 'serif:italic'
|
rcParams['mathtext.it'] = 'serif:italic'
|
||||||
rcParams['mathtext.bf'] = 'serif:bold'
|
rcParams['mathtext.bf'] = 'serif:bold'
|
||||||
|
|
18
runAll.sh
18
runAll.sh
|
@ -3,14 +3,24 @@
|
||||||
MAX_JOBS=4
|
MAX_JOBS=4
|
||||||
|
|
||||||
echo "Starting:"
|
echo "Starting:"
|
||||||
for n in 1 2 3 4 5 6; do
|
for n in 5 6; do
|
||||||
while [[ $MAX_JOBS -le $(jobs -l | wc -l) ]]; do sleep 0.1; done
|
while [[ $MAX_JOBS -le $(jobs -l | wc -l) ]]; do sleep 0.1; done
|
||||||
(
|
(
|
||||||
echo " start $n →"
|
echo " start $n →"
|
||||||
./tankPlot.py -n $n -sq &>/dev/null
|
./tankPlot.py -n $n -sq 1>/dev/null
|
||||||
./tankPlot.py -n $n -rq &>/dev/null
|
./tankPlot.py -n $n -rq 1>/dev/null
|
||||||
echo " 止 end $n"
|
echo " 止 end $n"
|
||||||
) &
|
) &
|
||||||
done
|
done
|
||||||
while [[ $(jobs -l | wc -l) -gt 1 ]]; do sleep 0.1; done
|
for n in 1 2 3 4 11 12 13 14; do
|
||||||
|
while [[ $MAX_JOBS -le $(jobs -l | wc -l) ]]; do sleep 0.1; done
|
||||||
|
(
|
||||||
|
echo " start $n →"
|
||||||
|
./tankPlot.py -n $n --subplot -sq 1>/dev/null
|
||||||
|
./tankPlot.py -n $n --subplot -rq 1>/dev/null
|
||||||
|
echo " 止 end $n"
|
||||||
|
) &
|
||||||
|
done
|
||||||
|
|
||||||
|
while [[ $(jobs -lr | wc -l) -gt 0 ]]; do sleep 0.1; done
|
||||||
echo "DONE!"
|
echo "DONE!"
|
||||||
|
|
100
tankPlot.py
100
tankPlot.py
|
@ -16,6 +16,8 @@ args_parser.add_argument('--raster','-r', action='store_true',
|
||||||
help='save as raster')
|
help='save as raster')
|
||||||
args_parser.add_argument('--debug','-d', action='store_true',
|
args_parser.add_argument('--debug','-d', action='store_true',
|
||||||
help='hold for debugging')
|
help='hold for debugging')
|
||||||
|
args_parser.add_argument('--subplot', action='store_true',
|
||||||
|
help='use subplots when available')
|
||||||
args_parser.add_argument('--headless','-q', action='store_true',
|
args_parser.add_argument('--headless','-q', action='store_true',
|
||||||
help='Remain neadless even if we aren\'t saving files.')
|
help='Remain neadless even if we aren\'t saving files.')
|
||||||
args = args_parser.parse_args()
|
args = args_parser.parse_args()
|
||||||
|
@ -50,7 +52,10 @@ else:
|
||||||
################################################################################
|
################################################################################
|
||||||
# Override the defaults for this script
|
# Override the defaults for this script
|
||||||
figScaleSize = 1.0 if args.save else 1.6
|
figScaleSize = 1.0 if args.save else 1.6
|
||||||
rcParams['figure.figsize'] = [3.4*figScaleSize,2.25*figScaleSize]
|
if args.subplot:
|
||||||
|
rcParams['figure.figsize'] = [3.4*figScaleSize,4*figScaleSize]
|
||||||
|
else:
|
||||||
|
rcParams['figure.figsize'] = [3.4*figScaleSize,2.25*figScaleSize]
|
||||||
default_window_position=['+20+80', '+120+80']
|
default_window_position=['+20+80', '+120+80']
|
||||||
|
|
||||||
################################################################################
|
################################################################################
|
||||||
|
@ -69,7 +74,7 @@ if plot_list[0] in [11, 12, 13, 14]:
|
||||||
else:
|
else:
|
||||||
gain_variation = 0 # dB
|
gain_variation = 0 # dB
|
||||||
|
|
||||||
if plot_list[0] in [4, 5]:
|
if plot_list[0] in [14, 4, 5]:
|
||||||
S.bw_plt = 0.5
|
S.bw_plt = 0.5
|
||||||
B.bw_plt = S.bw_plt
|
B.bw_plt = S.bw_plt
|
||||||
freq_pts = 51
|
freq_pts = 51
|
||||||
|
@ -162,8 +167,13 @@ if 6 in plot_list:
|
||||||
|
|
||||||
################################################################################
|
################################################################################
|
||||||
if 1 in plot_list or 11 in plot_list:
|
if 1 in plot_list or 11 in plot_list:
|
||||||
h1 = [pp.figure() for x in range(2)]
|
if not args.subplot:
|
||||||
ax1 = [hT.add_subplot(1,1,1) for hT in h1]
|
h1 = [pp.figure() for x in range(2)]
|
||||||
|
ax1 = [hT.add_subplot(1,1,1) for hT in h1]
|
||||||
|
else:
|
||||||
|
h1 = [pp.figure() for x in range(1)]
|
||||||
|
ax1 = h1[0].subplots(2,1)
|
||||||
|
|
||||||
ax1[0].plot(f.hz,dB20(tf))
|
ax1[0].plot(f.hz,dB20(tf))
|
||||||
ax1[1].plot(f.hz,ang_unwrap(tf))
|
ax1[1].plot(f.hz,ang_unwrap(tf))
|
||||||
|
|
||||||
|
@ -185,11 +195,23 @@ if 1 in plot_list or 11 in plot_list:
|
||||||
|
|
||||||
if args.save:
|
if args.save:
|
||||||
if 11 in plot_list:
|
if 11 in plot_list:
|
||||||
h1[0].savefig('%s/%s.%s' % (figdir, 'NA-11.0', fig_ext))
|
if args.subplot:
|
||||||
h1[1].savefig('%s/%s.%s' % (figdir, 'NA-11.1', fig_ext))
|
h1[0].savefig('%s/%s.%s' % (figdir,
|
||||||
|
'01d-ideal-AbsGainPhase-wgv', fig_ext))
|
||||||
|
else:
|
||||||
|
h1[0].savefig('%s/%s.%s' % (figdir,
|
||||||
|
'010-AbsGain-wgv', fig_ext))
|
||||||
|
h1[1].savefig('%s/%s.%s' % (figdir,
|
||||||
|
'011-AbsPhase-wgv', fig_ext))
|
||||||
else:
|
else:
|
||||||
h1[0].savefig('%s/%s.%s' % (figdir, 'NA-01.0', fig_ext))
|
if args.subplot:
|
||||||
h1[1].savefig('%s/%s.%s' % (figdir, 'NA-01.1', fig_ext))
|
h1[0].savefig('%s/%s.%s' % (figdir,
|
||||||
|
'01d-ideal-AbsGainPhase', fig_ext))
|
||||||
|
else:
|
||||||
|
h1[0].savefig('%s/%s.%s' % (figdir,
|
||||||
|
'010-AbsGain', fig_ext))
|
||||||
|
h1[1].savefig('%s/%s.%s' % (figdir,
|
||||||
|
'011-AbsPhase', fig_ext))
|
||||||
if HEADLESS:
|
if HEADLESS:
|
||||||
pp.close()
|
pp.close()
|
||||||
else:
|
else:
|
||||||
|
@ -220,14 +242,14 @@ if 4 in plot_list or 14 in plot_list:
|
||||||
if args.save:
|
if args.save:
|
||||||
if 14 in plot_list:
|
if 14 in plot_list:
|
||||||
h4[0].savefig('%s/%s.%s' % (figdir,
|
h4[0].savefig('%s/%s.%s' % (figdir,
|
||||||
'ideal-smith_tank_impedance_wgv', fig_ext))
|
'040-ideal-smith_tank_impedance-wgv', fig_ext))
|
||||||
h4[1].savefig('%s/%s.%s' % (figdir,
|
h4[1].savefig('%s/%s.%s' % (figdir,
|
||||||
'ideal-polar_gain_plot_wgv', fig_ext))
|
'041-ideal-polar_gain_plot-wgv', fig_ext))
|
||||||
else:
|
else:
|
||||||
h4[0].savefig('%s/%s.%s' % (figdir,
|
h4[0].savefig('%s/%s.%s' % (figdir,
|
||||||
'ideal-smith_tank_impedance', fig_ext))
|
'040-ideal-smith_tank_impedance', fig_ext))
|
||||||
h4[1].savefig('%s/%s.%s' % (figdir,
|
h4[1].savefig('%s/%s.%s' % (figdir,
|
||||||
'ideal-polar_gain_plot', fig_ext))
|
'041-ideal-polar_gain_plot', fig_ext))
|
||||||
if HEADLESS:
|
if HEADLESS:
|
||||||
pp.close()
|
pp.close()
|
||||||
else:
|
else:
|
||||||
|
@ -252,9 +274,9 @@ if 5 in plot_list:
|
||||||
#[hT.tight_layout() for hT in h5]
|
#[hT.tight_layout() for hT in h5]
|
||||||
if args.save:
|
if args.save:
|
||||||
h5[0].savefig('%s/%s.%s' % (figdir,
|
h5[0].savefig('%s/%s.%s' % (figdir,
|
||||||
'ideal-flat_g1-smith_tank_impedance', fig_ext))
|
'050-ideal-flat_g1-smith_tank_impedance', fig_ext))
|
||||||
h5[1].savefig('%s/%s.%s' % (figdir,
|
h5[1].savefig('%s/%s.%s' % (figdir,
|
||||||
'ideal-flat_g1-polar_gain_plot', fig_ext))
|
'050-ideal-flat_g1-polar_gain_plot', fig_ext))
|
||||||
if HEADLESS:
|
if HEADLESS:
|
||||||
pp.close()
|
pp.close()
|
||||||
else:
|
else:
|
||||||
|
@ -263,9 +285,13 @@ if 5 in plot_list:
|
||||||
|
|
||||||
################################################################################
|
################################################################################
|
||||||
if 2 in plot_list or 12 in plot_list:
|
if 2 in plot_list or 12 in plot_list:
|
||||||
h2 = [pp.figure() for x in range(2)]
|
if not args.subplot:
|
||||||
|
h2 = [pp.figure() for x in range(2)]
|
||||||
|
ax2 = [hT.add_subplot(1,1,1) for hT in h2]
|
||||||
|
else:
|
||||||
|
h2 = [pp.figure() for x in range(1)]
|
||||||
|
ax2 = h2[0].subplots(2,1)
|
||||||
|
|
||||||
ax2 = [hT.add_subplot(1,1,1) for hT in h2]
|
|
||||||
ax2[0].plot(f.hz,dB20(tf_r))
|
ax2[0].plot(f.hz,dB20(tf_r))
|
||||||
setLimitsTicks(ax2[0], dB20(tf_r), 6)
|
setLimitsTicks(ax2[0], dB20(tf_r), 6)
|
||||||
ax2[1].plot(f.hz,ang_unwrap(tf_r.T).T)
|
ax2[1].plot(f.hz,ang_unwrap(tf_r.T).T)
|
||||||
|
@ -288,11 +314,23 @@ if 2 in plot_list or 12 in plot_list:
|
||||||
|
|
||||||
if args.save:
|
if args.save:
|
||||||
if 12 in plot_list:
|
if 12 in plot_list:
|
||||||
h2[0].savefig('%s/%s.%s' % (figdir, 'NA-12.0', fig_ext))
|
if not args.subplot:
|
||||||
h2[1].savefig('%s/%s.%s' % (figdir, 'NA-12.1', fig_ext))
|
h2[0].savefig('%s/%s.%s' % (figdir,
|
||||||
|
'020-TF_RelativeGainPhase-wgv', fig_ext))
|
||||||
|
h2[1].savefig('%s/%s.%s' % (figdir,
|
||||||
|
'021-TF_RelativePhase-wgv', fig_ext))
|
||||||
|
else:
|
||||||
|
h2[0].savefig('%s/%s.%s' % (figdir,
|
||||||
|
'02d-TF_RelativeGain-gv', fig_ext))
|
||||||
else:
|
else:
|
||||||
h2[0].savefig('%s/%s.%s' % (figdir, 'NA-02.0', fig_ext))
|
if not args.subplot:
|
||||||
h2[1].savefig('%s/%s.%s' % (figdir, 'NA-02.1', fig_ext))
|
h2[0].savefig('%s/%s.%s' % (figdir,
|
||||||
|
'020-TF_RelativeGainPhase', fig_ext))
|
||||||
|
h2[1].savefig('%s/%s.%s' % (figdir,
|
||||||
|
'021-TF_RelativePhase', fig_ext))
|
||||||
|
else:
|
||||||
|
h2[0].savefig('%s/%s.%s' % (figdir,
|
||||||
|
'02d-TF_RelativeGain', fig_ext))
|
||||||
if HEADLESS:
|
if HEADLESS:
|
||||||
pp.close()
|
pp.close()
|
||||||
else:
|
else:
|
||||||
|
@ -301,9 +339,13 @@ if 2 in plot_list or 12 in plot_list:
|
||||||
|
|
||||||
################################################################################
|
################################################################################
|
||||||
if 3 in plot_list or 13 in plot_list:
|
if 3 in plot_list or 13 in plot_list:
|
||||||
h3 = [pp.figure() for x in range(2)]
|
if not args.subplot:
|
||||||
|
h3 = [pp.figure() for x in range(2)]
|
||||||
|
ax3 = [hT.add_subplot(1,1,1) for hT in h3]
|
||||||
|
else:
|
||||||
|
h3 = [pp.figure() for x in range(1)]
|
||||||
|
ax3 = h3[0].subplots(2,1)
|
||||||
|
|
||||||
ax3 = [hT.add_subplot(1,1,1) for hT in h3]
|
|
||||||
ax3[0].plot(bw_mag,dB20(rms_gain_swp))
|
ax3[0].plot(bw_mag,dB20(rms_gain_swp))
|
||||||
ax3[1].plot(bw_ang,rms_ang_swp*180/np.pi)
|
ax3[1].plot(bw_ang,rms_ang_swp*180/np.pi)
|
||||||
|
|
||||||
|
@ -326,11 +368,17 @@ if 3 in plot_list or 13 in plot_list:
|
||||||
|
|
||||||
if args.save:
|
if args.save:
|
||||||
if 13 in plot_list:
|
if 13 in plot_list:
|
||||||
h3[0].savefig('%s/%s.%s' % (figdir, 'NA-13.0', fig_ext))
|
if not args.subplot:
|
||||||
h3[1].savefig('%s/%s.%s' % (figdir, 'NA-13.1', fig_ext))
|
h3[0].savefig('%s/%s.%s' % (figdir, 'NA-13.0', fig_ext))
|
||||||
|
h3[1].savefig('%s/%s.%s' % (figdir, 'NA-13.1', fig_ext))
|
||||||
|
else:
|
||||||
|
h3[0].savefig('%s/%s.%s' % (figdir, 'NA-13.0', fig_ext))
|
||||||
else:
|
else:
|
||||||
h3[0].savefig('%s/%s.%s' % (figdir, 'NA-03.0', fig_ext))
|
if not args.subplot:
|
||||||
h3[1].savefig('%s/%s.%s' % (figdir, 'NA-03.1', fig_ext))
|
h3[0].savefig('%s/%s.%s' % (figdir, 'NA-03.0', fig_ext))
|
||||||
|
h3[1].savefig('%s/%s.%s' % (figdir, 'NA-03.1', fig_ext))
|
||||||
|
else:
|
||||||
|
h3[0].savefig('%s/%s.%s' % (figdir, 'NA-03.d', fig_ext))
|
||||||
if HEADLESS:
|
if HEADLESS:
|
||||||
pp.close()
|
pp.close()
|
||||||
else:
|
else:
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue