/* $Id$ */
//_________________________________________________________________________
-// Algorithm class for the reconstruction: clusterizer
-// track segment maker
-// particle identifier
//*--
-//*-- Author: Gines Martinez & Yves Schutz (SUBATECH)
-
+//*-- Author: Gines Martinez & Yves Schutz (SUBATECH)
+//*-- Compleetely redesigned by Dmitri Peressounko (SUBATECH & RRC KI) March 2001
+/////////////////////////////////////////////////////////////////////////////////////
+// Wrapping class for reconstruction. Allows to produce reconstruction from
+// different steps: from previously produced hits,sdigits, etc. Each new reconstruction
+// flow (e.g. digits, made from them RecPoints, subsequently made TrackSegments,
+// subsequently made RecParticles) are distinguished by the title of created branches. One can
+// use this title as a comment, see use case below.
+// Thanks to getters, one can set
+// parameters to reconstruction briks. The full set of parameters is saved in the
+// corresponding branch: e.g. parameters of clusterizer are stored in branch
+// TreeR::AliPHOSClusterizer with the same title as the branch containing the RecPoints. // TTree does not support overwriting, therefore one can not produce several
+// branches with the same names and titles - use different titles.
+//
+// Use case:
+//
+// root [0] AliPHOSReconstructioner * r = new AliPHOSReconstructioner("galice.root")
+// // Set the header file
+// root [1] r->ExecuteTask()
+// // Make full chain of reconstruction
+//
+// // One can specify the title for each branch
+// root [2] r->SetBranchFileName("RecPoints","RecPoints1") ;
+//
+// // One can change parameters of reconstruction algorithms
+// root [3] r->GetClusterizer()->SetEmcLocalMaxCut(0.02)
+//
+// // One can specify the starting point of the reconstruction and title of all
+// // branches produced in this pass
+// root [4] r->StartFrom("AliPHOSClusterizer","Local max cut 0.02")
+// // means that will use already generated Digits and produce only RecPoints,
+// // TS and RecParticles
+//
+// // And finally one can call ExecuteTask() with the following options
+// root [5] r->ExecuteTask("debug all timing")
+// // deb - prints the numbers of RecPoints, TrackSegments, RecParticles
+// // deb all - prints in addition list of RecPoints, TrackSegments, RecParticles
+// // timing - prints benchmarking results
+///////////////////////////////////////////////////////////////////////////////////////////////////
// --- ROOT system ---
-#include "TClonesArray.h"
-
// --- Standard library ---
-#include <iomanip.h>
-
// --- AliRoot header files ---
-
+#include "AliESD.h"
+#include "AliESDCaloTrack.h"
#include "AliPHOSReconstructioner.h"
-#include "AliPHOSClusterizer.h"
-#include "AliPHOSFastRecParticle.h"
-#include "AliPHOSCpvRecPoint.h"
+#include "AliPHOSClusterizerv1.h"
+#include "AliPHOSTrackSegmentMakerv1.h"
+#include "AliPHOSPIDv1.h"
+#include "AliPHOSGetter.h"
+
ClassImp(AliPHOSReconstructioner)
//____________________________________________________________________________
-AliPHOSReconstructioner::AliPHOSReconstructioner(AliPHOSClusterizer * Clusterizer,
- AliPHOSTrackSegmentMaker * Tracker,
- AliPHOSPID * Pid)
+ AliPHOSReconstructioner::AliPHOSReconstructioner():TTask("AliPHOSReconstructioner","")
{
// ctor
-
- fClusterizer = Clusterizer ;
- fTrackSegmentMaker = Tracker ;
- fPID = Pid ;
- fDebugReconstruction = kFALSE ;
-}
+ fClusterizer = 0 ;
+ fTSMaker = 0 ;
+ fPID = 0 ;
+ fFirstEvent = 0 ;
+ fLastEvent = -1 ;
+ fIsInitialized = kFALSE ;
+}
//____________________________________________________________________________
- void AliPHOSReconstructioner::Init(AliPHOSClusterizer * Clusterizer,
- AliPHOSTrackSegmentMaker * Tracker,
- AliPHOSPID * Pid)
+AliPHOSReconstructioner::AliPHOSReconstructioner(const char* evFoldName,const char * branchName,const TString taskName):
+TTask("AliPHOSReconstructioner",evFoldName)
{
- // Initialisation
+ // Create a PHOS reconstructioner for the tasks defined by taskName
+ // "C" - clusterization
+ // "T" - track segment making
+ // "P" - PID
- fClusterizer = Clusterizer ;
- fTrackSegmentMaker = Tracker ;
- fPID = Pid ;
- fDebugReconstruction = kFALSE ;
-}
+ AliPHOSGetter::Instance(evFoldName) ;
-//____________________________________________________________________________
- void AliPHOSReconstructioner::MakePPSD(DigitsList * dl,
- AliPHOSRecPoint::RecPointsList * emccl,
- AliPHOSRecPoint::RecPointsList * ppsdl,
- AliPHOSTrackSegment::TrackSegmentsList * trsl,
- AliPHOSRecParticle::RecParticlesList * rpl)
-{
- // Launches the Reconstruction process in the sequence: Make the reconstructed poins (clusterize)
- // Make the track segments
- // Make the reconstructed particles
- Int_t index ;
- if (fDebugReconstruction)
- cout << "\n\nDebugReconstruction>>> " << "Start making reconstructed points (clusterizing!!)" << endl;
+ if (taskName.Contains("C")) {
+ fRecPointBranch=branchName ;
+ fClusterizer = new AliPHOSClusterizerv1(evFoldName, GetTitle());
+ Add(fClusterizer);
+ }
- fClusterizer->MakeClusters(dl, emccl, ppsdl);
+ if (taskName.Contains("T")) {
+ fTSBranch=branchName ;
+ fTSMaker = new AliPHOSTrackSegmentMakerv1(evFoldName, GetTitle());
+ Add(fTSMaker) ;
+ }
- if (fDebugReconstruction){
- cout << "DebugReconstruction>>> " << "AliPHOSReconstructioner: Digit list entries is " << dl->GetEntries() << endl ;
- cout << "AliPHOSReconstructioner: Emc list entries is " << emccl->GetEntries() << endl ;
- cout << "AliPHOSReconstructioner: Ppsd list entries is " << ppsdl->GetEntries() << endl ;
+ if (taskName.Contains("P")) {
+ fRecPartBranch=branchName ;
+ fPID = new AliPHOSPIDv1(evFoldName, GetTitle());
+ Add(fPID);
}
+
+ fIsInitialized = kTRUE ;
+}
+//____________________________________________________________________________
+void AliPHOSReconstructioner::Exec(Option_t *opt)
+{
+ //check, if the names of branches, which should be made coincide with already
+ //existing
+ if (!opt)
+ return ;
+ if(!fIsInitialized)
+ Init() ;
+}
+//____________________________________________________________________________
+void AliPHOSReconstructioner:: Clusters2Tracks(Int_t ievent, AliESD *event)
+{
+ // Convert PHOS reconstructed particles into ESD object for event# ievent.
+ // ESD object is returned as an argument event
- // Digit Debuging
- if (fDebugReconstruction) {
- cout << ">>>>>>>>>>>>>>>>>>>>>> DebugReconstruction <<<<<<<<<<<<<<<<<<<<<<<<<<" << endl ;
- cout << "DebugReconstruction>>> Digit list entries is " << dl->GetEntries() << endl ;
- AliPHOSDigit * digit;
- Bool_t calorimeter ;
- Float_t factor;
- cout << "DebugReconstruction>>> Vol Id " <<
- " Ene(MeV, KeV) " <<
- " Index " <<
- " Nprim " <<
- " Primaries list " << endl;
- for (index = 0 ; index < dl->GetEntries() ; index++) {
- digit = (AliPHOSDigit * ) dl->At(index) ;
- calorimeter = fClusterizer->IsInEmc(digit);
- if (calorimeter) factor =1000. ; else factor=1000000.;
- cout << "DebugReconstruction>>> " <<
- setw(8) << digit->GetId() << " " <<
- setw(3) << (Int_t) calorimeter <<
- setw(10) << factor*fClusterizer->Calibrate(digit->GetAmp()) << " " <<
- setw(6) << digit->GetIndexInList() << " " <<
- setw(5) << digit->GetNprimary() <<" ";
- for (Int_t iprimary=0; iprimary<digit->GetNprimary(); iprimary++)
- cout << setw(5) << digit->GetPrimary(iprimary+1) << " ";
- cout << endl;
- }
-
- }
+ if(!fIsInitialized) Init() ;
- // Making Clusters
- if (fDebugReconstruction) cout << "DebugReconstruction>>> Start making reconstructed points (clusterizing)" << endl;
+ fClusterizer->SetEventRange(ievent,ievent);
+ fClusterizer->ExecuteTask();
- // mark the position of the RecPoints in the array
- AliPHOSEmcRecPoint * emcrp ;
- for (index = 0 ; index < emccl->GetEntries() ; index++) {
- emcrp = (AliPHOSEmcRecPoint * )emccl->At(index) ;
- emcrp->SetIndexInList(index) ;
- }
- AliPHOSPpsdRecPoint * ppsdrp ;
- for (index = 0 ; index < ppsdl->GetEntries() ; index++) {
- ppsdrp = (AliPHOSPpsdRecPoint * )ppsdl->At(index) ;
- ppsdrp->SetIndexInList(index) ;
- }
-
- if (fDebugReconstruction) {
- cout << "DebugReconstruction>>> Cluster emc list entries is " << emccl->GetEntries() << endl ;
- AliPHOSEmcRecPoint * recpoint;
- cout << "DebugReconstruction>>> Module " <<
- "Ene(MeV) " <<
- "Index " <<
- "Multi " <<
- " X " <<
- " Y " <<
- " Z " <<
- " Lambda 1 " <<
- " Lambda 2 " <<
- "MaxEnergy(MeV) " <<
- "Nprim " <<
- " Primaries list " << endl;
- for (index = 0 ; index < emccl->GetEntries() ; index++) {
- recpoint = (AliPHOSEmcRecPoint * )emccl->At(index) ;
- TVector3 locpos; recpoint->GetLocalPosition(locpos);
- Float_t lambda[2]; recpoint->GetElipsAxis(lambda);
- Int_t * primaries;
- Int_t nprimaries;
- primaries = recpoint->GetPrimaries(nprimaries);
- cout << "DebugReconstruction>>> " <<
- setw(2) <<recpoint->GetPHOSMod() << " " <<
- setw(9) << 1000.*recpoint->GetTotalEnergy() << " " <<
- setw(6) << recpoint->GetIndexInList() << " " <<
- setw(5) << recpoint->GetMultiplicity() <<" " <<
- setw(8) << locpos.X() <<" " <<
- setw(8) << locpos.Y() <<" " <<
- setw(8) << locpos.Z() << " " <<
- setw(10) << lambda[0] << " " <<
- setw(10) << lambda[1] << " " <<
- setw(9) << 1000*recpoint->GetMaximalEnergy() << " " <<
- setw(9) << nprimaries << " ";
- for (Int_t iprimary=0; iprimary<nprimaries; iprimary++)
- cout << setw(4) << primaries[iprimary] << " ";
- cout << endl;
- }
-
- cout << "DebugReconstruction>>> Cluster ppsd list entries is " << ppsdl->GetEntries() << endl ;
- AliPHOSPpsdRecPoint * ppsdrecpoint;
- Text_t detector[4];
- cout << "DebugReconstruction>>> Module " <<
- "Det " <<
- "Ene(KeV) " <<
- "Index " <<
- "Multi " <<
- " X " <<
- " Y " <<
- " Z " <<
- "Nprim " <<
- " Primaries list " << endl;
- for (index = 0 ; index < ppsdl->GetEntries() ; index++) {
- ppsdrecpoint = (AliPHOSPpsdRecPoint * ) ppsdl->At(index) ;
- TVector3 locpos; ppsdrecpoint->GetLocalPosition(locpos);
- Int_t * primaries;
- Int_t nprimaries;
- if (ppsdrecpoint->GetUp())
- strcpy(detector, "CPV");
- else
- strcpy(detector, "PC ");
- primaries = ppsdrecpoint->GetPrimaries(nprimaries);
- cout << "DebugReconstruction>>> " <<
- setw(4) << ppsdrecpoint->GetPHOSMod() << " " <<
- setw(4) << detector << " " <<
- setw(9) << 1000000.*ppsdrecpoint->GetTotalEnergy() << " " <<
- setw(6) << ppsdrecpoint->GetIndexInList() << " " <<
- setw(5) << ppsdrecpoint->GetMultiplicity() <<" " <<
- setw(8) << locpos.X() <<" " <<
- setw(8) << locpos.Y() <<" " <<
- setw(8) << locpos.Z() << " " <<
- setw(9) << nprimaries << " ";
- for (Int_t iprimary=0; iprimary<nprimaries; iprimary++)
- cout << setw(4) << primaries[iprimary] << " ";
- cout << endl;
- }
- }
-
+ fTSMaker ->SetEventRange(ievent,ievent);
+ fTSMaker ->ExecuteTask();
- if (fDebugReconstruction) cout << "DebugReconstruction>>>> Start making track segments(unfolding+tracksegments)" << endl;
- fTrackSegmentMaker->MakeTrackSegments(dl, emccl, ppsdl, trsl) ;
-
- // mark the position of the TrackSegments in the array
- AliPHOSTrackSegment * trs ;
- for (index = 0 ; index < trsl->GetEntries() ; index++) {
- trs = (AliPHOSTrackSegment * )trsl->At(index) ;
- trs->SetIndexInList(index) ;
- }
- if (fDebugReconstruction){
- cout << "DebugReconstruction>>> Track segment list entries is " << trsl->GetEntries() << endl ;
- cout << "DebugReconstruction>>> Module " <<
- "Ene(KeV) " <<
- "Index " <<
- " X " <<
- " Y " <<
- " Z " <<
- " rX " <<
- " rY " <<
- " rZ " <<
- "Nprim " <<
- " Primaries list " << endl;
-
- for (index = 0 ; index < trsl->GetEntries() ; index++) {
- trs = (AliPHOSTrackSegment * )trsl->At(index) ;
- TVector3 locpos; trs->GetPosition(locpos);
- Int_t * primaries;
- Int_t nprimaries;
- primaries = trs->GetPrimariesEmc(nprimaries);
- cout << "DebugReconstruction>>> " <<
- setw(4) << trs->GetPHOSMod() << " " <<
- setw(9) << 1000.*trs->GetEnergy() << " " <<
- setw(3) << trs->GetIndexInList() << " " <<
- setw(9) << locpos.X() <<" " <<
- setw(9) << locpos.Y() <<" " <<
- setw(9) << locpos.Z() << " " <<
- setw(10) << (trs->GetMomentumDirection()).X() << " " <<
- setw(10) << (trs->GetMomentumDirection()).Y() << " " <<
- setw(10) << (trs->GetMomentumDirection()).Z() << " " <<
- setw(4) << nprimaries << " ";
- for (Int_t iprimary=0; iprimary<nprimaries; iprimary++)
- cout << setw(4) << primaries[iprimary] << " ";
- cout << endl;
- }
-
+ fPID ->SetEventRange(ievent,ievent);
+ fPID ->ExecuteTask();
+
+ AliPHOSGetter *gime = AliPHOSGetter::Instance();
+ TClonesArray *recParticles = gime->RecParticles();
+ Int_t nOfRecParticles = recParticles->GetEntries();
+ for (Int_t recpart=0; recpart<nOfRecParticles; recpart++) {
+ AliESDCaloTrack *ct = new AliESDCaloTrack((AliPHOSRecParticle*)recParticles->At(recpart));
+ event->AddCaloTrack(ct);
}
- if (fDebugReconstruction) cout << "DebugReconstruction>>>> Start making reconstructed particles" << endl;
- fPID->MakeParticles(trsl, rpl) ;
-
- // mark the position of the RecParticles in the array
- AliPHOSRecParticle * rp ;
- for (index = 0 ; index < rpl->GetEntries() ; index++) {
- rp = (AliPHOSRecParticle * )rpl->At(index) ;
- rp->SetIndexInList(index) ;
- }
- //Debugger of RecParticles
- if (fDebugReconstruction){
- cout << "DebugReconstruction>>> Reconstructed particle list entries is " << rpl->GetEntries() << endl ;
- cout << "DebugReconstruction>>> Module " <<
- " PARTICLE " <<
- "Ene(KeV) " <<
- "Index " <<
- " X " <<
- " Y " <<
- " Z " <<
- "Nprim " <<
- " Primaries list " << endl;
- for (index = 0 ; index < rpl->GetEntries() ; index++) {
- rp = (AliPHOSRecParticle * ) rpl->At(index) ;
- TVector3 locpos; (rp->GetPHOSTrackSegment())->GetPosition(locpos);
- Int_t * primaries;
- Int_t nprimaries;
- Text_t particle[11];
- primaries = (rp->GetPHOSTrackSegment())->GetPrimariesEmc(nprimaries);
- switch(rp->GetType())
- {
- case AliPHOSFastRecParticle::kNEUTRALEM:
- strcpy( particle, "NEUTRAL_EM");
- break;
- case AliPHOSFastRecParticle::kNEUTRALHA:
- strcpy(particle, "NEUTRAL_HA");
- break;
- case AliPHOSFastRecParticle::kGAMMA:
- strcpy(particle, "GAMMA");
- break ;
- case AliPHOSFastRecParticle::kGAMMAHA:
- strcpy(particle, "GAMMA_H");
- break ;
- case AliPHOSFastRecParticle::kABSURDEM:
- strcpy(particle, "ABSURD_EM") ;
- break ;
- case AliPHOSFastRecParticle::kABSURDHA:
- strcpy(particle, "ABSURD_HA") ;
- break ;
- case AliPHOSFastRecParticle::kELECTRON:
- strcpy(particle, "ELECTRON") ;
- break ;
- case AliPHOSFastRecParticle::kCHARGEDHA:
- strcpy(particle, "CHARGED_HA") ;
- break ;
- }
-
- cout << "DebugReconstruction>>> " <<
- setw(4) << (rp->GetPHOSTrackSegment())->GetPHOSMod() << " " <<
- setw(15) << particle << " " <<
- setw(9) << 1000.*(rp->GetPHOSTrackSegment())->GetEnergy() << " " <<
- setw(3) << rp->GetIndexInList() << " " <<
- setw(9) << locpos.X() <<" " <<
- setw(9) << locpos.Y() <<" " <<
- setw(9) << locpos.Z() << " " <<
- setw(4) << nprimaries << " ";
- for (Int_t iprimary=0; iprimary<nprimaries; iprimary++)
- cout << setw(4) << primaries[iprimary] << " ";
- cout << endl;
- }
+}
+//____________________________________________________________________________
+ void AliPHOSReconstructioner::Init()
+{
+ // initiliaze Reconstructioner if necessary: we can not do this in default constructor
+
+ if(!fIsInitialized){
- }
+ fRecPointBranch="Default" ;
+ fClusterizer = new AliPHOSClusterizerv1(GetTitle(),fRecPointBranch.Data());
+ Add(fClusterizer) ;
+ fTSBranch="Default" ;
+ fTSMaker = new AliPHOSTrackSegmentMakerv1(GetTitle(),fTSBranch.Data());
+ Add(fTSMaker) ;
-}
+ fRecPartBranch="Default";
+ fPID = new AliPHOSPIDv1(GetTitle(),fRecPartBranch.Data()) ;
+ Add(fPID) ;
+
+ fIsInitialized = kTRUE ;
+
+ }
+}
//____________________________________________________________________________
- void AliPHOSReconstructioner::MakeCPV(DigitsList * dl,
- AliPHOSRecPoint::RecPointsList * emccl,
- AliPHOSRecPoint::RecPointsList * cpvcl)
+AliPHOSReconstructioner::~AliPHOSReconstructioner()
{
+ // Delete data members if any
+}
- // Launches the Reconstruction process of EMC and CPV in the sequence:
- // Make the reconstructed poins (clusterize)
- // Make the track segments
- // Particle identification is not made here
- // EMC and CPV rec.points are the same yet
- //
- // Yuri Kharlov. 20 October 2000
+void AliPHOSReconstructioner::Print()const {
+ // Print reconstructioner data
- Int_t index ;
+ TString message ;
+ message = "-----------------AliPHOSReconstructioner---------------\n" ;
+ message += " Reconstruction of the header file %s\n" ;
+ message += " with the following modules:\n" ;
- // Making Clusters
- if (fDebugReconstruction)
- cout << "DebugReconstruction>>> Start clusterizing reconstructed points" << endl;
- fClusterizer->MakeClusters(dl, emccl, cpvcl);
-
- if (fDebugReconstruction){
- // Digit Debuging
- cout << "AliPHOSReconstructioner: Digit list entries are " << dl->GetEntries() << endl ;
- cout << "AliPHOSReconstructioner: EMC list entries are " << emccl->GetEntries() << endl ;
- cout << "AliPHOSReconstructioner: CPV list entries are " << cpvcl->GetEntries() << endl ;
- cout << ">>>>>>>>>>>>>>>>>>>>>> DebugReconstruction <<<<<<<<<<<<<<<<<<<<<<<<<<" << endl ;
- cout << "DebugReconstruction>>> Digit list entries is " << dl->GetEntries() << endl ;
- AliPHOSDigit * digit;
- Bool_t calorimeter ;
- Float_t factor;
- cout << "DebugReconstruction>>> Vol Id " <<
- " Ene(MeV, KeV) " <<
- " Index " <<
- " Nprim " <<
- " Primaries list " << endl;
- for (index = 0 ; index < dl->GetEntries() ; index++) {
- digit = (AliPHOSDigit * ) dl->At(index) ;
- calorimeter = fClusterizer->IsInEmc(digit);
- if (calorimeter) factor =1000. ; else factor=1000000.;
- cout << "DebugReconstruction>>> " <<
- setw(8) << digit->GetId() << " " <<
- setw(3) << (Int_t) calorimeter <<
- setw(10) << factor*fClusterizer->Calibrate(digit->GetAmp()) << " " <<
- setw(6) << digit->GetIndexInList() << " " <<
- setw(5) << digit->GetNprimary() <<" ";
- for (Int_t iprimary=0; iprimary<digit->GetNprimary(); iprimary++)
- cout << setw(5) << digit->GetPrimary(iprimary+1) << " ";
- cout << endl;
- }
-
+ if(fClusterizer->IsActive()){
+ message += " (+) %s to branch %s\n" ;
}
- // mark the position of the RecPoints in the array
- AliPHOSEmcRecPoint * emcrp ;
- Int_t currentPHOSModule;
- for (index = 0 ; index < emccl->GetEntries() ; index++) {
- emcrp = (AliPHOSEmcRecPoint * )emccl->At(index) ;
- emcrp ->SetIndexInList(index) ;
- TVector3 locpos; emcrp->GetLocalPosition(locpos);
- currentPHOSModule = emcrp->GetPHOSMod();
+ if(fTSMaker->IsActive()){
+ message += " (+) %s to branch %s\n" ;
}
- AliPHOSCpvRecPoint * cpvrp ;
- for (index = 0 ; index < cpvcl->GetEntries() ; index++) {
- cpvrp = (AliPHOSCpvRecPoint * )cpvcl->At(index) ;
- if (cpvrp->IsCPV()) break;
- cpvrp ->SetIndexInList(index) ;
- TVector3 locpos; cpvrp->GetLocalPosition(locpos);
- Int_t lengX,lengZ; cpvrp->GetClusterLengths(lengX,lengZ);
- currentPHOSModule = cpvrp->GetPHOSMod();
+
+ if(fPID->IsActive()){
+ message += " (+) %s to branch %s\n" ;
}
-
- if (fDebugReconstruction)
- cout << "DebugReconstruction>>>> Start unfolding reconstructed points" << endl;
- fTrackSegmentMaker->MakeTrackSegmentsCPV(dl, emccl, cpvcl) ;
+ Info("Print", message.Data(),
+ GetTitle(),
+ fClusterizer->GetName(), fRecPointBranch.Data(),
+ fTSMaker->GetName(), fTSBranch.Data() ,
+ fPID->GetName(), fRecPartBranch.Data() ) ;
+}
+
+//____________________________________________________________________________
+void AliPHOSReconstructioner::SetEventRange(Int_t first, Int_t last)
+{
+ // Set the event range to process
+ fFirstEvent=first;
+ fLastEvent=last;
+ fClusterizer->SetEventRange(fFirstEvent, fLastEvent) ;
+ fTSMaker->SetEventRange(fFirstEvent, fLastEvent) ;
+ fPID->SetEventRange(fFirstEvent, fLastEvent) ;
}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff