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,