//_________________________________________________________________________
// Implementation version v1 of the PHOS particle identifier
-// Identification is based on information from PPSD and EMC
+// Particle identification based on the
+// - CPV information,
+// - Preshower information (in MIXT or GPS2 geometries)
+// - shower width.
+//
+// CPV or Preshower clusters should be closer in PHOS plane than fCpvEmcDistance (in cm).
+// This parameter can be set by method SetCpvtoEmcDistanceCut(Float_t cut)
+//
+// One can set desirable ID method by the function SetIdentificationMethod(option).
+// Presently the following options can be used together or separately :
+// - "disp": use dispersion cut on shower width
+// (width can be set by method SetDispersionCut(Float_t cut)
+// - "ell" : use cut on the axis of the ellipse, drawn around shower
+// (this cut can be changed by SetShowerProfileCut(char* formula),
+// where formula - any function of two variables f(lambda[0],lambda[1]).
+// Shower is considered as EM if f() > 0 )
+// One can visualize current cuts calling method PlotDispersionCuts().
+//
+// use case:
+// root [0] AliPHOSPIDv1 * p1 = new AliPHOSPIDv1("galice.root")
+// Warning in <TDatabasePDG::TDatabasePDG>: object already instantiated
+// root [1] p1->SetIdentificationMethod("disp ellipse")
+// root [2] p1->ExecuteTask()
+// root [3] AliPHOSPIDv1 * p2 = new AliPHOSPIDv1("galice1.root","ts1")
+// Warning in <TDatabasePDG::TDatabasePDG>: object already instantiated
+// // reading headers from file galice1.root and TrackSegments
+// // with title "ts1"
+// root [4] p2->SetRecParticlesBranch("rp1")
+// // set file name for the branch RecParticles
+// root [5] p2->ExecuteTask("deb all time")
+// // available options
+// // "deb" - prints # of reconstructed particles
+// // "deb all" - prints # and list of RecParticles
+// // "time" - prints benchmarking results
//
-//*-- Author: Yves Schutz (SUBATECH) & Gines Martinez (SUBATECH)
-
+//*-- Author: Yves Schutz (SUBATECH) & Gines Martinez (SUBATECH) &
+// Dmitri Peressounko (SUBATECH & Kurchatov Institute)
+// Complitely 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 "AliRun.h"
+#include "AliGenerator.h"
+#include "AliPHOS.h"
#include "AliPHOSPIDv1.h"
+#include "AliPHOSClusterizerv1.h"
#include "AliPHOSTrackSegment.h"
+#include "AliPHOSTrackSegmentMakerv1.h"
#include "AliPHOSRecParticle.h"
-#include "AliPHOSIndexToObject.h"
+#include "AliPHOSGeometry.h"
+#include "AliPHOSGetter.h"
ClassImp( AliPHOSPIDv1)
//____________________________________________________________________________
-Float_t AliPHOSPIDv1::GetDistanceInPHOSPlane(AliPHOSEmcRecPoint * emcclu,AliPHOSPpsdRecPoint * PpsdClu, Bool_t &toofar, Option_t * Axis)
+AliPHOSPIDv1::AliPHOSPIDv1():AliPHOSPID()
+{
+ // default ctor
+ fFormula = 0 ;
+ fDispersion = 0. ;
+ fCpvEmcDistance = 0 ;
+ fHeaderFileName = "" ;
+ fTrackSegmentsTitle= "" ;
+ fRecPointsTitle = "" ;
+ fRecParticlesTitle = "" ;
+ fIDOptions = "" ;
+ fRecParticlesInRun = 0 ;
+}
+
+//____________________________________________________________________________
+AliPHOSPIDv1::AliPHOSPIDv1(const char * headerFile,const char * name) : AliPHOSPID(headerFile, name)
+{
+ //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 ;
+
+ fHeaderFileName = GetTitle() ;
+ fTrackSegmentsTitle = GetName() ;
+ fRecPointsTitle = GetName() ;
+ fRecParticlesTitle = GetName() ;
+ fIDOptions = "" ;
+
+ TString tempo(GetName()) ;
+ tempo.Append(":") ;
+ tempo.Append(Version()) ;
+ SetName(tempo) ;
+ fRecParticlesInRun = 0 ;
+
+ Init() ;
+
+}
+
+//____________________________________________________________________________
+AliPHOSPIDv1::~AliPHOSPIDv1()
+{
+ //dtor
+}
+
+//____________________________________________________________________________
+Bool_t AliPHOSPIDv1::ReadTrackSegments(Int_t event)
{
- // Calculates the distance between the EMC RecPoint and the PPSD RecPoint
+ // Reads TrackSegments an extracts the title of the RecPoints
+ // branch from which TS were made of.
+ // Then reads both TrackSegments and RecPoints.
+
+ //Fist read Track Segment Branch and extract RecPointsBranch from fTSMaker
+
+ // Get TreeR header from file
+ if(gAlice->TreeR()==0){
+ cerr << "ERROR: AliPHOSPIDv1::ReadTrackSegments -> There is no Reconstruction Tree" << endl;
+ return kFALSE;
+ }
- Float_t r;
- TVector3 vecEmc ;
- TVector3 vecPpsd ;
-
- emcclu->GetLocalPosition(vecEmc) ;
- PpsdClu->GetLocalPosition(vecPpsd) ;
- if(emcclu->GetPHOSMod() == PpsdClu->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 = fGeom->GetIPtoOuterCoverDistance();
- Float_t dEMC = fGeom->GetIPtoCrystalSurface() ;
- dEMC = dEMC / dCPV ;
- vecPpsd = dEMC * vecPpsd - vecEmc ;
- r = vecPpsd.Mag() ;
- if (Axis == "X") r = vecPpsd.X();
- if (Axis == "Y") r = vecPpsd.Y();
- if (Axis == "Z") r = vecPpsd.Z();
- if (Axis == "R") r = vecPpsd.Mag();
+ // Find TrackSegments
+ TBranch * tsbranch = 0;
+ TBranch * tsmakerbranch = 0;
+ TObjArray * lob = (TObjArray*)gAlice->TreeR()->GetListOfBranches() ;
+ TIter next(lob) ;
+ TBranch * branch = 0 ;
+ Bool_t phostsfound = kFALSE, tsmakerfound = kFALSE ;
+
+ TString taskName(GetName()) ;
+ taskName.ReplaceAll(Version(), "") ;
+
+ while ( (branch = (TBranch*)next()) && (!phostsfound || !tsmakerfound) ) {
+ if ( (strcmp(branch->GetName(), "PHOSTS")==0) && (strcmp(branch->GetTitle(), taskName.Data())==0) ) {
+ phostsfound = kTRUE ;
+ tsbranch = branch ;
+
+ } else if ( (strcmp(branch->GetName(), "AliPHOSTrackSegmentMaker")==0) && (strcmp(branch->GetTitle(), taskName.Data())==0) ) {
+ tsmakerfound = kTRUE ;
+ tsmakerbranch = branch ;
+ }
+ }
+ if ( !phostsfound || !tsmakerfound ) {
+ cerr << "WARNING: AliPHOSPIDv1::ReadTrackSegments -> TrackSegments and/or TrackSegmentMaker branch with name " << taskName.Data()
+ << " not found" << endl ;
+ return kFALSE ;
+ }
+
+ AliPHOSGetter * gime = AliPHOSGetter::GetInstance() ;
+
+ TClonesArray * trackSegments = gime->TrackSegments() ;
+ trackSegments->Clear() ;
+ tsbranch->SetAddress(&trackSegments) ;
+
+ AliPHOSTrackSegmentMaker * tsmaker = 0 ;
+ tsmakerbranch->SetAddress(&tsmaker) ;
+ tsmakerbranch->GetEntry(0) ;
+ TString tsmakerName( fRecParticlesTitle ) ;
+ tsmakerName.Append(tsmaker->Version()) ;
+ tsmaker = gime->TrackSegmentMaker(tsmakerName) ;
+
+ tsbranch->GetEntry(0) ;
+ tsmakerbranch->GetEntry(0) ;
+
+ fRecPointsTitle = tsmaker->GetRecPointsBranch() ;
+
+ // Find RecPoints
+ TBranch * emcbranch = 0;
+ TBranch * cpvbranch = 0;
+ TBranch * clusterizerbranch = 0;
+ lob = (TObjArray*)gAlice->TreeR()->GetListOfBranches() ;
+ TIter next2(lob) ;
+ branch = 0 ;
+ Bool_t phosemcfound = kFALSE, phoscpvfound = kFALSE, clusterizerfound = kFALSE ;
+
+ while ( (branch = (TBranch*)next2()) && (!phosemcfound || !phoscpvfound || !clusterizerfound) ) {
+ if ( (strcmp(branch->GetName(), "PHOSEmcRP")==0) && (strcmp(branch->GetTitle(), taskName.Data())==0) ) {
+ phosemcfound = kTRUE ;
+ emcbranch = branch ;
+ }
+
+ else if ( (strcmp(branch->GetName(), "PHOSCpvRP")==0) && (strcmp(branch->GetTitle(), taskName.Data())==0) ) {
+ phoscpvfound = kTRUE ;
+ cpvbranch = branch ;
- }
- else
- {
- toofar = kTRUE ;
+ } else if ( (strcmp(branch->GetName(), "AliPHOSClusterizer")==0) && (strcmp(branch->GetTitle(), taskName.Data())==0) ) {
+ clusterizerfound = kTRUE ;
+ clusterizerbranch = branch ;
}
- return r ;
+ }
+ if ( !phosemcfound || !phoscpvfound || !clusterizerfound ) {
+ cerr << "WARNING: AliPHOSTrackPIDv1::ReadTrackSegments -> emc(cpv)RecPoints and/or Clusterizer branch with name " << taskName.Data()
+ << " not found" << endl ;
+ return kFALSE ;
+ }
+
+ TObjArray * emcRecPoints = gime->EmcRecPoints() ;
+ emcRecPoints->Clear() ;
+ emcbranch->SetAddress(&emcRecPoints) ;
+
+ TObjArray * cpvRecPoints = gime->CpvRecPoints() ;
+ cpvRecPoints->Clear() ;
+ cpvbranch->SetAddress(&cpvRecPoints) ;
+
+
+ AliPHOSClusterizer * clusterizer = 0 ;
+ clusterizerbranch->SetAddress(&clusterizer) ;
+ clusterizerbranch->GetEntry(0) ;
+ TString clusterizerName( fTrackSegmentsTitle ) ;
+ clusterizerName.Append(clusterizer->Version()) ;
+ clusterizer = gime->Clusterizer(clusterizerName) ;
+
+ emcbranch->GetEntry(0) ;
+ cpvbranch->GetEntry(0) ;
+ clusterizerbranch->GetEntry(0) ;
+
+ return kTRUE ;
+
+}
+
+//____________________________________________________________________________
+Float_t AliPHOSPIDv1::GetDistance(AliPHOSEmcRecPoint * emc,AliPHOSRecPoint * cpv, Option_t * Axis)const
+{
+ // 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 ;
}
+//____________________________________________________________________________
+void AliPHOSPIDv1::Exec(Option_t * option)
+{
+ //Steering method
+
+ if( strcmp(GetName(), "")== 0 )
+ Init() ;
+
+ if(strstr(option,"tim"))
+ gBenchmark->Start("PHOSPID");
+
+ if(strstr(option,"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 ;
+
+ 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 ;
+ }
+
+ if ( phospidfound || pidfound ) {
+ cerr << "WARNING: AliPHOSPIDv1::Exec -> RecParticles and/or PIDtMaker branch with name "
+ << taskName.Data() << " already exits" << endl ;
+ return ;
+ }
+
+ Int_t nevents = (Int_t) gAlice->TreeE()->GetEntries() ;
+ Int_t ievent ;
+ AliPHOSGetter * gime = AliPHOSGetter::GetInstance() ;
+
+ for(ievent = 0; ievent < nevents; ievent++){
+
+ gime->Event(ievent,"R") ;
+ // if(!ReadTrackSegments(ievent)) //reads TrackSegments for event ievent
+ // continue ;
+ MakeRecParticles() ;
+ WriteRecParticles(ievent);
+
+ 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 ;
+ }
+
+}
//____________________________________________________________________________
-void AliPHOSPIDv1::MakeParticles(AliPHOSTrackSegment::TrackSegmentsList * trsl,
- AliPHOSRecParticle::RecParticlesList * rpl)
+void AliPHOSPIDv1::Init()
{
+ // 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) ;
+
+ AliPHOSGetter * gime = AliPHOSGetter::GetInstance(GetTitle(), taskName.Data()) ;
+ if ( gime == 0 ) {
+ cerr << "ERROR: AliPHOSPIDv1::Init -> Could not obtain the Getter object !" << endl ;
+ return ;
+ }
+
+ //add Task to //YSAlice/tasks/Reconstructioner/PHOS
+ // TTask * aliceRe = (TTask*)gROOT->FindObjectAny("YSAlice/tasks/Reconstructioner") ;
+ // TTask * phosRe = (TTask*)aliceRe->GetListOfTasks()->FindObject("PHOS") ;
+ // phosRe->Add(this) ;
+ gime->PostPID(this) ;
+ // create a folder on the white board //YSAlice/WhiteBoard/RecParticles/PHOS/recparticlesName
+ gime->PostRecParticles(taskName.Data() ) ;
+
+}
+
+//____________________________________________________________________________
+void AliPHOSPIDv1::MakeRecParticles(){
+
// Makes a RecParticle out of a TrackSegment
+ TString taskName(GetName()) ;
+ taskName.Remove(taskName.Index(Version())-1) ;
- TIter next(trsl) ;
- AliPHOSTrackSegment * tracksegment ;
+ AliPHOSGetter * gime = AliPHOSGetter::GetInstance() ;
+ TObjArray * emcRecPoints = gime->EmcRecPoints(taskName) ;
+ TObjArray * cpvRecPoints = gime->CpvRecPoints(taskName) ;
+ TClonesArray * trackSegments = gime->TrackSegments(taskName) ;
+ TClonesArray * recParticles = gime->RecParticles(taskName) ;
+ recParticles->Clear();
+
+ TIter next(trackSegments) ;
+ AliPHOSTrackSegment * ts ;
Int_t index = 0 ;
AliPHOSRecParticle * rp ;
- Bool_t tDistance;
- Int_t type ;
- Int_t showerprofile; // 0 narrow and 1 wide
- Int_t cpvdetector; // 1 hit and 0 no hit
- Int_t pcdetector; // 1 hit and 0 no hit
-
- while ( (tracksegment = (AliPHOSTrackSegment *)next()) ) {
- new( (*rpl)[index] ) AliPHOSRecParticle(tracksegment) ;
- rp = (AliPHOSRecParticle *)rpl->At(index) ;
- AliPHOSEmcRecPoint * recp = tracksegment->GetEmcRecPoint() ;
- AliPHOSPpsdRecPoint * rpcpv = tracksegment->GetPpsdUpRecPoint() ;
- AliPHOSPpsdRecPoint * rppc = tracksegment->GetPpsdLowRecPoint() ;
-
-// Float_t * lambda = new Float_t[2];
-// recp->GetElipsAxis(lambda) ;
-
-// // Looking at the lateral development of the shower
-// if ( ( lambda[0] > fLambda1m && lambda[0] < fLambda1M ) && // shower profile cut
-// ( lambda[1] > fLambda2m && lambda[1] < fLambda2M ) )
-// // Float_t R ;
-// //R=(lambda[0]-1.386)*(lambda[0]-1.386)+1.707*1.707*(lambda[1]-1.008)*(lambda[1]-1.008) ;
-// //if(R<0.35*0.35)
-
- Float_t Dispersion;
- Dispersion = recp->GetDispersion();
- if (Dispersion < fCutOnDispersion)
- showerprofile = 0 ; // NARROW PROFILE
- else
- showerprofile = 1 ;// WIDE PROFILE
- // Looking at the photon conversion detector
- if( tracksegment->GetPpsdLowRecPoint() == 0 )
- pcdetector = 0 ; // No hit
- else
- if (GetDistanceInPHOSPlane(recp, rppc, tDistance, "R")< fCutOnRelativeDistance) pcdetector = 1 ; // hit
+ Bool_t ellips = fIDOptions.Contains("ell",TString::kIgnoreCase ) ;
+ Bool_t disp = fIDOptions.Contains("dis",TString::kIgnoreCase ) ;
+ while ( (ts = (AliPHOSTrackSegment *)next()) ) {
+
+ new( (*recParticles)[index] ) AliPHOSRecParticle() ;
+ rp = (AliPHOSRecParticle *)recParticles->At(index) ;
+ rp->SetTraskSegment(index) ;
+
+ AliPHOSEmcRecPoint * emc = 0 ;
+ if(ts->GetEmcIndex()>=0)
+ emc = (AliPHOSEmcRecPoint *) emcRecPoints->At(ts->GetEmcIndex()) ;
+
+ AliPHOSRecPoint * cpv = 0 ;
+ if(ts->GetCpvIndex()>=0)
+ cpv = (AliPHOSRecPoint *) cpvRecPoints->At(ts->GetCpvIndex()) ;
+
+ AliPHOSRecPoint * ppsd = 0 ;
+ if(ts->GetPpsdIndex()>=0)
+ ppsd= (AliPHOSRecPoint *) cpvRecPoints->At(ts->GetPpsdIndex()) ;
+
+ //set momentum and energy first
+ Float_t e = emc->GetEnergy() ;
+ TVector3 dir = GetMomentumDirection(emc,cpv,ppsd) ;
+ dir.SetMag(e) ;
+
+ rp->SetMomentum(dir.X(),dir.Y(),dir.Z(),e) ;
+ rp->SetCalcMass(0);
+
+ //now set type (reconstructed) of the particle
+ Int_t showerprofile = 0; // 0 narrow and 1 wide
+
+ if(ellips){
+ Float_t lambda[2] ;
+ emc->GetElipsAxis(lambda) ;
+ if(fFormula->Eval(lambda[0],lambda[1]) <= 0 )
+ showerprofile = 1 ; // not narrow
+ }
+
+ if(disp)
+ if(emc->GetDispersion() > fDispersion )
+ showerprofile = 1 ; // not narrow
+
+
// Looking at the photon conversion detector
- if( tracksegment->GetPpsdUpRecPoint() == 0 )
- cpvdetector = 0 ; // No hit
- else
- if (GetDistanceInPHOSPlane(recp, rpcpv, tDistance, "R")< fCutOnRelativeDistance) cpvdetector = 1 ; // Hit
-
- type = showerprofile + 2 * pcdetector + 4 * cpvdetector ;
- rp->SetType(type) ;
+ Int_t pcdetector= 0 ; //1 hit and 0 no hit
+ if(ppsd)
+ if(GetDistance(emc, ppsd, "R") < fCpvEmcDistance)
+ pcdetector = 1 ;
+
+ // Looking at the CPV detector
+ Int_t cpvdetector= 0 ; //1 hit and 0 no hit
+ if(cpv)
+ if(GetDistance(emc, cpv, "R") < fCpvEmcDistance)
+ cpvdetector = 1 ;
+
+ Int_t type = showerprofile + 2 * pcdetector + 4 * cpvdetector ;
+ rp->SetType(type) ;
+ rp->SetIndexInList(index) ;
index++ ;
}
-
+
}
//____________________________________________________________________________
-void AliPHOSPIDv1:: Print(const char * opt)
+void AliPHOSPIDv1:: Print(Option_t * option) 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 (PPSD) 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 ;
+ }
+ cout << "============================================" << endl ;
+}
+
+//____________________________________________________________________________
+void AliPHOSPIDv1::SetShowerProfileCut(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.
+ if(fFormula)
+ delete fFormula;
+ fFormula = new TFormula("Lambda Cut",formula) ;
+}
+//____________________________________________________________________________
+void AliPHOSPIDv1::WriteRecParticles(Int_t event)
+{
+
+ AliPHOSGetter *gime = AliPHOSGetter::GetInstance() ;
+ TString taskName(GetName()) ;
+ taskName.Remove(taskName.Index(Version())-1) ;
+ TClonesArray * recParticles = gime->RecParticles(taskName) ;
+ recParticles->Expand(recParticles->GetEntriesFast() ) ;
- cout << "AliPHOSPIDv1 : cuts for the particle idendification based on the shower profile " << endl
- << fLambda1m << " < value1 < " << fLambda1M << endl
- << fLambda2m << " < value2 < " << fLambda2M << endl ;
+ //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()) ;
+ }
+
+ TDirectory *cwd = gDirectory;
+
+ //First rp
+ Int_t bufferSize = 32000 ;
+ TBranch * rpBranch = gAlice->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 ;
+ AliPHOSPIDv1 * pid = this ;
+ TBranch * pidBranch = gAlice->TreeR()->Branch("AliPHOSPID","AliPHOSPIDv1",&pid,bufferSize,splitlevel);
+ pidBranch->SetTitle(fRecParticlesTitle);
+ 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) ;
+
}
//____________________________________________________________________________
-void AliPHOSPIDv1::SetShowerProfileCuts(Float_t l1m, Float_t l1M, Float_t l2m, Float_t l2M)
+void AliPHOSPIDv1::PlotDispersionCuts()const
{
- // Modifies the parameters used for the particle type identification
-
- fLambda1m = l1m ;
- fLambda1M = l1M ;
- fLambda2m = l2m ;
- fLambda2M = l2M ;
+ // produces a plot of the dispersion cut
+ TCanvas* lambdas = new TCanvas("lambdas","Cuts on the ellipse axis",200,10,700,500);
+
+ if(fIDOptions.Contains("ell",TString::kIgnoreCase ) ){
+ TF2 * ell = new TF2("Elliptic Cuts",fFormula->GetName(),0,3,0,3) ;
+ ell->SetMinimum(0.0000001) ;
+ ell->SetMaximum(0.001) ;
+ ell->SetLineStyle(1) ;
+ ell->SetLineWidth(2) ;
+ ell->Draw() ;
+ }
+
+ if( fIDOptions.Contains("dis",TString::kIgnoreCase ) ){
+ TF2 * dsp = new TF2("dispersion","(y<x)*(x*x+y*y < [0]*[0])",0,3,0,3) ;
+ dsp->SetParameter(0,fDispersion) ;
+ dsp->SetMinimum(0.0000001) ;
+ dsp->SetMaximum(0.001) ;
+ dsp->SetLineStyle(1) ;
+ dsp->SetLineColor(2) ;
+ dsp->SetLineWidth(2) ;
+ dsp->SetNpx(200) ;
+ dsp->SetNpy(200) ;
+ if(fIDOptions.Contains("ell",TString::kIgnoreCase ) )
+ dsp->Draw("same") ;
+ else
+ dsp->Draw() ;
+ }
+ lambdas->Update();
}
//____________________________________________________________________________
-void AliPHOSPIDv1::SetRelativeDistanceCut(Float_t CutOnRelativeDistance)
+TVector3 AliPHOSPIDv1::GetMomentumDirection(AliPHOSEmcRecPoint * emc, AliPHOSRecPoint * cpv,AliPHOSRecPoint * ppsd)const
+{
+ // Calculates the momentum direction:
+ // 1. if only a EMC RecPoint, direction is given by IP and this RecPoint
+ // 2. if a EMC RecPoint and one PPSD RecPoint, direction is given by the line through the 2 recpoints
+ // 3. if a EMC RecPoint and two PPSD RecPoints, dirrection is given by the average line through
+ // the 2 pairs of recpoints
+ // However because of the poor position resolution of PPSD the direction is always taken as if we were
+ // in case 1.
+
+ TVector3 dir(0,0,0) ;
+
+ TVector3 emcglobalpos ;
+ TMatrix dummy ;
+
+ emc->GetGlobalPosition(emcglobalpos, dummy) ;
+
+
+ // The following commented code becomes valid once the PPSD provides
+ // a reasonable position resolution, at least as good as EMC !
+ // TVector3 ppsdlglobalpos ;
+ // TVector3 ppsduglobalpos ;
+ // if( fPpsdLowRecPoint ){ // certainly a photon that has concerted
+ // fPpsdLowRecPoint->GetGlobalPosition(ppsdlglobalpos, mdummy) ;
+ // dir = emcglobalpos - ppsdlglobalpos ;
+ // if( fPpsdUpRecPoint ){ // not looks like a charged
+ // fPpsdUpRecPoint->GetGlobalPosition(ppsduglobalpos, mdummy) ;
+ // dir = ( dir + emcglobalpos - ppsduglobalpos ) * 0.5 ;
+ // }
+ // }
+ // else { // looks like a neutral
+ // dir = emcglobalpos ;
+ // }
+
+ dir = emcglobalpos ;
+ dir.SetZ( -dir.Z() ) ; // why ?
+ dir.SetMag(1.) ;
+
+ //account correction to the position of IP
+ Float_t xo,yo,zo ; //Coordinates of the origin
+ gAlice->Generator()->GetOrigin(xo,yo,zo) ;
+ TVector3 origin(xo,yo,zo);
+ dir = dir - origin ;
+
+ return dir ;
+}
+//____________________________________________________________________________
+void AliPHOSPIDv1::PrintRecParticles(Option_t * option)
{
- // Modifies the parameters used for the particle type identification
+ // Print table of reconstructed particles
+
+ AliPHOSGetter *gime = AliPHOSGetter::GetInstance() ;
+
+ TString taskName(GetName()) ;
+ taskName.Remove(taskName.Index(Version())-1) ;
+ TClonesArray * recParticles = gime->RecParticles(taskName) ;
- fCutOnRelativeDistance = CutOnRelativeDistance ;
+ cout << "AliPHOSPIDv1: event "<<gAlice->GetEvNumber() << endl ;
+ cout << " found " << recParticles->GetEntriesFast() << " RecParticles " << endl ;
+
+ fRecParticlesInRun += recParticles->GetEntriesFast() ;
+
+ if(strstr(option,"all")) { // printing found TS
+
+ cout << " PARTICLE "
+ << " Index " << endl ;
+ // << " X "
+ // << " Y "
+ // << " Z "
+ // << " # of primaries "
+ // << " Primaries list " << endl;
+
+ Int_t index ;
+ for (index = 0 ; index < recParticles->GetEntries() ; index++) {
+ AliPHOSRecParticle * rp = (AliPHOSRecParticle * ) recParticles->At(index) ;
+
+ Text_t particle[11];
+ 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 ;
+ }
+
+ // Int_t * primaries;
+ // Int_t nprimaries;
+ // primaries = rp->GetPrimaries(nprimaries);
+
+ cout << setw(15) << particle << " "
+ << setw(3) << rp->GetIndexInList() << " " ;
+ // << setw(4) << nprimaries << " ";
+ // for (Int_t iprimary=0; iprimary<nprimaries; iprimary++)
+ // cout << setw(4) << primaries[iprimary] << " ";
+ cout << endl;
+ }
+ cout << "-------------------------------------------" << endl ;
+ }
+
}
+
+
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff