///////////////////////////////////////////////////////////////////////////////
// //
-// class for TRD reconstruction //
+// Class for TRD reconstruction //
// //
///////////////////////////////////////////////////////////////////////////////
#include "AliRunLoader.h"
#include "AliRawReader.h"
#include "AliLog.h"
-#include "AliRun.h"
#include "AliESDTrdTrack.h"
-#include "AliESD.h"
+#include "AliESDEvent.h"
#include "AliTRDReconstructor.h"
-#include "AliTRDclusterizerV1.h"
+#include "AliTRDclusterizer.h"
#include "AliTRDtracker.h"
#include "AliTRDpidESD.h"
-#include "AliTRDtrigger.h"
-#include "AliTRDtrigParam.h"
#include "AliTRDgtuTrack.h"
+#include "AliTRDrawData.h"
+#include "AliTRDdigitsManager.h"
+#include "AliTRDtrackerV1.h"
+#include "AliTRDrecoParam.h"
ClassImp(AliTRDReconstructor)
-Bool_t AliTRDReconstructor::fgkSeedingOn = kFALSE;
-Int_t AliTRDReconstructor::fgStreamLevel = 0; // Stream (debug) level
+AliTRDrecoParam* AliTRDReconstructor::fgRecoParam = 0x0;
//_____________________________________________________________________________
-void AliTRDReconstructor::Reconstruct(AliRunLoader *runLoader) const
+AliTRDReconstructor::~AliTRDReconstructor() {
+ if(fgRecoParam) delete fgRecoParam;
+}
+
+
+
+//_____________________________________________________________________________
+void AliTRDReconstructor::ConvertDigits(AliRawReader *rawReader
+ , TTree *digitsTree) const
{
//
- // Reconstruct clusters
+ // Convert raw data digits into digit objects in a root tree
//
- AliLoader *loader = runLoader->GetLoader("TRDLoader");
- loader->LoadRecPoints("recreate");
+ AliInfo("Convert raw data digits into digit objects [RawReader -> Digit TTree]");
- runLoader->CdGAFile();
- Int_t nEvents = runLoader->GetNumberOfEvents();
+ AliTRDrawData rawData;
+ rawReader->Reset();
+ rawReader->Select("TRD");
+ AliTRDdigitsManager *manager = rawData.Raw2Digits(rawReader);
+ manager->MakeBranch(digitsTree);
+ manager->WriteDigits();
+ delete manager;
- for (Int_t iEvent = 0; iEvent < nEvents; iEvent++) {
- AliTRDclusterizerV1 clusterer("clusterer","TRD clusterizer");
- clusterer.Open(runLoader->GetFileName(),iEvent);
- clusterer.ReadDigits();
- clusterer.MakeClusters();
- clusterer.WriteClusters(-1);
- }
-
- loader->UnloadRecPoints();
+}
+//_____________________________________________________________________________
+void AliTRDReconstructor::Reconstruct(AliRawReader *rawReader
+ , TTree *clusterTree) const
+{
//
- // Trigger (tracklets, LTU)
+ // Reconstruct clusters
//
- loader->LoadTracks("RECREATE");
- AliInfo("Trigger tracklets will be produced");
- AliTRDtrigger trdTrigger("Trigger","Trigger class");
+ AliInfo("Reconstruct TRD clusters from RAW data [RawReader -> Cluster TTree]");
- AliTRDtrigParam *trigp = new AliTRDtrigParam("TRDtrigParam"
- ,"TRD Trigger parameters");
+ // TODO move it to rec.C. check TPC
+ if (!fgRecoParam) fgRecoParam = AliTRDrecoParam::GetLowFluxParam();
- 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));
+ rawReader->Reset();
+ rawReader->Select("TRD");
- 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();
+ // New (fast) cluster finder
+ AliTRDclusterizer clusterer("clusterer","TRD clusterizer");
+ clusterer.OpenOutput(clusterTree);
+ clusterer.SetAddLabels(kFALSE);
+ clusterer.Raw2ClustersChamber(rawReader);
}
//_____________________________________________________________________________
-void AliTRDReconstructor::Reconstruct(AliRunLoader *runLoader
- , AliRawReader *rawReader) const
+void AliTRDReconstructor::Reconstruct(TTree *digitsTree
+ , TTree *clusterTree) const
{
//
// Reconstruct clusters
//
- AliInfo("Reconstruct TRD clusters from RAW data");
-
- AliLoader *loader = runLoader->GetLoader("TRDLoader");
- loader->LoadRecPoints("recreate");
+ AliInfo("Reconstruct TRD clusters from Digits [Digit TTree -> Cluster TTree]");
- runLoader->CdGAFile();
- Int_t nEvents = runLoader->GetNumberOfEvents();
+ // TODO move it to rec.C. check TPC
+ if (!fgRecoParam) fgRecoParam = AliTRDrecoParam::GetLowFluxParam();
- 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();
+ AliTRDclusterizer clusterer("clusterer","TRD clusterizer");
+ clusterer.OpenOutput(clusterTree);
+ clusterer.ReadDigits(digitsTree);
+ clusterer.MakeClusters();
}
//_____________________________________________________________________________
-AliTracker *AliTRDReconstructor::CreateTracker(AliRunLoader *runLoader) const
+AliTracker *AliTRDReconstructor::CreateTracker() const
{
//
// Create a TRD tracker
//
- runLoader->CdGAFile();
+ //return new AliTRDtracker(NULL);
- return new AliTRDtracker(gFile);
+ // TODO move it to rec.C. check TPC
+ if (!fgRecoParam) fgRecoParam = AliTRDrecoParam::GetLowFluxParam();
+ return new AliTRDtrackerV1();
}
//_____________________________________________________________________________
-void AliTRDReconstructor::FillESD(AliRunLoader *runLoader
- , AliESD *esd) const
+void AliTRDReconstructor::FillESD(TTree* /*digitsTree*/
+ , TTree* /*clusterTree*/
+ , AliESDEvent* /*esd*/) const
{
//
- // Make PID
- //
-
- AliTRDpidESD trdPID;
- trdPID.MakePID(esd);
-
- //
- // Trigger (tracks, GTU)
+ // Fill ESD
//
- 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);
- 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);
-
- }
}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff