]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - TRD/AliTRDReconstructor.cxx
Added methods to list available reports and detectors. Changed Fill(..) method to...
[u/mrichter/AliRoot.git] / TRD / AliTRDReconstructor.cxx
index 74afd72ec9982c5f60e65d765002804a6c65a6ca..d39b371c4ef6c94fb3362a9096cd203c4bcf9b51 100644 (file)
 #include "AliRawReader.h"
 #include "AliLog.h"
 #include "AliESDTrdTrack.h"
-#include "AliESD.h"
+#include "AliESDEvent.h"
 
 #include "AliTRDReconstructor.h"
 #include "AliTRDclusterizerV1.h"
+#include "AliTRDclusterizerV2.h"
 #include "AliTRDtracker.h"
 #include "AliTRDpidESD.h"
-#include "AliTRDtrigger.h"
-#include "AliTRDtrigParam.h"
 #include "AliTRDgtuTrack.h"
+#include "AliTRDrawData.h"
+#include "AliTRDdigitsManager.h"
 
 ClassImp(AliTRDReconstructor)
 
@@ -43,14 +44,21 @@ Bool_t AliTRDReconstructor::fgkSeedingOn  = kFALSE;
 Int_t  AliTRDReconstructor::fgStreamLevel = 0;      // Stream (debug) level
 
 //_____________________________________________________________________________
-void AliTRDReconstructor::ConvertDigits(AliRawReader* /*rawReader*/
-                                     , TTree* /*digitsTree*/) const
+void AliTRDReconstructor::ConvertDigits(AliRawReader *rawReader
+                                     , TTree *digitsTree) const
 {
   //
   // Convert raw data digits into digit objects in a root tree
   //
 
-  AliError("conversion of raw data digits into digit objects not implemented");
+  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();
 
 }
 
@@ -62,7 +70,7 @@ void AliTRDReconstructor::Reconstruct(AliRunLoader *runLoader
   // Reconstruct clusters
   //
 
-  AliInfo("Reconstruct TRD clusters from RAW data");
+  AliInfo("Reconstruct TRD clusters from RAW data [RunLoader, RawReader]");
 
   AliLoader *loader = runLoader->GetLoader("TRDLoader");
   loader->LoadRecPoints("recreate");
@@ -70,51 +78,32 @@ void AliTRDReconstructor::Reconstruct(AliRunLoader *runLoader
   runLoader->CdGAFile();
   Int_t nEvents = runLoader->GetNumberOfEvents();
 
-  for (Int_t iEvent = 0; iEvent < nEvents; iEvent++) {
-    if (!rawReader->NextEvent()) break;
-    AliTRDclusterizerV1 clusterer("clusterer","TRD clusterizer");
-    clusterer.Open(runLoader->GetFileName(),iEvent);
-    clusterer.ReadDigits(rawReader);
-    clusterer.MakeClusters();
-    clusterer.WriteClusters(-1);
-  }
-
-  loader->UnloadRecPoints();
+  rawReader->Reset();
+  rawReader->Select("TRD");
 
-  //
-  // Trigger (tracklets, LTU)
-  //
-//   loader->LoadTracks();
-//   if (loader->TreeT()) {
-//     AliError("Tracklets already exist");
-//     return;
-//   }
-//   AliInfo("Trigger tracklets will be produced");
+  for (Int_t iEvent = 0; iEvent < nEvents; iEvent++) {
 
-//   AliTRDtrigger trdTrigger("Trigger","Trigger class"); 
+    if (!rawReader->NextEvent()) break;
 
-//   AliTRDtrigParam *trigp = new AliTRDtrigParam("TRDtrigParam"
-//                                               ,"TRD Trigger parameters");
+    // Old (slow) cluster finder
+    //AliTRDclusterizerV1 clusterer("clusterer","TRD clusterizer");
+    //clusterer.Open(runLoader->GetFileName(),iEvent);
+    //clusterer.ReadDigits(rawReader);
+    //clusterer.MakeClusters();
 
-//   Float_t field = AliTracker::GetBz() * 0.1; // Tesla
-//   AliInfo(Form("Trigger set for magnetic field = %f Tesla \n",field));
-//   trigp->SetField(field);
-//   trigp->Init();
-//   trdTrigger.SetParameter(trigp);
+    // New (fast) cluster finder
+    AliTRDclusterizerV2 clusterer("clusterer","TRD clusterizer");
+    clusterer.Open(runLoader->GetFileName(),iEvent);
+    clusterer.Raw2ClustersChamber(rawReader);
 
-//   rawReader->RewindEvents();
+    clusterer.WriteClusters(-1);
 
-//   for (Int_t iEvent = 0; iEvent < nEvents; iEvent++) {
-//     if (!rawReader->NextEvent()) break;
-//     trdTrigger.Open(runLoader->GetFileName(),iEvent);
-//     trdTrigger.ReadDigits(rawReader);
-//     trdTrigger.MakeTracklets();
-//     trdTrigger.WriteTracklets(-1);
-//   }
+  }
 
-//   loader->UnloadTracks();
+  loader->UnloadRecPoints();
 
 }
+
 //_____________________________________________________________________________
 void AliTRDReconstructor::Reconstruct(AliRawReader *rawReader
                                     , TTree *clusterTree) const
@@ -123,37 +112,38 @@ void AliTRDReconstructor::Reconstruct(AliRawReader *rawReader
   // Reconstruct clusters
   //
 
-  AliInfo("Reconstruct TRD clusters from RAW data");
+  AliInfo("Reconstruct TRD clusters from RAW data [RawReader -> Cluster TTree]");
 
-  AliTRDclusterizerV1 clusterer("clusterer","TRD clusterizer");
-  clusterer.OpenOutput(clusterTree);
-  clusterer.ReadDigits(rawReader);
-  clusterer.MakeClusters();
-  clusterer.WriteClusters(-1);
+  rawReader->Reset();
+  rawReader->Select("TRD");
 
-  //
-  // No trigger, since we don't have the output tree for tracklets
-  //
+  // 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
+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");
+  //AliTRDclusterizerV1 clusterer("clusterer","TRD clusterizer");
+  AliTRDclusterizerV2 clusterer("clusterer","TRD clusterizer");
   clusterer.OpenOutput(clusterTree);
   clusterer.ReadDigits(digitsTree);
   clusterer.MakeClusters();
-  clusterer.WriteClusters(-1);
-
-  //
-  // No trigger, since we don't have the output tree for tracklets
-  //
-
 }
 
 //_____________________________________________________________________________
@@ -163,6 +153,7 @@ void AliTRDReconstructor::Reconstruct(AliRunLoader *runLoader) const
   // Reconstruct clusters
   //
 
+  AliInfo("Reconstruct TRD clusters [AliRunLoader]");
   AliLoader *loader = runLoader->GetLoader("TRDLoader");
   loader->LoadRecPoints("recreate");
 
@@ -171,6 +162,7 @@ void AliTRDReconstructor::Reconstruct(AliRunLoader *runLoader) const
 
   for (Int_t iEvent = 0; iEvent < nEvents; iEvent++) {
     AliTRDclusterizerV1 clusterer("clusterer","TRD clusterizer");
+    //AliTRDclusterizerV2 clusterer("clusterer","TRD clusterizer");
     clusterer.Open(runLoader->GetFileName(),iEvent);
     clusterer.ReadDigits();
     clusterer.MakeClusters();
@@ -179,32 +171,6 @@ void AliTRDReconstructor::Reconstruct(AliRunLoader *runLoader) const
 
   loader->UnloadRecPoints();
 
-  //
-  // Trigger (tracklets, LTU)
-  //
-  loader->LoadTracks("RECREATE");
-  AliInfo("Trigger tracklets will be produced");
-
-  AliTRDtrigger trdTrigger("Trigger","Trigger class"); 
-
-  AliTRDtrigParam *trigp = new AliTRDtrigParam("TRDtrigParam"
-                                              ,"TRD Trigger parameters");
-
-  Float_t field = AliTracker::GetBz() * 0.1; // Tesla
-  AliInfo(Form("Trigger set for magnetic field = %f Tesla \n",field));
-  trigp->SetField(field);
-  trigp->Init();
-  trdTrigger.SetParameter(trigp);
-
-  for (Int_t iEvent = 0; iEvent < nEvents; iEvent++) {
-    trdTrigger.Open(runLoader->GetFileName(),iEvent);
-    trdTrigger.ReadDigits();
-    trdTrigger.MakeTracklets();
-    trdTrigger.WriteTracklets(-1);
-  }
-
-  loader->UnloadTracks();
-
 }
 
 //_____________________________________________________________________________
@@ -223,113 +189,54 @@ AliTracker *AliTRDReconstructor::CreateTracker(AliRunLoader *runLoader) const
 //_____________________________________________________________________________
 void AliTRDReconstructor::FillESD(AliRunLoader* /*runLoader*/
                                , AliRawReader* /*rawReader*/
-                               , AliESD *esd) const
+                               , AliESDEvent* /*esd*/) const
 {
   //
   // Make PID
   //
 
-  AliTRDpidESD trdPID;
-  trdPID.MakePID(esd);
-
-  //
-  // No trigger, since we don't have the output tree for tracklets
-  //
+  //AliTRDpidESD trdPID;
+  //trdPID.MakePID(esd);
 
 }
 
 //_____________________________________________________________________________
 void AliTRDReconstructor::FillESD(AliRawReader* /*rawReader*/
                                , TTree* /*clusterTree*/
-                               , AliESD *esd) const
+                               , AliESDEvent* /*esd*/) const
 {
   //
   // Make PID
   //
 
-  AliTRDpidESD trdPID;
-  trdPID.MakePID(esd);
-
-  //
-  // No trigger, since we don't have the output tree for tracklets
-  //
+  //AliTRDpidESD trdPID;
+  //trdPID.MakePID(esd);
 
 }
 
 //_____________________________________________________________________________
 void AliTRDReconstructor::FillESD(TTree* /*digitsTree*/
                                , TTree* /*clusterTree*/
-                               , AliESD *esd) const
+                               , AliESDEvent* /*esd*/) const
 {
   //
   // Make PID
   //
 
-  AliTRDpidESD trdPID;
-  trdPID.MakePID(esd);
-
-  //
-  // No trigger, since we don't have the output tree for tracklets
-  //
+  //AliTRDpidESD trdPID;
+  //trdPID.MakePID(esd);
 
 }
 
 //_____________________________________________________________________________
-void AliTRDReconstructor::FillESD(AliRunLoader *runLoader
-                               , AliESD *esd) const
+void AliTRDReconstructor::FillESD(AliRunLoader* /*runLoader*/
+                               , AliESDEvent* /*esd*/) const
 {
   //
   // Make PID
   //
 
-  AliTRDpidESD trdPID;
-  trdPID.MakePID(esd);
-
-  //
-  // Trigger (tracks, GTU)
-  //
-  AliTRDtrigger trdTrigger("Trigger","Trigger class"); 
-
-  AliTRDtrigParam *trigp = new AliTRDtrigParam("TRDtrigParam"
-                                              ,"TRD Trigger parameters");
-
-  Float_t field = AliTracker::GetBz() * 0.1; // Tesla
-  AliInfo(Form("Trigger set for magnetic field = %f Tesla \n",field));
-  trigp->SetField(field);
-  trigp->Init();
-
-  trdTrigger.SetParameter(trigp);
-  trdTrigger.SetRunLoader(runLoader);
-  trdTrigger.Init();
-
-  Int_t iEvent = runLoader->GetEventNumber(); 
-  runLoader->GetEvent(iEvent);
-  trdTrigger.ReadTracklets(runLoader);
-
-  AliESDTrdTrack *TrdTrack = new AliESDTrdTrack();
-  AliTRDgtuTrack *GtuTrack;
-
-  Int_t nTracks = trdTrigger.GetNumberOfTracks();
-  for (Int_t iTrack = 0; iTrack < nTracks; iTrack++) {
-
-    GtuTrack = trdTrigger.GetTrack(iTrack);
-
-    TrdTrack->SetYproj(GtuTrack->GetYproj());
-    TrdTrack->SetZproj(GtuTrack->GetZproj());
-    TrdTrack->SetSlope(GtuTrack->GetSlope());
-    TrdTrack->SetDetector(GtuTrack->GetDetector());
-    TrdTrack->SetTracklets(GtuTrack->GetTracklets());
-    TrdTrack->SetPlanes(GtuTrack->GetPlanes());
-    TrdTrack->SetClusters(GtuTrack->GetClusters());
-    TrdTrack->SetPt(GtuTrack->GetPt());
-    TrdTrack->SetPhi(GtuTrack->GetPhi());
-    TrdTrack->SetEta(GtuTrack->GetEta());
-    TrdTrack->SetLabel(GtuTrack->GetLabel());
-    TrdTrack->SetPID(GtuTrack->GetPID());
-    TrdTrack->SetIsElectron(GtuTrack->IsElectron());
-
-    esd->AddTrdTrack(TrdTrack);
-
-  }
+  //AliTRDpidESD trdPID;
+  //trdPID.MakePID(esd);
 
 }