Moving some functionality to selection base class.
[u/mrichter/AliRoot.git] / PWGLF / totEt / AliAnalysisEtSelector.cxx
1 //_________________________________________________________________________
2 //  Utility Class for transverse energy studies
3 //  Selector Base class 
4 //  -  
5 // implementation file
6 //
7 //*-- Authors: Oystein Djuvsland (Bergen)
8 //_________________________________________________________________________
9 #include "AliAnalysisEtSelector.h"
10 #include "AliAnalysisEtCuts.h"
11 #include "AliESDCaloCluster.h"
12 #include "AliStack.h"
13 #include "TParticle.h"
14 #include "TParticlePDG.h"
15 #include "AliAnalysisEtCommon.h"
16 #include <iostream>
17
18 ClassImp(AliAnalysisEtSelector);
19
20 AliAnalysisEtSelector::AliAnalysisEtSelector(AliAnalysisEtCuts *cuts) : AliAnalysisEtCommon()
21 ,fEvent(0)
22 ,fClusterArray(0)
23 ,fCuts(cuts)
24 ,fRunNumber(0)
25 ,fInitialized(kFALSE)
26 {
27 }
28
29 AliAnalysisEtSelector::~AliAnalysisEtSelector()
30 { // dtor
31   if(fClusterArray)
32   {
33     delete fClusterArray;
34   }
35 }
36 void AliAnalysisEtSelector::SetEvent(const AliESDEvent* event)
37 {
38     fEvent = event;
39     if(!fInitialized) Init(event);
40 }
41
42 Bool_t AliAnalysisEtSelector::IsNeutralMcParticle(Int_t pIdx, AliStack& s, const TParticlePDG& pdg) const
43 {
44   return s.IsPhysicalPrimary(pIdx) &&(TMath::Abs(TMath::Abs(pdg.Charge()) - fCuts->GetMonteCarloNeutralParticle())<1e-3);
45 }
46
47 Bool_t AliAnalysisEtSelector::IsEmEtParticle(const Int_t pdgCode) const
48 {
49   return pdgCode == fgGammaCode || pdgCode == fgPi0Code || pdgCode == fgEtaCode || pdgCode == fgEPlusCode || pdgCode == fgEMinusCode;
50 }
51
52
53 Bool_t AliAnalysisEtSelector::PrimaryIsEmEtParticle(const Int_t pIdx, AliStack& stack) const
54 {
55   return IsEmEtParticle(stack.Particle(GetPrimary(pIdx, stack))->GetPdgCode());
56 }
57 Int_t AliAnalysisEtSelector::GetPrimary(const Int_t partIdx, AliStack& stack) const
58 { // get primary
59   if(partIdx >= 0) 
60   {
61     Int_t mothIdx = stack.Particle(partIdx)->GetMother(0);
62     if(mothIdx < 0) return -1;
63     TParticle *mother = stack.Particle(mothIdx);
64     if(mother)
65     {
66       if(stack.IsPhysicalPrimary(mothIdx)) return mothIdx;
67       else return GetPrimary(mothIdx, stack);
68     }
69     else 
70     {
71       std::cout << "WAT!" << std::endl;
72       return -1;
73     }
74   }
75   return -1;
76 }
77 Bool_t AliAnalysisEtSelector::FromSecondaryInteraction(const TParticle& part, AliStack &stack) const
78 {
79 //   Bool_t partVtxSecondary = (
80 //                            TMath::Sqrt(part.Vx()*part.Vx() + part.Vy()*part.Vy()) > fCuts->GetPrimaryVertexCutXY() 
81 //                            || TMath::Abs(part.Vz()) > fCuts->GetPrimaryVertexCutZ()
82 //                          )
83 //                          && TMath::Sqrt(part.Vx()*part.Vx()+part.Vy()*part.Vy() + part.Vz()*part.Vz())<(fCuts->GetGeometryPhosDetectorRadius()-10);
84   
85   //Let's find suspect decay (typical for secondary interaction)...
86   
87   return SuspeciousDecayInChain(211, 111, part, stack);
88   
89                             
90   
91   
92 }
93
94 Bool_t AliAnalysisEtSelector::SuspeciousDecayInChain(const UInt_t suspectMotherPdg, const UInt_t suspectDaughterPdg, const TParticle &part, AliStack& stack) const
95 {
96   UInt_t partPdg = TMath::Abs(part.GetPdgCode());
97   if(part.GetFirstMother() == -1)
98   {
99     return kFALSE;
100   }
101   TParticle *mother = stack.Particle(part.GetFirstMother()); 
102   UInt_t motherPdg = TMath::Abs(mother->GetPdgCode());
103   if((suspectDaughterPdg==partPdg || 2112 == partPdg )&& suspectMotherPdg == motherPdg)
104   {
105     return kTRUE;
106   }
107   return SuspeciousDecayInChain(suspectMotherPdg, suspectDaughterPdg, *mother, stack);
108 }