Added additional D0 topology cuts and variables

Author: axti98

PWGDQ/Core/CutsLibrary.cxx

@@ -3702,6 +3702,11 @@ AnalysisCompositeCut* o2::aod::dqcuts::GetCompositeCut(const char* cutName)
     return cut;
   }
 
+  if (nameStr == "pairD0Topology") {
+    cut->AddCut(GetAnalysisCut("pairD0Topology"));
+    return cut;
+  }
+
   if (nameStr == "pairD0HighPt1") {
     cut->AddCut(GetAnalysisCut("pairLxyzProjected3sigma"));
     cut->AddCut(GetAnalysisCut("pairPtLow5"));
@@ -3923,6 +3928,24 @@ AnalysisCompositeCut* o2::aod::dqcuts::GetCompositeCut(const char* cutName)
     return cut;
   }
 
+  if (nameStr == "alice3StandardTrackD0") {
+    cut->AddCut(GetAnalysisCut("alice3StandardKineD0"));
+    cut->AddCut(GetAnalysisCut("alice3TrackQuality"));
+    return cut;
+  }
+
+  if (nameStr == "alice3StandardTrackTOFAcceptanceD0") {
+    cut->AddCut(GetAnalysisCut("alice3KineTOFAcceptanceD0"));
+    cut->AddCut(GetAnalysisCut("alice3TrackQuality"));
+    return cut;
+  }
+
+  if (nameStr == "alice3StandardTrackRICHAcceptanceD0") {
+    cut->AddCut(GetAnalysisCut("alice3KineRICHAcceptanceD0"));
+    cut->AddCut(GetAnalysisCut("alice3TrackQuality"));
+    return cut;
+  }
+
   if (nameStr == "alice3TightDCATrack") {
     cut->AddCut(GetAnalysisCut("alice3StandardKine"));
     cut->AddCut(GetAnalysisCut("alice3TrackQualityTightDCA"));
@@ -4108,6 +4131,18 @@ AnalysisCompositeCut* o2::aod::dqcuts::GetCompositeCut(const char* cutName)
     return cut;
   }
 
+  if (nameStr == "alice3TOFPIDPi") {
+    cut->AddCut(GetAnalysisCut("alice3iTOFPIDPi"));
+    cut->AddCut(GetAnalysisCut("alice3oTOFPIDPi"));
+    return cut;
+  }
+
+  if (nameStr == "alice3TOFPIDKa") {
+    cut->AddCut(GetAnalysisCut("alice3iTOFPIDKa"));
+    cut->AddCut(GetAnalysisCut("alice3oTOFPIDKa"));
+    return cut;
+  }
+
   delete cut;
   LOGF(fatal, Form("Did not find cut %s. Returning nullptr", cutName));
   return nullptr;
@@ -7083,6 +7118,15 @@ AnalysisCut* o2::aod::dqcuts::GetAnalysisCut(const char* cutName)
     return cut;
   }
 
+  if (nameStr == "pairD0Topology") {
+    cut->AddCut(VarManager::kCosPointingAngle, 0.9, 1.0);
+    cut->AddCut(VarManager::kD0DcaDaughters, 0.0, 0.03);
+    cut->AddCut(VarManager::kD0ImpParProduct, -1000.0, 0.0);
+    cut->AddCut(VarManager::kDCAxy1, -0.1, 0.1);
+    cut->AddCut(VarManager::kDCAxy2, -0.1, 0.1);
+    return cut;
+  }
+
   if (nameStr == "pairJpsi") {
     cut->AddCut(VarManager::kMass, 2.8, 3.3);
     return cut;
@@ -7339,6 +7383,24 @@ AnalysisCut* o2::aod::dqcuts::GetAnalysisCut(const char* cutName)
     return cut;
   }
 
+  if (nameStr == "alice3StandardKineD0") {
+    cut->AddCut(VarManager::kPt, 0.7, 1000.0);
+    cut->AddCut(VarManager::kEta, -2.5, 2.5); // Total tracker acceptance in v3b geomety
+    return cut;
+  }
+
+  if (nameStr == "alice3KineTOFAcceptanceD0") {
+    cut->AddCut(VarManager::kPt, 0.7, 1000.0);
+    cut->AddCut(VarManager::kEta, -2.0, 2.0); // TOF acceptance in v3b geomety
+    return cut;
+  }
+
+  if (nameStr == "alice3KineRICHAcceptanceD0") {
+    cut->AddCut(VarManager::kPt, 0.7, 1000.0);
+    cut->AddCut(VarManager::kEta, -0.8, 0.8); // TOF acceptance in v3b geomety
+    return cut;
+  }
+
   if (nameStr == "alice3LambdaCKine") {
     cut->AddCut(VarManager::kPt, 0.2, 1000.0);
     cut->AddCut(VarManager::kEta, -2.0, 2.0);

PWGDQ/Core/VarManager.cxx

@@ -1347,6 +1347,10 @@ void VarManager::SetDefaultVarNames()
   fgVariableUnits[kCosPointingAngle] = "";
   fgVariableNames[kMCCosPointingAngle] = "MC cos(#theta_{pointing})";
   fgVariableUnits[kMCCosPointingAngle] = "";
+  fgVariableNames[kD0DcaDaughters] = "DCA of D0 daughters";
+  fgVariableUnits[kD0DcaDaughters] = "";
+  fgVariableNames[kD0ImpParProduct] = "D0 impact parameter product";
+  fgVariableUnits[kD0ImpParProduct] = "";
   fgVariableNames[kVertexingPz] = "Pz Pair";
   fgVariableUnits[kVertexingPz] = "GeV/c";
   fgVariableNames[kVertexingSV] = "Secondary Vertexing z";
@@ -2633,6 +2637,8 @@ void VarManager::SetDefaultVarNames()
   fgVarNamesMap["kSignQuadDCAsigXY"] = kSignQuadDCAsigXY;
   fgVarNamesMap["kCosPointingAngle"] = kCosPointingAngle;
   fgVarNamesMap["kMCCosPointingAngle"] = kMCCosPointingAngle;
+  fgVarNamesMap["kD0ImpParProduct"] = kD0ImpParProduct;
+  fgVarNamesMap["kD0DcaDaughters"] = kD0DcaDaughters;
   fgVarNamesMap["kImpParXYJpsi"] = kImpParXYJpsi;
   fgVarNamesMap["kImpParXYK"] = kImpParXYK;
   fgVarNamesMap["kDCATrackProd"] = kDCATrackProd;

PWGDQ/Core/VarManager.h

@@ -874,6 +874,8 @@ class VarManager : public TObject
     kQuadDCAsigXYZ,
     kSignQuadDCAsigXY,
     kCosPointingAngle,
+    kD0ImpParProduct,
+    kD0DcaDaughters,
     kImpParXYJpsi,
     kImpParXYK,
     kDCATrackProd,
@@ -7688,6 +7690,14 @@ void VarManager::FillPairAlice3(T1 const& t1, T2 const& t2, float* values)
   if (fgUsedVars[kPairPhiv]) {
     values[kPairPhiv] = calculatePhiV<pairType>(t1, t2);
   }
+
+  if constexpr (pairType == kDecayToKPi) {
+    if constexpr ((fillMap & TrackDCA) > 0 || (fillMap & ReducedTrackBarrel) > 0) {
+      values[kDCAxy1] = t1.dcaXY();
+      values[kDCAxy2] = t2.dcaXY();
+      values[kD0ImpParProduct] = t1.dcaXY() * t2.dcaXY();
+    }
+  }
 }
 
 template <typename M, typename T>
@@ -7792,7 +7802,17 @@ void VarManager::FillPairVertexingAlice3(C const& collision, T const& t1, T cons
         v1 = {trackParVar0.getPt(), trackParVar0.getEta(), trackParVar0.getPhi(), m1};
         v2 = {trackParVar1.getPt(), trackParVar1.getEta(), trackParVar1.getPhi(), m2};
         v12 = v1 + v2;
+
+        const auto pos1 = trackParVar0.getXYZGlo();
+        const auto pos2 = trackParVar1.getXYZGlo();
+
+        const double dx = pos1.X() - pos2.X();
+        const double dy = pos1.Y() - pos2.Y();
+        const double dz = pos1.Z() - pos2.Z();
+
+        values[kD0DcaDaughters] = std::sqrt(dx * dx + dy * dy + dz * dz);
       }
+
       double phi = std::atan2(secondaryVertex[1] - collision.posY(), secondaryVertex[0] - collision.posX());
       double theta = std::atan2(secondaryVertex[2] - collision.posZ(),
                                 std::sqrt((secondaryVertex[0] - collision.posX()) * (secondaryVertex[0] - collision.posX()) +