1 //_________________________________________________________________________
2 // Utility Class for transverse energy studies
7 //*-- Authors: Oystein Djuvsland (Bergen)
8 //_________________________________________________________________________
9 #include "AliAnalysisEtSelector.h"
10 #include "AliAnalysisEtCuts.h"
11 #include "AliESDCaloCluster.h"
13 #include "TParticle.h"
14 #include "TParticlePDG.h"
15 #include "AliAnalysisEtCommon.h"
18 ClassImp(AliAnalysisEtSelector);
20 AliAnalysisEtSelector::AliAnalysisEtSelector(AliAnalysisEtCuts *cuts) : AliAnalysisEtCommon()
28 AliAnalysisEtSelector::AliAnalysisEtSelector() : AliAnalysisEtCommon()
35 fCuts = new AliAnalysisEtCuts();
38 AliAnalysisEtSelector::~AliAnalysisEtSelector()
45 void AliAnalysisEtSelector::SetEvent(const AliESDEvent* event)
48 if(!fInitialized) Init(event);
51 Bool_t AliAnalysisEtSelector::IsNeutralMcParticle(Int_t pIdx, AliStack& s, const TParticlePDG& pdg) const
53 return s.IsPhysicalPrimary(pIdx) &&(TMath::Abs(TMath::Abs(pdg.Charge()) - fCuts->GetMonteCarloNeutralParticle())<1e-3);
56 Bool_t AliAnalysisEtSelector::IsEmEtParticle(const Int_t pdgCode) const
58 return pdgCode == fgGammaCode || pdgCode == fgPi0Code || pdgCode == fgEtaCode || pdgCode == fgEPlusCode || pdgCode == fgEMinusCode;
62 Bool_t AliAnalysisEtSelector::PrimaryIsEmEtParticle(const Int_t pIdx, AliStack& stack) const
64 return IsEmEtParticle(stack.Particle(GetPrimary(pIdx, stack))->GetPdgCode());
66 Int_t AliAnalysisEtSelector::GetPrimary(const Int_t partIdx, AliStack& stack) const
70 Int_t mothIdx = stack.Particle(partIdx)->GetMother(0);
71 if(mothIdx < 0) return -1;
72 TParticle *mother = stack.Particle(mothIdx);
75 if(stack.IsPhysicalPrimary(mothIdx)) return mothIdx;
76 else return GetPrimary(mothIdx, stack);
80 std::cout << "WAT!" << std::endl;
86 Bool_t AliAnalysisEtSelector::FromSecondaryInteraction(const TParticle& part, AliStack &stack) const
88 // Bool_t partVtxSecondary = (
89 // TMath::Sqrt(part.Vx()*part.Vx() + part.Vy()*part.Vy()) > fCuts->GetPrimaryVertexCutXY()
90 // || TMath::Abs(part.Vz()) > fCuts->GetPrimaryVertexCutZ()
92 // && TMath::Sqrt(part.Vx()*part.Vx()+part.Vy()*part.Vy() + part.Vz()*part.Vz())<(fCuts->GetGeometryPhosDetectorRadius()-10);
95 // Bool_t partVtxSecondary = (TMath::Sqrt(part.Vx()*part.Vx() + part.Vy()*part.Vy()) <300);
96 // //Let's find suspect decay (typical for secondary interaction)...
97 // if(partVtxSecondary){
98 return SuspiciousDecayInChain(211, 111, part, stack);
109 Bool_t AliAnalysisEtSelector::SuspiciousDecayInChain(const UInt_t suspectMotherPdg, const UInt_t suspectDaughterPdg, const TParticle &part, AliStack& stack) const
111 UInt_t partPdg = TMath::Abs(part.GetPdgCode());
112 if(part.GetFirstMother() == -1)
116 TParticle *mother = stack.Particle(part.GetFirstMother());
117 UInt_t motherPdg = TMath::Abs(mother->GetPdgCode());
118 if((suspectDaughterPdg==partPdg || 2112 == partPdg )&& suspectMotherPdg == motherPdg)
122 return SuspiciousDecayInChain(suspectMotherPdg, suspectDaughterPdg, *mother, stack);
125 Float_t AliAnalysisEtSelector::ShiftAngle(Float_t phi){//Always returns an angle in radians between -pi<phi<pi
127 while(myphi>TMath::Pi()){//angle is too high, decrease the angle
128 myphi = myphi - 2*TMath::Pi();
130 while(myphi<-TMath::Pi()){//angle is too low, increase the angle
131 myphi = myphi + 2*TMath::Pi();