X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=TRD%2FAliTRDReconstructor.cxx;h=8638476688165261dbf228a7d0808e7f00daa6ac;hb=7fdb2519b5c46634a48500f3702063714f9277d9;hp=4caa6d869fc513589026f4e9fc3540b4b2b8c3a9;hpb=91834b8d2c5ca0e5bd173c0bf76176cfc006ecce;p=u%2Fmrichter%2FAliRoot.git

diff --git a/TRD/AliTRDReconstructor.cxx b/TRD/AliTRDReconstructor.cxx
index 4caa6d869fc..86384766881 100644
--- a/TRD/AliTRDReconstructor.cxx
+++ b/TRD/AliTRDReconstructor.cxx
@@ -21,62 +21,98 @@
 //                                                                           //
 ///////////////////////////////////////////////////////////////////////////////
 
-#include <TFile.h>
 #include <TObjString.h>
 #include <TObjArray.h>
-#include <TClonesArray.h>
+#include <TTreeStream.h>
+#include <TDirectory.h>
 
-#include "AliRunLoader.h"
 #include "AliRawReader.h"
-#include "AliLog.h"
-#include "AliESDTrdTrack.h"
-#include "AliESDEvent.h"
 
 #include "AliTRDReconstructor.h"
 #include "AliTRDclusterizer.h"
-#include "AliTRDtracker.h"
-#include "AliTRDpidESD.h"
-#include "AliTRDgtuTrack.h"
 #include "AliTRDrawData.h"
+#include "AliTRDrawStreamBase.h"
 #include "AliTRDdigitsManager.h"
 #include "AliTRDtrackerV1.h"
-#include "AliTRDrecoParam.h"
+
+#define SETFLG(n,f) ((n) |= f)
+#define CLRFLG(n,f) ((n) &= ~f)
 
 ClassImp(AliTRDReconstructor)
 
-TClonesArray *AliTRDReconstructor::fgClusters = 0x0;
+TClonesArray *AliTRDReconstructor::fgClusters = NULL;
+TClonesArray *AliTRDReconstructor::fgTracklets = NULL;
+Char_t const * AliTRDReconstructor::fgSteerNames[kNsteer] = {
+  "DigitsConversion       "
+ ,"Write Clusters         "
+ ,"Write Online Tracklets "
+ ,"Stand Alone Tracking   "
+ ,"HLT Mode               "
+ ,"Process Online Tracklets"
+ ,"Debug Streaming        "
+};
+Char_t const * AliTRDReconstructor::fgSteerFlags[kNsteer] = {
+  "dc"// digits conversion [false]
+ ,"cw"// write clusters [true]
+ ,"tw"// write online tracklets [false]
+ ,"sa"// track seeding (stand alone tracking) [true]
+ ,"hlt"// HLT reconstruction [false]
+ ,"tp"// also use online tracklets for reconstruction [false]
+ ,"deb"// Write debug stream [false]
+};
+Char_t const * AliTRDReconstructor::fgTaskNames[AliTRDrecoParam::kTRDreconstructionTasks] = {
+  "Clusterizer"
+ ,"Tracker"
+ ,"PID"
+};
+Char_t const * AliTRDReconstructor::fgTaskFlags[AliTRDrecoParam::kTRDreconstructionTasks] = {
+  "cl"
+ ,"tr"
+ ,"pd"
+};
+Int_t AliTRDReconstructor::fgNTimeBins = -1;
+
 //_____________________________________________________________________________
 AliTRDReconstructor::AliTRDReconstructor()
   :AliReconstructor()
-  ,fSteerParam(0x00000007)
+  ,fSteerParam(0)
+  ,fClusterizer(NULL)
 {
-  memset(fStreamLevel, 0, 5*sizeof(UChar_t));
-  // Xe tail cancellation parameters
-  fTCParams[0] = 1.156; // r1
-  fTCParams[1] = 0.130; // r2
-  fTCParams[2] = 0.114; // c1
-  fTCParams[3] = 0.624; // c2
-  // Ar tail cancellation parameters
-  fTCParams[4] = 6.;    // r1
-  fTCParams[5] = 0.62;  // r2
-  fTCParams[6] = 0.0087;// c1
-  fTCParams[7] = 0.07;  // c2
-}
-
-//_____________________________________________________________________________
-AliTRDReconstructor::AliTRDReconstructor(const AliTRDReconstructor &r)
-  :AliReconstructor(r)
-  ,fSteerParam(r.fSteerParam)
-{
-  memcpy(fStreamLevel, r.fStreamLevel, 5*sizeof(UChar_t));
-  memcpy(fTCParams, r.fTCParams, 8*sizeof(Double_t));
+  // setting default "ON" steering parameters
+  // owner of debug streamers 
+  SETFLG(fSteerParam, kOwner);
+  // write clusters [cw]
+  SETFLG(fSteerParam, kWriteClusters);
+  // track seeding (stand alone tracking) [sa]
+  SETFLG(fSteerParam, kSeeding);
+
+  memset(fDebugStream, 0, sizeof(TTreeSRedirector *) * AliTRDrecoParam::kTRDreconstructionTasks);
 }
 
 //_____________________________________________________________________________
 AliTRDReconstructor::~AliTRDReconstructor()
 {
+  //
+  // Destructor
+  //
+
   if(fgClusters) {
-    fgClusters->Delete(); delete fgClusters;
+    fgClusters->Delete();
+    delete fgClusters;
+    fgClusters = NULL;
+  }
+  if(fgTracklets) {
+    fgTracklets->Delete();
+    delete fgTracklets;
+    fgTracklets = NULL;
+  }
+  if(fSteerParam&kOwner){
+    for(Int_t itask = 0; itask < AliTRDrecoParam::kTRDreconstructionTasks; itask++)
+      if(fDebugStream[itask]) delete fDebugStream[itask];
+  }
+  if(fClusterizer){
+    delete fClusterizer;
+    fClusterizer = NULL;
   }
 }
 
@@ -87,16 +123,22 @@ void AliTRDReconstructor::Init(){
   // Init Options
   //
   SetOption(GetOption());
+  Options(fSteerParam);
 
-  AliInfo("TRD reconstruction will use the following settings:");
-  printf("\tWrite Clusters         [cw] : %s\n", fSteerParam&kWriteClusters?"yes":"no");
-  printf("\tWrite Online Tracklets [tw] : %s\n", fSteerParam&kWriteTracklets?"yes":"no");
-  printf("\tDrift Gas Argon        [ar] : %s\n", fSteerParam&kDriftGas?"yes":"no");
-  printf("\tStand Alone Tracking   [sa] : %s\n", fSteerParam&kSeeding?"yes":"no");
-  printf("\tHLT         Tracking  [hlt] : %s\n", fSteerParam&kHLT?"yes":"no");
-  printf("\tCosmic Reconstruction [cos] : %s\n", fSteerParam&kCosmic?"yes":"no");
-  printf("\tNN PID                 [nn] : %s\n", fSteerParam&kSteerPID?"yes":"no");
-  printf("\tStreaming Levels            : Clusterizer[%d] Tracker[%d] PID[%d]\n", fStreamLevel[kClusterizer], fStreamLevel[kTracker], fStreamLevel[kPID]);
+  if(!fClusterizer){
+    fClusterizer = new AliTRDclusterizer(fgTaskNames[AliTRDrecoParam::kClusterizer], fgTaskNames[AliTRDrecoParam::kClusterizer]);
+    fClusterizer->SetReconstructor(this);
+  }
+  
+  // Make Debug Streams when Debug Streaming
+  if(IsDebugStreaming()){
+    for(Int_t task = 0; task < AliTRDrecoParam::kTRDreconstructionTasks; task++){
+      TDirectory *savedir = gDirectory;
+      fDebugStream[task] = new TTreeSRedirector(Form("TRD.Debug%s.root", fgTaskNames[task]));
+      savedir->cd();
+      SETFLG(fSteerParam, kOwner);
+    }
+  }
 }
 
 //_____________________________________________________________________________
@@ -107,11 +149,13 @@ void AliTRDReconstructor::ConvertDigits(AliRawReader *rawReader
   // Convert raw data digits into digit objects in a root tree
   //
 
-  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();
   rawReader->Select("TRD");
+  rawData.OpenOutput();
+  AliTRDrawStreamBase::SetRawStreamVersion(GetRecoParam()->GetRawStreamVersion()->Data());
   AliTRDdigitsManager *manager = rawData.Raw2Digits(rawReader);
   manager->MakeBranch(digitsTree);
   manager->WriteDigits();
@@ -132,19 +176,31 @@ void AliTRDReconstructor::Reconstruct(AliRawReader *rawReader
 
   rawReader->Reset();
   rawReader->Select("TRD");
+  AliTRDrawStreamBase::SetRawStreamVersion(GetRecoParam()->GetRawStreamVersion()->Data());
+
+  if(!fClusterizer){
+    AliFatal("Clusterizer not available!");
+    return;
+  }
+
+  fClusterizer->ResetRecPoints();
 
-  // New (fast) cluster finder
-  AliTRDclusterizer clusterer("clusterer","TRD clusterizer");
-  clusterer.SetReconstructor(this);
-  clusterer.OpenOutput(clusterTree);
-  clusterer.SetAddLabels(kFALSE);
-  clusterer.Raw2ClustersChamber(rawReader);
+  fClusterizer->OpenOutput(clusterTree);
+  fClusterizer->OpenTrackletOutput();
+  fClusterizer->SetUseLabels(kFALSE);
+  fClusterizer->Raw2ClustersChamber(rawReader);
   
   if(IsWritingClusters()) return;
 
   // take over ownership of clusters
-  fgClusters = clusterer.RecPoints();
-  clusterer.SetClustersOwner(kFALSE);
+  fgClusters = fClusterizer->RecPoints();
+  fClusterizer->SetClustersOwner(kFALSE);
+
+  // take over ownership of online tracklets
+  fgTracklets = fClusterizer->TrackletsArray();
+  fClusterizer->SetTrackletsOwner(kFALSE);
+
+  fgNTimeBins = fClusterizer->GetNTimeBins();
 }
 
 //_____________________________________________________________________________
@@ -156,18 +212,29 @@ void AliTRDReconstructor::Reconstruct(TTree *digitsTree
   //
 
   //AliInfo("Reconstruct TRD clusters from Digits [Digit TTree -> Cluster TTree]");
+  
+  if(!fClusterizer){
+    AliFatal("Clusterizer not available!");
+    return;
+  }
+
+  fClusterizer->ResetRecPoints();
 
-  AliTRDclusterizer clusterer("clusterer","TRD clusterizer");
-  clusterer.SetReconstructor(this);
-  clusterer.OpenOutput(clusterTree);
-  clusterer.ReadDigits(digitsTree);
-  clusterer.MakeClusters();
+  fClusterizer->OpenOutput(clusterTree);
+  fClusterizer->ReadDigits(digitsTree);
+  fClusterizer->MakeClusters();
 
   if(IsWritingClusters()) return;
 
   // take over ownership of clusters
-  fgClusters = clusterer.RecPoints();
-  clusterer.SetClustersOwner(kFALSE);
+  fgClusters = fClusterizer->RecPoints();
+  fClusterizer->SetClustersOwner(kFALSE);
+
+  // take over ownership of online tracklets
+  fgTracklets = fClusterizer->TrackletsArray();
+  fClusterizer->SetTrackletsOwner(kFALSE);
+
+  fgNTimeBins = fClusterizer->GetNTimeBins();
 }
 
 //_____________________________________________________________________________
@@ -195,82 +262,42 @@ void AliTRDReconstructor::FillESD(TTree* /*digitsTree*/
 
 }
 
-
 //_____________________________________________________________________________
 void AliTRDReconstructor::SetOption(Option_t *opt)
 {
-// Read option string into the steer param.
-//
-// Default steer param values
-//
-// write clusters [cw] = true
-// track seeding (stand alone tracking) [sa] = true
-// PID method in reconstruction (NN) [nn] = true
-// write online tracklets [tw] = false
-// drift gas [ar] = false
-// HLT tracking [hlt] = false
-// Cosmic Reconstruction [cos] = false
-//
-  fSteerParam = 0x00000007;
+  //
+  // Read option string into the steer param.
+  //
+
+  AliReconstructor::SetOption(opt);
 
   TString s(opt);
   TObjArray *opar = s.Tokenize(",");
   for(Int_t ipar=0; ipar<opar->GetEntriesFast(); ipar++){
+    Bool_t processed = kFALSE;
     TString sopt(((TObjString*)(*opar)[ipar])->String());
-    if(sopt.Contains("!cw")){ 
-      fSteerParam &= ~kWriteClusters;
-      continue;
-    } else if(sopt.Contains("!sa")){
-      fSteerParam &= ~kSeeding;
-      continue;
-    } else if(sopt.Contains("!nn")){
-      fSteerParam &= ~kSteerPID;
-      continue;
-    } else if(sopt.Contains("tw")){
-      if(!sopt.Contains("!")) fSteerParam |= kWriteTracklets;
-      continue;	
-    } else if(sopt.Contains("ar")){
-      if(!sopt.Contains("!")) fSteerParam |= kDriftGas;
-      continue;	
-    } else if(sopt.Contains("hlt")){
-      if(!sopt.Contains("!")) fSteerParam |= kHLT;
-      continue;	
-    } else if(sopt.Contains("cos")){
-      if(!sopt.Contains("!")) fSteerParam |= kCosmic;
-    } else if(sopt.Contains("sl")){
-      TObjArray *stl = sopt.Tokenize("_");
-      if(stl->GetEntriesFast() < 3) continue;
-      TString taskstr(((TObjString*)(*stl)[1])->String());
-      TString levelstring(((TObjString*)(*stl)[2])->String());
-      // Set the stream Level
-      Int_t level = levelstring.Atoi();
-      AliTRDReconstructorTask task = kTracker;
-      if(taskstr.CompareTo("cl") == 0) task = kClusterizer;	
-      else if(taskstr.CompareTo("tr") == 0) task = kTracker;
-      else if(taskstr.CompareTo("pi") == 0) task = kPID;
-      SetStreamLevel(level, task);
-      continue;
+    for(Int_t iopt=0; iopt<kNsteer; iopt++){
+      if(!sopt.Contains(fgSteerFlags[iopt])) continue;
+      SETFLG(fSteerParam, BIT(iopt));
+      if(sopt.Contains("!")) CLRFLG(fSteerParam, BIT(iopt));
+      processed = kTRUE;
+      break;	
     }
+    if(processed) continue;
+
+    AliWarning(Form("Unknown option flag %s.", sopt.Data()));
   }
 }
 
 //_____________________________________________________________________________
-void AliTRDReconstructor::SetStreamLevel(Int_t level, AliTRDReconstructorTask task){
+void AliTRDReconstructor::Options(UInt_t steer)
+{
   //
-  // Set the Stream Level for one of the tasks Clusterizer, Tracker or PID
+  // Print the options
   //
-  TString taskname;
-  switch(task){
-  case kClusterizer: 
-    taskname = "Clusterizer";
-    break;
-  case kTracker: 
-    taskname = "Tracker";
-    break;
-  case kPID: 
-    taskname = "PID";
-    break;
+
+  for(Int_t iopt=0; iopt<kNsteer; iopt++){
+    AliDebugGeneral("AliTRDReconstructor", 1, Form(" %s[%s]%s", fgSteerNames[iopt], fgSteerFlags[iopt], steer ?(((steer>>iopt)&1)?" : ON":" : OFF"):""));
   }
-  //AliInfo(Form("Setting Stream Level for Task %s to %d", taskname.Data(),level));
-  fStreamLevel[(Int_t)task] = level;
 }
+