Update for Ds
[u/mrichter/AliRoot.git] / ITS / AliITSVertexer.h
CommitLineData
c5f0f3c1 1#ifndef ALIITSVERTEXER_H
2#define ALIITSVERTEXER_H
3
2257f27e 4#include<AliVertexer.h>
ff44c37c 5#include "AliITSDetTypeRec.h"
c5f0f3c1 6
7///////////////////////////////////////////////////////////////////
8// //
2257f27e 9// Base class for primary vertex reconstruction for ITS //
c5f0f3c1 10// //
11///////////////////////////////////////////////////////////////////
12
8c42830a 13/* $Id$ */
14
41b19549 15class TString;
c5f0f3c1 16
2257f27e 17class AliITSVertexer : public AliVertexer {
c5f0f3c1 18
19 public:
20 // default constructor
21 AliITSVertexer();
27167524 22 virtual ~AliITSVertexer();
308c2f7c 23 virtual AliESDVertex *FindVertexForCurrentEvent(TTree *itsClusterTree)=0;
24 virtual void PrintStatus() const = 0;
25
6b4d9537 26 virtual void ResetVertex();
308c2f7c 27 void FindMultiplicity(TTree *itsClusterTree);
28 void SetFirstEvent(Int_t ev){fFirstEvent = ev;}
29 void SetLastEvent(Int_t ev){fLastEvent = ev;}
5af4a2d0 30 static Float_t GetPipeRadius() {return fgkPipeRadius;}
308c2f7c 31 void SetLaddersOnLayer2(Int_t ladwid=4);
1ff24d0a 32 void SetComputeMultiplicity(Bool_t opt=kTRUE){fComputeMultiplicity=opt;}
8c42830a 33 virtual void SetUseModule(Int_t imod, Bool_t optUse){
34 if(imod>=0 && imod<kNSPDMod) fUseModule[imod]=optUse;
1ff24d0a 35 }
8c42830a 36 virtual Bool_t IsModuleUsed(Int_t imod) const {
37 if(imod>=0 && imod<kNSPDMod) return fUseModule[imod];
38 else return 0;
39 }
b96ee725 40
6b4d9537 41 virtual AliESDVertex* GetAllVertices(Int_t &novertices) const {novertices = fNoVertices; return fVertArray; }
42
06a7cbee 43 AliITSDetTypeRec *GetDetTypeRec() const {return fDetTypeRec;}
44 virtual void SetDetTypeRec(AliITSDetTypeRec *ptr){fDetTypeRec = ptr;}
8c42830a 45 enum{kNSPDMod=240};
308c2f7c 46
47 // Methods containing run-loaders, should be moved to some other class
48 void Init(TString filename);
49 void WriteCurrentVertex();
50 void FindVertices();
c5f0f3c1 51
8c32ba44 52 Int_t GetMinTrackletsForPilup() const {return fMinTrackletsForPilup;}
53 void SetMinTrackletsForPilup(Int_t mint=2) {fMinTrackletsForPilup=mint;}
54 Bool_t IsPileup() const{return fIsPileup;}
55 Float_t GetZPileupV() const {
56 if(IsPileup()) return fZpuv;
57 else return -9999999.;
58 }
59 Int_t GetNTrackletsPileupV() const {
60 if(IsPileup()) return fNTrpuv;
61 else return -2;
62 }
c5f0f3c1 63 protected:
308c2f7c 64 static const Float_t fgkPipeRadius; // beam pipe radius (cm)
65 UShort_t *fLadders; // array with layer1-layer2 ladders correspondances
66 Int_t fLadOnLay2; // (2*fLadOnLay2+1)=number of layer2 ladders
67 // associated to a layer1 ladder
1ff24d0a 68 Bool_t fComputeMultiplicity; // flag to switch on/off tracklet calculation
8c42830a 69 Bool_t fUseModule[kNSPDMod]; // flag for enabling/disabling SPD modules
06a7cbee 70 AliITSDetTypeRec *fDetTypeRec; //! pointer to DetTypeRec
8c32ba44 71 Int_t fMinTrackletsForPilup; // min. n. of tracklets for pilup definition
72 Bool_t fIsPileup; // flag for pileup
73 Int_t fNTrpuv; // tracklets in pile-up vertex
74 Float_t fZpuv; // Z of second pile-up vertex
6b4d9537 75 Int_t fNoVertices; //! number of vertices found
76 AliESDVertex* fVertArray; //! vertices (main+pileupped)
8c32ba44 77
308c2f7c 78 private:
41b19549 79 // copy constructor (NO copy allowed: the constructor is protected
80 // to avoid misuse)
81 AliITSVertexer(const AliITSVertexer& vtxr);
82 // assignment operator (NO assignment allowed)
83 AliITSVertexer& operator=(const AliITSVertexer& /* vtxr */);
84
308c2f7c 85 Int_t fFirstEvent; // First event to be processed by FindVertices
ff44c37c 86 Int_t fLastEvent; // Last event to be processed by FindVertices
27167524 87
1ff24d0a 88 ClassDef(AliITSVertexer,11);
c5f0f3c1 89};
90
91#endif