load xrd

loadxrd.py

def loadXRD(filename):
    with open(filename, 'r') as f:
        lines = f.readlines()
    angles = []
    intensity = []
    for line in lines[1:-2]:
        x, y = line.split('\n')[0].split(",")[:-1]
        angles.append(float(x))
        intensity.append(float(y))
    x = np.array(angles)
    y = np.array(intensity)

    scale = np.diff(x).mean()
    offset = x[0]

    s = hs.signals.Signal1D(y)
    s.axes_manager[-1].offset = offset
    s.axes_manager[-1].scale = scale
    s.axes_manager[-1].units = "2 $\theta$"

    return s