_tester_utils.py

import numpy as np
from ecgdetectors import Detectors
from datetime import datetime

det_names = ['two_average', 'matched_filter', 'swt', 'engzee', 'christov', 'hamilton', 'pan_tompkins']

def sort_MIT_annotations(ann):
    beat_labels = ['N', 'L', 'R', 'B', 'A', 'a', 'J', 'S', 'V', 'r', 'F', 'e', 'j', 'n', 'E', '/', 'f', 'Q', '?']
    in_beat_labels = np.in1d(ann.symbol, beat_labels)

    sorted_anno = ann.sample[in_beat_labels]
    sorted_anno = np.unique(sorted_anno)

    return sorted_anno


"""
From the detected R peaks the function works itself backwards to
calculate the median delay the detector introduces. This is used
for benchmarking to compensate for different delays the detctors
introduce.
"""
def calcMedianDelay(detected_peaks, unfiltered_ecg, search_samples):

    r_peaks = []
    window = int(search_samples)

    for i in detected_peaks:
        i = int(i)
        if i<window:
            section = unfiltered_ecg[:i]
            r_peaks.append(np.argmax(section))
        else:
            section = unfiltered_ecg[i-window:i]
            r_peaks.append(window - np.argmax(section))

    m = int(np.median(r_peaks))
    #print("Delay = ",m)
    return m


def evaluate_detector(test, annotation, delay, tol=0):

    test = np.unique(test)
    reference = np.unique(annotation)
    
    TP = 0

    for anno_value in test:
        test_range = np.arange(anno_value-tol-delay, anno_value+1+tol-delay)
        in1d = np.in1d(test_range, reference)
        if np.any(in1d):
            TP = TP + 1
    
    FP = len(test)-TP
    FN = len(reference)-TP 

    return TP, FP, FN


def det_from_name(detector_name, fs):

        detectors = Detectors(fs)
        detectors.engzee_fake_delay = 10

        if detector_name=='two_average':
                return detectors.two_average_detector
        elif detector_name=='matched_filter':
                return detectors.matched_filter_detector
        elif detector_name=='swt':
                return detectors.swt_detector
        elif detector_name=='engzee':
                return detectors.engzee_detector
        elif detector_name=='christov':
                return detectors.christov_detector
        elif detector_name=='hamilton':
                return detectors.hamilton_detector
        elif detector_name=='pan_tompkins':
                return detectors.pan_tompkins_detector
        else:
                raise RuntimeError('invalid detector name!')


def get_time():

        time = str(datetime.now().time())
        time = time[:5]
        time = time.replace(':', '.')
        
        return time