Allow more fine-grained selection of panels

NNPDF · Aug 15, 2024 · a20ed80 · a20ed80
1 parent 1fc2b0f
commit a20ed80
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 13 deletions.
diff --git a/pineappl_cli/src/plot.py b/pineappl_cli/src/plot.py
@@ -72,19 +72,19 @@
 
 def main():
     panels = [
-        {inte}plot_int,
-        {nint}plot_abs,
-        {nint}plot_rel_ewonoff,
-        {nconvs}plot_abs_pdfs,
-        {nconvs}plot_ratio_pdf,
-        {nconvs}plot_double_ratio_pdf,
-        {nconvs}plot_rel_pdfunc,
-        {nconvs}plot_rel_pdfpull,
+        {enable_int}plot_int,
+        {enable_abs}plot_abs,
+        {enable_rel_ewonoff}plot_rel_ewonoff,
+        {enable_abs_pdfs}plot_abs_pdfs,
+        {enable_ratio_pdf}plot_ratio_pdf,
+        {enable_double_ratio_pdf}plot_double_ratio_pdf,
+        {enable_rel_pdfunc}plot_rel_pdfunc,
+        {enable_rel_pdfpull}plot_rel_pdfpull,
     ]
 
     mpl.rcParams.update(stylesheet)
-    {nint}plt.rc("figure", figsize=(6.4, 2.4 * len(panels)))
-    {inte}plt.rc("figure", figsize=(4.2, 2.6))
+    {enable_abs}plt.rc("figure", figsize=(6.4, 2.4 * len(panels)))
+    {enable_int}plt.rc("figure", figsize=(4.2, 2.6))
 
     data_slices = data()
 

diff --git a/pineappl_cli/src/plot.rs b/pineappl_cli/src/plot.rs
@@ -513,11 +513,30 @@ impl Subcommand for Opts {
             let bins = grid.bin_info().bins();
             let nconvs = self.conv_funs.len();
 
+            let enable_int = if bins == 1 { "" } else { "# " };
+            let enable_abs = if bins == 1 { "# " } else { "" };
+            // TODO: only enable if there are EW corrections
+            let enable_rel_ewonoff = enable_abs;
+            let enable_abs_pdfs = if nconvs == 1 || bins == 1 { "# " } else { "" };
+            let enable_ratio_pdf = enable_abs_pdfs;
+            let enable_double_ratio_pdf = enable_abs_pdfs;
+            let enable_rel_pdfunc = if nconvs == 1 || bins == 1 || self.no_conv_fun_unc {
+                "# "
+            } else {
+                ""
+            };
+            let enable_rel_pdfpull = enable_rel_pdfunc;
+
             print!(
                 include_str!("plot.py"),
-                inte = if bins == 1 { "" } else { "# " },
-                nint = if bins == 1 { "# " } else { "" },
-                nconvs = if nconvs == 1 || bins == 1 { "# " } else { "" },
+                enable_int = enable_int,
+                enable_abs = enable_abs,
+                enable_rel_ewonoff = enable_rel_ewonoff,
+                enable_abs_pdfs = enable_abs_pdfs,
+                enable_ratio_pdf = enable_ratio_pdf,
+                enable_double_ratio_pdf = enable_double_ratio_pdf,
+                enable_rel_pdfunc = enable_rel_pdfunc,
+                enable_rel_pdfpull = enable_rel_pdfpull,
                 xlabel = xlabel,
                 ylabel = ylabel,
                 xlog = xlog,