Added methods to list available reports and detectors. Changed Fill(..) method to...

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

diff --git a/TRD/AliTRDReconstructor.cxx b/TRD/AliTRDReconstructor.cxx
index d48f229f9b1e7f20db09fb73eebcf2ff59e224bf..d39b371c4ef6c94fb3362a9096cd203c4bcf9b51 100644 (file)
--- a/TRD/AliTRDReconstructor.cxx
+++ b/TRD/AliTRDReconstructor.cxx
@@ -17,89 +17,226 @@
 
 ///////////////////////////////////////////////////////////////////////////////
 //                                                                           //
-// class for TRD reconstruction                                              //
+// Class for TRD reconstruction                                              //
 //                                                                           //
 ///////////////////////////////////////////////////////////////////////////////
 
+#include <TFile.h>
 
-#include "AliTRDReconstructor.h"
 #include "AliRunLoader.h"
-#include "AliTRDparameter.h"
+#include "AliRawReader.h"
+#include "AliLog.h"
+#include "AliESDTrdTrack.h"
+#include "AliESDEvent.h"
+
+#include "AliTRDReconstructor.h"
 #include "AliTRDclusterizerV1.h"
+#include "AliTRDclusterizerV2.h"
 #include "AliTRDtracker.h"
 #include "AliTRDpidESD.h"
-#include <TFile.h>
-
+#include "AliTRDgtuTrack.h"
+#include "AliTRDrawData.h"
+#include "AliTRDdigitsManager.h"
 
 ClassImp(AliTRDReconstructor)
 
+Bool_t AliTRDReconstructor::fgkSeedingOn  = kFALSE;
+Int_t  AliTRDReconstructor::fgStreamLevel = 0;      // Stream (debug) level
 
 //_____________________________________________________________________________
-void AliTRDReconstructor::Reconstruct(AliRunLoader* runLoader) const
+void AliTRDReconstructor::ConvertDigits(AliRawReader *rawReader
+                                     , TTree *digitsTree) const
 {
-// reconstruct clusters
+  //
+  // Convert raw data digits into digit objects in a root tree
+  //
+
+  AliInfo("Convert raw data digits into digit objects [RawReader -> Digit TTree]");
+
+  AliTRDrawData rawData;
+  rawReader->Reset();
+  rawReader->Select("TRD");
+  AliTRDdigitsManager *manager = rawData.Raw2Digits(rawReader);
+  manager->MakeBranch(digitsTree);
+  manager->WriteDigits();
 
-  AliLoader *loader=runLoader->GetLoader("TRDLoader");
+}
+
+//_____________________________________________________________________________
+void AliTRDReconstructor::Reconstruct(AliRunLoader *runLoader
+                                    , AliRawReader *rawReader) const
+{
+  //
+  // Reconstruct clusters
+  //
+
+  AliInfo("Reconstruct TRD clusters from RAW data [RunLoader, RawReader]");
+
+  AliLoader *loader = runLoader->GetLoader("TRDLoader");
   loader->LoadRecPoints("recreate");
 
-  AliTRDclusterizerV1 clusterer("clusterer", "TRD clusterizer");
   runLoader->CdGAFile();
-  AliTRDparameter* trdParam = GetTRDparameter(runLoader); 
-  if (!trdParam) {
-    Error("Reconstruct", "no TRD parameters found");
-    return;
+  Int_t nEvents = runLoader->GetNumberOfEvents();
+
+  rawReader->Reset();
+  rawReader->Select("TRD");
+
+  for (Int_t iEvent = 0; iEvent < nEvents; iEvent++) {
+
+    if (!rawReader->NextEvent()) break;
+
+    // Old (slow) cluster finder
+    //AliTRDclusterizerV1 clusterer("clusterer","TRD clusterizer");
+    //clusterer.Open(runLoader->GetFileName(),iEvent);
+    //clusterer.ReadDigits(rawReader);
+    //clusterer.MakeClusters();
+
+    // New (fast) cluster finder
+    AliTRDclusterizerV2 clusterer("clusterer","TRD clusterizer");
+    clusterer.Open(runLoader->GetFileName(),iEvent);
+    clusterer.Raw2ClustersChamber(rawReader);
+
+    clusterer.WriteClusters(-1);
+
   }
-  trdParam->ReInit();
-  clusterer.SetParameter(trdParam);
+
+  loader->UnloadRecPoints();
+
+}
+
+//_____________________________________________________________________________
+void AliTRDReconstructor::Reconstruct(AliRawReader *rawReader
+                                    , TTree *clusterTree) const
+{
+  //
+  // Reconstruct clusters
+  //
+
+  AliInfo("Reconstruct TRD clusters from RAW data [RawReader -> Cluster TTree]");
+
+  rawReader->Reset();
+  rawReader->Select("TRD");
+
+  // Old (slow) cluster finder
+  //AliTRDclusterizerV1 clusterer("clusterer","TRD clusterizer");
+  //clusterer.OpenOutput(clusterTree);
+  //clusterer.ReadDigits(rawReader);
+  //clusterer.MakeClusters();
+
+  // New (fast) cluster finder
+  AliTRDclusterizerV2 clusterer("clusterer","TRD clusterizer");
+  clusterer.OpenOutput(clusterTree);
+  clusterer.SetAddLabels(kFALSE);
+  clusterer.Raw2ClustersChamber(rawReader);
+}
+
+//_____________________________________________________________________________
+void AliTRDReconstructor::Reconstruct(TTree *digitsTree
+                                    , TTree *clusterTree) const
+{
+  //
+  // Reconstruct clusters
+  //
+  AliInfo("Reconstruct TRD clusters from Digits [Digit TTree -> Cluster TTree]");
+
+  //AliTRDclusterizerV1 clusterer("clusterer","TRD clusterizer");
+  AliTRDclusterizerV2 clusterer("clusterer","TRD clusterizer");
+  clusterer.OpenOutput(clusterTree);
+  clusterer.ReadDigits(digitsTree);
+  clusterer.MakeClusters();
+}
+
+//_____________________________________________________________________________
+void AliTRDReconstructor::Reconstruct(AliRunLoader *runLoader) const
+{
+  //
+  // Reconstruct clusters
+  //
+
+  AliInfo("Reconstruct TRD clusters [AliRunLoader]");
+  AliLoader *loader = runLoader->GetLoader("TRDLoader");
+  loader->LoadRecPoints("recreate");
+
+  runLoader->CdGAFile();
   Int_t nEvents = runLoader->GetNumberOfEvents();
 
   for (Int_t iEvent = 0; iEvent < nEvents; iEvent++) {
-    clusterer.Open(runLoader->GetFileName(), iEvent);
+    AliTRDclusterizerV1 clusterer("clusterer","TRD clusterizer");
+    //AliTRDclusterizerV2 clusterer("clusterer","TRD clusterizer");
+    clusterer.Open(runLoader->GetFileName(),iEvent);
     clusterer.ReadDigits();
     clusterer.MakeClusters();
     clusterer.WriteClusters(-1);
   }
 
   loader->UnloadRecPoints();
+
 }
 
 //_____________________________________________________________________________
-AliTracker* AliTRDReconstructor::CreateTracker(AliRunLoader* runLoader) const
+AliTracker *AliTRDReconstructor::CreateTracker(AliRunLoader *runLoader) const
 {
-// create a TRD tracker
+  //
+  // Create a TRD tracker
+  //
 
   runLoader->CdGAFile();
+
   return new AliTRDtracker(gFile);
+
 }
 
 //_____________________________________________________________________________
-void AliTRDReconstructor::FillESD(AliRunLoader* /*runLoader*/, 
-                                 AliESD* esd) const
+void AliTRDReconstructor::FillESD(AliRunLoader* /*runLoader*/
+                               , AliRawReader* /*rawReader*/
+                               , AliESDEvent* /*esd*/) const
 {
-// make PID
-
-  Double_t parTRD[] = {
-    187., // Min. Ionizing Particle signal.  Check it !!!
-    0.23, // relative resolution             Check it !!!
-    10.   // PID range (in sigmas)
-  };
-  AliTRDpidESD trdPID(parTRD);
-  trdPID.MakePID(esd);
+  //
+  // Make PID
+  //
+
+  //AliTRDpidESD trdPID;
+  //trdPID.MakePID(esd);
+
 }
 
+//_____________________________________________________________________________
+void AliTRDReconstructor::FillESD(AliRawReader* /*rawReader*/
+                               , TTree* /*clusterTree*/
+                               , AliESDEvent* /*esd*/) const
+{
+  //
+  // Make PID
+  //
+
+  //AliTRDpidESD trdPID;
+  //trdPID.MakePID(esd);
+
+}
 
 //_____________________________________________________________________________
-AliTRDparameter* AliTRDReconstructor::GetTRDparameter(AliRunLoader* runLoader) const
+void AliTRDReconstructor::FillESD(TTree* /*digitsTree*/
+                               , TTree* /*clusterTree*/
+                               , AliESDEvent* /*esd*/) const
 {
-// get the TRD parameters
+  //
+  // Make PID
+  //
+
+  //AliTRDpidESD trdPID;
+  //trdPID.MakePID(esd);
 
-  runLoader->CdGAFile();
-  AliTRDparameter* trdParam = (AliTRDparameter*) gFile->Get("TRDparameter"); 
-  if (!trdParam) {
-    Error("GetTRDparameter", "no TRD parameters available");
-    return NULL;
-  }
-  return trdParam;
 }
 
+//_____________________________________________________________________________
+void AliTRDReconstructor::FillESD(AliRunLoader* /*runLoader*/
+                               , AliESDEvent* /*esd*/) const
+{
+  //
+  // Make PID
+  //
+
+  //AliTRDpidESD trdPID;
+  //trdPID.MakePID(esd);
 
+}