Running with different scenario at the same time

Author: rbailhac

ALICE3/Tasks/alice3-dilepton.cxx

@@ -16,6 +16,7 @@
 
 #include "ALICE3/DataModel/OTFRICH.h"
 #include "ALICE3/DataModel/OTFTOF.h"
+#include "ALICE3/DataModel/OTFCollision.h"
 #include "ALICE3/DataModel/tracksAlice3.h"
 #include "Common/DataModel/TrackSelectionTables.h"
 
@@ -25,9 +26,11 @@
 #include <Framework/AnalysisTask.h>
 #include <Framework/HistogramRegistry.h>
 #include <Framework/runDataProcessing.h>
+#include <Framework/O2DatabasePDGPlugin.h>
 
 #include <Math/Vector4D.h>
 
+
 #include <vector>
 
 using namespace o2;
@@ -54,6 +57,8 @@ struct Alice3Dilepton {
   SliceCache cache_mc;
   SliceCache cache_rec;
 
+  Service<o2::framework::O2DatabasePDG> inspdg;
+
   Configurable<int> pdg{"pdg", 11, "pdg code for analysis. dielectron:11, dimuon:13"};
   Configurable<bool> requireHFEid{"requireHFEid", true, "Require HFE identification for both leptons"};
   Configurable<float> ptMin{"pt-min", 0.f, "Lower limit in pT"};
@@ -68,6 +73,7 @@ struct Alice3Dilepton {
   Configurable<float> nSigmaPionCutInnerTOF{"nSigmaPionCutInnerTOF", 3., "Pion exclusion in inner TOF"};
   Configurable<float> nSigmaElectronRich{"nSigmaElectronRich", 3., "Electron inclusion RICH"};
   Configurable<float> nSigmaPionRich{"nSigmaPionRich", 3., "Pion exclusion RICH"};
+  Configurable<int> otfConfig{"otfConfig", 0, "OTF configuration flag"};
 
   HistogramRegistry registry{"Histos", {}, OutputObjHandlingPolicy::AnalysisObject};
 
@@ -100,6 +106,7 @@ struct Alice3Dilepton {
       registry.add("Generated/Particle/prodVy", "Particle Prod. Vertex Y", kTH1F, {axisPrody});
       registry.add("Generated/Particle/prodVz", "Particle Prod. Vertex Z", kTH1F, {axisProdz});
       registry.add("Generated/Particle/ParticlesPerEvent", "Particles per event", kTH1F, {{100, 0, 100}});
+      registry.add("Generated/Particle/ParticlesFit", "Charged Particles in Fit acceptance per event", kTH1F, {{15000,0,15000}});
 
       registry.add("Generated/Pair/ULS/Tried", "Pair tries", kTH1F, {{10, -0.5, 9.5}});
       registry.add("Generated/Pair/ULS/Mass", "Pair Mass", kTH1F, {axisM});
@@ -601,7 +608,19 @@ struct Alice3Dilepton {
 
       auto mcParticles_per_coll = mcParticles.sliceBy(perMCCollision, mccollision.globalIndex());
       int nParticlesInEvent = 0;
+      int nParticlesFIT=0;
       for (const auto& mcParticle : mcParticles_per_coll) {
+        if (mcParticle.isPhysicalPrimary()) {
+          if ((2.2<mcParticle.eta()&&mcParticle.eta()< 5.0) || (-3.4<mcParticle.eta()&&mcParticle.eta()<-2.3)) {
+            auto pdgParticle = inspdg->GetParticle(mcParticle.pdgCode());
+            if (pdgParticle) {
+              float charge = pdgParticle->Charge() / 3.f; // Charge in units of |e|
+              if (std::abs(charge) >= 1.) {
+                nParticlesFIT++;
+              }
+            }
+          }
+        }
         if (std::abs(mcParticle.pdgCode()) != pdg) {
           continue;
         }
@@ -624,6 +643,7 @@ struct Alice3Dilepton {
 
       } // end of mc particle loop
       registry.fill(HIST("Generated/Particle/ParticlesPerEvent"), nParticlesInEvent);
+      registry.fill(HIST("Generated/Particle/ParticlesFIT"), nParticlesFIT);
 
       auto neg_mcParticles_coll = neg_mcParticles->sliceByCached(o2::aod::mcparticle::mcCollisionId, mccollision.globalIndex(), cache_mc);
       auto pos_mcParticles_coll = pos_mcParticles->sliceByCached(o2::aod::mcparticle::mcCollisionId, mccollision.globalIndex(), cache_mc);
@@ -636,18 +656,22 @@ struct Alice3Dilepton {
   } // end of processGen
 
   using MyTracksMC = soa::Join<aod::Tracks, aod::TracksCov, aod::TracksDCA, aod::McTrackLabels, aod::UpgradeTofs, aod::UpgradeTofMCs, aod::UpgradeRichs, aod::TracksAlice3>;
+  using Alice3Collision = soa::Join<aod::Collisions, aod::OTFLUTConfigId>;
+
   // Filter trackFilter = etaMin < o2::aod::track::eta &&
   //                      o2::aod::track::eta < etaMax &&
   //                      ptMin < o2::aod::track::pt &&
   //                      o2::aod::track::pt < ptMax &&
   //                      o2::aod::track_alice3::isReconstructed == selectReconstructed;
   Filter trackFilter = o2::aod::track_alice3::isReconstructed == selectReconstructed;
   using MyFilteredTracksMC = soa::Filtered<MyTracksMC>;
+  Filter configFilter = (aod::upgrade_collision::lutConfigId == otfConfig);
+  using MyFilteredAlice3Collision = soa::Filtered<Alice3Collision>;
   Preslice<MyFilteredTracksMC> perCollision = aod::track::collisionId;
   Partition<MyFilteredTracksMC> posTracks = o2::aod::track::signed1Pt > 0.f;
   Partition<MyFilteredTracksMC> negTracks = o2::aod::track::signed1Pt < 0.f;
 
-  void processRec(const o2::aod::Collisions& collisions,
+  void processRec(MyFilteredAlice3Collision const& collisions,
                   MyFilteredTracksMC const& tracks,
                   const o2::aod::McCollisions&,
                   const aod::McParticles& mcParticles)