[WIP] Implement Charge ASY for CMS 13TeV

nnpdf_data/nnpdf_data/commondata/CMS_WPWM_13TEV_ETA/data_ASY.yaml

@@ -0,0 +1,19 @@
+data_central:
+- 0.0886
+- 0.0889
+- 0.0878
+- 0.0911
+- 0.0936
+- 0.0951
+- 0.0986
+- 0.1016
+- 0.1058
+- 0.1126
+- 0.1184
+- 0.1255
+- 0.1312
+- 0.1413
+- 0.1558
+- 0.1747
+- 0.1939
+- 0.2144

nnpdf_data/nnpdf_data/commondata/CMS_WPWM_13TEV_ETA/filter.py

@@ -3,6 +3,8 @@
 
 from nnpdf_data.filter_utils.utils import prettify_float
 
+import numpy as np
+
 yaml.add_representer(float, prettify_float)
 
 
@@ -87,6 +89,76 @@ def filter_CMS_W_13TEV_uncertainties(observable, figure):
             yaml.dump(uncertainties_yaml, file, sort_keys=False)
 
 
+def get_data_CMS_W_13TEV_ASY():
+
+    data_central = []
+    with open("metadata.yaml", "r") as file:
+        metadata = yaml.safe_load(file)
+    version = metadata["hepdata"]["version"]
+    figure = "18"
+    kinematics = get_kinematics(version, figure)
+    hepdata_table = f"rawdata/HEPData-ins1810913-v{version}-Figure_{figure}.yaml"
+
+    with open(hepdata_table, "r") as f:
+        input = yaml.safe_load(f)
+
+    data_values = input["dependent_variables"][0]["values"]
+
+    for data_value in data_values:
+        data_central.append(data_value["value"])
+
+    ndata = len(data_central)
+    syst_dict = {}
+
+    # Luminosity uncertainty?
+    value_id = 0
+
+    for point in data_values:
+        for err in point["errors"]:
+            label = err["label"]
+            symerr = err['symerror']
+
+            if label not in syst_dict:
+                syst_dict[label] = np.zeros(ndata)
+
+            syst_dict[label][value_id] = symerr
+
+        value_id += 1
+
+    sys_list = []
+    for label, values in syst_dict.items():
+        sys_list.append({"name": label, "values": values.tolist()})
+
+    return data_central, kinematics, sys_list
+
+
+def filter_CMS_W_13TEV_ASY():
+    central_values, kinematics, uncertainties = get_data_CMS_W_13TEV_ASY()
+    data_central_yaml = {"data_central": central_values}
+
+    kinematics_yaml = {"bins": kinematics}
+    definitions = {
+        uncertainties[0]['name']: {
+            "description": uncertainties[0]['name'],
+            "treatment": "ADD",
+            "type": "UNCORR",
+        }
+    }
+    errors_yaml = []
+    unc_name = uncertainties[0]['name']
+    for bin in range(len(central_values)):
+        errors_yaml.append({unc_name: uncertainties[0]['values'][bin]})
+
+    uncertainties_yaml = {"definitions": definitions, "bins": errors_yaml}
+    with open("data_ASY.yaml", "w") as file:
+        yaml.dump(data_central_yaml, file, sort_keys=False)
+    with open("kinematics_ASY.yaml", "w") as file:
+        yaml.dump(kinematics_yaml, file, sort_keys=False)
+    with open("uncertainties_ASY.yaml", "w") as file:
+        yaml.dump(uncertainties_yaml, file, sort_keys=False)
+    return
+
+
 if __name__ == "__main__":
     # WP data
     filter_CMS_W_13TEV_data_kinetic(figure="17a")
@@ -95,3 +167,6 @@ def filter_CMS_W_13TEV_uncertainties(observable, figure):
     # WM data
     filter_CMS_W_13TEV_data_kinetic(figure="17b")
     filter_CMS_W_13TEV_uncertainties(observable="W-", figure="17b")
+
+    # ASY data
+    filter_CMS_W_13TEV_ASY()

nnpdf_data/nnpdf_data/commondata/CMS_WPWM_13TEV_ETA/filter_utils.py

@@ -168,5 +168,5 @@ def get_systematics(observable, version, figure):
     return uncertainties
 
 
-if __name__ == "__main__":
-    get_systematics(observable="W+", version=1, figure='17a')
+# if __name__ == "__main__":
+#     get_systematics(observable="W+", version=1, figure='17a')

nnpdf_data/nnpdf_data/commondata/CMS_WPWM_13TEV_ETA/kinematics_ASY.yaml

@@ -0,0 +1,217 @@
+bins:
+- eta:
+    min: 0.0
+    mid: 0.05
+    max: 0.1
+  m_W2:
+    min: null
+    mid: 6460.5
+    max: null
+  sqrts:
+    min: null
+    mid: 13000.0
+    max: null
+- eta:
+    min: 0.1
+    mid: 1.50000000e-01
+    max: 0.2
+  m_W2:
+    min: null
+    mid: 6460.5
+    max: null
+  sqrts:
+    min: null
+    mid: 13000.0
+    max: null
+- eta:
+    min: 0.2
+    mid: 0.25
+    max: 0.3
+  m_W2:
+    min: null
+    mid: 6460.5
+    max: null
+  sqrts:
+    min: null
+    mid: 13000.0
+    max: null
+- eta:
+    min: 0.3
+    mid: 0.35
+    max: 0.4
+  m_W2:
+    min: null
+    mid: 6460.5
+    max: null
+  sqrts:
+    min: null
+    mid: 13000.0
+    max: null
+- eta:
+    min: 0.4
+    mid: 0.45
+    max: 0.5
+  m_W2:
+    min: null
+    mid: 6460.5
+    max: null
+  sqrts:
+    min: null
+    mid: 13000.0
+    max: null
+- eta:
+    min: 0.5
+    mid: 0.55
+    max: 0.6
+  m_W2:
+    min: null
+    mid: 6460.5
+    max: null
+  sqrts:
+    min: null
+    mid: 13000.0
+    max: null
+- eta:
+    min: 0.6
+    mid: 6.50000000e-01
+    max: 0.7
+  m_W2:
+    min: null
+    mid: 6460.5
+    max: null
+  sqrts:
+    min: null
+    mid: 13000.0
+    max: null
+- eta:
+    min: 0.7
+    mid: 0.75
+    max: 0.8
+  m_W2:
+    min: null
+    mid: 6460.5
+    max: null
+  sqrts:
+    min: null
+    mid: 13000.0
+    max: null
+- eta:
+    min: 0.8
+    mid: 8.50000000e-01
+    max: 0.9
+  m_W2:
+    min: null
+    mid: 6460.5
+    max: null
+  sqrts:
+    min: null
+    mid: 13000.0
+    max: null
+- eta:
+    min: 0.9
+    mid: 0.95
+    max: 1.0
+  m_W2:
+    min: null
+    mid: 6460.5
+    max: null
+  sqrts:
+    min: null
+    mid: 13000.0
+    max: null
+- eta:
+    min: 1.0
+    mid: 1.05
+    max: 1.1
+  m_W2:
+    min: null
+    mid: 6460.5
+    max: null
+  sqrts:
+    min: null
+    mid: 13000.0
+    max: null
+- eta:
+    min: 1.1
+    mid: 1.15
+    max: 1.2
+  m_W2:
+    min: null
+    mid: 6460.5
+    max: null
+  sqrts:
+    min: null
+    mid: 13000.0
+    max: null
+- eta:
+    min: 1.2
+    mid: 1.25
+    max: 1.3
+  m_W2:
+    min: null
+    mid: 6460.5
+    max: null
+  sqrts:
+    min: null
+    mid: 13000.0
+    max: null
+- eta:
+    min: 1.3
+    mid: 1.4
+    max: 1.5
+  m_W2:
+    min: null
+    mid: 6460.5
+    max: null
+  sqrts:
+    min: null
+    mid: 13000.0
+    max: null
+- eta:
+    min: 1.5
+    mid: 1.6
+    max: 1.7
+  m_W2:
+    min: null
+    mid: 6460.5
+    max: null
+  sqrts:
+    min: null
+    mid: 13000.0
+    max: null
+- eta:
+    min: 1.7
+    mid: 1.80000000e+00
+    max: 1.9
+  m_W2:
+    min: null
+    mid: 6460.5
+    max: null
+  sqrts:
+    min: null
+    mid: 13000.0
+    max: null
+- eta:
+    min: 1.9
+    mid: 2.0
+    max: 2.1
+  m_W2:
+    min: null
+    mid: 6460.5
+    max: null
+  sqrts:
+    min: null
+    mid: 13000.0
+    max: null
+- eta:
+    min: 2.1
+    mid: 2.25
+    max: 2.4
+  m_W2:
+    min: null
+    mid: 6460.5
+    max: null
+  sqrts:
+    min: null
+    mid: 13000.0
+    max: null

nnpdf_data/nnpdf_data/commondata/CMS_WPWM_13TEV_ETA/metadata.yaml

@@ -94,3 +94,40 @@ implemented_observables:
     FK_tables:
     - - CMS_WPWM_13TEV_ETA_WM
     operation: 'null'
+
+- observable_name: "ASY"
+  observable:
+    description: "CMS 13 TeV, charge asymmetry. Combination of muon and electron channel"
+    label: r"A"
+    units: " "
+
+  ndata: 18
+
+  tables: [18]
+  process_type: DY_W_ETA
+
+  plotting:
+    dataset_label: 'CMS 13 TeV $W$ charge asymmetry'
+    plot_x: eta
+
+  kinematic_coverage: [eta, m_W2, sqrts]
+
+  kinematics:
+    variables:
+      eta: {description: "W > l v pseudorapidity", label: '$\eta$', units: ""}
+      m_W2: {description: "W boson mass squared", label: '$M_W^2$', units: "GeV"}
+      sqrts: {description: "center of mass energy", label: '$\sqrt(s)$', units: "GeV"}
+    file: kinematics_ASY.yaml
+
+    # Data
+  data_central: data_ASY.yaml
+  data_uncertainties:
+  - uncertainties_ASY.yaml
+
+
+    # Theory
+  theory:
+    FK_tables:
+    - - CMS_WPWM_13TEV_ETA_WP
+    - - CMS_WPWM_13TEV_ETA_WM
+    operation: ASY