]>
Commit | Line | Data |
---|---|---|
121a60bd | 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 | ||
3bc9d03e | 6 | /* $Id$ */ |
7 | ||
121a60bd | 8 | /////////////////////////////////////////////////////////////////////////////// |
9 | // // | |
3bc9d03e | 10 | // Class for TRD reconstruction // |
121a60bd | 11 | // // |
12 | /////////////////////////////////////////////////////////////////////////////// | |
13 | ||
121a60bd | 14 | #include "AliReconstructor.h" |
a2fbb6ec | 15 | #include "AliRecoParam.h" |
7e88424f | 16 | #include "AliDetectorRecoParam.h" |
0d83b3a5 | 17 | #include "AliTRDpidUtil.h" |
d39401ad | 18 | #include "AliTRDrecoParam.h" |
828c6f80 | 19 | #include "AliTRDdigitsParam.h" |
7e88424f | 20 | |
66f6bfd9 | 21 | class TClonesArray; |
29f95561 | 22 | class TTreeSRedirector; |
3a039a31 | 23 | class AliRawReader; |
5127281e | 24 | class AliTRDclusterizer; |
eb38ed55 | 25 | class AliTRDReconstructor: public AliReconstructor |
26 | { | |
3a039a31 | 27 | public: |
89971837 | 28 | enum ETRDReconstructorSteer { |
00872461 | 29 | kDigitsConversion= BIT(0) |
a2fbb6ec | 30 | ,kWriteClusters = BIT(1) |
31 | ,kWriteTracklets = BIT(2) | |
32 | ,kSeeding = BIT(3) | |
33 | ,kHLT = BIT(4) | |
34 | ,kProcTracklets = BIT(5) // process online tracklets | |
35 | ,kDebug = BIT(6) | |
8fc736d7 | 36 | ,kClRadialCorr = BIT(7) // toggle radial correction in clusters |
37 | ,kOwner = BIT(8) | |
38 | ,kNsteer = 8 // number of tasks | |
3a039a31 | 39 | }; |
3a039a31 | 40 | AliTRDReconstructor(); |
48f8adf3 | 41 | virtual ~AliTRDReconstructor(); |
acd241e9 | 42 | |
43 | virtual void Init(); | |
3bc9d03e | 44 | |
3a039a31 | 45 | virtual void ConvertDigits(AliRawReader *rawReader, TTree *digitsTree) const; |
46 | virtual AliTracker* CreateTracker() const; | |
8fc736d7 | 47 | TTreeSRedirector* GetDebugStream(AliTRDrecoParam::ETRDReconstructionTask task) const { return task < AliTRDrecoParam::kTRDreconstructionTasks ? fDebugStream[task] : NULL; } |
25ca55ce | 48 | |
3a039a31 | 49 | virtual void FillESD(AliRawReader *, TTree *clusterTree, AliESDEvent *esd) const { FillESD((TTree * )NULL, clusterTree, esd); } |
50 | virtual void FillESD(TTree *digitsTree, TTree *clusterTree, AliESDEvent *esd) const; | |
952051c5 | 51 | static TClonesArray* GetClusters() {return fgClusters;} |
52 | static TClonesArray* GetTracklets() { return fgTracklets;} | |
cb8b99ee | 53 | static TClonesArray* GetTracks() { return fgTracks;} |
952051c5 | 54 | static Int_t GetNTimeBins() { return fgNTimeBins;} |
0d83b3a5 | 55 | Int_t GetNdEdxSlices() const { return (Int_t)AliTRDpidUtil::GetNdEdxSlices(GetPIDMethod());} |
a2fbb6ec | 56 | AliTRDpidUtil::ETRDPIDMethod GetPIDMethod() const { return GetRecoParam()->IsPIDNeuralNetwork() ? AliTRDpidUtil::kNN : AliTRDpidUtil::kLQ;} |
952051c5 | 57 | static const AliTRDrecoParam* GetRecoParam() { return dynamic_cast<const AliTRDrecoParam*>(AliReconstructor::GetRecoParam(2)); } |
58 | static Float_t GetMinClustersInTrack() { return fgkMinClustersInTrack;} | |
59 | static Float_t GetLabelFraction() { return fgkLabelFraction;} | |
60 | static Double_t GetMaxChi2() { return fgkMaxChi2;} | |
61 | static Double_t GetMaxSnp() { return fgkMaxSnp;} | |
62 | static Double_t GetMaxStep() { return fgkMaxStep;} | |
63 | static Double_t GetEpsilon() { return fgkEpsilon;} | |
a0446ff1 | 64 | |
b72f4eaf | 65 | virtual Bool_t HasDigitConversion() const { return fSteerParam&kDigitsConversion; }; |
952051c5 | 66 | Bool_t IsCosmic() const { return GetRecoParam()->GetEventSpecie() & AliRecoParam::kCosmic;} |
3a039a31 | 67 | Bool_t IsWritingClusters() const { return fSteerParam&kWriteClusters;} |
68 | Bool_t IsWritingTracklets() const { return fSteerParam&kWriteTracklets;} | |
4302c900 | 69 | Bool_t IsHLT() const { return fSteerParam&kHLT;} |
3a039a31 | 70 | Bool_t IsSeeding() const { return fSteerParam&kSeeding;} |
a5b99acd | 71 | Bool_t IsProcessingTracklets() const { return fSteerParam&kProcTracklets;} |
952051c5 | 72 | Bool_t IsDebugStreaming() const { return fSteerParam&kDebug;} |
8fc736d7 | 73 | Bool_t UseClusterRadialCorrection() const { return fSteerParam&kClRadialCorr;} |
a2fbb6ec | 74 | |
75 | static void Options(UInt_t steer=0); | |
3a039a31 | 76 | virtual void Reconstruct(AliRawReader *rawReader, TTree *clusterTree) const; |
77 | virtual void Reconstruct(TTree *digitsTree, TTree *clusterTree) const; | |
3bc9d03e | 78 | |
828c6f80 | 79 | static void SetClusters(TClonesArray *clusters) { fgClusters = clusters;} |
80 | static void SetTracklets(TClonesArray *tracklets) { fgTracklets = tracklets;} | |
cb8b99ee | 81 | static void SetTracks(TClonesArray *tracks) { fgTracks = tracks;} |
3a039a31 | 82 | void SetOption(Option_t *opt); |
3bc9d03e | 83 | |
3a039a31 | 84 | private: |
3abc001d | 85 | AliTRDReconstructor(const AliTRDReconstructor &r); //Not implemented |
86 | AliTRDReconstructor& operator = (const AliTRDReconstructor&); //Not implemented | |
87 | ||
828c6f80 | 88 | static Char_t const *fgSteerNames[kNsteer];//! steering names |
89 | static Char_t const *fgSteerFlags[kNsteer];//! steering flags | |
90 | static Char_t const *fgTaskNames[AliTRDrecoParam::kTRDreconstructionTasks]; //! tasks names | |
91 | static Char_t const *fgTaskFlags[AliTRDrecoParam::kTRDreconstructionTasks]; //! tasks flags | |
b72f4eaf | 92 | UInt_t fSteerParam; // steering bits |
952051c5 | 93 | // configuration vars for tracking |
94 | static const Double_t fgkMaxChi2; // Max increment in track chi2 | |
95 | static const Float_t fgkMinClustersInTrack; // Min number of clusters in track | |
96 | static const Float_t fgkLabelFraction; // Min fraction of same label | |
97 | static const Double_t fgkMaxSnp; // Maximal snp for tracking | |
98 | static const Double_t fgkMaxStep; // Maximal step for tracking | |
99 | static const Double_t fgkEpsilon; // Precision of radial coordinate | |
100 | ||
a2fbb6ec | 101 | TTreeSRedirector *fDebugStream[AliTRDrecoParam::kTRDreconstructionTasks];// Debug Streamer container; |
3a039a31 | 102 | |
5127281e | 103 | static TClonesArray *fgClusters; // list of clusters for local reconstructor |
104 | static TClonesArray *fgTracklets; // list of online tracklets for local reconstructor | |
cb8b99ee | 105 | static TClonesArray *fgTracks; // list of GTU tracks for local reconstructor |
5127281e | 106 | static Int_t fgNTimeBins; // number of time bins as given by the clusterizer |
107 | AliTRDclusterizer *fClusterizer; //! instance of TRD clusterizer | |
3bc9d03e | 108 | |
cb8b99ee | 109 | ClassDef(AliTRDReconstructor, 5) // Class for the TRD reconstruction |
3bc9d03e | 110 | |
121a60bd | 111 | }; |
112 | ||
3a039a31 | 113 | |
3a039a31 | 114 | |
121a60bd | 115 | #endif |
5127281e | 116 |