def draw_several_cams(geom, ncams=4):
cmaps = ['jet', 'afmhot', 'terrain', 'autumn']
fig, axs = plt.subplots(1, ncams, figsize=(15, 4), sharey=True, sharex=True)
for ii in range(ncams):
disp = visualization.CameraDisplay(
geom,
ax=axs[ii],
title="CT{}".format(ii + 1),
)
disp.cmap = cmaps[ii]
model = toymodel.generate_2d_shower_model(
centroid=(0.2 - ii * 0.1, -ii * 0.05),
width=0.005 + 0.001 * ii,
length=0.1 + 0.05 * ii,
psi=ii * 20 * u.deg,
)
image, sig, bg = toymodel.make_toymodel_shower_image(
geom,
model.pdf,
intensity=50,
nsb_level_pe=1000,
)
clean = image.copy()
clean[image <= 3.0 * image.mean()] = 0.0
hillas = hillas_parameters(geom.pix_x, geom.pix_y, clean)
disp.image = image
disp.add_colorbar(ax=axs[ii])
disp.set_limits_percent(95)
disp.overlay_moments(hillas, linewidth=3, color='blue')
"""
if __name__ == '__main__':
# Prepare the camera geometry
geom = io.CameraGeometry.from_name('hess', 1)
disp = visualization.CameraDisplay(geom)
disp.set_limits_minmax(0, 350)
disp.add_colorbar()
# make a toymodel shower model
model = toymodel.generate_2d_shower_model(centroid=(0.2, 0.2), width=0.01, length=0.1, psi='45d')
# generate toymodel image in camera for this shower model.
image, signal, noise = toymodel.make_toymodel_shower_image(geom, model.pdf, intensity=50, nsb_level_pe=100)
#Image cleaning
clean_mask = tailcuts_clean(geom, image, 1, 10, 5) #pedvars = 1 and core and boundary threshold in pe
image[~clean_mask] = 0
#Pixel values in the camera
pix_x = geom.pix_x.value
pix_y = geom.pix_y.value
# Hillas parameters
hillas1, hillas2 = hillas_parameters(pix_x, pix_y, image)
print(hillas1, hillas2)
#Overlay moments