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

simtools/corsika/corsika_histograms.py

@@ -1,3 +1,5 @@
+"""Extract Cherenkov photons information from a CORSIKA IACT file."""
+
 import functools
 import logging
 import operator
@@ -17,18 +19,22 @@ from eventio import IACTFile
 from simtools import version
 from simtools.io_operations import io_handler
 from simtools.io_operations.hdf5_handler import fill_hdf5_table
-from simtools.utils.general import collect_data_from_file_or_dict
+from simtools.utils.general import collect_data_from_file
 from simtools.utils.geometry import convert_2d_to_radial_distr, rotate
 from simtools.utils.names import sanitize_name
 
+X_AXIS_STRING = "x axis"
+Y_AXIS_STRING = "y axis"
+Z_AXIS_STRING = "z axis"
+
 
-class HistogramNotCreated(Exception):
+class HistogramNotCreatedError(Exception):
     """Exception for histogram not created."""
 
 
 class CorsikaHistograms:
-    """CorsikaHistograms extracts the Cherenkov photons information from a CORSIKA IACT file
-     using pyeventio.
+    """
+    Extracts the Cherenkov photons information from a CORSIKA IACT file.
 
     Parameters
     ----------
@@ -109,15 +115,16 @@ class CorsikaHistograms:
     def hdf5_file_name(self):
         """
         Property for the hdf5 file name.
+
         The idea of this property is to allow setting (or changing) the name of the hdf5 file
-         even after creating the class instance.
+        even after creating the class instance.
         """
         return self._hdf5_file_name
 
     @hdf5_file_name.setter
     def hdf5_file_name(self, hdf5_file_name):
         """
-        Sets the hdf5_file_name to the argument passed.
+        Set the hdf5_file_name to the argument passed.
 
         Parameters
         ----------
@@ -134,9 +141,8 @@ class CorsikaHistograms:
         Returns
         -------
         float:
-            The version of CORSIKA used to produce the CORSIKA IACT file given by `self.input_file`.
+            The version of CORSIKA used to produce the CORSIKA IACT file given by self.input_file.
         """
-
         if self._corsika_version is None:
             all_corsika_versions = list(run_header.run_header_types.keys())
             header = list(self.iact_file.header)
@@ -158,9 +164,7 @@ class CorsikaHistograms:
         return self._corsika_version
 
     def _initialize_header(self):
-        """
-        Initialize the header.
-        """
+        """Initialize the header."""
         self.all_run_keys = list(
             run_header.run_header_types[np.around(self.corsika_version, 1)].names
         )
@@ -192,11 +196,11 @@ class CorsikaHistograms:
     def read_event_information(self):
         """
         Read the information about the events from their headers and save as a class instance.
+
         The main information can also be fetched individually through the functions below.
         For the remaining information (such as px, py, pz), use this function.
 
         """
-
         if self.event_information is None:
             with IACTFile(self.input_file) as self.iact_file:
                 self.telescope_positions = np.array(self.iact_file.telescope_positions)
@@ -246,7 +250,6 @@ class CorsikaHistograms:
         dict:
             A dictionary with the astropy units.
         """
-
         # Build a dictionary with astropy units for the unit of the event's (header's) parameters.
         all_event_astropy_units = {}
         for key in parameters[1:]:  # starting at the second
@@ -277,6 +280,7 @@ class CorsikaHistograms:
     def telescope_indices(self, telescope_new_indices):
         """
         Set the telescope index (or indices).
+
         If self.individual_telescopes is True, the indices of the telescopes passed are analyzed
         individually (different histograms for each telescope) even if all telescopes are listed.
 
@@ -292,14 +296,13 @@ class CorsikaHistograms:
         TypeError:
             if the indices passed through telescope_index are not of type int.
         """
-
         if telescope_new_indices is None:
             self._telescope_indices = np.arange(self.num_telescopes)
         else:
-            if not isinstance(telescope_new_indices, (list, np.ndarray)):
+            if not isinstance(telescope_new_indices, list | np.ndarray):
                 telescope_new_indices = np.array([telescope_new_indices])
             for i_telescope in telescope_new_indices:
-                if not isinstance(i_telescope, (int, np.integer)):
+                if not isinstance(i_telescope, int | np.integer):
                     msg = "The index or indices given are not of type int."
                     self._logger.error(msg)
                     raise TypeError
@@ -328,6 +331,7 @@ class CorsikaHistograms:
     def hist_config(self, input_config):
         """
         Set the configuration for the histograms (e.g., bin size, min and max values, etc).
+
         The input is allowed either through a yaml file or a dictionary. If nothing is given,
         the dictionary is created with default values.
 
@@ -335,16 +339,14 @@ class CorsikaHistograms:
         ----------
         input_config: str, Path, dict or NoneType
             yaml file with the configuration parameters to create the histograms. For the correct
-            format, please look at the docstring at `_create_histogram_default_config`.
+            format, please look at the docstring at _create_histogram_default_config.
             Alternatively, it can be a dictionary with the configuration parameters to create
             the histograms.
         """
-        input_dict, input_yaml = None, None
         if isinstance(input_config, dict):
-            input_dict = input_config
+            self._hist_config = input_config
         else:
-            input_yaml = input_config
-        self._hist_config = collect_data_from_file_or_dict(input_yaml, input_dict, allow_empty=True)
+            self._hist_config = collect_data_from_file(input_config) if input_config else None
 
     def hist_config_to_yaml(self, file_name=None):
         """
@@ -356,7 +358,6 @@ class CorsikaHistograms:
             Name of the output file, in which to save the histogram configuration.
 
         """
-
         if file_name is None:
             file_name = "hist_config"
         file_name = Path(file_name).with_suffix(".yml")
@@ -366,8 +367,9 @@ class CorsikaHistograms:
 
     def _create_histogram_default_config(self):
         """
-        Create a dictionary with the configuration necessary to create the histograms. It is used
-        only in case the configuration is not provided in a yaml file or dict.
+        Create a dictionary with the configuration necessary to create the histograms.
+
+        It is used only in case the configuration is not provided in a yaml file or dict.
 
         Three histograms are created: hist_position with 3 dimensions (x, y positions and the
         wavelength), hist_direction with 2 dimensions (direction cosines in x and y directions),
@@ -387,7 +389,6 @@ class CorsikaHistograms:
         dict:
             Dictionary with the configuration parameters to create the histograms.
         """
-
         if self.individual_telescopes is False:
             xy_maximum = 1000 * u.m
             xy_bin = 100
@@ -395,22 +396,21 @@ class CorsikaHistograms:
         else:
             xy_maximum = 16 * u.m
             xy_bin = 64
-
-        histogram_config = {
+        return {
             "hist_position": {
-                "x axis": {
+                X_AXIS_STRING: {
                     "bins": xy_bin,
                     "start": -xy_maximum,
                     "stop": xy_maximum,
                     "scale": "linear",
                 },
-                "y axis": {
+                Y_AXIS_STRING: {
                     "bins": xy_bin,
                     "start": -xy_maximum,
                     "stop": xy_maximum,
                     "scale": "linear",
                 },
-                "z axis": {
+                Z_AXIS_STRING: {
                     "bins": 80,
                     "start": 200 * u.nm,
                     "stop": 1000 * u.nm,
@@ -418,13 +418,13 @@ class CorsikaHistograms:
                 },
             },
             "hist_direction": {
-                "x axis": {
+                X_AXIS_STRING: {
                     "bins": 100,
                     "start": -1,
                     "stop": 1,
                     "scale": "linear",
                 },
-                "y axis": {
+                Y_AXIS_STRING: {
                     "bins": 100,
                     "start": -1,
                     "stop": 1,
@@ -432,20 +432,24 @@ class CorsikaHistograms:
                 },
             },
             "hist_time_altitude": {
-                "x axis": {
+                X_AXIS_STRING: {
                     "bins": 100,
                     "start": -2000 * u.ns,
                     "stop": 2000 * u.ns,
                     "scale": "linear",
                 },
-                "y axis": {"bins": 100, "start": 120 * u.km, "stop": 0 * u.km, "scale": "linear"},
+                Y_AXIS_STRING: {
+                    "bins": 100,
+                    "start": 120 * u.km,
+                    "stop": 0 * u.km,
+                    "scale": "linear",
+                },
             },
         }
-        return histogram_config
 
     def _create_regular_axes(self, label):
         """
-        Helper function to _create_histograms.
+        Create regular axis for histograms.
 
         Parameters
         ----------
@@ -462,11 +466,11 @@ class CorsikaHistograms:
         if label not in self._allowed_histograms:
             msg = f"allowed labels must be one of the following: {self._allowed_histograms}"
             self._logger.error(msg)
-            raise ValueError
+            raise ValueError(msg)
 
-        all_axes = ["x axis", "y axis"]
+        all_axes = [X_AXIS_STRING, Y_AXIS_STRING]
         if label == "hist_position":
-            all_axes.append("z axis")
+            all_axes.append(Z_AXIS_STRING)
 
         boost_axes = []
         for axis in all_axes:
@@ -527,10 +531,11 @@ class CorsikaHistograms:
             )
 
     def _fill_histograms(self, photons, rotation_around_z_axis=None, rotation_around_y_axis=None):
-        """Fill all the histograms created by self._create_histogram with the information of the
-         photons on the ground.
-         If the azimuth and zenith angles are provided, the Cherenkov photon's coordinates are
-         filled in the plane perpendicular to the incoming direction of the particle.
+        """
+        Fill histograms with the information of the photons on the ground.
+
+        if the azimuth and zenith angles are provided, the Cherenkov photon's coordinates are
+        filled in the plane perpendicular to the incoming direction of the particle.
 
         Parameters
         ----------
@@ -545,7 +550,7 @@ class CorsikaHistograms:
              cy: direction cosine in the y direction, i.e., the cosine of the angle between the
              incoming direction and the y axis,
              time: time of arrival of the photon in ns. The clock starts when the particle crosses
-             the top of the atmosphere (CORSIKA-defined) if `self.event_first_interaction_heights`
+             the top of the atmosphere (CORSIKA-defined) if self.event_first_interaction_heights
              is positive or at first interaction if otherwise.
              zem: altitude where the photon was generated in cm,
              photons: number of photons associated to this bunch,
@@ -567,7 +572,6 @@ class CorsikaHistograms:
         IndexError:
             If the index or indices passed though telescope_index are out of range.
         """
-
         hist_num = 0
         for i_tel_info, photons_info in np.array(
             list(zip(self.telescope_positions, photons)), dtype=object
@@ -603,8 +607,7 @@ class CorsikaHistograms:
 
     def set_histograms(self, telescope_indices=None, individual_telescopes=None, hist_config=None):
         """
-        Extract the information of the Cherenkov photons from a CORSIKA output IACT file, create
-         and fill the histograms
+        Create and fill Cherenkov photons histograms using information from the CORSIKA IACT file.
 
         Parameters
         ----------
@@ -615,7 +618,7 @@ class CorsikaHistograms:
             if True, one histogram is set for each telescope separately.
         hist_config:
             yaml file with the configuration parameters to create the histograms. For the correct
-            format, please look at the docstring of `_create_histogram_default_config`.
+            format, please look at the docstring of _create_histogram_default_config.
             Alternatively, it can be a dictionary with the configuration parameters to create
             the histograms.
 
@@ -663,16 +666,13 @@ class CorsikaHistograms:
 
     @property
     def individual_telescopes(self):
-        """
-        Return the individual telescopes as property.
-        """
+        """Return the individual telescopes as property."""
         return self._individual_telescopes
 
     @individual_telescopes.setter
     def individual_telescopes(self, new_individual_telescopes: bool):
         """
-        The following lines allow `individual_telescopes` to be defined before using this function
-        but if any parameter is passed in this function, it overwrites the class attribute.
+        Define individual telescopes.
 
         Parameters
         ----------
@@ -680,7 +680,6 @@ class CorsikaHistograms:
             if False, the histograms are supposed to be filled for all telescopes.
             if True, one histogram is set for each telescope sepparately.
         """
-
         if new_individual_telescopes is None:
             self._individual_telescopes = False
         else:
@@ -692,22 +691,21 @@ class CorsikaHistograms:
 
         Raises
         ------
-        HistogramNotCreated:
+        HistogramNotCreatedError:
             if the histogram was not previously created.
         """
-
         for histogram in self._allowed_histograms:
             if not hasattr(self, histogram) or getattr(self, histogram) is None:
                 msg = (
-                    "The histograms were not created. Please, use `create_histograms` to create "
+                    "The histograms were not created. Please, use create_histograms to create "
                     "histograms from the CORSIKA output file."
                 )
                 self._logger.error(msg)
-                raise HistogramNotCreated
+                raise HistogramNotCreatedError
 
     def _get_hist_2d_projection(self, label):
         """
-        Helper function to get 2D distributions.
+        Get 2D distributions.
 
         Parameters
         ----------
@@ -728,11 +726,10 @@ class CorsikaHistograms:
         ValueError:
             if label is not valid.
         """
-
         if label not in self._allowed_2d_labels:
             msg = f"label is not valid. Valid entries are {self._allowed_2d_labels}"
             self._logger.error(msg)
-            raise ValueError
+            raise ValueError(msg)
         self._raise_if_no_histogram()
 
         num_telescopes_to_fill = (
@@ -741,6 +738,7 @@ class CorsikaHistograms:
 
         x_bin_edges, y_bin_edges, hist_values = [], [], []
         for i_telescope in range(num_telescopes_to_fill):
+            mini_hist = None
             if label == "counts":
                 mini_hist = self.hist_position[i_telescope][:, :, sum]
                 hist_values.append(mini_hist.view().T)
@@ -776,8 +774,9 @@ class CorsikaHistograms:
 
     def get_2d_photon_density_distr(self):
         """
-        Get 2D histograms of position of the Cherenkov photons on the ground. It returns the photon
-        density per square meter.
+        Get 2D histograms of position of the Cherenkov photons on the ground.
+
+        It returns the photon density per square meter.
 
         Returns
         -------
@@ -822,8 +821,10 @@ class CorsikaHistograms:
 
     def get_2d_num_photons_distr(self):
         """
-        Get the distribution of Cherenkov photons per event per telescope. It returns the 2D array
-        accounting for the events from the telescopes given by `self.telescope_indices`.
+        Get the distribution of Cherenkov photons per event per telescope.
+
+        It returns the 2D array accounting for the events from the telescopes given
+        by self.telescope_indices.
 
         Returns
         -------
@@ -846,7 +847,7 @@ class CorsikaHistograms:
 
     def _get_hist_1d_projection(self, label):
         """
-        Helper function to get 1D distributions.
+        Get 1D distributions.
 
         Parameters
         ----------
@@ -865,15 +866,15 @@ class CorsikaHistograms:
         ValueError:
             if label is not valid.
         """
-
         if label not in self._allowed_1d_labels:
             msg = f"{label} is not valid. Valid entries are {self._allowed_1d_labels}"
             self._logger.error(msg)
-            raise ValueError
+            raise ValueError(msg)
         self._raise_if_no_histogram()
 
         x_bin_edges_list, hist_1d_list = [], []
         for i_hist, _ in enumerate(self.hist_position):
+            mini_hist = None
             if label == "wavelength":
                 mini_hist = self.hist_position[i_hist][sum, sum, :]
             elif label == "time":
@@ -886,8 +887,8 @@ class CorsikaHistograms:
         return np.array(hist_1d_list), np.array(x_bin_edges_list)
 
     def _get_bins_max_dist(self, bins=None, max_dist=None):
-        """Auxiliary function to get the number of bins and the max distance to generate the
-        radial and the density histograms
+        """
+        Get the number of bins and the max distance to generate the radial and density histograms.
 
         Parameters
         ----------
@@ -896,21 +897,22 @@ class CorsikaHistograms:
         max_dist: float
             Maximum distance to consider in the 1D histogram (in meters).
         """
+        hist_position = "hist_position"
         if max_dist is None:
             max_dist = np.amax(
                 [
-                    self.hist_config["hist_position"]["x axis"]["start"].to(u.m).value,
-                    self.hist_config["hist_position"]["x axis"]["stop"].to(u.m).value,
-                    self.hist_config["hist_position"]["y axis"]["start"].to(u.m).value,
-                    self.hist_config["hist_position"]["y axis"]["stop"].to(u.m).value,
+                    self.hist_config[hist_position][X_AXIS_STRING]["start"].to(u.m).value,
+                    self.hist_config[hist_position][X_AXIS_STRING]["stop"].to(u.m).value,
+                    self.hist_config[hist_position][Y_AXIS_STRING]["start"].to(u.m).value,
+                    self.hist_config[hist_position][Y_AXIS_STRING]["stop"].to(u.m).value,
                 ]
             )
         if bins is None:
             bins = (
                 np.amax(
                     [
-                        self.hist_config["hist_position"]["x axis"]["bins"],
-                        self.hist_config["hist_position"]["y axis"]["bins"],
+                        self.hist_config[hist_position][X_AXIS_STRING]["bins"],
+                        self.hist_config[hist_position][Y_AXIS_STRING]["bins"],
                     ]
                 )
                 // 2
@@ -920,8 +922,7 @@ class CorsikaHistograms:
 
     def get_photon_radial_distr(self, bins=None, max_dist=None):
         """
-        Get the radial distribution of the photons on the ground in relation to the center of the
-        array.
+        Get the phton radial distribution on the ground in relation to the center of the array.
 
         Parameters
         ----------
@@ -938,7 +939,6 @@ class CorsikaHistograms:
             The bin edges of the 1D histogram in meters with size = int(max_dist/bin_size) + 1,
             usually in meter.
         """
-
         bins, max_dist = self._get_bins_max_dist(bins=bins, max_dist=max_dist)
         bin_edges_1d_list, hist_1d_list = [], []
 
@@ -958,8 +958,7 @@ class CorsikaHistograms:
 
     def get_photon_density_distr(self, bins=None, max_dist=None):
         """
-        Get the density distribution of the photons on the ground in relation to the center of the
-        array.
+        Get the photon density distribution on the ground in relation to the center of the array.
 
         Parameters
         ----------
@@ -1010,9 +1009,10 @@ class CorsikaHistograms:
 
     def get_photon_time_of_emission_distr(self):
         """
-        Get the distribution of the emitted time of the Cherenkov photons. The clock starts when the
-         particle crosses the top of the atmosphere (CORSIKA-defined) if
-         `self.event_first_interaction_heights` is positive or at first interaction if otherwise.
+        Get the distribution of the emitted time of the Cherenkov photons.
+
+        The clock starts when the particle crosses the top of the atmosphere (CORSIKA-defined) if
+        self.event_first_interaction_heights is positive or at first interaction if otherwise.
 
         Returns
         -------
@@ -1040,16 +1040,12 @@ class CorsikaHistograms:
 
     @property
     def num_photons_per_event_per_telescope(self):
-        """
-        The number of photons per event per telescope.
-        """
+        """The number of photons per event per telescope."""
         return self._num_photons_per_event_per_telescope
 
     @num_photons_per_event_per_telescope.setter
     def num_photons_per_event_per_telescope(self, num_photons_per_event_per_telescope_to_set):
-        """
-        Set the number of photons per event per telescope.
-        """
+        """Set the number of photons per event per telescope."""
         self._num_photons_per_event_per_telescope = (
             np.array(num_photons_per_event_per_telescope_to_set)
             .reshape(self.num_events, len(self.telescope_indices))
@@ -1059,8 +1055,9 @@ class CorsikaHistograms:
     @property
     def num_photons_per_event(self):
         """
-        Get the distribution of the number of photons amongst the events,
-         including the telescopes indicated by `self.telescope_indices`.
+        Get the the number of photons per events.
+
+        Includes the telescopes indicated by self.telescope_indices.
 
         Returns
         -------
@@ -1109,7 +1106,6 @@ class CorsikaHistograms:
         numpy.array
             Number of photons per telescope.
         """
-
         hist, bin_edges = np.histogram(self.num_photons_per_telescope, bins=bins, range=hist_range)
         return hist.reshape(1, bins), bin_edges.reshape(1, bins + 1)
 
@@ -1135,7 +1131,6 @@ class CorsikaHistograms:
         dict
             Meta dictionary for the hdf5 files with the histograms.
         """
-
         if self.__meta_dict is None:
             self.__meta_dict = {
                 "corsika_version": self.corsika_version,
@@ -1157,55 +1152,60 @@ class CorsikaHistograms:
         dict:
             The dictionary with information about the 1D distributions.
         """
+        fn_key = "function"
+        file_name = "file name"
+        title = "title"
+        bin_edges = "bin edges"
+        axis_unit = "axis unit"
         self._dict_1d_distributions = {
             "wavelength": {
-                "function": "get_photon_wavelength_distr",
-                "file name": "hist_1d_photon_wavelength_distr",
-                "title": "Photon wavelength distribution",
-                "bin edges": "wavelength",
-                "axis unit": self.hist_config["hist_position"]["z axis"]["start"].unit,
+                fn_key: "get_photon_wavelength_distr",
+                file_name: "hist_1d_photon_wavelength_distr",
+                title: "Photon wavelength distribution",
+                bin_edges: "wavelength",
+                axis_unit: self.hist_config["hist_position"][Z_AXIS_STRING]["start"].unit,
             },
             "counts": {
-                "function": "get_photon_radial_distr",
-                "file name": "hist_1d_photon_radial_distr",
-                "title": "Radial photon distribution on the ground",
-                "bin edges": "Distance to center",
-                "axis unit": self.hist_config["hist_position"]["x axis"]["start"].unit,
+                fn_key: "get_photon_radial_distr",
+                file_name: "hist_1d_photon_radial_distr",
+                title: "Radial photon distribution on the ground",
+                bin_edges: "Distance to center",
+                axis_unit: self.hist_config["hist_position"][X_AXIS_STRING]["start"].unit,
             },
             "density": {
-                "function": "get_photon_density_distr",
-                "file name": "hist_1d_photon_density_distr",
-                "title": "Photon density distribution on the ground",
-                "bin edges": "Distance to center",
-                "axis unit": self.hist_config["hist_position"]["x axis"]["start"].unit,
+                fn_key: "get_photon_density_distr",
+                file_name: "hist_1d_photon_density_distr",
+                title: "Photon density distribution on the ground",
+                bin_edges: "Distance to center",
+                axis_unit: self.hist_config["hist_position"][X_AXIS_STRING]["start"].unit,
             },
             "time": {
-                "function": "get_photon_time_of_emission_distr",
-                "file name": "hist_1d_photon_time_distr",
-                "title": "Photon time of arrival distribution",
-                "bin edges": "Time of arrival",
-                "axis unit": self.hist_config["hist_time_altitude"]["x axis"]["start"].unit,
+                fn_key: "get_photon_time_of_emission_distr",
+                file_name: "hist_1d_photon_time_distr",
+                title: "Photon time of arrival distribution",
+                bin_edges: "Time of arrival",
+                axis_unit: self.hist_config["hist_time_altitude"][X_AXIS_STRING]["start"].unit,
             },
             "altitude": {
-                "function": "get_photon_altitude_distr",
-                "file name": "hist_1d_photon_altitude_distr",
-                "title": "Photon altitude of emission distribution",
-                "bin edges": "Altitude of emission",
-                "axis unit": self.hist_config["hist_time_altitude"]["y axis"]["start"].unit,
+                fn_key: "get_photon_altitude_distr",
+                file_name: "hist_1d_photon_altitude_distr",
+                title: "Photon altitude of emission distribution",
+                bin_edges: "Altitude of emission",
+                axis_unit: self.hist_config["hist_time_altitude"][Y_AXIS_STRING]["start"].unit,
             },
             "num_photons_per_event": {
-                "function": "get_num_photons_per_event_distr",
-                "file name": "hist_1d_photon_per_event_distr",
-                "title": "Photons per event distribution",
-                "bin edges": "Event counter",
-                "axis unit": u.dimensionless_unscaled,
+                fn_key: "get_num_photons_per_event_distr",
+                file_name: "hist_1d_photon_per_event_distr",
+                title: "Photons per event distribution",
+                bin_edges: "Event counter",
+                axis_unit: u.dimensionless_unscaled,
             },
             "num_photons_per_telescope": {
-                "function": "get_num_photons_per_telescope_distr",
-                "file name": "hist_1d_photon_per_telescope_distr",
-                "title": "Photons per telescope distribution",
-                "bin edges": "Telescope counter",
-                "axis unit": u.dimensionless_unscaled,
+                fn_key: "get_num_photons_per_telescope_distr",
+                file_name: "hist_1d_photon_per_telescope_distr",
+                title: "Photons per telescope distribution",
+                bin_edges: "Telescope counter",
+                axis_unit: u.dimensionless_unscaled,
             },
         }
         return self._dict_1d_distributions
@@ -1219,14 +1219,14 @@ class CorsikaHistograms:
         overwrite: bool
             If True overwrites the histograms already saved in the hdf5 file.
         """
-
+        axis_unit = "axis unit"
         for _, function_dict in self.dict_1d_distributions.items():
             self._meta_dict["Title"] = sanitize_name(function_dict["title"])
-            function = getattr(self, function_dict["function"])
-            hist_1d_list, x_bin_edges_list = function()
-            x_bin_edges_list = x_bin_edges_list * function_dict["axis unit"]
+            histogram_function = getattr(self, function_dict["function"])
+            hist_1d_list, x_bin_edges_list = histogram_function()
+            x_bin_edges_list = x_bin_edges_list * function_dict[axis_unit]
             if function_dict["function"] == "get_photon_density_distr":
-                histogram_value_unit = 1 / (function_dict["axis unit"] ** 2)
+                histogram_value_unit = 1 / (function_dict[axis_unit] ** 2)
             else:
                 histogram_value_unit = u.dimensionless_unscaled
             hist_1d_list = hist_1d_list * histogram_value_unit
@@ -1270,52 +1270,63 @@ class CorsikaHistograms:
         dict:
             The dictionary with information about the 2D distributions.
         """
+        fn_key = "function"
+        file_name = "file name"
+        title = "title"
+        x_bin_edges = "x bin edges"
+        x_axis_unit = "x axis unit"
+        y_bin_edges = "y bin edges"
+        y_axis_unit = "y axis unit"
         if self._dict_2d_distributions is None:
             self._dict_2d_distributions = {
                 "counts": {
-                    "function": "get_2d_photon_position_distr",
-                    "file name": "hist_2d_photon_count_distr",
-                    "title": "Photon count distribution on the ground",
-                    "x bin edges": "x position on the ground",
-                    "x axis unit": self.hist_config["hist_position"]["x axis"]["start"].unit,
-                    "y bin edges": "y position on the ground",
-                    "y axis unit": self.hist_config["hist_position"]["y axis"]["start"].unit,
+                    fn_key: "get_2d_photon_position_distr",
+                    file_name: "hist_2d_photon_count_distr",
+                    title: "Photon count distribution on the ground",
+                    x_bin_edges: "x position on the ground",
+                    x_axis_unit: self.hist_config["hist_position"][X_AXIS_STRING]["start"].unit,
+                    y_bin_edges: "y position on the ground",
+                    y_axis_unit: self.hist_config["hist_position"][Y_AXIS_STRING]["start"].unit,
                 },
                 "density": {
-                    "function": "get_2d_photon_density_distr",
-                    "file name": "hist_2d_photon_density_distr",
-                    "title": "Photon density distribution on the ground",
-                    "x bin edges": "x position on the ground",
-                    "x axis unit": self.hist_config["hist_position"]["x axis"]["start"].unit,
-                    "y bin edges": "y position on the ground",
-                    "y axis unit": self.hist_config["hist_position"]["y axis"]["start"].unit,
+                    fn_key: "get_2d_photon_density_distr",
+                    file_name: "hist_2d_photon_density_distr",
+                    title: "Photon density distribution on the ground",
+                    x_bin_edges: "x position on the ground",
+                    x_axis_unit: self.hist_config["hist_position"][X_AXIS_STRING]["start"].unit,
+                    y_bin_edges: "y position on the ground",
+                    y_axis_unit: self.hist_config["hist_position"][Y_AXIS_STRING]["start"].unit,
                 },
                 "direction": {
-                    "function": "get_2d_photon_direction_distr",
-                    "file name": "hist_2d_photon_direction_distr",
-                    "title": "Photon arrival direction",
-                    "x bin edges": "x direction cosine",
-                    "x axis unit": u.dimensionless_unscaled,
-                    "y bin edges": "y direction cosine",
-                    "y axis unit": u.dimensionless_unscaled,
+                    fn_key: "get_2d_photon_direction_distr",
+                    file_name: "hist_2d_photon_direction_distr",
+                    title: "Photon arrival direction",
+                    x_bin_edges: "x direction cosine",
+                    x_axis_unit: u.dimensionless_unscaled,
+                    y_bin_edges: "y direction cosine",
+                    y_axis_unit: u.dimensionless_unscaled,
                 },
                 "time_altitude": {
-                    "function": "get_2d_photon_time_altitude_distr",
-                    "file name": "hist_2d_photon_time_altitude_distr",
-                    "title": "Time of arrival vs altitude of emission",
-                    "x bin edges": "Time of arrival",
-                    "x axis unit": self.hist_config["hist_time_altitude"]["x axis"]["start"].unit,
-                    "y bin edges": "Altitude of emission",
-                    "y axis unit": self.hist_config["hist_time_altitude"]["y axis"]["start"].unit,
+                    fn_key: "get_2d_photon_time_altitude_distr",
+                    file_name: "hist_2d_photon_time_altitude_distr",
+                    title: "Time of arrival vs altitude of emission",
+                    x_bin_edges: "Time of arrival",
+                    x_axis_unit: self.hist_config["hist_time_altitude"][X_AXIS_STRING][
+                        "start"
+                    ].unit,
+                    y_bin_edges: "Altitude of emission",
+                    y_axis_unit: self.hist_config["hist_time_altitude"][Y_AXIS_STRING][
+                        "start"
+                    ].unit,
                 },
                 "num_photons_per_telescope": {
-                    "function": "get_2d_num_photons_distr",
-                    "file name": "hist_2d_photon_telescope_event_distr",
-                    "title": "Number of photons per telescope and per event",
-                    "x bin edges": "Telescope counter",
-                    "x axis unit": u.dimensionless_unscaled,
-                    "y bin edges": "Event counter",
-                    "y axis unit": u.dimensionless_unscaled,
+                    fn_key: "get_2d_num_photons_distr",
+                    file_name: "hist_2d_photon_telescope_event_distr",
+                    title: "Number of photons per telescope and per event",
+                    x_bin_edges: "Telescope counter",
+                    x_axis_unit: u.dimensionless_unscaled,
+                    y_bin_edges: "Event counter",
+                    y_axis_unit: u.dimensionless_unscaled,
                 },
             }
         return self._dict_2d_distributions
@@ -1329,23 +1340,26 @@ class CorsikaHistograms:
         overwrite: bool
             If True overwrites the histograms already saved in the hdf5 file.
         """
+        x_axis_unit = "x axis unit"
+        y_axis_unit = "y axis unit"
+
         for property_name, function_dict in self.dict_2d_distributions.items():
             self._meta_dict["Title"] = sanitize_name(function_dict["title"])
-            function = getattr(self, function_dict["function"])
+            histogram_function = getattr(self, function_dict["function"])
 
-            hist_2d_list, x_bin_edges_list, y_bin_edges_list = function()
+            hist_2d_list, x_bin_edges_list, y_bin_edges_list = histogram_function()
             if function_dict["function"] == "get_2d_photon_density_distr":
                 histogram_value_unit = 1 / (
-                    self.dict_2d_distributions[property_name]["x axis unit"]
-                    * self.dict_2d_distributions[property_name]["y axis unit"]
+                    self.dict_2d_distributions[property_name][x_axis_unit]
+                    * self.dict_2d_distributions[property_name][y_axis_unit]
                 )
             else:
                 histogram_value_unit = u.dimensionless_unscaled
 
             hist_2d_list, x_bin_edges_list, y_bin_edges_list = (
                 hist_2d_list * histogram_value_unit,
-                x_bin_edges_list * self.dict_2d_distributions[property_name]["x axis unit"],
-                y_bin_edges_list * self.dict_2d_distributions[property_name]["y axis unit"],
+                x_bin_edges_list * self.dict_2d_distributions[property_name][x_axis_unit],
+                y_bin_edges_list * self.dict_2d_distributions[property_name][y_axis_unit],
             )
 
             for i_histogram, _ in enumerate(x_bin_edges_list):
@@ -1356,7 +1370,7 @@ class CorsikaHistograms:
                     )
                 else:
                     hdf5_table_name = (
-                        f"/{self.dict_2d_distributions[property_name]['file name']}" f"_all_tels"
+                        f"/{self.dict_2d_distributions[property_name]['file name']}_all_tels"
                     )
                 table = fill_hdf5_table(
                     hist=hist_2d_list[i_histogram],
@@ -1381,14 +1395,13 @@ class CorsikaHistograms:
         self, event_header_element, bins=50, hist_range=None, overwrite=False
     ):
         """
-        Export to a hdf5 file the 1D histogram for the key `event_header_element` from the CORSIKA
-        event header.
+        Export 'event_header_element' from CORSIKA to hd5 for a 1D histogram.
 
         Parameters
         ----------
         event_header_element: str
             The key to the CORSIKA event header element.
-            Possible choices are stored in `self.all_event_keys`.
+            Possible choices are stored in 'self.all_event_keys'.
         bins: float
             Number of bins for the histogram.
         hist_range: 2-tuple
@@ -1396,7 +1409,6 @@ class CorsikaHistograms:
         overwrite: bool
             If True overwrites the histograms already saved in the hdf5 file.
         """
-
         hist, bin_edges = self.event_1d_histogram(
             event_header_element, bins=bins, hist_range=hist_range
         )
@@ -1430,8 +1442,10 @@ class CorsikaHistograms:
         overwrite=False,
     ):
         """
-        Export to a hdf5 file the 2D histogram for the key `event_header_element_1` and
-        `event_header_element_2`from the CORSIKA event header.
+        Export event_header of a 2D histogram to a hdf5 file.
+
+        Searches the 2D histogram for the key 'event_header_element_1' and
+        'event_header_element_2'from the CORSIKA event header.
 
         Parameters
         ----------
@@ -1439,14 +1453,15 @@ class CorsikaHistograms:
             The key to the CORSIKA event header element.
         event_header_element_2: str
             The key to the CORSIKA event header element.
-            Possible choices for `event_header_element_1` and `event_header_element_2` are stored
-            in `self.all_event_keys`.
+            Possible choices for 'event_header_element_1' and 'event_header_element_2' are stored
+            in 'self.all_event_keys'.
         bins: float
             Number of bins for the histogram.
         hist_range: 2-tuple
             Tuple to define the range of the histogram.
         overwrite: bool
             If True overwrites the histograms already saved in the hdf5 file.
+
         """
         hist, x_bin_edges, y_bin_edges = self.event_2d_histogram(
             event_header_element_1, event_header_element_2, bins=bins, hist_range=hist_range
@@ -1478,8 +1493,7 @@ class CorsikaHistograms:
     @property
     def num_photons_per_telescope(self):
         """
-        The number of photons per event, considering the telescopes given by
-        self.telescope_indices.
+        The number of photons per event, considering the telescopes given by self.telescope_indices.
 
         Returns
         -------
@@ -1508,7 +1522,8 @@ class CorsikaHistograms:
     def telescope_positions(self):
         """
         The telescope positions found in the CORSIKA output file.
-        It does not depend on the `telescope_indices` attribute.
+
+        It does not depend on the telescope_indices attribute.
 
         Returns
         -------
@@ -1533,7 +1548,7 @@ class CorsikaHistograms:
 
     # In the next five functions, we provide dedicated functions to retrieve specific information
     # about the runs, i.e. zenith, azimuth, total energy, interaction height and Earth magnetic
-    # field defined for the run. For other information, please use the `get_event_parameter_info`
+    # field defined for the run. For other information, please use the get_event_parameter_info
     # function.
     @property
     def event_zenith_angles(self):
@@ -1590,8 +1605,9 @@ class CorsikaHistograms:
     def event_first_interaction_heights(self):
         """
         Get the height of the first interaction in astropy units of km.
-        If negative, tracking starts at margin of atmosphere, see TSTART in the CORSIKA 7 user guide
-        .
+
+        If negative, tracking starts at margin of atmosphere,
+        see TSTART in the CORSIKA 7 user guide.
 
         Returns
         -------
@@ -1609,6 +1625,7 @@ class CorsikaHistograms:
     def magnetic_field(self):
         """
         Get the Earth magnetic field from the events header in astropy units of microT.
+
         A tuple with Earth's magnetic field in the x and z directions are returned.
 
         Returns
@@ -1626,8 +1643,10 @@ class CorsikaHistograms:
 
     def get_event_parameter_info(self, parameter):
         """
-        Get specific information (i.e. any parameter) of the events. The parameter is passed through
-        the key word `parameter`. Available options are to be found under `self.all_event_keys`.
+        Get specific information (i.e. any parameter) of the events.
+
+        The parameter is passed through the key word parameter.
+        Available options are to be found under self.all_event_keys.
         The unit of the parameter, if any, is given according to the CORSIKA version
         (please see user guide in this case).
 
@@ -1635,7 +1654,7 @@ class CorsikaHistograms:
         ----------
         parameter: str
             The parameter of interest. Available options are to be found under
-            `self.all_event_keys`.
+            self.all_event_keys.
 
         Returns
         -------
@@ -1647,17 +1666,18 @@ class CorsikaHistograms:
         KeyError:
             If parameter is not valid.
         """
-
         if parameter not in self.all_event_keys:
-            msg = f"`key` is not valid. Valid entries are {self.all_event_keys}"
+            msg = f"key is not valid. Valid entries are {self.all_event_keys}"
             self._logger.error(msg)
             raise KeyError
         return self.event_information[parameter]
 
     def get_run_info(self, parameter):
         """
-        Get specific information (i.e. any parameter) of the run. The parameter is passed through
-        the key word `parameter`. Available options are to be found under `self.all_run_keys`.
+        Get specific information (i.e. any parameter) of the run.
+
+        The parameter is passed through the key word parameter.
+        Available options are to be found under self.all_run_keys.
         The unit of the parameter, if any, is given according to the CORSIKA version
         (please see user guide in this case).
 
@@ -1665,7 +1685,7 @@ class CorsikaHistograms:
         ----------
         parameter: str
             The parameter of interest. Available options are to be found under
-            `self.all_run_keys`.
+            self.all_run_keys.
 
         Raises
         ------
@@ -1673,15 +1693,16 @@ class CorsikaHistograms:
             If parameter is not valid.
         """
         if parameter not in self.all_run_keys:
-            msg = f"`key` is not valid. Valid entries are {self.all_run_keys}"
+            msg = f"key is not valid. Valid entries are {self.all_run_keys}"
             self._logger.error(msg)
             raise KeyError
         return self.header[parameter]
 
     def event_1d_histogram(self, key, bins=50, hist_range=None):
         """
-        Create a histogram for the all events using `key` as parameter.
-        Valid keys are stored in `self.all_event_keys` (CORSIKA defined).
+        Create a histogram for the all events using key as parameter.
+
+        Valid keys are stored in self.all_event_keys (CORSIKA defined).
 
         Parameters
         ----------
@@ -1707,7 +1728,7 @@ class CorsikaHistograms:
             If key is not valid.
         """
         if key not in self.all_event_keys:
-            msg = f"`key` is not valid. Valid entries are {self.all_event_keys}"
+            msg = f"key is not valid. Valid entries are {self.all_event_keys}"
             self._logger.error(msg)
             raise KeyError
         hist, bin_edges = np.histogram(
@@ -1719,8 +1740,9 @@ class CorsikaHistograms:
 
     def event_2d_histogram(self, key_1, key_2, bins=50, hist_range=None):
         """
-        Create a 2D histogram for the all events using `key_1` and `key_2` as parameters.
-        Valid keys are stored in `self.all_event_keys` (CORSIKA defined).
+        Create a 2D histogram for the all events using key_1 and key_2 as parameters.
+
+        Valid keys are stored in self.all_event_keys (CORSIKA defined).
 
         Parameters
         ----------