Add runcards (and runner) for integrability observables

pinecards/NNPDF_INTEG_XT3_40/integrability.yaml

@@ -0,0 +1,5 @@
+hadron_pid: 2212
+lepton_pid: 11
+flavour: 103
+xgrid:
+  - 1.0e-9

pinecards/NNPDF_INTEG_XT3_40/metadata.txt

@@ -0,0 +1,4 @@
+description=Integrability for T3
+y_label=xT3
+y_label_tex=$\sum_i(x_i T_3(x_i))$
+y_unit=

pinecards/NNPDF_INTEG_XT8_40/integrability.yaml

@@ -0,0 +1,5 @@
+hadron_pid: 2212
+lepton_pid: 11
+flavour: 108
+xgrid:
+  - 1.0e-9

pinecards/NNPDF_INTEG_XT8_40/metadata.txt

@@ -0,0 +1,4 @@
+description=Integrability for T8
+y_label=xT8
+y_label_tex=$\sum_i(x_i T_8(x_i))$
+y_unit=

pinefarm/external/integrability.py

@@ -0,0 +1,104 @@
+# -*- coding: utf-8 -*-
+import dataclasses
+import json
+import typing
+
+import lhapdf
+import numpy as np
+import pandas as pd
+import pineappl
+import yaml
+from eko import basis_rotation as br
+
+from .. import configs
+from . import interface
+
+_RUNCARD = "integrability.yaml"
+
+
+def is_integrability(name):
+    """Is this an integrability dataset?
+    The decision is based on the existence of `integrability.yaml`"""
+    return (configs.configs["paths"]["runcards"] / name / _RUNCARD).exists()
+
+
+def evolution_to_flavour(evol_fl):
+    """Uses eko to transform an evolution pid (>100) to flavour basis
+    in a pineappl-usable way (flavour, weight)
+    """
+    if evol_fl < 100:
+        return [(evol_fl, 1.0)]
+    idx = br.evol_basis_pids.index(evol_fl)
+    row = br.rotate_flavor_to_evolution[idx]
+    lumis = []
+    for i, w in enumerate(row):
+        if w != 0.0:
+            lumis.append((br.flavor_basis_pids[i], w))
+    return lumis
+
+
+@dataclasses.dataclass
+class _IntegrabilityRuncard:
+    hadron_pid: int
+    lepton_pid: int
+    flavour: int
+    xgrid: typing.List[float]
+
+    def asdict(self):
+        return dataclasses.asdict(self)
+
+
+class Integrability(interface.External):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+        input_card = self.source / _RUNCARD
+        yaml_dict = yaml.safe_load(input_card.open("r", encoding="utf-8"))
+        self._q2 = np.power(self.theory["Q0"], 2)
+        self._info = _IntegrabilityRuncard(**yaml_dict)
+        self._evo2fl = evolution_to_flavour(self._info.flavour)
+
+    def run(self):
+        pass
+
+    def generate_pineappl(self):
+        """Generate the pineappl grid for the integrability observable"""
+        ## Generate the grid
+        lumi_entries = [(fl, self._info.lepton_pid, w) for fl, w in self._evo2fl]
+        luminosities = [pineappl.lumi.LumiEntry(lumi_entries)]
+        # Set default parameters
+        orders = [pineappl.grid.Order(0, 0, 0, 0)]
+        params = pineappl.subgrid.SubgridParams()
+        # Initialize and parametrize grid
+        grid = pineappl.grid.Grid.create(luminosities, orders, [0.0, 1.0], params)
+        grid.set_key_value("initial_state_1", str(self._info.hadron_pid))
+        grid.set_key_value("initial_state_2", str(self._info.lepton_pid))
+        grid.set_key_value("runcard", json.dumps(self._info.asdict()))
+        grid.set_key_value("lumi_id_types", "pdg_mc_ids")
+        # Fill grid with x*Tn
+        # use subgrid because fill doesn't work?
+        x = self._info.xgrid
+        w = np.array(x).reshape((1, -1, 1))
+        sg = pineappl.import_only_subgrid.ImportOnlySubgridV1(w, [self._q2], x, x)
+        grid.set_subgrid(0, 0, 0, sg)
+        grid.write(self.grid)
+
+    def collect_versions(self):
+        return {"integrability_version": "1.0"}
+
+    def results(self):
+        pdf = lhapdf.mkPDF(self.pdf)
+        final_result = 0.0
+        q2 = self._q2 * np.ones_like(self._info.xgrid)
+
+        for fl, w in self._evo2fl:
+            final_result += w * np.sum(pdf.xfxQ2(fl, self._info.xgrid, q2))
+        final_cv = [final_result]
+
+        d = {
+            "result": final_cv,
+            "error": np.zeros_like(final_cv),
+            "sv_min": np.zeros_like(final_cv),
+            "sv_max": np.zeros_like(final_cv),
+        }
+        return pd.DataFrame(data=d)

pinefarm/info.py

@@ -3,14 +3,15 @@
 import enum
 import typing
 
-from .external import interface, mg5, positivity, vrap, yad
+from .external import integrability, interface, mg5, positivity, vrap, yad
 
 
 class Kind(enum.Enum):
     dis = enum.auto()
     positivity = enum.auto()
     ftdy = enum.auto()
     hadronic = enum.auto()
+    integrability = enum.auto()
 
 
 @dataclasses.dataclass
@@ -29,5 +30,9 @@ def label(dataset: str) -> Info:
         )
     if vrap.is_vrap(dataset):
         return Info(color="green", external=vrap.Vrap, kind=Kind.ftdy)
+    if integrability.is_integrability(dataset):
+        return Info(
+            color="brown", external=integrability.Integrability, kind=Kind.integrability
+        )
 
     return Info(color="blue", external=mg5.Mg5, kind=Kind.hadronic)