def _plot_accel_Bode(bode, fi, linetype='-', \
bode_opts=None, **kwargs):
if bode_opts is None:
AccelFB_Bode_opts[1]
BPO._plot_bode(bode, bode_opts, \
f, fignum=fi, linetype=linetype, \
**kwargs)
def plot_one_bode(self, f, index, fignum=1, clear=False, **kwargs):
"""index refers to which bode option to pull from
self.bode_opts. This is how the method determines which bode
to plot."""
if not hasattr(self, "bodes"):
self.calc_bodes(f)
if not kwargs.has_key("label"):
kwargs["label"] = self.label
opt = self.bode_opts[index]
bode = self.find_bode(opt.output_label, opt.input_label)
BPO._plot_bode(bode, opt, f, fignum=fignum, clear=clear, **kwargs)
def plot_bodes(self, f, startfi=1, clear=False, **kwargs):
"""This method is for compatability with
bode_plot_overlayer.py"""
if not hasattr(self, "bodes"):
self.calc_bodes(f)
if not kwargs.has_key("label"):
kwargs["label"] = self.label
for i, opt in enumerate(self.bode_opts):
fignum = startfi + i
bode = self.find_bode(opt.output_label, opt.input_label)
BPO._plot_bode(bode, opt, f, fignum=fignum, clear=clear, **kwargs)
def ThetaFB_bode_plots(fi=3, resave=0):
ThetaFB_Exp_Bode_Plotter = build_ThetaFB_exp_Bode_plotter()
design_dir = rwkos.FindFullPath('SFLR_2010/vibration_suppression_design/TMM')
if design_dir not in sys.path:
sys.path.append(design_dir)
Gth_TMM = build_Gth_TMM_model()
import Gth
reload(Gth)
ThetaFB_TMM_Bode_Plotter = BPO.TMM_bode_object(Gth_TMM, \
ThetaFB_Bode_opts)
ThetaFB_Plotters =[ThetaFB_Exp_Bode_Plotter, \
ThetaFB_TMM_Bode_Plotter, \
]
ThetaFB_overlayer = BPO.Bode_Overlayer(ThetaFB_Plotters, \
ThetaFB_Bode_opts)
ThetaFB_overlayer.plot_bodes(f, startfi=fi)
th_comp, a_comp = model_w_bm_bodes_ThetaFB_editted.Bodes(s, \
params, \
Gth.Gth)
PU.SetLegend(fi,axis=0,loc=3)
PU.SetLegend(fi+1,axis=0,loc=2)
if resave:
mysave('thetafb_theta_thd_bode.eps', fi)
mysave('thetafb_a_thd_bode.eps', fi+1)
th_bode_opt = ThetaFB_Bode_opts[0]
a_bode_opt = ThetaFB_Bode_opts[1]
th_bode = BPO.comp_to_Bode(th_comp, f, bode_opt=th_bode_opt)
a_bode = BPO.comp_to_Bode(a_comp, f, bode_opt=a_bode_opt)
BPO._plot_bode(th_bode, th_bode_opt, f, fignum=fi, linetype='k:')
BPO._plot_bode(a_bode, a_bode_opt, f, fignum=fi+1, linetype='k:')
def plot_Accel_comp(bode, fi, linetype='-', **kwargs):
BPO._plot_bode(bode, Accel_Comp_Bode_opts[1], \
f, fignum=fi, linetype=linetype, **kwargs)
