]> git.uio.no Git - u/mrichter/AliRoot.git/blame - JETAN/jetan2004/AliJetEventParticles.cxx
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Changes in the ACORDE libraries to compile on Windows/Cygwin
[u/mrichter/AliRoot.git] / JETAN / jetan2004 / AliJetEventParticles.cxx
CommitLineData
d7c6ab14 1// $Id$
2
3//__________________________________________________________
4///////////////////////////////////////////////////////////////////
5//
6// class AliJetEventParticles
7//
8// loizides@ikf.uni-frankfurt.de
9//
10///////////////////////////////////////////////////////////////////
11
12#include <Riostream.h>
13#include <TClonesArray.h>
14#include "AliJetParticle.h"
15#include "AliJetEventParticles.h"
16
17ClassImp(AliJetEventParticles)
18
d7c6ab14 19AliJetEventParticles::AliJetEventParticles(Int_t size) :
962f5fc0 20 TObject(),
21 fHeader(),
d7c6ab14 22 fNParticles(0),
23 fParticles(new TClonesArray("AliJetParticle",size)),
24 fVertexX(0.),
25 fVertexY(0.),
04a02430 26 fVertexZ(0.),
27 fTrials(0),
28 fNJets(0),
29 fNUQJets(0),
30 fXJet(-1),
3bf434de 31 fYJet(-1),
32 fImpact(0.),
33 fNHardScatters(0),
4a8324e6 34 fNwNwColl(0),
35 fEventNr(0)
d7c6ab14 36{
04a02430 37 // Default Constructor
38 for (Int_t i = 0; i < 4; i++) fZquench[i] = 0.;
39 for (Int_t i = 0; i < 10; i++)
40 for (Int_t j = 0; j < 4; j++) {
41 fJets[j][i]=0; // Trigger jets
42 fUQJets[j][i]=0; // Unquenched trigger jets
43 }
9e431626 44 for (Int_t i = 0; i < 5; i++){
45 fHard[i][0]=0;
46 fHard[i][1]=0;
47 }
d7c6ab14 48}
49
d7c6ab14 50AliJetEventParticles::AliJetEventParticles(const AliJetEventParticles& source) :
51 TObject(source),
962f5fc0 52 fHeader(),
d7c6ab14 53 fNParticles(source.fNParticles),
04a02430 54 fParticles(new TClonesArray("AliJetParticle",source.fNParticles)),
55 fVertexX(source.GetVertexX()),
56 fVertexY(source.GetVertexY()),
57 fVertexZ(source.GetVertexZ()),
58 fTrials(source.Trials()),
59 fNJets(source.NTriggerJets()),
60 fNUQJets(source.NUQTriggerJets()),
61 fXJet(source.GetXJet()),
3bf434de 62 fYJet(source.GetXJet()),
63 fImpact(source.GetImpact()),
64 fNHardScatters(source.GetNhard()),
4a8324e6 65 fNwNwColl(source.GetNpart()),
66 fEventNr(source.GetEventNr())
d7c6ab14 67{
68 //copy constructor
69 for(Int_t i =0; i<fNParticles; i++)
70 {
71 const AliJetParticle *kjp=(const AliJetParticle *)source.fParticles->At(i);
04a02430 72 new((*fParticles)[i]) AliJetParticle(*(kjp));
73 }
74 for (Int_t i = 0; i < 4; i++) fZquench[i] = 0.;
75 for (Int_t i = 0; i < 10; i++)
76 for (Int_t j = 0; j < 4; j++) {
77 fJets[j][i]=0; // Trigger jets
78 fUQJets[j][i]=0; // Unquenched trigger jets
d7c6ab14 79 }
04a02430 80 source.GetZQuench(fZquench);
81 for (Int_t i = 0; i < NTriggerJets(); i++){
82 source.TriggerJet(i,fJets[0][i],fJets[1][i],fJets[2][i],fJets[3][i]);
83 }
84 for (Int_t i = 0; i < NUQTriggerJets(); i++){
85 source.UQJet(i,fUQJets[0][i],fUQJets[1][i],fUQJets[2][i],fUQJets[3][i]);
86 }
9e431626 87
88 source.Hard(0,fHard[0][0],fHard[1][0],fHard[2][0],fHard[3][0],fHard[4][0]);
89 source.Hard(1,fHard[0][1],fHard[1][1],fHard[2][1],fHard[3][1],fHard[4][1]);
d7c6ab14 90}
91
962f5fc0 92void AliJetEventParticles::Set(const AliJetEventParticles& source)
93{
94 source.Copy(*this);
95 fHeader=source.GetHeader();
96 fNParticles=source.fNParticles;
97 if(fParticles) delete fParticles;
f4ecde7e 98 fParticles=new TClonesArray("AliJetParticle",source.GetParticles()->GetEntries());
962f5fc0 99 fVertexX=source.GetVertexX();
100 fVertexY=source.GetVertexY();
101 fVertexZ=source.GetVertexZ();
102 fTrials=source.Trials();
103 fNJets=source.NTriggerJets();
104 fNUQJets=source.NUQTriggerJets();
105 fXJet=source.GetXJet();
106 fYJet=source.GetXJet();
107 fImpact=source.GetImpact();
108 fNHardScatters=source.GetNhard();
109 fNwNwColl=source.GetNpart();
4a8324e6 110 fEventNr=source.GetEventNr();
962f5fc0 111
f4ecde7e 112 for(Int_t i =0; i<source.GetParticles()->GetEntries(); i++){
113 const AliJetParticle *kjp=(const AliJetParticle *)source.fParticles->At(i);
114 new((*fParticles)[i]) AliJetParticle(*(kjp));
115 }
962f5fc0 116 for (Int_t i = 0; i < 4; i++) fZquench[i] = 0.;
117 for (Int_t i = 0; i < 10; i++)
118 for (Int_t j = 0; j < 4; j++) {
119 fJets[j][i]=0; // Trigger jets
120 fUQJets[j][i]=0; // Unquenched trigger jets
121 }
122 source.GetZQuench(fZquench);
123 for (Int_t i = 0; i < NTriggerJets(); i++){
124 source.TriggerJet(i,fJets[0][i],fJets[1][i],fJets[2][i],fJets[3][i]);
125 }
126 for (Int_t i = 0; i < NUQTriggerJets(); i++){
127 source.UQJet(i,fUQJets[0][i],fUQJets[1][i],fUQJets[2][i],fUQJets[3][i]);
128 }
129
130 source.Hard(0,fHard[0][0],fHard[1][0],fHard[2][0],fHard[3][0],fHard[4][0]);
131 source.Hard(1,fHard[0][1],fHard[1][1],fHard[2][1],fHard[3][1],fHard[4][1]);
132}
133
d7c6ab14 134AliJetEventParticles::~AliJetEventParticles()
135{
136 //destructor
137 Reset();
138 delete fParticles;
139}
140
d7c6ab14 141void AliJetEventParticles::Reset(Int_t size)
142{
962f5fc0 143 fHeader="";
d7c6ab14 144 //deletes all particles from the event
d7c6ab14 145 if(fParticles) fParticles->Clear();
146 if(size>=0) fParticles->Expand(size);
147 fNParticles = 0;
d7c6ab14 148
04a02430 149 fVertexX=0.;
150 fVertexY=0.;
151 fVertexZ=0.;
152 fTrials=0;
153 fNJets=0;
154 fNUQJets=0;
155 fXJet=-1;
156 fYJet=-1;
3bf434de 157 fImpact=0.;
158 fNHardScatters=0;
159 fNwNwColl=0;
4a8324e6 160 fEventNr=0;
04a02430 161 for (Int_t i = 0; i < 4; i++) fZquench[i] = 0.;
162 for (Int_t i = 0; i < 10; i++)
163 for (Int_t j = 0; j < 4; j++) {
164 fJets[j][i]=0; // Trigger jets
165 fUQJets[j][i]=0; // Unquenched trigger jets
166 }
9e431626 167 for (Int_t i = 0; i < 5; i++){
168 fHard[i][0]=0;
169 fHard[i][1]=0;
170 }
04a02430 171}
d7c6ab14 172
301a24f1 173void AliJetEventParticles::AddSignal(const AliJetEventParticles& source)
174{ //mark signal particles and add them to TClonesArray
175 //note that fNParticles still keeps only background particles
176
177 Int_t nSignalParts=source.GetNParticles();
178 for(Int_t i=0; i<nSignalParts; i++)
179 {
180 const AliJetParticle *kjp=source.GetParticle(i);
181
182 AliJetParticle *ap=new((*fParticles)[fNParticles+i]) AliJetParticle(*(kjp));
183 ap->SetType(-123); //mark pythia particle
184 }
185 for(Int_t i=nSignalParts+fNParticles;i<fParticles->GetEntriesFast();i++)
186 fParticles->RemoveAt(i);
187 //cout << fParticles->GetEntries() << " " << fNParticles << " " << nSignalParts << endl;
188
189 /* should we transform the vertex???
190 fVertexX=source.GetVertexX();
191 fVertexY=source.GetVertexY();
192 fVertexZ=source.GetVertexZ();
193 */
194 fTrials=source.Trials();
195 fNJets=source.NTriggerJets();
196 fNUQJets=source.NUQTriggerJets();
197 fXJet=source.GetXJet();
198 fYJet=source.GetXJet();
7f48d376 199 fEventNr=source.GetEventNr();
301a24f1 200
301a24f1 201 for (Int_t i = 0; i < 4; i++) fZquench[i] = 0.;
202 for (Int_t i = 0; i < 10; i++)
203 for (Int_t j = 0; j < 4; j++) {
204 fJets[j][i]=0; // Trigger jets
205 fUQJets[j][i]=0; // Unquenched trigger jets
206 }
207 source.GetZQuench(fZquench);
208 for (Int_t i = 0; i < NTriggerJets(); i++){
209 source.TriggerJet(i,fJets[0][i],fJets[1][i],fJets[2][i],fJets[3][i]);
210 }
211 for (Int_t i = 0; i < NUQTriggerJets(); i++){
212 source.UQJet(i,fUQJets[0][i],fUQJets[1][i],fUQJets[2][i],fUQJets[3][i]);
213 }
214 source.Hard(0,fHard[0][0],fHard[1][0],fHard[2][0],fHard[3][0],fHard[4][0]);
215 source.Hard(1,fHard[0][1],fHard[1][1],fHard[2][1],fHard[3][1],fHard[4][1]);
216}
217
d7c6ab14 218void AliJetEventParticles::AddParticle(AliJetParticle* part)
219{
220 //Adds new particle to the event
221 fParticles->AddAt(part,fNParticles++);
222}
223
d7c6ab14 224void AliJetEventParticles::AddParticle(const AliJetParticle* part)
225{
226 //Adds new particle to the event
b2760c9e 227 new((*fParticles)[fNParticles++]) AliJetParticle(*part);
d7c6ab14 228}
229
301a24f1 230void AliJetEventParticles::AddParticle(const TParticle* part,Int_t idx, Int_t l, Int_t ncl)
d7c6ab14 231{
232 //Adds new particle to the event
301a24f1 233 new((*fParticles)[fNParticles++]) AliJetParticle(part,idx,l,ncl);
d7c6ab14 234}
235
d7c6ab14 236void AliJetEventParticles::AddParticle(Float_t px, Float_t py, Float_t pz,
301a24f1 237 Float_t etot, Int_t idx, Int_t l, Int_t ncl)
d7c6ab14 238{
239 //Adds new particle to the event
301a24f1 240 new((*fParticles)[fNParticles++]) AliJetParticle(px,py,pz,etot,idx,l,ncl);
d7c6ab14 241}
242
d7c6ab14 243void AliJetEventParticles::AddParticle(Float_t px, Float_t py, Float_t pz, Float_t etot, Int_t idx, Int_t l,
301a24f1 244 Int_t ncl, Float_t pt, Float_t phi, Float_t eta)
d7c6ab14 245{
246 //Adds new particle to the event
301a24f1 247 new((*fParticles)[fNParticles++]) AliJetParticle(px,py,pz,etot,idx,l,ncl,pt,phi,eta);
d7c6ab14 248}
249
d7c6ab14 250const AliJetParticle* AliJetEventParticles::GetParticleSafely(Int_t n)
251{
252 //returns nth particle with range check
253 if( (n<0) || (fNParticles<=n) ) return 0;
254 return (const AliJetParticle*)fParticles->At(n);
255}
256
04a02430 257void AliJetEventParticles::AddJet(Float_t px, Float_t py, Float_t pz, Float_t e)
258{
259 //
260 // Add a jet
261 //
262 if (fNJets < 10) {
263 fJets[0][fNJets] = px;
264 fJets[1][fNJets] = py;
265 fJets[2][fNJets] = pz;
266 fJets[3][fNJets] = e;
267 fNJets++;
268 } else {
269 printf("\nWarning: More than 10 jets triggered !!\n");
270 }
271}
272
273void AliJetEventParticles::AddJet(Float_t p[4])
274{
275 //
276 // Add a jet
277 //
278 if (fNJets < 10) {
279 fJets[0][fNJets] = p[0];
280 fJets[1][fNJets] = p[1];
281 fJets[2][fNJets] = p[2];
282 fJets[3][fNJets] = p[3];
283 fNJets++;
284 } else {
285 printf("\nWarning: More than 10 jets triggered !!\n");
286 }
287}
288
289void AliJetEventParticles::AddUQJet(Float_t px, Float_t py, Float_t pz, Float_t e)
290{
291 //
292 // Add a jet
293 //
294 if (fNUQJets < 10) {
295 fUQJets[0][fNUQJets] = px;
296 fUQJets[1][fNUQJets] = py;
297 fUQJets[2][fNUQJets] = pz;
298 fUQJets[3][fNUQJets] = e;
299 fNUQJets++;
300 } else {
301 printf("\nWarning: More than 10 jets triggered !!\n");
302 }
303}
304
305void AliJetEventParticles::AddUQJet(Float_t p[4])
306{
307 //
308 // Add a jet
309 //
310 if (fNUQJets < 10) {
311 fUQJets[0][fNUQJets] = p[0];
312 fUQJets[1][fNUQJets] = p[1];
313 fUQJets[2][fNUQJets] = p[2];
314 fUQJets[3][fNUQJets] = p[3];
315 fNUQJets++;
316 } else {
317 printf("\nWarning: More than 10 jets triggered !!\n");
318 }
319}
320
9e431626 321void AliJetEventParticles::AddHard(Int_t i,Float_t px, Float_t py, Float_t pz, Float_t e, Float_t type)
322{
323 //
324 // Add a had parton
325 //
326 if (i < 2) {
327 fHard[0][i] = px;
328 fHard[1][i] = py;
329 fHard[2][i] = pz;
330 fHard[3][i] = e;
331 fHard[4][i] = type;
332 } else {
333 printf("\nWarning: More than 2 partons !!\n");
334 }
335}
336
04a02430 337void AliJetEventParticles::SetZQuench(Double_t z[4])
338{
339 //
340 // Set quenching fraction
341 //
342 for (Int_t i = 0; i < 4; i++) fZquench[i] = z[i];
343}
344
345void AliJetEventParticles::GetZQuench(Double_t z[4]) const
346{
347 //
348 // Get quenching fraction
349 //
350 for (Int_t i = 0; i < 4; i++) z[i] = fZquench[i];
351}
352
353void AliJetEventParticles::TriggerJet(Int_t i, Float_t p[4]) const
354{
355 //
356 // Give back jet #i
357 //
358 if (i >= fNJets) {
359 printf("\nWarning: TriggerJet, index out of Range!!\n");
360 } else {
361 p[0] = fJets[0][i];
362 p[1] = fJets[1][i];
363 p[2] = fJets[2][i];
364 p[3] = fJets[3][i];
365 }
366}
367
368void AliJetEventParticles::TriggerJet(Int_t i, Float_t &p1, Float_t &p2, Float_t &p3, Float_t &E) const
369{
370 //
371 // Give back jet #i
372 //
373 if (i >= fNJets) {
374 printf("\nWarning: TriggerJet, index out of Range!!\n");
375 } else {
9e431626 376 p1 = fJets[0][i];
377 p2 = fJets[1][i];
378 p3 = fJets[2][i];
379 E = fJets[3][i];
04a02430 380 }
381}
382
383void AliJetEventParticles::UQJet(Int_t i, Float_t p[4]) const
384{
385 //
386 // Give back jet #i
387 //
388 if (i >= fNUQJets) {
389 printf("\nWarning: Unquenched Jets, index out of Range!!\n");
390 } else {
391 p[0] = fUQJets[0][i];
392 p[1] = fUQJets[1][i];
393 p[2] = fUQJets[2][i];
394 p[3] = fUQJets[3][i];
395 }
396}
397
398void AliJetEventParticles::UQJet(Int_t i, Float_t &p1, Float_t &p2, Float_t &p3, Float_t &E) const
399{
400 //
401 // Give back jet #i
402 //
403 if (i >= fNUQJets) {
404 printf("\nWarning: Unquenched Jets, index out of Range!!\n");
405 } else {
9e431626 406 p1 = fUQJets[0][i];
407 p2 = fUQJets[1][i];
408 p3 = fUQJets[2][i];
409 E = fUQJets[3][i];
410 }
411}
412
413void AliJetEventParticles::Hard(Int_t i, Float_t &p1, Float_t &p2, Float_t &p3, Float_t &E, Float_t &type) const
414{
415 //
416 // Give back jet #i
417 //
418 if (i >= 2) {
419 printf("\nWarning: Hard partons, index out of Range!!\n");
420 } else {
421 p1 = fHard[0][i];
422 p2 = fHard[1][i];
423 p3 = fHard[2][i];
424 E = fHard[3][i];
425 type = fHard[4][i];
04a02430 426 }
427}
428
3bf434de 429void AliJetEventParticles::Hard(Int_t i, Float_t p[4], Float_t &type) const
430{
431 //
432 // Give back jet #i
433 //
434 if (i >= 2) {
435 printf("\nWarning: Hard partons, index out of Range!!\n");
436 } else {
437 p[0] = fHard[0][i];
438 p[1] = fHard[1][i];
439 p[2] = fHard[2][i];
440 p[3] = fHard[3][i];
441 type = fHard[4][i];
442 }
443}
444
04a02430 445void AliJetEventParticles::SetXYJet(Double_t x, Double_t y)
446{
447 //
448 // Add jet production point
449 //
450 fXJet = x;
451 fYJet = y;
452}
d7c6ab14 453
454void AliJetEventParticles::Print(Option_t* /*t*/) const
455{
456 cout << "--- AliJetEventParticles ---" << endl;
457 if(fHeader.Length()) cout << fHeader.Data() << endl;
4a8324e6 458 cout << "Event Number: " << fEventNr << endl;
04a02430 459 cout << "Particles in Event: " << fNParticles << endl;
460 if(fNUQJets){
461 cout << "Unquenched Jets: " << fNUQJets << endl;
462 for(Int_t i = 0; i<fNUQJets; i++){
463 Float_t x=fUQJets[0][i];
464 Float_t y=fUQJets[1][i];
465 Float_t z=fUQJets[2][i];
466 Float_t e=fUQJets[3][i];
467 Float_t ptj=TMath::Sqrt(x*x+y*y);
468 Float_t thj=TMath::ATan2(ptj,z);
469 Float_t etaj=-TMath::Log(TMath::Tan(thj/2));
470 Float_t phj=TMath::Pi()+TMath::ATan2(-y,-x);
471 Float_t et=e*TMath::Sin(thj);
472 cout << i << " " << et << " " << etaj << " " << phj << endl;
473 }
474 }
475 if(fNJets){
476 cout << "Triggered Jets: " << fNJets << endl;
477 for(Int_t i = 0; i<fNJets; i++){
478 Float_t x=fJets[0][i];
479 Float_t y=fJets[1][i];
480 Float_t z=fJets[2][i];
481 Float_t e=fJets[3][i];
482 Float_t ptj=TMath::Sqrt(x*x+y*y);
483 Float_t thj=TMath::ATan2(ptj,z);
484 Float_t etaj=-TMath::Log(TMath::Tan(thj/2));
485 Float_t phj=TMath::Pi()+TMath::ATan2(-y,-x);
486 Float_t et=e*TMath::Sin(thj);
487 cout << i << " " << et << " " << etaj << " " << phj << endl;
488 }
489 }
d7c6ab14 490}