Added ability to disable a script for comparisons

Author: JBorrow

CHANGELOG.txt

@@ -1,5 +1,16 @@
 Changelog for the ``swiftpipeline`` repository.
 
+Version 0.2.0
+-------------
+
+Titles update!
+
++ Adds ability to overwrite run names in output, to enable easier comparisons.
++ Adds by default a redshift tag to each simulation name when comparing different
+  snapshots that have different redshifts.
++ Adds ability to disable scripts for comparisons (`use_for_comparison: false` in
+  yaml file).
+
 Version 0.1.8
 -------------
 

example/config/config.yml

@@ -28,5 +28,6 @@ scripts:
     output_file: density_temperature.png
     section: Density-Temperature
     title: Density-Temperature
+    use_for_comparison: false
     additional_arguments:
       quantity_type: hydro

swift-pipeline

@@ -194,7 +194,9 @@ if __name__ == "__main__":
         f"{config.config_directory}/{config.observational_data_directory}/data"
     )
 
-    if len(args.snapshots) == 1:
+    is_comparison = len(args.snapshots) > 1
+
+    if not is_comparison:
         # Run the pipeline based on the arguments if only a single simulation is
         # included and generate the metadata yaml file.
         print_if_debug(
@@ -313,7 +315,9 @@ if __name__ == "__main__":
     if args.debug:
         config.scripts = tqdm(config.scripts, desc="Running Scripts")
 
-    for script in config.scripts:
+    scripts_to_use = config.comparison_scripts if is_comparison else config.scripts
+
+    for script in scripts_to_use:
         full_script_path = f"{config.config_directory}/{script.filename}"
 
         run(
@@ -340,7 +344,7 @@ if __name__ == "__main__":
     print_if_debug("Creating webpage.")
     webpage = WebpageCreator()
     webpage.add_auto_plotter_metadata(auto_plotter_metadata=auto_plotter_metadata)
-    webpage.add_config_metadata(config=config)
+    webpage.add_config_metadata(config=config, is_comparison=is_comparison)
     webpage.add_metadata(page_name=" | ".join(run_names))
     webpage.add_run_metadata(config=config, snapshots=snapshots)
     webpage.render_webpage()

swiftpipeline/config.py

@@ -38,6 +38,9 @@ class Script(object):
     show_on_webpage: bool
     # additional arguments to be fed to a given script
     additional_arguments: dict
+    # Use in the case where we have comparisons? The scripts may be disabled
+    # in comparison cases for performance reasons.
+    use_for_comparison: bool
 
     def __init__(self, script_dict: dict):
         """
@@ -51,6 +54,7 @@ def __init__(self, script_dict: dict):
         self.title = script_dict.get("title", "")
         self.show_on_webpage = script_dict.get("show_on_webpage", True)
         self.additional_arguments = script_dict.get("additional_arguments", {})
+        self.use_for_comparison = script_dict.get("use_for_comparison", True)
         return
 
     def __str__(self):
@@ -85,7 +89,7 @@ class Config(object):
 
     # Raw config read directly from the file, before processing.
     raw_config: dict
-    scripts: List[Script]
+    raw_scripts: List[Script]
 
     # Set up the object.
     __slots__ = list(direct_read.keys()) + ["scripts", "config_directory", "raw_config"]
@@ -143,6 +147,23 @@ def __extract_scripts(self):
         """
 
         raw_scripts = self.raw_config.get("scripts", [])
-        self.scripts = [Script(script_dict=script_dict) for script_dict in raw_scripts]
+        self.raw_scripts = [
+            Script(script_dict=script_dict) for script_dict in raw_scripts
+        ]
 
         return
+
+    @property
+    def scripts(self):
+        """
+        Gets all of the scripts defined in the parameter file.
+        """
+        return self.raw_scripts
+
+    @property
+    def comparison_scripts(self):
+        """
+        Gets the scripts only to be used in comparisons from the parameter
+        file.
+        """
+        return [script for script in self.raw_scripts if script.use_for_comparison]

swiftpipeline/html.py

@@ -196,7 +196,7 @@ def add_auto_plotter_metadata(self, auto_plotter_metadata: AutoPlotterMetadata):
 
         return
 
-    def add_config_metadata(self, config: Config):
+    def add_config_metadata(self, config: Config, is_comparison: bool = False):
         """
         Adds the section metadata from the additional plots
         defined under ``config.yml::scripts``.
@@ -206,13 +206,18 @@ def add_config_metadata(self, config: Config):
 
         config: Config
             Configuration object from ``swift-pipeline``.
+
+        is_comparison: bool, optional
+            Is this webpage being created for a comparison? Default: False.
         """
 
         self.config = config
 
         # Unique sections
         sections = {script.section for script in config.scripts}
 
+        scripts_to_use = config.comparison_scripts if is_comparison else config.scripts
+
         for section in sections:
             plots = [
                 dict(
@@ -221,7 +226,7 @@ def add_config_metadata(self, config: Config):
                     caption=script.caption,
                     hash=abs(hash(script.caption + script.title)),
                 )
-                for script in config.scripts
+                for script in scripts_to_use
                 if script.section == section and script.show_on_webpage
             ]