Line data Source code
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 :
16 : // $Id$
17 :
18 : #include "AliMUONESDInterface.h"
19 : #include "AliMUONTrack.h"
20 : #include "AliMUONVTrackStore.h"
21 : #include "AliMUONVCluster.h"
22 : #include "AliMUONVClusterStore.h"
23 : #include "AliMUONVDigit.h"
24 : #include "AliMUONVDigitStore.h"
25 : #include "AliMUONLocalTrigger.h"
26 : #include "AliMUONTriggerTrack.h"
27 : #include "AliMUONVTriggerStore.h"
28 : #include "AliMUONVTriggerTrackStore.h"
29 : #include "AliMUON2DMapIterator.h"
30 : #include "AliMUONTrackParam.h"
31 : #include "AliMUONTrackExtrap.h"
32 : #include "AliMUONConstants.h"
33 : #include "AliMUONTracker.h"
34 : #include "AliMUONRecoParam.h"
35 : #include "AliMUONVTrackReconstructor.h"
36 : #include "AliMUONCDB.h"
37 :
38 : #include "AliMpExMapIterator.h"
39 : #include "AliMpVSegmentation.h"
40 : #include "AliMpSegmentation.h"
41 : #include "AliMpPad.h"
42 :
43 : #include "AliESDEvent.h"
44 : #include "AliESDMuonTrack.h"
45 : #include "AliESDMuonCluster.h"
46 : #include "AliESDMuonPad.h"
47 : #include "AliLog.h"
48 :
49 : #include <TROOT.h>
50 : #include <TClass.h>
51 : #include <TIterator.h>
52 : #include <TMath.h>
53 : #include <TMatrixD.h>
54 : #include <Riostream.h>
55 : #include <TGeoGlobalMagField.h>
56 : #include <TVirtualMagField.h>
57 :
58 : //-----------------------------------------------------------------------------
59 : /// \class AliMUONESDInterface
60 : ///
61 : /// There are 2 way of using thid converter between MUON track/cluster/digit
62 : /// and ESDMuon track/cluster/pad:
63 : ///
64 : /// 1) using the static methods converting the objects one by one
65 : ///
66 : /// 2) loading a whole ESDEvent and using the finders and/or the iterators
67 : /// to access the corresponding MUON objects
68 : ///
69 : /// note: You can set the recoParam used to refit the MUON track with ResetTracker(...);
70 : /// By default we use Kalman filter + Smoother
71 : ///
72 : /// \author Philippe Pillot
73 : //-----------------------------------------------------------------------------
74 :
75 : using std::endl;
76 : using std::cout;
77 : /// \cond CLASSIMP
78 18 : ClassImp(AliMUONESDInterface)
79 : /// \endcond
80 :
81 : AliMUONRecoParam* AliMUONESDInterface::fgRecoParam = 0x0;
82 : AliMUONVTrackReconstructor* AliMUONESDInterface::fgTracker = 0x0;
83 :
84 18 : TString AliMUONESDInterface::fgTrackStoreName = "AliMUONTrackStoreV1";
85 18 : TString AliMUONESDInterface::fgClusterStoreName = "AliMUONClusterStoreV2";
86 18 : TString AliMUONESDInterface::fgDigitStoreName = "AliMUONDigitStoreV2R";
87 18 : TString AliMUONESDInterface::fgTriggerStoreName = "AliMUONTriggerStoreV1";
88 18 : TString AliMUONESDInterface::fgTriggerTrackStoreName = "AliMUONTriggerTrackStoreV1";
89 :
90 : //_____________________________________________________________________________
91 : AliMUONESDInterface::AliMUONESDInterface()
92 0 : : TObject(),
93 0 : fTracks(0x0),
94 0 : fDigits(0x0),
95 0 : fTriggers(0x0),
96 0 : fClusterMap(0x0),
97 0 : fDigitMap(0x0)
98 0 : {
99 : /// Default constructor
100 0 : }
101 :
102 : //_____________________________________________________________________________
103 : AliMUONESDInterface::~AliMUONESDInterface()
104 0 : {
105 : /// Destructor
106 0 : delete fTracks;
107 0 : delete fDigits;
108 0 : delete fTriggers;
109 0 : delete fClusterMap;
110 0 : delete fDigitMap;
111 0 : }
112 :
113 : //+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++//
114 : // methods to play with internal objects //
115 : //+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++//
116 :
117 : //_____________________________________________________________________________
118 : void AliMUONESDInterface::Clear(Option_t*)
119 : {
120 : /// clear memory
121 0 : delete fTracks; fTracks = 0x0;
122 0 : delete fDigits; fDigits = 0x0;
123 0 : delete fTriggers; fTriggers = 0x0;
124 0 : delete fClusterMap; fClusterMap = 0x0;
125 0 : delete fDigitMap; fDigitMap = 0x0;
126 0 : }
127 :
128 : //_____________________________________________________________________________
129 : void AliMUONESDInterface::Reset()
130 : {
131 : /// reset stores and maps
132 :
133 0 : if (fTracks) fTracks->Clear("C");
134 0 : else fTracks = NewTrackStore();
135 :
136 0 : if (fDigits) fDigits->Clear("C");
137 0 : else fDigits = NewDigitStore();
138 :
139 0 : if (fTriggers) fTriggers->Clear("C");
140 0 : else fTriggers = NewTriggerStore();
141 :
142 0 : if (fClusterMap) fClusterMap->Clear();
143 0 : else fClusterMap = new AliMpExMap;
144 0 : fClusterMap->SetOwner(kTRUE);
145 :
146 0 : if (fDigitMap) fDigitMap->Clear();
147 0 : else fDigitMap = new AliMpExMap;
148 0 : fDigitMap->SetOwner(kTRUE);
149 0 : }
150 :
151 : //_____________________________________________________________________________
152 : void AliMUONESDInterface::LoadEvent(AliESDEvent& esdEvent, Bool_t refit)
153 : {
154 : /// Extract MUON data from the given ESD event
155 :
156 : // reset data members
157 0 : Reset();
158 :
159 : // loop over ESD pads and fill the digit store
160 0 : for (Int_t iPad = 0; iPad < esdEvent.GetNumberOfMuonPads(); iPad++)
161 0 : Add(*(esdEvent.GetMuonPad(iPad)), *fDigits);
162 :
163 : // loop over ESD tracks and fill the stores
164 0 : for (Int_t iTrack = 0; iTrack < esdEvent.GetNumberOfMuonTracks(); iTrack++) {
165 :
166 : // get ESD track
167 0 : AliESDMuonTrack* esdTrack = esdEvent.GetMuonTrack(iTrack);
168 :
169 : // fill trigger store if related info are availables
170 0 : if (esdTrack->ContainTriggerData()) Add(*esdTrack, *fTriggers);
171 :
172 : // fill tracker data if availables
173 0 : if (!esdTrack->ContainTrackerData()) continue;
174 :
175 : // add it to track store
176 0 : AliMUONTrack* track = Add(*esdTrack, *fTracks, refit);
177 :
178 : // prepare cluster map
179 0 : AliMpExMap* cMap = new AliMpExMap;
180 0 : cMap->SetOwner(kFALSE);
181 0 : fClusterMap->Add(esdTrack->GetUniqueID(), cMap);
182 :
183 : // prepare digit maps
184 0 : AliMpExMap* dMaps = new AliMpExMap;
185 0 : dMaps->SetOwner(kTRUE);
186 0 : fDigitMap->Add(esdTrack->GetUniqueID(), dMaps);
187 :
188 : // loop over clusters
189 0 : for (Int_t iCluster = 0; iCluster < track->GetNClusters(); iCluster++) {
190 :
191 : // get cluster
192 0 : AliMUONVCluster* cluster = ((AliMUONTrackParam*)track->GetTrackParamAtCluster()->UncheckedAt(iCluster))->GetClusterPtr();
193 :
194 : // fill cluster map
195 0 : cMap->Add(cluster->GetUniqueID(), cluster);
196 :
197 : // prepare digit map
198 0 : AliMpExMap* dMap =new AliMpExMap;
199 0 : dMap->SetOwner(kFALSE);
200 0 : dMaps->Add(cluster->GetUniqueID(), dMap);
201 :
202 : // loop over digits
203 0 : for (Int_t iDigit = 0; iDigit < cluster->GetNDigits(); iDigit++) {
204 :
205 : // find the digit
206 0 : AliMUONVDigit* digit = fDigits->FindObject(cluster->GetDigitId(iDigit));
207 :
208 : // fill digit map
209 0 : if (digit) dMap->Add(digit->GetUniqueID(), digit);
210 0 : else AliError("a digit is missing");
211 :
212 : } // end of loop over pads
213 :
214 : } // end of loop over clusters
215 :
216 0 : } // end of loop over tracks
217 :
218 0 : }
219 :
220 : //___________________________________________________________________________
221 : Int_t AliMUONESDInterface::GetNTracks() const
222 : {
223 : /// return the number of tracks
224 0 : return fTracks ? fTracks->GetSize() : 0;
225 : }
226 :
227 : //___________________________________________________________________________
228 : Int_t AliMUONESDInterface::GetNClusters() const
229 : {
230 : /// return the number of clusters
231 : Int_t nClusters = 0;
232 : AliMUONTrack *track;
233 0 : TIter next(CreateTrackIterator());
234 0 : while ((track = static_cast<AliMUONTrack*>(next()))) nClusters += track->GetNClusters();
235 : return nClusters;
236 0 : }
237 :
238 : //___________________________________________________________________________
239 : Int_t AliMUONESDInterface::GetNClusters(UInt_t trackId) const
240 : {
241 : /// return the number of clusters in track "trackId"
242 0 : AliMUONTrack* track = FindTrack(trackId);
243 0 : return track ? track->GetNClusters() : 0;
244 : }
245 :
246 : //___________________________________________________________________________
247 : Int_t AliMUONESDInterface::GetNDigits() const
248 : {
249 : /// return the number of digits
250 0 : return fDigits ? fDigits->GetSize() : 0;
251 : }
252 :
253 : //___________________________________________________________________________
254 : Int_t AliMUONESDInterface::GetNDigits(UInt_t trackId) const
255 : {
256 : /// return the number of digits in all clusters of track "trackId"
257 : Int_t nDigits = 0;
258 : AliMUONVCluster *cluster;
259 0 : TIter next(CreateClusterIterator(trackId));
260 0 : while ((cluster = static_cast<AliMUONVCluster*>(next()))) nDigits += cluster->GetNDigits();
261 : return nDigits;
262 0 : }
263 :
264 : //___________________________________________________________________________
265 : Int_t AliMUONESDInterface::GetNDigits(UInt_t trackId, UInt_t clusterId) const
266 : {
267 : /// return the number of digits in cluster numbered "iCluster" of track "iTrack"
268 0 : AliMUONVCluster* cluster = FindCluster(trackId, clusterId);
269 0 : return cluster ? cluster->GetNDigits() : 0;
270 : }
271 :
272 : //___________________________________________________________________________
273 : Int_t AliMUONESDInterface::GetNDigitsInCluster(UInt_t clusterId) const
274 : {
275 : /// return the number of digits in cluster "clusterId"
276 0 : AliMUONVCluster* cluster = FindCluster(clusterId);
277 0 : return cluster ? cluster->GetNDigits() : 0;
278 : }
279 :
280 : //___________________________________________________________________________
281 : Int_t AliMUONESDInterface::GetNTriggers() const
282 : {
283 : /// return the number of triggers
284 0 : return fTriggers ? fTriggers->GetSize() : 0;
285 : }
286 :
287 : //___________________________________________________________________________
288 : Bool_t AliMUONESDInterface::DigitsStored(UInt_t trackId) const
289 : {
290 : /// return kTRUE if digits have been stored for all clusters of track "trackId"
291 : AliMUONVCluster *cluster;
292 0 : TIter next(CreateClusterIterator(trackId));
293 0 : while ((cluster = static_cast<AliMUONVCluster*>(next())))
294 0 : if (cluster->GetNDigits() == 0) return kFALSE;
295 0 : return kTRUE;
296 0 : }
297 :
298 : //___________________________________________________________________________
299 : AliMUONTrack* AliMUONESDInterface::FindTrack(UInt_t trackId) const
300 : {
301 : /// return track "trackId" (0x0 if not found)
302 0 : AliMUONTrack *track = fTracks ? static_cast<AliMUONTrack*>(fTracks->FindObject(trackId)) : 0x0;
303 0 : if (!track) AliWarning(Form("track %d does not exist",trackId));
304 0 : return track;
305 : }
306 :
307 : //___________________________________________________________________________
308 : AliMUONVCluster* AliMUONESDInterface::FindCluster(UInt_t clusterId) const
309 : {
310 : /// loop over tracks and return the first cluster "clusterId" found (0x0 if not found)
311 : AliMpExMap *cMap;
312 : AliMUONVCluster* cluster = 0x0;
313 :
314 0 : if (fClusterMap) {
315 :
316 0 : TIter next(fClusterMap->CreateIterator());
317 0 : while ((cMap = static_cast<AliMpExMap*>(next()))) {
318 :
319 0 : cluster = static_cast<AliMUONVCluster*>(cMap->GetValue(clusterId));
320 0 : if (cluster) return cluster;
321 :
322 : }
323 :
324 0 : }
325 :
326 0 : if (!cluster) AliWarning(Form("cluster %d does not exist",clusterId));
327 0 : return 0x0;
328 0 : }
329 :
330 : //___________________________________________________________________________
331 : AliMUONVCluster* AliMUONESDInterface::FindCluster(UInt_t trackId, UInt_t clusterId) const
332 : {
333 : /// return cluster "clusterId" in track "trackId" (0x0 if not found)
334 0 : AliMpExMap *cMap = fClusterMap ? static_cast<AliMpExMap*>(fClusterMap->GetValue(trackId)) : 0x0;
335 0 : AliMUONVCluster* cluster = cMap ? static_cast<AliMUONVCluster*>(cMap->GetValue(clusterId)) : 0x0;
336 0 : if (!cluster) AliWarning(Form("cluster %d does not exist in track %d", clusterId, trackId));
337 0 : return cluster;
338 : }
339 :
340 : //___________________________________________________________________________
341 : AliMUONVDigit* AliMUONESDInterface::FindDigit(UInt_t digitId) const
342 : {
343 : /// return digit "digitId" (0x0 if not found)
344 0 : AliMUONVDigit *digit = fDigits ? fDigits->FindObject(digitId) : 0x0;
345 0 : if (!digit) AliWarning(Form("digit %d does not exist",digitId));
346 0 : return digit;
347 : }
348 :
349 : //___________________________________________________________________________
350 : AliMUONLocalTrigger* AliMUONESDInterface::FindLocalTrigger(Int_t boardNumber) const
351 : {
352 : /// return MUON local trigger "boardNumber"
353 0 : return (fTriggers) ? fTriggers->FindLocal(boardNumber) : 0x0;
354 : }
355 :
356 : //___________________________________________________________________________
357 : TIterator* AliMUONESDInterface::CreateTrackIterator() const
358 : {
359 : /// return iterator over all tracks
360 0 : return fTracks ? fTracks->CreateIterator() : 0x0;
361 : }
362 :
363 : //___________________________________________________________________________
364 : TIterator* AliMUONESDInterface::CreateClusterIterator() const
365 : {
366 : /// return iterator over all clusters
367 0 : return fClusterMap ? new AliMUON2DMapIterator(*fClusterMap) : 0x0;
368 0 : }
369 :
370 : //___________________________________________________________________________
371 : TIterator* AliMUONESDInterface::CreateClusterIterator(UInt_t trackId) const
372 : {
373 : /// return iterator over clusters of track "trackId"
374 0 : AliMpExMap *cMap = fClusterMap ? static_cast<AliMpExMap*>(fClusterMap->GetValue(trackId)) : 0x0;
375 0 : return cMap ? cMap->CreateIterator() : 0x0;
376 : }
377 :
378 : //___________________________________________________________________________
379 : TIterator* AliMUONESDInterface::CreateDigitIterator() const
380 : {
381 : /// return iterator over all digits
382 0 : return fDigits ? fDigits->CreateIterator() : 0x0;
383 : }
384 :
385 : //___________________________________________________________________________
386 : TIterator* AliMUONESDInterface::CreateDigitIterator(UInt_t trackId) const
387 : {
388 : /// return iterator over all digits of track "trackId"
389 0 : AliMpExMap* dMaps = fDigitMap ? static_cast<AliMpExMap*>(fDigitMap->GetValue(trackId)) : 0x0;
390 0 : return dMaps ? new AliMUON2DMapIterator(*dMaps) : 0x0;
391 0 : }
392 :
393 : //___________________________________________________________________________
394 : TIterator* AliMUONESDInterface::CreateDigitIterator(UInt_t trackId, UInt_t clusterId) const
395 : {
396 : /// return iterator over digits of cluster "clusterId" in track "trackId"
397 0 : AliMpExMap* dMaps = fDigitMap ? static_cast<AliMpExMap*>(fDigitMap->GetValue(trackId)) : 0x0;
398 0 : AliMpExMap* dMap = dMaps ? static_cast<AliMpExMap*>(dMaps->GetValue(clusterId)) : 0x0;
399 0 : return dMap ? dMap->CreateIterator() : 0x0;
400 : }
401 :
402 : //___________________________________________________________________________
403 : TIterator* AliMUONESDInterface::CreateDigitIteratorInCluster(UInt_t clusterId) const
404 : {
405 : /// return iterator over digits of the first cluster "clusterId" found by looping over all tracks
406 : AliMpExMap *dMaps;
407 : AliMpExMap* dMap = 0x0;
408 :
409 0 : if (fDigitMap) {
410 :
411 0 : TIter next(fDigitMap->CreateIterator());
412 0 : while ((dMaps = static_cast<AliMpExMap*>(next()))) {
413 :
414 0 : dMap = static_cast<AliMpExMap*>(dMaps->GetValue(clusterId));
415 0 : if (dMap) return dMap->CreateIterator();
416 :
417 : }
418 :
419 0 : }
420 :
421 0 : return 0x0;
422 0 : }
423 :
424 : //___________________________________________________________________________
425 : TIterator* AliMUONESDInterface::CreateLocalTriggerIterator() const
426 : {
427 : /// return iterator over all local trigger
428 0 : return fTriggers ? fTriggers->CreateLocalIterator() : 0x0;
429 : }
430 :
431 : //+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++//
432 : // static methods //
433 : //+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++//
434 :
435 : //_____________________________________________________________________________
436 : void AliMUONESDInterface::ResetTracker(const AliMUONRecoParam* recoParam, Bool_t info)
437 : {
438 : /// Reset the MUON tracker using "recoParam" if provided.
439 : /// If not provided, will get them from OCDB.
440 : /// Load the magnetic field from OCDB if not already set.
441 :
442 0 : delete fgTracker;
443 0 : delete fgRecoParam;
444 :
445 0 : if (recoParam) {
446 :
447 : // use provided recoParam
448 0 : fgRecoParam = new AliMUONRecoParam(*recoParam);
449 :
450 0 : if (info) AliInfoClass("will refit tracks with provided RecoParam:");
451 :
452 : } else {
453 :
454 : // or get them from OCDB
455 0 : fgRecoParam = AliMUONCDB::LoadRecoParam();
456 :
457 0 : if (!fgRecoParam) AliFatalClass("RecoParam have not been set!");
458 :
459 : }
460 :
461 : // print useful parameters for refitting
462 0 : if (info) {
463 0 : cout<<" --> Tracking mode = "<<fgRecoParam->GetTrackingMode()<<endl;
464 0 : if (fgRecoParam->UseSmoother()) cout<<" --> Use Smoother"<<endl;
465 0 : else cout<<" --> Do not use smoother"<<endl;
466 0 : cout<<" --> Vertex dispersion in bending direction = "
467 0 : <<fgRecoParam->GetBendingVertexDispersion()<<" cm"<<endl;
468 0 : }
469 :
470 : // load magnetic field for track extrapolation if not already set
471 0 : if (!TGeoGlobalMagField::Instance()->GetField() && !AliMUONCDB::LoadField())
472 0 : AliFatalClass("Magnetic field has not been set!");
473 :
474 0 : fgTracker = AliMUONTracker::CreateTrackReconstructor(fgRecoParam,0x0,0x0);
475 :
476 0 : }
477 :
478 : //_____________________________________________________________________________
479 : AliMUONVTrackStore* AliMUONESDInterface::NewTrackStore()
480 : {
481 : /// Create an empty track store of type fgTrackStoreName
482 0 : TClass* classPtr = TClass::GetClass(fgTrackStoreName);
483 0 : if (!classPtr || !classPtr->InheritsFrom("AliMUONVTrackStore")) {
484 0 : AliErrorClass(Form("Unable to create store of type %s", fgTrackStoreName.Data()));
485 0 : return 0x0;
486 : }
487 0 : return reinterpret_cast<AliMUONVTrackStore*>(gROOT->ProcessLineFast(Form("new %s()",fgTrackStoreName.Data())));
488 0 : }
489 :
490 : //_____________________________________________________________________________
491 : AliMUONVClusterStore* AliMUONESDInterface::NewClusterStore()
492 : {
493 : /// Create an empty cluster store of type fgClusterStoreName
494 0 : TClass* classPtr = TClass::GetClass(fgClusterStoreName);
495 0 : if (!classPtr || !classPtr->InheritsFrom("AliMUONVClusterStore")) {
496 0 : AliErrorClass(Form("Unable to create store of type %s", fgClusterStoreName.Data()));
497 0 : return 0x0;
498 : }
499 0 : return reinterpret_cast<AliMUONVClusterStore*>(gROOT->ProcessLineFast(Form("new %s()",fgClusterStoreName.Data())));
500 0 : }
501 :
502 : //_____________________________________________________________________________
503 : AliMUONVCluster* AliMUONESDInterface::NewCluster()
504 : {
505 : /// Create an empty cluster of type associated to the cluster store of type fgClusterStoreName
506 0 : TClass* classPtr = TClass::GetClass(fgClusterStoreName);
507 0 : if (!classPtr || !classPtr->InheritsFrom("AliMUONVClusterStore")) {
508 0 : AliErrorClass(Form("Unable to create store of type %s", fgClusterStoreName.Data()));
509 0 : return 0x0;
510 : }
511 0 : AliMUONVClusterStore* cStore = reinterpret_cast<AliMUONVClusterStore*>(classPtr->New());
512 0 : AliMUONVCluster* cluster = cStore->CreateCluster(0,0,0);
513 0 : delete cStore;
514 : return cluster;
515 0 : }
516 :
517 : //_____________________________________________________________________________
518 : AliMUONVDigitStore* AliMUONESDInterface::NewDigitStore()
519 : {
520 : /// Create an empty digit store of type fgDigitStoreName
521 0 : TClass* classPtr = TClass::GetClass(fgDigitStoreName);
522 0 : if (!classPtr || !classPtr->InheritsFrom("AliMUONVDigitStore")) {
523 0 : AliErrorClass(Form("Unable to create store of type %s", fgDigitStoreName.Data()));
524 0 : return 0x0;
525 : }
526 0 : return reinterpret_cast<AliMUONVDigitStore*>(gROOT->ProcessLineFast(Form("new %s()",fgDigitStoreName.Data())));
527 0 : }
528 :
529 : //_____________________________________________________________________________
530 : AliMUONVDigit* AliMUONESDInterface::NewDigit()
531 : {
532 : /// Create an empty digit of type associated to the digit store of type fgDigitStoreName
533 0 : TClass* classPtr = TClass::GetClass(fgDigitStoreName);
534 0 : if (!classPtr || !classPtr->InheritsFrom("AliMUONVDigitStore")) {
535 0 : AliErrorClass(Form("Unable to create store of type %s", fgDigitStoreName.Data()));
536 0 : return 0x0;
537 : }
538 0 : AliMUONVDigitStore* dStore = reinterpret_cast<AliMUONVDigitStore*>(classPtr->New());
539 0 : AliMUONVDigit* digit = dStore->CreateDigit(0,0,0,0);
540 0 : delete dStore;
541 : return digit;
542 0 : }
543 :
544 : //_____________________________________________________________________________
545 : AliMUONVTriggerStore* AliMUONESDInterface::NewTriggerStore()
546 : {
547 : /// Create an empty trigger store of type fgTriggerStoreName
548 0 : TClass* classPtr = TClass::GetClass(fgTriggerStoreName);
549 0 : if (!classPtr || !classPtr->InheritsFrom("AliMUONVTriggerStore")) {
550 0 : AliErrorClass(Form("Unable to create store of type %s", fgTriggerStoreName.Data()));
551 0 : return 0x0;
552 : }
553 0 : return reinterpret_cast<AliMUONVTriggerStore*>(gROOT->ProcessLineFast(Form("new %s()",fgTriggerStoreName.Data())));
554 0 : }
555 :
556 : //_____________________________________________________________________________
557 : AliMUONVTriggerTrackStore* AliMUONESDInterface::NewTriggerTrackStore()
558 : {
559 : /// Create an empty trigger track store of type fgTriggerTrackStoreName
560 0 : TClass* classPtr = TClass::GetClass(fgTriggerTrackStoreName);
561 0 : if (!classPtr || !classPtr->InheritsFrom("AliMUONVTriggerTrackStore")) {
562 0 : AliErrorClass(Form("Unable to create store of type %s", fgTriggerTrackStoreName.Data()));
563 0 : return 0x0;
564 : }
565 0 : return reinterpret_cast<AliMUONVTriggerTrackStore*>(gROOT->ProcessLineFast(Form("new %s()",fgTriggerTrackStoreName.Data())));
566 0 : }
567 :
568 : //_________________________________________________________________________
569 : void AliMUONESDInterface::GetParamAtVertex(const AliESDMuonTrack& esdTrack, AliMUONTrackParam& trackParam)
570 : {
571 : /// Get parameters at vertex from ESDMuon track
572 0 : trackParam.SetZ(esdTrack.GetZ());
573 0 : trackParam.SetNonBendingCoor(esdTrack.GetNonBendingCoor());
574 0 : trackParam.SetNonBendingSlope(TMath::Tan(esdTrack.GetThetaX()));
575 0 : trackParam.SetBendingCoor(esdTrack.GetBendingCoor());
576 0 : trackParam.SetBendingSlope(TMath::Tan(esdTrack.GetThetaY()));
577 0 : trackParam.SetInverseBendingMomentum(esdTrack.GetInverseBendingMomentum());
578 0 : }
579 :
580 : //_________________________________________________________________________
581 : void AliMUONESDInterface::GetParamAtDCA(const AliESDMuonTrack& esdTrack, AliMUONTrackParam& trackParam)
582 : {
583 : /// Get parameters at DCA from ESDMuon track
584 0 : trackParam.SetZ(esdTrack.GetZ());
585 0 : trackParam.SetNonBendingCoor(esdTrack.GetNonBendingCoorAtDCA());
586 0 : trackParam.SetNonBendingSlope(TMath::Tan(esdTrack.GetThetaXAtDCA()));
587 0 : trackParam.SetBendingCoor(esdTrack.GetBendingCoorAtDCA());
588 0 : trackParam.SetBendingSlope(TMath::Tan(esdTrack.GetThetaYAtDCA()));
589 0 : trackParam.SetInverseBendingMomentum(esdTrack.GetInverseBendingMomentumAtDCA());
590 0 : }
591 :
592 : //_________________________________________________________________________
593 : void AliMUONESDInterface::GetParamAtFirstCluster(const AliESDMuonTrack& esdTrack, AliMUONTrackParam& trackParam)
594 : {
595 : /// Get parameters at first cluster from ESDMuon track
596 0 : trackParam.SetZ(esdTrack.GetZUncorrected());
597 0 : trackParam.SetNonBendingCoor(esdTrack.GetNonBendingCoorUncorrected());
598 0 : trackParam.SetNonBendingSlope(TMath::Tan(esdTrack.GetThetaXUncorrected()));
599 0 : trackParam.SetBendingCoor(esdTrack.GetBendingCoorUncorrected());
600 0 : trackParam.SetBendingSlope(TMath::Tan(esdTrack.GetThetaYUncorrected()));
601 0 : trackParam.SetInverseBendingMomentum(esdTrack.GetInverseBendingMomentumUncorrected());
602 0 : }
603 :
604 : //_________________________________________________________________________
605 : void AliMUONESDInterface::GetParamCov(const AliESDMuonTrack& esdTrack, AliMUONTrackParam& trackParam)
606 : {
607 : /// Get parameters covariances from ESD track
608 :
609 : // get ESD covariance matrix
610 0 : TMatrixD covariances(5,5);
611 0 : esdTrack.GetCovariances(covariances);
612 :
613 : // compute Jacobian to change the coordinate system
614 : // from (X,thetaX,Y,thetaY,c/pYZ) to (X,slopeX,Y,slopeY,c/pYZ)
615 0 : Double_t cosThetaX = TMath::Cos(esdTrack.GetThetaXUncorrected());
616 0 : Double_t cosThetaY = TMath::Cos(esdTrack.GetThetaYUncorrected());
617 0 : TMatrixD jacob(5,5);
618 0 : jacob.Zero();
619 0 : jacob(0,0) = 1.;
620 0 : jacob(1,1) = 1. / cosThetaX / cosThetaX;
621 0 : jacob(2,2) = 1.;
622 0 : jacob(3,3) = 1. / cosThetaY / cosThetaY;
623 0 : jacob(4,4) = 1.;
624 :
625 : // compute covariance matrix in ESD coordinate system
626 0 : TMatrixD tmp(covariances,TMatrixD::kMultTranspose,jacob);
627 0 : trackParam.SetCovariances(TMatrixD(jacob,TMatrixD::kMult,tmp));
628 :
629 0 : }
630 :
631 : //_________________________________________________________________________
632 : void AliMUONESDInterface::SetParamAtVertex(const AliMUONTrackParam& trackParam, AliESDMuonTrack& esdTrack)
633 : {
634 : /// Set parameters in ESD track
635 32 : esdTrack.SetZ(trackParam.GetZ());
636 16 : esdTrack.SetNonBendingCoor(trackParam.GetNonBendingCoor());
637 16 : esdTrack.SetThetaX(TMath::ATan(trackParam.GetNonBendingSlope()));
638 16 : esdTrack.SetBendingCoor(trackParam.GetBendingCoor());
639 16 : esdTrack.SetThetaY(TMath::ATan(trackParam.GetBendingSlope()));
640 16 : esdTrack.SetInverseBendingMomentum(trackParam.GetInverseBendingMomentum());
641 16 : }
642 :
643 : //_________________________________________________________________________
644 : void AliMUONESDInterface::SetParamAtDCA(const AliMUONTrackParam& trackParam, AliESDMuonTrack& esdTrack)
645 : {
646 : /// Set parameters in ESD track
647 32 : esdTrack.SetNonBendingCoorAtDCA(trackParam.GetNonBendingCoor());
648 16 : esdTrack.SetThetaXAtDCA(TMath::ATan(trackParam.GetNonBendingSlope()));
649 16 : esdTrack.SetBendingCoorAtDCA(trackParam.GetBendingCoor());
650 16 : esdTrack.SetThetaYAtDCA(TMath::ATan(trackParam.GetBendingSlope()));
651 16 : esdTrack.SetInverseBendingMomentumAtDCA(trackParam.GetInverseBendingMomentum());
652 16 : }
653 :
654 : //_________________________________________________________________________
655 : void AliMUONESDInterface::SetParamAtFirstCluster(const AliMUONTrackParam& trackParam, AliESDMuonTrack& esdTrack)
656 : {
657 : /// Set parameters in ESD track
658 32 : esdTrack.SetZUncorrected(trackParam.GetZ());
659 16 : esdTrack.SetNonBendingCoorUncorrected(trackParam.GetNonBendingCoor());
660 16 : esdTrack.SetThetaXUncorrected(TMath::ATan(trackParam.GetNonBendingSlope()));
661 16 : esdTrack.SetBendingCoorUncorrected(trackParam.GetBendingCoor());
662 16 : esdTrack.SetThetaYUncorrected(TMath::ATan(trackParam.GetBendingSlope()));
663 16 : esdTrack.SetInverseBendingMomentumUncorrected(trackParam.GetInverseBendingMomentum());
664 16 : }
665 :
666 : //_________________________________________________________________________
667 : void AliMUONESDInterface::SetParamCov(const AliMUONTrackParam& trackParam, AliESDMuonTrack& esdTrack)
668 : {
669 : /// Set parameters covariances in ESD track
670 :
671 : // set null matrix if covariances does not exist
672 32 : if (!trackParam.CovariancesExist()) {
673 0 : TMatrixD tmp(5,5);
674 0 : tmp.Zero();
675 0 : esdTrack.SetCovariances(tmp);
676 : return;
677 0 : }
678 :
679 : // compute Jacobian to change the coordinate system
680 : // from (X,slopeX,Y,slopeY,c/pYZ) to (X,thetaX,Y,thetaY,c/pYZ)
681 16 : Double_t cosThetaX = TMath::Cos(TMath::ATan(trackParam.GetNonBendingSlope()));
682 16 : Double_t cosThetaY = TMath::Cos(TMath::ATan(trackParam.GetBendingSlope()));
683 16 : TMatrixD jacob(5,5);
684 16 : jacob.Zero();
685 32 : jacob(0,0) = 1.;
686 32 : jacob(1,1) = cosThetaX * cosThetaX;
687 32 : jacob(2,2) = 1.;
688 32 : jacob(3,3) = cosThetaY * cosThetaY;
689 32 : jacob(4,4) = 1.;
690 :
691 : // compute covariance matrix in ESD coordinate system
692 32 : TMatrixD tmp(trackParam.GetCovariances(),TMatrixD::kMultTranspose,jacob);
693 48 : esdTrack.SetCovariances(TMatrixD(jacob,TMatrixD::kMult,tmp));
694 :
695 32 : }
696 :
697 : //_____________________________________________________________________________
698 : void AliMUONESDInterface::ESDToMUON(const AliESDMuonTrack& esdTrack, AliMUONTrack& track, Bool_t refit)
699 : {
700 : /// Transfert data from ESDMuon track to MUON track.
701 : /// ESDMuon track must hold the pointer to the ESD event otherwise we cannot recover the clusters.
702 : /// If refit = kTRUE, the track parameters at each cluster are obtained by refitting the track
703 : /// or by extrapolating the parameters at the first one if the refit failed.
704 : /// If refit = kFALSE, only the track parameters at first cluster are valid.
705 : /// note: You can set the recoParam used to refit the MUON track with ResetTracker(...);
706 : /// By default we use Kalman filter + Smoother
707 :
708 : // if the ESDMuon track is a ghost then return an empty MUON track
709 0 : if (!esdTrack.ContainTrackerData()) {
710 0 : track.Reset();
711 0 : track.SetUniqueID(esdTrack.GetUniqueID());
712 0 : return;
713 : }
714 :
715 0 : track.Clear("C");
716 :
717 : // global info
718 0 : track.SetUniqueID(esdTrack.GetUniqueID());
719 0 : track.FitWithVertex(kFALSE);
720 0 : track.FitWithMCS(kFALSE);
721 0 : track.SetImproved(kFALSE);
722 0 : track.SetVertexErrXY2(0.,0.);
723 0 : track.SetGlobalChi2(esdTrack.GetChi2());
724 0 : track.SetMatchTrigger(esdTrack.GetMatchTrigger());
725 0 : track.SetChi2MatchTrigger(esdTrack.GetChi2MatchTrigger());
726 0 : track.SetHitsPatternInTrigCh(esdTrack.GetHitsPatternInTrigCh());
727 0 : track.SetHitsPatternInTrigChTrk(esdTrack.GetHitsPatternInTrigChTrk());
728 0 : track.SetLocalTrigger(esdTrack.LoCircuit(), esdTrack.LoStripX(), esdTrack.LoStripY(),
729 0 : esdTrack.LoDev(), esdTrack.LoLpt(), esdTrack.LoHpt(),
730 0 : esdTrack.GetTriggerWithoutChamber());
731 0 : track.Connected(esdTrack.IsConnected());
732 :
733 : // track parameters at vertex
734 0 : AliMUONTrackParam paramAtVertex;
735 0 : GetParamAtVertex(esdTrack, paramAtVertex);
736 0 : track.SetTrackParamAtVertex(¶mAtVertex);
737 :
738 : // track parameters at first cluster
739 0 : AliMUONTrackParam param;
740 0 : GetParamAtFirstCluster(esdTrack, param);
741 0 : GetParamCov(esdTrack, param);
742 :
743 : // create empty cluster
744 0 : AliMUONVCluster* cluster = NewCluster();
745 :
746 : // fill TrackParamAtCluster with track parameters at each cluster if available
747 : Bool_t clusterFound = kFALSE;
748 0 : if(esdTrack.GetESDEvent()) {
749 :
750 : // loop over cluster Id
751 0 : for (Int_t i = 0; i < esdTrack.GetNClusters(); i++) {
752 :
753 : // recover the ESDMuon cluster
754 0 : AliESDMuonCluster *esdCluster = esdTrack.GetESDEvent()->FindMuonCluster(esdTrack.GetClusterId(i));
755 0 : if (esdCluster) clusterFound = kTRUE;
756 : else {
757 0 : AliErrorClass("a cluster is missing in ESD");
758 0 : continue;
759 : }
760 :
761 : // copy cluster information
762 0 : ESDToMUON(*esdCluster, *cluster);
763 :
764 : // only set the Z parameter to avoid error in the AddTrackParamAtCluster(...) method
765 0 : param.SetZ(cluster->GetZ());
766 :
767 : // add common track parameters at current cluster
768 0 : track.AddTrackParamAtCluster(param, *cluster, kTRUE);
769 :
770 0 : }
771 :
772 : // refit the track to get better parameters and covariances at each cluster (temporary disable track improvement)
773 0 : if (clusterFound && refit) {
774 :
775 0 : AliMUONTrackParam *firstTrackParam = (AliMUONTrackParam*) track.GetTrackParamAtCluster()->First();
776 0 : AliMUONTrackParam paramSave(*firstTrackParam);
777 :
778 0 : if (!fgTracker) ResetTracker();
779 :
780 0 : if (!fgTracker->RefitTrack(track, kFALSE) && track.GetGlobalChi2() < AliMUONTrack::MaxChi2()) {
781 0 : *firstTrackParam = paramSave;
782 0 : track.UpdateCovTrackParamAtCluster();
783 : }
784 :
785 0 : }
786 :
787 : }
788 :
789 : // fill TrackParamAtCluster with only track parameters at first (fake) cluster if no cluster found in ESD
790 0 : if (!clusterFound) {
791 :
792 : // get number of the first hit chamber according to the MUONClusterMap
793 : Int_t firstCh = 0;
794 0 : while (firstCh < 10 && !esdTrack.IsInMuonClusterMap(firstCh)) firstCh++;
795 :
796 : // produce fake cluster at this chamber
797 0 : cluster->SetUniqueID(AliMUONVCluster::BuildUniqueID(firstCh, 0, 0));
798 0 : cluster->SetXYZ(param.GetNonBendingCoor(), param.GetBendingCoor(), param.GetZ());
799 0 : cluster->SetErrXY(0., 0.);
800 :
801 : // add track parameters at first (fake) cluster
802 0 : track.AddTrackParamAtCluster(param, *cluster, kTRUE);
803 :
804 0 : }
805 :
806 : // set the MC label from ESD track
807 0 : track.SetMCLabel(esdTrack.GetLabel());
808 :
809 0 : delete cluster;
810 :
811 0 : }
812 :
813 : //_____________________________________________________________________________
814 : void AliMUONESDInterface::ESDToMUON(const AliESDMuonTrack& esdTrack, AliMUONLocalTrigger& locTrg)
815 : {
816 : /// Transfert trigger data from ESDMuon track to the MUONLocalTtrigger object
817 :
818 : // reset the local trigger content
819 0 : AliMUONLocalTrigger emptyLocTrg;
820 0 : locTrg = emptyLocTrg;
821 :
822 : // if the ESDMuon track is a ghost then return an empty local trigger
823 0 : if (!esdTrack.ContainTriggerData()) return;
824 :
825 0 : locTrg.SetUniqueID(esdTrack.GetUniqueID());
826 0 : locTrg.SetLoCircuit(esdTrack.LoCircuit());
827 0 : locTrg.SetLoStripX(esdTrack.LoStripX());
828 0 : locTrg.SetLoStripY(esdTrack.LoStripY());
829 0 : locTrg.SetDeviation(esdTrack.LoDev());
830 0 : locTrg.SetLoLpt(esdTrack.LoLpt());
831 0 : locTrg.SetLoHpt(esdTrack.LoHpt());
832 0 : for (Int_t ich = 0; ich < 4; ich++)
833 0 : if (esdTrack.TriggerFiredWithoutChamber(ich))
834 0 : locTrg.SetTriggerWithoutChamber(ich);
835 0 : locTrg.SetLoTrigY(0);
836 0 : locTrg.SetX1Pattern(esdTrack.GetTriggerX1Pattern());
837 0 : locTrg.SetX2Pattern(esdTrack.GetTriggerX2Pattern());
838 0 : locTrg.SetX3Pattern(esdTrack.GetTriggerX3Pattern());
839 0 : locTrg.SetX4Pattern(esdTrack.GetTriggerX4Pattern());
840 0 : locTrg.SetY1Pattern(esdTrack.GetTriggerY1Pattern());
841 0 : locTrg.SetY2Pattern(esdTrack.GetTriggerY2Pattern());
842 0 : locTrg.SetY3Pattern(esdTrack.GetTriggerY3Pattern());
843 0 : locTrg.SetY4Pattern(esdTrack.GetTriggerY4Pattern());
844 :
845 0 : }
846 :
847 : //_____________________________________________________________________________
848 : void AliMUONESDInterface::ESDToMUON(const AliESDMuonCluster& esdCluster, AliMUONVCluster& cluster)
849 : {
850 : /// Transfert data from ESDMuon cluster to MUON cluster
851 :
852 0 : cluster.Clear("C");
853 :
854 0 : cluster.SetUniqueID(esdCluster.GetUniqueID());
855 0 : cluster.SetXYZ(esdCluster.GetX(), esdCluster.GetY(), esdCluster.GetZ());
856 0 : cluster.SetErrXY(esdCluster.GetErrX(),esdCluster.GetErrY());
857 0 : cluster.SetCharge(esdCluster.GetCharge());
858 0 : cluster.SetChi2(esdCluster.GetChi2());
859 0 : cluster.SetMCLabel(esdCluster.GetLabel());
860 :
861 0 : cluster.SetDigitsId(esdCluster.GetNPads(), esdCluster.GetPadsId());
862 :
863 0 : }
864 :
865 : //___________________________________________________________________________
866 : void AliMUONESDInterface::ESDToMUON(const AliESDMuonPad& esdPad, AliMUONVDigit& digit)
867 : {
868 : /// Transfert data from ESDMuon pad to MUON digit
869 : /// Load mapping from OCDB if not already set
870 :
871 0 : if (!AliMpSegmentation::Instance(kFALSE) && !AliMUONCDB::LoadMapping(kTRUE))
872 0 : AliFatalClass("mapping segmentation has not been set!");
873 :
874 0 : const AliMpVSegmentation* seg = AliMpSegmentation::Instance()->GetMpSegmentationByElectronics(esdPad.GetDetElemId(), esdPad.GetManuId());
875 0 : AliMpPad pad = seg->PadByLocation(esdPad.GetManuId(), esdPad.GetManuChannel(), kFALSE);
876 :
877 0 : digit.Saturated(esdPad.IsSaturated());
878 0 : digit.Used(kFALSE);
879 0 : digit.Calibrated(esdPad.IsCalibrated());
880 0 : digit.SetUniqueID(esdPad.GetUniqueID());
881 0 : digit.SetCharge(esdPad.GetCharge());
882 0 : digit.SetADC(esdPad.GetADC());
883 0 : digit.SetPadXY(pad.GetIx(), pad.GetIy());
884 :
885 0 : }
886 :
887 : //_____________________________________________________________________________
888 : void AliMUONESDInterface::MUONToESD(const AliMUONTrack& track, AliESDEvent& esd, const Double_t vertex[3],
889 : const AliMUONVDigitStore* digits, const AliMUONLocalTrigger* locTrg)
890 : {
891 : /// Transfert data from MUON track to ESD event (ESDTrack + ESDClusters)
892 : /// Incorporate the ESDPads if the digits are provided
893 : /// Add trigger info if the MUON track is matched with a trigger track
894 :
895 : // transfert track info
896 32 : AliESDMuonTrack *esdTrack = esd.NewMuonTrack();
897 16 : MUONToESD(track, *esdTrack, vertex, locTrg);
898 :
899 : // transfert cluster info if any
900 356 : for (Int_t i = 0; i < track.GetNClusters(); i++) {
901 162 : AliMUONTrackParam *param = static_cast<AliMUONTrackParam*>(track.GetTrackParamAtCluster()->UncheckedAt(i));
902 162 : MUONToESD(*(param->GetClusterPtr()), esd, digits);
903 : }
904 :
905 16 : }
906 :
907 : //_____________________________________________________________________________
908 : void AliMUONESDInterface::MUONToESD(const AliMUONTrack& track, AliESDMuonTrack& esdTrack,
909 : const Double_t vertex[3], const AliMUONLocalTrigger* locTrg)
910 : {
911 : /// Transfert data from MUON track to ESDMuon track (ESDMuon track only contains clusters'Id, not cluster themselves)
912 : /// Add trigger info if the MUON track is matched with a trigger track
913 :
914 : // empty MUON track -> produce a ghost ESDMuon track if trigger info are available otherwise produce an empty track
915 32 : if (track.GetNClusters() == 0) {
916 0 : if (locTrg) MUONToESD(*locTrg, esdTrack, track.GetUniqueID());
917 : else {
918 0 : cout<<"W-AliMUONESDInterface::MUONToESD: will produce an empty ESDMuon track"<<endl;
919 0 : esdTrack.Reset();
920 0 : esdTrack.SetUniqueID(0xFFFFFFFF);
921 : }
922 : return;
923 : }
924 :
925 16 : esdTrack.Clear("C");
926 :
927 : // set global info
928 16 : esdTrack.SetUniqueID(track.GetUniqueID());
929 16 : esdTrack.SetChi2(track.GetGlobalChi2());
930 16 : esdTrack.SetLabel(track.GetMCLabel());
931 :
932 : // set param at first cluster
933 16 : AliMUONTrackParam* trackParam = static_cast<AliMUONTrackParam*>((track.GetTrackParamAtCluster())->First());
934 16 : SetParamAtFirstCluster(*trackParam, esdTrack);
935 16 : SetParamCov(*trackParam, esdTrack);
936 :
937 : // set transverse position at the end of the absorber
938 16 : AliMUONTrackParam trackParamAtAbsEnd(*trackParam);
939 16 : AliMUONTrackExtrap::ExtrapToZ(&trackParamAtAbsEnd, AliMUONConstants::AbsZEnd());
940 16 : Double_t xAbs = trackParamAtAbsEnd.GetNonBendingCoor();
941 16 : Double_t yAbs = trackParamAtAbsEnd.GetBendingCoor();
942 16 : esdTrack.SetRAtAbsorberEnd(TMath::Sqrt(xAbs*xAbs + yAbs*yAbs));
943 :
944 : // set param at vertex
945 16 : AliMUONTrackParam trackParamAtVtx(*trackParam);
946 16 : AliMUONTrackExtrap::ExtrapToVertex(&trackParamAtVtx, vertex[0], vertex[1], vertex[2], 0., 0.);
947 16 : SetParamAtVertex(trackParamAtVtx, esdTrack);
948 :
949 : // set param at Distance of Closest Approach
950 16 : AliMUONTrackParam trackParamAtDCA(*trackParam);
951 16 : AliMUONTrackExtrap::ExtrapToVertexWithoutBranson(&trackParamAtDCA, vertex[2]);
952 16 : SetParamAtDCA(trackParamAtDCA, esdTrack);
953 :
954 : // set muon cluster info
955 16 : esdTrack.SetMuonClusterMap(0);
956 356 : while (trackParam) {
957 162 : AliMUONVCluster *cluster = trackParam->GetClusterPtr();
958 324 : esdTrack.AddClusterId(cluster->GetUniqueID());
959 324 : esdTrack.AddInMuonClusterMap(cluster->GetChamberId());
960 486 : trackParam = static_cast<AliMUONTrackParam*>(track.GetTrackParamAtCluster()->After(trackParam));
961 : }
962 :
963 : // set connected flag
964 16 : esdTrack.Connected(track.IsConnected());
965 :
966 : // set trigger info
967 16 : esdTrack.SetLocalTrigger(track.GetLocalTrigger());
968 16 : esdTrack.SetChi2MatchTrigger(track.GetChi2MatchTrigger());
969 16 : esdTrack.SetHitsPatternInTrigCh(track.GetHitsPatternInTrigCh());
970 16 : esdTrack.SetHitsPatternInTrigChTrk(track.GetHitsPatternInTrigChTrk());
971 16 : if (locTrg) {
972 12 : esdTrack.SetTriggerX1Pattern(locTrg->GetX1Pattern());
973 12 : esdTrack.SetTriggerY1Pattern(locTrg->GetY1Pattern());
974 12 : esdTrack.SetTriggerX2Pattern(locTrg->GetX2Pattern());
975 12 : esdTrack.SetTriggerY2Pattern(locTrg->GetY2Pattern());
976 12 : esdTrack.SetTriggerX3Pattern(locTrg->GetX3Pattern());
977 12 : esdTrack.SetTriggerY3Pattern(locTrg->GetY3Pattern());
978 12 : esdTrack.SetTriggerX4Pattern(locTrg->GetX4Pattern());
979 12 : esdTrack.SetTriggerY4Pattern(locTrg->GetY4Pattern());
980 12 : } else {
981 4 : esdTrack.SetTriggerX1Pattern(0);
982 4 : esdTrack.SetTriggerY1Pattern(0);
983 4 : esdTrack.SetTriggerX2Pattern(0);
984 4 : esdTrack.SetTriggerY2Pattern(0);
985 4 : esdTrack.SetTriggerX3Pattern(0);
986 4 : esdTrack.SetTriggerY3Pattern(0);
987 4 : esdTrack.SetTriggerX4Pattern(0);
988 4 : esdTrack.SetTriggerY4Pattern(0);
989 : }
990 :
991 32 : }
992 :
993 : //_____________________________________________________________________________
994 : void AliMUONESDInterface::MUONToESD(const AliMUONLocalTrigger& locTrg, AliESDEvent& esd,
995 : UInt_t trackId, const AliMUONTriggerTrack* triggerTrack)
996 : {
997 : /// Add in ESD event a ghost ESDMuon track containing only informations about trigger track
998 4 : AliESDMuonTrack *esdTrack = esd.NewMuonTrack();
999 2 : MUONToESD(locTrg, *esdTrack, trackId, triggerTrack);
1000 2 : }
1001 :
1002 : //_____________________________________________________________________________
1003 : void AliMUONESDInterface::MUONToESD(const AliMUONLocalTrigger& locTrg, AliESDMuonTrack& esdTrack,
1004 : UInt_t trackId, const AliMUONTriggerTrack* triggerTrack)
1005 : {
1006 : /// Build ghost ESDMuon track containing only informations about trigger track
1007 :
1008 4 : esdTrack.Reset();
1009 2 : esdTrack.SetUniqueID(trackId);
1010 :
1011 : // set trigger info
1012 2 : AliMUONTrack muonTrack;
1013 4 : muonTrack.SetLocalTrigger(locTrg.LoCircuit(),
1014 2 : locTrg.LoStripX(),
1015 2 : locTrg.LoStripY(),
1016 2 : locTrg.GetDeviation(),
1017 2 : locTrg.LoLpt(),
1018 2 : locTrg.LoHpt(),
1019 2 : locTrg.GetTriggerWithoutChamber());
1020 2 : esdTrack.SetLocalTrigger(muonTrack.GetLocalTrigger());
1021 2 : esdTrack.SetChi2MatchTrigger(0.);
1022 2 : esdTrack.SetTriggerX1Pattern(locTrg.GetX1Pattern());
1023 2 : esdTrack.SetTriggerY1Pattern(locTrg.GetY1Pattern());
1024 2 : esdTrack.SetTriggerX2Pattern(locTrg.GetX2Pattern());
1025 2 : esdTrack.SetTriggerY2Pattern(locTrg.GetY2Pattern());
1026 2 : esdTrack.SetTriggerX3Pattern(locTrg.GetX3Pattern());
1027 2 : esdTrack.SetTriggerY3Pattern(locTrg.GetY3Pattern());
1028 2 : esdTrack.SetTriggerX4Pattern(locTrg.GetX4Pattern());
1029 2 : esdTrack.SetTriggerY4Pattern(locTrg.GetY4Pattern());
1030 : UShort_t hitPattern = 0;
1031 2 : esdTrack.SetHitsPatternInTrigChTrk(hitPattern);
1032 2 : if(triggerTrack){
1033 2 : hitPattern = triggerTrack->GetHitsPatternInTrigCh();
1034 2 : esdTrack.SetHitsPatternInTrigCh(hitPattern);
1035 2 : esdTrack.SetThetaXUncorrected(triggerTrack->GetThetax());
1036 2 : esdTrack.SetThetaYUncorrected(triggerTrack->GetThetay());
1037 2 : esdTrack.SetNonBendingCoorUncorrected(triggerTrack->GetX11());
1038 2 : esdTrack.SetBendingCoorUncorrected(triggerTrack->GetY11());
1039 2 : esdTrack.SetZUncorrected(triggerTrack->GetZ11());
1040 2 : }
1041 2 : }
1042 :
1043 : //_____________________________________________________________________________
1044 : void AliMUONESDInterface::MUONToESD(const AliMUONVCluster& cluster, AliESDEvent& esd, const AliMUONVDigitStore* digits)
1045 : {
1046 : /// Transfert data from MUON cluster to ESD event if it does not already exist
1047 : /// Also transfert digits'Id to cluster and digits to ESD if they are provided
1048 :
1049 324 : AliESDMuonCluster *esdCluster = esd.FindMuonCluster(cluster.GetUniqueID());
1050 324 : if (!esdCluster) esdCluster = esd.NewMuonCluster();
1051 0 : else if (!digits || esdCluster->GetNPads() > 0) return;
1052 :
1053 162 : if (digits) {
1054 :
1055 0 : MUONToESD(cluster, *esdCluster, kTRUE);
1056 :
1057 0 : for (Int_t i=0; i<cluster.GetNDigits(); i++) {
1058 0 : AliMUONVDigit* digit = digits->FindObject(cluster.GetDigitId(i));
1059 0 : if (!digit) {
1060 0 : AliErrorClass(Form("digit %u not found", cluster.GetDigitId(i)));
1061 0 : continue;
1062 : }
1063 0 : MUONToESD(*digit, esd);
1064 0 : }
1065 :
1066 0 : } else {
1067 :
1068 162 : MUONToESD(cluster, *esdCluster, kFALSE);
1069 :
1070 : }
1071 :
1072 324 : }
1073 :
1074 : //_____________________________________________________________________________
1075 : void AliMUONESDInterface::MUONToESD(const AliMUONVCluster& cluster, AliESDMuonCluster& esdCluster, Bool_t copyPadsId)
1076 : {
1077 : /// Transfert data from MUON cluster to ESDMuon cluster
1078 : /// Also transfert digits'Is if required
1079 :
1080 162 : esdCluster.Clear("C");
1081 :
1082 162 : esdCluster.SetUniqueID(cluster.GetUniqueID());
1083 162 : esdCluster.SetXYZ(cluster.GetX(), cluster.GetY(), cluster.GetZ());
1084 162 : esdCluster.SetErrXY(cluster.GetErrX(), cluster.GetErrY());
1085 162 : esdCluster.SetCharge(cluster.GetCharge());
1086 162 : esdCluster.SetChi2(cluster.GetChi2());
1087 162 : esdCluster.SetLabel(cluster.GetMCLabel());
1088 :
1089 162 : if (copyPadsId) esdCluster.SetPadsId(cluster.GetNDigits(), cluster.GetDigitsId());
1090 :
1091 162 : }
1092 :
1093 : //_____________________________________________________________________________
1094 : void AliMUONESDInterface::MUONToESD(const AliMUONVDigit& digit, AliESDEvent& esd)
1095 : {
1096 : /// Transfert data from MUON digit to ESD event if it does not already exist
1097 0 : if (esd.FindMuonPad(digit.GetUniqueID())) return;
1098 0 : AliESDMuonPad *esdPad = esd.NewMuonPad();
1099 0 : MUONToESD(digit, *esdPad);
1100 0 : }
1101 :
1102 : //_____________________________________________________________________________
1103 : void AliMUONESDInterface::MUONToESD(const AliMUONVDigit& digit, AliESDMuonPad& esdPad)
1104 : {
1105 : /// Transfert data from MUON digit to ESDMuon pad
1106 0 : esdPad.SetUniqueID(digit.GetUniqueID());
1107 0 : esdPad.SetADC(digit.ADC());
1108 0 : esdPad.SetCharge(digit.Charge());
1109 0 : esdPad.SetCalibrated(digit.IsCalibrated());
1110 0 : esdPad.SetSaturated(digit.IsSaturated());
1111 0 : }
1112 :
1113 : //___________________________________________________________________________
1114 : AliMUONTrack* AliMUONESDInterface::Add(const AliESDMuonTrack& esdTrack, AliMUONVTrackStore& trackStore, Bool_t refit)
1115 : {
1116 : /// Create MUON track from ESDMuon track and add it to the store
1117 : /// Track parameters at each clusters are recomputed or not depending on the flag "refit"
1118 : /// return a pointer to the track into the store (0x0 if the track already exist)
1119 0 : if(trackStore.FindObject(esdTrack.GetUniqueID())) return 0x0;
1120 0 : AliMUONTrack* track = trackStore.Add(AliMUONTrack());
1121 0 : ESDToMUON(esdTrack, *track, refit);
1122 : return track;
1123 0 : }
1124 :
1125 : //___________________________________________________________________________
1126 : void AliMUONESDInterface::Add(const AliESDMuonTrack& esdTrack, AliMUONVTriggerStore& triggerStore)
1127 : {
1128 : /// Create MUON local trigger from ESDMuon track and add it to the store if not already there
1129 0 : if (!triggerStore.FindLocal(esdTrack.LoCircuit())->IsNull()) return;
1130 0 : AliMUONLocalTrigger locTrg;
1131 0 : ESDToMUON(esdTrack, locTrg);
1132 0 : triggerStore.Add(locTrg);
1133 0 : }
1134 :
1135 : //___________________________________________________________________________
1136 : AliMUONVCluster* AliMUONESDInterface::Add(const AliESDMuonCluster& esdCluster, AliMUONVClusterStore& clusterStore)
1137 : {
1138 : /// Create MUON cluster from ESDMuon cluster and add it to the store
1139 : /// return a pointer to the cluster into the store (0x0 if the cluster already exist)
1140 0 : AliMUONVCluster* cluster = clusterStore.Add(esdCluster.GetChamberId(), esdCluster.GetDetElemId(), esdCluster.GetClusterIndex());
1141 0 : if (cluster) ESDToMUON(esdCluster, *cluster);
1142 0 : return cluster;
1143 : }
1144 :
1145 : //___________________________________________________________________________
1146 : AliMUONVDigit* AliMUONESDInterface::Add(const AliESDMuonPad& esdPad, AliMUONVDigitStore& digitStore)
1147 : {
1148 : /// Create MUON digit from ESDMuon digit and add it to the store
1149 : /// return a pointer to the digit into the store (0x0 if the digit already exist)
1150 0 : AliMUONVDigit* digit = digitStore.Add(esdPad.GetDetElemId(), esdPad.GetManuId(), esdPad.GetManuChannel(), esdPad.GetCathode(), AliMUONVDigitStore::kDeny);
1151 0 : if (digit) ESDToMUON(esdPad, *digit);
1152 0 : return digit;
1153 : }
1154 :
|
