Broke out individual plotting commands, and updated CLI

Calling the tool with the debug flag will now automatically enter into an
interactive python console following execution. This is a handy way to try to
break into screwing around with the code if we are working with things.

.gitignore

@@ -1,3 +1,6 @@
+# Figures directory
+figures/
+
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]

LPRDefaultPlotting.py

@@ -37,3 +37,6 @@ rcParams['axes.prop_cycle'] = (cycler('linestyle',['-','--'])*cycler(color=COLOR
 		
 for tri in COLOR_CYCLE_LIST:
 	color = '0x' + ''.join([ "%02x" % int(255*x) for x in tri])
+
+figures_directory='figures'
+

TankGlobals.py

@@ -12,12 +12,15 @@ def g1_map_default(system):
 	g1_swp = system.g1 * np.sin(np.pi/2-system.phase_swp) / system.alpha_swp
 	return g1_swp
 	
+def g1_map_flat(system):
+	return system.g1*np.ones(system.phase_swp.shape)
+
 # Operating Enviornment
 #####
 class ampSystem:
 	f0		= 28
 	bw0		= 8
-	bw_plt	= 0.5
+	bw_plt	= 3
 	"""define global (hardware descriptive) variables for use in our system."""
 	def __init__(self, quiet=False):
 		self.f0		= self.__class__.f0 # design frequency (GHz)

tankPlot.py

@@ -1,16 +1,51 @@
 #!/usr/bin/env python3
 
 import numpy as np
+import matplotlib
+import argparse
+import os
+import code
 
+################################################################################
+args_parser = argparse.ArgumentParser()
+args_parser.add_argument('-n', type=int, default=1,\
+	help='plot testing number')
+args_parser.add_argument('--save','-s', action='store_true',\
+	help='save to files')
+args_parser.add_argument('--raster','-r', action='store_true',\
+	help='save as raster')
+args_parser.add_argument('--debug','-d', action='store_true',\
+	help='hold for debugging')
+args_parser.add_argument('--headless','-q', action='store_true',\
+	help='Remain neadless even if we aren\'t saving files.')
+args = args_parser.parse_args()
+
+#exit()
+
+HEADLESS = not 'DISPLAY' in os.environ.keys()
+if args.headless: HEADLESS = True	# Overide Manually if request
+if HEADLESS: matplotlib.use('Agg');
+
+################################################################################
 from matplotlib import rcParams, pyplot as pp
 import LPRDefaultPlotting
 
+figdir = LPRDefaultPlotting.figures_directory
+if args.save: os.makedirs(figdir, exist_ok=True)
+
 import sys
+
 sys.path.append("./pySmithPlot")
 import smithplot
 from smithplot import SmithAxes
 
-plot_list = [4]
+plot_list = [args.n]
+
+if args.raster:
+	args.save = True
+	fig_ext = 'png'
+else:
+	fig_ext = 'pdf'
 
 ################################################################################
 # Override the defaults for this script
@@ -22,10 +57,18 @@ default_window_position=['+20+80', '+120+80']
 import TankGlobals
 from FreqClass import FreqClass
 from tankComputers import *
+freq_pts = 501
 
 S=TankGlobals.ampSystem()
 B=TankGlobals.bufferSystem()
-f=FreqClass(501, S.f0, S.bw_plt)
+if plot_list[0] in [4, 5]:
+	S.bw_plt = 0.5
+	B.bw_plt = S.bw_plt
+	freq_pts = 51
+if plot_list[0] == 5:
+	S.set_g1_swp(TankGlobals.g1_map_flat)
+
+f=FreqClass(freq_pts, S.f0, S.bw_plt)
 
 ################################################################################
 # We want a smooth transition out to alpha. So For now assume a squares
@@ -101,6 +144,11 @@ if 6 in plot_list:
 		axT.yaxis.label.set_color(c_color)
 		axT.tick_params('y', colors=c_color)
 	h6.tight_layout()
+	if args.save:
+		h6.savefig('%s/%s.%s' % (figdir, 'NA-6.0', fig_ext))
+	if HEADLESS:
+		pp.close()
+	else:
 		mgr.window.geometry(default_window_position[0])
 		h6.show()
 
@@ -122,6 +170,12 @@ if 1 in plot_list:
 		axT.set_xlim(f.hz_range)
 	
 	[hT.tight_layout() for hT in h1]
+	if args.save:
+		h1[0].savefig('%s/%s.%s' % (figdir, 'NA-1.0', fig_ext))
+		h1[1].savefig('%s/%s.%s' % (figdir, 'NA-1.1', fig_ext))
+	if HEADLESS:
+		pp.close()
+	else:
 		mgr.window.geometry(default_window_position[0])
 		[hT.show() for hT in h1]
 
@@ -141,9 +195,40 @@ if 4 in plot_list:
 	ax4[1].title.set_position((old_pos[0], 1.1))
 	h4[1].tight_layout()
 	#[hT.tight_layout() for hT in h4]
+	if args.save:
+		h4[0].savefig('%s/%s.%s' % (figdir, 'ideal-smith_tank_impedance', fig_ext))
+		h4[1].savefig('%s/%s.%s' % (figdir, 'ideal-polar_gain_plot', fig_ext))
+	if HEADLESS:
+		pp.close()
+	else:
 		mgr.window.geometry(default_window_position[0])
 		[hT.show() for hT in h4]
 
+if 5 in plot_list:
+	h5 = [pp.figure(figsize=(3.4,3.4)) for x in range(2)]
+	ax5 = []
+	ax5.append(h5[0].add_subplot(1,1,1, projection='smith'))
+	ax5.append(h5[1].add_subplot(1,1,1, projection='polar'))
+
+	ax5[0].plot(y_tank, datatype=SmithAxes.Y_PARAMETER, marker="None")
+	ax5[1].plot(np.angle(tf), dB20(tf))
+
+	ax5[0].set_title('Tank Impedance')
+	ax5[1].set_title('Transfer Function')
+
+	old_pos = ax5[1].title.get_position()
+	ax5[1].title.set_position((old_pos[0], 1.1))
+	h5[1].tight_layout()
+	#[hT.tight_layout() for hT in h5]
+	if args.save:
+		h5[0].savefig('%s/%s.%s' % (figdir, 'ideal-flat_g1-smith_tank_impedance', fig_ext))
+		h5[1].savefig('%s/%s.%s' % (figdir, 'ideal-flat_g1-polar_gain_plot', fig_ext))
+	if HEADLESS:
+		pp.close()
+	else:
+		mgr.window.geometry(default_window_position[0])
+		[hT.show() for hT in h5]
+	
 ################################################################################
 if 2 in plot_list:
 	h2 = [pp.figure() for x in range(2)]
@@ -164,7 +249,13 @@ if 2 in plot_list:
 		axT.set_xlabel('Freq (GHz)')
 		axT.set_xlim(f.hz_range)
 	[hT.tight_layout() for hT in h2]
-	mgr.window.geometry(default_window_position[1])
+	if args.save:
+		h2[0].savefig('%s/%s.%s' % (figdir, 'NA-2.0', fig_ext))
+		h2[1].savefig('%s/%s.%s' % (figdir, 'NA-2.1', fig_ext))
+	if HEADLESS:
+		pp.close()
+	else:
+		mgr.window.geometry(default_window_position[0])
 		[hT.show() for hT in h2]
 
 ################################################################################
@@ -185,5 +276,20 @@ if 3 in plot_list:
 		axT.set_xlabel('Bandwidth (GHz)')
 
 	[hT.tight_layout() for hT in h3]
+	if args.save:
+		h3[0].savefig('%s/%s.%s' % (figdir, 'NA-3.0', fig_ext))
+		h3[1].savefig('%s/%s.%s' % (figdir, 'NA-3.1', fig_ext))
+	if HEADLESS:
+		pp.close()
+	else:
+		mgr.window.geometry(default_window_position[0])
 		[hT.show() for hT in h3]
 
+if args.debug:
+	print("")
+	print("#"*80)
+	print("# Finished execution.")
+	print("# Debugging Mode active.")
+	print("# Falling back to an interactive prompt.")
+	print("#"*80)
+	code.interact(local=dict(globals(), **locals()))