]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - TRD/AliTRDReconstructor.cxx
- version HLT-v0-4 ready
[u/mrichter/AliRoot.git] / TRD / AliTRDReconstructor.cxx
index 82a9e40a9ad267c1a3ff521dba4929e2de0e0a31..dbe04b02e43327964d848c1605cf8a7aa2ac0756 100644 (file)
 //                                                                           //
 ///////////////////////////////////////////////////////////////////////////////
 
+#include <TFile.h>
 
-#include "AliTRDReconstructor.h"
 #include "AliRunLoader.h"
-#include "AliTRDparameter.h"
+#include "AliRawReader.h"
+#include "AliLog.h"
+#include "AliRun.h"
+#include "AliESDTrdTrack.h"
+#include "AliESD.h"
+
+#include "AliTRDReconstructor.h"
 #include "AliTRDclusterizerV1.h"
 #include "AliTRDtracker.h"
 #include "AliTRDpidESD.h"
-#include <TFile.h>
-
+#include "AliTRDtrigger.h"
+#include "AliTRDtrigParam.h"
+#include "AliTRDgtuTrack.h"
 
 ClassImp(AliTRDReconstructor)
 
+Bool_t AliTRDReconstructor::fgkSeedingOn  = kFALSE;
+Int_t  AliTRDReconstructor::fgStreamLevel = 0;      // Stream (debug) level
 
 //_____________________________________________________________________________
-void AliTRDReconstructor::Reconstruct(AliRunLoaderrunLoader) 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();
-  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);
+    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");
+
+  if (runLoader->GetAliRun() == 0x0) {
+    runLoader->LoadgAlice();
+  }
+  gAlice = runLoader->GetAliRun();
+  Double_t x[3]  = { 0.0, 0.0, 0.0 };
+  Double_t b[3];
+  gAlice->Field(x,b);          // b[] is in kilo Gauss
+  Float_t  field = b[2] * 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();
+
 }
 
 //_____________________________________________________________________________
-AliTracker* AliTRDReconstructor::CreateTracker(AliRunLoader* runLoader) const
+void AliTRDReconstructor::Reconstruct(AliRunLoader *runLoader
+                                    , AliRawReader *rawReader) const
 {
-// create a TRD tracker
+  //
+  // Reconstruct clusters
+  //
+
+  AliInfo("Reconstruct TRD clusters from RAW data");
+
+  AliLoader *loader = runLoader->GetLoader("TRDLoader");
+  loader->LoadRecPoints("recreate");
 
   runLoader->CdGAFile();
-  return new AliTRDtracker(gFile);
+  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();
+
+  //
+  // 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");
+
+  if (runLoader->GetAliRun() == 0x0) {
+    runLoader->LoadgAlice();
+  }
+  gAlice = runLoader->GetAliRun();
+  Double_t x[3]  = { 0.0, 0.0, 0.0 };
+  Double_t b[3];
+  gAlice->Field(x,b);          // b[] is in kilo Gauss
+  Float_t  field = b[2] * 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();
+
 }
 
 //_____________________________________________________________________________
-void AliTRDReconstructor::FillESD(AliRunLoader* /*runLoader*/, 
-                                 AliESD* esd) const
+AliTracker *AliTRDReconstructor::CreateTracker(AliRunLoader *runLoader) 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
+  //
+
+  runLoader->CdGAFile();
 
+  return new AliTRDtracker(gFile);
+
+}
 
 //_____________________________________________________________________________
-AliTRDparameter* AliTRDReconstructor::GetTRDparameter(AliRunLoader* runLoader) const
+void AliTRDReconstructor::FillESD(AliRunLoader *runLoader
+                                , AliESD *esd) const
 {
-// get the TRD parameters
+  //
+  // Make PID
+  //
 
-  runLoader->CdGAFile();
-  AliTRDparameter* trdParam = (AliTRDparameter*) gFile->Get("TRDparameter"); 
-  if (!trdParam) {
-    Error("GetTRDparameter", "no TRD parameters available");
-    return NULL;
+  AliTRDpidESD trdPID;
+  trdPID.MakePID(esd);
+
+  //
+  // Trigger (tracks, GTU)
+  //
+  AliTRDtrigger trdTrigger("Trigger","Trigger class"); 
+
+  AliTRDtrigParam *trigp = new AliTRDtrigParam("TRDtrigParam"
+                                              ,"TRD Trigger parameters");
+
+  if (runLoader->GetAliRun() == 0x0) {
+    runLoader->LoadgAlice();
   }
-  return trdParam;
-}
+  gAlice = runLoader->GetAliRun();
+  Double_t x[3]  = { 0.0, 0.0, 0.0 };
+  Double_t b[3];
+  gAlice->Field(x,b);          // b[] is in kilo Gauss
+  Float_t  field = b[2] * 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);
 
+  }
 
+}