[PWGDQ] Refactor the AnalysisDileptonTrackTrack and add process function for … (#18673)

Co-authored-by: Ionut Cristian Arsene <iarsene@cern.ch>
Co-authored-by: ALICE Action Bot <alibuild@cern.ch>

Author: 3 people

PWGDQ/Core/VarManager.h

@@ -156,16 +156,19 @@ class VarManager : public TObject
 
   enum PairCandidateType {
     // TODO: need to agree on a scheme to incorporate all various hypotheses (e.g. e - mu, jpsi - K+, Jpsi - pipi,...)
-    kDecayToEE = 0, // e.g. J/psi        -> e+ e-
-    kDecayToMuMu,   // e.g. J/psi        -> mu+ mu-
-    kDecayToPiPi,
+    kDecayToEE = 0,             // e.g. J/psi        -> e+ e-
+    kDecayToMuMu,               // e.g. J/psi        -> mu+ mu-
+    kDecayToPiPi,               // decay channel to pi+ pi-
+    kDecayToKchKch,             // decay channel to K+ K-
+    kDecayToPrAPr,              // decay channel to p+ p-
     kElectronMuon,              // e.g. Electron - muon correlations
     kBcToThreeMuons,            // e.g. Bc           -> mu+ mu- mu+
     kBtoJpsiEEK,                // e.g. B+           -> e+ e- K+
     kJpsiEEProton,              // e.g. Jpsi-proton correlation, Jpsi to e+e-
     kXtoJpsiPiPi,               // e.g. X(3872)      -> J/psi pi+ pi-
     kPsi2StoJpsiPiPi,           // e.g. Psi(2S)      -> J/psi pi+ pi-
     kChictoJpsiEE,              // e.g. Chi_c1      -> J/psi e+ e-
+    kNonResonantToKpKmPiPi,     // e.g. J/psi -> K+ K- pi+ pi+
     kDstarToD0KPiPi,            // e.g. D*+ -> D0 pi+ -> K- pi+ pi+
     kTripleCandidateToEEPhoton, // e.g. chi_c   -> e+ e- photon or pi0 -> e+ e- photon
     kDecayToKPi,                // e.g. D0           -> K+ pi- or cc.
@@ -3841,6 +3844,16 @@ void VarManager::FillPair(T1 const& t1, T2 const& t2, float* values)
     m2 = o2::constants::physics::MassPionCharged;
   }
 
+  if constexpr (pairType == kDecayToKchKch) {
+    m1 = o2::constants::physics::MassKaonCharged;
+    m2 = o2::constants::physics::MassKaonCharged;
+  }
+
+  if constexpr (pairType == kDecayToPrAPr) {
+    m1 = o2::constants::physics::MassProton;
+    m2 = o2::constants::physics::MassProton;
+  }
+
   if constexpr (pairType == kDecayToKPi) {
     m1 = o2::constants::physics::MassKaonCharged;
     m2 = o2::constants::physics::MassPionCharged;
@@ -6910,6 +6923,11 @@ void VarManager::FillDileptonTrackTrack(T1 const& dilepton, T2 const& hadron1, T
     hadronMass1 = o2::constants::physics::MassElectron;
     hadronMass2 = o2::constants::physics::MassElectron;
   }
+  if (candidateType == kNonResonantToKpKmPiPi) {
+    defaultDileptonMass = dilepton.mass();
+    hadronMass1 = o2::constants::physics::MassPionCharged;
+    hadronMass2 = o2::constants::physics::MassPionCharged;
+  }
 
   ROOT::Math::PtEtaPhiMVector v1(dilepton.pt(), dilepton.eta(), dilepton.phi(), dilepton.mass());
   ROOT::Math::PtEtaPhiMVector v2(hadron1.pt(), hadron1.eta(), hadron1.phi(), hadronMass1);