Making online tracklets usable in offline reconstruction
[u/mrichter/AliRoot.git] / TRD / AliTRDReconstructor.h
1 #ifndef ALITRDRECONSTRUCTOR_H
2 #define ALITRDRECONSTRUCTOR_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 /* $Id$ */
7
8 ///////////////////////////////////////////////////////////////////////////////
9 //                                                                           //
10 // Class for TRD reconstruction                                              //
11 //                                                                           //
12 ///////////////////////////////////////////////////////////////////////////////
13
14 #include "AliReconstructor.h"
15 #include "AliDetectorRecoParam.h"
16 #include "AliTRDrecoParam.h"
17 #include "AliTRDpidUtil.h"
18
19 class TClonesArray;
20 class TTreeSRedirector;
21 class AliRawReader;
22 class AliTRDReconstructor: public AliReconstructor 
23 {
24 public:
25   enum ETRDReconstructorSteer {
26     kDigitsConversion= BIT(0)
27     ,kTC             = BIT(1) // tail cancelation
28     ,kLUT            = BIT(2) // look up table for cluster position determination 
29     ,kGAUS           = BIT(3) // look up table for cluster position determination 
30     ,kClusterSharing = BIT(4) // Toggle cluster sharing
31     ,kSteerPID       = BIT(5)
32     ,kEightSlices    = BIT(6)
33     ,kWriteClusters  = BIT(7)
34     ,kWriteTracklets = BIT(8)
35     ,kDriftGas       = BIT(9)
36     ,kSeeding        = BIT(10)
37     ,kVertexConstrained = BIT(11) // Perform vertex constrained fit
38     ,kImproveTracklet   = BIT(12) // Improve tracklet in the SA TRD track finder 
39     ,kHLT            = BIT(13)
40     ,kCosmic         = BIT(14)
41     ,kProcTracklets  = BIT(15) // process online tracklets
42     ,kOwner          = BIT(16)
43     ,kNsteer         = 16       // number of tasks
44   };
45   enum ETRDReconstructorTask {
46     kRawReader    = 0
47     ,kClusterizer = 1
48     ,kTracker     = 2
49     ,kPID         = 3
50     ,kNtasks      = 4  // number of reconsruction tasks
51   };
52   enum ETRDReconstructorGas {
53     kXe = 0
54     ,kAr = 1
55   };
56
57   AliTRDReconstructor();
58   AliTRDReconstructor(const AliTRDReconstructor &r);
59   virtual ~AliTRDReconstructor();
60   AliTRDReconstructor& operator = (const AliTRDReconstructor&)          { return *this;}
61         
62         virtual void        Init();
63
64   virtual void        ConvertDigits(AliRawReader *rawReader, TTree *digitsTree) const;
65   virtual AliTracker* CreateTracker() const;
66   TTreeSRedirector*   GetDebugStream(ETRDReconstructorTask task) const { return task < kNtasks ? fDebugStream[task] : 0x0; }
67
68   virtual void        FillESD(AliRawReader *, TTree *clusterTree, AliESDEvent *esd) const { FillESD((TTree * )NULL, clusterTree, esd);                    }
69   virtual void        FillESD(TTree *digitsTree, TTree *clusterTree, AliESDEvent *esd) const;
70   static TClonesArray* GetClusters() {return fgClusters;}
71   static TClonesArray* GetTracklets() {return fgTracklets;}
72   Int_t               GetNdEdxSlices() const     { return (Int_t)AliTRDpidUtil::GetNdEdxSlices(GetPIDMethod());}
73   ETRDReconstructorGas GetDriftGas() const        { return fSteerParam&kDriftGas ? kAr : kXe;}
74   AliTRDpidUtil::ETRDPIDMethod       GetPIDMethod() const       { return fSteerParam&kSteerPID ? AliTRDpidUtil::kNN : AliTRDpidUtil::kLQ;}
75   static const AliTRDrecoParam* GetRecoParam() { return dynamic_cast<const AliTRDrecoParam*>(AliReconstructor::GetRecoParam(2)); }
76   Int_t               GetStreamLevel(ETRDReconstructorTask task) const    { return fStreamLevel[task];} 
77   inline void         GetTCParams(Double_t *par) const;
78   virtual Bool_t      HasDigitConversion() const { return fSteerParam&kDigitsConversion;  };
79   Bool_t              HasVertexConstrained() const { return fSteerParam&kVertexConstrained; }
80   Bool_t              HasImproveTracklets() const  { return fSteerParam&kImproveTracklet; }
81   Bool_t              IsWritingClusters() const  { return fSteerParam&kWriteClusters;}
82   Bool_t              IsWritingTracklets() const { return fSteerParam&kWriteTracklets;}
83   Bool_t              IsHLT() const              { return fSteerParam&kHLT;}
84   Bool_t              IsSeeding() const          { return fSteerParam&kSeeding;}
85   Bool_t              IsCosmic() const           { return fSteerParam&kCosmic;}
86   Bool_t              IsEightSlices() const      { return fSteerParam&kEightSlices;}
87   Bool_t              UseClusterSharing() const  { return fSteerParam&kClusterSharing;}
88   Bool_t              UseLUT() const             { return fSteerParam&kLUT;}
89   Bool_t              UseGAUS() const             { return fSteerParam&kGAUS;}
90   Bool_t              UseTailCancelation() const { return fSteerParam&kTC;}
91   Bool_t              IsProcessingTracklets() const { return fSteerParam&kProcTracklets;}
92   
93   static void         Options(UInt_t steer=0, UChar_t *stream=0x0);
94   virtual void        Reconstruct(AliRawReader *rawReader, TTree *clusterTree) const;
95   virtual void        Reconstruct(TTree *digitsTree, TTree *clusterTree) const;
96
97   static void         SetClusters(TClonesArray *clusters) {fgClusters = clusters;}
98   static void         SetTracklets(TClonesArray *tracklets) {fgTracklets = tracklets;}
99   void                SetOption(Option_t *opt);
100   inline void         SetTCParams(Double_t *par);
101   void                SetStreamLevel(Int_t level, ETRDReconstructorTask task= kTracker);
102
103 private:
104   static Char_t    *fgSteerNames[kNsteer];//! steering names
105   static Char_t    *fgSteerFlags[kNsteer];//! steering flags
106   static Char_t    *fgTaskNames[kNtasks]; //! tasks names
107   static Char_t    *fgTaskFlags[kNtasks]; //! tasks flags
108   UChar_t           fStreamLevel[kNtasks];// stream level for each reconstruction task
109   UInt_t            fSteerParam;          // steering bits
110   Double_t          fTCParams[8];         // Tail Cancellation parameters for drift gases 
111   TTreeSRedirector *fDebugStream[kNtasks];// Debug Streamer container;
112  
113   static TClonesArray *fgClusters;    // list of clusters for local reconstructor
114   static TClonesArray *fgTracklets;   // list of online tracklets for local reconstructor
115
116   ClassDef(AliTRDReconstructor, 2)    //  Class for the TRD reconstruction
117
118 };
119
120 //___________________________________________________
121 inline void AliTRDReconstructor::GetTCParams(Double_t *par) const
122 {
123   if(!par) return;
124   if(GetDriftGas()==kAr) memcpy(par, &fTCParams[4], 4*sizeof(Double_t));
125   else memcpy(par, &fTCParams[0], 4*sizeof(Double_t));
126 }
127
128 //___________________________________________________
129 inline void AliTRDReconstructor::SetTCParams(Double_t *par)
130 {
131   if(!par) return;
132   memcpy(fTCParams, par, 8*sizeof(Double_t));
133 }
134
135 #endif