def plot_envelope(self, signal, legend=False, num=None):
'''
Sliding window amplitude envelope
'''
if self.L is None:
print('Set a window size for the sliding window first!')
return
if self.ax_signal is None:
fig = ppl.figure(num, figsize=(6, 3.5))
self.ax_signal = pl.mk_signal_ax(self.time_unit_label, fig=fig)
tvec = np.arange(len(signal)) * self.dt
envelope = self.get_envelope(signal, self.L)
pl.draw_envelope(self.ax_signal, tvec, envelope)
if legend:
self.ax_signal.legend(fontsize=pl.label_size, ncol=3)
ymin, ymax = self.ax_signal.get_ylim()
self.ax_signal.set_ylim((ymin, 1.3 * ymax))
fig = ppl.gcf()
fig.subplots_adjust(bottom=0.18)
fig.tight_layout()
def doPlot(self):
'''
Checks the checkboxes for trend and envelope..
'''
# update raw_signal and tvec
succ = self.vector_prep(self.signal_id) # error handling done here
if not succ:
return False
if self.debug:
print(
"called Plotting [raw] [trend] [detrended] [envelope]",
self.cb_raw.isChecked(),
self.cb_trend.isChecked(),
self.cb_detrend.isChecked(),
self.cb_envelope.isChecked(),
)
# check if trend is needed
if self.cb_trend.isChecked() or self.cb_detrend.isChecked():
trend = self.calc_trend()
if trend is None:
return
else:
trend = None
# envelope calculation
if self.cb_envelope.isChecked():
envelope = self.calc_envelope()
if envelope is None:
return
else:
envelope = None
self.tsCanvas.fig1.clf()
ax1 = pl.mk_signal_ax(self.time_unit, fig=self.tsCanvas.fig1)
self.tsCanvas.fig1.add_axes(ax1)
if self.debug:
print(
f"plotting signal and trend with {self.tvec[:10]}, {self.raw_signal[:10]}"
)
if self.cb_raw.isChecked():
pl.draw_signal(ax1, time_vector=self.tvec, signal=self.raw_signal)
if trend is not None and self.cb_trend.isChecked():
pl.draw_trend(ax1, time_vector=self.tvec, trend=trend)
if trend is not None and self.cb_detrend.isChecked():
ax2 = pl.draw_detrended(
ax1, time_vector=self.tvec, detrended=self.raw_signal - trend
)
ax2.legend(fontsize=pl.tick_label_size)
if envelope is not None and not self.cb_detrend.isChecked():
pl.draw_envelope(ax1, time_vector=self.tvec, envelope=envelope)
# plot on detrended axis
if envelope is not None and self.cb_detrend.isChecked():
pl.draw_envelope(ax2, time_vector=self.tvec, envelope=envelope)
ax2.legend(fontsize=pl.tick_label_size)
self.tsCanvas.fig1.subplots_adjust(bottom=0.15, left=0.15, right=0.85)
# add a simple legend
ax1.legend(fontsize=pl.tick_label_size)
self.tsCanvas.draw()
self.tsCanvas.show()
def doPlot(self):
if self.raw_signal is None:
self.NoSignal = MessageWindow("Please create a signal first!",
"No Signal")
if self.debug:
print(
"called Plotting [raw] [trend] [detrended] [envelope]",
self.cb_raw.isChecked(),
self.cb_trend.isChecked(),
self.cb_detrend.isChecked(),
self.cb_envelope.isChecked(),
)
# check if trend is needed
if self.T_c and (self.cb_trend.isChecked()
or self.cb_detrend.isChecked()):
if self.debug:
print("Calculating trend with T_c = ", self.T_c)
trend = self.calc_trend()
else:
trend = None
# envelope calculation
if self.L and self.cb_envelope.isChecked():
if self.debug:
print("Calculating envelope with L = ", self.L)
envelope = self.calc_envelope()
else:
envelope = None
self.tsCanvas.fig1.clf()
ax1 = pl.mk_signal_ax(self.time_unit, fig=self.tsCanvas.fig1)
self.tsCanvas.fig1.add_axes(ax1)
if self.debug:
print(
f"plotting signal and trend with {self.tvec[:10]}, {self.raw_signal[:10]}"
)
if self.cb_raw.isChecked():
pl.draw_signal(ax1, time_vector=self.tvec, signal=self.raw_signal)
if trend is not None and self.cb_trend.isChecked():
pl.draw_trend(ax1, time_vector=self.tvec, trend=trend)
if trend is not None and self.cb_detrend.isChecked():
ax2 = pl.draw_detrended(ax1,
time_vector=self.tvec,
detrended=self.raw_signal - trend)
ax2.legend(fontsize=pl.tick_label_size)
if envelope is not None and not self.cb_detrend.isChecked():
pl.draw_envelope(ax1, time_vector=self.tvec, envelope=envelope)
# plot on detrended axis
if envelope is not None and self.cb_detrend.isChecked():
pl.draw_envelope(ax2, time_vector=self.tvec, envelope=envelope)
ax2.legend(fontsize=pl.tick_label_size)
self.tsCanvas.fig1.subplots_adjust(bottom=0.15, left=0.15, right=0.85)
# add a simple legend
ax1.legend(fontsize=pl.tick_label_size)
self.tsCanvas.draw()
self.tsCanvas.show()