1 #ifndef ALIMUONESDINTERFACE_H
2 #define ALIMUONESDINTERFACE_H
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5 * See cxx source for full Copyright notice */
10 /// \class AliMUONESDInterface
11 /// \brief Converter between MUON track/cluster/digit and ESDMuon track/cluster/pad
13 // Author Philippe Pillot
15 #include <AliMpExMap.h>
21 class AliMUONVTrackStore;
22 class AliMUONVCluster;
23 class AliMUONVClusterStore;
25 class AliMUONVDigitStore;
26 class AliMUONLocalTrigger;
27 class AliMUONTriggerTrack;
28 class AliMUONVTriggerStore;
29 class AliMUONTrackParam;
30 class AliMUONVTrackReconstructor;
32 class AliESDMuonTrack;
33 class AliESDMuonCluster;
36 class AliMUONRecoParam;
38 class AliMUONESDInterface : public TObject
40 public: // methods to play with internal objects
42 AliMUONESDInterface();
43 virtual ~AliMUONESDInterface();
45 virtual void Clear(Option_t* = "");
47 void LoadEvent(AliESDEvent& esdEvent, Bool_t refit = kTRUE);
49 /// Return internal track store
50 AliMUONVTrackStore* GetTracks() const {return fTracks;}
51 /// Return internal track store
52 AliMUONVDigitStore* GetDigits() const {return fDigits;}
53 /// Return internal trigger store
54 AliMUONVTriggerStore* GetTriggers() const {return fTriggers;}
56 // Return numbers of tracks/clusters/digits
57 Int_t GetNTracks() const;
58 Int_t GetNClusters() const;
59 Int_t GetNClusters(UInt_t trackId) const;
60 Int_t GetNDigits() const;
61 Int_t GetNDigits(UInt_t trackId) const;
62 Int_t GetNDigits(UInt_t trackId, UInt_t clusterId) const;
63 Int_t GetNDigitsInCluster(UInt_t clusterId) const;
64 Int_t GetNTriggers() const;
66 // Find internal MUON objects
67 AliMUONTrack* FindTrack(UInt_t trackId) const;
68 AliMUONVCluster* FindCluster(UInt_t clusterId) const;
69 AliMUONVCluster* FindCluster(UInt_t trackId, UInt_t clusterId) const;
70 AliMUONVDigit* FindDigit(UInt_t digitId) const;
71 AliMUONLocalTrigger* FindLocalTrigger(Int_t boardNumber) const;
73 // iterate over internal MUON objects
74 TIterator* CreateTrackIterator() const;
75 TIterator* CreateClusterIterator() const;
76 TIterator* CreateClusterIterator(UInt_t trackId) const;
77 TIterator* CreateDigitIterator() const;
78 TIterator* CreateDigitIterator(UInt_t trackId) const;
79 TIterator* CreateDigitIterator(UInt_t trackId, UInt_t clusterId) const;
80 TIterator* CreateDigitIteratorInCluster(UInt_t clusterId) const;
81 TIterator* CreateLocalTriggerIterator() const;
84 public: // static methods
86 // Reset the MUON tracker (using "recoParam" if provided)
87 static void ResetTracker(const AliMUONRecoParam* recoParam = 0x0, Bool_t info = kTRUE);
88 /// Return pointer to the MUON tracker if any (use first ResetTracker(...) to create it)
89 static AliMUONVTrackReconstructor* GetTracker() {return fgTracker;}
91 /// Set the version of track store
92 static void UseTrackStore(TString name) {fgTrackStoreName = name;}
93 /// Set the version of cluster store
94 static void UseClusterStore(TString name) {fgClusterStoreName = name;}
95 /// Set the version of digit store
96 static void UseDigitStore(TString name) {fgDigitStoreName = name;}
97 /// Set the version of trigger store
98 static void UseTriggerStore(TString name) {fgTriggerStoreName = name;}
100 // Create empty stores (use the version defined in this interface)
101 static AliMUONVTrackStore* NewTrackStore();
102 static AliMUONVClusterStore* NewClusterStore();
103 static AliMUONVDigitStore* NewDigitStore();
104 static AliMUONVTriggerStore* NewTriggerStore();
106 // ESD track parameters --> MUON track parameters
107 static void GetParamAtVertex(const AliESDMuonTrack& esdTrack, AliMUONTrackParam& trackParam);
108 static void GetParamAtDCA(const AliESDMuonTrack& esdTrack, AliMUONTrackParam& trackParam);
109 static void GetParamAtFirstCluster(const AliESDMuonTrack& esdTrack, AliMUONTrackParam& trackParam);
110 static void GetParamCov(const AliESDMuonTrack& esdTrack, AliMUONTrackParam& trackParam);
112 // MUON track parameters --> ESD track parameters
113 static void SetParamAtVertex(const AliMUONTrackParam& trackParam, AliESDMuonTrack& esdTrack);
114 static void SetParamAtDCA(const AliMUONTrackParam& trackParam, AliESDMuonTrack& esdTrack);
115 static void SetParamAtFirstCluster(const AliMUONTrackParam& trackParam, AliESDMuonTrack& esdTrack);
116 static void SetParamCov(const AliMUONTrackParam& trackParam, AliESDMuonTrack& esdTrack);
118 // ESDMuon objects --> MUON objects conversion
119 static void ESDToMUON(const AliESDMuonTrack& esdTrack, AliMUONTrack& track, Bool_t refit = kTRUE);
120 static void ESDToMUON(const AliESDMuonTrack& esdTrack, AliMUONLocalTrigger& locTrg);
121 static void ESDToMUON(const AliESDMuonCluster& esdCluster, AliMUONVCluster& cluster);
122 static void ESDToMUON(const AliESDMuonPad& esdPad, AliMUONVDigit& digit);
124 // MUON objects --> ESDMuon objects conversion
125 static void MUONToESD(const AliMUONTrack& track, AliESDMuonTrack& esdTrack, const Double_t vertex[3],
126 const AliMUONVDigitStore* digits = 0x0, const AliMUONLocalTrigger* locTrg = 0x0);
127 static void MUONToESD(const AliMUONLocalTrigger& locTrg, AliESDMuonTrack& esdTrack, UInt_t trackId,
128 const AliMUONTriggerTrack* triggerTrack = 0x0);
129 static void MUONToESD(const AliMUONVCluster& cluster, AliESDMuonCluster& esdCluster, const AliMUONVDigitStore* digits = 0x0);
130 static void MUONToESD(const AliMUONVDigit& digit, AliESDMuonPad& esdPad);
132 // Add ESD object into the corresponding MUON store
133 // return a pointer to the corresponding MUON object into the store
134 static AliMUONTrack* Add(const AliESDMuonTrack& esdTrack, AliMUONVTrackStore& trackStore, Bool_t refit = kTRUE);
135 static void Add(const AliESDMuonTrack& esdTrack, AliMUONVTriggerStore& triggerStore);
136 static AliMUONVCluster* Add(const AliESDMuonCluster& esdCluster, AliMUONVClusterStore& clusterStore);
137 static AliMUONVDigit* Add(const AliESDMuonPad& esdPad, AliMUONVDigitStore& digitStore);
142 AliMUONESDInterface (const AliMUONESDInterface&); ///< copy constructor
143 AliMUONESDInterface& operator=(const AliMUONESDInterface&); ///< assignment operator
149 AliMUONVCluster* FindClusterInTrack(const AliMUONTrack& track, UInt_t clusterId) const;
154 static AliMUONRecoParam* fgRecoParam; ///< reconstruction parameters for refitting
155 static AliMUONVTrackReconstructor* fgTracker; ///< track reconstructor for refitting
157 static TString fgTrackStoreName; ///< class name of the track store to use
158 static TString fgClusterStoreName; ///< class name of the cluster store to use
159 static TString fgDigitStoreName; ///< class name of the digit store to use
160 static TString fgTriggerStoreName; ///< class name of the trigger store to use
163 AliMUONVTrackStore* fTracks; ///< track container
164 AliMUONVDigitStore* fDigits; ///< digit container
165 AliMUONVTriggerStore* fTriggers; ///< trigger container
167 // maps (to speed up data retrieval)
168 AliMpExMap* fClusterMap; ///< map of clusters
169 AliMpExMap* fDigitMap; ///< map of digits
172 ClassDef(AliMUONESDInterface,0)