Possibility to reconstruct D*s and not D0s (Xiaoming)

[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliAnalysisTaskSEVertexingHF.cxx

diff --git a/PWG3/vertexingHF/AliAnalysisTaskSEVertexingHF.cxx b/PWG3/vertexingHF/AliAnalysisTaskSEVertexingHF.cxx
index 79c38ecf9aa21cc87f703442099f99297f29220a..c4aae0739ae0b58f0cef43d39482fd9fead600d2 100644 (file)
--- a/PWG3/vertexingHF/AliAnalysisTaskSEVertexingHF.cxx
+++ b/PWG3/vertexingHF/AliAnalysisTaskSEVertexingHF.cxx
@@ -43,7 +43,8 @@ fVerticesHFTClArr(0),
 fD0toKpiTClArr(0),
 fJPSItoEleTClArr(0),
 fCharm3ProngTClArr(0),
-fCharm4ProngTClArr(0)
+fCharm4ProngTClArr(0),
+fDstarTClArr(0)
 {
   // Default constructor
 }
@@ -56,7 +57,8 @@ fVerticesHFTClArr(0),
 fD0toKpiTClArr(0),
 fJPSItoEleTClArr(0),
 fCharm3ProngTClArr(0),
-fCharm4ProngTClArr(0)
+fCharm4ProngTClArr(0),
+fDstarTClArr(0)
 {
   // Default constructor
 }
@@ -99,7 +101,7 @@ void AliAnalysisTaskSEVertexingHF::UserCreateOutputObjects()
   fVerticesHFTClArr->SetName("VerticesHF");
   AddAODBranch("TClonesArray", &fVerticesHFTClArr);
 
-  if(fVHF->GetD0toKpi()) {
+  if(fVHF->GetD0toKpi() || fVHF->GetDstar()) {
     fD0toKpiTClArr = new TClonesArray("AliAODRecoDecayHF2Prong", 0);
     fD0toKpiTClArr->SetName("D0toKpi");
     AddAODBranch("TClonesArray", &fD0toKpiTClArr);
@@ -123,6 +125,12 @@ void AliAnalysisTaskSEVertexingHF::UserCreateOutputObjects()
     AddAODBranch("TClonesArray", &fCharm4ProngTClArr);
   }
 
+  if(fVHF->GetDstar()) {
+    fDstarTClArr = new TClonesArray("AliAODRecoCascadeHF", 0);
+    fDstarTClArr->SetName("Dstar");
+    AddAODBranch("TClonesArray", &fDstarTClArr);
+  }
+
   return;
 }
 
@@ -140,7 +148,8 @@ void AliAnalysisTaskSEVertexingHF::UserExec(Option_t */*option*/)
                       fD0toKpiTClArr,
                       fJPSItoEleTClArr,
                       fCharm3ProngTClArr,
-                      fCharm4ProngTClArr);
+                      fCharm4ProngTClArr,
+                      fDstarTClArr);
   
   return;
 }