HLT/global/AliFlatESDVertex.h

   1 #ifndef ALIFLATESDVERTEX_H
   2 #define ALIFLATESDVERTEX_H
   3
   4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   5  * See cxx source for full Copyright notice                               *
   6  * Primary Authors : Sergey Gorbunov, Jochen Thaeder, Chiara Zampolli     */
   7
   8 /**
   9  * >> Flat structure representing a ESD vertex <<
  10  */
  11
  12 #include "Rtypes.h"
  13 #include "AliVMisc.h"
  14 #include "AliESDVertex.h"
  15
  16 struct AliFlatESDVertex
  17 {
  18   // -- Constructor / Destructors
  19
  20   AliFlatESDVertex();
  21  ~AliFlatESDVertex() {}
  22
  23   // constructor and method for reinitialisation of virtual table
  24   AliFlatESDVertex( AliVConstructorReinitialisationFlag );
  25   void Reinitialize() const {} // no virtual table - do nothing
  26
  27   //--
  28
  29   Double32_t fPosition[3];    // vertex position
  30   Double32_t fCov[6];  // vertex covariance matrix
  31   Int_t    fNContributors;  // # of tracklets/tracks used for the estimate
  32   Double32_t fChi2;  // chi2 of vertex fit
  33   /*
  34     Double32_t fSigma;          // track dispersion around found vertex
  35     Char_t fID;       // ID of this vertex within an ESD event
  36     Char_t fBCID;     // BC ID assigned to vertex
  37   */
  38
  39
  40   void SetFromESDVertex(const AliESDVertex &v );
  41   void GetESDVertex( AliESDVertex &v ) const;
  42
  43   Double32_t GetX() const { return fPosition[0]; }
  44   Double32_t GetY() const { return fPosition[1]; }
  45   Double32_t GetZ() const { return fPosition[2]; }
  46
  47   void GetXYZ(Double_t pos[3]) const { for(Int_t j=0; j<3; j++) pos[j]=fPosition[j]; }
  48   void SetXYZ(Double_t pos[3]) { for(Int_t j=0; j<3; j++) fPosition[j]=pos[j]; }
  49
  50   void   SetX(Double_t xVert) {fPosition[0]=xVert; }
  51   void   SetY(Double_t yVert) {fPosition[1]=yVert; }
  52   void   SetZ(Double_t zVert) {fPosition[2]=zVert; }
  53   void   SetNContributors(Int_t nContr) {fNContributors=nContr; }
  54
  55
  56   Int_t    GetNContributors() const { return fNContributors; }
  57
  58   /*
  59   void     GetCovarianceMatrix(Double_t covmatrix[6]) const;
  60   void     SetCovarianceMatrix(const Double_t *) {}
  61
  62   Double_t GetChi2perNDF() const {return -999.;}
  63   Double_t GetChi2() const {return -999.;}
  64   void     SetChi2(Double_t ) {}
  65   Int_t    GetNDF() const {return -999;}
  66
  67   void     GetSigmaXYZ(Double_t sigma[3]) const;
  68   void     GetCovMatrix(Double_t covmatrix[6]) const;
  69   void     GetCovarianceMatrix(Double_t covmatrix[6]) const
  70                     {GetCovMatrix(covmatrix);}
  71   void     GetSNR(Double_t snr[3]) const;
  72   void     SetCovarianceMatrix(const Double_t *cov);
  73
  74   Double_t GetXRes() const {return TMath::Sqrt(fCovXX);}
  75   Double_t GetYRes() const {return TMath::Sqrt(fCovYY);}
  76   Double_t GetZRes() const {return TMath::Sqrt(fCovZZ);}
  77   Double_t GetXSNR() const { return fSNR[0]; }
  78   Double_t GetYSNR() const { return fSNR[1]; }
  79   Double_t GetZSNR() const { return fSNR[2]; }
  80   void     SetSNR(double snr, int i) {if (i<3 && i>=0) fSNR[i] = snr;}
  81
  82   Double_t GetChi2() const { return fChi2; }
  83   void     SetChi2(Double_t chi) { fChi2 = chi; }
  84   Double_t GetChi2toNDF() const
  85     { return fChi2/(2.*(Double_t)fNContributors-3.); }
  86   Double_t GetChi2perNDF() const { return GetChi2toNDF();}
  87   Int_t    GetNDF() const {return (2*fNContributors-3);}
  88
  89   void     Print(Option_t* option = "") const;
  90   void     PrintStatus() const {Print();}
  91
  92   void     Reset() { SetToZero(); SetName("Vertex"); }
  93
  94   void     SetID(Char_t id) {fID=id;}
  95   Char_t   GetID() const {return fID;}
  96   //
  97   Double_t GetWDist(const AliESDVertex* v) const;
  98   */
  99
 100   static size_t GetSize() { return sizeof(AliFlatESDVertex); }
 101
 102 };
 103
 104 inline AliFlatESDVertex::AliFlatESDVertex() :
 105 fNContributors(0),
 106 fChi2(0)
 107 {
 108   for( int i=0; i<3; i++) fPosition[i] = -9999;
 109   for( int i=0; i<6; i++) fCov[i] = -9999;
 110 }
 111
 112 #pragma GCC diagnostic ignored "-Weffc++"
 113 inline AliFlatESDVertex::AliFlatESDVertex( AliVConstructorReinitialisationFlag ){}  // do nothing
 114 #pragma GCC diagnostic warning "-Weffc++"
 115
 116 inline void AliFlatESDVertex::SetFromESDVertex(const AliESDVertex &v )
 117 {
 118   fPosition[0] = v.GetX();
 119   fPosition[1] = v.GetY();
 120   fPosition[2] = v.GetZ();
 121   Double_t c[6];
 122   v.GetCovarianceMatrix( c );
 123   for( int i=0; i<6; i++) fCov[i] = c[i];
 124   fNContributors = v.GetNContributors();
 125   fChi2 = v.GetChi2();
 126 }
 127
 128 inline void AliFlatESDVertex::GetESDVertex( AliESDVertex &v ) const
 129 {
 130   v.Reset();
 131   Double_t p[3] = { fPosition[0], fPosition[1], fPosition[2] };
 132   v.SetXYZ( p );
 133   Double_t c[6];
 134   for( int i=0; i<6; i++) c[i] = fCov[i];
 135   v.SetCovarianceMatrix( c );
 136   v.SetNContributors( fNContributors );
 137   v.SetChi2( fChi2 );
 138 }
 139
 140 typedef struct AliFlatESDVertex AliFlatESDVertex;
 141
 142 #endif