Cluster Map moved to ANALYSIS
[u/mrichter/AliRoot.git] / HBTAN / AliHBTEvent.cxx
index 003a062980c1358f42858285a09f059917d67f8f..9f5275523d788512a24ff307affcfa947f90f3be 100644 (file)
@@ -1,38 +1,86 @@
+//__________________________________________________________
+///////////////////////////////////////////////////////////////////
+//
+// class AliHBTEvent
+//
+// This class is container for paticles coming from one event
+//
+// Piotr.Skowronski@cern.ch 
+//
+///////////////////////////////////////////////////////////////////
+
+
 #include "AliHBTEvent.h"
 #include "AliHBTParticle.h"
 
 ClassImp(AliHBTEvent)
 
-const UInt_t AliHBTEvent::fgkInitEventSize = 10000;
-
-
+const UInt_t AliHBTEvent::fgkInitEventSize = 100;
 
 /**************************************************************************/ 
  
-AliHBTEvent::AliHBTEvent()
+AliHBTEvent::AliHBTEvent():
+ fSize(fgkInitEventSize),
+ fParticles(new AliHBTParticle* [fSize]),
+ fNParticles(0),
+ fOwner(kTRUE),
+ fRandomized(kFALSE)
  {
-    if(fgkInitEventSize<1) 
-     {
-      Fatal("AliHBTEvent::AliHBTEvent()",
-            "fgkInitEventSize has a stiupid value (%d). Change it to positive number and recompile",
-             fgkInitEventSize);
-      
-     }
-    fSize=fgkInitEventSize;
-    fParticles = new AliHBTParticle* [fSize];
-    fNParticles = 0;
-    fOwner = kTRUE;
+//default constructor   
+  if(fgkInitEventSize<1) 
+   {
+    Fatal("AliHBTEvent::AliHBTEvent()",
+          "fgkInitEventSize has a stiupid value (%d). Change it to positive number and recompile",
+           fgkInitEventSize);
+
+   }
  }
 /**************************************************************************/ 
+AliHBTEvent::AliHBTEvent(const AliHBTEvent& source):
+ TObject(source),
+ fSize(source.fSize),
+ fParticles(new AliHBTParticle* [fSize]),
+ fNParticles(source.fNParticles),
+ fOwner(source.fNParticles),
+ fRandomized(source.fRandomized)
+{
+//copy constructor
+  for(Int_t i =0; i<fNParticles; i++)
+   {
+      fParticles[i] = new AliHBTParticle( *(source.fParticles[i]) );
+   }
+  
+}
+/**************************************************************************/ 
 
+AliHBTEvent& AliHBTEvent::operator=(const AliHBTEvent& source)
+{
+  // assigment operator
+  Reset();
+  if (fParticles) delete [] fParticles;
+  fSize = source.fSize;
+  fParticles = new AliHBTParticle* [fSize];
+  fNParticles = source.fNParticles;
+  fOwner = source.fNParticles;
+  fRandomized = source.fRandomized;
+      
+  for(Int_t i =0; i<fNParticles; i++)
+   {
+      fParticles[i] = new AliHBTParticle( *(source.fParticles[i]) );
+   }
+  return *this;
+}
+/**************************************************************************/ 
 AliHBTEvent::~AliHBTEvent()
  {
-   this->Reset();//delete all particles
-   if(fParticles)
-    { 
-     delete [] fParticles; //and delete array itself
-    }
-   fParticles = 0x0;
+//destructor   
+  this->Reset();//delete all particles
+  if(fParticles)
+   { 
+    delete [] fParticles; //and delete array itself
+   }
+  fParticles = 0x0;
  }
 /**************************************************************************/ 
 void  AliHBTEvent::Reset()
@@ -41,60 +89,76 @@ void  AliHBTEvent::Reset()
   if(fParticles && fOwner)
     {
       for(Int_t i =0; i<fNParticles; i++)
-        delete fParticles[i];
+       {
+         for (Int_t j = i+1; j<fNParticles; j++)
+           if (fParticles[j] == fParticles[i]) fParticles[j] = 0x0;
+         delete fParticles[i];
+       }
     }
    fNParticles = 0;
+   fRandomized = kFALSE;
 } 
+/**************************************************************************/ 
 
 AliHBTParticle* AliHBTEvent::GetParticleSafely(Int_t n)
- {
-   if( (n<0) || (fNParticles<=n) ) return 0x0;
-   else return fParticles[n];
-   
- }
+{
+  //returns nth particle with range check
+  if( (n<0) || (fNParticles<=n) ) return 0x0;
+  else return fParticles[n];
+}
 /**************************************************************************/ 
 
 void  AliHBTEvent:: AddParticle(AliHBTParticle* hbtpart)
- {
-   //Adds new perticle to the event
-   if ( fNParticles+1 >= fSize) Expand(); //if there is no space in array, expand it
-   fParticles[fNParticles++] = hbtpart; //add a pointer
- }
-
+{
+  //Adds new perticle to the event
+  if ( fNParticles+1 >= fSize) Expand(); //if there is no space in array, expand it
+  fParticles[fNParticles++] = hbtpart; //add a pointer
+}
 /**************************************************************************/ 
-void  AliHBTEvent::AddParticle(TParticle* part)
- {
-   AddParticle( new AliHBTParticle(*part) );
- }
+
+void  AliHBTEvent::AddParticle(TParticle* part, Int_t idx)
+{
+  //Adds TParticle to event
+  AddParticle( new AliHBTParticle(*part,idx) );
+}
 /**************************************************************************/ 
-void  AliHBTEvent::
-AddParticle(Int_t pdg, Double_t px, Double_t py, Double_t pz, Double_t etot,
-            Double_t vx, Double_t vy, Double_t vz, Double_t time)
- {
-   AddParticle(new  AliHBTParticle(pdg,px,py,pz,etot,vx,vy,vz,time) );
- }
+void  AliHBTEvent::AddParticle(Int_t pdg, Int_t idx, 
+             Double_t px, Double_t py, Double_t pz, Double_t etot,
+             Double_t vx, Double_t vy, Double_t vz, Double_t time)
+{
+  //adds particle to event
+  AddParticle(new  AliHBTParticle(pdg,idx,px,py,pz,etot,vx,vy,vz,time) );
+}
 /**************************************************************************/ 
 
 void AliHBTEvent::Expand()
- {
- //expands the array with pointers to particles
- //about the size defined in fgkInitEventSize
-  fSize+=fgkInitEventSize;
-  AliHBTParticle** tmpParticles = new AliHBTParticle* [fSize]; //create new array of pointers
-  //check if we got memory / if not Abort
-  if (!tmpParticles) Fatal("AliHBTEvent::Expand()","No more space in memory");
-  
-  
-  for(Int_t i = 0; i<fNParticles ;i++)
-   //copy pointers to the new array
-   {
-     tmpParticles[i] = fParticles[i];
-   }
-  delete [] fParticles; //delete old array
-   fParticles = tmpParticles; //copy new pointer to the array of pointers to particles
+{
+//expands the array with pointers to particles
+//about the size defined in fgkInitEventSize
+
+ fSize+=fgkInitEventSize;
+ AliHBTParticle** tmpParticles = new AliHBTParticle* [fSize]; //create new array of pointers
+ //check if we got memory / if not Abort
+ if (!tmpParticles) Fatal("AliHBTEvent::Expand()","No more space in memory");
+
+
+ for(Int_t i = 0; i<fNParticles ;i++)
+  //copy pointers to the new array
+  {
+    tmpParticles[i] = fParticles[i];
+  }
+ delete [] fParticles; //delete old array
+  fParticles = tmpParticles; //copy new pointer to the array of pointers to particles
+}
+/**************************************************************************/ 
+
+void AliHBTEvent::SwapParticles(Int_t i, Int_t j)
+{
+//swaps particles positions; used by AliHBTEvent::Blend
+  if ( (i<0) || (i>=fNParticles)) return;
+  if ( (j<0) || (j>=fNParticles)) return;
   
- }
+  AliHBTParticle* tmp = fParticles[i];
+  fParticles[i] = fParticles[j];
+  fParticles[j] = tmp;
+}