]>
Commit | Line | Data |
---|---|---|
35b8d778 | 1 | /************************************************************************** |
2 | * Copyright(c) 1998-1999, 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 | /* $Id$ */ | |
16 | ||
17 | /* History of cvs commits: | |
18 | * | |
19 | * $Log$ | |
3788def4 | 20 | * Revision 1.1 2007/10/29 13:50:39 gustavo |
21 | * New class for particle selection | |
22 | * | |
35b8d778 | 23 | * |
24 | * | |
25 | */ | |
26 | ||
27 | //_________________________________________________________________________ | |
28 | // Class for plotting particle/cluster/track distributions without cuts | |
29 | // and select clusters/tracks/particles needed in the analysis | |
30 | // depending on PID criteria or other. | |
31 | // | |
32 | // | |
33 | //*-- Author: Gustavo Conesa (LNF-INFN) | |
34 | ////////////////////////////////////////////////////////////////////////////// | |
35 | ||
36 | ||
37 | // --- ROOT system --- | |
38 | #include <TParticle.h> | |
39 | #include <TH2.h> | |
40 | #include <TList.h> | |
41 | #include "Riostream.h" | |
42 | #include "TROOT.h" | |
43 | ||
44 | // --- AliRoot system --- | |
45 | #include "AliAnaGammaSelection.h" | |
46 | #include "AliLog.h" | |
47 | ||
48 | ClassImp(AliAnaGammaSelection) | |
49 | ||
50 | //____________________________________________________________________________ | |
51 | AliAnaGammaSelection::AliAnaGammaSelection() : | |
52 | TObject(), fAnaMC(0), fFillCTS(0), | |
53 | fntEMCAL(0), fntPHOS(0), fntCTS(0) | |
54 | { | |
55 | //default ctor | |
56 | ||
57 | //Initialize parameters | |
58 | InitParameters(); | |
59 | ||
60 | } | |
61 | ||
62 | //____________________________________________________________________________ | |
63 | AliAnaGammaSelection::AliAnaGammaSelection(const AliAnaGammaSelection & g) : | |
64 | TObject(g), fAnaMC(g.fAnaMC), fFillCTS(g.fFillCTS), | |
65 | fntEMCAL(g.fntEMCAL), fntPHOS(g.fntPHOS), fntCTS(g.fntCTS) | |
66 | { | |
67 | // cpy ctor | |
68 | ||
69 | } | |
70 | ||
71 | //_________________________________________________________________________ | |
72 | AliAnaGammaSelection & AliAnaGammaSelection::operator = (const AliAnaGammaSelection & source) | |
73 | { | |
74 | // assignment operator | |
75 | ||
76 | if(&source == this) return *this; | |
77 | ||
78 | fAnaMC = source.fAnaMC ; | |
79 | fFillCTS = source.fFillCTS ; | |
80 | fntEMCAL = source.fntEMCAL ; | |
81 | fntPHOS = source.fntPHOS ; | |
82 | fntCTS = source.fntCTS ; | |
83 | ||
84 | return *this; | |
85 | ||
86 | } | |
87 | ||
88 | //____________________________________________________________________________ | |
89 | AliAnaGammaSelection::~AliAnaGammaSelection() | |
90 | { | |
3788def4 | 91 | // Remove all pointers except analysis output pointers. |
35b8d778 | 92 | |
93 | } | |
94 | ||
95 | //________________________________________________________________________ | |
96 | TList * AliAnaGammaSelection::GetCreateOutputObjects() | |
97 | { | |
98 | ||
99 | // Create histograms to be saved in output file and | |
100 | // store them in outputContainer | |
101 | TList * outputContainer = new TList() ; | |
102 | outputContainer->SetName("SelectionHistos") ; | |
103 | ||
104 | // //Histograms of highest gamma identified in Event | |
105 | // fhNGamma = new TH1F("NGamma","Number of #gamma over calorimeter",240,0,120); | |
106 | // fhNGamma->SetYTitle("N"); | |
107 | // fhNGamma->SetXTitle("p_{T #gamma}(GeV/c)"); | |
108 | // outputContainer->Add(fhNGamma) ; | |
109 | ||
110 | ||
111 | //NTUPLE | |
112 | fntEMCAL = new TNtuple("ntEMCAL", "Tree of EMCAL particles before selection", "pt:phi:eta:pdg:status:ptprimary:phiprimary:etaprimary:pdgprimary:statusprimary"); | |
113 | outputContainer->Add(fntEMCAL) ; | |
114 | fntPHOS = new TNtuple("ntPHOS", "Tree of PHOS particles before selection", "pt:phi:eta:pdg:status:ptprimary:phiprimary:etaprimary:pdgprimary:statusprimary"); | |
115 | //outputContainer->Add(fntPHOS) ; | |
116 | fntCTS = new TNtuple("ntCTS", "Tree of CTS particles before selection", "pt:phi:eta:pdg:status:ptprimary:phiprimary:etaprimary:pdgprimary:statusprimary"); | |
117 | //outputContainer->Add(fntCTS) ; | |
118 | ||
3788def4 | 119 | // gROOT->cd(); |
35b8d778 | 120 | |
121 | return outputContainer ; | |
122 | ||
123 | } | |
124 | ||
125 | //____________________________________________________________________________ | |
126 | void AliAnaGammaSelection::Selection(TString det, TClonesArray * pl, TClonesArray * plPrim) const | |
127 | { | |
128 | //Select particles from detector "det" | |
129 | ||
130 | //Keep some particle info before selection | |
131 | Float_t pt = 0, ptprimary = 0 ; | |
132 | Float_t phi = 0, phiprimary = 0 ; | |
133 | Float_t eta = 0, etaprimary = 0 ; | |
134 | Int_t pdg = 0, pdgprimary = 0 ; | |
135 | Int_t status = 0, statusprimary = 0 ; | |
136 | ||
137 | for(Int_t ipr = 0;ipr < pl->GetEntries() ; ipr ++ ){ | |
138 | TParticle * particle = dynamic_cast<TParticle *>(pl->At(ipr)) ; | |
139 | pt = particle->Pt(); | |
140 | phi = particle->Phi(); | |
141 | eta = particle->Eta(); | |
142 | pdg = TMath::Abs(particle->GetPdgCode()) ; | |
143 | status = particle->GetStatusCode(); // = 2 means decay gamma!!! | |
144 | ||
145 | if(fAnaMC){ | |
146 | TParticle * primary = dynamic_cast<TParticle *>(plPrim->At(ipr)) ; | |
147 | ptprimary = primary->Pt(); | |
148 | phiprimary = primary->Phi(); | |
149 | etaprimary = primary->Eta(); | |
150 | pdgprimary = TMath::Abs(primary->GetPdgCode()) ; | |
151 | statusprimary = primary->GetStatusCode(); // = 2 means decay gamma!!! | |
152 | } | |
153 | ||
154 | gROOT->cd(); | |
155 | if(det == "EMCAL") | |
156 | fntEMCAL->Fill(pt,phi,eta,pdg,status,ptprimary,phiprimary, etaprimary,pdgprimary,statusprimary); | |
157 | else if(det == "PHOS") | |
158 | fntPHOS->Fill(pt,phi,eta,pdg,status,ptprimary,phiprimary, etaprimary,pdgprimary,statusprimary); | |
159 | else | |
160 | fntCTS->Fill(pt,phi,eta,pdg,status,ptprimary,phiprimary, etaprimary,pdgprimary,statusprimary); | |
161 | } | |
162 | ||
163 | } | |
164 | ||
165 | //____________________________________________________________________________ | |
166 | void AliAnaGammaSelection::InitParameters() | |
167 | { | |
168 | ||
169 | //Initialize the parameters of the analysis. | |
170 | ||
171 | fAnaMC = kFALSE ; | |
172 | ||
173 | fFillCTS = kFALSE ; | |
174 | } | |
175 | ||
176 | ||
177 | //__________________________________________________________________ | |
178 | void AliAnaGammaSelection::Print(const Option_t * opt) const | |
179 | { | |
180 | ||
181 | //Print some relevant parameters set for the analysis | |
182 | if(! opt) | |
183 | return; | |
184 | ||
185 | Info("Print", "%s %s", GetName(), GetTitle() ) ; | |
186 | printf("Is MC option selected %d\n", fAnaMC) ; | |
187 | printf(" \n") ; | |
188 | ||
189 | } |