QA update by Sylwester

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

diff --git a/TRD/AliTRDReconstructor.cxx b/TRD/AliTRDReconstructor.cxx
index 425a1bd9bddd08d9270e3193ff91012287773bf0..fb506015059e20ac5b397203e0976cffc4697982 100644 (file)
--- a/TRD/AliTRDReconstructor.cxx
+++ b/TRD/AliTRDReconstructor.cxx
@@ -17,78 +17,118 @@
 
 ///////////////////////////////////////////////////////////////////////////////
 //                                                                           //
-// class for TRD reconstruction                                              //
+// Class for TRD reconstruction                                              //
 //                                                                           //
 ///////////////////////////////////////////////////////////////////////////////
 
+#include <TFile.h>
 
-#include "AliTRDReconstructor.h"
 #include "AliRunLoader.h"
-#include "AliTRDparameter.h"
-#include "AliTRD.h"
-#include "AliTRDclusterizerV1.h"
-#include "AliTRDtracker.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; 
 
 //_____________________________________________________________________________
-void AliTRDReconstructor::Reconstruct(AliRunLoader* runLoader) const
-{
-// reconstruct clusters
-
-  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;
-  }
-  trdParam->ReInit();
-  clusterer.SetParameter(trdParam);
-  Int_t nEvents = runLoader->GetNumberOfEvents();
-
-  for (Int_t iEvent = 0; iEvent < nEvents; iEvent++) {
-    clusterer.Open(runLoader->GetFileName(), iEvent);
-    clusterer.ReadDigits();
-    clusterer.MakeClusters();
-    clusterer.WriteClusters(-1);
-  }
+AliTRDReconstructor::~AliTRDReconstructor()                  { 
+       if(fgRecoParam) delete fgRecoParam;
 }
 
+
+
 //_____________________________________________________________________________
-AliTracker* AliTRDReconstructor::CreateTracker(AliRunLoader* runLoader) const
+void AliTRDReconstructor::ConvertDigits(AliRawReader *rawReader
+                                     , TTree *digitsTree) const
 {
-// create a TRD tracker
+  //
+  // 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();
+  delete manager;
 
-  runLoader->CdGAFile();
-  return new AliTRDtracker(gFile);
 }
 
 //_____________________________________________________________________________
-void AliTRDReconstructor::FillESD(AliRunLoader* /*runLoader*/, 
-                                 AliESD* /*esd*/) const
+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");
+
+  // New (fast) cluster finder
+  AliTRDclusterizer 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]");
+
+  AliTRDclusterizer clusterer("clusterer","TRD clusterizer");
+  clusterer.OpenOutput(clusterTree);
+  clusterer.ReadDigits(digitsTree);
+  clusterer.MakeClusters();
+
+}
 
 //_____________________________________________________________________________
-AliTRDparameter* AliTRDReconstructor::GetTRDparameter(AliRunLoader* runLoader) const
+AliTracker *AliTRDReconstructor::CreateTracker() const
 {
-// get the TRD parameters
-
-  runLoader->CdGAFile();
-  AliTRDparameter* trdParam = (AliTRDparameter*) gFile->Get("TRDparameter"); 
-  if (!trdParam) {
-    Error("GetTRDparameter", "no TRD parameters available");
-    return NULL;
-  }
-  return trdParam;
+  //
+  // Create a TRD tracker
+  //
+
+  //return new AliTRDtracker(NULL);
+
+  return new AliTRDtrackerV1();
+
 }
 
+//_____________________________________________________________________________
+void AliTRDReconstructor::FillESD(TTree* /*digitsTree*/
+                               , TTree* /*clusterTree*/
+                               , AliESDEvent* /*esd*/) const
+{
+  //
+  // Fill ESD
+  //
 
+}