]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGDQ/dielectron/macrosJPSI/ConfigJpsi_mf_PbPb.C
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Merge branch 'master' of https://git.cern.ch/reps/AliRoot
[u/mrichter/AliRoot.git] / PWGDQ / dielectron / macrosJPSI / ConfigJpsi_mf_PbPb.C
CommitLineData
bffbeda1 1void SetupTrackCutsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD);
2void SetupPairCutsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD);
490b806c 3void InitHistogramsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD,const char* triggerName);
bffbeda1 4void InitCFDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD);
6fdf1dbb 5void AddMCSignals(AliDielectron *diele);
bffbeda1 6AliESDtrackCuts *SetupESDtrackCutsDieleData(Int_t cutDefinition);
7
6fdf1dbb 8TString namesDieleData=("EMCal+QA;TPC+QA;EMCal+ITSfirstclass");
bffbeda1 9TObjArray *arrNamesDieleData=namesDieleData.Tokenize("; ");
6fdf1dbb 10Bool_t hasMC=kFALSE;
11TString list = gSystem->Getenv("LIST");
bffbeda1 12const Int_t nDie=arrNamesDieleData->GetEntries();
13
490b806c 14AliDielectron* ConfigJpsi_mf_PbPb(Int_t cutDefinition, Bool_t isAOD=kFALSE,const char* triggerName)
bffbeda1 15{
16 //
17 // Setup the instance of AliDielectron
18 //
6fdf1dbb 19 // find mc or not?
20// if( list.IsNull()) list=prod;
21 if( list.Contains("LHC10h") || list.Contains("LHC11h") ) hasMC=kFALSE;
22 if( list.Contains("LHC11a10") || list.Contains("LHC12a17") ) hasMC=kTRUE;
bffbeda1 23
24 // create the actual framework object
25 TString name=Form("%02d",cutDefinition);
26 if (cutDefinition<arrNamesDieleData->GetEntriesFast()){
27 name=arrNamesDieleData->At(cutDefinition)->GetName();
28 }
6fdf1dbb 29
30
bffbeda1 31 AliDielectron *diele = new AliDielectron(Form("%s",name.Data()),
32 Form("Track cuts: %s",name.Data()));
33
6fdf1dbb 34 diele->SetHasMC(hasMC);
35
36// printf(" Add %s %s config %s for %s \n",(isESD?"ESD":"AOD"),(hasMC?"MC":""),name.Data(),list.Data());
37
38 // Monte Carlo Signals and TRD efficiency tables
39 if(hasMC) {
40 AddMCSignals(diele);
41 printf(" Add %d MC signals \n",diele->GetMCSignals()->GetEntriesFast());
42 }
43
44
bffbeda1 45 // cut setup
6fdf1dbb 46// SetupEventCutsDieleFilter(diele, cutDefinition, isAOD);
bffbeda1 47 SetupTrackCutsDieleData(diele, cutDefinition, isAOD);
48 SetupPairCutsDieleData(diele, cutDefinition, isAOD);
49
50 //
51 // histogram setup
52 // only if an AliDielectronHistos object is attached to the
53 // dielelectron framework histograms will be filled
54 //
55 InitHistogramsDieleData(diele, cutDefinition, isAOD);
56
57 // the last definition uses no cuts and only the QA histograms should be filled!
6fdf1dbb 58 if(cutDefinition==2||cutDefinition==0) InitCFDieleData(diele, cutDefinition, isAOD);
59
60 if(!hasMC){
61 AliDielectronTrackRotator *rot=new AliDielectronTrackRotator;
62 rot->SetIterations(10);
63 rot->SetConeAnglePhi(TMath::Pi());
64 rot->SetStartAnglePhi(TMath::Pi());
65 if(cutDefinition>=2) diele->SetTrackRotator(rot);
66
67 AliDielectronMixingHandler *mix=new AliDielectronMixingHandler;
68 mix->AddVariable(AliDielectronVarManager::kZvPrim, 20,-10.,10.);
69 mix->AddVariable(AliDielectronVarManager::kCentrality, 9, 0.,90.);
70 mix->SetMixType(AliDielectronMixingHandler::kAll);
71 mix->SetDepth(120);
72 if(cutDefinition>=2) diele->SetMixingHandler(mix);
73 }
74
bffbeda1 75 return diele;
76}
77
78//______________________________________________________________________________________
79void SetupEventCutsDieleFilter(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD=kFALSE)
80{
81 //
82 // Setup the event cuts
83 //
84 AliDielectronVarCuts *vtxZ = new AliDielectronVarCuts("vtxZ","Vertex z cut");
85 vtxZ->AddCut(AliDielectronVarManager::kZvPrim,-10.,10.);
86
87 AliDielectronVarCuts *Centrality = new AliDielectronVarCuts("Centrality","Centrality Percentile");
88 Centrality->AddCut(AliDielectronVarManager::kCentrality,0.,90.);
89
90 diele->GetEventFilter().AddCuts(vtxZ);
91 diele->GetEventFilter().AddCuts(Centrality);
92
93}
94
95
96//______________________________________________________________________________________
97void SetupTrackCutsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD)
98{
99 //
100 // Setup the track cuts
101 //
102
103 //ESD quality cuts DielectronTrackCuts
104 if (!isAOD) {
105 diele->GetTrackFilter().AddCuts(SetupESDtrackCutsDieleData(cutDefinition));
106 } else {
107 AliDielectronTrackCuts *trackCuts=new AliDielectronTrackCuts("trackCuts","trackCuts");
bffbeda1 108 trackCuts->SetRequireTPCRefit(kTRUE);
109 trackCuts->SetRequireITSRefit(kTRUE);
110 diele->GetTrackFilter().AddCuts(trackCuts);
111 }
112
113 //Pt cut ----------------------------------------------------------
114 AliDielectronVarCuts *pt = new AliDielectronVarCuts("ptCut","pt cut");
6fdf1dbb 115 pt->AddCut(AliDielectronVarManager::kPt,1.,100.);
bffbeda1 116 pt->AddCut(AliDielectronVarManager::kKinkIndex0,0.);
117 //AOD additions since there are no AliESDtrackCuts -----------------
118 pt->AddCut(AliDielectronVarManager::kEta,-0.9,0.9);
119 pt->AddCut(AliDielectronVarManager::kImpactParXY, -1.0, 1.0);
120 pt->AddCut(AliDielectronVarManager::kImpactParZ, -3.0, 3.0);
121 //pt->AddCut(AliDielectronVarManager::kTPCchi2Cl, 0.0, 4.0);
6fdf1dbb 122 pt->AddCut(AliDielectronVarManager::AliDielectronVarManager::kTPCnSigmaEle,-6,6.);
123
124 if(cutDefinition==0) pt->AddCut(AliDielectronVarManager::AliDielectronVarManager::kTPCnSigmaEle,-2,3.);
125 if(cutDefinition==1) pt->AddCut(AliDielectronVarManager::kEMCALEoverP,0.9,1.2);
126 if(cutDefinition<=1) pt->AddCut(AliDielectronVarManager::kNclsTPC,90.,160.);
127
128 if(cutDefinition>=2){
129 pt->AddCut(AliDielectronVarManager::kNclsTPC,80.,160.);
130 pt->AddCut(AliDielectronVarManager::AliDielectronVarManager::kTPCnSigmaEle,-3.5,4.1);
131 pt->AddCut(AliDielectronVarManager::kPt,1.1,100.);
132 }
133
bffbeda1 134 diele->GetTrackFilter().AddCuts(pt);
135}
136
137//______________________________________________________________________________________
138void SetupPairCutsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD)
139{
140 //
141 // Setup the pair cuts
142 //
143 //Invariant mass and rapidity selection
144AliDielectronVarCuts *pairCut=new AliDielectronVarCuts("0<M<5+|Y|<.9+PtEMCalleg","0<M<5 + |Y|<.9+PtEMCalleg");
145pairCut->AddCut(AliDielectronVarManager::kM,1.,5.);
146pairCut->AddCut(AliDielectronVarManager::kY,-0.9,0.9);
6fdf1dbb 147pairCut->AddCut(AliDielectronVarManager::kPt,4.5,100.);
bffbeda1 148diele->GetPairFilter().AddCuts(pairCut);
149
150AliDielectronVarCuts *mycut = new AliDielectronVarCuts("ptCutEMCAL","cut for EMCal");
6fdf1dbb 151mycut->AddCut(AliDielectronVarManager::kEMCALE,4.,100.);
152mycut->AddCut(AliDielectronVarManager::kEMCALEoverP,0.65,2.);
bffbeda1 153AliDielectronPairLegCuts *varpair=new AliDielectronPairLegCuts();
154varpair->GetLeg1Filter().AddCuts(mycut);
155varpair->GetLeg2Filter().AddCuts(mycut);
156varpair->SetCutType(AliDielectronPairLegCuts::kAnyLeg);
6fdf1dbb 157if(cutDefinition>=2) diele->GetPairFilter().AddCuts(varpair);
bffbeda1 158
159}
160
161//______________________________________________________________________________________
162AliESDtrackCuts *SetupESDtrackCutsDieleData(Int_t cutDefinition)
163{
164 //
165 // Setup default AliESDtrackCuts
166 //
167 AliESDtrackCuts *esdTrackCuts = new AliESDtrackCuts;
168
169 // basic track quality cuts (basicQ)
170 esdTrackCuts->SetMaxDCAToVertexZ(3.0);
171 esdTrackCuts->SetMaxDCAToVertexXY(1.0);
172 esdTrackCuts->SetEtaRange( -0.9 , 0.9 );
173 esdTrackCuts->SetAcceptKinkDaughters(kFALSE);
174 esdTrackCuts->SetRequireITSRefit(kTRUE);
175 esdTrackCuts->SetRequireTPCRefit(kTRUE);
176 esdTrackCuts->SetPtRange(1.,1e30);
177 esdTrackCuts->SetMinNClustersTPC(70);
178 esdTrackCuts->SetMaxChi2PerClusterTPC(4);
179 esdTrackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD,AliESDtrackCuts::kAny);
180
181 return esdTrackCuts;
182}
183
184
185//______________________________________________________________________________________
490b806c 186void InitHistogramsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD,const char* triggerName)
bffbeda1 187{
188 //
189 // Initialise the histograms
190 //
490b806c
JB		 191 TString uniname;
192 if(cutDefinition==0) uniname="TPC";
193 else if(cutDefinition==1) uniname="EMC";
194 else uniname="Cuts";
195
bffbeda1 196
197 //Setup histogram Manager
198 AliDielectronHistos *histos=new AliDielectronHistos(diele->GetName(),diele->GetTitle());
199
200 //Initialise histogram classes
201 histos->SetReservedWords("Track;Pair");
202
203 //Track classes
204 //to fill also track info from 2nd event loop until 2
205 for (Int_t i=0; i<2; ++i){
206 histos->AddClass(Form("Track_%s",AliDielectron::TrackClassName(i)));
207 }
208
209 //Pair classes
210 // to fill also mixed event histograms loop until 10
6fdf1dbb 211 if (cutDefinition==2) {
212 for (Int_t i=0; i<10; ++i){
213 histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(i)));
214 }
215 }
bffbeda1 216
217 //legs from pair
218 for (Int_t i=0; i<3; ++i){
219 histos->AddClass(Form("Track_Legs_%s",AliDielectron::PairClassName(i)));
220 }
221
222 //track rotation
6fdf1dbb 223 if (cutDefinition==2) {
bffbeda1 224 histos->AddClass(Form("Pair_%s",PairClassName(AliDielectron::kEv1PMRot)));
225 histos->AddClass(Form("Track_Legs_%s",PairClassName(AliDielectron::kEv1PMRot)));
226 }
227
228 //add histograms to event class
229 if (cutDefinition==0) {
230 histos->AddClass("Event");
490b806c
JB		 231 histos->UserHistogram("Event",Form("VtxZ_%s",triggerName),"Vertex Z;Z[cm]",300,-15.,15.,AliDielectronVarManager::kZvPrim);
232 histos->UserHistogram("Event",Form("Centrality_%s",triggerName),"Centrality;Cent(%)",100,0.,100.,AliDielectronVarManager::kCentrality);
233 histos->UserHistogram("Event",Form("Multiplicity_%s",triggerName),"Multiplicity V0;Multiplicity V0",500,0.,25000.,AliDielectronVarManager::kMultV0);
234 histos->UserHistogram("Event",Form("Multiplicity_cent_%s",triggerName),"Multiplicity V0 x Cent;Cent(%);Multiplicity V0",10,0,100.,500,0.,25000.,AliDielectronVarManager::kCentrality,AliDielectronVarManager::kMultV0);
bffbeda1 235 }
236
6fdf1dbb 237 //add histograms to Track classes
490b806c
JB		 238 histos->UserHistogram("Track",Form("Pt_%s_%s",uniname.Data(),triggerName),"Pt;Pt [GeV];#tracks",100,0,20.,AliDielectronVarManager::kPt);
239 histos->UserHistogram("Track",Form("TPCnCls_%s_%s",uniname.Data(),triggerName),"Number of Clusters TPC;TPC number clusteres;#tracks",160,-0.5,159.5,AliDielectronVarManager::kNclsTPC);
240 histos->UserHistogram("Track",Form("dXY_%s_%s",uniname.Data(),triggerName),"dXY;dXY [cm];#tracks",200,-1.,1.,AliDielectronVarManager::kImpactParXY);
241 histos->UserHistogram("Track",Form("dZ_%s_%s",uniname.Data(),triggerName),"dZ;dZ [cm];#tracks",200,-3.,3.,AliDielectronVarManager::kImpactParZ);
242 histos->UserHistogram("Track",Form("Eta_Phi_%s_%s",uniname.Data(),triggerName),"Eta Phi Map; Eta; Phi;#tracks",
bffbeda1 243 100,-1,1,144,0,6.285,AliDielectronVarManager::kEta,AliDielectronVarManager::kPhi);
490b806c
JB		 244 histos->UserHistogram("Track",Form("dEdx_P_%s_%s",uniname.Data(),triggerName),"dEdx_P;P [GeV];TPC signal (arb units);#tracks",
245 200,0.2,20.,800,20.,200.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCsignal);
bffbeda1 246
490b806c
JB		 247 histos->UserHistogram("Track",Form("dEdx_Pt_%s_%s",uniname.Data(),triggerName),"dEdx;Pt [GeV];TPC signal (arb units);#tracks",
248 200,0.2,20.,800,20.,200.,AliDielectronVarManager::kPt,AliDielectronVarManager::kTPCsignal);
6fdf1dbb 249
490b806c
JB		 250 histos->UserHistogram("Track",Form("TPCnSigmaPion_P_%s_%s",uniname.Data(),triggerName),"TPCnSigmaPion;P [GeV];TPCnSigmaEle;#tracks",
251 200,0.2,20.,800,-12.,12.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaPio);
6fdf1dbb 252
490b806c
JB		 253 histos->UserHistogram("Track",Form("TPCnSigmaEle_P_%s_%s",uniname.Data(),triggerName),"TPCnSigmaEle;P [GeV];TPCnSigmaEle;#tracks",
254 200,0.2,20.,800,-12.,12.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaEle);
6fdf1dbb 255
490b806c
JB		 256 histos->UserHistogram("Track",Form("TPCnSigmaEle_Pt_%s_%s",uniname.Data(),triggerName),"TPCnSigmaEle;Pt [GeV];TPCnSigmaEle;#tracks",
257 200,0.2,20.,800,-12.,12.,AliDielectronVarManager::kPt,AliDielectronVarManager::kTPCnSigmaEle);
6fdf1dbb 258
6fdf1dbb 259
490b806c
JB		 260 histos->UserHistogram("Track",Form("TPCnSigmaEle_Eta_%s_%s",uniname.Data(),triggerName),"TPCnSigmaEle;#eta;TPCnSigmaEle;#tracks",
261 200,-1.,1.,800,-12.,12.,AliDielectronVarManager::kEta,AliDielectronVarManager::kTPCnSigmaEle);
6fdf1dbb 262
490b806c
JB		 263
264 histos->UserHistogram("Track",Form("TPCnSigmaEle_Phi_%s_%s",uniname.Data(),triggerName),"TPCnSigmaEle;#phi [rad];TPCnSigmaEle;#tracks",
265 AliDielectronHelper::MakeLinBinning(200,0.,2*TMath::Pi()),AliDielectronHelper::MakeLinBinning(800,20.,200.),AliDielectronVarManager::kPhi,AliDielectronVarManager::kTPCnSigmaEle);
266
bffbeda1 267
490b806c
JB		 268 histos->UserHistogram("Track",Form("dEdx_Phi_%s_%s",uniname.Data(),triggerName),"dEdx;#phi [rad];TPC signal (arb units);#tracks",
269 AliDielectronHelper::MakeLinBinning(200,0.,2*TMath::Pi()),AliDielectronHelper::MakeLinBinning(800,20.,200.),AliDielectronVarManager::kPhi,AliDielectronVarManager::kTPCsignal);
6fdf1dbb 270
490b806c
JB		 271 histos->UserHistogram("Track",Form("dEdx_Eta_%s_%s",uniname.Data(),triggerName),"dEdx;#eta;TPC signal (arb units);#tracks",
272 200,-1.,1.,800,20.,200.,AliDielectronVarManager::kEta,AliDielectronVarManager::kTPCsignal);
bffbeda1 273
490b806c
JB		 274 histos->UserHistogram("Track",Form("dEdx_nSigmaEMCal_%s_%s",uniname.Data(),triggerName),"dEdx;NsigmaEmcal;TPC signal (arb units);NSigmaEMCAL",
275 200,-5.,5.,800,20.,200.,AliDielectronVarManager::kEMCALnSigmaEle,AliDielectronVarManager::kTPCsignal);
bffbeda1 276
490b806c
JB		 277 histos->UserHistogram("Track",Form("dEdx_TPCnSigmaEle_%s_%s",uniname.Data(),triggerName),"dEdx;TPC signal (arbunits);TPC number of sigmas Electrons;TPC signal (a.u.);#tracks",
278 100,-10.,10.,800,20.,200.,AliDielectronVarManager::kTPCnSigmaEle,AliDielectronVarManager::kTPCsignal);
bffbeda1 279
490b806c 280 histos->UserHistogram("Track",Form("dEdx_EoverP_%s_%s",uniname.Data(),triggerName),"dEdx;EoverP;TPC signal (arbunits);E/P",100,0.,5.,800,20.,200.,AliDielectronVarManager::kEMCALEoverP,AliDielectronVarManager::kTPCsignal);
bffbeda1 281
490b806c 282 histos->UserHistogram("Track",Form("nSigmaEMCal_EoverP_%s_%s",uniname.Data(),triggerName),"NsigmaEmcal;EoverP;NSigmaEMCAL;E/P",100,0.,5.,200,-5.,5.,AliDielectronVarManager::kEMCALEoverP,AliDielectronVarManager::kEMCALnSigmaEle);
6fdf1dbb 283
490b806c 284 histos->UserHistogram("Track",Form("EMCal_E_%s_%s",uniname.Data(),triggerName),"EmcalE;Cluster Energy [GeV];#Clusters",200,0.,40.,AliDielectronVarManager::kEMCALE);
bffbeda1 285
490b806c
JB		 286 histos->UserHistogram("Track",Form("EMCal_E_nCells_%s_%s",uniname.Data(),triggerName),"EmcalE_nCells;# cells;Cluster Energy [GeV]",20,0,20,200,0.,40.,AliDielectronVarManager::kEMCALNCells,AliDielectronVarManager::kEMCALE);
287 histos->UserHistogram("Track",Form("ITS_FirstCls_%s_%s",uniname.Data(),triggerName),"ITS First Layer;ITS First Layer;#Entries",AliDielectronHelper::MakeLinBinning(6,0.,6.),AliDielectronVarManager::kITSLayerFirstCls);
6fdf1dbb 288
bffbeda1 289 //add histograms to Pair classes
490b806c 290 histos->UserHistogram("Pair",Form("InvMass_%s_%s",uniname.Data(),triggerName),"Inv.Mass;Inv. Mass [GeV];#pairs",
bffbeda1 291 251,-.01,5.01,AliDielectronVarManager::kM);
490b806c 292 histos->UserHistogram("Pair",Form("InvMass2D_%s_%s",uniname.Data(),triggerName),"Inv.Mass;Pt [GeV]; Inv. Mass [GeV]",
bffbeda1 293 20,0.,20.,251,-.01,5.01,AliDielectronVarManager::kPt,AliDielectronVarManager::kM);
490b806c 294 histos->UserHistogram("Pair",Form("Rapidity_%s_%s",uniname.Data(),triggerName),"Rapidity;Rapidity;#pairs",
bffbeda1 295 100,-1.,1.,AliDielectronVarManager::kY);
490b806c 296 histos->UserHistogram("Pair",Form("OpeningAngle_%s_%s",uniname.Data(),triggerName),"Opening angle;angle",
bffbeda1 297 100,0.,3.15,AliDielectronVarManager::kOpeningAngle);
298
299 diele->SetHistogramManager(histos);
300}
301
302void InitCFDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD)
303{
304 //
305 // Setupd the CF Manager if needed
306 //
307
308 AliDielectronCF *cf=new AliDielectronCF(diele->GetName(),diele->GetTitle());
309
310 //pair variables
6fdf1dbb 311 cf->AddVariable(AliDielectronVarManager::kPt,"4.0, 4.5, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 20.0, 100.0");
bffbeda1 312 cf->AddVariable(AliDielectronVarManager::kM,125,0.,125*.04); //40Mev Steps
313 cf->AddVariable(AliDielectronVarManager::kPairType,12,0,12);
6fdf1dbb 314 cf->AddVariable(AliDielectronVarManager::kOpeningAngle,"0.,0.1,0.2,0.4,0.6,0.8,1.0,2.0,3.14");
315 cf->AddVariable(AliDielectronVarManager::kY,20,-1.,1.);
bffbeda1 316 //leg variables
6fdf1dbb 317 cf->AddVariable(AliDielectronVarManager::kPt,"1.0, 1.1, 1.2, 1.3, 1.4, 1.5,1.6,1.7,1.8,1.9, 2.0, 3.0, 10.0, 100.0",kTRUE);
bffbeda1 318// cf->AddVariable(AliDielectronVarManager::kP,"0.0, 1.0, 1.2, 1.5, 2.0, 2.5, 3.0, 3.5, 4.0, 5.0, 10.0, 100.0",kTRUE);
6fdf1dbb 319 cf->AddVariable(AliDielectronVarManager::kNclsTPC,"80, 85, 90, 95, 100, 110, 120, 160",kTRUE);
bffbeda1 320// cf->AddVariable(AliDielectronVarManager::kEta,"-5.0,-1.0,-0.9,-0.7,0.7,0.9,1.0,5.0",kTRUE);
490b806c 321 cf->AddVariable(AliDielectronVarManager::kPhi,"0.0,1.0,2.0,3.0,4.0,5.0,6.0",kTRUE);
6fdf1dbb 322 cf->AddVariable(AliDielectronVarManager::kEMCALE,"1.0,2.0, 3.0, 4.0, 4.5, 5.0, 5.5, 6.0, 7.0, 8.0, 9.0, 10.0,11.0,12.0,16.0,100.0",kTRUE);
bffbeda1 323 cf->AddVariable(AliDielectronVarManager::kEMCALnSigmaEle,"-3.5,-3.0,-2.0,-1.0,1.0,2.0,3.0,4.0,5.0",kTRUE);
bffbeda1 324 cf->AddVariable(AliDielectronVarManager::kEMCALNCells,25,0,25,kTRUE);
6fdf1dbb 325 cf->AddVariable(AliDielectronVarManager::kEMCALEoverP,"0.6, 0.65,0.7,0.8,0.9,1.1,1.2,1.3,1.4,1.8,2.0,2.1",kTRUE);
326 cf->AddVariable(AliDielectronVarManager::kTPCnSigmaEle,"-3.5,-3.0,-2.5,-2.0,-1.5,-1.0,0.0,1.0,2.0,3.0,3.5,4.0",kTRUE);
490b806c 327 cf->AddVariable(AliDielectronVarManager::kTPCnSigmaPio,"2.,2.5,3.0,3.5,4.0,4.5,5.0,5.5,100",kTRUE);
bffbeda1 328 //event variables
329 cf->AddVariable(AliDielectronVarManager::kCentrality,20,0.,100.);
330 cf->AddVariable(AliDielectronVarManager::kMultV0,26,0.,26000.);
6fdf1dbb 331 cf->AddVariable(AliDielectronVarManager::kZvPrim,20,-10.,10.);
bffbeda1 332
bffbeda1 333 cf->AddVariable(AliDielectronVarManager::kITSLayerFirstCls,6,0.,6.,kTRUE);
334
6fdf1dbb 335 if (hasMC){
bffbeda1 336 cf->AddVariable(AliDielectronVarManager::kPdgCode,10000,-5000.5,4999.5,kTRUE);
337 cf->AddVariable(AliDielectronVarManager::kPdgCodeMother,10000,-5000.5,4999.5,kTRUE);
338 cf->AddVariable(AliDielectronVarManager::kPdgCodeGrandMother,10000,-5000.5,4999.5,kTRUE);
bffbeda1 339 }
340 //only in this case write MC truth info
341 if (cutDefinition==0){
342 cf->SetStepForMCtruth();
6fdf1dbb 343// cf->SetStepsForMCtruthOnly();
344 }
bffbeda1 345
346 diele->SetCFManagerPair(cf);
347
348}
349
6fdf1dbb 350void AddMCSignals(AliDielectron *diele){
351 //Do we have an MC handler?
352 //Bool_t hasMC=(AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler()!=0x0);
353 if (!hasMC) return;
354
355 AliDielectronSignalMC* inclusiveJpsi = new AliDielectronSignalMC("inclusiveJpsi","Inclusive J/psi");
356 inclusiveJpsi->SetLegPDGs(11,-11);
357 inclusiveJpsi->SetMotherPDGs(443,443);
358 inclusiveJpsi->SetMothersRelation(AliDielectronSignalMC::kSame);
359 inclusiveJpsi->SetFillPureMCStep(kTRUE);
360 inclusiveJpsi->SetCheckBothChargesLegs(kTRUE,kTRUE);
361 inclusiveJpsi->SetCheckBothChargesMothers(kTRUE,kTRUE);
362 diele->AddSignalMC(inclusiveJpsi);
363
364 AliDielectronSignalMC* promptJpsi = new AliDielectronSignalMC("promptJpsi","Prompt J/psi"); // prompt J/psi (not from beauty decays)
365 promptJpsi->SetLegPDGs(11,-11);
366 promptJpsi->SetMotherPDGs(443,443);
367 promptJpsi->SetGrandMotherPDGs(503,503,kTRUE,kTRUE); // not from beauty hadrons
368 promptJpsi->SetMothersRelation(AliDielectronSignalMC::kSame);
369 promptJpsi->SetFillPureMCStep(kTRUE);
370 promptJpsi->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
371 promptJpsi->SetCheckBothChargesLegs(kTRUE,kTRUE);
372 promptJpsi->SetCheckBothChargesMothers(kTRUE,kTRUE);
373 promptJpsi->SetCheckBothChargesGrandMothers(kTRUE,kTRUE);
374 diele->AddSignalMC(promptJpsi);
375
376 AliDielectronSignalMC* beautyJpsi = new AliDielectronSignalMC("beautyJpsi","Beauty J/psi");
377 beautyJpsi->SetLegPDGs(11,-11);
378 beautyJpsi->SetMotherPDGs(443,443);
379 beautyJpsi->SetMothersRelation(AliDielectronSignalMC::kSame);
380 beautyJpsi->SetGrandMotherPDGs(500,500);
381 beautyJpsi->SetFillPureMCStep(kTRUE);
382 beautyJpsi->SetCheckBothChargesLegs(kTRUE,kTRUE);
383 beautyJpsi->SetCheckBothChargesMothers(kTRUE,kTRUE);
384 beautyJpsi->SetCheckBothChargesGrandMothers(kTRUE,kTRUE);
385 diele->AddSignalMC(beautyJpsi);
386
387 AliDielectronSignalMC* directJpsi = new AliDielectronSignalMC("directJpsi","Direct J/psi"); // embedded J/psi
388 directJpsi->SetLegPDGs(11,-11);
389 directJpsi->SetMotherPDGs(443,443);
390 directJpsi->SetMothersRelation(AliDielectronSignalMC::kSame);
391 directJpsi->SetFillPureMCStep(kTRUE);
392 directJpsi->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
393 directJpsi->SetMotherSources(AliDielectronSignalMC::kDirect, AliDielectronSignalMC::kDirect);
394 directJpsi->SetCheckBothChargesLegs(kTRUE,kTRUE);
395 directJpsi->SetCheckBothChargesMothers(kTRUE,kTRUE);
396 diele->AddSignalMC(directJpsi);
397}