From Cvetan: when opening an event from ESD get the first tree-entry so that run...

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

diff --git a/TRD/AliTRDReconstructor.cxx b/TRD/AliTRDReconstructor.cxx
index 6ef49234b4d7700fbf4e8296286a26a03480d899..fb506015059e20ac5b397203e0976cffc4697982 100644 (file)
--- a/TRD/AliTRDReconstructor.cxx
+++ b/TRD/AliTRDReconstructor.cxx
@@ -17,96 +17,118 @@
 
 ///////////////////////////////////////////////////////////////////////////////
 //                                                                           //
-// class for TRD reconstruction                                              //
+// Class for TRD reconstruction                                              //
 //                                                                           //
 ///////////////////////////////////////////////////////////////////////////////
 
+#include <TFile.h>
 
-#include "AliTRDReconstructor.h"
 #include "AliRunLoader.h"
-#include "AliTRDclusterizerV1.h"
-#include "AliTRDtracker.h"
-#include "AliTRDpidESD.h"
-#include <TFile.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 "AliTRDdigitsManager.h"
+#include "AliTRDtrackerV1.h"
+#include "AliTRDrecoParam.h"
 
 ClassImp(AliTRDReconstructor)
 
+AliTRDrecoParam* AliTRDReconstructor::fgRecoParam = 0x0; 
+
+//_____________________________________________________________________________
+AliTRDReconstructor::~AliTRDReconstructor()                  { 
+       if(fgRecoParam) delete fgRecoParam;
+}
+
 
-Bool_t  AliTRDReconstructor::fgkSeedingOn = kFALSE;
 
 //_____________________________________________________________________________
-void AliTRDReconstructor::Reconstruct(AliRunLoader* runLoader) const
+void AliTRDReconstructor::ConvertDigits(AliRawReader *rawReader
+                                     , TTree *digitsTree) const
 {
-// reconstruct clusters
-
-  AliLoader *loader=runLoader->GetLoader("TRDLoader");
-  loader->LoadRecPoints("recreate");
+  //
+  // Convert raw data digits into digit objects in a root tree
+  //
 
-  AliTRDclusterizerV1 clusterer("clusterer", "TRD clusterizer");
-  runLoader->CdGAFile();
-  Int_t nEvents = runLoader->GetNumberOfEvents();
+  AliInfo("Convert raw data digits into digit objects [RawReader -> Digit TTree]");
 
-  for (Int_t iEvent = 0; iEvent < nEvents; iEvent++) {
-    clusterer.Open(runLoader->GetFileName(), iEvent);
-    clusterer.ReadDigits();
-    clusterer.MakeClusters();
-    clusterer.WriteClusters(-1);
-  }
+  AliTRDrawData rawData;
+  rawReader->Reset();
+  rawReader->Select("TRD");
+  AliTRDdigitsManager *manager = rawData.Raw2Digits(rawReader);
+  manager->MakeBranch(digitsTree);
+  manager->WriteDigits();
+  delete manager;
 
-  loader->UnloadRecPoints();
 }
 
 //_____________________________________________________________________________
-void AliTRDReconstructor::Reconstruct(AliRunLoader* runLoader,
-                                      AliRawReader* rawReader) const
+void AliTRDReconstructor::Reconstruct(AliRawReader *rawReader
+                                    , TTree *clusterTree) const
 {
-// reconstruct clusters
+  //
+  // Reconstruct clusters
+  //
 
-  AliInfo("Reconstruct TRD clusters from RAW data");
+  AliInfo("Reconstruct TRD clusters from RAW data [RawReader -> Cluster TTree]");
 
-  AliLoader *loader=runLoader->GetLoader("TRDLoader");
-  loader->LoadRecPoints("recreate");
 
-  AliTRDclusterizerV1 clusterer("clusterer", "TRD clusterizer");
-  runLoader->CdGAFile();
-  Int_t nEvents = runLoader->GetNumberOfEvents();
+  rawReader->Reset();
+  rawReader->Select("TRD");
 
-  for (Int_t iEvent = 0; iEvent < nEvents; iEvent++) {
-    if (!rawReader->NextEvent()) break;
-    clusterer.Open(runLoader->GetFileName(), iEvent);
-    clusterer.ReadDigits(rawReader);
-    clusterer.MakeClusters();
-    clusterer.WriteClusters(-1);
-  }
+  // New (fast) cluster finder
+  AliTRDclusterizer clusterer("clusterer","TRD clusterizer");
+  clusterer.OpenOutput(clusterTree);
+  clusterer.SetAddLabels(kFALSE);
+  clusterer.Raw2ClustersChamber(rawReader);
 
-  loader->UnloadRecPoints();
 }
 
 //_____________________________________________________________________________
-AliTracker* AliTRDReconstructor::CreateTracker(AliRunLoader* runLoader) const
+void AliTRDReconstructor::Reconstruct(TTree *digitsTree
+                                    , TTree *clusterTree) const
 {
-// create a TRD tracker
+  //
+  // Reconstruct clusters
+  //
+
+  AliInfo("Reconstruct TRD clusters from Digits [Digit TTree -> Cluster TTree]");
+
+  AliTRDclusterizer clusterer("clusterer","TRD clusterizer");
+  clusterer.OpenOutput(clusterTree);
+  clusterer.ReadDigits(digitsTree);
+  clusterer.MakeClusters();
 
-  runLoader->CdGAFile();
-  return new AliTRDtracker(gFile);
 }
 
 //_____________________________________________________________________________
-void AliTRDReconstructor::FillESD(AliRunLoader* /*runLoader*/, 
-                                 AliESD* esd) const
+AliTracker *AliTRDReconstructor::CreateTracker() const
 {
-// make PID
-
-  Double_t parTRD[] = {
-    280., // Min. Ionizing Particle signal.  Check it !!!
-    0.23, // relative resolution             Check it !!!
-    10.   // PID range (in sigmas)
-  };
-  AliTRDpidESD trdPID(parTRD);
-  trdPID.MakePID(esd);
-}
+  //
+  // Create a TRD tracker
+  //
 
+  //return new AliTRDtracker(NULL);
 
+  return new AliTRDtrackerV1();
 
+}
+
+//_____________________________________________________________________________
+void AliTRDReconstructor::FillESD(TTree* /*digitsTree*/
+                               , TTree* /*clusterTree*/
+                               , AliESDEvent* /*esd*/) const
+{
+  //
+  // Fill ESD
+  //
+
+}