Included also 3 and 4 prong decays; added variables to ntuple

[u/mrichter/AliRoot.git] / TRD / AliTRDReconstructor.cxx

diff --git a/TRD/AliTRDReconstructor.cxx b/TRD/AliTRDReconstructor.cxx
index b45dcc0a85da085b3b412286ab2f46ef3c20591d..d54a2fa01adfbaa9fec95bb0a2bac479762a95c3 100644 (file)
--- a/TRD/AliTRDReconstructor.cxx
+++ b/TRD/AliTRDReconstructor.cxx
@@ -49,6 +49,7 @@
 ClassImp(AliTRDReconstructor)
 
 TClonesArray *AliTRDReconstructor::fgClusters = 0x0;
+TClonesArray *AliTRDReconstructor::fgTracklets = 0x0;
 Char_t* AliTRDReconstructor::fgSteerNames[kNsteer] = {
   "DigitsConversion       "
  ,"Tail Cancellation      "
@@ -65,6 +66,7 @@ Char_t* AliTRDReconstructor::fgSteerNames[kNsteer] = {
  ,"Tracklet Improve       "
  ,"HLT Mode              "
  ,"Cosmic Reconstruction "
+ ,"Process Online Tracklets"
 };
 Char_t* AliTRDReconstructor::fgSteerFlags[kNsteer] = {
   "dc"// digits conversion [false]
@@ -82,6 +84,7 @@ Char_t* AliTRDReconstructor::fgSteerFlags[kNsteer] = {
  ,"ti"// improve tracklets in stand alone track finder [true]
  ,"hlt"// HLT reconstruction [false]
  ,"cos"// Cosmic Reconstruction [false]
+ ,"tp"// also use online tracklets for reconstruction [false]
 };
 Char_t* AliTRDReconstructor::fgTaskNames[kNtasks] = {
   "RawReader"
@@ -153,6 +156,9 @@ AliTRDReconstructor::~AliTRDReconstructor()
   if(fgClusters) {
     fgClusters->Delete(); delete fgClusters;
   }
+  if(fgTracklets) {
+    fgTracklets->Delete(); delete fgTracklets;
+  }
   if(fSteerParam&kOwner){
     for(Int_t itask = 0; itask < kNtasks; itask++)
       if(fDebugStream[itask]) delete fDebugStream[itask];
@@ -176,9 +182,8 @@ void AliTRDReconstructor::ConvertDigits(AliRawReader *rawReader
   //
   // Convert raw data digits into digit objects in a root tree
   //
-  AliInfo("Feature not available for the moment."); return;
 
-  AliInfo("Convert raw data digits into digit objects [RawReader -> Digit TTree]");
+  //AliInfo("Convert raw data digits into digit objects [RawReader -> Digit TTree]");
 
   AliTRDrawData rawData;
   rawReader->Reset();
@@ -216,6 +221,10 @@ void AliTRDReconstructor::Reconstruct(AliRawReader *rawReader
   // take over ownership of clusters
   fgClusters = clusterer.RecPoints();
   clusterer.SetClustersOwner(kFALSE);
+
+  // take over ownership of online tracklets
+  fgTracklets = clusterer.TrackletsArray();
+  clusterer.SetTrackletsOwner(kFALSE);
 }
 
 //_____________________________________________________________________________
@@ -239,6 +248,10 @@ void AliTRDReconstructor::Reconstruct(TTree *digitsTree
   // take over ownership of clusters
   fgClusters = clusterer.RecPoints();
   clusterer.SetClustersOwner(kFALSE);
+
+  // take over ownership of online tracklets
+  fgTracklets = clusterer.TrackletsArray();
+  clusterer.SetTrackletsOwner(kFALSE);
 }
 
 //_____________________________________________________________________________