STEER/AliAODVertex.h

   1 #ifndef AliAODVertex_H
   2 #define AliAODVertex_H
   3 /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id$ */
   7
   8 //-------------------------------------------------------------------------
   9 //     AOD vertex base class
  10 //     Author: Markus Oldenburg, CERN
  11 //-------------------------------------------------------------------------
  12
  13 #include <TRef.h>
  14 #include <TRefArray.h>
  15 #include <TMath.h>
  16
  17 #include "AliAODRedCov.h"
  18
  19 class AliAODVertex : public TObject {
  20
  21  public :
  22
  23   enum AODVtx_t {kUndef=-1, kPrimary, kKink, kV0, kCascade, kMulti};
  24
  25   AliAODVertex();
  26   AliAODVertex(const Double_t *position,
  27                const Double_t *covMatrix=0x0,
  28                Double_t chi2perNDF = -999.,
  29                TObject *parent = 0x0,
  30                Short_t id=-1,
  31                Char_t vtype=kUndef);
  32   AliAODVertex(const Float_t *position,
  33                const Float_t *covMatrix=0x0,
  34                Double_t chi2perNDF = -999.,
  35                TObject *parent = 0x0,
  36                Short_t id=-1,
  37                Char_t vtype=kUndef);
  38     AliAODVertex(const Double_t *position,
  39                  Double_t chi2perNDF,
  40                  Char_t vtype=kUndef);
  41     AliAODVertex(const Float_t *position,
  42                  Double_t chi2perNDF,
  43                  Char_t vtype=kUndef);
  44
  45   virtual ~AliAODVertex();
  46   AliAODVertex(const AliAODVertex& vtx);
  47   AliAODVertex& operator=(const AliAODVertex& vtx);
  48
  49   void     SetX(Double_t x) { fPosition[0] = x; }
  50   void     SetY(Double_t y) { fPosition[1] = y; }
  51   void     SetZ(Double_t z) { fPosition[2] = z; }
  52   void     SetPosition(Double_t x, Double_t y, Double_t z) { fPosition[0] = x; fPosition[1] = y; fPosition[2] = z; }
  53   template <class T> void SetPosition(T *pos)
  54     { fPosition[0] = pos[0]; fPosition[1] = pos[1]; fPosition[2] = pos[2]; }
  55
  56   void     SetChi2perNDF(Double_t chi2perNDF) { fChi2perNDF = chi2perNDF; }
  57
  58   void     SetParent(TObject *parent) { fParent = parent; }
  59
  60   Double_t GetX() const { return fPosition[0]; }
  61   Double_t GetY() const { return fPosition[1]; }
  62   Double_t GetZ() const { return fPosition[2]; }
  63   template <class T> void GetPosition(T *pos) const
  64     {pos[0]=fPosition[0]; pos[1]=fPosition[1]; pos[2]=fPosition[2];}
  65
  66   template <class T> void SetCovMatrix(const T *covMatrix) {
  67     if(!fCovMatrix) fCovMatrix=new AliAODRedCov<3>();
  68     fCovMatrix->SetCovMatrix(covMatrix);}
  69
  70   template <class T> Bool_t GetCovMatrix(T *covMatrix) const {
  71     if(!fCovMatrix) return kFALSE;
  72     fCovMatrix->GetCovMatrix(covMatrix); return kTRUE;}
  73
  74   void RemoveCovMatrix() {delete fCovMatrix; fCovMatrix=NULL;}
  75
  76   template <class T> void     GetSigmaXYZ(T *sigma) const;
  77
  78   Double_t GetChi2perNDF() const { return fChi2perNDF; }
  79   Double_t GetChi2() const { return fChi2perNDF*(Double_t)GetNDF(); }
  80   Int_t    GetNDF() const { return 2*GetNContributors()-3; }
  81
  82   Short_t  GetID() const { return fID; }
  83   void     SetID(Short_t id) { fID=id; }
  84
  85   Char_t   GetType() const { return fType; }
  86   void     SetType(AODVtx_t vtype) { fType=vtype; }
  87
  88   TObject* GetParent() const   { return fParent.GetObject(); }
  89   Bool_t   HasParent(TObject *parent) const { return (fParent.GetObject() == parent) ? kTRUE : kFALSE; }
  90
  91   void     AddDaughter(TObject *daughter) { fDaughters.Add(daughter);}
  92   void     RemoveDaughter(TObject *daughter) { fDaughters.Remove(daughter); }
  93   TObject* GetDaughter(Int_t i) { return fDaughters.At(i); }
  94   Bool_t   HasDaughter(TObject *daughter) const;
  95   Int_t    GetNDaughters() const { return fDaughters.GetEntriesFast(); }
  96   Int_t    GetNContributors() const;
  97
  98   // covariance matrix elements after rotation by phi around z-axis
  99   // and, then, by theta around new y-axis
 100   Double_t  RotatedCovMatrixXX(Double_t phi = 0., Double_t theta = 0.) const;
 101   Double_t  RotatedCovMatrixXY(Double_t phi = 0., Double_t theta = 0.) const;
 102   Double_t  RotatedCovMatrixYY(Double_t phi = 0.) const;
 103   Double_t  RotatedCovMatrixXZ(Double_t phi = 0., Double_t theta = 0.) const;
 104   Double_t  RotatedCovMatrixYZ(Double_t phi = 0., Double_t theta = 0.) const;
 105   Double_t  RotatedCovMatrixZZ(Double_t phi = 0., Double_t theta = 0.) const;
 106
 107   template <class T, class P> void     PhiAndThetaToVertex(AliAODVertex *vtx, P &phi, T &theta) const;
 108   Double_t  DistanceToVertex(AliAODVertex *vtx) const;
 109   Double_t  ErrorDistanceToVertex(AliAODVertex *vtx) const;
 110   Double_t  DistanceXYToVertex(AliAODVertex *vtx) const;
 111   Double_t  ErrorDistanceXYToVertex(AliAODVertex *vtx) const;
 112
 113   void     PrintIndices() const;
 114   void     Print(Option_t* option = "") const;
 115
 116  private :
 117
 118   Double32_t      fPosition[3]; // vertex position
 119   Double32_t      fChi2perNDF;  // chi2/NDF of vertex fit
 120   Short_t         fID;          // vertex ID; corresponds to the array index of the appropriate ESD container
 121   Char_t          fType;        // vertex type
 122
 123   AliAODRedCov<3> *fCovMatrix;  // vertex covariance matrix; values of and below the diagonal
 124   TRef            fParent;      // reference to the parent particle
 125   TRefArray       fDaughters;   // references to the daughter particles
 126
 127   ClassDef(AliAODVertex,3);
 128 };
 129
 130 #endif