code cleaning
[u/mrichter/AliRoot.git] / HBTAN / AliHBTEvent.cxx
1 #include "AliHBTEvent.h"
2 #include "AliHBTParticle.h"
3 //_________________________________________________________________________
4 ///////////////////////////////////////////////////////////////////////////
5 //                                                                       //
6 // class AliHBTEvent                                                     //
7 //                                                                       //
8 // This class stores HBT perticles for one event                         //
9 // more info: http://alisoft.cern.ch/people/skowron/analyzer/index.html  //
10 //                                                                       //
11 ///////////////////////////////////////////////////////////////////////////
12
13 ClassImp(AliHBTEvent)
14
15 const UInt_t AliHBTEvent::fgkInitEventSize = 10000;
16
17
18 /**************************************************************************/ 
19  
20 AliHBTEvent::AliHBTEvent():
21   fSize(fgkInitEventSize),
22   fParticles(new AliHBTParticle* [fSize]),
23   fNParticles(0),
24   fOwner(kTRUE)
25  {
26  //default constructor    
27  }
28 /**************************************************************************/ 
29
30 AliHBTEvent::~AliHBTEvent()
31  {
32  //destructor
33    this->Reset();//delete all particles
34    if(fParticles)
35     { 
36      delete [] fParticles; //and delete array itself
37     }
38    fParticles = 0x0;
39  }
40 /**************************************************************************/ 
41 void  AliHBTEvent::Reset()
42 {
43  //deletes all particles from the event
44   if(fParticles && fOwner)
45     {
46       for(Int_t i =0; i<fNParticles; i++)
47         delete fParticles[i];
48     }
49    fNParticles = 0;
50
51 /**************************************************************************/ 
52
53 AliHBTParticle* AliHBTEvent::GetParticleSafely(Int_t n)
54  {
55  //returns nth particle  with range check
56    if( (n<0) || (fNParticles<=n) ) return 0x0;
57    else return fParticles[n];
58    
59  }
60 /**************************************************************************/ 
61
62 void  AliHBTEvent:: AddParticle(AliHBTParticle* hbtpart)
63  {
64  //Adds new perticle to the event
65    if ( fNParticles+1 >= fSize) Expand(); //if there is no space in array, expand it
66    fParticles[fNParticles++] = hbtpart; //add a pointer
67  }
68
69 /**************************************************************************/ 
70 void  AliHBTEvent::AddParticle(TParticle* part)
71  {
72  //Adds TParticle to event
73    AddParticle( new AliHBTParticle(*part) );
74  }
75 /**************************************************************************/ 
76 void  AliHBTEvent::
77 AddParticle(Int_t pdg, Double_t px, Double_t py, Double_t pz, Double_t etot,
78             Double_t vx, Double_t vy, Double_t vz, Double_t time)
79  {
80  //adds particle to event
81    AddParticle(new  AliHBTParticle(pdg,px,py,pz,etot,vx,vy,vz,time) );
82  }
83 /**************************************************************************/ 
84
85 void AliHBTEvent::Expand()
86  {
87  //expands the array with pointers to particles
88  //about the size defined in fgkInitEventSize
89  
90   fSize+=fgkInitEventSize;
91   AliHBTParticle** tmpParticles = new AliHBTParticle* [fSize]; //create new array of pointers
92   //check if we got memory / if not Abort
93   if (!tmpParticles) Fatal("AliHBTEvent::Expand()","No more space in memory");
94   
95   
96   for(Int_t i = 0; i<fNParticles ;i++)
97    //copy pointers to the new array
98    {
99      tmpParticles[i] = fParticles[i];
100    }
101   delete [] fParticles; //delete old array
102    fParticles = tmpParticles; //copy new pointer to the array of pointers to particles
103   
104  }
105  
106