import numpy as np
import matplotlib.pyplot as plt
from spectractor.extractor.psf import *
Nx = 50
Ny = 50
yy, xx = np.mgrid[:Nx, :Ny]
amplitude = 10
a = evaluate_moffat2d(xx, yy, amplitude=amplitude, y_c=Ny/2, x_c=Nx/2, gamma=5, alpha=2)
im = plt.pcolor(xx, yy, a)
plt.grid()
plt.xlabel("x")
plt.ylabel("y")
plt.colorbar(im, label="Moffat 2D")
plt.show()