Update

Author: njacazio

ALICE3/Tasks/CMakeLists.txt

@@ -40,7 +40,7 @@ o2physics_add_dpl_workflow(alice3-pid-ftof-qa
                     COMPONENT_NAME Analysis)
 
 o2physics_add_dpl_workflow(alice3-pid-separation-power
-                    SOURCES pidSeparationPower.cxx
+                    SOURCES alice3SeparationPower.cxx
                     PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore
                     COMPONENT_NAME Analysis)
 

ALICE3/Tasks/alice3Efficiency.cxx

@@ -30,18 +30,24 @@
 using namespace o2;
 using namespace o2::framework;
 std::map<int, TEfficiency*> effVsPt;
+std::map<int, TEfficiency*> effVsEta;
 
 struct alice3Efficiency {
   Configurable<std::vector<int>> pdgCodes{"pdgCodes", {211}, "List of PDG codes to consider for efficiency calculation"};
   OutputObj<THashList> outList{"output"};
   void init(o2::framework::InitContext&)
   {
     outList.setObject(new THashList);
+    auto createEff = [&](const char* baseName, const char* axisTitle, int pdg, int nBins, double min, double max) {
+      auto eff = new TEfficiency(Form("%s_pdg%d", baseName, pdg),
+                                 Form("Efficiency for PDG %d; %s; Efficiency", pdg, axisTitle),
+                                 nBins, min, max);
+      outList->Add(eff);
+      return eff;
+    };
     for (auto pdg : pdgCodes.value) {
-      effVsPt[pdg] = new TEfficiency(Form("efficiency_pdg%d", pdg),
-                                     Form("Efficiency for PDG %d; p_{T} (GeV/c); Efficiency", pdg),
-                                     100, 0, 10);
-      outList->Add(effVsPt[pdg]);
+      effVsPt[pdg] = createEff("efficiency", "p_{T} (GeV/c)", pdg, 100, 0, 10);
+      effVsEta[pdg] = createEff("efficiency_eta", "#eta", pdg, 100, -5, 5);
     }
   }
 
@@ -67,9 +73,9 @@ struct alice3Efficiency {
 
       std::vector<int64_t>& indices = pdgIndices[mcParticle.pdgCode()];
       // Fill efficiency histogram
-      auto& eff = effVsPt[mcParticle.pdgCode()];
       const bool found = std::find(indices.begin(), indices.end(), track.globalIndex()) != indices.end();
-      eff->Fill(track.pt(), found);
+      effVsPt[mcParticle.pdgCode()]->Fill(found, track.pt());
+      effVsEta[mcParticle.pdgCode()]->Fill(found, track.eta());
     }
   }
 };

ALICE3/Tasks/pidSeparationPower.cxx ALICE3/Tasks/alice3SeparationPower.cxxALICE3/Tasks/pidSeparationPower.cxx renamed to ALICE3/Tasks/alice3SeparationPower.cxx

@@ -9,7 +9,7 @@
 // granted to it by virtue of its status as an Intergovernmental Organization
 // or submit itself to any jurisdiction.
 ///
-/// \file pidSeparationPower.cxx
+/// \file alice3SeparationPower.cxx
 ///
 /// \brief This task produces the separation power of the ALICE3 detector
 ///
@@ -38,7 +38,7 @@ using namespace o2::framework;
 std::array<TProfile2D*, 5> separationInnerTOF;
 std::array<TProfile2D*, 5> separationOuterTOF;
 std::array<TProfile2D*, 5> separationRICH;
-struct pidSeparationPower {
+struct alice3SeparationPower {
 
   ConfigurableAxis etaAxis{"etaAxis", {100, -1.f, 1.f}, "Binning in eta"};
   HistogramRegistry histos{"Histos", {}, OutputObjHandlingPolicy::AnalysisObject};
@@ -103,4 +103,4 @@ struct pidSeparationPower {
   }
 };
 
-WorkflowSpec defineDataProcessing(ConfigContext const& cfgc) { return WorkflowSpec{adaptAnalysisTask<pidSeparationPower>(cfgc)}; }
+WorkflowSpec defineDataProcessing(ConfigContext const& cfgc) { return WorkflowSpec{adaptAnalysisTask<alice3SeparationPower>(cfgc)}; }