Commit fd30b183 authored by Jameson Graef Rollins's avatar Jameson Graef Rollins

return figure handle from plot_noise

parent 7799b43c
......@@ -74,12 +74,23 @@ def plot_noise(
"""Plot a GWINC noise budget from calculated noises
If an axis handle is provided it will be used for the plot.
`displacement` may be set to False to supress the right had
displacement axis.
Returns the figure handle used.
f = noises['Freq']
if ax is None:
import matplotlib.pyplot as plt
fig = plt.figure()
ax = fig.add_subplot(1, 1, 1)
fig = ax.figure
def plot_dict(noises):
#use sorted to force a consistent ordering
......@@ -132,3 +143,5 @@ def plot_noise(
y1, y2 = ax.get_ylim()
ax_d.set_ylim(y1*ifo.Infrastructure.Length, y2*ifo.Infrastructure.Length)
ax_d.set_ylabel(u"Displacement [m/\u221AHz]")
return fig
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment