Addition of the dimuon analysis task (#4342)

Author: vfeuilla

Analysis/Core/include/Analysis/VarManager.h

@@ -27,6 +27,7 @@
 
 // TODO: create an array holding these constants for all needed particles or check for a place where these are already defined
 static const float fgkElectronMass = 0.000511; // GeV
+static const float fgkMuonMass = 0.105;        // GeV
 
 //_________________________________________________________________________
 class VarManager : public TObject
@@ -126,6 +127,7 @@ class VarManager : public TObject
     kMuonZMu,
     kMuonBendingCoor,
     kMuonNonBendingCoor,
+    kMuonRAtAbsorberEnd,
     kMuonChi2,
     kMuonChi2MatchTrigger,
     kNMuonTrackVariables,
@@ -359,6 +361,7 @@ void VarManager::FillTrack(T const& track, float* values)
     values[kMuonZMu] = track.zMu();
     values[kMuonBendingCoor] = track.bendingCoor();
     values[kMuonNonBendingCoor] = track.nonBendingCoor();
+    values[kMuonRAtAbsorberEnd] = track.rAtAbsorberEnd();
     values[kMuonChi2] = track.chi2();
     values[kMuonChi2MatchTrigger] = track.chi2MatchTrigger();
   }
@@ -376,13 +379,27 @@ void VarManager::FillPair(T const& t1, T const& t2, float* values)
   if (!values)
     values = fgValues;
 
-  ROOT::Math::PtEtaPhiMVector v1(t1.pt(), t1.eta(), t1.phi(), fgkElectronMass);
-  ROOT::Math::PtEtaPhiMVector v2(t2.pt(), t2.eta(), t2.phi(), fgkElectronMass);
+  float mass1;
+  float mass2;
+
+  if (t1.filteringFlags() > 0)
+    mass1 = fgkElectronMass;
+  else
+    mass1 = fgkMuonMass;
+
+  if (t1.filteringFlags() > 0)
+    mass2 = fgkElectronMass;
+  else
+    mass2 = fgkMuonMass;
+
+  ROOT::Math::PtEtaPhiMVector v1(t1.pt(), t1.eta(), t1.phi(), mass1);
+  ROOT::Math::PtEtaPhiMVector v2(t2.pt(), t2.eta(), t2.phi(), mass2);
   ROOT::Math::PtEtaPhiMVector v12 = v1 + v2;
   values[kMass] = v12.M();
   values[kPt] = v12.Pt();
   values[kEta] = v12.Eta();
   values[kPhi] = v12.Phi();
+  values[kRap] = -v12.Rapidity();
 }
 
 template <typename T1, typename T2>

Analysis/Core/src/VarManager.cxx

@@ -209,6 +209,8 @@ void VarManager::SetDefaultVarNames()
   fgVariableUnits[kMuonBendingCoor] = "";
   fgVariableNames[kMuonNonBendingCoor] = "non-bending coor.";
   fgVariableUnits[kMuonNonBendingCoor] = "";
+  fgVariableNames[kMuonRAtAbsorberEnd] = "R at the end of the absorber";
+  fgVariableUnits[kMuonRAtAbsorberEnd] = "";
   fgVariableNames[kMuonChi2] = "#chi 2";
   fgVariableUnits[kMuonChi2] = "";
   fgVariableNames[kMuonChi2MatchTrigger] = "#chi 2 trigger match";

Analysis/DataModel/include/Analysis/ReducedInfoTables.h

@@ -133,7 +133,8 @@ DECLARE_SOA_TABLE(ReducedMuonsExtended, "AOD", "RTMUONEXTENDED",
                   muon::InverseBendingMomentum,
                   muon::ThetaX, muon::ThetaY, muon::ZMu,
                   muon::BendingCoor, muon::NonBendingCoor,
-                  muon::Chi2, muon::Chi2MatchTrigger);
+                  muon::Chi2, muon::Chi2MatchTrigger,
+                  muon::RAtAbsorberEnd<muon::BendingCoor, muon::NonBendingCoor, muon::ThetaX, muon::ThetaY, muon::ZMu>);
 
 // iterators
 using ReducedTrack = ReducedTracks::iterator;

Analysis/Tasks/PWGDQ/CMakeLists.txt

@@ -13,6 +13,11 @@ o2_add_dpl_workflow(dilepton-ee
                     PUBLIC_LINK_LIBRARIES O2::Framework O2::AnalysisDataModel O2::DetectorsBase O2::AnalysisCore
                     COMPONENT_NAME Analysis)
 
+o2_add_dpl_workflow(dilepton-mumu
+                    SOURCES dileptonMuMu.cxx
+                    PUBLIC_LINK_LIBRARIES O2::Framework O2::AnalysisDataModel O2::DetectorsBase O2::AnalysisCore
+                    COMPONENT_NAME Analysis)
+
 o2_add_dpl_workflow(table-maker-pp
                     SOURCES tableMaker_pp.cxx
                     PUBLIC_LINK_LIBRARIES O2::Framework O2::AnalysisDataModel O2::DetectorsBase O2::AnalysisCore