From 67ed4433d60da71bed11f4e8fdfb9abf5472bc2a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Nicol=C3=B2=20Jacazio?= <nicolo.jacazio@cern.ch>
Date: Mon, 22 Nov 2021 16:18:51 +0100
Subject: [PATCH] Include track selection in efficiency computation

---
 MC/bin/o2dpg_sim_workflow.py          |   2 +-
 MC/config/QC/json/event-track-qa.json | 116 ++++++++++++++++++++++++++
 2 files changed, 117 insertions(+), 1 deletion(-)

diff --git a/MC/bin/o2dpg_sim_workflow.py b/MC/bin/o2dpg_sim_workflow.py
index 3c2830da8..560aa1dd0 100755
--- a/MC/bin/o2dpg_sim_workflow.py
+++ b/MC/bin/o2dpg_sim_workflow.py
@@ -842,7 +842,7 @@ def addAnalysisTask(tag, cmd, output, needs=[AOD_merge_task['name']],
 
    # Efficiency
    addAnalysisTask(tag="Efficiency",
-                   cmd="o2-analysis-qa-efficiency --make-eff 1 --eff-pi 1 --eff-el 1 --eff-ka 1 --eff-pr 1 --eff-mu 1 --eff-de 1 --eff-he 1 --eff-tr 1 --eta-min -0.8 --eta-max 0.8 --log-pt 1 --pt-bins 100 --pt-min 0.01 --pt-max 10. --sel-prim 1", output="AnalysisResults.root")
+                   cmd="o2-analysis-trackextension --configuration json://${O2DPG_ROOT}/MC/config/QC/json/event-track-qa.json | o2-analysis-trackselection --configuration json://${O2DPG_ROOT}/MC/config/QC/json/event-track-qa.json | o2-analysis-qa-efficiency --eff-el 1 --eff-mu 1 --eff-pi 1 --eff-ka 1 --eff-pr 1 --eff-de 1 --eff-tr 1 --eff-he 1 --configuration json://${O2DPG_ROOT}/MC/config/QC/json/event-track-qa.json ", output="AnalysisResults.root")
 
    # Event and track QA
    addAnalysisTask(tag="EventTrackQA",
diff --git a/MC/config/QC/json/event-track-qa.json b/MC/config/QC/json/event-track-qa.json
index ecf864a21..2a28f345a 100644
--- a/MC/config/QC/json/event-track-qa.json
+++ b/MC/config/QC/json/event-track-qa.json
@@ -3,6 +3,122 @@
         "processRun2": "false",
         "processRun3": "true"
     },
+    "qa-tracking-efficiency-data": {
+        "eta-min": "-0.8",
+        "eta-max": "0.8",
+        "pt-min": "0.1",
+        "pt-max": "10",
+        "track-sel": "true",
+        "pt-bins": "500",
+        "log-pt": "1",
+        "eta-bins": "500",
+        "phi-bins": "500",
+        "make-eff": "true"
+    },
+    "qa-tracking-efficiency-muon": {
+        "eta-min": "-0.8",
+        "eta-max": "0.8",
+        "pt-min": "0.1",
+        "pt-max": "10",
+        "track-sel": "true",
+        "pt-bins": "500",
+        "log-pt": "1",
+        "eta-bins": "500",
+        "phi-bins": "500",
+        "make-eff": "true",
+        "sel-prim": "true"
+    },
+    "qa-tracking-efficiency-pion": {
+        "eta-min": "-0.8",
+        "eta-max": "0.8",
+        "pt-min": "0.1",
+        "pt-max": "10",
+        "track-sel": "true",
+        "pt-bins": "500",
+        "log-pt": "1",
+        "eta-bins": "500",
+        "phi-bins": "500",
+        "make-eff": "true",
+        "sel-prim": "true"
+    },
+    "qa-tracking-efficiency-electron": {
+        "eta-min": "-0.8",
+        "eta-max": "0.8",
+        "pt-min": "0.1",
+        "pt-max": "10",
+        "track-sel": "true",
+        "pt-bins": "500",
+        "log-pt": "1",
+        "eta-bins": "500",
+        "phi-bins": "500",
+        "make-eff": "true",
+        "sel-prim": "true"
+    },
+    "qa-tracking-efficiency-kaon": {
+        "eta-min": "-0.8",
+        "eta-max": "0.8",
+        "pt-min": "0.1",
+        "pt-max": "10",
+        "track-sel": "true",
+        "pt-bins": "500",
+        "log-pt": "1",
+        "eta-bins": "500",
+        "phi-bins": "500",
+        "make-eff": "true",
+        "sel-prim": "true"
+    },
+    "qa-tracking-efficiency-proton": {
+        "eta-min": "-0.8",
+        "eta-max": "0.8",
+        "pt-min": "0.1",
+        "pt-max": "10",
+        "track-sel": "true",
+        "pt-bins": "500",
+        "log-pt": "1",
+        "eta-bins": "500",
+        "phi-bins": "500",
+        "make-eff": "true",
+        "sel-prim": "true"
+    },
+    "qa-tracking-efficiency-deuteron": {
+        "eta-min": "-0.8",
+        "eta-max": "0.8",
+        "pt-min": "0.1",
+        "pt-max": "10",
+        "track-sel": "true",
+        "pt-bins": "500",
+        "log-pt": "1",
+        "eta-bins": "500",
+        "phi-bins": "500",
+        "make-eff": "true",
+        "sel-prim": "true"
+    },
+    "qa-tracking-efficiency-triton": {
+        "eta-min": "-0.8",
+        "eta-max": "0.8",
+        "pt-min": "0.1",
+        "pt-max": "10",
+        "track-sel": "true",
+        "pt-bins": "500",
+        "log-pt": "1",
+        "eta-bins": "500",
+        "phi-bins": "500",
+        "make-eff": "true",
+        "sel-prim": "true"
+    },
+    "qa-tracking-efficiency-helium3": {
+        "eta-min": "-0.8",
+        "eta-max": "0.8",
+        "pt-min": "0.1",
+        "pt-max": "10",
+        "track-sel": "true",
+        "pt-bins": "500",
+        "log-pt": "1",
+        "eta-bins": "500",
+        "phi-bins": "500",
+        "make-eff": "true",
+        "sel-prim": "true"
+    },
     "event-selection-task": {
         "syst": "pp",
         "muonSelection": "0",