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

diff --git a/TRD/AliTRDReconstructor.cxx b/TRD/AliTRDReconstructor.cxx
index cc735de661a..86384766881 100644
--- a/TRD/AliTRDReconstructor.cxx
+++ b/TRD/AliTRDReconstructor.cxx
@@ -21,84 +21,99 @@
 //                                                                           //
 ///////////////////////////////////////////////////////////////////////////////
 
-#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"
-
-#include "TTreeStream.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(0)
+  ,fClusterizer(NULL)
 {
   // 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);
-  // PID method in reconstruction (NN) [nn]
-  SETFLG(fSteerParam, kSteerPID);
-
-  memset(fStreamLevel, 0, 5*sizeof(UChar_t));
-  memset(fDebugStream, 0, sizeof(TTreeSRedirector *) * 4);
-  // 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
-  SetBit(kOwner, kTRUE);
-}
 
-//_____________________________________________________________________________
-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));
-  memcpy(fDebugStream, r.fDebugStream, sizeof(TTreeSRedirector *) *4);
-  SetBit(kOwner, kFALSE);
+  memset(fDebugStream, 0, sizeof(TTreeSRedirector *) * AliTRDrecoParam::kTRDreconstructionTasks);
 }
 
 //_____________________________________________________________________________
 AliTRDReconstructor::~AliTRDReconstructor()
 {
+  //
+  // Destructor
+  //
+
   if(fgClusters) {
-    fgClusters->Delete(); delete fgClusters;
+    fgClusters->Delete();
+    delete fgClusters;
+    fgClusters = NULL;
   }
-  if(TestBit(kOwner)){
-    for(Int_t itask = 0; itask < 4; itask++)
+  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;
+  }
 }
 
 
@@ -108,17 +123,22 @@ void AliTRDReconstructor::Init(){
   // Init Options
   //
   SetOption(GetOption());
+  Options(fSteerParam);
 
-  AliInfo(Form("\tDigitsConversion       [dc] : %s", fSteerParam&kDigitsConversion?"yes":"no"));
-  AliInfo(Form("\tWrite Clusters         [cw] : %s", fSteerParam&kWriteClusters?"yes":"no"));
-  AliInfo(Form("\tWrite Online Tracklets [tw] : %s", fSteerParam&kWriteTracklets?"yes":"no"));
-  AliInfo(Form("\tDrift Gas Argon        [ar] : %s", fSteerParam&kDriftGas?"yes":"no"));
-  AliInfo(Form("\tStand Alone Tracking   [sa] : %s", fSteerParam&kSeeding?"yes":"no"));
-  AliInfo(Form("\tHLT         Tracking  [hlt] : %s", fSteerParam&kHLT?"yes":"no"));
-  AliInfo(Form("\tCosmic Reconstruction [cos] : %s", fSteerParam&kCosmic?"yes":"no"));
-  AliInfo(Form("\tNN PID                 [nn] : %s", fSteerParam&kSteerPID?"yes":"no"));
-  AliInfo(Form("\t8 dEdx slices in ESD   [8s] : %s", fSteerParam&kEightSlices?"yes":"no"));
-  AliInfo(Form("\tStreaming Levels            : Clusterizer[%d] Tracker[%d] PID[%d]", 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);
+    }
+  }
 }
 
 //_____________________________________________________________________________
@@ -128,13 +148,14 @@ 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();
   rawReader->Select("TRD");
+  rawData.OpenOutput();
+  AliTRDrawStreamBase::SetRawStreamVersion(GetRecoParam()->GetRawStreamVersion()->Data());
   AliTRDdigitsManager *manager = rawData.Raw2Digits(rawReader);
   manager->MakeBranch(digitsTree);
   manager->WriteDigits();
@@ -155,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;
+  }
 
-  // New (fast) cluster finder
-  AliTRDclusterizer clusterer("clusterer","TRD clusterizer");
-  clusterer.SetReconstructor(this);
-  clusterer.OpenOutput(clusterTree);
-  clusterer.SetAddLabels(kFALSE);
-  clusterer.Raw2ClustersChamber(rawReader);
+  fClusterizer->ResetRecPoints();
+
+  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();
 }
 
 //_____________________________________________________________________________
@@ -179,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();
 }
 
 //_____________________________________________________________________________
@@ -218,97 +262,42 @@ void AliTRDReconstructor::FillESD(TTree* /*digitsTree*/
 
 }
 
-
 //_____________________________________________________________________________
 void AliTRDReconstructor::SetOption(Option_t *opt)
 {
-// Read option string into the steer param.
-//
-// Default steer param values
-//
-// digits conversion [dc] = false
-// drift gas [ar] = false - do not update the number of exponentials in the TC !
-// write clusters [cw] = true
-// write online tracklets [tw] = false
-// track seeding (stand alone tracking) [sa] = true
-// PID method in reconstruction (NN) [nn] = true
-// 8 dEdx slices in ESD [8s] = false 
-// HLT tracking [hlt] = false
-// Cosmic Reconstruction [cos] = false
-//
+  //
+  // 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("dc")){
-      SETFLG(fSteerParam, kDigitsConversion);
-      if(sopt.Contains("!")) CLRFLG(fSteerParam, kDigitsConversion);
-      continue;	
-    } else if(sopt.Contains("cw")){ 
-      SETFLG(fSteerParam, kWriteClusters);
-      if(sopt.Contains("!")) CLRFLG(fSteerParam, kWriteClusters);
-      continue;
-    } else if(sopt.Contains("sa")){
-      SETFLG(fSteerParam, kSeeding);
-      if(sopt.Contains("!")) CLRFLG(fSteerParam, kSeeding);
-      continue;
-    } else if(sopt.Contains("nn")){
-      SETFLG(fSteerParam, kSteerPID);
-      if(sopt.Contains("!")) CLRFLG(fSteerParam, kSteerPID);
-      continue;
-    } else if(sopt.Contains("8s")){
-      SETFLG(fSteerParam, kEightSlices);
-      if(sopt.Contains("!")) CLRFLG(fSteerParam, kEightSlices);
-      continue;
-    } else if(sopt.Contains("tw")){
-      SETFLG(fSteerParam, kWriteTracklets);
-      if(sopt.Contains("!")) CLRFLG(fSteerParam, kWriteTracklets);
-      continue;	
-    } else if(sopt.Contains("ar")){
-      SETFLG(fSteerParam, kDriftGas);
-      if(sopt.Contains("!")) CLRFLG(fSteerParam, kDriftGas);
-      continue;	
-    } else if(sopt.Contains("hlt")){
-      SETFLG(fSteerParam, kHLT);
-      if(sopt.Contains("!")) CLRFLG(fSteerParam, kHLT);
-      continue;	
-    } else if(sopt.Contains("cos")){
-      SETFLG(fSteerParam, kCosmic);
-      if(sopt.Contains("!")) CLRFLG(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("raw") == 0) task = kRawReader;	
-      else 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[4] = {"RawReader", "Clusterizer", "Tracker", "PID"};
-  const Int_t minLevel[4] = {1, 1, 2, 1}; // the minimum debug level upon which a debug stream is created for different tasks
-  //AliInfo(Form("Setting Stream Level for Task %s to %d", taskname.Data(),level));
-  fStreamLevel[(Int_t)task] = level;
-  // Initialize DebugStreamer if not yet done
-  if(level >= minLevel[task] && !fDebugStream[task]){
-    TDirectory *savedir = gDirectory;
-    fDebugStream[task] = new TTreeSRedirector(Form("TRD.%sDebug.root", taskname[task].Data()));
-    savedir->cd();
+
+  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"):""));
   }
 }
+