Updates for pile-up vertex (F. Prino)

[u/mrichter/AliRoot.git] / STEER / AliVertex.cxx

diff --git a/STEER/AliVertex.cxx b/STEER/AliVertex.cxx
index fbdb2abb87a1b090ad295181a55452c05ae53e05..e107055ccecde650c490757304583be7ecdddd54 100644 (file)
--- a/STEER/AliVertex.cxx
+++ b/STEER/AliVertex.cxx
@@ -27,37 +27,53 @@
 ClassImp(AliVertex)
 
 //--------------------------------------------------------------------------
-AliVertex::AliVertex() : TNamed() {
+AliVertex::AliVertex() :
+  AliVVertex(),
+  fSigma(0),
+  fNContributors(0),
+  fNIndices(0),
+  fIndices(0)
+{
 //
 // Default Constructor, set everything to 0
 //
   for(Int_t k=0;k<3;k++) fPosition[k]   = 0;
-  fSigma = 0;
-  fNContributors=0;
 }
 
 //--------------------------------------------------------------------------
 AliVertex::AliVertex(Double_t position[3],Double_t dispersion,
-               Int_t nContributors): TNamed() {
+                    Int_t nContributors):
+  AliVVertex(),
+  fSigma(dispersion),
+  fNContributors(nContributors),
+  fNIndices(0),
+  fIndices(0)
+{
   //
   // Standard Constructor
   //
 
   for(Int_t k=0;k<3;k++) fPosition[k]   = position[k];
-  fSigma         = dispersion;
-  fNContributors = nContributors;
   SetName("BaseVertex");
 
 }
 
 //--------------------------------------------------------------------------
-AliVertex::AliVertex(const AliVertex &source): TNamed(source) {
+AliVertex::AliVertex(const AliVertex &source):
+  AliVVertex(source),
+  fSigma(source.GetDispersion()),
+  fNContributors(source.GetNContributors()),
+  fNIndices(source.GetNIndices()),
+  fIndices(0x0)
+{
   //
   // Copy constructor
   //
   for(Int_t i=0;i<3;i++)fPosition[i] = source.fPosition[i];
-  fSigma = source.GetDispersion();
-  fNContributors = source.GetNContributors();
+  if(source.fNIndices>0) {
+    fIndices = new UShort_t[fNIndices];
+    memcpy(fIndices,source.fIndices,fNIndices*sizeof(UShort_t));
+  }
 }
 
 //--------------------------------------------------------------------------
@@ -65,12 +81,19 @@ AliVertex &AliVertex::operator=(const AliVertex &source){
   //
   // assignment operator
   //
-  if(&source == this) return *this;
-  this->SetName(source.GetName());
-  this->SetTitle(source.GetTitle());
-  for(Int_t i=0;i<3;i++)fPosition[i] = source.fPosition[i];
-  fSigma = source.GetDispersion();
-  fNContributors = source.GetNContributors();
+  if(&source != this){
+    AliVVertex::operator=(source);
+    for(Int_t i=0;i<3;i++)fPosition[i] = source.fPosition[i];
+    fSigma = source.GetDispersion();
+    fNContributors = source.GetNContributors();
+    fNIndices = source.GetNIndices();
+    if(fIndices)delete [] fIndices;
+    fIndices = 0;
+    if(fNIndices>0) {
+      fIndices = new UShort_t[fNIndices];
+      memcpy(fIndices,source.fIndices,fNIndices*sizeof(UShort_t));
+    }
+  }
   return *this;
 }
 
@@ -80,8 +103,18 @@ AliVertex::~AliVertex() {
 //  
 // Default Destructor
 //
+  delete [] fIndices;
+  fIndices = 0;
+}
 
+void AliVertex::Clear(Option_t* option) 
+{
+    // Delete allocated memory
+    delete [] fIndices;
+    fIndices = 0;
+    AliVVertex::Clear(option);
 }
+
 //--------------------------------------------------------------------------
 void AliVertex::GetXYZ(Double_t position[3]) const {
 //
@@ -94,6 +127,38 @@ void AliVertex::GetXYZ(Double_t position[3]) const {
   return;
 }
 //--------------------------------------------------------------------------
+void AliVertex::GetCovarianceMatrix(Double_t covmatrix[6]) const {
+//
+// Fake method (is implmented in AliESDVertex)
+//
+  for(Int_t i=0;i<6;i++) covmatrix[i] = -999.;
+
+  return;
+}
+//--------------------------------------------------------------------------
+void AliVertex::SetIndices(Int_t nindices,UShort_t *indices) {
+//
+// Set indices of tracks used for vertex determination 
+//
+  if(fNContributors<1)  { printf("fNContributors<1"); return; }
+  fNIndices = nindices;
+  delete [] fIndices;
+  fIndices = new UShort_t[fNIndices];
+  for(Int_t i=0;i<fNIndices;i++) fIndices[i] = indices[i]; 
+  return;
+}
+//--------------------------------------------------------------------------
+Bool_t AliVertex::UsesTrack(Int_t index) const {
+//
+// checks if a track is used for the vertex 
+//
+  if(fNIndices<1)  {/* printf("fNIndices<1"); */return kFALSE; }
+  for(Int_t i=0;i<fNIndices;i++) {
+    if((Int_t)fIndices[i]==index) return kTRUE;
+  }
+  return kFALSE;
+}
+//--------------------------------------------------------------------------
 void AliVertex::Print(Option_t* /*option*/) const {
 //
 // Print out information on all data members