gammasimtools 0.8.1__py3-none-any.whl → 0.9.0__py3-none-any.whl

simtools/model/camera.py

@@ -1,11 +1,10 @@
 """Definition and modeling of camera."""
 
 import logging
+from pathlib import Path
 
 import astropy.units as u
 import numpy as np
-from scipy.spatial import cKDTree as KDTree
-from scipy.spatial import distance
 
 from simtools.utils.geometry import rotate
 
@@ -20,9 +19,9 @@ class Camera:
 
     Parameters
     ----------
-    telescope_model_name: string
+    telescope_model_name: str
         As provided by the telescope model method TelescopeModel (e.g., LSTN-01)
-    camera_config_file: string
+    camera_config_file: str or Path
         The sim_telarray file name.
     focal_length: float
         The focal length of the camera in (preferably the effective focal length), assumed to be \
@@ -34,7 +33,9 @@ class Camera:
     SIPM_NEIGHBOR_RADIUS_FACTOR = 1.4
     SIPM_ROW_COLUMN_DIST_FACTOR = 0.2
 
-    def __init__(self, telescope_model_name, camera_config_file, focal_length):
+    def __init__(
+        self, telescope_model_name: str, camera_config_file: str | Path, focal_length: float
+    ):
         """
         Initialize Camera class, defining pixel layout.
 
@@ -58,13 +59,13 @@ class Camera:
         self._edge_pixel_indices = None
 
     @staticmethod
-    def read_pixel_list(camera_config_file):
+    def read_pixel_list(camera_config_file: str | Path) -> dict:
         """
         Read the pixel layout from the camera config file, assumed to be in a sim_telarray format.
 
         Parameters
         ----------
-        camera_config_file: string
+        camera_config_file: str or Path
             The sim_telarray file name.
 
         Returns
@@ -90,7 +91,7 @@ class Camera:
         return pixels
 
     @staticmethod
-    def initialize_pixel_dict():
+    def initialize_pixel_dict() -> dict:
         """
         Initialize the pixel dictionary with default values.
 
@@ -113,16 +114,16 @@ class Camera:
         }
 
     @staticmethod
-    def process_line(line, pixels):
+    def process_line(line: str, pixels: dict):
         """
-        Process a line from the camera config file and update the pixel dictionary.
+        Process a line from the camera config file and update the pixels dictionary.
 
         Parameters
         ----------
-        line: string
+        line: str
             A line from the camera config file.
         pixels: dict
-            The pixel dictionary to update.
+            The dictionary to update with pixel information.
         """
         pix_info = line.split()
 
@@ -150,9 +151,9 @@ class Camera:
                 pixels["pix_on"].append(True)
 
     @staticmethod
-    def validate_pixels(pixels, camera_config_file):
+    def validate_pixels(pixels: dict, camera_config_file: str | Path):
         """
-        Validate the pixel dictionary to ensure it has correct values.
+        Validate the pixel dictionary to ensure all required fields are present.
 
         Parameters
         ----------
@@ -174,7 +175,7 @@ class Camera:
                 f"Pixel shape in {camera_config_file} unrecognized (has to be 1, 2 or 3)"
             )
 
-    def _rotate_pixels(self, pixels):
+    def _rotate_pixels(self, pixels: dict) -> dict:
         """
         Rotate the pixels according to the rotation angle given in pixels['rotate_angle'].
 
@@ -184,7 +185,7 @@ class Camera:
 
         Parameters
         ----------
-        pixels: dictionary
+        pixels: dict
             The dictionary produced by the read_pixel_list method of this class
 
         Returns
@@ -219,9 +220,9 @@ class Camera:
 
         return pixels
 
-    def get_number_of_pixels(self):
+    def get_number_of_pixels(self) -> int:
         """
-        Get the number of pixels in the camera (all pixels, including those defined as "off".
+        Get the number of pixels in the camera (all pixels, including those defined as "off").
 
         Returns
         -------
@@ -230,7 +231,7 @@ class Camera:
         """
         return len(self.pixels["x"])
 
-    def get_pixel_diameter(self):
+    def get_pixel_diameter(self) -> float:
         """
         Get pixel diameter contained in _pixels.
 
@@ -241,7 +242,7 @@ class Camera:
         """
         return self.pixels["pixel_diameter"]
 
-    def get_pixel_active_solid_angle(self):
+    def get_pixel_active_solid_angle(self) -> float:
         """
         Get the active solid angle of a pixel in sr.
 
@@ -256,7 +257,7 @@ class Camera:
             pixel_area *= np.sqrt(3) / 2
         return pixel_area / (self.focal_length**2)
 
-    def get_pixel_shape(self):
+    def get_pixel_shape(self) -> int:
         """
         Get pixel shape code 1, 2 or 3.
 
@@ -270,7 +271,7 @@ class Camera:
         """
         return self.pixels["pixel_shape"]
 
-    def get_lightguide_efficiency_angle_file_name(self):
+    def get_lightguide_efficiency_angle_file_name(self) -> str:
         """
         Get the file name of the light guide efficiency as a function of incidence angle.
 
@@ -281,7 +282,7 @@ class Camera:
         """
         return self.pixels["lightguide_efficiency_angle_file"]
 
-    def get_lightguide_efficiency_wavelength_file_name(self):
+    def get_lightguide_efficiency_wavelength_file_name(self) -> str:
         """
         Get the file name of the light guide efficiency as a function of wavelength.
 
@@ -292,7 +293,7 @@ class Camera:
         """
         return self.pixels["lightguide_efficiency_wavelength_file"]
 
-    def get_camera_fill_factor(self):
+    def get_camera_fill_factor(self) -> float:
         """
         Calculate the fill factor of the camera, defined as (pixel_diameter/pixel_spacing)**2.
 
@@ -301,6 +302,8 @@ class Camera:
         float
             The camera fill factor.
         """
+        from scipy.spatial import distance  # pylint: disable=import-outside-toplevel
+
         if self.pixels["pixel_spacing"] == 9999:
             points = np.array([self.pixels["x"], self.pixels["y"]]).T
             pixel_distances = distance.cdist(points, points, "euclidean")
@@ -311,7 +314,7 @@ class Camera:
 
         return (self.pixels["pixel_diameter"] / self.pixels["pixel_spacing"]) ** 2
 
-    def calc_fov(self):
+    def calc_fov(self) -> tuple[float, float]:
         """
         Calculate the FOV of the camera in degrees, taking into account the focal length.
 
@@ -327,7 +330,6 @@ class Camera:
         The x,y pixel positions and focal length are assumed to have the same unit (usually cm)
         """
         self._logger.debug("Calculating the FoV")
-
         return self._calc_fov(
             self.pixels["x"],
             self.pixels["y"],
@@ -335,7 +337,13 @@ class Camera:
             self.focal_length,
         )
 
-    def _calc_fov(self, x_pixel, y_pixel, edge_pixel_indices, focal_length):
+    def _calc_fov(
+        self,
+        x_pixel: list[float],
+        y_pixel: list[float],
+        edge_pixel_indices: list[int],
+        focal_length: float,
+    ) -> tuple[float, float]:
         """
         Calculate the FOV of the camera in degrees, taking into account the focal length.
 
@@ -374,7 +382,7 @@ class Camera:
         return fov, average_edge_distance
 
     @staticmethod
-    def _find_neighbors(x_pos, y_pos, radius):
+    def _find_neighbors(x_pos: np.ndarray, y_pos: np.ndarray, radius: float) -> list[list[int]]:
         """
         Use a KD-Tree to quickly find nearest neighbors.
 
@@ -392,20 +400,18 @@ class Camera:
 
         Returns
         -------
-        neighbors: numpy.array_like
-            Array of neighbor indices in a list for each e.g., pixel.
+        list of lists
+            Array of neighbor indices in a list for each pixel
         """
-        points = np.array([x_pos, y_pos]).T
-        indices = np.arange(len(x_pos))
-        kdtree = KDTree(points)
-        neighbors = [kdtree.query_ball_point(p, r=radius) for p in points]
-
-        for neighbor_now, index_now in zip(neighbors, indices):
-            neighbor_now.remove(index_now)  # get rid of the pixel or mirror itself
+        from scipy.spatial import cKDTree as KDTree  # pylint: disable=import-outside-toplevel
 
-        return neighbors
+        tree = KDTree(np.column_stack([x_pos, y_pos]))
+        neighbors = tree.query_ball_tree(tree, radius)
+        return [list(np.setdiff1d(neigh, [i])) for i, neigh in enumerate(neighbors)]
 
-    def _find_adjacent_neighbor_pixels(self, x_pos, y_pos, radius, row_column_dist):
+    def _find_adjacent_neighbor_pixels(
+        self, x_pos: np.ndarray, y_pos: np.ndarray, radius: float, row_column_dist: float
+    ) -> list[list[int]]:
         """
         Find adjacent neighbor pixels in cameras with square pixels.
 
@@ -413,13 +419,13 @@ class Camera:
 
         Parameters
         ----------
-        x_pos : numpy.array_like
+        x_pos: np.ndarray
             x position of each pixel
-        y_pos : numpy.array_like
+        y_pos: np.ndarray
             y position of each pixel
-        radius : float
+        radius: float
             Radius within which to find neighbors
-        row_column_dist : float
+        row_column_dist: float
             Distance to consider for row/column adjacency.
             Should be around 20% of the pixel diameter.
 
@@ -441,23 +447,31 @@ class Camera:
 
         return neighbors
 
-    def _add_additional_neighbors(self, i_pix, nn, x_pos, y_pos, radius, row_column_dist):
+    def _add_additional_neighbors(
+        self,
+        i_pix: int,
+        nn: list[int],
+        x_pos: np.ndarray,
+        y_pos: np.ndarray,
+        radius: float,
+        row_column_dist: float,
+    ):
         """
         Add neighbors for a given pixel if they are not already neighbors and are adjacent.
 
         Parameters
         ----------
-        i_pix : int
+        i_pix: int
             Index of the pixel to find neighbors for
-        nn : list
+        nn: list
             Current list of neighbors for the pixel
-        x_pos : numpy.array_like
+        x_pos: np.ndarray
             x position of each pixel
-        y_pos : numpy.array_like
+        y_pos: np.ndarray
             y position of each pixel
-        radius : float
+        radius: float
             Radius within which to find neighbors
-        row_column_dist : float
+        row_column_dist: float
             Distance to consider for row/column adjacency
         """
         for j_pix, _ in enumerate(x_pos):
@@ -477,20 +491,20 @@ class Camera:
                 ) and dist < 1.2 * radius:
                     nn.append(j_pix)
 
-    def _calc_neighbor_pixels(self, pixels):
+    def _calc_neighbor_pixels(self, pixels: dict) -> list[list[int]]:
         """
         Find adjacent neighbor pixels in cameras with hexagonal or square pixels.
 
-        Only directly  adjacent neighbors are searched for, no diagonals.
+        Only directly adjacent neighbors are searched for, no diagonals.
 
         Parameters
         ----------
-        pixels: dictionary
+        pixels: dict
             The dictionary produced by the read_pixel_list method of this class
 
         Returns
         -------
-        neighbors: numpy.array_like
+        neighbors: list of lists
             Array of neighbor indices in a list for each pixel
         """
         self._logger.debug("Searching for neighbor pixels")
@@ -515,7 +529,7 @@ class Camera:
 
         return self._neighbors
 
-    def get_neighbor_pixels(self, pixels=None):
+    def get_neighbor_pixels(self, pixels: dict | None = None) -> list[list[int]]:
         """
         Get a list of neighbor pixels by calling calc_neighbor_pixels() when necessary.
 
@@ -529,7 +543,7 @@ class Camera:
 
         Returns
         -------
-        neighbors: numpy.array_like
+        neighbors: list of lists
             Array of neighbor indices in a list for each pixel.
         """
         if self._neighbors is None:
@@ -539,20 +553,20 @@ class Camera:
 
         return self._neighbors
 
-    def _calc_edge_pixels(self, pixels, neighbors):
+    def _calc_edge_pixels(self, pixels: dict, neighbors: list[list[int]]) -> list[int]:
         """
         Find the edge pixels of the camera.
 
         Parameters
         ----------
-        pixels: dictionary
+        pixels: dict
             The dictionary produced by the read_pixel_list method of this class.
-        neighbors: numpy.array_like
+        neighbors: list of lists
             Array of neighbor indices in a list for each pixel.
 
         Returns
         -------
-        edge_pixel_indices: numpy.array_like
+        edge_pixel_indices: list
             Array of edge pixel indices.
         """
         self._logger.debug("Searching for edge pixels")
@@ -574,7 +588,9 @@ class Camera:
 
         return edge_pixel_indices
 
-    def get_edge_pixels(self, pixels=None, neighbors=None):
+    def get_edge_pixels(
+        self, pixels: dict | None = None, neighbors: list[list[int]] | None = None
+    ) -> list[int]:
         """
         Get the indices of the edge pixels of the camera.
 
@@ -582,12 +598,12 @@ class Camera:
         ----------
         pixels: dict
             The dictionary produced by the read_pixel_list method of this class.
-        neighbors: numpy.array_like
+        neighbors: list of lists
             Array of neighbor indices in a list for each pixel.
 
         Returns
         -------
-        edge_pixel_indices: numpy.array_like
+        edge_pixel_indices: list
             Array of edge pixel indices.
         """
         if self._edge_pixel_indices is None:

simtools/model/mirrors.py

@@ -1,6 +1,7 @@
 """Definition and modeling of mirror panels."""
 
 import logging
+from pathlib import Path
 
 import astropy.io.ascii
 import astropy.units as u
@@ -20,13 +21,13 @@ class Mirrors:
 
     Parameters
     ----------
-    mirror_list_file: str
-        mirror list in sim_telarray or ecsv format (with panel focal length only).
-    parameters: dict
+    mirror_list_file: Union[str, Path]
+        Mirror list in sim_telarray or ecsv format (with panel focal length only).
+    parameters: dict, optional
         Dictionary of parameters from the database.
     """
 
-    def __init__(self, mirror_list_file, parameters=None):
+    def __init__(self, mirror_list_file: str | Path, parameters: dict | None = None):
         """Initialize Mirrors."""
         self._logger = logging.getLogger(__name__)
         self._logger.debug("Mirrors Init")
@@ -180,7 +181,7 @@ class Mirrors:
         self._logger.debug(f"Mirror diameter = {self.mirror_diameter}")
         self._logger.debug(f"Number of Mirrors = {self.number_of_mirrors}")
 
-    def get_single_mirror_parameters(self, number):
+    def get_single_mirror_parameters(self, number: int) -> tuple:
         """
         Get parameters for a single mirror given by number.
 
@@ -191,7 +192,8 @@ class Mirrors:
 
         Returns
         -------
-        (pos_x, pos_y, mirror_diameter, focal_length, shape_type): tuple of float
+        tuple
+            (pos_x, pos_y, mirror_diameter, focal_length, shape_type): tuple of float
             X, Y positions, mirror_diameter, focal length and shape_type.
         """
         mask = self.mirror_table["mirror_panel_id"] == number

simtools/model/model_parameter.py

@@ -6,10 +6,12 @@ import shutil
 from copy import copy
 
 import astropy.units as u
+from astropy.table import Table
 
 import simtools.utils.general as gen
 from simtools.db import db_handler
 from simtools.io_operations import io_handler
+from simtools.simtel import simtel_table_reader
 from simtools.simtel.simtel_config_writer import SimtelConfigWriter
 from simtools.utils import names
 
@@ -273,7 +275,7 @@ class ModelParameter:
 
     def _set_config_file_directory_and_name(self):
         """Set and create the directory and the name of the config file."""
-        if self.name is None:
+        if self.name is None and self.site is None:
             return
 
         self._config_file_directory = self.io_handler.get_output_directory(
@@ -281,15 +283,14 @@ class ModelParameter:
         )
 
         # Setting file name and the location
-        if self.site is not None and self.name is not None:
-            config_file_name = names.simtel_config_file_name(
-                self.site,
-                self.model_version,
-                telescope_model_name=self.name,
-                label=self.label,
-                extra_label=self._extra_label,
-            )
-            self._config_file_path = self.config_file_directory.joinpath(config_file_name)
+        config_file_name = names.simtel_config_file_name(
+            self.site,
+            self.model_version,
+            telescope_model_name=self.name,
+            label=self.label,
+            extra_label=self._extra_label,
+        )
+        self._config_file_path = self.config_file_directory.joinpath(config_file_name)
 
         self._logger.debug(f"Config file path: {self._config_file_path}")
 
@@ -456,7 +457,7 @@ class ModelParameter:
         Change values of multiple existing parameters in the model from a file.
 
         This function does not modify the DB, it affects only the current instance.
-        Experimental feature: insufficient validation of parameters.
+        This feature is intended for developers and lacks validation.
 
         Parameters
         ----------
@@ -464,7 +465,7 @@ class ModelParameter:
             File containing the parameters to be changed.
         """
         self._logger.warning(
-            "Changing multiple parameters from file is an experimental feature."
+            "Changing multiple parameters from file is a feature for developers."
             "Insufficient validation of parameters."
         )
         self._logger.debug(f"Changing parameters from file {file_name}")
@@ -515,6 +516,35 @@ class ModelParameter:
         self.db.export_model_files(pars_from_db, self.config_file_directory)
         self._is_exported_model_files_up_to_date = True
 
+    def get_model_file_as_table(self, par_name):
+        """
+        Return tabular data from file as astropy table.
+
+        Parameters
+        ----------
+        par_name: str
+            Name of the parameter.
+
+        Returns
+        -------
+        Table
+            Astropy table.
+        """
+        _par_entry = {}
+        try:
+            _par_entry[par_name] = self._parameters[par_name]
+        except KeyError as exc:
+            raise ValueError(f"Parameter {par_name} not found in the model.") from exc
+        self.db.export_model_files(_par_entry, self.config_file_directory)
+        if _par_entry[par_name]["value"].endswith("ecsv"):
+            return Table.read(
+                self.config_file_directory.joinpath(_par_entry[par_name]["value"]),
+                format="ascii.ecsv",
+            )
+        return simtel_table_reader.read_simtel_table(
+            par_name, self.config_file_directory.joinpath(_par_entry[par_name]["value"])
+        )
+
     def export_config_file(self):
         """Export the config file used by sim_telarray."""
         # Exporting model file

simtools/model/model_utils.py

@@ -11,7 +11,7 @@ __all__ = [
 ]
 
 
-def compute_telescope_transmission(pars, off_axis):
+def compute_telescope_transmission(pars: list[float], off_axis: float) -> float:
     """
     Compute telescope transmission (0 < T < 1) for a given off-axis angle.
 
@@ -20,7 +20,7 @@ def compute_telescope_transmission(pars, off_axis):
     Parameters
     ----------
     pars: list of float
-        Parameters of the telescope transmission. Len(pars) should be 4.
+        Parameters of the telescope transmission. Len(pars) should be 5 or 6.
     off_axis: float
         Off-axis angle in deg.
 
@@ -37,19 +37,19 @@ def compute_telescope_transmission(pars, off_axis):
     return pars[0] / (1.0 + pars[2] * t ** pars[4])
 
 
-def is_two_mirror_telescope(telescope_model_name):
+def is_two_mirror_telescope(telescope_model_name: str) -> bool:
     """
-    Check if the telescope is a two mirror design.
+    Determine if the telescope model is a two-mirror telescope.
 
     Parameters
     ----------
     telescope_model_name: str
-        Telescope model name (ex. LSTN-01).
+        Name of the telescope model.
 
     Returns
     -------
     bool
-        True if the telescope is a two mirror one.
+        True if it is a two-mirror telescope.
     """
     tel_type = names.get_array_element_type_from_name(telescope_model_name)
     if "SST" in tel_type or "SCT" in tel_type:

simtools/model/site_model.py

@@ -21,7 +21,7 @@ class SiteModel(ModelParameter):
         MongoDB configuration.
     model_version: str
         Model version.
-    label: str
+    label: str, optional
         Instance label. Important for output file naming.
     """
 
@@ -80,7 +80,7 @@ class SiteModel(ModelParameter):
             Site-related CORSIKA parameters as dict
         """
         if config_file_style:
-            model_directory = model_directory or Path("")
+            model_directory = model_directory or Path()
             return {
                 "OBSLEV": [
                     self.get_parameter_value_with_unit("corsika_observation_level").to_value("cm")