import matplotlib.pyplot as plt
plt.ion()
import numpy as np


from simple_eta import gaussian_2d, plot_gaussian
from simple_eta import generate

sigma_um = 12
resolution = 200
grid_size = 2
pixel_size = 25


# pos = (37.5,47.5) #x,y
# t = gaussian_2d(mx=pos[0], my = pos[1], sigma = sigma_um, res = resolution, grid_size = grid_size, pixel_size = pixel_size)

# fig, ax = plot_gaussian(t, pixel_size=pixel_size, grid_size = grid_size)
# plt.show()

# res = generate.sum_pixels(t, grid_size)


# fig,ax = plt.subplots()
# im = ax.imshow(res, origin = 'lower')

mx,my, pixels = generate.generate_uniform_hits(sigma = sigma_um, pixel_size = pixel_size, grid_size = grid_size, resolution=resolution, N=300)
# pixels2 = generate.sum2x2(t)

fig, ax = plt.subplots(figsize = (8,8))
ax.scatter(mx,my)

ax.set_xlim(0,grid_size*pixel_size)
ax.set_ylim(0,grid_size*pixel_size)
ticks = [tick for tick in range(0,pixel_size*grid_size+1, pixel_size)]
ax.set_xticks(ticks)
ax.set_yticks(ticks)
ax.grid()