Reflecting changes in AliPreprocessor

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

diff --git a/TRD/AliTRDReconstructor.cxx b/TRD/AliTRDReconstructor.cxx
index 6ef49234b4d7700fbf4e8296286a26a03480d899..ab56022cb7e9e01ace05d9ace61e7fdb830919ce 100644 (file)
--- a/TRD/AliTRDReconstructor.cxx
+++ b/TRD/AliTRDReconstructor.cxx
@@ -17,96 +17,213 @@
 
 ///////////////////////////////////////////////////////////////////////////////
 //                                                                           //
-// class for TRD reconstruction                                              //
+// Class for TRD reconstruction                                              //
 //                                                                           //
 ///////////////////////////////////////////////////////////////////////////////
 
+#include <TFile.h>
 
-#include "AliTRDReconstructor.h"
 #include "AliRunLoader.h"
+#include "AliRawReader.h"
+#include "AliLog.h"
+#include "AliESDTrdTrack.h"
+#include "AliESD.h"
+
+#include "AliTRDReconstructor.h"
 #include "AliTRDclusterizerV1.h"
 #include "AliTRDtracker.h"
 #include "AliTRDpidESD.h"
-#include <TFile.h>
-#include "AliRawReader.h"
-#include "AliLog.h"
+#include "AliTRDtrigger.h"
+#include "AliTRDtrigParam.h"
+#include "AliTRDgtuTrack.h"
 
 ClassImp(AliTRDReconstructor)
 
-
-Bool_t  AliTRDReconstructor::fgkSeedingOn = kFALSE;
+Bool_t AliTRDReconstructor::fgkSeedingOn  = kFALSE;
+Int_t  AliTRDReconstructor::fgStreamLevel = 0;      // Stream (debug) level
 
 //_____________________________________________________________________________
-void AliTRDReconstructor::Reconstruct(AliRunLoader* runLoader) const
+void AliTRDReconstructor::Reconstruct(AliRunLoader *runLoader) const
 {
-// reconstruct clusters
+  //
+  // Reconstruct clusters
+  //
 
-  AliLoader *loader=runLoader->GetLoader("TRDLoader");
+  AliLoader *loader = runLoader->GetLoader("TRDLoader");
   loader->LoadRecPoints("recreate");
 
-  AliTRDclusterizerV1 clusterer("clusterer", "TRD clusterizer");
   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");
+    clusterer.Open(runLoader->GetFileName(),iEvent);
     clusterer.ReadDigits();
     clusterer.MakeClusters();
     clusterer.WriteClusters(-1);
   }
 
   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();
+
 }
 
 //_____________________________________________________________________________
-void AliTRDReconstructor::Reconstruct(AliRunLoader* runLoader,
-                                      AliRawReader* rawReader) const
+void AliTRDReconstructor::Reconstruct(AliRunLoader *runLoader
+                                    , AliRawReader *rawReader) const
 {
-// reconstruct clusters
+  //
+  // Reconstruct clusters
+  //
 
   AliInfo("Reconstruct TRD clusters from RAW data");
 
-  AliLoader *loader=runLoader->GetLoader("TRDLoader");
+  AliLoader *loader = runLoader->GetLoader("TRDLoader");
   loader->LoadRecPoints("recreate");
 
-  AliTRDclusterizerV1 clusterer("clusterer", "TRD clusterizer");
   runLoader->CdGAFile();
   Int_t nEvents = runLoader->GetNumberOfEvents();
 
   for (Int_t iEvent = 0; iEvent < nEvents; iEvent++) {
     if (!rawReader->NextEvent()) break;
-    clusterer.Open(runLoader->GetFileName(), iEvent);
+    AliTRDclusterizerV1 clusterer("clusterer","TRD clusterizer");
+    clusterer.Open(runLoader->GetFileName(),iEvent);
     clusterer.ReadDigits(rawReader);
     clusterer.MakeClusters();
     clusterer.WriteClusters(-1);
   }
 
   loader->UnloadRecPoints();
+
+  //
+  // Trigger (tracklets, LTU)
+  //
+//   loader->LoadTracks();
+//   if (loader->TreeT()) {
+//     AliError("Tracklets already exist");
+//     return;
+//   }
+//   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);
+
+//   rawReader->RewindEvents();
+
+//   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();
+
 }
 
 //_____________________________________________________________________________
-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
+                               , AliESD *esd) 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);
+  //
+  // 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);
+
+  }
+
+}