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