Refactor the AnalysisDileptonTrackTrack and add process function for K+K-pi+pi-

Author: Ionut Cristian Arsene

PWGDQ/Core/VarManager.h

@@ -158,14 +158,17 @@ class VarManager : public TObject
     // 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,
+    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.
@@ -3840,6 +3843,16 @@ void VarManager::FillPair(T1 const& t1, T2 const& t2, float* values)
     m1 = o2::constants::physics::MassPionCharged;
     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;
@@ -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);