]>
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; | |
66f6bfd9 | 51 | static TClonesArray* GetClusters() {return fgClusters;} |
a5b99acd | 52 | static TClonesArray* GetTracklets() {return fgTracklets;} |
c6f7c6cb | 53 | static Int_t GetNTimeBins() {return fgNTimeBins;} |
0d83b3a5 | 54 | Int_t GetNdEdxSlices() const { return (Int_t)AliTRDpidUtil::GetNdEdxSlices(GetPIDMethod());} |
a2fbb6ec | 55 | AliTRDpidUtil::ETRDPIDMethod GetPIDMethod() const { return GetRecoParam()->IsPIDNeuralNetwork() ? AliTRDpidUtil::kNN : AliTRDpidUtil::kLQ;} |
3a039a31 | 56 | static const AliTRDrecoParam* GetRecoParam() { return dynamic_cast<const AliTRDrecoParam*>(AliReconstructor::GetRecoParam(2)); } |
a0446ff1 | 57 | |
b72f4eaf | 58 | virtual Bool_t HasDigitConversion() const { return fSteerParam&kDigitsConversion; }; |
a2fbb6ec | 59 | Bool_t IsCosmic() const { return GetRecoParam()->GetEventSpecie() & AliRecoParam::kCosmic;} |
3a039a31 | 60 | Bool_t IsWritingClusters() const { return fSteerParam&kWriteClusters;} |
61 | Bool_t IsWritingTracklets() const { return fSteerParam&kWriteTracklets;} | |
4302c900 | 62 | Bool_t IsHLT() const { return fSteerParam&kHLT;} |
3a039a31 | 63 | Bool_t IsSeeding() const { return fSteerParam&kSeeding;} |
a5b99acd | 64 | Bool_t IsProcessingTracklets() const { return fSteerParam&kProcTracklets;} |
a2fbb6ec | 65 | Bool_t IsDebugStreaming() const { return fSteerParam&kDebug;} |
8fc736d7 | 66 | Bool_t UseClusterRadialCorrection() const { return fSteerParam&kClRadialCorr;} |
a2fbb6ec | 67 | |
68 | static void Options(UInt_t steer=0); | |
3a039a31 | 69 | virtual void Reconstruct(AliRawReader *rawReader, TTree *clusterTree) const; |
70 | virtual void Reconstruct(TTree *digitsTree, TTree *clusterTree) const; | |
3bc9d03e | 71 | |
828c6f80 | 72 | static void SetClusters(TClonesArray *clusters) { fgClusters = clusters;} |
73 | static void SetTracklets(TClonesArray *tracklets) { fgTracklets = tracklets;} | |
3a039a31 | 74 | void SetOption(Option_t *opt); |
3bc9d03e | 75 | |
3a039a31 | 76 | private: |
3abc001d | 77 | AliTRDReconstructor(const AliTRDReconstructor &r); //Not implemented |
78 | AliTRDReconstructor& operator = (const AliTRDReconstructor&); //Not implemented | |
79 | ||
828c6f80 | 80 | static Char_t const *fgSteerNames[kNsteer];//! steering names |
81 | static Char_t const *fgSteerFlags[kNsteer];//! steering flags | |
82 | static Char_t const *fgTaskNames[AliTRDrecoParam::kTRDreconstructionTasks]; //! tasks names | |
83 | static Char_t const *fgTaskFlags[AliTRDrecoParam::kTRDreconstructionTasks]; //! tasks flags | |
b72f4eaf | 84 | UInt_t fSteerParam; // steering bits |
a2fbb6ec | 85 | TTreeSRedirector *fDebugStream[AliTRDrecoParam::kTRDreconstructionTasks];// Debug Streamer container; |
3a039a31 | 86 | |
5127281e | 87 | static TClonesArray *fgClusters; // list of clusters for local reconstructor |
88 | static TClonesArray *fgTracklets; // list of online tracklets for local reconstructor | |
89 | static Int_t fgNTimeBins; // number of time bins as given by the clusterizer | |
90 | AliTRDclusterizer *fClusterizer; //! instance of TRD clusterizer | |
3bc9d03e | 91 | |
5127281e | 92 | ClassDef(AliTRDReconstructor, 4) // Class for the TRD reconstruction |
3bc9d03e | 93 | |
121a60bd | 94 | }; |
95 | ||
3a039a31 | 96 | |
3a039a31 | 97 | |
121a60bd | 98 | #endif |
5127281e | 99 |