gammasimtools 0.17.0__py3-none-any.whl → 0.18.0__py3-none-any.whl

simtools/visualization/plot_tables.py

@@ -1,10 +1,13 @@
 #!/usr/bin/python3
 """Plot tabular data."""
 
+from pathlib import Path
+
 import numpy as np
 from astropy.table import Table
 
 import simtools.utils.general as gen
+from simtools.constants import SCHEMA_PATH
 from simtools.db import db_handler
 from simtools.io_operations import legacy_data_handler
 from simtools.visualization import visualize
@@ -20,6 +23,8 @@ def plot(config, output_file, db_config=None):
         Configuration dictionary for plotting.
     output_file: str
         Output file.
+    db_config: dict, optional
+        Database configuration dictionary for accessing the model parameter database.
     """
     data = read_table_data(config, db_config)
 
@@ -29,6 +34,8 @@ def plot(config, output_file, db_config=None):
     )
     visualize.save_figure(fig, output_file)
 
+    return output_file
+
 
 def read_table_data(config, db_config):
     """
@@ -69,7 +76,8 @@ def read_table_data(config, db_config):
                 _config["select_values"]["column_name"],
                 _config["select_values"]["value"],
             )
-        data[_config["label"]] = gen.get_structure_array_from_table(
+        label = _config.get("label", f"{_config.get('column_x')} vs {_config.get('column_y')}")
+        data[label] = gen.get_structure_array_from_table(
             table,
             [
                 _config["column_x"],
@@ -93,7 +101,7 @@ def _read_table_from_model_database(table_config, db_config):
     Returns
     -------
     Table
-        Astropy table.
+        Astropy table
     """
     db = db_handler.DatabaseHandler(mongo_db_config=db_config)
     return db.export_model_file(
@@ -109,3 +117,74 @@ def _read_table_from_model_database(table_config, db_config):
 def _select_values_from_table(table, column_name, value):
     """Return a table with only the rows where column_name == value."""
     return table[np.isclose(table[column_name], value)]
+
+
+def generate_plot_configurations(
+    parameter, parameter_version, site, telescope, output_path, plot_type
+):
+    """
+    Generate plot configurations for a model parameter from schema files.
+
+    Parameters
+    ----------
+    parameter: str
+        Model parameter name.
+
+    Returns
+    -------
+    tuple
+        Tuple containing a list of plot configurations and a list of output file names.
+        Return None, if no plot configurations are found.
+    """
+    schema = gen.change_dict_keys_case(
+        gen.collect_data_from_file(
+            file_name=SCHEMA_PATH / "model_parameters" / f"{parameter}.schema.yml"
+        )
+    )
+    configs = schema.get("plot_configuration")
+    if not configs:
+        return None
+    if plot_type != "all":
+        configs = [config for config in configs if config.get("type") == plot_type]
+        if not configs:
+            raise ValueError(
+                f"No plot configuration found for type '{plot_type}' in parameter '{parameter}'."
+            )
+
+    output_files = []
+    for _config in configs:
+        for _table in _config.get("tables", []):
+            _table["parameter_version"] = parameter_version
+            _table["site"] = site
+        output_files.append(
+            _generate_output_file_name(
+                parameter=parameter,
+                parameter_version=parameter_version,
+                site=site,
+                telescope=telescope,
+                plot_type=_config.get("type"),
+                output_path=output_path,
+            )
+        )
+
+    return configs, output_files
+
+
+def _generate_output_file_name(
+    parameter,
+    parameter_version,
+    site,
+    telescope,
+    plot_type,
+    output_path=None,
+    file_extension=".pdf",
+):
+    """Generate output file name based on table file and appendix."""
+    parts = [parameter, parameter_version, site]
+    if telescope:
+        parts.append(telescope)
+    if plot_type != parameter:
+        parts.append(plot_type)
+    filename = "_".join(parts) + file_extension
+
+    return Path(output_path) / filename if output_path else Path(filename)