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 */
8 ///////////////////////////////////////////////////////////////////////////////
10 // Class for TRD reconstruction //
12 ///////////////////////////////////////////////////////////////////////////////
14 #include "AliReconstructor.h"
15 #include "AliDetectorRecoParam.h"
16 #include "AliTRDpidUtil.h"
17 #include "AliTRDrecoParam.h"
20 class TTreeSRedirector;
22 class AliTRDReconstructor: public AliReconstructor
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
32 ,kEightSlices = BIT(6)
33 ,kWriteClusters = BIT(7)
34 ,kWriteTracklets = BIT(8)
37 ,kVertexConstrained = BIT(11) // Perform vertex constrained fit
38 ,kImproveTracklet = BIT(12) // Improve tracklet in the SA TRD track finder
41 ,kProcTracklets = BIT(15) // process online tracklets
43 ,kNsteer = 16 // number of tasks
45 enum ETRDReconstructorTask {
50 ,kNtasks = 4 // number of reconsruction tasks
52 enum ETRDReconstructorGas {
57 AliTRDReconstructor();
58 AliTRDReconstructor(const AliTRDReconstructor &r);
59 virtual ~AliTRDReconstructor();
60 AliTRDReconstructor& operator = (const AliTRDReconstructor&) { return *this;}
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; }
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;}
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;
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);
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;
113 static TClonesArray *fgClusters; // list of clusters for local reconstructor
114 static TClonesArray *fgTracklets; // list of online tracklets for local reconstructor
116 ClassDef(AliTRDReconstructor, 2) // Class for the TRD reconstruction
120 //___________________________________________________
121 inline void AliTRDReconstructor::GetTCParams(Double_t *par) const
124 if(GetDriftGas()==kAr) memcpy(par, &fTCParams[4], 4*sizeof(Double_t));
125 else memcpy(par, &fTCParams[0], 4*sizeof(Double_t));
128 //___________________________________________________
129 inline void AliTRDReconstructor::SetTCParams(Double_t *par)
132 memcpy(fTCParams, par, 8*sizeof(Double_t));