STEER/AliESDv0.h

   1 #ifndef ALIESDV0_H
   2 #define ALIESDV0_H
   3
   4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   5  * See cxx source for full Copyright notice                               */
   6
   7 /* $Id$ */
   8
   9 //-------------------------------------------------------------------------
  10 //                          ESD V0 Vertex Class
  11 //          This class is part of the Event Summary Data set of classes
  12 //    Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch
  13 //    Modified by: Marian Ivanov,  CERN, Marian.Ivanov@cern.ch
  14 //            and  Boris Hippolyte,IPHC, hippolyt@in2p3.fr
  15 //-------------------------------------------------------------------------
  16
  17 #include <TObject.h>
  18 #include <TPDGCode.h>
  19 #include "AliESDV0Params.h"
  20 #include "AliExternalTrackParam.h"
  21
  22 class AliESDv0 : public TObject {
  23 public:
  24   AliESDv0();
  25   AliESDv0(const AliExternalTrackParam &t1, Int_t i1,
  26            const AliExternalTrackParam &t2, Int_t i2);
  27
  28   AliESDv0(const AliESDv0&);
  29   virtual ~AliESDv0();
  30
  31   Double_t ChangeMassHypothesis(Int_t code=kK0Short);
  32
  33   Int_t    GetPdgCode() const {return fPdgCode;}
  34   Double_t GetEffMass() const {return fEffMass;}
  35   Double_t GetChi2V0() const {return fChi2V0;}
  36   void     GetPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const;
  37   void     GetNPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const;
  38   void     GetPPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const;
  39   void     GetXYZ(Double_t &x, Double_t &y, Double_t &z) const;
  40   Double_t GetD(Double_t x0=0.,Double_t y0=0.,Double_t z0=0.) const;
  41   Int_t    GetNindex() const {return fNidx;}
  42   Int_t    GetPindex() const {return fPidx;}
  43   void     SetESDindexes(Int_t ip, Int_t im){fNidx=ip;fPidx=im;}
  44   void     SetDcaV0Daughters(Double_t rDcaV0Daughters=0.);
  45   Double_t GetDcaV0Daughters() {return fDcaV0Daughters;}
  46   Double_t GetV0CosineOfPointingAngle(Double_t&, Double_t&, Double_t&) const;
  47   void SetOnFlyStatus(Bool_t status){fOnFlyStatus=status;}
  48   Bool_t GetOnFlyStatus() const {return fOnFlyStatus;}
  49
  50   // **** The following member functions need to be revised ***
  51
  52   Double_t GetSigmaY();     // sigma of y coordinate at vertex posistion
  53   Double_t GetSigmaZ();     // sigma of z coordinate at vertex posistion
  54   Double_t GetSigmaAP0();   // calculate sigma of Point angle resolution at vertex pos.
  55   Double_t GetSigmaD0();    // calculate sigma of position resolution at vertex pos.
  56   Double_t GetEffectiveSigmaAP0();   // calculate sigma of point angle resolution at vertex pos. effecive parameterization
  57   Double_t GetEffectiveSigmaD0();    // calculate sigma of position resolution at vertex pos.
  58   Double_t GetMinimaxSigmaAP0();    // calculate mini-max sigma of point angle resolution
  59   Double_t GetMinimaxSigmaD0();     // calculate mini-max sigma of dca resolution
  60   Double_t GetLikelihoodAP(Int_t mode0, Int_t mode1);   // get likelihood for point angle
  61   Double_t GetLikelihoodD(Int_t mode0, Int_t mode1);    // get likelihood for DCA
  62   Double_t GetLikelihoodC(Int_t mode0, Int_t mode1);    // get likelihood for Causality
  63   //
  64   //
  65   const AliExternalTrackParam *GetParamP() const {return &fParamP;}
  66   const AliExternalTrackParam *GetParamM() const {return &fParamN;}
  67   static const AliESDV0Params & GetParameterization(){return fgkParams;}
  68   void SetP(const AliExternalTrackParam & paramp);
  69   void SetM(const AliExternalTrackParam & paramd);
  70   void SetRp(const Double_t *rp);
  71   void SetRm(const Double_t *rm);
  72   void UpdatePID(Double_t pidp[5], Double_t pidm[5]);
  73   void SetStatus(Int_t status){fStatus=status;}
  74   Int_t GetStatus() const {return fStatus;}
  75   Float_t GetEffMass(UInt_t p1, UInt_t p2);
  76   Float_t GetProb(UInt_t p1, UInt_t p2);
  77   void SetID(Int_t id){fID =id;}
  78   Int_t GetID() const { return fID;}
  79   Int_t GetIndex(Int_t i) const {return fIndex[i];}
  80   void SetIndex(Int_t i, Int_t ind) {fIndex[i]=ind;}
  81   void SetDist1(Double_t d1) {fDist1=d1;}
  82   void SetDist2(Double_t d2) {fDist2=d2;}
  83   Double_t GetDist1() const {return fDist1;}
  84   Double_t GetDist2() const {return fDist2;}
  85   Double_t *GetAnglep() {return fAngle;}
  86   Double_t GetRr() const {return fRr;}
  87   void SetRr(Double_t rr) {fRr=rr;}
  88   Double_t *GetPMp() {return fPM;}
  89   Double_t *GetPPp() {return fPP;}
  90   Double_t *GetXrp() {return fXr;}
  91   Double_t GetXr(Int_t i) const {return fXr[i];}
  92   Double_t GetDistSigma() const {return fDistSigma;}
  93   void SetDistSigma(Double_t ds) {fDistSigma=ds;}
  94   Double_t GetDistNorm() const {return fDistNorm;}
  95   void SetDistNorm(Double_t ds) {fDistNorm=ds;}
  96   Float_t GetChi2Before() const {return fChi2Before;}
  97   void SetChi2Before(Float_t cb) {fChi2Before=cb;}
  98   Float_t GetChi2After() const {return fChi2After;}
  99   void SetChi2After(Float_t ca) {fChi2After=ca;}
 100   Float_t GetPointAngle() const {return fPointAngle;}
 101   void SetOrder(Int_t i, Int_t ord) {fOrder[i]=ord;}
 102   Float_t GetNAfter() const {return fNAfter;}
 103   void SetNAfter(Float_t na) {fNAfter=na;}
 104   Float_t GetNBefore() const {return fNBefore;}
 105   void SetNBefore(Float_t nb) {fNBefore=nb;}
 106   Int_t GetLab(Int_t i) const {return fLab[i];}
 107   void  SetLab(Int_t i, Int_t lab) {fLab[i]=lab;}
 108   void SetCausality(Float_t pb0, Float_t pb1, Float_t pa0, Float_t pa1);
 109   const Float_t * GetCausalityP() const {return fCausality;}
 110   void SetClusters(Int_t *clp, Int_t *clm);
 111   const Int_t * GetClusters(Int_t i) const {return fClusters[i];}
 112   void SetNormDCAPrim(Float_t nd0, Float_t nd1){fNormDCAPrim[0] = nd0; fNormDCAPrim[1]=nd1;}
 113   const Float_t  *GetNormDCAPrimP() const {return fNormDCAPrim;}
 114
 115 protected:
 116   Bool_t   fOnFlyStatus;    // if kTRUE, then this V0 is recontructed
 117                             // "on fly" during the tracking
 118
 119   Int_t    fPdgCode;        // reconstructed V0's type (PDG code)
 120   Double_t fEffMass;        // reconstructed V0's effective mass
 121   Double_t fDcaV0Daughters; // dca between V0's daughters
 122   Double_t fChi2V0;         // V0's chi2 value
 123   Double_t fPos[3];         // V0's position (global)
 124   Double_t fPosCov[6];      // covariance matrix of the vertex position
 125
 126   Int_t    fNidx;           // index of the negative daughter
 127   Double_t fNmom[3];        // momentum of the negative daughter (global)
 128   Double_t fNmomCov[6];     // covariance matrix of the negative daughter mom.
 129
 130   Int_t    fPidx;           // index of the positive daughter
 131   Double_t fPmom[3];        // momentum of the positive daughter (global)
 132   Double_t fPmomCov[6];     // covariance matrix of the positive daughter mom.
 133
 134   // **** The following data members need to be revised ***
 135
 136   AliExternalTrackParam fParamP;  // external parameters of positive particle
 137   AliExternalTrackParam fParamN;  // external parameters of negative particle
 138   Float_t        fRP[5];         // combined pid positive
 139   Float_t        fRM[5];         // combined pid positive
 140   Int_t          fID;            // ID number of the V0 in the ESDV0 container
 141   Int_t          fLab[2];         // MC label of the particle
 142   Int_t          fIndex[2];       // reconstructed labels of the tracks
 143   Int_t          fClusters[2][6]; //! its clusters
 144   //
 145   //
 146   Float_t        fNormDCAPrim[2];  // normalize distance to the priary vertex
 147   Double_t       fDist1;    //info about closest distance according closest MC - linear DCA
 148   Double_t       fDist2;    //info about closest distance parabolic DCA
 149   //
 150   Double_t       fPP[3];    //momentum  positive   - according approx at DCA
 151   Double_t       fPM[3];    //momentum negative
 152   //
 153   Double_t       fXr[3];      //rec. position according helix
 154   Double_t       fAngle[3];   //three angles
 155   Double_t       fRr;         //rec position of the vertex
 156   Int_t          fStatus;       //status
 157   Int_t          fRow0;         // critical layer
 158   Int_t          fOrder[3]; //order of the vertex
 159   //  quality information
 160   Double_t       fDistNorm; //normalized  DCA
 161   Double_t       fDistSigma; //sigma of distance
 162   Float_t        fCausality[4];  // causality information - see comments in SetCausality
 163   Float_t        fChi2Before;   //chi2 of the tracks before V0
 164   Float_t        fNBefore;      // number of possible points before V0
 165   Float_t        fChi2After;   // chi2 of the tracks after V0
 166   Float_t        fNAfter;      // number of possible points after V0
 167   Float_t        fPointAngleFi; //point angle fi
 168   Float_t        fPointAngleTh; //point angle theta
 169   Double_t       fPointAngle;   //point angle full
 170   //
 171   // parameterization coefficients
 172   static AliESDV0Params fgkParams;  // resolution and likelihood parameterization
 173
 174 private:
 175   AliESDv0& operator=(const AliESDv0&);
 176
 177   ClassDef(AliESDv0,2)      // ESD V0 vertex
 178 };
 179
 180 inline
 181 void AliESDv0::GetNPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const {
 182 px=fNmom[0]; py=fNmom[1]; pz=fNmom[2];
 183 }
 184
 185 inline
 186 void AliESDv0::GetPPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const {
 187 px=fPmom[0]; py=fPmom[1]; pz=fPmom[2];
 188 }
 189
 190 inline
 191 void AliESDv0::SetDcaV0Daughters(Double_t rDcaV0Daughters){
 192   fDcaV0Daughters=rDcaV0Daughters;
 193 }
 194
 195 #endif