New version of RALICE introduced
[u/mrichter/AliRoot.git] / RALICE / AliJet.cxx
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
16 /*
17 $Log$
18 Revision 1.2  1999/09/29 09:24:28  fca
19 Introduction of the Copyright and cvs Log
20
21 */
22
23 ///////////////////////////////////////////////////////////////////////////
24 // Class AliJet
25 // Creation and investigation of a jet of particle tracks.
26 // An AliJet can be constructed by adding AliTracks.
27 //
28 // Coding example to make 2 jets j1 and j2.
29 // ----------------------------------------
30 // j1 contains the AliTracks 1 and 2
31 // j2 contains the AliTracks 3 and 4
32 //
33 // AliTrack t1,t2,t3,t4;
34 //  ...
35 //  ... // code to fill the AliTrack data
36 //  ...
37 // AliJet j1(5);
38 // AliJet j2(12);
39 // j1.Add(t1);
40 // j1.Add(t2);
41 // j2.Add(t3);
42 // j2.Add(t4);
43 //
44 // j1.Info();
45 // j2.Info("sph");
46 //
47 // Float_t e1=j1.GetEnergy();
48 // Float_t pnorm=j1->GetMomentum();
49 // Ali3Vector p=j1->Get3Momentum();
50 // Float_t m=j1.GetInvmass();
51 // Int_t ntk=j1.GetNtracks();
52 // AliTrack* tj=j1.GetTrack(1);
53 //
54 // Note : All quantities are in GeV, GeV/c or GeV/c**2
55 //
56 //--- Author: Nick van Eijndhoven 10-jul-1997 UU-SAP Utrecht
57 //- Modified: NvE 06-apr-1999 UU-SAP Utrecht to inherit from Ali4Vector
58 ///////////////////////////////////////////////////////////////////////////
59
60 #include "AliJet.h"
61  
62 ClassImp(AliJet) // Class implementation to enable ROOT I/O
63  
64 AliJet::AliJet()
65 {
66 // Default constructor
67 // All variables initialised to 0
68 // Initial maximum number of tracks is set to the default value
69  fTracks=0;
70  fNtinit=0;
71  Reset();
72  SetNtinit();
73 }
74 ///////////////////////////////////////////////////////////////////////////
75 AliJet::AliJet(Int_t n)
76 {
77 // Create a jet to hold initially a maximum of n tracks
78 // All variables initialised to 0
79  fTracks=0;
80  fNtinit=0;
81  Reset();
82  if (n > 0)
83  {
84   SetNtinit(n);
85  }
86  else
87  {
88   cout << endl;
89   cout << " *AliJet* Initial max. number of tracks entered : " << n << endl;
90   cout << " This is invalid. Default initial maximum will be used." << endl;
91   cout << endl;
92   SetNtinit();
93  }
94 }
95 ///////////////////////////////////////////////////////////////////////////
96 AliJet::~AliJet()
97 {
98 // Default destructor
99  if (fTracks) delete fTracks;
100  fTracks=0;
101 }
102 ///////////////////////////////////////////////////////////////////////////
103 void AliJet::SetNtinit(Int_t n)
104 {
105 // Set the initial maximum number of tracks for this jet
106  fNtinit=n;
107  fNtmax=n;
108  if (fTracks) delete fTracks;
109  fTracks=new TObjArray(fNtmax);
110 }
111 ///////////////////////////////////////////////////////////////////////////
112 void AliJet::Reset()
113 {
114 // Reset all variables to 0
115 // The max. number of tracks is set to the initial value again
116  fNtrk=0;
117  fQ=0;
118  Double_t a[4]={0,0,0,0};
119  SetVector(a,"sph");
120  if (fNtinit > 0) SetNtinit(fNtinit);
121 }
122 ///////////////////////////////////////////////////////////////////////////
123 void AliJet::Add(AliTrack& t)
124 {
125 // Add a track to the jet
126 // In case the maximum number of tracks has been reached
127 // space will be extended to hold an additional amount of tracks as
128 // was initially reserved
129  if (fNtrk == fNtmax) // Check if maximum track number is reached
130  {
131   fNtmax+=fNtinit;
132   fTracks->Expand(fNtmax);
133  }
134  
135  // Add the track to this jet
136  fNtrk++;
137  fTracks->Add(&t);
138  (*this)+=(Ali4Vector&)t;
139  fQ+=t.GetCharge();
140 }
141 ///////////////////////////////////////////////////////////////////////////
142 void AliJet::Info(TString f)
143 {
144 // Provide jet information within the coordinate frame f
145  cout << " *AliJet::Info* Invmass : " << GetInvmass() << " Charge : " << fQ
146       << " Momentum : " << GetMomentum() << " Ntracks : " << fNtrk << endl;
147  cout << " ";
148  Ali4Vector::Info(f); 
149
150 ///////////////////////////////////////////////////////////////////////////
151 void AliJet::List(TString f)
152 {
153 // Provide jet and primary track information within the coordinate frame f
154
155  Info(f); // Information of the current jet
156
157  // The tracks of this jet
158  AliTrack* t; 
159  for (Int_t it=1; it<=fNtrk; it++)
160  {
161   t=GetTrack(it);
162   if (t)
163   {
164    cout << "  ---Track no. " << it << endl;
165    cout << " ";
166    t->Info(f); 
167   }
168   else
169   {
170    cout << " *AliJet::List* Error : No track present." << endl; 
171   }
172  }
173
174 ///////////////////////////////////////////////////////////////////////////
175 void AliJet::ListAll(TString f)
176 {
177 // Provide jet and prim.+sec. track information within the coordinate frame f
178
179  Info(f); // Information of the current jet
180
181  // The tracks of this jet
182  AliTrack* t; 
183  for (Int_t it=1; it<=fNtrk; it++)
184  {
185   t=GetTrack(it);
186   if (t)
187   {
188    cout << "  ---Track no. " << it << endl;
189    cout << " ";
190    t->ListAll(f); 
191   }
192   else
193   {
194    cout << " *AliJet::List* Error : No track present." << endl; 
195   }
196  }
197
198 ///////////////////////////////////////////////////////////////////////////
199 Int_t AliJet::GetNtracks()
200 {
201 // Return the current number of tracks of this jet
202  return fNtrk;
203 }
204 ///////////////////////////////////////////////////////////////////////////
205 Double_t AliJet::GetEnergy()
206 {
207 // Return the total energy of the jet
208  return GetScalar();
209 }
210 ///////////////////////////////////////////////////////////////////////////
211 Double_t AliJet::GetMomentum()
212 {
213 // Return the value of the total jet 3-momentum
214  Ali3Vector p=Get3Vector();
215  Double_t p2=p.Dot(p);
216  return sqrt(p2);
217 }
218 ///////////////////////////////////////////////////////////////////////////
219 Ali3Vector AliJet::Get3Momentum()
220 {
221 // Return the the total jet 3-momentum
222  Ali3Vector p=Get3Vector();
223  return p;
224 }
225 ///////////////////////////////////////////////////////////////////////////
226 Double_t AliJet::GetInvmass()
227 {
228 // Return the invariant mass of the jet
229  Double_t m2=Dot(*this);
230  if (m2>0)
231  {
232   return sqrt(m2);
233  }
234  else
235  {
236   return 0;
237  }
238 }
239 ///////////////////////////////////////////////////////////////////////////
240 Float_t AliJet::GetCharge()
241 {
242 // Return the total charge of the jet
243  return fQ;
244 }
245 ///////////////////////////////////////////////////////////////////////////
246 AliTrack* AliJet::GetTrack(Int_t i)
247 {
248 // Return the i-th track of this jet
249  return (AliTrack*)fTracks->At(i-1);
250 }
251 ///////////////////////////////////////////////////////////////////////////