gammasimtools 0.5.1__py3-none-any.whl → 0.6.1__py3-none-any.whl

simtools/corsika/corsika_histograms_visualize.py

@@ -1,24 +1,26 @@
 import logging
+from pathlib import Path
 
-import matplotlib.colors as colors
 import matplotlib.pyplot as plt
 import numpy as np
 from astropy import units as u
+from matplotlib import colors
+from matplotlib.backends.backend_pdf import PdfPages
 
 _logger = logging.getLogger(__name__)
 
 
-def _kernel_plot_2D_photons(corsika_histograms_instance, property_name, log_z=False):
+def _kernel_plot_2d_photons(histograms_instance, property_name, log_z=False):
     """
-    The next functions below are used by the the CorsikaHistograms class to plot all sort of information
-    from the Cherenkov photons saved.
+    The next functions below are used by the the CorsikaHistograms class to plot all sort of
+    information from the Cherenkov photons saved.
 
     Create the figure of a 2D plot. The parameter `name` indicate which plot.
     Choices are "counts", "density", "direction", "time_altitude", and "num_photons_per_telescope".
 
     Parameters
     ----------
-    corsika_histograms_instance: corsika.corsika_histograms.CorsikaHistograms
+    histograms_instance: corsika.corsika_histograms.CorsikaHistograms
         instance of corsika.corsika_histograms.CorsikaHistograms.
     property_name: string
         Name of the quantity. Options are: "counts", "density", "direction", "time_altitude" and
@@ -30,63 +32,68 @@ def _kernel_plot_2D_photons(corsika_histograms_instance, property_name, log_z=Fa
     -------
     list
         List of figures for the given telescopes.
-    list
-        List of the figure names.
 
     Raises
     ------
     ValueError
         if `property` is not allowed.
     """
-    if property_name not in corsika_histograms_instance._dict_2D_distributions:
-        msg = (f"This property does not exist. The valid entries are "
-               f"{corsika_histograms_instance._dict_2D_distributions}")
+    if property_name not in histograms_instance.dict_2d_distributions:
+        msg = (
+            f"This property does not exist. The valid entries are "
+            f"{histograms_instance.dict_2d_distributions}"
+        )
         _logger.error(msg)
         raise ValueError
     function = getattr(
-        corsika_histograms_instance,
-        corsika_histograms_instance._dict_2D_distributions[property_name]["function"],
+        histograms_instance,
+        histograms_instance.dict_2d_distributions[property_name]["function"],
     )
-    hist_values, x_edges, y_edges = function()
+    hist_values, x_bin_edges, y_bin_edges = function()
 
     all_figs = []
-    fig_names = []
-    for i_hist, _ in enumerate(x_edges):
+    for i_hist, _ in enumerate(x_bin_edges):
         fig, ax = plt.subplots()
         if log_z is True:
             norm = colors.LogNorm(vmin=1, vmax=np.amax([np.amax(hist_values[i_hist]), 2]))
         else:
             norm = None
-        mesh = ax.pcolormesh(x_edges[i_hist], y_edges[i_hist], hist_values[i_hist], norm=norm)
+        mesh = ax.pcolormesh(
+            x_bin_edges[i_hist], y_bin_edges[i_hist], hist_values[i_hist], norm=norm
+        )
         if (
-                corsika_histograms_instance._dict_2D_distributions[property_name]["x edges unit"] is
-                not u.dimensionless_unscaled
+            histograms_instance.dict_2d_distributions[property_name]["x axis unit"]
+            is not u.dimensionless_unscaled
         ):
             ax.set_xlabel(
-                f"{corsika_histograms_instance._dict_2D_distributions[property_name]['x edges']} "
-                f"({corsika_histograms_instance._dict_2D_distributions[property_name]['x edges unit']})"
+                f"{histograms_instance.dict_2d_distributions[property_name]['x bin edges']} "
+                f"({histograms_instance.dict_2d_distributions[property_name]['x axis unit']})"
             )
         else:
             ax.set_xlabel(
-                f"{corsika_histograms_instance._dict_2D_distributions[property_name]['x edges']} ")
-        if (corsika_histograms_instance._dict_2D_distributions[property_name]["y edges"] is not
-                u.dimensionless_unscaled):
+                f"{histograms_instance.dict_2d_distributions[property_name]['x bin edges']} "
+            )
+        if (
+            histograms_instance.dict_2d_distributions[property_name]["y axis unit"]
+            is not u.dimensionless_unscaled
+        ):
             ax.set_ylabel(
-                f"{corsika_histograms_instance._dict_2D_distributions[property_name]['y edges']} "
-                f"({corsika_histograms_instance._dict_2D_distributions[property_name]['y edges unit']})"
+                f"{histograms_instance.dict_2d_distributions[property_name]['y bin edges']} "
+                f"({histograms_instance.dict_2d_distributions[property_name]['y axis unit']})"
             )
         else:
             ax.set_ylabel(
-                f"{corsika_histograms_instance._dict_2D_distributions[property_name]['y edges']} ")
-        ax.set_xlim(np.amin(x_edges[i_hist]), np.amax(x_edges[i_hist]))
-        ax.set_ylim(np.amin(y_edges[i_hist]), np.amax(y_edges[i_hist]))
+                f"{histograms_instance.dict_2d_distributions[property_name]['y bin edges']} "
+            )
+        ax.set_xlim(np.amin(x_bin_edges[i_hist]), np.amax(x_bin_edges[i_hist]))
+        ax.set_ylim(np.amin(y_bin_edges[i_hist]), np.amax(y_bin_edges[i_hist]))
         ax.set_facecolor("black")
         fig.colorbar(mesh)
         all_figs.append(fig)
-        if corsika_histograms_instance.individual_telescopes is False:
-            fig_names.append(
-                f"{corsika_histograms_instance._dict_2D_distributions[property_name]['file name']}"
-                f"_all_tels.png"
+        if histograms_instance.individual_telescopes is False:
+            ax.set_title(
+                f"{histograms_instance.dict_2d_distributions[property_name]['file name']}"
+                "_all_tels"
             )
         else:
             ax.text(
@@ -98,22 +105,22 @@ def _kernel_plot_2D_photons(corsika_histograms_instance, property_name, log_z=Fa
                 transform=ax.transAxes,
                 color="white",
             )
-            fig_names.append(
-                f"{corsika_histograms_instance._dict_2D_distributions[property_name]['file name']}"
-                f"_tel_index_{corsika_histograms_instance.telescope_indices[i_hist]}.png",
+            ax.set_title(
+                f"{histograms_instance.dict_2d_distributions[property_name]['file name']}"
+                f"_tel_index_{histograms_instance.telescope_indices[i_hist]}",
             )
         plt.close()
 
-    return all_figs, fig_names
+    return all_figs
 
 
-def plot_2D_counts(corsika_histograms_instance, log_z=True):
+def plot_2d_counts(histograms_instance, log_z=True):
     """
     Plot the 2D histogram of the photon positions on the ground.
 
     Parameters
     ----------
-    corsika_histograms_instance: corsika.corsika_histograms.CorsikaHistograms
+    histograms_instance: corsika.corsika_histograms.CorsikaHistograms
         instance of corsika.corsika_histograms.CorsikaHistograms.
     log_z: bool
         if True, the intensity of the color bar is given in logarithmic scale.
@@ -122,20 +129,17 @@ def plot_2D_counts(corsika_histograms_instance, log_z=True):
     -------
     list
         List of figures for the given telescopes.
-    list
-        List of the figure names.
-
     """
-    return _kernel_plot_2D_photons(corsika_histograms_instance, "counts", log_z=log_z)
+    return _kernel_plot_2d_photons(histograms_instance, "counts", log_z=log_z)
 
 
-def plot_2D_density(corsika_histograms_instance, log_z=True):
+def plot_2d_density(histograms_instance, log_z=True):
     """
     Plot the 2D histogram of the photon density distribution on the ground.
 
     Parameters
     ----------
-    corsika_histograms_instance: corsika.corsika_histograms.CorsikaHistograms
+    histograms_instance: corsika.corsika_histograms.CorsikaHistograms
         instance of corsika.corsika_histograms.CorsikaHistograms.
     log_z: bool
         if True, the intensity of the color bar is given in logarithmic scale.
@@ -144,20 +148,18 @@ def plot_2D_density(corsika_histograms_instance, log_z=True):
     -------
     list
         List of figures for the given telescopes.
-    list
-        List of the figure names.
 
     """
-    return _kernel_plot_2D_photons(corsika_histograms_instance, "density", log_z=log_z)
+    return _kernel_plot_2d_photons(histograms_instance, "density", log_z=log_z)
 
 
-def plot_2D_direction(corsika_histograms_instance, log_z=True):
+def plot_2d_direction(histograms_instance, log_z=True):
     """
     Plot the 2D histogram of the incoming direction of photons.
 
     Parameters
     ----------
-    corsika_histograms_instance: corsika.corsika_histograms.CorsikaHistograms
+    histograms_instance: corsika.corsika_histograms.CorsikaHistograms
         instance of corsika.corsika_histograms.CorsikaHistograms.
     log_z: bool
         if True, the intensity of the color bar is given in logarithmic scale.
@@ -166,20 +168,18 @@ def plot_2D_direction(corsika_histograms_instance, log_z=True):
     -------
     list
         List of figures for the given telescopes.
-    list
-        List of the figure names.
 
     """
-    return _kernel_plot_2D_photons(corsika_histograms_instance, "direction", log_z=log_z)
+    return _kernel_plot_2d_photons(histograms_instance, "direction", log_z=log_z)
 
 
-def plot_2D_time_altitude(corsika_histograms_instance, log_z=True):
+def plot_2d_time_altitude(histograms_instance, log_z=True):
     """
     Plot the 2D histogram of the time and altitude where the photon was produced.
 
     Parameters
     ----------
-    corsika_histograms_instance: corsika.corsika_histograms.CorsikaHistograms
+    histograms_instance: corsika.corsika_histograms.CorsikaHistograms
         instance of corsika.corsika_histograms.CorsikaHistograms.
     log_z: bool
         if True, the intensity of the color bar is given in logarithmic scale.
@@ -188,20 +188,18 @@ def plot_2D_time_altitude(corsika_histograms_instance, log_z=True):
     -------
     list
         List of figures for the given telescopes.
-    list
-        List of the figure names.
 
     """
-    return _kernel_plot_2D_photons(corsika_histograms_instance, "time_altitude", log_z=log_z)
+    return _kernel_plot_2d_photons(histograms_instance, "time_altitude", log_z=log_z)
 
 
-def plot_2D_num_photons_per_telescope(corsika_histograms_instance, log_z=True):
+def plot_2d_num_photons_per_telescope(histograms_instance, log_z=True):
     """
     Plot the 2D histogram of the number of photons per event and per telescope.
 
     Parameters
     ----------
-    corsika_histograms_instance: corsika.corsika_histograms.CorsikaHistograms
+    histograms_instance: corsika.corsika_histograms.CorsikaHistograms
         instance of corsika.corsika_histograms.CorsikaHistograms.
     log_z: bool
         if True, the intensity of the color bar is given in logarithmic scale.
@@ -210,22 +208,18 @@ def plot_2D_num_photons_per_telescope(corsika_histograms_instance, log_z=True):
     -------
     list
         List of figures for the given telescopes.
-    list
-        List of the figure names.
 
     """
-    return _kernel_plot_2D_photons(
-        corsika_histograms_instance, "num_photons_per_telescope", log_z=log_z
-    )
+    return _kernel_plot_2d_photons(histograms_instance, "num_photons_per_telescope", log_z=log_z)
 
 
-def _kernel_plot_1D_photons(corsika_histograms_instance, property_name, log_y=True):
+def _kernel_plot_1d_photons(histograms_instance, property_name, log_y=True):
     """
     Create the figure of a 1D plot. The parameter `property` indicate which plot.
 
     Parameters
     ----------
-    corsika_histograms_instance: corsika.corsika_histograms.CorsikaHistograms
+    histograms_instance: corsika.corsika_histograms.CorsikaHistograms
         instance of corsika.corsika_histograms.CorsikaHistograms.
     property_name: string
         Name of the quantity. Choices are
@@ -238,71 +232,78 @@ def _kernel_plot_1D_photons(corsika_histograms_instance, property_name, log_y=Tr
     -------
     list
         List of figures for the given telescopes.
-    list
-        List of the figure names.
 
     Raises
     ------
     ValueError
         if `property` is not allowed.
     """
-    if property_name not in corsika_histograms_instance._dict_1D_distributions:
-        msg = (f"This property does not exist. The valid entries are "
-               f"{corsika_histograms_instance._dict_1D_distributions}")
+    if property_name not in histograms_instance.dict_1d_distributions:
+        msg = (
+            f"This property does not exist. The valid entries are "
+            f"{histograms_instance.dict_1d_distributions}"
+        )
         _logger.error(msg)
         raise ValueError
 
     function = getattr(
-        corsika_histograms_instance,
-        corsika_histograms_instance._dict_1D_distributions[property_name]["function"],
+        histograms_instance,
+        histograms_instance.dict_1d_distributions[property_name]["function"],
     )
-    hist_values, edges = function()
+    hist_values, bin_edges = function()
     all_figs = []
-    fig_names = []
-    for i_hist, _ in enumerate(edges):
+    for i_hist, _ in enumerate(bin_edges):
         fig, ax = plt.subplots()
         ax.bar(
-            edges[i_hist][:-1],
+            bin_edges[i_hist][:-1],
             hist_values[i_hist],
             align="edge",
-            width=np.abs(np.diff(edges[i_hist])),
+            width=np.abs(np.diff(bin_edges[i_hist])),
         )
         if (
-                corsika_histograms_instance._dict_1D_distributions[property_name]["edges unit"]
-                is not u.dimensionless_unscaled
+            histograms_instance.dict_1d_distributions[property_name]["axis unit"]
+            is not u.dimensionless_unscaled
         ):
             ax.set_xlabel(
-                f"{corsika_histograms_instance._dict_1D_distributions[property_name]['edges']} "
-                f"({corsika_histograms_instance._dict_1D_distributions[property_name]['edges unit']})"
+                f"{histograms_instance.dict_1d_distributions[property_name]['bin edges']} "
+                f"({histograms_instance.dict_1d_distributions[property_name]['axis unit']})"
             )
         else:
             ax.set_xlabel(
-                f"{corsika_histograms_instance._dict_1D_distributions[property_name]['edges']} ")
-        ax.set_ylabel("Counts")
+                f"{histograms_instance.dict_1d_distributions[property_name]['bin edges']} "
+            )
+        if property_name == "density":
+            ax.set_ylabel(
+                f"Density ({histograms_instance.dict_1d_distributions[property_name]['axis unit']}"
+                r"$^{-2}$)"
+            )
+        else:
+            ax.set_ylabel("Counts")
 
         if log_y is True:
             ax.set_yscale("log")
-        if corsika_histograms_instance.individual_telescopes is False:
-            fig_names.append(
-                f"{corsika_histograms_instance._dict_1D_distributions[property_name]['file name']}"
-                f"_all_tels.png"
+        if histograms_instance.individual_telescopes is False:
+            ax.set_title(
+                f"{histograms_instance.dict_1d_distributions[property_name]['file name']}"
+                "_all_tels"
             )
         else:
-            fig_names.append(
-                f"{corsika_histograms_instance._dict_1D_distributions[property_name]['file name']}"
-                f"_tel_index_{corsika_histograms_instance.telescope_indices[i_hist]}.png",
+            ax.set_title(
+                f"{histograms_instance.dict_1d_distributions[property_name]['file name']}"
+                f"_tel_index_{histograms_instance.telescope_indices[i_hist]}",
             )
         all_figs.append(fig)
-    return all_figs, fig_names
+        plt.close(fig)
+    return all_figs
 
 
-def plot_wavelength_distr(corsika_histograms_instance, log_y=True):
+def plot_wavelength_distr(histograms_instance, log_y=True):
     """
     Plots the 1D distribution of the photon wavelengths
 
     Parameters
     ----------
-    corsika_histograms_instance: corsika.corsika_histograms.CorsikaHistograms
+    histograms_instance: corsika.corsika_histograms.CorsikaHistograms
         instance of corsika.corsika_histograms.CorsikaHistograms.
     log_y: bool
         if True, the intensity of the Y axis is given in logarithmic scale.
@@ -311,19 +312,17 @@ def plot_wavelength_distr(corsika_histograms_instance, log_y=True):
     -------
     list
         List of figures for the given telescopes.
-    list
-        List of the figure names.
     """
-    return _kernel_plot_1D_photons(corsika_histograms_instance, "wavelength", log_y=log_y)
+    return _kernel_plot_1d_photons(histograms_instance, "wavelength", log_y=log_y)
 
 
-def plot_counts_distr(corsika_histograms_instance, log_y=True):
+def plot_counts_distr(histograms_instance, log_y=True):
     """
     Plots the 1D distribution, i.e. the radial distribution, of the photons on the ground.
 
     Parameters
     ----------
-    corsika_histograms_instance: corsika.corsika_histograms.CorsikaHistograms
+    histograms_instance: corsika.corsika_histograms.CorsikaHistograms
         instance of corsika.corsika_histograms.CorsikaHistograms.
     log_y: bool
         if True, the intensity of the Y axis is given in logarithmic scale.
@@ -332,19 +331,17 @@ def plot_counts_distr(corsika_histograms_instance, log_y=True):
     -------
     list
         List of figures for the given telescopes.
-    list
-        List of the figure names.
     """
-    return _kernel_plot_1D_photons(corsika_histograms_instance, "counts", log_y=log_y)
+    return _kernel_plot_1d_photons(histograms_instance, "counts", log_y=log_y)
 
 
-def plot_density_distr(corsika_histograms_instance, log_y=True):
+def plot_density_distr(histograms_instance, log_y=True):
     """
     Plots the photon density distribution on the ground.
 
     Parameters
     ----------
-    corsika_histograms_instance: corsika.corsika_histograms.CorsikaHistograms
+    histograms_instance: corsika.corsika_histograms.CorsikaHistograms
         instance of corsika.corsika_histograms.CorsikaHistograms.
     log_y: bool
         if True, the intensity of the Y axis is given in logarithmic scale.
@@ -353,19 +350,17 @@ def plot_density_distr(corsika_histograms_instance, log_y=True):
     -------
     list
         List of figures for the given telescopes.
-    list
-        List of the figure names.
     """
-    return _kernel_plot_1D_photons(corsika_histograms_instance, "density", log_y=log_y)
+    return _kernel_plot_1d_photons(histograms_instance, "density", log_y=log_y)
 
 
-def plot_time_distr(corsika_histograms_instance, log_y=True):
+def plot_time_distr(histograms_instance, log_y=True):
     """
     Plots the distribution times in which the photons were generated in ns.
 
     Parameters
     ----------
-    corsika_histograms_instance: corsika.corsika_histograms.CorsikaHistograms
+    histograms_instance: corsika.corsika_histograms.CorsikaHistograms
         instance of corsika.corsika_histograms.CorsikaHistograms.
     log_y: bool
         if True, the intensity of the Y axis is given in logarithmic scale.
@@ -374,19 +369,17 @@ def plot_time_distr(corsika_histograms_instance, log_y=True):
     -------
     list
         List of figures for the given telescopes.
-    list
-        List of the figure names.
     """
-    return _kernel_plot_1D_photons(corsika_histograms_instance, "time", log_y=log_y)
+    return _kernel_plot_1d_photons(histograms_instance, "time", log_y=log_y)
 
 
-def plot_altitude_distr(corsika_histograms_instance, log_y=True):
+def plot_altitude_distr(histograms_instance, log_y=True):
     """
     Plots the distribution of altitude in which the photons were generated in km.
 
     Parameters
     ----------
-    corsika_histograms_instance: corsika.corsika_histograms.CorsikaHistograms
+    histograms_instance: corsika.corsika_histograms.CorsikaHistograms
         instance of corsika.corsika_histograms.CorsikaHistograms.
     log_y: bool
         if True, the intensity of the Y axis is given in logarithmic scale.
@@ -395,19 +388,17 @@ def plot_altitude_distr(corsika_histograms_instance, log_y=True):
     -------
     list
         List of figures for the given telescopes.
-    list
-        List of the figure names.
     """
-    return _kernel_plot_1D_photons(corsika_histograms_instance, "altitude", log_y=log_y)
+    return _kernel_plot_1d_photons(histograms_instance, "altitude", log_y=log_y)
 
 
-def plot_photon_per_event_distr(corsika_histograms_instance, log_y=True):
+def plot_photon_per_event_distr(histograms_instance, log_y=True):
     """
     Plots the distribution of the number of Cherenkov photons per event.
 
     Parameters
     ----------
-    corsika_histograms_instance: corsika.corsika_histograms.CorsikaHistograms
+    histograms_instance: corsika.corsika_histograms.CorsikaHistograms
         instance of corsika.corsika_histograms.CorsikaHistograms.
     log_y: bool
         if True, the intensity of the Y axis is given in logarithmic scale.
@@ -416,22 +407,19 @@ def plot_photon_per_event_distr(corsika_histograms_instance, log_y=True):
     -------
     list
         List of figures for the given telescopes.
-    list
-        List of the figure names.
 
     """
 
-    return _kernel_plot_1D_photons(corsika_histograms_instance, "num_photons_per_event",
-                                   log_y=log_y)
+    return _kernel_plot_1d_photons(histograms_instance, "num_photons_per_event", log_y=log_y)
 
 
-def plot_photon_per_telescope_distr(corsika_histograms_instance, log_y=True):
+def plot_photon_per_telescope_distr(histograms_instance, log_y=True):
     """
     Plots the distribution of the number of Cherenkov photons per telescope.
 
     Parameters
     ----------
-    corsika_histograms_instance: corsika.corsika_histograms.CorsikaHistograms
+    histograms_instance: corsika.corsika_histograms.CorsikaHistograms
         instance of corsika.corsika_histograms.CorsikaHistograms.
     log_y: bool
         if True, the intensity of the Y axis is given in logarithmic scale.
@@ -440,25 +428,21 @@ def plot_photon_per_telescope_distr(corsika_histograms_instance, log_y=True):
     -------
     list
         List of figures for the given telescopes.
-    list
-        List of the figure names.
 
     """
 
-    return _kernel_plot_1D_photons(
-        corsika_histograms_instance, "num_photons_per_telescope", log_y=log_y
-    )
+    return _kernel_plot_1d_photons(histograms_instance, "num_photons_per_telescope", log_y=log_y)
 
 
-def plot_1D_event_header_distribution(
-        corsika_histograms_instance, event_header_element, log_y=True, bins=50, hist_range=None
+def plot_1d_event_header_distribution(
+    histograms_instance, event_header_element, log_y=True, bins=50, hist_range=None
 ):
     """
     Plots the distribution of the quantity given by .
 
     Parameters
     ----------
-    corsika_histograms_instance: corsika.corsika_histograms.CorsikaHistograms
+    histograms_instance: corsika.corsika_histograms.CorsikaHistograms
         instance of corsika.corsika_histograms.CorsikaHistograms.
     event_header_element: str
         The key to the CORSIKA event header element.
@@ -473,27 +457,25 @@ def plot_1D_event_header_distribution(
     -------
     list
         List of figures for the given telescopes.
-    list
-        List of the figure names.
 
     """
-    hist_values, edges = corsika_histograms_instance.event_1D_histogram(
+    hist_values, bin_edges = histograms_instance.event_1d_histogram(
         event_header_element, bins=bins, hist_range=hist_range
     )
     fig, ax = plt.subplots()
     ax.bar(
-        edges[:-1],
+        bin_edges[:-1],
         hist_values,
         align="edge",
-        width=np.abs(np.diff(edges)),
+        width=np.abs(np.diff(bin_edges)),
     )
     if (
-            corsika_histograms_instance.event_information[
-                event_header_element].unit is not u.dimensionless_unscaled
+        histograms_instance.event_information[event_header_element].unit
+        is not u.dimensionless_unscaled
     ):
         ax.set_xlabel(
             f"{event_header_element} ("
-            f"{corsika_histograms_instance.event_information[event_header_element].unit})"
+            f"{histograms_instance.event_information[event_header_element].unit})"
         )
     else:
         ax.set_xlabel(f"{event_header_element}")
@@ -501,24 +483,24 @@ def plot_1D_event_header_distribution(
 
     if log_y is True:
         ax.set_yscale("log")
-    fig_name = f"hist_1D_{event_header_element}"
-    return fig, fig_name
+    ax.set_title(f"hist_1d_{event_header_element}")
+    return fig
 
 
-def plot_2D_event_header_distribution(
-        corsika_histograms_instance,
-        event_header_element_1,
-        event_header_element_2,
-        log_z=True,
-        bins=50,
-        hist_range=None,
+def plot_2d_event_header_distribution(
+    histograms_instance,
+    event_header_element_1,
+    event_header_element_2,
+    log_z=True,
+    bins=50,
+    hist_range=None,
 ):
     """
     Plots the distribution of the quantity given by .
 
     Parameters
     ----------
-    corsika_histograms_instance: corsika.corsika_histograms.CorsikaHistograms
+    histograms_instance: corsika.corsika_histograms.CorsikaHistograms
         instance of corsika.corsika_histograms.CorsikaHistograms.
     event_header_element_1: str
         The first key to the CORSIKA event header element
@@ -535,11 +517,9 @@ def plot_2D_event_header_distribution(
     -------
     list
         List of figures for the given telescopes.
-    list
-        List of the figure names.
 
     """
-    hist_values, x_edges, y_edges = corsika_histograms_instance.event_2D_histogram(
+    hist_values, x_bin_edges, y_bin_edges = histograms_instance.event_2d_histogram(
         event_header_element_1, event_header_element_2, bins=bins, hist_range=hist_range
     )
     fig, ax = plt.subplots()
@@ -547,29 +527,48 @@ def plot_2D_event_header_distribution(
         norm = colors.LogNorm(vmin=1, vmax=np.amax([np.amax(hist_values), 2]))
     else:
         norm = None
-    mesh = ax.pcolormesh(x_edges, y_edges, hist_values, norm=norm)
+    mesh = ax.pcolormesh(x_bin_edges, y_bin_edges, hist_values, norm=norm)
 
     if (
-            corsika_histograms_instance.event_information[event_header_element_1].unit
-            is not u.dimensionless_unscaled
+        histograms_instance.event_information[event_header_element_1].unit
+        is not u.dimensionless_unscaled
     ):
         ax.set_xlabel(
             f"{event_header_element_1} ("
-            f"{corsika_histograms_instance.event_information[event_header_element_1].unit})"
+            f"{histograms_instance.event_information[event_header_element_1].unit})"
         )
     else:
         ax.set_xlabel(f"{event_header_element_2}")
     if (
-            corsika_histograms_instance.event_information[event_header_element_2].unit
-            is not u.dimensionless_unscaled
+        histograms_instance.event_information[event_header_element_2].unit
+        is not u.dimensionless_unscaled
     ):
         ax.set_ylabel(
             f"{event_header_element_2} "
-            f"({corsika_histograms_instance.event_information[event_header_element_2].unit})")
+            f"({histograms_instance.event_information[event_header_element_2].unit})"
+        )
     else:
         ax.set_ylabel(f"{event_header_element_2}")
 
     ax.set_facecolor("black")
+    ax.set_title(f"hist_2d_{event_header_element_1}_{event_header_element_2}")
     fig.colorbar(mesh)
-    fig_name = f"hist_2D_{event_header_element_1}_{event_header_element_2}"
-    return fig, fig_name
+    return fig
+
+
+def save_figs_to_pdf(figs, pdf_file_name):
+    """
+    Save figures from corsika histograms to an output pdf file.
+
+    Parameters
+    ----------
+    figs: list or numpy.array
+        List with the figures output by `corsika_output_visualize.py`.
+    pdf_file_name: str or Path
+        Name of the pdf file.
+    """
+    pdf_pages = PdfPages(Path(pdf_file_name).absolute().as_posix())
+    for fig in figs:
+        plt.tight_layout()
+        pdf_pages.savefig(fig)
+    pdf_pages.close()