Merge branch 'feature/gbt-configuration' into 'main'

Create the script producing GBT configuration

See merge request !17

gemos/analysis/gbt_phase_scan.py

+"""GBT phase scan routines"""
+
+import os
+
+import pandas as pd
+
+
+def create_configuration(input_filenames, output_directory):
+    def write_configuration(df):
+        """Write the configuration file for a given OptoHybrid"""
+        fed = df.iloc[0]["fed"]
+        slot = df.iloc[0]["slot"]
+        oh = df.iloc[0]["oh"]
+
+        output_filename = "{}/fed{}-slot{}/config-oh{}-gbt.cfg".format(
+            output_directory, fed, slot, oh
+        )
+        os.makedirs(os.path.dirname(output_filename), exist_ok=True)
+
+        df.to_csv(
+            output_filename, sep=" ", index=False, header=False, columns=["config-name", "phase"]
+        )
+
+    df = pd.concat((pd.read_csv(f, sep=";") for f in input_filenames), ignore_index=True)
+
+    # Create a new column with the configuration name
+    df["config-name"] = "PHASE_VFAT" + df["vfat"].astype(str)
+
+    # Create one file per OptoHybrid
+    groups = df.groupby(["fed", "slot", "oh"])
+    groups.apply(write_configuration)

gemos/cli.py

+"""Main CLI interface"""
+
+import argparse
+import pathlib
+
+from gemos.analysis import gbt_phase_scan
+
+
+def main():
+    parser = argparse.ArgumentParser()
+    subparsers = parser.add_subparsers(required=True, dest="command")
+
+    # "create-config" command
+    configuration_parser = subparsers.add_parser("create-config")
+    configuration_subparsers = configuration_parser.add_subparsers(required=True, dest="subcommand")
+
+    # "create-config gbt" subcommand
+    parser_gbt = configuration_subparsers.add_parser("gbt")
+    parser_gbt.add_argument(
+        "inputfiles",
+        type=pathlib.Path,
+        nargs="+",
+        help="Files containing the optimal GBT phases",
+    )
+    parser_gbt.add_argument(
+        "outputdir",
+        type=pathlib.Path,
+        help="Output directory in which to store the GBT configuration files",
+    )
+    parser_gbt.set_defaults(
+        func=lambda args: gbt_phase_scan.create_configuration(args.inputfiles, args.outputdir)
+    )
+
+    # Parse the command line
+    args = parser.parse_args()
+    args.func(args)

pyproject.toml

@@ -44,6 +44,9 @@ sphinx-rtd-theme = "^0.5.2"
 
 [tool.poetry.scripts]
 gem-unpack = "gdh.unpacker:main"
+gemos = "gemos.cli:main"
+
+# Legacy interface
 gem-sbit-rate-analyze = "gemos.analysis.sbit_rate:main"
 gem-dac-scan-fit = "gemos.analysis.dac_scan:fit"
 gem-dac-scan-plot = "gemos.analysis.dac_scan:plot"