/* $Id$ */
+/* History of cvs commits:
+ *
+ * $Log$
+ * Revision 1.14 2006/09/07 18:31:08 kharlov
+ * Effective c++ corrections (T.Pocheptsov)
+ *
+ * Revision 1.13 2005/05/28 14:19:04 schutz
+ * Compilation warnings fixed by T.P.
+ *
+ */
+
//_________________________________________________________________________
// Implementation version v0 of the PHOS particle identifier
// Particle identification based on the
// Completely redesined by Dmitri Peressounko, March 2001
// --- ROOT system ---
-#include "TROOT.h"
#include "TTree.h"
-#include "TFile.h"
#include "TF2.h"
#include "TFormula.h"
#include "TCanvas.h"
-#include "TFolder.h"
-#include "TSystem.h"
+
#include "TBenchmark.h"
// --- Standard library ---
-#include <iostream.h>
-#include <iomanip.h>
-
// --- AliRoot header files ---
-
+#include "AliLog.h"
#include "AliRun.h"
#include "AliGenerator.h"
-#include "AliPHOS.h"
#include "AliPHOSPIDv0.h"
-#include "AliPHOSClusterizerv1.h"
+#include "AliPHOSEmcRecPoint.h"
#include "AliPHOSTrackSegment.h"
-#include "AliPHOSTrackSegmentMakerv1.h"
#include "AliPHOSRecParticle.h"
#include "AliPHOSGeometry.h"
-#include "AliPHOSGetter.h"
+#include "AliPHOSLoader.h"
ClassImp( AliPHOSPIDv0)
//____________________________________________________________________________
-AliPHOSPIDv0::AliPHOSPIDv0():AliPHOSPID()
+AliPHOSPIDv0::AliPHOSPIDv0():
+ fTrackSegmentsTitle(""),
+ fRecPointsTitle(""),
+ fRecParticlesTitle(""),
+ fIDOptions("dis time"),
+ fNEvent(0),
+ fClusterizer(0),
+ fTSMaker(0),
+ fFormula(0),
+ fDispersion(0.f),
+ fCpvEmcDistance(0.f),
+ fTimeGate(2.e-9f),
+ fRecParticlesInRun(0)
{
// default ctor
- fFormula = 0 ;
- fDispersion = 0. ;
- fCpvEmcDistance = 0 ;
- fTimeGate = 2.e-9 ;
- fHeaderFileName = "" ;
- fTrackSegmentsTitle= "" ;
- fRecPointsTitle = "" ;
- fRecParticlesTitle = "" ;
- fIDOptions = "dis time" ;
- fRecParticlesInRun = 0 ;
- fClusterizer = 0;
- fTSMaker = 0;
+ fEventFolderName = "";
}
//____________________________________________________________________________
-AliPHOSPIDv0::AliPHOSPIDv0(const char * headerFile,const char * name) : AliPHOSPID(headerFile, name)
+AliPHOSPIDv0::AliPHOSPIDv0(const char * evFolderName,const char * name) :
+ AliPHOSPID(evFolderName, name),
+ fTrackSegmentsTitle(GetName()),
+ fRecPointsTitle(GetName()),
+ fRecParticlesTitle(GetName()),
+ fIDOptions("dis time"),
+ fNEvent(0),
+ fClusterizer(0),
+ fTSMaker(0),
+ fFormula(new TFormula("LambdaCuts","(x>1)*(x<2.5)*(y>0)*(y<x)")),
+ fDispersion(2.f),
+ fCpvEmcDistance(3.f),
+ fTimeGate(2.e-9f),
+ fRecParticlesInRun(0)
{
//ctor with the indication on where to look for the track segments
-
- fFormula = new TFormula("LambdaCuts","(x>1)*(x<2.5)*(y>0)*(y<x)") ;
- fDispersion = 2.0 ;
- fCpvEmcDistance = 3.0 ;
- fTimeGate = 2.e-9 ;
-
- fHeaderFileName = GetTitle() ;
- fTrackSegmentsTitle = GetName() ;
- fRecPointsTitle = GetName() ;
- fRecParticlesTitle = GetName() ;
- fIDOptions = "dis time" ;
-
- TString tempo(GetName()) ;
- tempo.Append(":") ;
- tempo.Append(Version()) ;
- SetName(tempo) ;
- fRecParticlesInRun = 0 ;
-
+ fEventFolderName = GetTitle() ;
Init() ;
-
}
//____________________________________________________________________________
-AliPHOSPIDv0::~AliPHOSPIDv0()
-{
+AliPHOSPIDv0::AliPHOSPIDv0(const AliPHOSPIDv0 & rhs) :
+ AliPHOSPID(rhs),
+ fTrackSegmentsTitle(rhs.fTrackSegmentsTitle),
+ fRecPointsTitle(rhs.fRecPointsTitle),
+ fRecParticlesTitle(rhs.fRecParticlesTitle),
+ fIDOptions(rhs.fIDOptions),
+ fNEvent(rhs.fNEvent),
+ fClusterizer(rhs.fClusterizer),
+ fTSMaker(rhs.fTSMaker),
+ fFormula(rhs.fFormula),
+ fDispersion(rhs.fDispersion),
+ fCpvEmcDistance(rhs.fCpvEmcDistance),
+ fTimeGate(rhs.fTimeGate),
+ fRecParticlesInRun(rhs.fRecParticlesInRun)
+{
+ //Copy ctor, the same as compiler-generated, possibly wrong if
+ //someone implements dtor correctly.
+}
+
+//____________________________________________________________________________
+AliPHOSPIDv0 & AliPHOSPIDv0::operator = (const AliPHOSPIDv0 & rhs)
+{
+ //Copy-assignment, emulates compiler generated, possibly wrong.
+ AliPHOSPID::operator = (rhs);
+ fTrackSegmentsTitle = rhs.fTrackSegmentsTitle;
+ fRecPointsTitle = rhs.fRecPointsTitle;
+ fRecParticlesTitle = rhs.fRecParticlesTitle;
+ fIDOptions = rhs.fIDOptions;
+ fNEvent = rhs.fNEvent;
+ fClusterizer = rhs.fClusterizer;
+ fTSMaker = rhs.fTSMaker;
+ fFormula = rhs.fFormula;
+ fDispersion = rhs.fDispersion;
+ fCpvEmcDistance = rhs.fCpvEmcDistance;
+ fTimeGate = rhs.fTimeGate;
+ fRecParticlesInRun = rhs.fRecParticlesInRun;
+
+ return *this;
}
-
//____________________________________________________________________________
-Float_t AliPHOSPIDv0::GetDistance(AliPHOSEmcRecPoint * emc,AliPHOSRecPoint * cpv, Option_t * Axis)const
+AliPHOSPIDv0::~AliPHOSPIDv0()
{
- // Calculates the distance between the EMC RecPoint and the PPSD RecPoint
-
- const AliPHOSGeometry * geom = AliPHOSGetter::GetInstance()->PHOSGeometry() ;
- TVector3 vecEmc ;
- TVector3 vecCpv ;
-
- emc->GetLocalPosition(vecEmc) ;
- cpv->GetLocalPosition(vecCpv) ;
- if(emc->GetPHOSMod() == cpv->GetPHOSMod()){
-
- // Correct to difference in CPV and EMC position due to different distance to center.
- // we assume, that particle moves from center
- Float_t dCPV = geom->GetIPtoOuterCoverDistance();
- Float_t dEMC = geom->GetIPtoCrystalSurface() ;
- dEMC = dEMC / dCPV ;
- vecCpv = dEMC * vecCpv - vecEmc ;
- if (Axis == "X") return vecCpv.X();
- if (Axis == "Y") return vecCpv.Y();
- if (Axis == "Z") return vecCpv.Z();
- if (Axis == "R") return vecCpv.Mag();
- }
-
- return 100000000 ;
+ //Empty dtor, fFormula leaks
}
+//DP
+////____________________________________________________________________________
+//Float_t AliPHOSPIDv0::GetDistance(AliPHOSEmcRecPoint * emc,AliPHOSRecPoint * cpv, Option_t * Axis)const
+//{
+// // Calculates the distance between the EMC RecPoint and the PPSD RecPoint
+//
+// const AliPHOSGeometry * geom = AliPHOSLoader::GetPHOSGeometry() ;
+// TVector3 vecEmc ;
+// TVector3 vecCpv ;
+//
+// emc->GetLocalPosition(vecEmc) ;
+// cpv->GetLocalPosition(vecCpv) ;
+// if(emc->GetPHOSMod() == cpv->GetPHOSMod()){
+//
+// // Correct to difference in CPV and EMC position due to different distance to center.
+// // we assume, that particle moves from center
+// Float_t dCPV = geom->GetIPtoOuterCoverDistance();
+// Float_t dEMC = geom->GetIPtoCrystalSurface() ;
+// dEMC = dEMC / dCPV ;
+// vecCpv = dEMC * vecCpv - vecEmc ;
+// if (Axis == "X") return vecCpv.X();
+// if (Axis == "Y") return vecCpv.Y();
+// if (Axis == "Z") return vecCpv.Z();
+// if (Axis == "R") return vecCpv.Mag();
+// }
+//
+// return 100000000 ;
+//}
+
//____________________________________________________________________________
void AliPHOSPIDv0::Exec(Option_t * option)
{
gBenchmark->Start("PHOSPID");
if(strstr(option,"print")) {
- Print("") ;
+ Print() ;
return ;
}
- gAlice->GetEvent(0) ;
- //check, if the branch with name of this" already exits?
- TObjArray * lob = (TObjArray*)gAlice->TreeR()->GetListOfBranches() ;
- TIter next(lob) ;
- TBranch * branch = 0 ;
- Bool_t phospidfound = kFALSE, pidfound = kFALSE ;
+ AliRunLoader* runget = AliRunLoader::GetRunLoader(GetTitle());
+ if(runget == 0x0)
+ {
+ AliError(Form("Can not find run getter in event folder \"%s\"",
+ GetTitle()));
+ return;
+ }
- TString taskName(GetName()) ;
- taskName.Remove(taskName.Index(Version())-1) ;
-
- while ( (branch = (TBranch*)next()) && (!phospidfound || !pidfound) ) {
- if ( (strcmp(branch->GetName(), "PHOSPID")==0) && (strcmp(branch->GetTitle(), taskName.Data())==0) )
- phospidfound = kTRUE ;
-
- else if ( (strcmp(branch->GetName(), "AliPHOSPID")==0) && (strcmp(branch->GetTitle(), taskName.Data())==0) )
- pidfound = kTRUE ;
- }
+ AliPHOSLoader* gime = dynamic_cast<AliPHOSLoader*>(runget->GetLoader("PHOSLoader"));
+ if ( gime == 0 )
+ {
+ AliError("Could not obtain the Loader object !");
+ return ;
+ }
+
+ if(gime->BranchExists("RecParticles") )
+ return ;
- if ( phospidfound || pidfound ) {
- cerr << "WARNING: AliPHOSPIDv0::Exec -> RecParticles and/or PIDtMaker branch with name "
- << taskName.Data() << " already exits" << endl ;
- return ;
- }
- Int_t nevents = (Int_t) gAlice->TreeE()->GetEntries() ;
+ Int_t nevents = runget->GetNumberOfEvents() ; //(Int_t) gAlice->TreeE()->GetEntries() ;
+
Int_t ievent ;
- AliPHOSGetter * gime = AliPHOSGetter::GetInstance() ;
for(ievent = 0; ievent < nevents; ievent++){
- gime->Event(ievent,"R") ;
-
+ runget->GetEvent(ievent);
+ AliInfo(Form("event %d %d %d",
+ ievent, gime->EmcRecPoints(),
+ gime->TrackSegments())) ;
MakeRecParticles() ;
- WriteRecParticles(ievent);
+ WriteRecParticles();
if(strstr(option,"deb"))
PrintRecParticles(option) ;
if(strstr(option,"tim")){
gBenchmark->Stop("PHOSPID");
- cout << "AliPHOSPID:" << endl ;
- cout << " took " << gBenchmark->GetCpuTime("PHOSPID") << " seconds for PID "
- << gBenchmark->GetCpuTime("PHOSPID")/nevents << " seconds per event " << endl ;
- cout << endl ;
+ AliInfo(Form("took %f seconds for PID %f seconds per event",
+ gBenchmark->GetCpuTime("PHOSPID"),
+ gBenchmark->GetCpuTime("PHOSPID")/nevents)) ;
}
}
// Make all memory allocations that are not possible in default constructor
// Add the PID task to the list of PHOS tasks
- if ( strcmp(GetTitle(), "") == 0 )
- SetTitle("galice.root") ;
-
- TString taskName(GetName()) ;
- taskName.Remove(taskName.Index(Version())-1) ;
+ AliRunLoader* runget = AliRunLoader::GetRunLoader(GetTitle());
+ if(runget == 0x0)
+ {
+ AliError(Form("Can not find run getter in event folder \"%s\"",
+ GetTitle()));
+ return;
+ }
- AliPHOSGetter * gime = AliPHOSGetter::GetInstance(GetTitle(), taskName.Data()) ;
- if ( gime == 0 ) {
- cerr << "ERROR: AliPHOSPIDv0::Init -> Could not obtain the Getter object !" << endl ;
- return ;
- }
+ AliPHOSLoader* gime = dynamic_cast<AliPHOSLoader*>(runget->GetLoader("PHOSLoader"));
+ if ( gime == 0 )
+ {
+ AliError("Could not obtain the Loader object !");
+ return ;
+ }
- gime->PostPID(this) ;
- // create a folder on the white board //YSAlice/WhiteBoard/RecParticles/PHOS/recparticlesName
- gime->PostRecParticles(taskName.Data() ) ;
+ gime->PostPID(this);
+ gime->LoadRecParticles("UPDATE");
}
//____________________________________________________________________________
-void AliPHOSPIDv0::MakeRecParticles(){
+void AliPHOSPIDv0::MakeRecParticles()
+{
+ // Reconstructs the particles from the tracksegments
- // Makes a RecParticle out of a TrackSegment
TString taskName(GetName()) ;
taskName.Remove(taskName.Index(Version())-1) ;
- AliPHOSGetter * gime = AliPHOSGetter::GetInstance() ;
- TObjArray * emcRecPoints = gime->EmcRecPoints(taskName) ;
- TObjArray * cpvRecPoints = gime->CpvRecPoints(taskName) ;
- TClonesArray * trackSegments = gime->TrackSegments(taskName) ;
- TClonesArray * recParticles = gime->RecParticles(taskName) ;
+ AliRunLoader* runget = AliRunLoader::GetRunLoader(GetTitle());
+ if(runget == 0x0)
+ {
+ AliError(Form("Can not find run getter in event folder \"%s\"",
+ GetTitle()));
+ return;
+ }
+
+ AliPHOSLoader* gime = dynamic_cast<AliPHOSLoader*>(runget->GetLoader("PHOSLoader"));
+ if ( gime == 0 )
+ {
+ AliError("Could not obtain the Loader object !");
+ return ;
+ }
+
+ TObjArray * emcRecPoints = gime->EmcRecPoints() ;
+ TObjArray * cpvRecPoints = gime->CpvRecPoints() ;
+ TClonesArray * trackSegments = gime->TrackSegments() ;
+ TClonesArray * recParticles = gime->RecParticles() ;
+
recParticles->Clear();
TIter next(trackSegments) ;
new( (*recParticles)[index] ) AliPHOSRecParticle() ;
rp = (AliPHOSRecParticle *)recParticles->At(index) ;
- rp->SetTraskSegment(index) ;
+ rp->SetTrackSegment(index) ;
rp->SetIndexInList(index) ;
AliPHOSEmcRecPoint * emc = 0 ;
// Looking at the CPV detector
Int_t cpvdetector= 0 ; //1 hit and 0 no hit
if(cpv)
- if(GetDistance(emc, cpv, "R") < fCpvEmcDistance)
+ if(ts->GetCpvDistance("R") < fCpvEmcDistance)
cpvdetector = 1 ;
Int_t type = showerprofile + 2 * slow + 4 * cpvdetector ;
}
//____________________________________________________________________________
-void AliPHOSPIDv0:: Print(Option_t * option) const
+void AliPHOSPIDv0:: Print(const Option_t *) const
{
// Print the parameters used for the particle type identification
- cout << "=============== AliPHOSPID1 ================" << endl ;
- cout << "Making PID "<< endl ;
- cout << " Headers file: " << fHeaderFileName.Data() << endl ;
- cout << " RecPoints branch title: " << fRecPointsTitle.Data() << endl ;
- cout << " TrackSegments Branch title: " << fTrackSegmentsTitle.Data() << endl ;
- cout << " RecParticles Branch title " << fRecParticlesTitle.Data() << endl;
- cout << "with parameters: " << endl ;
- cout << " Maximal EMC - CPV distance (cm) " << fCpvEmcDistance << endl ;
- if(fIDOptions.Contains("dis",TString::kIgnoreCase ))
- cout << " dispersion cut " << fDispersion << endl ;
- if(fIDOptions.Contains("ell",TString::kIgnoreCase )){
- cout << " Eliptic cuts function: " << endl ;
- cout << fFormula->GetTitle() << endl ;
- }
- if(fIDOptions.Contains("tim",TString::kIgnoreCase ))
- cout << " Time Gate uzed: " << fTimeGate << endl ;
- cout << "============================================" << endl ;
+ TString message ;
+ message = "=============== AliPHOSPIDv0 ================\n" ;
+ message += "Making PID\n" ;
+ message += " Headers file: %s\n" ;
+ message += " RecPoints branch title: %s\n" ;
+ message += " TrackSegments Branch title: %s\n" ;
+ message += " RecParticles Branch title %s\n" ;
+ message += "with parameters:\n" ;
+ message += " Maximal EMC - CPV distance (cm) %f\n" ;
+ AliInfo(Form( message.Data(),
+ GetTitle(),
+ fRecPointsTitle.Data(),
+ fTrackSegmentsTitle.Data(),
+ fRecParticlesTitle.Data(),
+ fCpvEmcDistance ));
+
+ if(fIDOptions.Contains("dis",TString::kIgnoreCase ))
+ AliInfo(Form(" dispersion cut %f", fDispersion )) ;
+ if(fIDOptions.Contains("ell",TString::kIgnoreCase ))
+ AliInfo(Form(" Eliptic cuts function: %s",
+ fFormula->GetTitle() )) ;
+ if(fIDOptions.Contains("tim",TString::kIgnoreCase ))
+ AliInfo(Form(" Time Gate used: %f", fTimeGate)) ;
}
//____________________________________________________________________________
-void AliPHOSPIDv0::SetShowerProfileCut(char * formula)
+void AliPHOSPIDv0::SetShowerProfileCut(const char * formula)
{
//set shape of the cut on the axis of ellipce, drown around shouer
//shower considered "narrow" if Formula(lambda[0],lambda[1]) > 0.
fFormula = new TFormula("Lambda Cut",formula) ;
}
//____________________________________________________________________________
-void AliPHOSPIDv0::WriteRecParticles(Int_t event)
+void AliPHOSPIDv0::WriteRecParticles()
{
+ // Saves the reconstructed particles too a file
- AliPHOSGetter *gime = AliPHOSGetter::GetInstance() ;
- TString taskName(GetName()) ;
- taskName.Remove(taskName.Index(Version())-1) ;
- TClonesArray * recParticles = gime->RecParticles(taskName) ;
+ AliRunLoader* runget = AliRunLoader::GetRunLoader(GetTitle());
+ if(runget == 0x0)
+ {
+ AliError(Form("Can not find run getter in event folder \"%s\"",
+ GetTitle()));
+ return;
+ }
+
+ AliPHOSLoader* gime = dynamic_cast<AliPHOSLoader*>(runget->GetLoader("PHOSLoader"));
+ if ( gime == 0 )
+ {
+ AliError("Could not obtain the Loader object !");
+ return ;
+ }
+
+ TClonesArray * recParticles = gime->RecParticles() ;
recParticles->Expand(recParticles->GetEntriesFast() ) ;
- //Make branch in TreeR for RecParticles
- char * filename = 0;
- if(gSystem->Getenv("CONFIG_SPLIT_FILE")!=0){ //generating file name
- filename = new char[strlen(gAlice->GetBaseFile())+20] ;
- sprintf(filename,"%s/PHOS.Reco.root",gAlice->GetBaseFile()) ;
- }
+ TTree * treeR = gime->TreeR();
- TDirectory *cwd = gDirectory;
+ if(!treeR){
+ gime->MakeTree("R");
+ treeR = gime->TreeR() ;
+ }
//First rp
Int_t bufferSize = 32000 ;
- TBranch * rpBranch = gAlice->TreeR()->Branch("PHOSRP",&recParticles,bufferSize);
+ TBranch * rpBranch = treeR->Branch("PHOSRP",&recParticles,bufferSize);
rpBranch->SetTitle(fRecParticlesTitle);
- if (filename) {
- rpBranch->SetFile(filename);
- TIter next( rpBranch->GetListOfBranches());
- TBranch * sb ;
- while ((sb=(TBranch*)next())) {
- sb->SetFile(filename);
- }
- cwd->cd();
- }
//second, pid
Int_t splitlevel = 0 ;
AliPHOSPIDv0 * pid = this ;
- TBranch * pidBranch = gAlice->TreeR()->Branch("AliPHOSPID","AliPHOSPIDv0",&pid,bufferSize,splitlevel);
+ TBranch * pidBranch = treeR->Branch("AliPHOSPID","AliPHOSPIDv0",&pid,bufferSize,splitlevel);
pidBranch->SetTitle(fRecParticlesTitle.Data());
- if (filename) {
- pidBranch->SetFile(filename);
- TIter next( pidBranch->GetListOfBranches());
- TBranch * sb ;
- while ((sb=(TBranch*)next())) {
- sb->SetFile(filename);
- }
- cwd->cd();
- }
rpBranch->Fill() ;
pidBranch->Fill() ;
-
- gAlice->TreeR()->Write(0,kOverwrite) ;
-
- delete [] filename ;
+
+ gime->WriteRecParticles("OVERWRITE");
+ gime->WritePID("OVERWRITE");
+
}
//____________________________________________________________________________
}
//____________________________________________________________________________
-TVector3 AliPHOSPIDv0::GetMomentumDirection(AliPHOSEmcRecPoint * emc, AliPHOSRecPoint * cpv)const
+TVector3 AliPHOSPIDv0::GetMomentumDirection(AliPHOSEmcRecPoint * emc, AliPHOSRecPoint * )const
{
// Calculates the momentum direction:
// 1. if only a EMC RecPoint, direction is given by IP and this RecPoint
{
// Print table of reconstructed particles
- AliPHOSGetter *gime = AliPHOSGetter::GetInstance() ;
+ AliRunLoader* runget = AliRunLoader::GetRunLoader(GetTitle());
+ if(runget == 0x0)
+ {
+ AliError(Form("Can not find run getter in event folder \"%s\"",
+ GetTitle()));
+ return;
+ }
+
+ AliPHOSLoader* gime = dynamic_cast<AliPHOSLoader*>(runget->GetLoader("PHOSLoader"));
+ if ( gime == 0 )
+ {
+ AliError("Could not obtain the Loader object !");
+ return ;
+ }
TString taskName(GetName()) ;
taskName.Remove(taskName.Index(Version())-1) ;
- TClonesArray * recParticles = gime->RecParticles(taskName) ;
-
- cout << "AliPHOSPIDv0: event "<<gAlice->GetEvNumber() << endl ;
- cout << " found " << recParticles->GetEntriesFast() << " RecParticles " << endl ;
+ TClonesArray * recParticles = gime->RecParticles() ;
+ TString message ;
+ message = "event %d\n" ;
+ message += " found %d RecParticles\n" ;
+ AliInfo(Form(message.Data(),
+ gAlice->GetEvNumber(), recParticles->GetEntriesFast() )) ;
+
if(strstr(option,"all")) { // printing found TS
-
- cout << " PARTICLE "
- << " Index " << endl ;
- // << " X "
- // << " Y "
- // << " Z "
- // << " # of primaries "
- // << " Primaries list " << endl;
-
+ AliInfo(" PARTICLE Index" ) ;
+
Int_t index ;
for (index = 0 ; index < recParticles->GetEntries() ; index++) {
AliPHOSRecParticle * rp = (AliPHOSRecParticle * ) recParticles->At(index) ;
// Int_t nprimaries;
// primaries = rp->GetPrimaries(nprimaries);
- cout << setw(10) << particle << " "
- << setw(5) << rp->GetIndexInList() << " " ;
- // << setw(4) << nprimaries << " ";
- // for (Int_t iprimary=0; iprimary<nprimaries; iprimary++)
- // cout << setw(4) << primaries[iprimary] << " ";
- cout << endl;
+ AliInfo(Form(" %s %d",
+ particle, rp->GetIndexInList())) ;
}
- cout << "-------------------------------------------" << endl ;
- }
-
+ }
}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff