bordas comb test.py

import abel
from abel.tools.analytical import PiecewisePolynomial
from itertools import chain
import matplotlib.pyplot as plt
import numpy as np

hw = 1     # peak half-width - Or is this the full-width?
step = 10  # center-to-center distance between peaks
n = 10     # number of peaks

rmax = int(n * step)

def peak(i):
    c = i * step
    if i:
        return [(c - hw, c, [1,  1], c, hw),
                (c, c + hw, [1, -1], c, hw)]
    else:
        return [(c, c + hw, [1, -1], c, hw)]


comb = PiecewisePolynomial(rmax + 1, rmax,
                           chain(*[peak(i) for i in range(1, n)]),
                           symmetric=False)

func = comb.abel

zeros_array = np.zeros_like(func)
IM = np.array([zeros_array, zeros_array, func, zeros_array, zeros_array])


trans = abel.onion_bordas.onion_bordas_transform(IM)



fig, axs = plt.subplots(1,2,figsize=(8,4))
axs[0].imshow(IM, aspect='auto')
axs[1].imshow(trans, aspect='auto')

plt.show()