1 void InitHistograms(AliDielectron *die, Int_t cutDefinition);
3 void SetupEventCuts(AliDielectron *die, Int_t cutDefinition);
4 void SetupTrackCuts(AliDielectron *die, Int_t cutDefinition);
5 void SetupV0Cuts( AliDielectron *die, Int_t cutDefinition);
6 void SetupPairCuts( AliDielectron *die, Int_t cutDefinition);
8 void ConfigEvtPlane(AliDielectron *die, Int_t cutDefinition);
9 void SetEtaCorrection(AliDielectron *die);
11 TString names=("NANO");
14 TObjArray *arrNames=names.Tokenize(";");
15 const Int_t nDie=arrNames->GetEntries();
18 TString periodLHC = "";
19 TString list = gSystem->Getenv("LIST");
22 AliDielectron* ConfigJpsi_nano_PbPb(Int_t cutDefinition, Bool_t hasMC=kFALSE, TString period="")
25 // Setup the instance of AliDielectron
29 printf("this is -%s- filtering \n",periodLHC.Data());
32 isESD=(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler()->IsA()==AliESDInputHandler::Class());
35 TString name=Form("%02d",cutDefinition);
36 if (cutDefinition<arrNames->GetEntriesFast()) name=arrNames->At(cutDefinition)->GetName();
37 printf(" Adding %s%s config %s for %s \n",(isESD?"ESD":"AOD"),(hasMC?" MC":""),name.Data(),list.Data());
40 AliDielectron *die = new AliDielectron(Form("%s",name.Data()), Form("%s",name.Data()));
43 /* vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv CUTS vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv */
44 SetupEventCuts(die,cutDefinition);
45 SetupTrackCuts(die,cutDefinition);
46 // SetupV0Cuts(die,cutDefinition); // switch off for nanoAODs??
47 // SetupPairCuts(die,cutDefinition);
49 /* vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv MISC vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv */
51 //SetEtaCorrection(die); //no eta corrction
55 // else die->SetPreFilterUnlikeOnly();
56 //die->SetPreFilterAllSigns();
58 /* vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv OUTPUT vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv */
60 InitHistograms(die,cutDefinition);
68 //______________________________________________________________________________________
69 void SetupEventCuts(AliDielectron *die, Int_t cutDefinition)
72 // Setup the event cuts
75 AliDielectronEventCuts *eventCuts=new AliDielectronEventCuts("eventCuts","eventCuts");
76 if(!isESD) eventCuts->SetVertexType(AliDielectronEventCuts::kVtxAny);
77 eventCuts->SetRequireVertex();
78 eventCuts->SetMinVtxContributors(1);
79 // eventCuts->SetVertexZ(-10.,+10.);
80 eventCuts->SetCentralityRange(0., 90.);
82 die->GetEventFilter().AddCuts(eventCuts);
86 //______________________________________________________________________________________
87 void SetupTrackCuts(AliDielectron *die, Int_t cutDefinition)
90 // Setup the track cuts
93 Bool_t hasMC=die->GetHasMC();
96 AliDielectronCutGroup* cuts = new AliDielectronCutGroup("cuts","cuts",AliDielectronCutGroup::kCompAND);
97 die->GetTrackFilter().AddCuts(cuts);
99 /* vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv FILTER CUTS vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv */
100 // AOD track filter (needs to be first cut to speed up)
101 AliDielectronTrackCuts *trkFilter = new AliDielectronTrackCuts("TrkFilter","TrkFilter");
102 trkFilter->SetAODFilterBit(AliDielectronTrackCuts::kTPCqual);
103 // trkFilter->SetAODFilterBit(AliDielectronTrackCuts::kTPCqualSPDany);
104 // trkFilter->SetMinNCrossedRowsOverFindable(0.6);
106 /* vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv TRACK CUTS vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv */
107 AliDielectronVarCuts *varCuts = new AliDielectronVarCuts("VarCuts","VarCuts");
108 AliDielectronTrackCuts *trkCuts = new AliDielectronTrackCuts("TrkCuts","TrkCuts");
110 varCuts->AddCut(AliDielectronVarManager::kPt, 0.85, 1e30);
111 varCuts->AddCut(AliDielectronVarManager::kEta, -0.9, 0.9);
112 if(periodLHC.Contains("LHC11h"))
113 varCuts->AddCut(AliDielectronVarManager::kTPCclsSegments,7., 8.0);
114 varCuts->AddCut(AliDielectronVarManager::kNclsTPC, 70.0, 160.0);
115 // trkCuts->SetITSclusterCut(AliDielectronTrackCuts::kOneOf, 3); // SPD any
118 varCuts->AddCut(AliDielectronVarManager::kImpactParXY, -1.0, 1.0);
119 varCuts->AddCut(AliDielectronVarManager::kImpactParZ, -3.0, 3.0);
120 varCuts->AddCut(AliDielectronVarManager::kTPCchi2Cl, 0.0, 4.0);
121 varCuts->AddCut(AliDielectronVarManager::kKinkIndex0, 0.);
122 trkCuts->SetRequireITSRefit(kTRUE);
123 trkCuts->SetRequireTPCRefit(kTRUE);
125 /* vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv PID CUTS vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv */
126 // AliDielectronVarCuts *pidVarCuts = new AliDielectronVarCuts("varPIDCuts","varPIDCuts");
127 AliDielectronPID *pidCuts = new AliDielectronPID("PIDCuts","PIDCuts");
129 // pidVarCuts->AddCut(AliDielectronVarManager::kTOFbeta, 0.2, 0.9, kTRUE);
130 // pidCuts->AddCut(AliDielectronPID::kTOF,AliPID::kElectron,-3,3.,0.,0.,kFALSE, AliDielectronPID::kIfAvailable);
132 if(periodLHC.Contains("LHC10h"))
133 pidCuts->AddCut(AliDielectronPID::kTPC,AliPID::kElectron,-2.6,4.7); // when eta correction OFF (LHC10h) [-2.0,+3.]
134 // pidCuts->AddCut(AliDielectronPID::kTPC,AliPID::kElectron,-4.0,4.7); // when eta correction OFF (LHC10h) [/*-2.0*/,+3.]
136 pidCuts->AddCut(AliDielectronPID::kTPC,AliPID::kElectron,-3.0,4.3); // when eta correction OFF (LHC11h) [-1.5,+3.]
137 // pidCuts->AddCut(AliDielectronPID::kTPC,AliPID::kPion,-100.,4.0,0.,0.,kTRUE);
138 // pidCuts->AddCut(AliDielectronPID::kTPC,AliPID::kProton,-100.,3.5,0.,0.,kTRUE);
140 /* vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv MC PID CUTS vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv */
141 AliDielectronVarCuts *pidCutsMC = new AliDielectronVarCuts("PIDCutsMC","PIDCutsMC");
142 pidCutsMC->SetCutType(AliDielectronVarCuts::kAny);
143 pidCutsMC->SetCutOnMCtruth(kTRUE);
144 pidCutsMC->AddCut(AliDielectronVarManager::kPdgCode, -11., -11.);
145 pidCutsMC->AddCut(AliDielectronVarManager::kPdgCode, +11., +11.);
147 /* vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv TENDER CUTS vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv */
148 // exclude conversion electrons selected by the tender
149 AliDielectronTrackCuts *noconv=new AliDielectronTrackCuts("noConv","noConv");
150 noconv->SetV0DaughterCut(AliPID::kElectron,kTRUE);
152 // activate the cut sets (order might be CPU timewise important)
154 // cuts->AddCut(pidCutsMC);
157 if(!isESD) cuts->AddCut(trkFilter);
158 cuts->AddCut(varCuts);
159 cuts->AddCut(trkCuts);
160 cuts->AddCut(pidCuts);
161 //cuts->AddCut(pidVarCuts);
162 //cuts->AddCut(noconv);
168 //______________________________________________________________________________________
169 void SetupV0Cuts(AliDielectron *die, Int_t cutDefinition)
175 // Quality cuts (add the gamma filter to the cut group)
176 TIter next(die->GetTrackFilter().GetCuts());
177 AliAnalysisCuts *cuts;
178 while((cuts = (AliAnalysisCuts*)next())) {
179 if(cuts->IsA() == AliDielectronCutGroup::Class()) break;
182 AliDielectronV0Cuts *gammaV0Cuts = new AliDielectronV0Cuts("IsGamma","IsGamma");
183 gammaV0Cuts->SetPdgCodes(22,11,11);
184 gammaV0Cuts->SetDefaultPID(16);
185 gammaV0Cuts->AddCut(AliDielectronVarManager::kCosPointingAngle, TMath::Cos(0.02), 1.0, kFALSE);
186 gammaV0Cuts->AddCut(AliDielectronVarManager::kChi2NDF, 0.0, 10.0, kFALSE);
187 gammaV0Cuts->AddCut(AliDielectronVarManager::kLegDist, 0.0, 0.25, kFALSE);
188 gammaV0Cuts->AddCut(AliDielectronVarManager::kR, 3.0, 90.0, kFALSE);
189 gammaV0Cuts->AddCut(AliDielectronVarManager::kPsiPair, 0.0, 0.05, kFALSE);
190 gammaV0Cuts->AddCut(AliDielectronVarManager::kM, 0.0, 0.05, kFALSE);
191 // gammaV0Cuts->AddCut(AliDielectronVarManager::kOpeningAngle, 0.0, 0.1, kFALSE);
192 gammaV0Cuts->AddCut(AliDielectronVarManager::kArmPt, 0.0, 0.05, kFALSE);
193 // gammaV0Cuts->AddCut(AliDielectronVarManager::kArmAlpha, -0.35, 0.35, kFALSE); // not sure if it works as expected
194 gammaV0Cuts->SetExcludeTracks(kTRUE);
195 gammaV0Cuts->Print();
197 // const Double_t |cutAlphaG| < 0.35; && const Double_t cutQTG < 0.05;
198 // const Double_t |cutAlphaG2|[2] = {0.6, 0.8}; && const Double_t cutQTG2 < 0.04;
201 ((AliDielectronCutGroup*)cuts)->AddCut(gammaV0Cuts);
203 die->GetTrackFilter().AddCuts(gammaV0Cuts);
206 //______________________________________________________________________________________
207 void SetupPairCuts(AliDielectron *die, Int_t cutDefinition)
210 // Setup the pair cuts
213 // conversion rejection
215 AliDielectronVarCuts *gammaCuts = new AliDielectronVarCuts("GammaCuts","GammaCuts");
216 gammaCuts->AddCut(AliDielectronVarManager::kM, 0.0, gCut);
217 // die->GetPairPreFilter().AddCuts(gammaCuts);
219 // rapidity selection
221 AliDielectronVarCuts *rapCut=new AliDielectronVarCuts(Form("|Y|<%.1f",yCut),Form("|Y|<%.1f",yCut));
222 rapCut->AddCut(AliDielectronVarManager::kY,-1.*yCut,yCut);
223 die->GetPairFilter().AddCuts(rapCut);
227 //______________________________________________________________________________________
228 void InitHistograms(AliDielectron *die, Int_t cutDefinition)
231 // Initialise the histograms
233 Bool_t hasMC=die->GetHasMC();
235 //Setup histogram Manager
236 AliDielectronHistos *histos=new AliDielectronHistos(die->GetName(),die->GetTitle());
238 //add histograms to event class
239 histos->AddClass("Event");
240 histos->UserHistogram("Event","","", 100, 0.0, 100.0, AliDielectronVarManager::kCentrality);
241 histos->UserHistogram("Event","","", 300,-15., +15.0, AliDielectronVarManager::kZvPrim);
242 // candidates monitoring
243 histos->UserProfile("Event","","", AliDielectronVarManager::kTracks, 100,0,100., AliDielectronVarManager::kCentrality);
244 histos->UserProfile("Event","","", AliDielectronVarManager::kPairs, 100,0,100., AliDielectronVarManager::kCentrality);
246 die->SetHistogramManager(histos);
249 //______________________________________________________________________________________
250 void SetEtaCorrection(AliDielectron *die) {
252 if (AliDielectronPID::GetCentroidCorrFunction()) return;
256 /* vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv DATA vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv */
257 if( !die->GetHasMC() ) {
258 // 2-dimensional eta correction for the centroid of electron sigmas
259 fCntrdCorr = new TF2("fCntrdCorr", "[0] + [1]*y + [2]*y*y + [3]*TMath::Power(y,3) + [4]*TMath::Power(y,4) + [5]*TMath::Power(y,5) + [6]*TMath::Power(y,6) + [7]*x",
260 //0.0, 3000.0, -0.9, +0.9); // Nacc
261 0.0, 90.0, -0.9, +0.9); // centrality
262 // fCntrdCorr->SetParameters(0.723106, 0.23958, -6.31221, -0.687976, 15.912, 0.579609, -11.6901, -0.000354381); // Nacc
263 fCntrdCorr->SetParameters(+0.149002, +0.214644 , -6.034930, -0.529588, +14.97902, +0.402640, -10.890027, +0.011248); // centrality
264 // 1-dimensional eta correction for the width of electron sigmas
265 // fWdthCorr = new TF1("fWdthCorr", "pol2", 0.0, 3000.0); // Nacc
266 // fWdthCorr->SetParameters(1.06108, 0.000217804,-5.80291e-08);
267 fWdthCorr = new TF1("fWdthCorr", "pol2", 0.0, 90.0); // centrality
268 fWdthCorr->SetParameters(+1.290755, -0.005261, +0.000021);
271 /* vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv MONTE CARLO vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv */
272 // 2-dimensional eta correction for the centroid of electron sigmas
273 fCntrdCorr = new TF2("fCntrdCorr", "[0] + [1]*y + [2]*y*y + [3]*TMath::Power(y,3) + [4]*TMath::Power(y,4) + [5]*TMath::Power(y,5) + [6]*TMath::Power(y,6) + [7]*x",
274 0.0, 3000.0, -0.9, +0.9);
275 fCntrdCorr->SetParameters(+0.378611, -0.070831, -3.076778, +0.121977, +8.576097, +0.113009, -5.001368, -0.000181);
277 // 1-dimensional eta correction for the width of electron sigmas
278 fWdthCorr = new TF1("fWdthCorr", "pol1", 0.0, 3000.0);
279 fWdthCorr->SetParameters(+0.881894, +0.000053);
283 // AliDielectronPID::SetCentroidCorrFunction(fCntrdCorr,AliDielectronVarManager::kNacc,AliDielectronVarManager::kEta);
284 // AliDielectronPID::SetWidthCorrFunction(fWdthCorr,AliDielectronVarManager::kNacc);
285 AliDielectronPID::SetCentroidCorrFunction(fCntrdCorr,AliDielectronVarManager::kCentrality,AliDielectronVarManager::kEta);
286 AliDielectronPID::SetWidthCorrFunction(fWdthCorr,AliDielectronVarManager::kCentrality);