Add --repair switch to the CLI's write

Author: cschwan

CHANGELOG.md

@@ -10,7 +10,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Added
 
 - added `Grid::repair` to repair bugs that survived by writing bugged grids to
-  disk, for example <https://github.com/NNPDF/pineappl/issues/338>
+  disk, for example <https://github.com/NNPDF/pineappl/issues/338>. The CLI
+  offers this functionality via `pineappl write --repair`
 
 ## [1.1.0] - 08/07/2025
 

pineappl_cli/src/write.rs

@@ -41,6 +41,7 @@ enum OpsArg {
     MulBinNorm(f64),
     Optimize(bool),
     OptimizeFkTable(FkAssumptions),
+    Repair(bool),
     RewriteChannel((usize, Channel)),
     RewriteOrder((usize, Order)),
     RotatePidBasis(PidBasis),
@@ -85,7 +86,7 @@ impl FromArgMatches for MoreArgs {
                         });
                     }
                 }
-                "merge_channel_factors" | "optimize" | "split_channels" | "upgrade" => {
+                "merge_channel_factors" | "optimize" | "repair" | "split_channels" | "upgrade" => {
                     let arguments: Vec<Vec<_>> = matches
                         .remove_occurrences(&id)
                         .unwrap()
@@ -98,6 +99,7 @@ impl FromArgMatches for MoreArgs {
                         args[index] = Some(match id.as_str() {
                             "merge_channel_factors" => OpsArg::MergeChannelFactors(arg[0]),
                             "optimize" => OpsArg::Optimize(arg[0]),
+                            "repair" => OpsArg::Repair(arg[0]),
                             "split_channels" => OpsArg::SplitChannels(arg[0]),
                             "upgrade" => OpsArg::Upgrade(arg[0]),
                             _ => unreachable!(),
@@ -394,6 +396,17 @@ impl Args for MoreArgs {
                     .try_map(|s| s.parse::<FkAssumptions>()),
                 ),
         )
+        .arg(
+            Arg::new("repair")
+                .action(ArgAction::Append)
+                .default_missing_value("true")
+                .help("Repair bugs saved in the grid")
+                .long("repair")
+                .num_args(0..=1)
+                .require_equals(true)
+                .value_name("ENABLE")
+                .value_parser(clap::value_parser!(bool)),
+        )
         .arg(
             Arg::new("rewrite_channel")
                 .action(ArgAction::Append)
@@ -583,6 +596,9 @@ impl Subcommand for Opts {
                     // UNWRAP: this cannot fail because we only modify the normalizations
                     .unwrap();
                 }
+                OpsArg::Repair(true) => {
+                    grid.repair();
+                }
                 OpsArg::RewriteChannel((index, new_channel)) => {
                     // TODO: check that `index` is valid
                     grid.channels_mut()[*index] = new_channel.clone();
@@ -618,6 +634,7 @@ impl Subcommand for Opts {
                 OpsArg::SplitChannels(true) => grid.split_channels(),
                 OpsArg::Upgrade(true) => grid.upgrade(),
                 OpsArg::MergeChannelFactors(false)
+                | OpsArg::Repair(false)
                 | OpsArg::Optimize(false)
                 | OpsArg::SplitChannels(false)
                 | OpsArg::Upgrade(false) => {}

pineappl_cli/tests/write.rs

@@ -24,6 +24,7 @@ Options:
       --mul-bin-norm <NORM>             Multiply all bin normalizations with the given factor
       --optimize[=<ENABLE>]             Optimize internal data structure to minimize memory and disk usage [possible values: true, false]
       --optimize-fk-table <OPTIMI>      Optimize internal data structure of an FkTable to minimize memory and disk usage [possible values: Nf6Ind, Nf6Sym, Nf5Ind, Nf5Sym, Nf4Ind, Nf4Sym, Nf3Ind, Nf3Sym]
+      --repair[=<ENABLE>]               Repair bugs saved in the grid [possible values: true, false]
       --rewrite-channel <IDX> <CHAN>    Rewrite the definition of the channel with index IDX
       --rewrite-order <IDX> <ORDER>     Rewrite the definition of the order with index IDX
       --rotate-pid-basis <BASIS>        Rotate the PID basis for this grid [possible values: PDG, EVOL]
@@ -876,6 +877,23 @@ fn multiple_arguments() {
         .stdout(MULTIPLE_ARGUMENTS_STR);
 }
 
+#[test]
+fn repair() {
+    let output = NamedTempFile::new("repaired-grid.pineappl.lz4").unwrap();
+
+    Command::cargo_bin("pineappl")
+        .unwrap()
+        .args([
+            "write",
+            "--repair",
+            "../test-data/LHCB_WP_7TEV_opt.pineappl.lz4",
+            output.path().to_str().unwrap(),
+        ])
+        .assert()
+        .success()
+        .stdout("");
+}
+
 #[test]
 fn rewrite_channels() {
     let output = NamedTempFile::new("ckm_channels.pineappl.lz4").unwrap();