New version of the V0 finder (M.Ivanov)
[u/mrichter/AliRoot.git] / STEER / AliESDV0MI.h
1 #ifndef ALIESDV0MI_H
2 #define ALIESDV0MI_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: Marian Ivanov marian.ivanov@cern.ch
13 //-------------------------------------------------------------------------
14
15 #include "AliESDv0.h"
16 #include "AliExternalTrackParam.h"
17
18 class AliESDtrack;
19
20 class AliESDV0MI :  public AliESDv0 {
21 public:
22   //  friend class AliITStrackerMI;
23   AliESDV0MI();             //constructor
24   //
25   const AliExternalTrackParam *GetParamP() const {return &fParamP;}
26   const AliExternalTrackParam *GetParamM() const {return &fParamM;}
27   void SetP(const AliExternalTrackParam & paramp); 
28   void SetM(const AliExternalTrackParam & paramd);
29   void SetRp(const Double_t *rp);
30   void SetRm(const Double_t *rm);
31   void UpdatePID(Double_t pidp[5], Double_t pidm[5]);
32   void SetStatus(Int_t status){fStatus=status;}
33   Int_t  GetStatus() const {return fStatus;}
34   Float_t GetEffMass(UInt_t p1, UInt_t p2);
35   Float_t GetProb(UInt_t p1, UInt_t p2);
36   void Update(Float_t vertex[3]);            //update
37   void SetID(Int_t id){fID =id;}
38   Int_t GetID() const { return fID;}
39   Int_t GetIndex(Int_t i) const {return fIndex[i];}
40   void SetIndex(Int_t i, Int_t ind) {fIndex[i]=ind;}
41   void SetDist1(Double_t d1) {fDist1=d1;}
42   void SetDist2(Double_t d2) {fDist2=d2;}
43   Double_t GetDist1() const {return fDist1;}
44   Double_t GetDist2() const {return fDist2;}
45   Double_t *GetAnglep() {return fAngle;}
46   Double_t GetRr() const {return fRr;}
47   void SetRr(Double_t rr) {fRr=rr;}
48   Double_t *GetPMp() {return fPM;}
49   Double_t *GetPPp() {return fPP;}
50   Double_t *GetXrp() {return fXr;}
51   Double_t GetXr(Int_t i) const {return fXr[i];}
52   Double_t GetDistSigma() const {return fDistSigma;}
53   void SetDistSigma(Double_t ds) {fDistSigma=ds;}
54   Double_t GetDistNorm() const {return fDistNorm;}
55   void SetDistNorm(Double_t ds) {fDistNorm=ds;}
56   Float_t GetChi2Before() const {return fChi2Before;}
57   void SetChi2Before(Float_t cb) {fChi2Before=cb;}
58   Float_t GetChi2After() const {return fChi2After;}
59   void SetChi2After(Float_t ca) {fChi2After=ca;}
60   Float_t GetPointAngle() const {return fPointAngle;}
61   void SetOrder(Int_t i, Int_t ord) {fOrder[i]=ord;}
62   Float_t GetNAfter() const {return fNAfter;}
63   void SetNAfter(Float_t na) {fNAfter=na;}
64   Float_t GetNBefore() const {return fNBefore;}
65   void SetNBefore(Float_t nb) {fNBefore=nb;}  
66   void SetLab(Int_t i, Int_t lab) {fLab[i]=lab;}
67   void SetCausality(Float_t pb0, Float_t pb1, Float_t pa0, Float_t pa1);
68   const Float_t * GetCausalityP() const {return fCausality;}
69 private:
70   AliExternalTrackParam fParamP;
71   AliExternalTrackParam fParamM;
72   Float_t        fRP[5];         // combined pid positive
73   Float_t        fRM[5];         // combined pid positive
74   Int_t          fID;
75   Int_t          fLab[2];     //MC label of the partecle
76   Int_t          fIndex[2];   //reconstructed labels of the tracks
77   //
78   //  
79   Double_t       fDist1;    //info about closest distance according closest MC - linear DCA
80   Double_t       fDist2;    //info about closest distance parabolic DCA
81   //
82   Double_t       fPP[3];    //momentum  positive   - according approx at DCA
83   Double_t       fPM[3];    //momentum negative
84   //
85   Double_t       fXr[3];      //rec. position according helix
86   Double_t       fAngle[3];   //three angles
87   Double_t       fRr;         //rec position of the vertex 
88   Int_t          fStatus;       //status  - 1 - TPC V0  2- ITS V0  4- accepted - 0 -rejected
89   Int_t          fRow0;         // critical layer
90   Int_t          fOrder[3]; //order of the vertex 
91   //  quality information
92   Double_t       fDistNorm; //normalized  DCA
93   Double_t       fDistSigma; //sigma of distance
94   Float_t        fCausality[4];  // causality information - see comments in SetCausality
95   Float_t        fChi2Before;   //chi2 of the tracks before V0
96   Float_t        fNBefore;      // number of possible points before V0
97   Float_t        fChi2After;   // chi2 of the tracks after V0
98   Float_t        fNAfter;      // number of possible points after V0
99   Float_t        fPointAngleFi; //point angle fi
100   Float_t        fPointAngleTh; //point angle theta
101   Float_t        fPointAngle;   //point angle full
102
103   ClassDef(AliESDV0MI,2)      // ESD V0 vertex
104 };
105
106
107 #endif