]>
Commit | Line | Data |
---|---|---|
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 Jpsi->ee | |
21 | // | |
22 | // Author: A.Dainese, andrea.dainese@pd.infn.it | |
23 | ///////////////////////////////////////////////////////////// | |
24 | ||
25 | #include <TDatabasePDG.h> | |
26 | #include <Riostream.h> | |
27 | ||
28 | #include "AliRDHFCutsJpsitoee.h" | |
29 | #include "AliAODRecoDecayHF2Prong.h" | |
30 | #include "AliAODTrack.h" | |
31 | #include "AliESDtrack.h" | |
32 | ||
33 | ClassImp(AliRDHFCutsJpsitoee) | |
34 | ||
35 | //-------------------------------------------------------------------------- | |
a9b75906 | 36 | AliRDHFCutsJpsitoee::AliRDHFCutsJpsitoee(const char* name) : |
37 | AliRDHFCuts(name) | |
e3d40058 | 38 | { |
39 | // | |
40 | // Default Constructor | |
41 | // | |
42 | Int_t nvars=9; | |
43 | SetNVars(nvars); | |
44 | TString varNames[9]={"inv. mass [GeV]", | |
45 | "dca [cm]", | |
46 | "cosThetaStar", // negative electron | |
47 | "pTP [GeV/c]", | |
48 | "pTN [GeV/c]", | |
49 | "d0P [cm]", | |
50 | "d0N [cm]", | |
51 | "d0d0 [cm^2]", | |
52 | "cosThetaPoint"}; | |
53 | Bool_t isUpperCut[9]={kTRUE, | |
54 | kTRUE, | |
55 | kTRUE, | |
56 | kFALSE, | |
57 | kFALSE, | |
58 | kTRUE, | |
59 | kTRUE, | |
60 | kTRUE, | |
61 | kFALSE}; | |
62 | SetVarNames(nvars,varNames,isUpperCut); | |
63 | Bool_t forOpt[9]={kFALSE, | |
64 | kTRUE, | |
65 | kTRUE, | |
66 | kFALSE, | |
67 | kFALSE, | |
68 | kFALSE, | |
69 | kFALSE, | |
70 | kTRUE, | |
71 | kTRUE}; | |
72 | SetVarsForOpt(4,forOpt); | |
73 | Float_t limits[2]={0,999999999.}; | |
74 | SetPtBins(2,limits); | |
75 | } | |
76 | //-------------------------------------------------------------------------- | |
77 | AliRDHFCutsJpsitoee::AliRDHFCutsJpsitoee(const AliRDHFCutsJpsitoee &source) : | |
78 | AliRDHFCuts(source) | |
79 | { | |
80 | // | |
81 | // Copy constructor | |
82 | // | |
83 | ||
84 | } | |
85 | //-------------------------------------------------------------------------- | |
86 | AliRDHFCutsJpsitoee &AliRDHFCutsJpsitoee::operator=(const AliRDHFCutsJpsitoee &source) | |
87 | { | |
88 | // | |
89 | // assignment operator | |
90 | // | |
91 | if(&source == this) return *this; | |
92 | ||
93 | AliRDHFCuts::operator=(source); | |
94 | ||
95 | return *this; | |
96 | } | |
97 | ||
98 | ||
99 | //--------------------------------------------------------------------------- | |
100 | void AliRDHFCutsJpsitoee::GetCutVarsForOpt(AliAODRecoDecayHF *d,Float_t *vars,Int_t nvars,Int_t *pdgdaughters) { | |
101 | // | |
102 | // Fills in vars the values of the variables | |
103 | // | |
104 | ||
105 | if(nvars!=fnVarsForOpt) { | |
106 | printf("AliRDHFCutsJpsitoee::GetCutsVarsForOpt: wrong number of variables\n"); | |
107 | return; | |
108 | } | |
109 | ||
110 | AliAODRecoDecayHF2Prong *dd = (AliAODRecoDecayHF2Prong*)d; | |
e3d40058 | 111 | |
112 | Int_t iter=-1; | |
113 | if(fVarsForOpt[0]){ | |
114 | iter++; | |
115 | vars[iter]=dd->InvMassJPSIee(); | |
116 | } | |
117 | if(fVarsForOpt[1]){ | |
118 | iter++; | |
119 | vars[iter]=dd->GetDCA(); | |
120 | } | |
121 | if(fVarsForOpt[2]){ | |
122 | iter++; | |
123 | vars[iter] = dd->CosThetaStarJPSI(); | |
124 | } | |
125 | if(fVarsForOpt[3]){ | |
126 | iter++; | |
4755453e | 127 | if(pdgdaughters[0]==11) vars[iter]=dd->PtProng(0); |
e3d40058 | 128 | } |
129 | if(fVarsForOpt[4]){ | |
130 | iter++; | |
4755453e | 131 | if(pdgdaughters[1]==11) vars[iter]=dd->PtProng(1); |
e3d40058 | 132 | } |
133 | if(fVarsForOpt[5]){ | |
134 | iter++; | |
135 | vars[iter]=dd->Getd0Prong(0); | |
136 | } | |
137 | if(fVarsForOpt[6]){ | |
138 | iter++; | |
139 | vars[iter]=dd->Getd0Prong(1); | |
140 | } | |
141 | if(fVarsForOpt[7]){ | |
142 | iter++; | |
143 | vars[iter]= dd->Prodd0d0(); | |
144 | } | |
145 | if(fVarsForOpt[8]){ | |
146 | iter++; | |
147 | vars[iter]=dd->CosPointingAngle(); | |
148 | } | |
149 | ||
150 | return; | |
151 | } | |
152 | //--------------------------------------------------------------------------- | |
153 | Int_t AliRDHFCutsJpsitoee::IsSelected(TObject* obj,Int_t selectionLevel) { | |
154 | // | |
155 | // Apply selection | |
156 | // | |
157 | ||
158 | if(!fCutsRD){ | |
4755453e | 159 | cout<<"Cut matrix not inizialized. Exit..."<<endl; |
e3d40058 | 160 | return 0; |
161 | } | |
162 | //PrintAll(); | |
163 | AliAODRecoDecayHF2Prong* d=(AliAODRecoDecayHF2Prong*)obj; | |
164 | ||
165 | if(!d){ | |
166 | cout<<"AliAODRecoDecayHF2Prong null"<<endl; | |
167 | return 0; | |
168 | } | |
169 | ||
ba27eecd | 170 | if(d->HasBadDaughters()) return 0; |
171 | ||
e3d40058 | 172 | |
173 | // selection on daughter tracks | |
174 | if(selectionLevel==AliRDHFCuts::kAll || | |
175 | selectionLevel==AliRDHFCuts::kTracks) { | |
176 | if(!AreDaughtersSelected(d)) return 0; | |
177 | } | |
178 | ||
179 | ||
180 | // selection on candidate | |
181 | if(selectionLevel==AliRDHFCuts::kAll || | |
182 | selectionLevel==AliRDHFCuts::kCandidate) { | |
183 | ||
184 | Double_t pt=d->Pt(); | |
185 | Double_t mJPsi,ctsJPsi; | |
186 | ||
187 | Double_t mJPSIPDG = TDatabasePDG::Instance()->GetParticle(443)->Mass(); | |
188 | ||
189 | Int_t ptbin=PtBin(pt); | |
190 | ||
191 | if(d->PtProng(1) < fCutsRD[GetGlobalIndex(3,ptbin)] || d->PtProng(0) < fCutsRD[GetGlobalIndex(4,ptbin)]) return 0; | |
192 | ||
193 | if(TMath::Abs(d->Getd0Prong(1)) > fCutsRD[GetGlobalIndex(5,ptbin)] || | |
194 | TMath::Abs(d->Getd0Prong(0)) > fCutsRD[GetGlobalIndex(6,ptbin)]) return 0; | |
195 | ||
196 | ||
197 | if(d->GetDCA() > fCutsRD[GetGlobalIndex(1,ptbin)]) return 0; | |
198 | ||
199 | mJPsi=d->InvMassJPSIee(); | |
200 | if(TMath::Abs(mJPsi-mJPSIPDG) > fCutsRD[GetGlobalIndex(0,ptbin)]) return 0; | |
201 | ||
202 | ctsJPsi=d->CosThetaStarJPSI(); | |
203 | if(TMath::Abs(ctsJPsi) > fCutsRD[GetGlobalIndex(2,ptbin)]) return 0; | |
204 | ||
205 | if(d->Prodd0d0() > fCutsRD[GetGlobalIndex(7,ptbin)]) return 0; | |
206 | ||
207 | if(d->CosPointingAngle() < fCutsRD[GetGlobalIndex(8,ptbin)]) return 0; | |
e3d40058 | 208 | } |
209 | ||
e3d40058 | 210 | return 1; |
211 | } | |
212 | //--------------------------------------------------------------------------- |