Transition from chi2 to chi2/NDF.
authormarkus <markus@f7af4fe6-9843-0410-8265-dc069ae4e863>
Fri, 23 Mar 2007 16:48:11 +0000 (16:48 +0000)
committermarkus <markus@f7af4fe6-9843-0410-8265-dc069ae4e863>
Fri, 23 Mar 2007 16:48:11 +0000 (16:48 +0000)
Additional functionality added to keep track of chi2, NDF, and the tracks used to fit the vertex.

STEER/AliAODVertex.cxx
STEER/AliAODVertex.h

index 97e452b..d841661 100644 (file)
 
 #include "AliAODVertex.h"
 
+#include "AliAODTrack.h"
+
 ClassImp(AliAODVertex)
 
 //______________________________________________________________________________
 AliAODVertex::AliAODVertex() : 
   TObject(),
-  fChi2(-999.),
+  fChi2perNDF(-999.),
   fCovMatrix(NULL),
   fParent(0x0),
   fDaughters(),
@@ -43,11 +45,11 @@ AliAODVertex::AliAODVertex() :
 //______________________________________________________________________________
 AliAODVertex::AliAODVertex(const Double_t position[3], 
                           const Double_t covMatrix[6],
-                          Double_t  chi2,
+                          Double_t  chi2perNDF,
                           TObject  *parent,
                           Char_t vtype) :
   TObject(),
-  fChi2(chi2),
+  fChi2perNDF(chi2perNDF),
   fCovMatrix(NULL),
   fParent(parent),
   fDaughters(),
@@ -62,12 +64,12 @@ AliAODVertex::AliAODVertex(const Double_t position[3],
 //______________________________________________________________________________
 AliAODVertex::AliAODVertex(const Float_t position[3], 
                           const Float_t  covMatrix[6],
-                          Double_t  chi2,
+                          Double_t  chi2perNDF,
                           TObject  *parent,
                           Char_t vtype) :
 
   TObject(),
-  fChi2(chi2),
+  fChi2perNDF(chi2perNDF),
   fCovMatrix(NULL),
   fParent(parent),
   fDaughters(),
@@ -81,10 +83,10 @@ AliAODVertex::AliAODVertex(const Float_t position[3],
 
 //______________________________________________________________________________
 AliAODVertex::AliAODVertex(const Double_t position[3], 
-                          Double_t  chi2,
+                          Double_t  chi2perNDF,
                           Char_t vtype) :
   TObject(),
-  fChi2(chi2),
+  fChi2perNDF(chi2perNDF),
   fCovMatrix(NULL),
   fParent(0x0),
   fDaughters(),
@@ -97,10 +99,10 @@ AliAODVertex::AliAODVertex(const Double_t position[3],
 
 //______________________________________________________________________________
 AliAODVertex::AliAODVertex(const Float_t position[3], 
-                          Double_t  chi2,
+                          Double_t  chi2perNDF,
                           Char_t vtype) :
   TObject(),
-  fChi2(chi2),
+  fChi2perNDF(chi2perNDF),
   fCovMatrix(NULL),
   fParent(0x0),
   fDaughters(),
@@ -122,7 +124,7 @@ AliAODVertex::~AliAODVertex()
 //______________________________________________________________________________
 AliAODVertex::AliAODVertex(const AliAODVertex& vtx) :
   TObject(vtx),
-  fChi2(vtx.fChi2),
+  fChi2perNDF(vtx.fChi2perNDF),
   fCovMatrix(NULL),
   fParent(vtx.fParent),
   fDaughters(vtx.fDaughters),
@@ -149,7 +151,7 @@ AliAODVertex& AliAODVertex::operator=(const AliAODVertex& vtx)
     for (int i = 0; i < 3; i++) 
       fPosition[i] = vtx.fPosition[i];
     
-    fChi2 = vtx.fChi2;
+    fChi2perNDF = vtx.fChi2perNDF;
 
     //covariance matrix
     delete fCovMatrix;
@@ -186,6 +188,20 @@ template <class T> void AliAODVertex::GetSigmaXYZ(T sigma[3]) const
 }
 
 //______________________________________________________________________________
+Int_t AliAODVertex::GetNContributors() const 
+{
+  // Returns the number of tracks used to fit this vertex.
+  
+  Int_t cont = 0;
+
+  for (Int_t iDaug = 0; iDaug < GetNDaughters(); iDaug++) {
+    if (((AliAODTrack*)fDaughters.At(iDaug))->GetUsedForVtxFit()) cont++;
+  }
+
+  return cont;
+}
+
+//______________________________________________________________________________
 Bool_t AliAODVertex::HasDaughter(TObject *daughter) const 
 {
   // Checks if the given daughter (particle) is part of this vertex.
@@ -462,6 +478,6 @@ void AliAODVertex::Print(Option_t* /*option*/) const
           fCovMatrix[4],
           fCovMatrix[5]); 
           } */
-  printf(" Chi^2 = %f\n", fChi2);
+  printf(" Chi^2/NDF = %f\n", fChi2perNDF);
 }
 
index b95da5e..325f416 100644 (file)
@@ -26,19 +26,19 @@ class AliAODVertex : public TObject {
   AliAODVertex();
   AliAODVertex(const Double_t *position, 
               const Double_t *covMatrix=0x0,
-              Double_t chi2 = -999.,
+              Double_t chi2perNDF = -999.,
               TObject *parent = 0x0,
               Char_t vtype=kUndef);
     AliAODVertex(const Float_t *position, 
                 const Float_t *covMatrix=0x0,
-                Double_t chi2 = -999.,
+                Double_t chi2perNDF = -999.,
                 TObject *parent = 0x0,
                 Char_t vtype=kUndef);
     AliAODVertex(const Double_t *position, 
-                Double_t chi2,
+                Double_t chi2perNDF,
                 Char_t vtype=kUndef);
     AliAODVertex(const Float_t *position, 
-                Double_t chi2,
+                Double_t chi2perNDF,
                 Char_t vtype=kUndef);
 
   virtual ~AliAODVertex();
@@ -52,7 +52,7 @@ class AliAODVertex : public TObject {
   template <class T> void SetPosition(T *pos)
     { fPosition[0] = pos[0]; fPosition[1] = pos[1]; fPosition[2] = pos[2]; }
 
-  void     SetChi2(Double_t chi2) { fChi2 = chi2; }
+  void     SetChi2perNDF(Double_t chi2perNDF) { fChi2perNDF = chi2perNDF; }
 
   void     SetParent(TObject *parent) { fParent = parent; }
 
@@ -74,9 +74,9 @@ class AliAODVertex : public TObject {
 
   template <class T> void     GetSigmaXYZ(T *sigma) const;
 
-  Double_t  GetChi2() const { return fChi2; }
-  Double_t  GetChi2perNDF() const
-    { return fChi2/(2.*fDaughters.GetEntriesFast()-3.); }
+  Double_t  GetChi2perNDF() const { return fChi2perNDF; }
+  Double_t  GetChi2() const { return fChi2perNDF*(Double_t)GetNDF(); }
+  Int_t     GetNDF() const { return 2*GetNContributors()-3; }
 
   Char_t    GetType() const { return fType; }
   void      SetType(AODVtx_t vtype) { fType=vtype; }
@@ -89,6 +89,7 @@ class AliAODVertex : public TObject {
   TObject* GetDaughter(Int_t i) { return fDaughters.At(i); }
   Bool_t   HasDaughter(TObject *daughter) const;
   Int_t    GetNDaughters() const { return fDaughters.GetEntriesFast(); }
+  Int_t    GetNContributors() const;
 
   // covariance matrix elements after rotation by phi around z-axis 
   // and, then, by theta around new y-axis
@@ -111,13 +112,13 @@ class AliAODVertex : public TObject {
  private :
 
   Double32_t    fPosition[3]; // vertex position
-  Double32_t    fChi2;        // chi2 of vertex fit
+  Double32_t    fChi2perNDF;  // chi2/NDF of vertex fit
   AliAODRedCov<3> *fCovMatrix;   // vertex covariance matrix; values of and below the diagonal
   TRef          fParent;      // reference to the parent particle
   TRefArray     fDaughters;   // references to the daughter particles
   Char_t        fType;        // Vertex type
 
-  ClassDef(AliAODVertex,1);
+  ClassDef(AliAODVertex,2);
 };
 
 #endif