data = {}
for chann in xrange(1, 4):
data[chann] = []
for chann, value in ekg_analyser._threshold_strategy.items():
data[chann].append((plotter._timeline,
ones(len(plotter._timeline)) * value,
False))
for chann in xrange(1, 4):
xdata = ekg_analyser.get_peaks(chann, seconds = True)
xdata_frames = ekg_analyser.get_peaks(chann)
ydata = signal.get_channel(chann)[xdata_frames]
data[chann].append((xdata, ydata, True))
plotter.plot_set(signal, extra_info = data)
# einthoven test
einthoven = data_preparator.einthoven_signal_set()
chann_list = [einthoven.get_channel(1) + einthoven.get_channel(3),
einthoven.get_channel(2)]
plotter.plot_channel_list(chann_list, title_list = ["Einthoven: I + III",
"Einthoven: II"])
## Analysing channel
# peaks
peaks = ekg_analyser.get_peaks(options.channel)
##Difference plot - beats
peaks_sec = ekg_analyser.get_peaks(options.channel, seconds = True)
y1 = ekg_analyser.inter_peak_diff(peaks, seconds = True)
data_preparator.apply_filters()
else:
data_preparator.apply_filters(exclude = [1])
print "Analysing channel %s\n" % options.channel
# plot signal
if options.mode == "trigger":
signal = data_preparator.trigger_signal_set()
elif options.mode == "simple":
signal = data_preparator.difference_signal_set()
else:
signal = data_preparator.trigger_signal_set(just_trigger = True)
emg_analyser.load_signal_set(signal)
plotter.plot_set(signal, ylabel = "potential [uV]")
print "Mean amplitude: %s [uV]" % emg_analyser.mean_amplitude(options.channel)
# plot full spectrum
freq, spec = emg_analyser.signal_spectrum(options.channel)
plotter.plot_channel(spec,
xaxis = freq,
xlabel = "frequency [Hz]",
ylabel = "magnitude [(uV s) ^ 2]",
title = "EMG spectrum")
if options.mode in ["trigger", "file"]:
trigger_moments = emg_analyser.get_trigger_times(seconds = True)
emg_moments = emg_analyser.get_emg_times(options.channel,
window = 10,
