noise_module.py

def _get_odd_res(res):
    return (res, res + 1)[res % 2 == 0]

def _meadian_window(samp_per_sym):
    res = round(samp_per_sym/2)
##    print(res)
    return _get_odd_res(res)

def _margin_bits(bit, half_window):
    return bit * half_window

def _get_capture_with_margins(capture, half_window):
    start_marg = _margin_bits(capture[0], half_window)
    end_marg = _margin_bits(capture[len(capture) - 1], half_window)
    return start_marg + capture + end_marg

def _bool_to_bit(boolean):
    return str(int(boolean))

def _get_median(capture, i, half_window):
    lst = [int(capture[j]) for j in range(i - half_window, i + half_window + 1)]
##    print(lst)
    return _bool_to_bit(sum(lst) > half_window)

def meadian_noise(capture, samp_per_sym):
    window = _meadian_window(samp_per_sym)
    half_window = window//2
    capture = _get_capture_with_margins(capture, half_window)
    res = ''
    
    for i in range(half_window, len(capture) - half_window):
        res += _get_median(capture, i, half_window)
    
    return res

##l1 = '110110001011111'
##print(l1)
##print(meadian_noise(l1, 5))