add initial version of the friend interface classes
[u/mrichter/AliRoot.git] / STEER / ESD / AliESDfriendTrack.h
1 #ifndef ALIESDFRIENDTRACK_H
2 #define ALIESDFRIENDTRACK_H
3
4 //-------------------------------------------------------------------------
5 //                     Class AliESDfriendTrack
6 //               This class contains ESD track additions
7 //       Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch 
8 //-------------------------------------------------------------------------
9
10 #include <TObject.h>
11 #include <TClonesArray.h>
12 #include <AliExternalTrackParam.h>
13
14 class AliTrackPointArray;
15 class AliKalmanTrack;
16 class TObjArrray;
17 class AliTPCseed;
18 class AliTRDseedV1;
19
20 //_____________________________________________________________________________
21 class AliESDfriendTrack : public TObject {
22 public:
23   enum {
24     kMaxITScluster=12,
25     kMaxTPCcluster=160,
26     kMaxTRDcluster=180
27   };
28   AliESDfriendTrack();
29   AliESDfriendTrack(const AliESDfriendTrack &t);
30   virtual ~AliESDfriendTrack();
31
32   void Set1P(Float_t p) {f1P=p;}
33   void SetTrackPointArray(AliTrackPointArray *points) {
34     fPoints=points;
35   }
36   Float_t Get1P() const  {return f1P;}
37   Int_t *GetITSindices() {return fITSindex;}
38   Int_t *GetTPCindices() {return fTPCindex;}
39   Int_t *GetTRDindices() {return fTRDindex;}
40   const AliTrackPointArray *GetTrackPointArray() const {return fPoints;}
41
42   void SetITStrack(AliKalmanTrack *t) {fITStrack=t;}
43   void SetTRDtrack(AliKalmanTrack *t) {fTRDtrack=t;}
44   AliKalmanTrack *GetTRDtrack() {return fTRDtrack;}
45   AliKalmanTrack *GetITStrack() {return fITStrack;}
46   void AddCalibObject(TObject * calibObject); 
47   TObject * GetCalibObject(Int_t index);
48   void AddTPCseed(AliTPCseed* /*seed*/) {}
49   void AddTRDseed(AliTRDseedV1* /*seed*/) {}
50   AliTPCseed* GetTPCseed() const {return NULL;}
51   AliTRDseedV1* GetTRDseed() const {return NULL;}
52   //
53   // parameters backup
54   void SetTPCOut(const AliExternalTrackParam &param);
55   void SetITSOut(const AliExternalTrackParam &param);
56   void SetTRDIn(const AliExternalTrackParam  &param);
57   //
58   
59   const AliExternalTrackParam * GetTPCOut() const {return  fTPCOut;} 
60   const AliExternalTrackParam * GetITSOut() const {return fITSOut;} 
61   const AliExternalTrackParam * GetTRDIn()  const {return fTRDIn;} 
62
63   void SetITSIndices(Int_t* indices, Int_t n);
64   void SetTPCIndices(Int_t* indices, Int_t n);
65   void SetTRDIndices(Int_t* indices, Int_t n);
66
67   Int_t GetMaxITScluster() {return fnMaxITScluster;}
68   Int_t GetMaxTPCcluster() {return fnMaxTPCcluster;}
69   Int_t GetMaxTRDcluster() {return fnMaxTRDcluster;}
70   
71   // bit manipulation for filtering
72   void SetSkipBit(Bool_t skip){SetBit(23,skip);}
73   Bool_t TestSkipBit() {return TestBit(23);}
74
75 protected:
76   Float_t f1P;                     // 1/P (1/(GeV/c))
77   Int_t fnMaxITScluster; // Max number of ITS clusters
78   Int_t fnMaxTPCcluster; // Max number of TPC clusters
79   Int_t fnMaxTRDcluster; // Max number of TRD clusters
80   Int_t* fITSindex; //[fnMaxITScluster] indices of the ITS clusters 
81   Int_t* fTPCindex; //[fnMaxTPCcluster] indices of the TPC clusters
82   Int_t* fTRDindex; //[fnMaxTRDcluster] indices of the TRD clusters
83
84   AliTrackPointArray *fPoints;//Array of track space points in the global frame
85   TObjArray      *fCalibContainer; //Array of objects for calibration    
86   AliKalmanTrack *fITStrack; //! pointer to the ITS track (debug purposes) 
87   AliKalmanTrack *fTRDtrack; //! pointer to the TRD track (debug purposes) 
88   //
89   //
90   AliExternalTrackParam * fTPCOut; // tpc outer parameters
91   AliExternalTrackParam * fITSOut; // its outer parameters
92   AliExternalTrackParam * fTRDIn;  // trd inner parameters
93
94 private:
95   AliESDfriendTrack &operator=(const AliESDfriendTrack & /* t */) {return *this;}
96
97   ClassDef(AliESDfriendTrack,6) //ESD friend track
98 };
99
100 #endif
101
102