Skip to content
Snippets Groups Projects
Commit 1191ecc2 authored by Gregory Ashton's avatar Gregory Ashton
Browse files

Minor improvements to general running

1) Adds check if the truths are not of the correct length (to prevent
corner.corner error)
2) Improve sliding window labels
3) Adds option to plot gps seconds on the sliding window
parent 6d8c913e
No related branches found
No related tags found
No related merge requests found
......@@ -399,7 +399,7 @@ class FrequencySlidingWindow(GridSearch):
self.input_data = np.array(input_data)
def plot_sliding_window(self, F0=None, ax=None, savefig=True,
colorbar=True):
colorbar=True, timestamps=False):
data = self.data
if ax is None:
ax = plt.subplot()
......@@ -411,7 +411,7 @@ class FrequencySlidingWindow(GridSearch):
dts = (tmids - self.minStartTime) / 86400.
if F0:
frequencies = frequencies - F0
ax.set_ylabel('Frequency - $f_0$ [Hz]')
ax.set_ylabel('Frequency - $f_0$ [Hz] \n $f_0={:0.2f}$'.format(F0))
else:
ax.set_ylabel('Frequency [Hz]')
twoF = twoF.reshape((len(tmids), len(frequencies)))
......@@ -421,10 +421,17 @@ class FrequencySlidingWindow(GridSearch):
cb = plt.colorbar(pax, ax=ax)
cb.set_label('$2\mathcal{F}$')
ax.set_xlabel(
r'Days from $t_\mathrm{{start}}$={}'.format(self.minStartTime))
r'Mid-point (days after $t_\mathrm{{start}}$={})'.format(
self.minStartTime))
ax.set_title(
'Sliding window length = {} days in increments of {} days'
.format(self.window_size/86400, self.window_delta/86400))
.format(self.window_size/86400, self.window_delta/86400),
)
if timestamps:
axT = ax.twiny()
axT.set_xlim(tmids[0]*1e-9, tmids[-1]*1e-9)
axT.set_xlabel('Mid-point timestamp [GPS $10^{9}$ s]')
ax.set_title(ax.get_title(), y=1.18)
if savefig:
plt.tight_layout()
plt.savefig(
......
......@@ -544,6 +544,10 @@ class MCMCSearch(core.BaseSearchClass):
"""
if 'truths' in kwargs and len(kwargs['truths']) != self.ndim:
logging.warning('len(Truths) != ndim, Truths will be ignored')
kwargs['truths'] = None
if self.ndim < 2:
with plt.rc_context(rc_context):
if fig_and_axes is None:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment