name change Int/Diff RP/POI
[u/mrichter/AliRoot.git] / PWG2 / FLOW / FlowEventMakers / FlowEventSimpleMaker.cxx
CommitLineData
334e3256 1/**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3 * *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
6 * *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
15/* $Id */
16
17#include "Riostream.h"
18#include "FlowEventSimpleMaker.h"
fdd8c18f 19#include "AliFlowCommon/AliFlowEventSimple.h"
20#include "AliFlowCommon/AliFlowTrackSimple.h"
334e3256 21#include "TTree.h"
22#include "TParticle.h"
fdd8c18f 23#include "AliFlowCommon/AliFlowTrackSimpleCuts.h"
334e3256 24
25
26// FlowEventSimpleMaker:
27// Class to fill the AliFlowEventSimple
28// with AliFlowTrackSimple objects
29// ouside the AliRoot Framework
30// Has fill methods for TTree,
31
32ClassImp(FlowEventSimpleMaker)
33//-----------------------------------------------------------------------
34FlowEventSimpleMaker::FlowEventSimpleMaker()
35{
36 //constructor
37}
38
39//-----------------------------------------------------------------------
40FlowEventSimpleMaker::~FlowEventSimpleMaker()
41{
42 //destructor
43}
44
45//-----------------------------------------------------------------------
1918addd 46AliFlowEventSimple* FlowEventSimpleMaker::FillTracks(TTree* anInput, AliFlowTrackSimpleCuts* RPCuts, AliFlowTrackSimpleCuts* POICuts)
334e3256 47{
48 //fills the event from a TTree of kinematic.root files
49
50 // number of times to use the same particle (trick to introduce nonflow)
51 Int_t iLoops = 1;
52
1918addd 53 //flags for particles passing RP and POI cuts
54 Bool_t bPassedRPCuts = kFALSE;
55 Bool_t bPassedPOICuts = kFALSE;
334e3256 56
57 //track cut values
1918addd 58 Double_t dPtMaxRP = RPCuts->GetPtMax();
59 Double_t dPtMinRP = RPCuts->GetPtMin();
60 Double_t dEtaMaxRP = RPCuts->GetEtaMax();
61 Double_t dEtaMinRP = RPCuts->GetEtaMin();
62 Double_t dPhiMaxRP = RPCuts->GetPhiMax();
63 Double_t dPhiMinRP = RPCuts->GetPhiMin();
64 Int_t iPIDRP = RPCuts->GetPID();
334e3256 65
1918addd 66 Double_t dPtMaxPOI = POICuts->GetPtMax();
67 Double_t dPtMinPOI = POICuts->GetPtMin();
68 Double_t dEtaMaxPOI = POICuts->GetEtaMax();
69 Double_t dEtaMinPOI = POICuts->GetEtaMin();
70 Double_t dPhiMaxPOI = POICuts->GetPhiMax();
71 Double_t dPhiMinPOI = POICuts->GetPhiMin();
72 Int_t iPIDPOI = POICuts->GetPID();
334e3256 73
74 Int_t iNumberOfInputTracks = anInput->GetEntries() ;
75 //cerr<<"iNumberOfInputTracks = "<<iNumberOfInputTracks<<endl;
76 TParticle* pParticle = new TParticle();
77 anInput->SetBranchAddress("Particles",&pParticle);
78 // AliFlowEventSimple* pEvent = new AliFlowEventSimple(iNumberOfInputTracks);
79 AliFlowEventSimple* pEvent = new AliFlowEventSimple(10);
80 //cerr<<pEvent<<" pEvent "<<endl;
81
82 Int_t iN = iNumberOfInputTracks; // additional variable to artificially fix the number of tracks
83 // Int_t iN = 576; //multiplicity for chi=1.5
84 // Int_t iN = 256; //multiplicity for chi=1
85 // Int_t iN = 164; //multiplicity for chi=0.8
86
87 Int_t iGoodTracks = 0;
88 Int_t itrkN = 0;
1918addd 89 Int_t iSelParticlesPOI = 0;
90 Int_t iSelParticlesRP = 0;
334e3256 91
92 while (itrkN < iNumberOfInputTracks) {
93 anInput->GetEntry(itrkN); //get input particle
94 //checking the cuts for int. and diff. flow
1918addd 95 if (pParticle->Pt() > dPtMinRP && pParticle->Pt() < dPtMaxRP &&
96 pParticle->Eta() > dEtaMinRP && pParticle->Eta() < dEtaMaxRP &&
97 pParticle->Phi() > dPhiMinRP && pParticle->Phi() < dPhiMaxRP &&
98 TMath::Abs(pParticle->GetPdgCode()) == iPIDRP) {
99 bPassedRPCuts = kTRUE;
334e3256 100 }
101
1918addd 102 if (pParticle->Pt() > dPtMinPOI && pParticle->Pt() < dPtMaxPOI &&
103 pParticle->Eta() > dEtaMinPOI && pParticle->Eta() < dEtaMaxPOI &&
104 pParticle->Phi() > dPhiMinPOI && pParticle->Phi() < dPhiMaxPOI &&
105 TMath::Abs(pParticle->GetPdgCode()) == iPIDPOI){
106 bPassedPOICuts = kTRUE;
334e3256 107 }
108
1918addd 109 if (bPassedRPCuts || bPassedPOICuts) {
334e3256 110 for(Int_t d=0;d<iLoops;d++) {
111 AliFlowTrackSimple* pTrack = new AliFlowTrackSimple();
112 pTrack->SetPt(pParticle->Pt());
113 pTrack->SetEta(pParticle->Eta());
114 pTrack->SetPhi(pParticle->Phi());
115
116 //marking the particles used for int. flow:
1918addd 117 if(bPassedRPCuts && iSelParticlesRP < iN*iLoops) {
118 //pTrack->SetForIntegratedFlow(kTRUE);
119 pTrack->SetForPRSelection(kTRUE);
120 iSelParticlesRP++;
334e3256 121 }
122 //marking the particles used for diff. flow:
1918addd 123 if(bPassedPOICuts) {
124 pTrack->SetForPOISelection(kTRUE);
125 iSelParticlesPOI++;
334e3256 126 }
127 //adding a particles which were used either for int. or diff. flow to the list
128 pEvent->TrackCollection()->Add(pTrack);
129 iGoodTracks++;
130 }//end of for(Int_t d=0;d<iLoops;d++)
131 }//end of if(bPassedIntFlowCuts || bPassedDiffFlowCuts)
132 itrkN++;
1918addd 133 bPassedRPCuts = kFALSE;
134 bPassedPOICuts = kFALSE;
334e3256 135 }//end of while (itrkN < iNumberOfInputTracks)
136
1918addd 137 pEvent->SetEventNSelTracksRP(iSelParticlesRP);
334e3256 138 pEvent->SetNumberOfTracks(iGoodTracks);//tracks used either for int. or for diff. flow
139
140 cout<<" iGoodTracks = "<<iGoodTracks<<endl;
1918addd 141 cout<<" # of selected tracks for RP = "<<iSelParticlesRP<<endl;
142 cout<<" # of selected tracks for POI = "<<iSelParticlesPOI<<endl;
334e3256 143
144 delete pParticle;
145 return pEvent;
146}
147
148
149
150
151