Add muon channel in the skim (#6342)

* Add Muon channel in the HF skim

* Clang Fix

* add space back

Co-authored-by: Gian Michele Innocenti <ginnocen@alicecerno2.cern.ch>

Author: ginnocen

Analysis/Core/include/AnalysisCore/HFConfigurables.h

@@ -38,6 +38,12 @@ class HFTrackIndexSkimsCreatorConfigs
   double mInvMassJpsiToEEMax = 4.1;
   double mCPAJpsiToEEMin = -2;
   double mImpParProductJpsiToEEMax = 1000.;
+  // JpsiToMuMu
+  double mPtJpsiToMuMuMin = 4.; //original value 0.
+  double mInvMassJpsiToMuMuMin = 2.5;
+  double mInvMassJpsiToMuMuMax = 4.1;
+  double mCPAJpsiToMuMuMin = -2;
+  double mImpParProductJpsiToMuMuMax = 1000.;
 
   // 3-prong cuts
 

Analysis/DataModel/include/AnalysisDataModel/HFSecondaryVertex.h

@@ -56,8 +56,9 @@ DECLARE_SOA_INDEX_COLUMN_FULL(Index3, index3, int, Tracks, "_3");          //!
 DECLARE_SOA_INDEX_COLUMN_FULL(IndexV0, indexV0, int, aod::V0Datas, "_V0"); //!
 DECLARE_SOA_COLUMN(HFflag, hfflag, uint8_t);                               //!
 
-DECLARE_SOA_COLUMN(D0ToKPiFlag, d0ToKPiFlag, uint8_t);   //!
-DECLARE_SOA_COLUMN(JpsiToEEFlag, jpsiToEEFlag, uint8_t); //!
+DECLARE_SOA_COLUMN(D0ToKPiFlag, d0ToKPiFlag, uint8_t);       //!
+DECLARE_SOA_COLUMN(JpsiToEEFlag, jpsiToEEFlag, uint8_t);     //!
+DECLARE_SOA_COLUMN(JpsiToMuMuFlag, jpsiToMuMuFlag, uint8_t); //!
 
 DECLARE_SOA_COLUMN(DPlusPiKPiFlag, dPlusPiKPiFlag, uint8_t); //!
 DECLARE_SOA_COLUMN(LcPKPiFlag, lcPKPiFlag, uint8_t);         //!
@@ -77,7 +78,8 @@ DECLARE_SOA_TABLE(HfTrackIndexCasc, "AOD", "HFTRACKIDXCASC", //!
 
 DECLARE_SOA_TABLE(HfCutStatusProng2, "AOD", "HFCUTSTATUSP2", //!
                   hf_track_index::D0ToKPiFlag,
-                  hf_track_index::JpsiToEEFlag);
+                  hf_track_index::JpsiToEEFlag,
+                  hf_track_index::JpsiToMuMuFlag);
 
 DECLARE_SOA_TABLE(HfTrackIndexProng3, "AOD", "HFTRACKIDXP3", //!
                   hf_track_index::Index0Id,
@@ -267,7 +269,7 @@ auto CosThetaStarD0bar(const T& candidate)
   return candidate.cosThetaStar(array{RecoDecay::getMassPDG(kKPlus), RecoDecay::getMassPDG(kPiPlus)}, RecoDecay::getMassPDG(pdg::Code::kD0), 0);
 }
 
-// J/ψ → e+ e−
+// J/ψ
 
 template <typename T>
 auto CtJpsi(const T& candidate)
@@ -287,11 +289,21 @@ auto EJpsi(const T& candidate)
   return candidate.e(RecoDecay::getMassPDG(pdg::Code::kJpsi));
 }
 
+// J/ψ → e+ e−
+
 template <typename T>
 auto InvMassJpsiToEE(const T& candidate)
 {
   return candidate.m(array{RecoDecay::getMassPDG(kElectron), RecoDecay::getMassPDG(kElectron)});
 }
+
+// J/ψ → µ+ µ−
+
+template <typename T>
+auto InvMassJpsiToMuMu(const T& candidate)
+{
+  return candidate.m(array{RecoDecay::getMassPDG(kMuonPlus), RecoDecay::getMassPDG(kMuonMinus)});
+}
 } // namespace hf_cand_prong2
 
 // general columns

Analysis/Tasks/PWGHF/HFTrackIndexSkimsCreator.cxx

@@ -297,6 +297,7 @@ struct HFTrackIndexSkimsCreator {
      {"hNCand2ProngVsNTracks", "2-prong candidates preselected;# of selected tracks;# of candidates;entries", {HistType::kTH2F, {{2500, 0., 25000.}, {2000, 0., 200000.}}}},
      {"hmassD0ToPiK", "D^{0} candidates;inv. mass (#pi K) (GeV/#it{c}^{2});entries", {HistType::kTH1F, {{500, 0., 5.}}}},
      {"hmassJpsiToEE", "J/#psi candidates;inv. mass (e^{#plus} e^{#minus}) (GeV/#it{c}^{2});entries", {HistType::kTH1F, {{500, 0., 5.}}}},
+     {"hmassJpsiToMuMu", "J/#psi candidates;inv. mass (#mu^{#plus} #mu^{#minus}) (GeV/#it{c}^{2});entries", {HistType::kTH1F, {{500, 0., 5.}}}},
      // 3-prong histograms
      {"hvtx3_x", "3-prong candidates;#it{x}_{sec. vtx.} (cm);entries", {HistType::kTH1F, {{1000, -2., 2.}}}},
      {"hvtx3_y", "3-prong candidates;#it{y}_{sec. vtx.} (cm);entries", {HistType::kTH1F, {{1000, -2., 2.}}}},
@@ -319,6 +320,7 @@ struct HFTrackIndexSkimsCreator {
   double massK = RecoDecay::getMassPDG(kKPlus);
   double massProton = RecoDecay::getMassPDG(kProton);
   double massElectron = RecoDecay::getMassPDG(kElectron);
+  double massMuon = RecoDecay::getMassPDG(kMuonMinus);
 
   // int nColls{0}; //can be added to run over limited collisions per file - for tesing purposes
 
@@ -376,6 +378,12 @@ struct HFTrackIndexSkimsCreator {
     cut2ProngCPACandMin[hf_cand_prong2::DecayType::JpsiToEE] = configs->mCPAJpsiToEEMin;
     cut2ProngImpParProductCandMax[hf_cand_prong2::DecayType::JpsiToEE] = configs->mImpParProductJpsiToEEMax;
 
+    cut2ProngPtCandMin[hf_cand_prong2::DecayType::JpsiToMuMu] = configs->mPtJpsiToMuMuMin;
+    cut2ProngInvMassCandMin[hf_cand_prong2::DecayType::JpsiToMuMu] = configs->mInvMassJpsiToMuMuMin;
+    cut2ProngInvMassCandMax[hf_cand_prong2::DecayType::JpsiToMuMu] = configs->mInvMassJpsiToMuMuMax;
+    cut2ProngCPACandMin[hf_cand_prong2::DecayType::JpsiToMuMu] = configs->mCPAJpsiToMuMuMin;
+    cut2ProngImpParProductCandMax[hf_cand_prong2::DecayType::JpsiToMuMu] = configs->mImpParProductJpsiToMuMuMax;
+
     const int nCuts3Prong = 4; // how many different selections are made on 3-prongs
     double cut3ProngPtCandMin[n3ProngDecays];
     double cut3ProngInvMassCandMin[n3ProngDecays];
@@ -415,10 +423,12 @@ struct HFTrackIndexSkimsCreator {
     array<array<double, 2>, n2ProngDecays> arr2Mass1;
     arr2Mass1[hf_cand_prong2::DecayType::D0ToPiK] = array{massPi, massK};
     arr2Mass1[hf_cand_prong2::DecayType::JpsiToEE] = array{massElectron, massElectron};
+    arr2Mass1[hf_cand_prong2::DecayType::JpsiToMuMu] = array{massMuon, massMuon};
 
     array<array<double, 2>, n2ProngDecays> arr2Mass2;
     arr2Mass2[hf_cand_prong2::DecayType::D0ToPiK] = array{massK, massPi};
     arr2Mass2[hf_cand_prong2::DecayType::JpsiToEE] = array{massElectron, massElectron};
+    arr2Mass2[hf_cand_prong2::DecayType::JpsiToMuMu] = array{massMuon, massMuon};
 
     array<array<double, 3>, n3ProngDecays> arr3Mass1;
     arr3Mass1[hf_cand_prong3::DecayType::DPlusToPiKPi] = array{massPi, massK, massPi};
@@ -591,7 +601,7 @@ struct HFTrackIndexSkimsCreator {
                     }
                   }
                 }
-                rowProng2CutStatus(Prong2CutStatus[0], Prong2CutStatus[1]); //FIXME when we can do this by looping over n2ProngDecays
+                rowProng2CutStatus(Prong2CutStatus[0], Prong2CutStatus[1], Prong2CutStatus[2]); //FIXME when we can do this by looping over n2ProngDecays
               }
 
               // fill histograms
@@ -611,6 +621,9 @@ struct HFTrackIndexSkimsCreator {
                       if (n2 == hf_cand_prong2::DecayType::JpsiToEE) {
                         registry.get<TH1>(HIST("hmassJpsiToEE"))->Fill(mass2ProngHypo1[n2]);
                       }
+                      if (n2 == hf_cand_prong2::DecayType::JpsiToMuMu) {
+                        registry.get<TH1>(HIST("hmassJpsiToMuMu"))->Fill(mass2ProngHypo1[n2]);
+                      }
                     }
                     if ((cut2ProngInvMassCandMin[n2] < 0. && cut2ProngInvMassCandMax[n2] <= 0.) || (mass2ProngHypo2[n2] >= cut2ProngInvMassCandMin[n2] && mass2ProngHypo2[n2] < cut2ProngInvMassCandMax[n2])) {
                       mass2ProngHypo2[n2] = RecoDecay::M(arrMom, arr2Mass2[n2]);