f51939efe53d8c823d2d1e53bb3f422dad6f8115
[u/mrichter/AliRoot.git] / ITS / AliITSVertexerTracks.h
1 #ifndef ALIITSVERTEXERTRACKS_H
2 #define ALIITSVERTEXERTRACKS_H
3 /* Copyright(c) 1998-2003, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6
7 //-------------------------------------------------------
8 // Class for primary vertex determination with ITS tracks
9 //
10 //   Origin: A.Dainese, Padova, andrea.dainese@pd.infn.it
11 //           M.Masera,  Torino, massimo.masera@to.infn.it 
12 //-------------------------------------------------------
13
14 /*****************************************************************************
15  *                                                                           *
16  * This class determines the primary vertex position using ITS tracks.       *
17  * This is done in two steps:                                                *
18  * 1) Vertex Finding: a reasonable estimate of the vertex position is        *
19  *    obtained from a mean of "points of closest approach" between all       *
20  *    possible pairs of tracks.                                              *
21  * 2) Vertex Fitting: once tracks are propagated to the position given by    *
22  *    first step, the optimal estimate of the position of vertex is obtained *
23  *    from a weighted average of the track positions. A covariance           *
24  *    matrix and a chi2 for the vertex are given.                            *
25  *                                                                           *
26  *****************************************************************************/
27
28 #include "AliKalmanTrack.h"
29 #include "AliITSVertexer.h"
30
31 #include <TObjArray.h>
32
33 class TTree; 
34 class AliESDVertex; 
35 class AliESD;
36
37 class AliITSVertexerTracks : public AliITSVertexer {
38   
39  public:
40   // default constructor
41   AliITSVertexerTracks();  
42   // standard constructor     
43   AliITSVertexerTracks(TFile *inFile,TFile *outFile,
44                 const AliMagF *map,Int_t fEv=0,Int_t lEv=0,
45                        Double_t xStart=0.,Double_t yStart=0.);
46   // alternative constructor
47   AliITSVertexerTracks(const AliMagF *map, TString fn,
48                        Double_t xStart=0,Double_t yStart=0); 
49   // destructor
50   virtual ~AliITSVertexerTracks();
51   // return vertex from the set of tracks in the tree
52   AliESDVertex* VertexOnTheFly(TTree &trkTree);
53   // computes the vertex for the current event
54   virtual AliESDVertex* FindVertexForCurrentEvent(Int_t evnumb);
55   // computes the vertex for the current event using the ESD
56   AliESDVertex*         FindVertexForCurrentEvent(AliESD *esdEvent);
57   // computes the vertex for each event and stores it on file
58   virtual void  FindVertices();
59   // computes the vertex for each event and stores it in the ESD
60   void FindVerticesESD();
61   virtual void  PrintStatus() const;
62   void  SetFieldMap(const AliMagF *map)const{AliKalmanTrack::SetFieldMap(map);}
63   void  SetMinTracks(Int_t n=2) { fMinTracks = n; return; }
64   void  SetSkipTracks(Int_t n,Int_t *skipped); 
65   void  SetVtxStart(Double_t x=0,Double_t y=0) 
66     { fNominalPos[0]=x; fNominalPos[1]=y; return; }
67   
68  private:
69     // copy constructor (NO copy allowed: the constructor is protected
70     // to avoid misuse)
71     AliITSVertexerTracks(const AliITSVertexerTracks& vtxr);
72     // assignment operator (NO assignment allowed)
73     AliITSVertexerTracks& operator=(const AliITSVertexerTracks& /* vtxr */);
74   TFile    *fInFile;          // input file (with tracks)
75   TFile    *fOutFile;         // output file for vertices
76   Double_t  fInitPos[3];      // vertex position after vertex finder
77   Double_t  fNominalPos[2];   // initial knowledge on vertex position
78   Int_t     fMinTracks;       // minimum number of tracks
79   Double_t  fMaxChi2PerTrack; // maximum contribition to the chi2 
80   TObjArray fTrkArray;        // array with tracks to be processed
81   Int_t     *fTrksToSkip;     // tracks to be skipped for find and fit 
82   Int_t     fNTrksToSkip;     // number of tracks to be skipped 
83
84   Bool_t   CheckField() const; 
85   void     ComputeMaxChi2PerTrack(Int_t nTracks);
86   Int_t    PrepareTracks(TTree &trkTree);
87   void     TooFewTracks();
88   void     VertexFinder();
89   void     VertexFitter();
90
91   ClassDef(AliITSVertexerTracks,1) // 3D Vertexing with ITS tracks 
92 };
93
94 #endif
95
96
97