generate.py

import argparse
import importlib
import os


class SplitArgs(argparse.Action):
    def __call__(self, parser, namespace, values, option_string=None):
        setattr(namespace, self.dest, values.split(","))


parser = argparse.ArgumentParser(description="Generate the C++ code for a given config")
parser.add_argument("--template", type=str, help="Path to the template")
parser.add_argument("--subset-template", type=str, help="Path to the subset template")
parser.add_argument("--output", type=str, help="Path to the output directory")
parser.add_argument("--analysis", type=str, help="Name of the analysis")
parser.add_argument("--config", type=str, help="Name of the config")
parser.add_argument(
    "--analysis-folder",
    type=str,
    help="The location of all analysis folders",
    default="analysis_configurations",
)
parser.add_argument(
    "--scopes",
    type=str,
    action=SplitArgs,
    help="Channels to be activated. To select multiple scopes, provide a comma separated list.",
)
parser.add_argument(
    "--shifts",
    type=str,
    action=SplitArgs,
    help='Shifts to be activated. To select multiple shifts, provide a comma separated list. To select all, choose "all"',
)
parser.add_argument(
    "--sample",
    type=str,
    help="Sample to be processed",
)
parser.add_argument(
    "--era",
    type=str,
    help="Era to be processed",
)
parser.add_argument("--threads", type=int, help="number of threads to be used")
parser.add_argument("--debug", type=str, help="set debug mode for building")
args = parser.parse_args()

# find available analyses, every folder in analysis_configurations is an analysis
available_analysis = [
    os.path.basename(f.path)
    for f in os.scandir(
        os.path.join(os.path.dirname(os.path.abspath(__file__)), args.analysis_folder)
    )
    if f.is_dir()
]

# sanitize arguments
available_analysis = [x.lower() for x in available_analysis]

analysis = args.analysis.lower()
# first check if the analysis is available
if analysis not in available_analysis:
    raise ValueError(
        f"The analysis {analysis} is not available. Available analysiss are: {available_analysis}"
    )
else:
    # load and run the generate.py function of the analysis
    if not os.path.exists(
        os.path.join(
            os.path.dirname(os.path.abspath(__file__)),
            args.analysis_folder,
            analysis,
            "generate.py",
        )
    ):
        raise ValueError(f"The generate.py file for analysis {analysis} does not exist")
    generator = importlib.import_module(f"analysis_configurations.{analysis}.generate")
    generator.run(args)