Update (Andrea)
[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliRDHFCutsLctopKpi.cxx
CommitLineData
e3d40058 1/**************************************************************************
2 * Copyright(c) 1998-2010, 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
27de2dfb 16/* $Id$ */
17
e3d40058 18/////////////////////////////////////////////////////////////
19//
20// Class for cuts on AOD reconstructed Lc->pKpi
21//
22// Author: A.Dainese, andrea.dainese@pd.infn.it
23/////////////////////////////////////////////////////////////
24
25#include <TDatabasePDG.h>
26#include <Riostream.h>
27
28#include "AliRDHFCutsLctopKpi.h"
29#include "AliAODRecoDecayHF3Prong.h"
7ad4b782 30#include "AliRDHFCuts.h"
e3d40058 31#include "AliAODTrack.h"
32#include "AliESDtrack.h"
7ad4b782 33#include "AliKFParticle.h"
34#include "AliESDVertex.h"
e3d40058 35
36ClassImp(AliRDHFCutsLctopKpi)
37
38//--------------------------------------------------------------------------
a9b75906 39AliRDHFCutsLctopKpi::AliRDHFCutsLctopKpi(const char* name) :
7ad4b782 40AliRDHFCuts(name),
41fPidObjprot(0),
42fPidObjpion(0),
43fRecoKF(kFALSE)
e3d40058 44{
45 //
46 // Default Constructor
47 //
a7b533b7 48 Int_t nvars=13;
e3d40058 49 SetNVars(nvars);
a7b533b7 50 TString varNames[13]={"inv. mass [GeV]",
a8b8e864 51 "pTK [GeV/c]",
a7b533b7 52 "pTP [GeV/c]",
a8b8e864 53 "d0K [cm] lower limit!",
e3d40058 54 "d0Pi [cm] lower limit!",
55 "dist12 (cm)",
56 "sigmavert (cm)",
57 "dist prim-sec (cm)",
58 "pM=Max{pT1,pT2,pT3} (GeV/c)",
59 "cosThetaPoint",
60 "Sum d0^2 (cm^2)",
a7b533b7 61 "dca cut (cm)",
62 "cut on pTpion [GeV/c]"};
63 Bool_t isUpperCut[13]={kTRUE,
e3d40058 64 kFALSE,
65 kFALSE,
66 kFALSE,
67 kFALSE,
68 kFALSE,
69 kTRUE,
70 kFALSE,
71 kFALSE,
72 kFALSE,
73 kFALSE,
a7b533b7 74 kTRUE,
75 kFALSE
76 };
e3d40058 77 SetVarNames(nvars,varNames,isUpperCut);
a7b533b7 78 Bool_t forOpt[13]={kFALSE,
79 kTRUE,
80 kTRUE,
e3d40058 81 kFALSE,
82 kFALSE,
83 kFALSE,
84 kFALSE,
85 kTRUE,
a7b533b7 86 kFALSE,
87 kFALSE,
88 kFALSE,
89 kFALSE,
90 kTRUE};
91 SetVarsForOpt(4,forOpt);
e3d40058 92 Float_t limits[2]={0,999999999.};
93 SetPtBins(2,limits);
94}
95//--------------------------------------------------------------------------
96AliRDHFCutsLctopKpi::AliRDHFCutsLctopKpi(const AliRDHFCutsLctopKpi &source) :
7ad4b782 97 AliRDHFCuts(source),
98 fPidObjprot(0),
99 fPidObjpion(0),
100 fRecoKF(kFALSE)
e3d40058 101{
102 //
103 // Copy constructor
104 //
7ad4b782 105 if(source.fPidObjprot) SetPidprot(source.fPidObjprot);
106 if(source.fPidObjpion) SetPidpion(source.fPidObjpion);
e3d40058 107}
108//--------------------------------------------------------------------------
109AliRDHFCutsLctopKpi &AliRDHFCutsLctopKpi::operator=(const AliRDHFCutsLctopKpi &source)
110{
111 //
112 // assignment operator
113 //
114 if(&source == this) return *this;
115
116 AliRDHFCuts::operator=(source);
7ad4b782 117 SetPidprot(source.GetPidprot());
118 SetPidpion(source.GetPidpion());
e3d40058 119
120 return *this;
121}
7ad4b782 122//---------------------------------------------------------------------------
123AliRDHFCutsLctopKpi::~AliRDHFCutsLctopKpi() {
124 //
125 // // Default Destructor
126 //
127 if(fPidObjpion){
128 delete fPidObjpion;
129 fPidObjpion=0;
130 }
131 if(fPidObjprot){
132 delete fPidObjprot;
133 fPidObjprot=0;
134 }
e3d40058 135
7ad4b782 136}
e3d40058 137
138//---------------------------------------------------------------------------
139void AliRDHFCutsLctopKpi::GetCutVarsForOpt(AliAODRecoDecayHF *d,Float_t *vars,Int_t nvars,Int_t *pdgdaughters) {
140 //
141 // Fills in vars the values of the variables
142 //
143
144 if(nvars!=fnVarsForOpt) {
145 printf("AliRDHFCutsLctopKpi::GetCutsVarsForOpt: wrong number of variables\n");
146 return;
147 }
148
149 AliAODRecoDecayHF3Prong *dd = (AliAODRecoDecayHF3Prong*)d;
150
151 Int_t iter=-1;
152 if(fVarsForOpt[0]){
153 iter++;
154 vars[iter]=dd->InvMassLcpKpi();
155 }
156 if(fVarsForOpt[1]){
157 iter++;
158 for(Int_t iprong=0;iprong<3;iprong++){
a7b533b7 159 if(TMath::Abs(pdgdaughters[iprong])==321) {
e3d40058 160 vars[iter]=dd->PtProng(iprong);
161 }
162 }
163 }
164 if(fVarsForOpt[2]){
165 iter++;
166 for(Int_t iprong=0;iprong<3;iprong++){
a7b533b7 167 if(TMath::Abs(pdgdaughters[iprong])==2212) {
e3d40058 168 vars[iter]=dd->PtProng(iprong);
169 }
170 }
171 }
172 if(fVarsForOpt[3]){
173 iter++;
174 for(Int_t iprong=0;iprong<3;iprong++){
175 if(TMath::Abs(pdgdaughters[iprong])==2212) {
176 vars[iter]=dd->Getd0Prong(iprong);
177 }
178 }
179 }
180 if(fVarsForOpt[4]){
181 iter++;
182 for(Int_t iprong=0;iprong<3;iprong++){
183 if(TMath::Abs(pdgdaughters[iprong])==211) {
184 vars[iter]=dd->Getd0Prong(iprong);
185 }
186 }
187 }
188 if(fVarsForOpt[5]){
189 iter++;
190 vars[iter]=dd->GetDist12toPrim();
191 }
192 if(fVarsForOpt[6]){
193 iter++;
194 vars[iter]=dd->GetSigmaVert();
195 }
196 if(fVarsForOpt[7]){
197 iter++;
198 vars[iter] = dd->DecayLength();
199 }
200 if(fVarsForOpt[8]){
201 iter++;
202 Float_t ptmax=0;
203 for(Int_t i=0;i<3;i++){
204 if(dd->PtProng(i)>ptmax)ptmax=dd->PtProng(i);
205 }
206 vars[iter]=ptmax;
207 }
208 if(fVarsForOpt[9]){
209 iter++;
210 vars[iter]=dd->CosPointingAngle();
211 }
212 if(fVarsForOpt[10]){
213 iter++;
214 vars[iter]=dd->Getd0Prong(0)*dd->Getd0Prong(0)+dd->Getd0Prong(1)*dd->Getd0Prong(1)+dd->Getd0Prong(2)*dd->Getd0Prong(2);
215 }
216 if(fVarsForOpt[11]){
217 iter++;
218 vars[iter]=dd->GetDCA();
219 }
a7b533b7 220 if(fVarsForOpt[12]){
221 iter++;
222 for(Int_t iprong=0;iprong<3;iprong++){
223 if(TMath::Abs(pdgdaughters[iprong])==211) {
224 vars[iter]=dd->PtProng(iprong);
225 }
226 }
227 }
e3d40058 228
229 return;
230}
231//---------------------------------------------------------------------------
7ad4b782 232Int_t AliRDHFCutsLctopKpi::IsSelected(TObject* obj,Int_t selectionLevel,AliAODEvent *aod) {
e3d40058 233 //
234 // Apply selection
235 //
236
237 if(!fCutsRD){
238 cout<<"Cut matrice not inizialized. Exit..."<<endl;
239 return 0;
240 }
241 //PrintAll();
242 AliAODRecoDecayHF3Prong* d=(AliAODRecoDecayHF3Prong*)obj;
243
244 if(!d){
245 cout<<"AliAODRecoDecayHF3Prong null"<<endl;
246 return 0;
247 }
248
249
5e938293 250 if(fKeepSignalMC) if(IsSignalMC(d,aod,4122)) return 3;
e3d40058 251
228974b3 252 Int_t returnvalue=3;
7ad4b782 253 Int_t returnvaluePID=3;
254
47aa3d55 255 if(d->Pt()<fMinPtCand) return 0;
256 if(d->Pt()>fMaxPtCand) return 0;
e3d40058 257
258 // selection on candidate
259 if(selectionLevel==AliRDHFCuts::kAll ||
260 selectionLevel==AliRDHFCuts::kCandidate) {
261
262 Double_t pt=d->Pt();
263
264 Int_t ptbin=PtBin(pt);
265
266 Double_t mLcpKpi,mLcpiKp;
267 Int_t okLcpKpi=1,okLcpiKp=1;
268
269 Double_t mLcPDG = TDatabasePDG::Instance()->GetParticle(4122)->Mass();
270
271 mLcpKpi=d->InvMassLcpKpi();
272 mLcpiKp=d->InvMassLcpiKp();
273
274 if(TMath::Abs(mLcpKpi-mLcPDG)>fCutsRD[GetGlobalIndex(0,ptbin)]) okLcpKpi = 0;
275 if(TMath::Abs(mLcpiKp-mLcPDG)>fCutsRD[GetGlobalIndex(0,ptbin)]) okLcpiKp = 0;
276 if(!okLcpKpi && !okLcpiKp) return 0;
277
278 if(TMath::Abs(d->PtProng(1)) < fCutsRD[GetGlobalIndex(1,ptbin)] || TMath::Abs(d->Getd0Prong(1))<fCutsRD[GetGlobalIndex(3,ptbin)]) return 0;//Kaon
a7b533b7 279 if((TMath::Abs(d->PtProng(0)) < fCutsRD[GetGlobalIndex(2,ptbin)]) || (TMath::Abs(d->PtProng(2)) < fCutsRD[GetGlobalIndex(12,ptbin)])) okLcpKpi=0;
280 if((TMath::Abs(d->PtProng(2)) < fCutsRD[GetGlobalIndex(2,ptbin)]) || (TMath::Abs(d->PtProng(0)) < fCutsRD[GetGlobalIndex(12,ptbin)]))okLcpiKp=0;
7ad4b782 281 if(!okLcpKpi && !okLcpiKp) return 0;
e3d40058 282
283
7ad4b782 284 if(fRecoKF){
285 Int_t valueTmp=3;
286 if(okLcpKpi) returnvalue=1; //cuts passed as Lc->pKpi
287 if(okLcpiKp) returnvalue=2; //cuts passed as Lc->piKp
288 if(okLcpKpi && okLcpiKp) returnvalue=3; //cuts passed as both pKpi and piKp
289
290 Int_t valueTotTmp=CombinePIDCuts(valueTmp,returnvaluePID);
291 Int_t pdgs[3]={2212,321,211};
292 if(valueTotTmp>=2) {
293 pdgs[0]=211;
294 pdgs[2]=2212;
295 }
296 if(!d->GetOwnPrimaryVtx()){
297 AliAODVertex *vtx1 = (AliAODVertex*)aod->GetPrimaryVertex();
298 d->SetOwnPrimaryVtx(vtx1);
299 }
300 Double_t field=aod->GetMagneticField();
301 ReconstructKF(d,pdgs,field);
302 }
e3d40058 303 //2track cuts
304 if(d->GetDist12toPrim()<fCutsRD[GetGlobalIndex(5,ptbin)]|| d->GetDist23toPrim()<fCutsRD[GetGlobalIndex(5,ptbin)]) return 0;
228974b3 305 if(d->GetDist12toPrim()>1.) return 0;
306 if(d->GetDist23toPrim()>1.) return 0;
e3d40058 307 if(d->Getd0Prong(0)*d->Getd0Prong(1)<0. && d->Getd0Prong(2)*d->Getd0Prong(1)<0.) return 0;
308
309 //sec vert
310 if(d->GetSigmaVert()>fCutsRD[GetGlobalIndex(6,ptbin)]) return 0;
311
312 if(d->DecayLength()<fCutsRD[GetGlobalIndex(7,ptbin)]) return 0;
228974b3 313 if(d->DecayLength()>0.5) return 0;
e3d40058 314
315 if(TMath::Abs(d->PtProng(0))<fCutsRD[GetGlobalIndex(8,ptbin)] && TMath::Abs(d->PtProng(1))<fCutsRD[GetGlobalIndex(8,ptbin)] && TMath::Abs(d->PtProng(2))<fCutsRD[GetGlobalIndex(8,ptbin)]) return 0;
316 if(d->CosPointingAngle()< fCutsRD[GetGlobalIndex(9,ptbin)]) return 0;
317 Double_t sum2=d->Getd0Prong(0)*d->Getd0Prong(0)+d->Getd0Prong(1)*d->Getd0Prong(1)+d->Getd0Prong(2)*d->Getd0Prong(2);
318 if(sum2<fCutsRD[GetGlobalIndex(10,ptbin)]) return 0;
319
320 //DCA
321 for(Int_t i=0;i<3;i++) if(d->GetDCA(i)>fCutsRD[GetGlobalIndex(11,ptbin)]) return 0;
322
323
324 if(okLcpKpi) returnvalue=1; //cuts passed as Lc->pKpi
325 if(okLcpiKp) returnvalue=2; //cuts passed as Lc->piKp
326 if(okLcpKpi && okLcpiKp) returnvalue=3; //cuts passed as both pKpi and piKp
dcc2ade0 327
dcc2ade0 328 }
329
1a141544 330 if(selectionLevel==AliRDHFCuts::kAll ||
331 selectionLevel==AliRDHFCuts::kCandidate||
332 selectionLevel==AliRDHFCuts::kPID) {
333 returnvaluePID = IsSelectedPID(d);
334 fIsSelectedPID=returnvaluePID;
335 }
11690a06 336 // if(fUsePID || selectionLevel==AliRDHFCuts::kPID) returnvaluePID = IsSelectedCombinedPID(d); // to test!!
dcc2ade0 337 if(returnvaluePID==0) return 0;
338
339 // selection on daughter tracks
340 if(selectionLevel==AliRDHFCuts::kAll ||
341 selectionLevel==AliRDHFCuts::kTracks) {
342 if(!AreDaughtersSelected(d)) return 0;
e3d40058 343 }
dcc2ade0 344
345
7ad4b782 346 Int_t returnvalueTot=CombinePIDCuts(returnvalue,returnvaluePID);
347 return returnvalueTot;
e3d40058 348}
349//---------------------------------------------------------------------------
7ad4b782 350Int_t AliRDHFCutsLctopKpi::IsSelectedPID(AliAODRecoDecayHF* obj) {
351
352
1a141544 353 if(!fUsePID || !obj) return 3;
7ad4b782 354 Int_t okLcpKpi=0,okLcpiKp=0;
355 Int_t returnvalue=0;
356 Bool_t isPeriodd=fPidHF->GetOnePad();
a7b533b7 357 Bool_t isMC=fPidHF->GetMC();
7ad4b782 358 Bool_t ispion0=kTRUE,ispion2=kTRUE;
359 Bool_t isproton0=kFALSE,isproton2=kFALSE;
360 Bool_t iskaon1=kFALSE;
a7b533b7 361 if(isPeriodd) {
362 fPidObjprot->SetOnePad(kTRUE);
363 fPidObjpion->SetOnePad(kTRUE);
364 }
365 if(isMC) {
366 fPidObjprot->SetMC(kTRUE);
367 fPidObjpion->SetMC(kTRUE);
368 }
7ad4b782 369
370 for(Int_t i=0;i<3;i++){
371 AliAODTrack *track=(AliAODTrack*)obj->GetDaughter(i);
372 if(!track) return 0;
373 // identify kaon
374 if(i==1) {
375 Int_t isKaon=fPidHF->MakeRawPid(track,3);
376 if(isKaon>=1) {
377 iskaon1=kTRUE;
378 if(fPidHF->MakeRawPid(track,2)>=1) iskaon1=kFALSE;
379 }
47aa3d55 380 if(!iskaon1) return 0;
7ad4b782 381
382 }else{
383 //pion or proton
384
385 Int_t isProton=fPidObjprot->MakeRawPid(track,4);
386 if(isProton>=1){
387 if(fPidHF->MakeRawPid(track,2)>=1) isProton=-1;
388 if(fPidHF->MakeRawPid(track,3)>=1) isProton=-1;
389 }
390
391 Int_t isPion=fPidObjpion->MakeRawPid(track,2);
392 if(fPidHF->MakeRawPid(track,3)>=1) isPion=-1;
393 if(fPidObjprot->MakeRawPid(track,4)>=1) isPion=-1;
394
395
396 if(i==0) {
397 if(isPion<0) ispion0=kFALSE;
398 if(isProton>=1) isproton0=kTRUE;
399
400 }
47aa3d55 401 if(!ispion0 && !isproton0) return 0;
7ad4b782 402 if(i==2) {
403 if(isPion<0) ispion2=kFALSE;
404 if(isProton>=1) isproton2=kTRUE;
405 }
406
407 }
408 }
409
410 if(ispion2 && isproton0 && iskaon1) okLcpKpi=1;
411 if(ispion0 && isproton2 && iskaon1) okLcpiKp=1;
412 if(okLcpKpi) returnvalue=1; //cuts passed as Lc->pKpi
413 if(okLcpiKp) returnvalue=2; //cuts passed as Lc->piKp
414 if(okLcpKpi && okLcpiKp) returnvalue=3; //cuts passed as both pKpi and piKp
415
416 return returnvalue;
417}
11690a06 418//---------------------------------------------------------------------------
419Int_t AliRDHFCutsLctopKpi::IsSelectedCombinedPID(AliAODRecoDecayHF* obj) {
420
421 // Printf(" -------- IsSelectedCombinedPID --------------");
422
423 if(!obj) {return 3;}
424 Int_t okLcpKpi=0,okLcpiKp=0;
425 Int_t returnvalue=0;
426 Bool_t isPeriodd=fPidHF->GetOnePad();
427 Bool_t isMC=fPidHF->GetMC();
428 Bool_t isKaon = kFALSE;
429 Bool_t isPion = kFALSE;
430 Bool_t isProton = kFALSE;
431
432 if(isPeriodd) {
433 fPidObjprot->SetOnePad(kTRUE);
434 fPidObjpion->SetOnePad(kTRUE);
435 }
436 if(isMC) {
437 fPidObjprot->SetMC(kTRUE);
438 fPidObjpion->SetMC(kTRUE);
439 }
440
441 Double_t probComb[AliPID::kSPECIES]={0.}; // array to store the info for the combined ITS|TPC|TOF probabilities
442 fPidHF->GetPidCombined()->SetDetectorMask(AliPIDResponse::kDetITS|AliPIDResponse::kDetTOF|AliPIDResponse::kDetTPC); // the mask could become a member of the cut object
443
444 for(Int_t i=0;i<3;i++){
445 AliAODTrack *track=(AliAODTrack*)obj->GetDaughter(i);
446 if(!track) return 0;
447 // identify kaon
2fd6ad5e 448 /*UInt_t detUsed =*/ fPidHF->GetPidCombined()->ComputeProbabilities(track, fPidHF->GetPidResponse(), probComb); // for the moment we don't check detUsed...
11690a06 449 // Printf("[%d] %x %f %f %f %f %f",i,detUsed,probComb[0],probComb[1],probComb[2],probComb[3],probComb[4]);
450 if(i==1) {
451 if(TMath::MaxElement(AliPID::kSPECIES,probComb) == probComb[3]) { // the probability to be a Kaon is the highest
452 isKaon=kTRUE;
453 // if ( probComb[3] > 0.8 ) isKaon=kTRUE;
454 // else return 0;
455 }
456 else return 0; // prong at position 1 is not a Kaon, returning 0
457 }
458 else {
459 //pion or proton
460
461 if(TMath::MaxElement(AliPID::kSPECIES,probComb) == probComb[4]) { // the probability to be a proton is the highest
462 isProton=kTRUE;
463 if (isPion) okLcpiKp = 1; // the pion was already identified, so here we must be at i == 2 --> Lc --> pi K p (K already found)
464 }
465 if(TMath::MaxElement(AliPID::kSPECIES,probComb) == probComb[2]) { // the probability to be a pion is the highest
466 isPion=kTRUE;
467 if (isProton) okLcpKpi = 1; // the proton was already identified, so here we must be at i == 2 --> Lc --> p K pi (K already found)
468 }
469
470 }
471 }
472
473 if(okLcpKpi) returnvalue=1; //cuts passed as Lc->pKpi
474 if(okLcpiKp) returnvalue=2; //cuts passed as Lc->piKp
475 if(okLcpKpi && okLcpiKp) returnvalue=3; //cuts passed as both pKpi and piKp
476
477 return returnvalue;
478}
7ad4b782 479//-----------------------
480Int_t AliRDHFCutsLctopKpi::CombinePIDCuts(Int_t returnvalue, Int_t returnvaluePID) const {
481
482 Int_t returnvalueTot=0;
483 Int_t okLcpKpi=0,okLcpiKp=0;
484 if(returnvaluePID==1){
485 if(returnvalue==1 || returnvalue==3) okLcpKpi=1;
486 }
487 if(returnvaluePID==2){
488 if(returnvalue>=2) okLcpiKp=1;
489 }
490 if(returnvaluePID==3 && returnvalue>0){
491 if(returnvalue==1 || returnvalue==3) okLcpKpi=1;
492 if(returnvalue>=2) okLcpiKp=1;
493 }
494
495 if(okLcpKpi) returnvalueTot=1; //cuts passed as Lc->pKpi
496 if(okLcpiKp) returnvalueTot=2; //cuts passed as Lc->piKp
497 if(okLcpKpi && okLcpiKp) returnvalueTot=3; //cuts passed as both pKpi and piKp
498 return returnvalueTot;
499}
500//----------------------------------
501void AliRDHFCutsLctopKpi::SetStandardCutsPP2010() {
502
503 SetName("LctopKpiProdCuts");
504 SetTitle("Production cuts for Lc analysis");
505
506 AliESDtrackCuts *esdTrackCuts = new AliESDtrackCuts("AliESDtrackCuts","default");
507 esdTrackCuts->SetRequireSigmaToVertex(kFALSE);
508 esdTrackCuts->SetRequireTPCRefit(kTRUE);
509 esdTrackCuts->SetMinNClustersTPC(70);
510 esdTrackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD,
511 AliESDtrackCuts::kAny);
512 esdTrackCuts->SetRequireITSRefit(kTRUE);
513 esdTrackCuts->SetMinNClustersITS(4);
514 esdTrackCuts->SetMinDCAToVertexXY(0.);
515 esdTrackCuts->SetEtaRange(-0.8,0.8);
516 esdTrackCuts->SetPtRange(0.3,1.e10);
517 AddTrackCuts(esdTrackCuts);
518
519 const Int_t nptbins=4;
a7b533b7 520 const Int_t nvars=13;
7ad4b782 521 Float_t* ptbins;
522 ptbins=new Float_t[nptbins+1];
523
524 ptbins[0]=0.;
525 ptbins[1]=2.;
526 ptbins[2]=3.;
527 ptbins[3]=4.;
528 ptbins[4]=9999.;
529
530 SetGlobalIndex(nvars,nptbins);
531 SetPtBins(nptbins+1,ptbins);
532
533 Float_t** prodcutsval;
534 prodcutsval=new Float_t*[nvars];
535 for(Int_t iv=0;iv<nvars;iv++){
536 prodcutsval[iv]=new Float_t[nptbins];
537 }
e11ae259 538
7ad4b782 539 for(Int_t ipt=0;ipt<nptbins;ipt++){
540 prodcutsval[0][ipt]=0.18;
541 prodcutsval[1][ipt]=0.4;
542 prodcutsval[2][ipt]=0.5;
543 prodcutsval[3][ipt]=0.;
544 prodcutsval[4][ipt]=0.;
545 prodcutsval[5][ipt]=0.01;
546 prodcutsval[6][ipt]=0.06;
547 prodcutsval[7][ipt]=0.005;
548 prodcutsval[8][ipt]=0.;
549 prodcutsval[9][ipt]=0.;
550 prodcutsval[10][ipt]=0.;
551 prodcutsval[11][ipt]=0.05;
a7b533b7 552 prodcutsval[12][ipt]=0.4;
7ad4b782 553 }
554 SetCuts(nvars,nptbins,prodcutsval);
555
556 AliAODPidHF* pidObjK=new AliAODPidHF();
557 Double_t sigmasK[5]={3.,1.,1.,3.,2.};
558 pidObjK->SetSigma(sigmasK);
559 pidObjK->SetAsym(kTRUE);
560 pidObjK->SetMatch(1);
561 pidObjK->SetTPC(kTRUE);
562 pidObjK->SetTOF(kTRUE);
563 pidObjK->SetITS(kTRUE);
564 Double_t plimK[2]={0.5,0.8};
565 pidObjK->SetPLimit(plimK,2);
a7b533b7 566 pidObjK->SetTOFdecide(kTRUE);
7ad4b782 567
568 SetPidHF(pidObjK);
569
570 AliAODPidHF* pidObjpi=new AliAODPidHF();
571 pidObjpi->SetTPC(kTRUE);
572 Double_t sigmaspi[5]={3.,0.,0.,0.,0.};
573 pidObjpi->SetSigma(sigmaspi);
a7b533b7 574 pidObjpi->SetTOFdecide(kTRUE);
7ad4b782 575 SetPidpion(pidObjpi);
576
577 AliAODPidHF* pidObjp=new AliAODPidHF();
578 Double_t sigmasp[5]={3.,1.,1.,3.,2.};
579 pidObjp->SetSigma(sigmasp);
580 pidObjp->SetAsym(kTRUE);
581 pidObjp->SetMatch(1);
582 pidObjp->SetTPC(kTRUE);
583 pidObjp->SetTOF(kTRUE);
584 pidObjp->SetITS(kTRUE);
585 Double_t plimp[2]={1.,2.};
586 pidObjp->SetPLimit(plimp,2);
a7b533b7 587 pidObjp->SetTOFdecide(kTRUE);
7ad4b782 588
589 SetPidprot(pidObjp);
590
591 SetUsePID(kTRUE);
592
593 PrintAll();
594
e11ae259 595 for(Int_t iiv=0;iiv<nvars;iiv++){
596 delete [] prodcutsval[iiv];
597 }
598 delete [] prodcutsval;
599 prodcutsval=NULL;
600 delete [] ptbins;
601 ptbins=NULL;
602
603 delete pidObjp;
604 pidObjp=NULL;
605
7ad4b782 606 return;
607}
a8b8e864 608//------------------
609void AliRDHFCutsLctopKpi::SetStandardCutsPbPb2010() {
610
611 SetName("LctopKpiProdCuts");
612 SetTitle("Production cuts for Lc analysis");
613
614 AliESDtrackCuts *esdTrackCuts = new AliESDtrackCuts("AliESDtrackCuts","default");
615
616 esdTrackCuts->SetRequireTPCRefit(kTRUE);
617 esdTrackCuts->SetMinNClustersTPC(70);
618 esdTrackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD,
619 AliESDtrackCuts::kAny);
620 esdTrackCuts->SetRequireITSRefit(kTRUE);
621 esdTrackCuts->SetMinNClustersITS(4);
622 esdTrackCuts->SetMinDCAToVertexXYPtDep("0.0100*TMath::Max(0.,(1-TMath::Floor(TMath::Abs(pt)/2.)))");
623 esdTrackCuts->SetEtaRange(-0.8,0.8);
624 esdTrackCuts->SetMaxDCAToVertexXY(1.);
625 esdTrackCuts->SetMaxDCAToVertexZ(1.);
626 esdTrackCuts->SetPtRange(0.8,1.e10);
627 AddTrackCuts(esdTrackCuts);
628
629 const Int_t nptbins=4;
a7b533b7 630 const Int_t nvars=13;
a8b8e864 631 Float_t* ptbins;
632 ptbins=new Float_t[nptbins+1];
633
634 ptbins[0]=0.;
635 ptbins[1]=2.;
636 ptbins[2]=3.;
637 ptbins[3]=4.;
638 ptbins[4]=9999.;
639
640 SetGlobalIndex(nvars,nptbins);
641 SetPtBins(nptbins+1,ptbins);
642
643 Float_t** prodcutsval;
644 prodcutsval=new Float_t*[nvars];
645 for(Int_t iv=0;iv<nvars;iv++){
646 prodcutsval[iv]=new Float_t[nptbins];
647 }
648
649 for(Int_t ipt=0;ipt<nptbins;ipt++){
650 prodcutsval[0][ipt]=0.13;
651 prodcutsval[1][ipt]=0.5;
652 prodcutsval[2][ipt]=0.6;
653 prodcutsval[3][ipt]=0.;
654 prodcutsval[4][ipt]=0.;
655 prodcutsval[5][ipt]=0.01;
656 prodcutsval[6][ipt]=0.04;
657 prodcutsval[7][ipt]=0.006;
658 prodcutsval[8][ipt]=0.8;
659 prodcutsval[9][ipt]=0.3;
660 prodcutsval[10][ipt]=0.;
661 prodcutsval[11][ipt]=0.05;
a7b533b7 662 prodcutsval[12][ipt]=0.4;
a8b8e864 663 }
664 SetCuts(nvars,nptbins,prodcutsval);
665
666 AliAODPidHF* pidObjK=new AliAODPidHF();
667 Double_t sigmasK[5]={3.,1.,1.,3.,2.};
668 pidObjK->SetSigma(sigmasK);
669 pidObjK->SetAsym(kTRUE);
670 pidObjK->SetMatch(1);
671 pidObjK->SetTPC(kTRUE);
672 pidObjK->SetTOF(kTRUE);
673 pidObjK->SetITS(kTRUE);
674 Double_t plimK[2]={0.5,0.8};
675 pidObjK->SetPLimit(plimK,2);
676
677 SetPidHF(pidObjK);
678
679 AliAODPidHF* pidObjpi=new AliAODPidHF();
680 pidObjpi->SetTPC(kTRUE);
681 Double_t sigmaspi[5]={3.,0.,0.,0.,0.};
682 pidObjpi->SetSigma(sigmaspi);
683 SetPidpion(pidObjpi);
7ad4b782 684
a8b8e864 685 AliAODPidHF* pidObjp=new AliAODPidHF();
686 Double_t sigmasp[5]={3.,1.,1.,3.,2.};
687 pidObjp->SetSigma(sigmasp);
688 pidObjp->SetAsym(kTRUE);
689 pidObjp->SetMatch(1);
690 pidObjp->SetTPC(kTRUE);
691 pidObjp->SetTOF(kTRUE);
692 pidObjp->SetITS(kTRUE);
693 Double_t plimp[2]={1.,2.};
694 pidObjp->SetPLimit(plimp,2);
695
696 SetPidprot(pidObjp);
697
698 SetUsePID(kTRUE);
699
700 PrintAll();
701
702 for(Int_t iiv=0;iiv<nvars;iiv++){
703 delete [] prodcutsval[iiv];
704 }
705 delete [] prodcutsval;
706 prodcutsval=NULL;
707 delete [] ptbins;
708 ptbins=NULL;
709
710 delete pidObjp;
711 pidObjp=NULL;
712
713 return;
714}
7ad4b782 715//------------------
716Bool_t AliRDHFCutsLctopKpi::ReconstructKF(AliAODRecoDecayHF3Prong *d,Int_t *pdgs,Double_t field) const{
717
718 Int_t nprongs=d->GetNProngs();
719 Int_t iprongs[nprongs];
720 for(Int_t i=0;i<nprongs;i++) iprongs[i]=i;
721
722 Double_t mass[2]={0.,0.};
723
724 AliKFParticle *decay=d->ApplyVertexingKF(iprongs,nprongs,pdgs,kTRUE,field,mass);
725 if(!decay) return kTRUE;
726 AliESDVertex *vertexESD = new AliESDVertex(decay->Parameters(),
727 decay->CovarianceMatrix(),
728 decay->GetChi2(),
729 nprongs);
730 Double_t pos[3],cov[6],chi2perNDF;
731 vertexESD->GetXYZ(pos);
732 vertexESD->GetCovMatrix(cov);
733 chi2perNDF = vertexESD->GetChi2toNDF();
734 delete vertexESD; vertexESD=NULL;
735 AliAODVertex *vertexAOD = new AliAODVertex(pos,cov,chi2perNDF,0x0,-1,AliAODVertex::kUndef,nprongs);
736 d->SetSecondaryVtx(vertexAOD);
737 return kTRUE;
738}
a8b8e864 739