]>
Commit | Line | Data |
---|---|---|
d0cf54cd | 1 | /************************************************************************** |
2 | * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
3 | * * | |
4 | * Author: The ALICE Off-line Project. * | |
5 | * Contributors are mentioned in the code where appropriate. * | |
6 | * * | |
7 | * Permission to use, copy, modify and distribute this software and its * | |
8 | * documentation strictly for non-commercial purposes is hereby granted * | |
9 | * without fee, provided that the above copyright notice appears in all * | |
10 | * copies and that both the copyright notice and this permission notice * | |
11 | * appear in the supporting documentation. The authors make no claims * | |
12 | * about the suitability of this software for any purpose. It is * | |
13 | * provided "as is" without express or implied warranty. * | |
14 | **************************************************************************/ | |
15 | ||
27de2dfb | 16 | /* $Id$ */ |
17 | ||
d0cf54cd | 18 | // ROOT includes |
19 | #include <TString.h> | |
20 | #include <TGeoManager.h> | |
21 | ||
22 | // STEER includes | |
23 | #include "AliESDEvent.h" | |
24 | #include "AliESDMuonTrack.h" | |
25 | #include "AliCDBManager.h" | |
26 | #include "AliCDBStorage.h" | |
27 | #include "AliGeomManager.h" | |
28 | ||
29 | // ANALYSIS includes | |
30 | #include "AliAnalysisManager.h" | |
31 | #include "AliInputEventHandler.h" | |
32 | #include "AliAnalysisTaskMuonRefit.h" | |
33 | ||
34 | // MUON includes | |
35 | #include "AliMUONCDB.h" | |
36 | #include "AliMUONRecoParam.h" | |
37 | #include "AliMUONESDInterface.h" | |
38 | #include "AliMUONRefitter.h" | |
39 | #include "AliMUONTrack.h" | |
40 | #include "AliMUONVCluster.h" | |
41 | #include "AliMUONVTrackStore.h" | |
42 | #include "AliMUONGeometryTransformer.h" | |
43 | ||
44 | #ifndef SafeDelete | |
45 | #define SafeDelete(x) if (x != NULL) { delete x; x = NULL; } | |
46 | #endif | |
47 | ||
48 | ClassImp(AliAnalysisTaskMuonRefit) | |
49 | ||
50 | //________________________________________________________________________ | |
51 | AliAnalysisTaskMuonRefit::AliAnalysisTaskMuonRefit() : | |
52 | AliAnalysisTaskSE(), | |
53 | fDefaultStorage(""), | |
54 | fImproveTracks(kFALSE), | |
55 | fSigmaCut(-1.), | |
56 | fSigmaCutForTrigger(-1.), | |
57 | fReAlign(kFALSE), | |
58 | fOldAlignStorage(""), | |
59 | fNewAlignStorage(""), | |
60 | fOldGeoTransformer(NULL), | |
61 | fNewGeoTransformer(NULL), | |
62 | fESDInterface(NULL), | |
63 | fRefitter(NULL) | |
64 | { | |
65 | /// Default constructor | |
66 | } | |
67 | ||
68 | //________________________________________________________________________ | |
69 | AliAnalysisTaskMuonRefit::AliAnalysisTaskMuonRefit(const char *name) : | |
70 | AliAnalysisTaskSE(name), | |
71 | fDefaultStorage("raw://"), | |
72 | fImproveTracks(kFALSE), | |
73 | fSigmaCut(-1.), | |
74 | fSigmaCutForTrigger(-1.), | |
75 | fReAlign(kFALSE), | |
76 | fOldAlignStorage(""), | |
77 | fNewAlignStorage(""), | |
78 | fOldGeoTransformer(NULL), | |
79 | fNewGeoTransformer(NULL), | |
80 | fESDInterface(NULL), | |
81 | fRefitter(NULL) | |
82 | { | |
83 | /// Constructor | |
84 | for (Int_t i = 0; i < AliMUONConstants::NTrackingCh(); i++) ResetClusterResolution(i, -1., -1.); | |
85 | } | |
86 | ||
87 | //________________________________________________________________________ | |
88 | AliAnalysisTaskMuonRefit::~AliAnalysisTaskMuonRefit() | |
89 | { | |
90 | /// Destructor | |
91 | SafeDelete(fOldGeoTransformer); | |
92 | SafeDelete(fNewGeoTransformer); | |
93 | SafeDelete(fESDInterface); | |
94 | SafeDelete(fRefitter); | |
95 | } | |
96 | ||
97 | //___________________________________________________________________________ | |
98 | void AliAnalysisTaskMuonRefit::UserCreateOutputObjects() | |
99 | { | |
100 | } | |
101 | ||
102 | //________________________________________________________________________ | |
103 | void AliAnalysisTaskMuonRefit::UserExec(Option_t *) | |
104 | { | |
105 | /// Main event loop | |
106 | ||
107 | // check if refitter properly created | |
108 | if (!fRefitter) return; | |
109 | ||
110 | AliESDEvent* esd = dynamic_cast<AliESDEvent*>(InputEvent()); | |
111 | if (!esd) return; | |
112 | ||
113 | Int_t nTracks = (Int_t)esd->GetNumberOfMuonTracks(); | |
114 | if (nTracks < 1) return; | |
115 | ||
116 | // load the current event | |
117 | fESDInterface->LoadEvent(*esd, kFALSE); | |
118 | ||
119 | // modify clusters | |
120 | AliMUONVCluster* cluster = 0x0; | |
121 | TIter nextCluster(fESDInterface->CreateClusterIterator()); | |
122 | while ((cluster = static_cast<AliMUONVCluster*>(nextCluster()))) ModifyCluster(*cluster); | |
123 | ||
124 | // refit the tracks from clusters | |
125 | AliMUONVTrackStore* newTrackStore = fRefitter->ReconstructFromClusters(); | |
126 | ||
127 | // loop over the list of ESD tracks | |
128 | TClonesArray *esdTracks = (TClonesArray*) esd->FindListObject("MuonTracks"); | |
129 | for (Int_t iTrack = 0; iTrack < nTracks; iTrack++) { | |
130 | ||
131 | // get the ESD track | |
132 | AliESDMuonTrack* esdTrack = (AliESDMuonTrack*) esdTracks->UncheckedAt(iTrack); | |
133 | ||
134 | // skip ghost tracks (leave them unchanged) | |
135 | if (!esdTrack->ContainTrackerData()) continue; | |
136 | ||
137 | // Find the corresponding re-fitted MUON track | |
138 | AliMUONTrack* newTrack = (AliMUONTrack*) newTrackStore->FindObject(esdTrack->GetUniqueID()); | |
139 | ||
140 | // replace the content of the current ESD track or remove it | |
141 | if (newTrack && (!fImproveTracks || newTrack->IsImproved())) { | |
142 | Double_t vertex[3] = {esdTrack->GetNonBendingCoor(), esdTrack->GetBendingCoor(), esdTrack->GetZ()}; | |
143 | AliMUONESDInterface::MUONToESD(*newTrack, *esdTrack, vertex, fESDInterface->GetDigits()); | |
144 | ||
145 | // eventually remove the trigger part if matching chi2 do not pass the new cut | |
146 | if (newTrack->GetChi2MatchTrigger() > fSigmaCutForTrigger*fSigmaCutForTrigger) { | |
147 | esdTrack->SetLocalTrigger(0); | |
148 | esdTrack->SetChi2MatchTrigger(0.); | |
149 | esdTrack->SetHitsPatternInTrigCh(0); | |
150 | esdTrack->SetTriggerX1Pattern(0); | |
151 | esdTrack->SetTriggerY1Pattern(0); | |
152 | esdTrack->SetTriggerX2Pattern(0); | |
153 | esdTrack->SetTriggerY2Pattern(0); | |
154 | esdTrack->SetTriggerX3Pattern(0); | |
155 | esdTrack->SetTriggerY3Pattern(0); | |
156 | esdTrack->SetTriggerX4Pattern(0); | |
157 | esdTrack->SetTriggerY4Pattern(0); | |
158 | } | |
159 | ||
160 | } else { | |
161 | esdTracks->Remove(esdTrack); | |
162 | } | |
163 | ||
164 | } | |
165 | ||
166 | // free memory | |
167 | delete newTrackStore; | |
168 | ||
169 | // compress the array of ESD tracks | |
170 | esdTracks->Compress(); | |
171 | ||
172 | } | |
173 | ||
174 | //________________________________________________________________________ | |
175 | void AliAnalysisTaskMuonRefit::NotifyRun() | |
176 | { | |
177 | /// load necessary data from OCDB and create the refitter | |
178 | ||
179 | AliCDBManager* cdbm = AliCDBManager::Instance(); | |
180 | cdbm->SetDefaultStorage(fDefaultStorage.Data()); | |
181 | cdbm->SetRun(fCurrentRunNumber); | |
182 | ||
183 | if (!AliMUONCDB::LoadField()) return; | |
184 | ||
185 | if (!AliMUONCDB::LoadMapping(kTRUE)) return; | |
186 | ||
187 | AliMUONRecoParam* recoParam = AliMUONCDB::LoadRecoParam(); | |
188 | if (!recoParam) return; | |
189 | ||
190 | if (fImproveTracks) { | |
191 | if (fSigmaCut > 0.) recoParam->ImproveTracks(kTRUE, fSigmaCut); | |
192 | else recoParam->ImproveTracks(kTRUE); | |
193 | } else recoParam->ImproveTracks(kFALSE); | |
194 | ||
195 | if (fSigmaCutForTrigger > 0.) recoParam->SetSigmaCutForTrigger(fSigmaCutForTrigger); | |
196 | else fSigmaCutForTrigger = recoParam->GetSigmaCutForTrigger(); | |
197 | ||
198 | for (Int_t i = 0; i < AliMUONConstants::NTrackingCh(); i++) { | |
199 | if (fClusterResNB[i] < 0.) fClusterResNB[i] = recoParam->GetDefaultNonBendingReso(i); | |
200 | if (fClusterResB[i] < 0.) fClusterResB[i] = recoParam->GetDefaultBendingReso(i); | |
201 | } | |
202 | ||
203 | if (fReAlign) { | |
204 | ||
205 | // recover default storage full name (raw:// cannot be used to set specific storage) | |
206 | TString defaultStorage(cdbm->GetDefaultStorage()->GetType()); | |
207 | if (defaultStorage == "alien") defaultStorage += Form("://folder=%s", cdbm->GetDefaultStorage()->GetBaseFolder().Data()); | |
208 | else defaultStorage += Form("://%s", cdbm->GetDefaultStorage()->GetBaseFolder().Data()); | |
209 | ||
210 | // reset existing geometry/alignment if any | |
211 | if (cdbm->GetEntryCache()->Contains("GRP/Geometry/Data")) cdbm->UnloadFromCache("GRP/Geometry/Data"); | |
212 | if (cdbm->GetEntryCache()->Contains("MUON/Align/Data")) cdbm->UnloadFromCache("MUON/Align/Data"); | |
213 | if (AliGeomManager::GetGeometry()) AliGeomManager::GetGeometry()->UnlockGeometry(); | |
214 | ||
215 | // get original geometry transformer | |
216 | AliGeomManager::LoadGeometry(); | |
217 | if (!AliGeomManager::GetGeometry()) return; | |
218 | if (fOldAlignStorage != "none") { | |
219 | if (!fOldAlignStorage.IsNull()) cdbm->SetSpecificStorage("MUON/Align/Data",fOldAlignStorage.Data()); | |
220 | else cdbm->SetSpecificStorage("MUON/Align/Data",defaultStorage.Data()); | |
221 | AliGeomManager::ApplyAlignObjsFromCDB("MUON"); | |
222 | } | |
223 | fOldGeoTransformer = new AliMUONGeometryTransformer(); | |
224 | fOldGeoTransformer->LoadGeometryData(); | |
225 | ||
226 | // get new geometry transformer | |
227 | cdbm->UnloadFromCache("GRP/Geometry/Data"); | |
228 | if (fOldAlignStorage != "none") cdbm->UnloadFromCache("MUON/Align/Data"); | |
229 | AliGeomManager::GetGeometry()->UnlockGeometry(); | |
230 | AliGeomManager::LoadGeometry(); | |
231 | if (!AliGeomManager::GetGeometry()) return; | |
232 | if (!fNewAlignStorage.IsNull()) cdbm->SetSpecificStorage("MUON/Align/Data",fNewAlignStorage.Data()); | |
233 | else cdbm->SetSpecificStorage("MUON/Align/Data",defaultStorage.Data()); | |
234 | AliGeomManager::ApplyAlignObjsFromCDB("MUON"); | |
235 | fNewGeoTransformer = new AliMUONGeometryTransformer(); | |
236 | fNewGeoTransformer->LoadGeometryData(); | |
237 | ||
238 | } else { | |
239 | ||
240 | // load geometry for track extrapolation to vertex | |
241 | if (cdbm->GetEntryCache()->Contains("GRP/Geometry/Data")) cdbm->UnloadFromCache("GRP/Geometry/Data"); | |
242 | if (AliGeomManager::GetGeometry()) AliGeomManager::GetGeometry()->UnlockGeometry(); | |
243 | AliGeomManager::LoadGeometry(); | |
244 | if (!AliGeomManager::GetGeometry()) return; | |
245 | ||
246 | } | |
247 | ||
248 | fESDInterface = new AliMUONESDInterface(); | |
249 | fRefitter = new AliMUONRefitter(recoParam); | |
250 | fRefitter->Connect(fESDInterface); | |
251 | } | |
252 | ||
253 | //________________________________________________________________________ | |
254 | void AliAnalysisTaskMuonRefit::Terminate(Option_t *) | |
255 | { | |
256 | } | |
257 | ||
258 | //________________________________________________________________________ | |
259 | void AliAnalysisTaskMuonRefit::ModifyCluster(AliMUONVCluster& cl) | |
260 | { | |
261 | /// Reset the cluster resolution to the one given to the task and change | |
262 | /// the cluster position according to the new alignment parameters if required | |
263 | ||
264 | Double_t gX,gY,gZ,lX,lY,lZ; | |
265 | ||
266 | // change their resolution | |
267 | cl.SetErrXY(fClusterResNB[cl.GetChamberId()], fClusterResB[cl.GetChamberId()]); | |
268 | ||
269 | // change their position | |
270 | if (fReAlign) { | |
271 | gX = cl.GetX(); | |
272 | gY = cl.GetY(); | |
273 | gZ = cl.GetZ(); | |
274 | fOldGeoTransformer->Global2Local(cl.GetDetElemId(),gX,gY,gZ,lX,lY,lZ); | |
275 | fNewGeoTransformer->Local2Global(cl.GetDetElemId(),lX,lY,lZ,gX,gY,gZ); | |
276 | cl.SetXYZ(gX,gY,gZ); | |
277 | } | |
278 | ||
279 | } | |
280 |