from skued import azimuthal_average, diffread, autocenter
import numpy as np
import matplotlib.pyplot as plt

im = diffread('data/Cr_1.tif')

mask = np.ones_like(im, dtype = bool)
mask[0:1250, 975:1225] = False

# Reduce size of images because of memory usage of ReadTheDocs
im = im[::3, ::3]
mask = mask[::3, ::3]

yc, xc = autocenter(im, mask=mask)
radius, intensity = azimuthal_average(im, (xc, yc), mask=mask)

plt.plot(radius, intensity)
plt.xlabel('Radius [px]')
plt.ylabel('Diffracted intensity [a.u.]')
plt.tight_layout()
plt.show()