Allow pineko to check if the grid contains SV

pyproject.toml

@@ -25,7 +25,7 @@ packages = [{ include = "pineko", from = "src" }]
 [tool.poetry.dependencies]
 python = ">=3.8,<3.11"
 eko = "^0.8.5"
-pineappl = "^0.5.2"
+pineappl = "^0.5.4"
 PyYAML = "^6.0"
 numpy = "^1.21.0"
 pandas = "^1.4.1"

src/pineko/check.py

@@ -65,3 +65,31 @@ def check_grid_and_eko_compatible(pineappl_grid, operators, xif):
     # x-grid
     if not np.all(in1d(np.unique(operators["targetgrid"]), np.array(x_grid))):
         raise ValueError("x grid in pineappl grid and eko operator are NOT compatible!")
+
+
+def check_grid_contains_fact_sv(grid):
+    """Checks whether factorization scale-variations are available in the pineappl grid.
+    Parameters
+    ----------
+        grid: pineappl.grid.Grid
+            Pineappl grid
+    """
+    order_list = [order.as_tuple() for order in grid.orders()]
+    for order in order_list:
+        if order[-1] != 0:
+            return
+    raise ValueError("Factorization scale variations are not available for this grid")
+
+
+def check_grid_contains_ren_sv(grid):
+    """Checks whether renormalization scale-variations are available in the pineappl grid.
+    Parameters
+    ----------
+        grid: pineappl.grid.Grid
+            Pineappl grid
+    """
+    order_list = [order.as_tuple() for order in grid.orders()]
+    for order in order_list:
+        if order[-2] != 0:
+            return
+    raise ValueError("Renormalization scale variations are not available for this grid")

src/pineko/cli/check.py

@@ -28,3 +28,33 @@ def subcommand(pineappl_path, eko_path, xif):
         rich.print("[green]Success:[/] grids are compatible")
     except ValueError as e:
         rich.print("[red]Error:[/]", e)
+
+
+@command.command("check_scalevar")
+@click.argument("pineappl_path", metavar="PINEAPPL", type=click.Path(exists=True))
+@click.argument(
+    "tocheck",
+    metavar="TOCHECK",
+    type=str,
+)
+def subcommand_sv(pineappl_path, tocheck):
+    """Check if PineAPPL grid contains requested scale variations"""
+    pineappl_grid = pineappl.grid.Grid.read(pineappl_path)
+    if tocheck == "xir":
+        try:
+            check.check_grid_contains_ren_sv(pineappl_grid)
+            rich.print(
+                "[green]Success:[/] grids contain renormalization scale variations"
+            )
+        except ValueError as e:
+            rich.print("[red]Error:[/]", e)
+    elif tocheck == "xif":
+        try:
+            check.check_grid_contains_fact_sv(pineappl_grid)
+            rich.print(
+                "[green]Success:[/] grids contain factorization scale variations"
+            )
+        except ValueError as e:
+            rich.print("[red]Error:[/]", e)
+    else:
+        raise ValueError("Scale variation to check can be one between xir and xif")

src/pineko/cli/convolute.py

@@ -1,13 +1,15 @@
 # -*- coding: utf-8 -*-
 """CLI entry point to convolution."""
 import click
+import pineappl
+import rich
 
 from .. import evolve
 from ._base import command
 
 
 @command.command("convolute")
-@click.argument("pineappl", type=click.Path(exists=True))
+@click.argument("pine", type=click.Path(exists=True))
 @click.argument("eko", type=click.Path(exists=True))
 @click.argument("fktable", type=click.Path())
 @click.argument("max_as", type=int)
@@ -23,10 +25,10 @@
     help="the flavor assumptions to be used",
     show_default=True,
 )
-def subcommand(pineappl, eko, fktable, max_as, max_al, xir, xif, pdf, assumptions):
+def subcommand(pine, eko, fktable, max_as, max_al, xir, xif, pdf, assumptions):
     """Convolute PineAPPL grid and EKO into an FK table.
 
-    PINEAPPL and EKO are the path to the respective elements to convolute, and
+    PINE and EKO are the path to the respective elements to convolute, and
     FKTABLE is the path where to dump the output.
 
     MAX_AS and MAX_AL are used to specify the order in QCD and QED
@@ -38,8 +40,16 @@ def subcommand(pineappl, eko, fktable, max_as, max_al, xir, xif, pdf, assumption
 
     PDF is an optional PDF set compatible with the EKO to compare grid and FK table.
     """
+    pineappl_grid = pineappl.grid.Grid.read(pine)
+    rich.print(
+        rich.panel.Panel.fit("Computing ...", style="magenta", box=rich.box.SQUARE),
+        f"   {pineappl}\n",
+        f"+ {eko}\n",
+        f"= {fktable}\n",
+        f"with max_as={max_as}, max_al={max_al}, xir={xir}, xif={xif}",
+    )
     _grid, _fk, comp = evolve.evolve_grid(
-        pineappl,
+        pineappl_grid,
         eko,
         fktable,
         max_as,

src/pineko/evolve.py

@@ -77,7 +77,7 @@ def write_operator_card(pineappl_grid, default_card, card_path, xi):
 
 
 def evolve_grid(
-    pineappl_path,
+    pineappl_grid,
     eko_path,
     fktable_path,
     max_as,
@@ -92,7 +92,7 @@ def evolve_grid(
 
     Parameters
     ----------
-    pineappl_path : str
+    pineappl_grid : pineappl.grid.Grid
         unconvoluted grid
     eko_path : str
         evolution operator
@@ -113,15 +113,6 @@ def evolve_grid(
     comparison_pdf : None or str
         if given, a comparison table (with / without evolution) will be printed
     """
-    rich.print(
-        rich.panel.Panel.fit("Computing ...", style="magenta", box=rich.box.SQUARE),
-        f"   {pineappl_path}\n",
-        f"+ {eko_path}\n",
-        f"= {fktable_path}\n",
-        f"with max_as={max_as}, max_al={max_al}, xir={xir}, xif={xif}",
-    )
-    # load
-    pineappl_grid = pineappl.grid.Grid.read(str(pineappl_path))
     _x_grid, _pids, mur2_grid, _muf2_grid = pineappl_grid.axes()
     operators = eko.output.Output.load_tar(eko_path)
     check.check_grid_and_eko_compatible(pineappl_grid, operators, xif)

src/pineko/theory.py

@@ -15,7 +15,7 @@
 import yaml
 from eko import strong_coupling as sc
 
-from . import configs, evolve, parser, theory_card
+from . import check, configs, evolve, parser, theory_card
 
 logger = logging.getLogger(__name__)
 
@@ -346,6 +346,15 @@ def fk(self, name, grid_path, tcard, pdf):
         do_log = self.activate_logging(
             paths["logs"]["fk"], f"{self.theory_id}-{name}-{pdf}.log"
         )
+        # check if grid contains SV if theory is requesting them
+        xir = tcard["XIR"]
+        xif = tcard["XIF"]
+        ftr = tcard["fact_to_ren_scale_ratio"]
+        pineappl_grid = pineappl.grid.Grid.read(grid_path)
+        if not np.isclose(xir, 1.0):
+            check.check_grid_contains_ren_sv(pineappl_grid)
+        if not (np.isclose(xif, 1.0) and np.isclose(ftr, 1.0)):
+            check.check_grid_contains_fact_sv(pineappl_grid)
         # setup data
         eko_filename = self.ekos_path() / f"{name}.tar"
         fk_filename = self.fks_path / f"{name}.{parser.EXT}"
@@ -362,8 +371,6 @@ def fk(self, name, grid_path, tcard, pdf):
         # q2_grid = ocard["Q2grid"]
         operators = eko.output.Output.load_tar(eko_filename)
         q2_grid = operators["Q2grid"].keys()
-        xir = tcard["XIR"]
-        xif = tcard["XIF"]
         # PineAPPL wants alpha_s = 4*pi*a_s
         alphas_values = [
             4.0 * np.pi * astrong.a_s(xir * xir * Q2 / xif / xif) for Q2 in q2_grid
@@ -374,8 +381,16 @@ def fk(self, name, grid_path, tcard, pdf):
         logger.info("Start computation of %s", name)
         logger.info("max_as=%d, max_al=%d, xir=%f, xif=%f", max_as, max_al, xir, xif)
         start_time = time.perf_counter()
+
+        rich.print(
+            rich.panel.Panel.fit("Computing ...", style="magenta", box=rich.box.SQUARE),
+            f"   {grid_path}\n",
+            f"+ {eko_filename}\n",
+            f"= {fk_filename}\n",
+            f"with max_as={max_as}, max_al={max_al}, xir={xir}, xif={xif}",
+        )
         _grid, _fk, comparison = evolve.evolve_grid(
-            grid_path,
+            pineappl_grid,
             eko_filename,
             fk_filename,
             max_as,