Method GetCellPosition added.

[u/mrichter/AliRoot.git] / STEER / AliVertexerTracks.h

diff --git a/STEER/AliVertexerTracks.h b/STEER/AliVertexerTracks.h
index cc693f4dd7b4a32158823b390d841c1085fcd220..a5e5f4f00fc69cd7f3f1e49d7bbcd38aff2f3072 100644 (file)
--- a/STEER/AliVertexerTracks.h
+++ b/STEER/AliVertexerTracks.h
@@ -28,7 +28,9 @@
 #include "AliExternalTrackParam.h"
 #include "AliLog.h"
 
+class AliVEvent;
 class AliESDEvent;
+class AliStrLine;
 
 class AliVertexerTracks : public TObject {
   
@@ -37,17 +39,22 @@ class AliVertexerTracks : public TObject {
   AliVertexerTracks(Double_t fieldkG); 
   virtual ~AliVertexerTracks();
 
-  AliESDVertex* FindPrimaryVertex(const AliESDEvent *esdEvent);
+  AliESDVertex* FindPrimaryVertex(AliVEvent *vEvent);
   AliESDVertex* FindPrimaryVertex(TObjArray *trkArrayOrig,UShort_t *idOrig);
   AliESDVertex* VertexForSelectedTracks(TObjArray *trkArray,UShort_t *id,
                                        Bool_t optUseFitter=kTRUE,
-                                       Bool_t optPropagate=kTRUE);
+                                       Bool_t optPropagate=kTRUE,
+                                       Bool_t optUseDiamondConstraint=kFALSE);
   AliESDVertex* VertexForSelectedESDTracks(TObjArray *trkArray,
                                        Bool_t optUseFitter=kTRUE,
-                                       Bool_t optPropagate=kTRUE);
+                                       Bool_t optPropagate=kTRUE,
+                                       Bool_t optUseDiamondConstraint=kFALSE);
   AliESDVertex* RemoveTracksFromVertex(AliESDVertex *inVtx,
                                       TObjArray *trkArray,UShort_t *id,
-                                      Float_t *diamondxy); 
+                                      Float_t *diamondxy) const; 
+  AliESDVertex* RemoveConstraintFromVertex(AliESDVertex *inVtx,
+                                          Float_t *diamondxyz,
+                                          Float_t *diamondcov) const;
   void  SetITSMode(Double_t dcacut=0.1,
                   Double_t dcacutIter0=0.1,
                   Double_t maxd0z0=0.5,
@@ -57,7 +64,9 @@ class AliVertexerTracks : public TObject {
                   Double_t mindetfitter=100.,
                   Double_t maxtgl=1000.,
                   Double_t fidR=3.,
-                  Double_t fidZ=30.); 
+                  Double_t fidZ=30.,
+                  Int_t finderAlgo=1,
+                  Int_t finderAlgoIter0=4); 
   void  SetTPCMode(Double_t dcacut=0.1,
                   Double_t dcacutIter0=1.0,
                   Double_t maxd0z0=5.0,
@@ -67,7 +76,9 @@ class AliVertexerTracks : public TObject {
                   Double_t mindetfitter=0.1,
                   Double_t maxtgl=1.5, 
                   Double_t fidR=3.,
-                  Double_t fidZ=30.);
+                  Double_t fidZ=30.,
+                  Int_t finderAlgo=1,
+                  Int_t finderAlgoIter0=4); 
   void  SetCuts(Double_t *cuts);
   void  SetConstraintOff() { fConstraint=kFALSE; return; }
   void  SetConstraintOn() { fConstraint=kTRUE; return; }
@@ -75,10 +86,12 @@ class AliVertexerTracks : public TObject {
   void  SetDCAcutIter0(Double_t maxdca) { fDCAcutIter0=maxdca; return; }
   void  SetFinderAlgorithm(Int_t opt=1) { fAlgo=opt; return; }
   void  SetITSrefitRequired() { fITSrefit=kTRUE; return; }
+  Bool_t GetITSrefitRequired() const { return fITSrefit; }
   void  SetITSrefitNotRequired() { fITSrefit=kFALSE; return; }
   void  SetFiducialRZ(Double_t r=3,Double_t z=30) { fFiducialR=r; fFiducialZ=z; return; }
   void  SetMaxd0z0(Double_t maxd0z0=0.5) { fMaxd0z0=maxd0z0; return; }
   void  SetMinClusters(Int_t n=5) { fMinClusters=n; return; }
+  Int_t GetMinClusters() const { return fMinClusters; }
   void  SetMinTracks(Int_t n=1) { fMinTracks=n; return; }
   void  SetNSigmad0(Double_t n=3) { fNSigma=n; return; }
   Double_t GetNSigmad0() const { return fNSigma; }
@@ -98,6 +111,7 @@ class AliVertexerTracks : public TObject {
   static void GetStrLinDerivMatrix(Double_t *p0,Double_t *p1,Double_t (*m)[3],Double_t *d);
   static void GetStrLinDerivMatrix(Double_t *p0,Double_t *p1,Double_t *sigmasq,Double_t (*m)[3],Double_t *d);
   static AliESDVertex TrackletVertexFinder(TClonesArray *lines, Int_t optUseWeights=0);
+  static AliESDVertex TrackletVertexFinder(AliStrLine **lines, const Int_t knacc, Int_t optUseWeights=0);
   void     SetFieldkG(Double_t field=-999.) { fFieldkG=field; return; }
   Double_t GetFieldkG() const { 
     if(fFieldkG<-99.) AliFatal("Field value not set");
@@ -149,6 +163,7 @@ class AliVertexerTracks : public TObject {
   Double_t  fFiducialZ;       // length of fiducial cylinder for tracks
   Double_t  fnSigmaForUi00;   // n. sigmas from finder in TrackToPoint
   Int_t     fAlgo;            // option for vertex finding algorythm
+  Int_t     fAlgoIter0;       // this is for iteration 0
   // fAlgo=1 (default) finds minimum-distance point among all selected tracks
   //         approximated as straight lines 
   //         and uses errors on track parameters as weights