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