/* History of cvs commits:
*
* $Log$
+ * Revision 1.12 2006/03/10 13:23:36 hristov
+ * Using AliESDCaloCluster instead of AliESDtrack
+ *
+ * Revision 1.11 2006/01/31 20:30:52 hristov
+ * Including TFile.h
+ *
+ * Revision 1.10 2006/01/23 18:04:08 hristov
+ * Removing meaningless const
+ *
+ * Revision 1.9 2006/01/12 16:23:26 schutz
+ * ESD is properly read with methods of macros/AliReadESD.C copied in it
+ *
+ * Revision 1.8 2005/12/20 07:08:32 schutz
+ * corrected error in call AliReadESD
+ *
* Revision 1.6 2005/05/28 14:19:04 schutz
* Compilation warnings fixed by T.P.
*
// --- ROOT system ---
-#include "TParticle.h"
-#include "TCanvas.h"
-#include "TPaveLabel.h"
-#include "TPad.h"
+#include <TFile.h>
+#include <TParticle.h>
+#include <TCanvas.h>
+#include <TPaveLabel.h>
+#include <TPad.h>
+#include <TH2.h>
+
#include "AliPHOSGammaJet.h"
#include "AliPHOSGetter.h"
-#include "TH2.h"
#include "AliPHOSGeometry.h"
#include "AliPHOSFastGlobalReconstruction.h"
#include "AliESD.h"
#include "AliESDtrack.h"
+#include "AliESDCaloCluster.h"
#include "Riostream.h"
-#include "macros/AliReadESD.C"
-//#include "../PYTHIA6/AliGenPythia.h"
+
ClassImp(AliPHOSGammaJet)
TTask("GammaJet","Analysis of gamma-jet correlations")
{
// ctor
-// gSystem->Load("$ALICE_ROOT/PHOS/macros/AliReadESD_C.so");
fInputFileName = inputfilename;
fFastRec = new AliPHOSFastGlobalReconstruction(fInputFileName);
AliPHOSGetter * gime = AliPHOSGetter::Instance(fInputFileName) ;
TClonesArray * plNe,
TClonesArray * plNePHOS,
const AliESD * esd){
- //, Int_t iEvent){
+
+ //Create a list of particles from the ESD. These particles have been measured
+ //by the Central Tracking system (TPC), PHOS and EMCAL
+ //(EMCAL not available for the moment).
//Info("CreateParticleListFromESD","Inside");
- //AliPHOSGetter * gime = AliPHOSGetter::Instance(fInputFileName) ;
- //gime->Event(iEvent, "P") ;
- Double_t v[3] ; //vertex ;
- esd->GetVertex()->GetXYZ(v) ;
Int_t index = pl->GetEntries() ;
+ Int_t npar = 0 ;
+ Float_t *pid = new Float_t[AliPID::kSPECIESN];
+
+ //########### PHOS ##############
+ //Info("CreateParticleListFromESD","Fill ESD PHOS list");
+ Int_t begphos = esd->GetFirstPHOSCluster();
+ Int_t endphos = esd->GetFirstPHOSCluster() +
+ esd->GetNumberOfPHOSClusters() ;
+ Int_t indexNePHOS = plNePHOS->GetEntries() ;
+ if(strstr(fOptionGJ,"deb all"))
+ Info("CreateParticleListFromESD","PHOS: first particle %d, last particle %d",
+ begphos,endphos);
- //PHOS
- Int_t begphos = esd->GetFirstPHOSParticle();
- Int_t endphos = esd->GetFirstPHOSParticle() + esd->GetNumberOfPHOSParticles() ;
+ for (npar = begphos; npar < endphos; npar++) {//////////////PHOS track loop
+ AliESDCaloCluster * clus = esd->GetCaloCluster(npar) ; // retrieve track from esd
+
+ //Create a TParticle to fill the particle list
- Int_t indexNePHOS = plNePHOS->GetEntries() ;
- Int_t nphP ;
- //cout<<"Fill PHOS"<<endl;
- //cout<<"begin "<<begphos<<" end "<<endphos<<endl;
- for (nphP = begphos; nphP < endphos; nphP++) {//////////////PHOS track loop
- AliESDtrack * track = esd->GetTrack(nphP) ; // retrieve track from esd
-
- // PID of this track
- Double_t pid[AliPID::kSPECIESN];
- track->GetPHOSpid(pid);
- TParticle * particle = new TParticle() ;
- Double_t en = track->GetPHOSsignal() ;
- Double_t * p = new Double_t();
- track->GetPHOSposition(p) ;
+ Float_t en = clus->GetClusterEnergy() ;
+ Float_t *p = new Float_t();
+ clus->GetGlobalPosition(p) ;
TVector3 pos(p[0],p[1],p[2]) ;
Double_t phi = pos.Phi();
Double_t theta= pos.Theta();
Double_t px = en*TMath::Cos(phi)*TMath::Sin(theta);;
Double_t py = en*TMath::Sin(phi)*TMath::Sin(theta);
Double_t pz = en*TMath::Cos(theta);
+
+ TParticle * particle = new TParticle() ;
particle->SetMomentum(px,py,pz,en) ;
+
+ //Select only photons
+
+ pid=clus->GetPid();
//cout<<"pid "<<pid[AliPID::kPhoton]<<endl ;
if( pid[AliPID::kPhoton] > 0.75)
- new((*plNePHOS)[indexNePHOS++]) TParticle(*particle) ;
+ new((*plNePHOS)[indexNePHOS++]) TParticle(*particle) ;
}
- //TPC
+
+ //########### TPC #####################
+ //Info("CreateParticleListFromESD","Fill ESD TPC list");
Int_t begtpc = 0 ;
Int_t endtpc = esd->GetNumberOfTracks() ;
Int_t indexCh = plCh->GetEntries() ;
- //cout<<"Fill TPC"<<endl;
- //cout<<"begin "<<begtpc<<" end "<<endtpc<<endl;
- for (nphP = begtpc; nphP < endtpc; nphP++) {////////////// track loop
- AliESDtrack * track = esd->GetTrack(nphP) ; // retrieve track from esd
+ if(strstr(fOptionGJ,"deb all"))
+ Info("CreateParticleListFromESD","TPC: first particle %d, last particle %d",
+ begtpc,endtpc);
+
+ for (npar = begtpc; npar < endtpc; npar++) {////////////// track loop
+ AliESDtrack * track = esd->GetTrack(npar) ; // retrieve track from esd
- // PID of this track
- //Double_t pid[AliPID::kSPECIESN];
- //track->GetEMCALpid(pid);
+ Double_t en = track ->GetTPCsignal() ;
+ Double_t mom[3];
+ track->GetPxPyPz(mom) ;
+ Double_t px = mom[0];
+ Double_t py = mom[1];
+ Double_t pz = mom[2]; //Check with TPC people if this is correct.
+
+ //cout<<"TPC signal "<<en<<endl;
+ //cout<<"px "<<px<<"; py "<<py<<"; pz "<<pz<<endl;
TParticle * particle = new TParticle() ;
- Double_t en = track->GetTPCsignal() ;
- //Double_t *p = 0;
- //track->GetEMCALposition(p) ;
- //TVector3 pos(p[0],p[1],p[2]) ;
- //Double_t phi = pos.Phi();
- //Double_t theta= pos.Theta();
-
- TVector3 mom = track->P3() ;
- Double_t px = mom.Px();
- Double_t py = mom.Py();
- Double_t pz = mom.Pz();
-
particle->SetMomentum(px,py,pz,en) ;
- //if( pid[AliPID::kPhoton] > 0.75)
+
new((*plCh)[indexCh++]) TParticle(*particle) ;
new((*pl)[index++]) TParticle(*particle) ;
}
- //EMCAL
- Int_t begem = esd->GetFirstEMCALParticle();
- Int_t endem = esd->GetFirstEMCALParticle() + esd->GetNumberOfEMCALParticles() ;
- Int_t indexNe = plNe->GetEntries() ;
- //cout<<"Fill EMCAL"<<endl;
- //cout<<"begin "<<begem<<" end "<<endem<<endl;
- for (nphP = begem; nphP < endem; nphP++) {//////////////PHOS track loop
- AliESDtrack * track = esd->GetTrack(nphP) ; // retrieve track from esd
-
- // PID of this track
- Double_t pid[AliPID::kSPECIESN];
- track->GetEMCALpid(pid);
- Double_t en = track->GetEMCALsignal() ;
- //cout<<"EMCAL signal "<<en<<endl;
- Double_t *p = new Double_t();
- track->GetEMCALposition(p) ;
- TVector3 pos(p[0],p[1],p[2]) ;
+ //################ EMCAL ##############
+ Double_t v[3] ; //vertex ;
+ esd->GetVertex()->GetXYZ(v) ;
+ //##########Uncomment when ESD for EMCAL works ##########
+ //Info("CreateParticleListFromESD","Fill ESD EMCAL list");
+
+ Int_t begem = esd->GetFirstEMCALCluster();
+ Int_t endem = esd->GetFirstEMCALCluster() +
+ esd->GetNumberOfEMCALClusters() ;
+ Int_t indexNe = plNe->GetEntries() ;
+ if(strstr(fOptionGJ,"deb all"))
+ Info("CreateParticleListFromESD","EMCAL: first particle %d, last particle %d",
+ begem,endem);
+
+ for (npar = begem; npar < endem; npar++) {//////////////EMCAL track loop
+ AliESDCaloCluster * clus = esd->GetCaloCluster(npar) ; // retrieve track from esd
+
+ Float_t en = clus->GetClusterEnergy() ;
+ Float_t *p = new Float_t();
+ clus->GetGlobalPosition(p) ;
+ TVector3 pos(p[0],p[1],p[2]) ;
Double_t phi = pos.Phi();
Double_t theta= pos.Theta();
-
Double_t px = en*TMath::Cos(phi)*TMath::Sin(theta);;
Double_t py = en*TMath::Sin(phi)*TMath::Sin(theta);
Double_t pz = en*TMath::Cos(theta);
- Int_t pdg = 0;
+ //cout<<"EMCAL signal "<<en<<endl;
+ //cout<<"px "<<px<<"; py "<<py<<"; pz "<<pz<<endl;
+ //TParticle * particle = new TParticle() ;
//particle->SetMomentum(px,py,pz,en) ;
- if( pid[AliPID::kPhoton] > 0.75) //This has to be fixen.
- pdg = 22;
- //particle->SetPdgCode(22);
- else if( pid[AliPID::kPi0] > 0.75)
- pdg = 111;
- // particle->SetPdgCode(111);
- pdg = 22 ; //Assume all photons, no PID for EMCAL available.
+
+ Int_t pdg = 0;
+ // //Uncomment if PID IS WORKING, photon and pi0 idenitification.
+ // //if( pid[AliPID::kPhoton] > 0.75) //This has to be fixen.
+ // //pdg = 22;
+ // //else if( pid[AliPID::kPi0] > 0.75)
+ // //pdg = 111;
+ pdg = 22; //No PID, assume all photons
TParticle * particle = new TParticle(pdg, 1, -1, -1, -1, -1,
px, py, pz, en, v[0], v[1], v[2], 0);
- //cout<<"pid; ph "<<pid[AliPID::kPhoton]<<" pi0 "<<pid[AliPID::kPi0]<<endl;
+
new((*plNe)[indexNe++]) TParticle(*particle) ;
new((*pl)[index++]) TParticle(*particle) ;
}
- //Info("CreateParticleListFromESD","End Inside");
-
+
+ // Info("CreateParticleListFromESD","End Inside");
+
}
if(fESDdata){
// Create chain of esd trees
- const UInt_t kNevent = static_cast<const UInt_t>(GetNEvent()) ;
- t = AliReadESD(kNevent, fDirName, fESDTree, fPattern) ;
+ const UInt_t kNevent = static_cast<UInt_t>(GetNEvent()) ;
+ t = ReadESD(kNevent, fDirName, fESDTree, fPattern) ;
if(!t) {
AliError("Could not create the TChain") ;
//break ;
}
+//__________________________________________________________________________
+TChain * AliPHOSGammaJet::ReadESDfromdisk(const UInt_t eventsToRead,
+ const TString dirName,
+ const TString esdTreeName,
+ const char * pattern)
+{
+ // Reads ESDs from Disk
+ TChain * rv = 0;
+
+ AliInfo( Form("\nReading files in %s \nESD tree name is %s \nReading %d events",
+ dirName.Data(), esdTreeName.Data(), eventsToRead) ) ;
+
+ // create a TChain of all the files
+ TChain * cESDTree = new TChain(esdTreeName) ;
+
+ // read from the directory file until the require number of events are collected
+ void * from = gSystem->OpenDirectory(dirName) ;
+ if (!from) {
+ AliError( Form("Directory %s does not exist") ) ;
+ rv = 0 ;
+ }
+ else{ // reading file names from directory
+ const char * subdir ;
+ // search all subdirectories witch matching pattern
+ while( (subdir = gSystem->GetDirEntry(from)) &&
+ (cESDTree->GetEntries() < eventsToRead)) {
+ if ( strstr(subdir, pattern) != 0 ) {
+ char file[200] ;
+ sprintf(file, "%s%s/AliESDs.root", dirName.Data(), subdir);
+ AliInfo( Form("Adding %s\n", file) );
+ cESDTree->Add(file) ;
+ }
+ } // while file
+
+ AliInfo( Form(" %d events read", cESDTree->GetEntriesFast()) ) ;
+ rv = cESDTree ;
+
+ } // reading file names from directory
+ return rv ;
+}
+
+//__________________________________________________________________________
+TChain * AliPHOSGammaJet::ReadESD(const UInt_t eventsToRead,
+ const TString dirName,
+ const TString esdTreeName,
+ const char * pattern)
+{
+ // Read AliESDs files and return a Chain of events
+
+ if ( dirName == "" ) {
+ AliError("Give the name of the DIR where to find files") ;
+ return 0 ;
+ }
+ if ( esdTreeName == "" )
+ return ReadESDfromdisk(eventsToRead, dirName) ;
+ else if ( strcmp(pattern, "") == 0 )
+ return ReadESDfromdisk(eventsToRead, dirName, esdTreeName) ;
+ else
+ return ReadESDfromdisk(eventsToRead, dirName, esdTreeName, pattern) ;
+}
+
//___________________________________________________________________
void AliPHOSGammaJet::SetJet(TParticle * part, Bool_t & b, Float_t cone,
Double_t eta, Double_t phi)
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff