1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
17 //-----------------------------
18 // Class AliMUONReconstructor
19 //-----------------------------
21 // MUON track reconstruction
23 #include "AliMUONReconstructor.h"
25 #include "AliMUONConstants.h"
26 #include "AliMUONCalibrationData.h"
27 #include "AliMUONClusterFinderAZ.h"
28 #include "AliMUONClusterReconstructor.h"
29 #include "AliMUONRecData.h"
30 #include "AliMUONDigitCalibrator.h"
31 #include "AliMUONEventRecoCombi.h"
32 #include "AliMUONDigitMaker.h"
33 #include "AliMUONTrack.h"
34 #include "AliMUONTrackParam.h"
35 #include "AliMUONTrackExtrap.h"
36 #include "AliMUONTracker.h"
37 #include "AliMUONVTrackReconstructor.h"
38 #include "AliMUONTrackReconstructor.h"
39 #include "AliMUONTrackReconstructorK.h"
40 #include "AliMUONTriggerTrack.h"
41 #include "AliMUONTriggerCircuit.h"
42 #include "AliMUONTriggerCrateStore.h"
43 #include "AliMUONSegFactory.h"
44 #include "AliMUONSegmentation.h"
45 #include "AliMUONPreClusterFinder.h"
46 #include "AliMUONClusterFinderCOG.h"
47 #include "AliMUONClusterFinderSimpleFit.h"
48 #include "AliMUONClusterFinderMLEM.h"
51 #include "AliESDMuonTrack.h"
53 #include "AliRawReader.h"
54 #include "AliRunLoader.h"
55 #include "AliCDBManager.h"
58 #include "TStopwatch.h"
59 #include "Riostream.h"
62 ClassImp(AliMUONReconstructor)
65 //_____________________________________________________________________________
66 AliMUONReconstructor::AliMUONReconstructor()
68 fDigitMaker(new AliMUONDigitMaker()),
69 fCalibrationData(0x0),
70 fCrateManager(new AliMUONTriggerCrateStore()),
71 fTriggerCircuit(new TClonesArray("AliMUONTriggerCircuit", 234)),
72 fTransformer(new AliMUONGeometryTransformer(kTRUE)),
74 fMUONData(new AliMUONRecData(0x0,"MUON","MUON"))
76 /// Default constructor
80 fCrateManager->ReadFromFile();
83 fDigitMaker->SetCrateManager(fCrateManager);
86 fTransformer->ReadGeometryData("volpath.dat", "geometry.root");
88 // create segmentation and pass it to EventRecoCombi
89 AliMUONSegFactory factory(fTransformer);
90 fSegmentation = factory.CreateSegmentation();
91 AliMUONEventRecoCombi::Instance(fSegmentation);
94 for (Int_t i = 0; i < AliMUONConstants::NTriggerCircuit(); i++) {
95 AliMUONTriggerCircuit* c = new AliMUONTriggerCircuit();
96 c->SetTransformer(fTransformer);
97 c->Init(i,*fCrateManager);
98 TClonesArray& circuit = *fTriggerCircuit;
99 new(circuit[circuit.GetEntriesFast()])AliMUONTriggerCircuit(*c);
106 //_____________________________________________________________________________
107 AliMUONReconstructor::~AliMUONReconstructor()
112 delete fCalibrationData;
114 delete fCrateManager;
115 delete fTriggerCircuit;
117 delete fSegmentation;
121 //_____________________________________________________________________________
123 AliMUONReconstructor::GetCalibrationTask() const
125 /// Create the calibration task(s).
127 //const AliRun* run = fRunLoader->GetAliRun();
128 //Int_t runNumber = run->GetRunNumber();
129 Int_t runNumber = AliCDBManager::Instance()->GetRun();
130 AliInfo("Calibration will occur.");
132 fCalibrationData = new AliMUONCalibrationData(runNumber);
133 if ( !fCalibrationData->IsValid() )
135 AliError("Could not retrieve calibrations !");
136 delete fCalibrationData;
137 fCalibrationData = 0x0;
140 // Check that we get all the calibrations we'll need
141 if ( !fCalibrationData->Pedestals() ||
142 !fCalibrationData->Gains() ||
143 !fCalibrationData->HV() )
145 AliFatal("Could not access all required calibration data");
147 TTask* calibration = new TTask("MUONCalibrator","MUON Digit calibrator");
149 TString opt(GetOption());
151 Bool_t statusMap(kTRUE);
153 if ( strstr(opt,"NOSTATUSMAP") )
155 AliWarning("Disconnecting status map : SHOULD BE USED FOR DEBUG ONLY. NOT FOR PRODUCTION !!!");
158 calibration->Add(new AliMUONDigitCalibrator(fMUONData,fCalibrationData,statusMap));
162 //_____________________________________________________________________________
163 AliMUONClusterReconstructor*
164 AliMUONReconstructor::CreateClusterReconstructor() const
166 /// Create cluster reconstructor
168 AliMUONVClusterFinder* clusterFinder(0x0);
170 TString opt(GetOption());
173 if ( strstr(opt,"PRECLUSTER") )
175 clusterFinder = new AliMUONPreClusterFinder;
177 else if ( strstr(opt,"COG") )
179 clusterFinder = new AliMUONClusterFinderCOG;
181 else if ( strstr(opt,"SIMPLEFIT") )
183 clusterFinder = new AliMUONClusterFinderSimpleFit;
185 else if ( strstr(opt,"MLEM:DRAW") )
187 clusterFinder = new AliMUONClusterFinderMLEM(kTRUE);
189 else if ( strstr(opt,"MLEM") )
191 clusterFinder = new AliMUONClusterFinderMLEM(kFALSE);
196 AliInfo(Form("Will use %s for clusterizing",clusterFinder->ClassName()));
199 AliMUONClusterReconstructor* clusterReco =
200 new AliMUONClusterReconstructor(fMUONData,clusterFinder,fTransformer);
204 //_____________________________________________________________________________
205 void AliMUONReconstructor::Reconstruct(AliRunLoader* runLoader) const
210 AliLoader* loader = runLoader->GetLoader("MUONLoader");
211 Int_t nEvents = runLoader->GetNumberOfEvents();
212 Int_t evtNumber = runLoader->GetEventNumber();
214 fMUONData->SetLoader(loader);
216 // passing loader as argument.
217 AliMUONVTrackReconstructor* recoEvent;
218 if (strstr(GetOption(),"Original")) recoEvent = new AliMUONTrackReconstructor(fMUONData);
219 else if (strstr(GetOption(),"Combi")) recoEvent = new AliMUONTrackReconstructorK(fMUONData,"Combi");
220 else recoEvent = new AliMUONTrackReconstructorK(fMUONData,"Kalman");
222 recoEvent->SetTriggerCircuit(fTriggerCircuit);
224 AliMUONClusterReconstructor* recoCluster = CreateClusterReconstructor();
226 AliMUONClusterFinderVS *recModel = recoCluster->GetRecoModel();
228 if (!strstr(GetOption(),"VS")) {
229 recModel = (AliMUONClusterFinderVS*) new AliMUONClusterFinderAZ();
230 recoCluster->SetRecoModel(recModel);
232 recModel->SetGhostChi2Cut(10);
233 recModel->SetEventNumber(evtNumber);
235 loader->LoadDigits("READ");
236 loader->LoadRecPoints("RECREATE");
237 loader->LoadTracks("RECREATE");
239 TTask* calibration = GetCalibrationTask();
241 Int_t chBeg = (strstr(GetOption(),"Combi") ? 6 : 0);
244 for(Int_t ievent = 0; ievent < nEvents; ievent++) {
246 AliDebug(1,Form("Event %d",ievent));
248 runLoader->GetEvent(ievent);
250 //----------------------- digit2cluster & Trigger2Trigger -------------------
251 if (!loader->TreeR()) loader->MakeRecPointsContainer();
254 if (!strstr(GetOption(),"Combi")) {
255 fMUONData->MakeBranch("RC");
256 fMUONData->SetTreeAddress("D,RC");
258 fMUONData->SetTreeAddress("D");
259 fMUONData->SetTreeAddress("RCC");
261 // Important for avoiding a memory leak when reading digits ( to be investigated more in detail)
262 // In any case the reading of GLT is needed for the Trigger2Tigger method below
263 fMUONData->SetTreeAddress("GLT");
265 fMUONData->GetDigits();
269 calibration->ExecuteTask();
272 recoCluster->Digits2Clusters(chBeg);
274 if (strstr(GetOption(),"Combi")) {
275 // Combined cluster / track finder
276 AliMUONEventRecoCombi::Instance()->FillEvent(fMUONData, (AliMUONClusterFinderAZ*)recModel);
277 ((AliMUONClusterFinderAZ*) recModel)->SetReco(2);
279 else fMUONData->Fill("RC");
282 fMUONData->MakeBranch("TC");
283 fMUONData->SetTreeAddress("TC");
284 recoCluster->Trigger2Trigger();
285 fMUONData->Fill("TC");
287 //AZ loader->WriteRecPoints("OVERWRITE");
289 //---------------------------- Track & TriggerTrack ---------------------
290 if (!loader->TreeT()) loader->MakeTracksContainer();
293 fMUONData->MakeBranch("RL"); //trigger track
294 fMUONData->SetTreeAddress("RL");
295 recoEvent->EventReconstructTrigger();
296 fMUONData->Fill("RL");
299 fMUONData->MakeBranch("RT"); //track
300 fMUONData->SetTreeAddress("RT");
301 recoEvent->EventReconstruct();
302 fMUONData->Fill("RT");
304 loader->WriteTracks("OVERWRITE");
306 if (strstr(GetOption(),"Combi")) {
307 // Combined cluster / track
308 ((AliMUONClusterFinderAZ*) recModel)->SetReco(1);
309 fMUONData->MakeBranch("RC");
310 fMUONData->SetTreeAddress("RC");
311 AliMUONEventRecoCombi::Instance()->FillRecP(fMUONData, (AliMUONTrackReconstructorK*)recoEvent);
312 fMUONData->Fill("RC");
314 loader->WriteRecPoints("OVERWRITE");
316 //--------------------------- Resetting branches -----------------------
317 fMUONData->ResetDigits();
318 fMUONData->ResetRawClusters();
319 fMUONData->ResetTrigger();
320 fMUONData->ResetRecTracks();
321 fMUONData->ResetRecTriggerTracks();
324 loader->UnloadDigits();
325 loader->UnloadRecPoints();
326 loader->UnloadTracks();
333 //_____________________________________________________________________________
334 void AliMUONReconstructor::Reconstruct(AliRunLoader* runLoader,
335 AliRawReader* rawReader) const
341 AliLoader* loader = runLoader->GetLoader("MUONLoader");
342 Int_t evtNumber = runLoader->GetEventNumber();
344 fMUONData->SetLoader(loader);
346 // passing loader as argument.
347 fDigitMaker->SetMUONData(fMUONData);
349 // disable trigger rawdata reading
350 if (strstr(GetOption(),"TriggerDisable"))
351 fDigitMaker->DisableTrigger();
353 AliMUONClusterReconstructor* recoCluster = CreateClusterReconstructor();
355 AliMUONClusterFinderVS *recModel = recoCluster->GetRecoModel();
357 if (!strstr(GetOption(),"VS"))
359 recModel = (AliMUONClusterFinderVS*) new AliMUONClusterFinderAZ();
360 recoCluster->SetRecoModel(recModel);
362 recModel->SetGhostChi2Cut(10);
363 recModel->SetEventNumber(evtNumber);
365 TTask* calibration = GetCalibrationTask();
367 loader->LoadRecPoints("RECREATE");
369 // Digits are not stored on disk and created on flight from rawdata.
370 // In order to write digits on disk the following line should be uncommented
371 // loader->LoadDigits("RECREATE");
376 TStopwatch totalTimer;
378 TStopwatch calibTimer;
379 TStopwatch clusterTimer;
381 rawTimer.Start(kTRUE); rawTimer.Stop();
382 calibTimer.Start(kTRUE); calibTimer.Stop();
383 clusterTimer.Start(kTRUE); clusterTimer.Stop();
385 totalTimer.Start(kTRUE);
387 while (rawReader->NextEvent())
389 AliDebug(1,Form("Event %d",iEvent));
391 runLoader->GetEvent(iEvent++);
393 //----------------------- raw2digits & raw2trigger-------------------
394 // if (!loader->TreeD())
396 // AliDebug(1,Form("Making Digit Container for event %d",iEvent));
397 // loader->MakeDigitsContainer();
399 // Write digits from raw data on disk
400 if (strstr(GetOption(),"SAVEDIGITS")) {
401 if (!loader->TreeD())fMUONData->GetLoader()->MakeDigitsContainer();
402 fMUONData->MakeBranch("D,GLT");
403 fMUONData->SetTreeAddress("D,GLT");
404 AliInfo("Digits from raw data will be stored.");
407 fMUONData->SetDataContainer("D, GLT");
409 rawTimer.Start(kFALSE);
410 fDigitMaker->Raw2Digits(rawReader);
415 calibTimer.Start(kFALSE);
416 calibration->ExecuteTask();
420 // Write digits from raw data on disk
421 if (strstr(GetOption(),"SAVEDIGITS")) {
422 fMUONData->Fill("D,GLT");
423 loader->WriteDigits("OVERWRITE");
425 //----------------------- digit2cluster & Trigger2Trigger -------------------
426 clusterTimer.Start(kFALSE);
428 if (!loader->TreeR()) loader->MakeRecPointsContainer();
431 fMUONData->MakeBranch("RC");
432 fMUONData->SetTreeAddress("RC");
433 recoCluster->Digits2Clusters();
434 fMUONData->Fill("RC");
437 fMUONData->MakeBranch("TC");
438 fMUONData->SetTreeAddress("TC");
439 fMUONData->Fill("TC");
441 loader->WriteRecPoints("OVERWRITE");
446 //--------------------------- Resetting branches -----------------------
447 fMUONData->ResetDigits();
448 fMUONData->ResetRawClusters();
449 fMUONData->ResetTrigger();
454 loader->UnloadRecPoints();
455 loader->UnloadDigits();
460 AliInfo(Form("Execution time for converting RAW data to digits in MUON : R:%.2fs C:%.2fs",
461 rawTimer.RealTime(),rawTimer.CpuTime()));
462 AliInfo(Form("Execution time for calibrating MUON : R:%.2fs C:%.2fs",
463 calibTimer.RealTime(),calibTimer.CpuTime()));
464 AliInfo(Form("Execution time for clusterizing MUON : R:%.2fs C:%.2fs",
465 clusterTimer.RealTime(),clusterTimer.CpuTime()));
466 AliInfo(Form("Total Execution time for Reconstruct(from raw) MUON : R:%.2fs C:%.2fs",
467 totalTimer.RealTime(),totalTimer.CpuTime()));
470 //_____________________________________________________________________________
471 void AliMUONReconstructor::FillESD(AliRunLoader* runLoader, AliESD* esd) const
476 TClonesArray* recTracksArray = 0;
477 TClonesArray* recTrigTracksArray = 0;
479 AliLoader* loader = runLoader->GetLoader("MUONLoader");
480 loader->LoadTracks("READ");
481 fMUONData->SetLoader(loader);
484 Int_t iEvent;// nPart;
485 Int_t nTrackHits;// nPrimary;
486 Double_t fitFmin, chi2MatchTrigger;
487 Double_t xRec, yRec, zRec, bendingSlope, nonBendingSlope, inverseBendingMomentum;
488 Double_t xVtx, yVtx, zVtx, bendingSlopeAtVtx, nonBendingSlopeAtVtx, inverseBendingMomentumAtVtx;
490 UShort_t hitsPatternInTrigCh;
492 // setting pointer for tracks, triggertracks & trackparam at vertex
493 AliMUONTrack* recTrack = 0;
494 AliMUONTrackParam trackParamAtVtx;
495 AliMUONTriggerTrack* recTriggerTrack = 0;
497 iEvent = runLoader->GetEventNumber();
498 runLoader->GetEvent(iEvent);
501 Double_t vertex[3] = {0};
502 const AliESDVertex *esdVert = esd->GetVertex();
503 if (esdVert->GetNContributors()) {
504 esdVert->GetXYZ(vertex);
505 AliDebug(1, "find vertex\n");
507 // setting ESD MUON class
508 AliESDMuonTrack* theESDTrack = new AliESDMuonTrack() ;
510 //-------------------- trigger tracks-------------
512 fMUONData->SetTreeAddress("RL");
513 fMUONData->GetRecTriggerTracks();
514 recTrigTracksArray = fMUONData->RecTriggerTracks();
516 // ready global trigger pattern from first track
517 if (recTrigTracksArray)
518 recTriggerTrack = (AliMUONTriggerTrack*) recTrigTracksArray->First();
519 if (recTriggerTrack) trigPat = recTriggerTrack->GetGTPattern();
521 //printf(">>> Event %d Number of Recconstructed tracks %d \n",iEvent, nrectracks);
523 // -------------------- tracks-------------
524 fMUONData->SetTreeAddress("RT");
525 fMUONData->GetRecTracks();
526 recTracksArray = fMUONData->RecTracks();
528 Int_t nRecTracks = 0;
530 nRecTracks = (Int_t) recTracksArray->GetEntriesFast(); //
533 for (Int_t iRecTracks = 0; iRecTracks < nRecTracks; iRecTracks++) {
535 // reading info from tracks
536 recTrack = (AliMUONTrack*) recTracksArray->At(iRecTracks);
537 AliMUONTrackParam *trackParam = (AliMUONTrackParam*) (recTrack->GetTrackParamAtHit())->First();
538 trackParamAtVtx = *trackParam;
540 // extrapolate to the vertex if available, else to (0,0,0)
541 if (esdVert->GetNContributors())
542 AliMUONTrackExtrap::ExtrapToVertex(&trackParamAtVtx, vertex[0],vertex[1],vertex[2]);
544 AliMUONTrackExtrap::ExtrapToVertex(&trackParamAtVtx, 0.,0.,0.);
546 // Track parameters at first station
547 bendingSlope = trackParam->GetBendingSlope();
548 nonBendingSlope = trackParam->GetNonBendingSlope();
549 inverseBendingMomentum = trackParam->GetInverseBendingMomentum();
550 xRec = trackParam->GetNonBendingCoor();
551 yRec = trackParam->GetBendingCoor();
552 zRec = trackParam->GetZ();
554 // Track parameters at vertex
555 bendingSlopeAtVtx = trackParamAtVtx.GetBendingSlope();
556 nonBendingSlopeAtVtx = trackParamAtVtx.GetNonBendingSlope();
557 inverseBendingMomentumAtVtx = trackParamAtVtx.GetInverseBendingMomentum();
558 xVtx = trackParamAtVtx.GetNonBendingCoor();
559 yVtx = trackParamAtVtx.GetBendingCoor();
560 zVtx = trackParamAtVtx.GetZ();
563 nTrackHits = recTrack->GetNTrackHits();
564 fitFmin = recTrack->GetFitFMin();
565 localTrigger = recTrack->GetLocalTrigger();
566 chi2MatchTrigger = recTrack->GetChi2MatchTrigger();
567 hitsPatternInTrigCh = recTrack->GetHitsPatternInTrigCh();
569 // setting data member of ESD MUON
571 theESDTrack->SetInverseBendingMomentumUncorrected(inverseBendingMomentum);
572 theESDTrack->SetThetaXUncorrected(TMath::ATan(nonBendingSlope));
573 theESDTrack->SetThetaYUncorrected(TMath::ATan(bendingSlope));
574 theESDTrack->SetZUncorrected(zRec);
575 theESDTrack->SetBendingCoorUncorrected(yRec);
576 theESDTrack->SetNonBendingCoorUncorrected(xRec);
578 theESDTrack->SetInverseBendingMomentum(inverseBendingMomentumAtVtx);
579 theESDTrack->SetThetaX(TMath::ATan(nonBendingSlopeAtVtx));
580 theESDTrack->SetThetaY(TMath::ATan(bendingSlopeAtVtx));
581 theESDTrack->SetZ(zVtx);
582 theESDTrack->SetBendingCoor(yVtx);
583 theESDTrack->SetNonBendingCoor(xVtx);
585 theESDTrack->SetChi2(fitFmin);
586 theESDTrack->SetNHit(nTrackHits);
587 theESDTrack->SetLocalTrigger(localTrigger);
588 theESDTrack->SetChi2MatchTrigger(chi2MatchTrigger);
589 theESDTrack->SetHitsPatternInTrigCh(hitsPatternInTrigCh);
591 // storing ESD MUON Track into ESD Event
593 esd->AddMuonTrack(theESDTrack);
597 fMUONData->ResetRecTracks();
598 fMUONData->ResetRecTriggerTracks();
600 //} // end loop on event
601 loader->UnloadTracks();
606 //_____________________________________________________________________________
607 void AliMUONReconstructor::FillESD(AliRunLoader* runLoader, AliRawReader* /*rawReader*/, AliESD* esd) const
612 // don't need rawReader ???
613 FillESD(runLoader, esd);
616 //_____________________________________________________________________________
617 AliTracker* AliMUONReconstructor::CreateTracker(AliRunLoader* runLoader) const
619 /// create tracker for MUON
620 /// go into the tracking framework finally (Ch.F)
622 AliLoader* loader = runLoader->GetLoader("MUONLoader");
624 fMUONData->SetLoader(loader);
626 AliMUONTracker* tracker = new AliMUONTracker();
627 tracker->SetMUONData(fMUONData);
628 tracker->SetTriggerCircuit(fTriggerCircuit);
629 tracker->SetOption(GetOption());