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 ///////////////////////////////////////////////////////////////////////////////
19 // class for MUON reconstruction //
21 ///////////////////////////////////////////////////////////////////////////////
22 #include <TParticle.h>
25 #include "AliRunLoader.h"
26 #include "AliHeader.h"
27 #include "AliGenEventHeader.h"
29 #include "AliMUONReconstructor.h"
31 #include "AliMUONData.h"
32 #include "AliMUONTrackReconstructor.h"
33 #include "AliMUONClusterReconstructor.h"
34 #include "AliMUONClusterFinderVS.h"
35 #include "AliMUONTrack.h"
36 #include "AliMUONTrackParam.h"
37 #include "AliMUONTriggerTrack.h"
38 #include "AliESDMuonTrack.h"
39 #include "AliMUONRawData.h"
41 #include "AliRawReader.h"
44 ClassImp(AliMUONReconstructor)
45 //_____________________________________________________________________________
46 AliMUONReconstructor::AliMUONReconstructor()
50 //_____________________________________________________________________________
51 AliMUONReconstructor::~AliMUONReconstructor()
54 //_____________________________________________________________________________
55 void AliMUONReconstructor::Reconstruct(AliRunLoader* runLoader) const
58 AliLoader* loader = runLoader->GetLoader("MUONLoader");
59 Int_t nEvents = runLoader->GetNumberOfEvents();
61 // used local container for each method
62 // passing fLoader as argument, could be avoided ???
63 AliMUONTrackReconstructor* recoEvent = new AliMUONTrackReconstructor(loader);
64 AliMUONData* dataEvent = recoEvent->GetMUONData();
66 AliMUONClusterReconstructor* recoCluster = new AliMUONClusterReconstructor(loader);
67 AliMUONData* dataCluster = recoCluster->GetMUONData();
68 AliMUONClusterFinderVS *recModel = recoCluster->GetRecoModel();
69 recModel->SetGhostChi2Cut(10);
71 loader->LoadDigits("READ");
72 loader->LoadRecPoints("RECREATE");
73 loader->LoadTracks("RECREATE");
76 for(Int_t ievent = 0; ievent < nEvents; ievent++) {
77 printf("Event %d\n",ievent);
78 runLoader->GetEvent(ievent);
80 //----------------------- digit2cluster & Trigger2Trigger -------------------
81 if (!loader->TreeR()) loader->MakeRecPointsContainer();
84 dataCluster->MakeBranch("RC");
85 dataCluster->SetTreeAddress("D,RC");
86 recoCluster->Digits2Clusters();
87 dataCluster->Fill("RC");
90 dataCluster->MakeBranch("TC");
91 dataCluster->SetTreeAddress("TC");
92 recoCluster->Trigger2Trigger();
93 dataCluster->Fill("TC");
95 loader->WriteRecPoints("OVERWRITE");
97 //---------------------------- Track & TriggerTrack ---------------------
98 if (!loader->TreeT()) loader->MakeTracksContainer();
101 dataEvent->MakeBranch("RL"); //trigger track
102 dataEvent->SetTreeAddress("RL");
103 recoEvent->EventReconstructTrigger();
104 dataEvent->Fill("RL");
107 dataEvent->MakeBranch("RT"); //track
108 dataEvent->SetTreeAddress("RT");
109 recoEvent->EventReconstruct();
110 dataEvent->Fill("RT");
112 loader->WriteTracks("OVERWRITE");
114 //--------------------------- Resetting branches -----------------------
115 dataCluster->ResetDigits();
116 dataCluster->ResetRawClusters();
117 dataCluster->ResetTrigger();
119 dataEvent->ResetRawClusters();
120 dataEvent->ResetTrigger();
121 dataEvent->ResetRecTracks();
122 dataEvent->ResetRecTriggerTracks();
125 loader->UnloadDigits();
126 loader->UnloadRecPoints();
127 loader->UnloadTracks();
133 //_____________________________________________________________________________
134 void AliMUONReconstructor::Reconstruct(AliRunLoader* runLoader, AliRawReader* rawReader) const
137 AliLoader* loader = runLoader->GetLoader("MUONLoader");
139 // used local container for each method
140 // passing fLoader as argument, could be avoided ???
141 AliMUONTrackReconstructor* recoEvent = new AliMUONTrackReconstructor(loader);
142 AliMUONData* dataEvent = recoEvent->GetMUONData();
144 AliMUONRawData* rawData = new AliMUONRawData(loader);
145 AliMUONData* dataCluster = rawData->GetMUONData();
147 AliMUONClusterReconstructor* recoCluster = new AliMUONClusterReconstructor(loader, dataCluster);
148 AliMUONClusterFinderVS *recModel = recoCluster->GetRecoModel();
149 recModel->SetGhostChi2Cut(10);
151 loader->LoadRecPoints("RECREATE");
152 loader->LoadTracks("RECREATE");
157 while (rawReader->NextEvent()) {
158 printf("Event %d\n",iEvent);
159 runLoader->GetEvent(iEvent++);
160 rawData->Raw2Digits(rawReader);
162 //----------------------- digit2cluster & Trigger2Trigger -------------------
163 if (!loader->TreeR()) loader->MakeRecPointsContainer();
166 dataCluster->MakeBranch("RC");
167 dataCluster->SetTreeAddress("D,RC");
168 recoCluster->Digits2Clusters();
169 dataCluster->Fill("RC");
172 dataCluster->MakeBranch("TC");
173 dataCluster->SetTreeAddress("TC");
174 recoCluster->Trigger2Trigger();
175 dataCluster->Fill("TC");
177 loader->WriteRecPoints("OVERWRITE");
179 //---------------------------- Track & TriggerTrack ---------------------
180 if (!loader->TreeT()) loader->MakeTracksContainer();
183 dataEvent->MakeBranch("RL"); //trigger track
184 dataEvent->SetTreeAddress("RL");
185 recoEvent->EventReconstructTrigger();
186 dataEvent->Fill("RL");
189 dataEvent->MakeBranch("RT"); //track
190 dataEvent->SetTreeAddress("RT");
191 recoEvent->EventReconstruct();
192 dataEvent->Fill("RT");
194 loader->WriteTracks("OVERWRITE");
196 //--------------------------- Resetting branches -----------------------
197 dataCluster->ResetDigits();
198 dataCluster->ResetRawClusters();
199 dataCluster->ResetTrigger();
201 dataEvent->ResetRawClusters();
202 dataEvent->ResetTrigger();
203 dataEvent->ResetRecTracks();
204 dataEvent->ResetRecTriggerTracks();
207 loader->UnloadRecPoints();
208 loader->UnloadTracks();
214 //_____________________________________________________________________________
215 void AliMUONReconstructor::FillESD(AliRunLoader* runLoader, AliESD* esd) const
217 TClonesArray* recTracksArray = 0;
218 TClonesArray* recTrigTracksArray = 0;
220 AliLoader* loader = runLoader->GetLoader("MUONLoader");
221 loader->LoadTracks("READ");
222 AliMUONData* muonData = new AliMUONData(loader,"MUON","MUON");
226 Int_t nTrackHits, nPrimary;
230 Double_t bendingSlope, nonBendingSlope, inverseBendingMomentum;
231 Double_t xRec, yRec, zRec, chi2MatchTrigger;
234 // setting pointer for tracks, triggertracks & trackparam at vertex
235 AliMUONTrack* recTrack = 0;
236 AliMUONTrackParam* trackParam = 0;
237 AliMUONTriggerTrack* recTriggerTrack = 0;
238 TParticle* particle = new TParticle();
239 AliGenEventHeader* header = 0;
240 iEvent = runLoader->GetEventNumber();
241 runLoader->GetEvent(iEvent);
243 // vertex calculation (maybe it exists already somewhere else)
244 vertex[0] = vertex[1] = vertex[2] = 0.;
246 if ( (header = runLoader->GetHeader()->GenEventHeader()) ) {
247 header->PrimaryVertex(vertex);
249 runLoader->LoadKinematics("READ");
250 runLoader->TreeK()->GetBranch("Particles")->SetAddress(&particle);
251 nPart = (Int_t)runLoader->TreeK()->GetEntries();
252 for(Int_t iPart = 0; iPart < nPart; iPart++) {
253 runLoader->TreeK()->GetEvent(iPart);
254 if (particle->GetFirstMother() == -1) {
255 vertex[0] += particle->Vx();
256 vertex[1] += particle->Vy();
257 vertex[2] += particle->Vz();
261 vertex[0] /= (double)nPrimary;
262 vertex[1] /= (double)nPrimary;
263 vertex[2] /= (double)nPrimary;
267 // setting ESD MUON class
268 AliESDMuonTrack* theESDTrack = new AliESDMuonTrack() ;
270 //-------------------- trigger tracks-------------
272 muonData->SetTreeAddress("RL");
273 muonData->GetRecTriggerTracks();
274 recTrigTracksArray = muonData->RecTriggerTracks();
276 // ready global trigger pattern from first track
277 if (recTrigTracksArray)
278 recTriggerTrack = (AliMUONTriggerTrack*) recTrigTracksArray->First();
279 if (recTriggerTrack) trigPat = recTriggerTrack->GetGTPattern();
281 //printf(">>> Event %d Number of Recconstructed tracks %d \n",iEvent, nrectracks);
283 // -------------------- tracks-------------
284 muonData->SetTreeAddress("RT");
285 muonData->GetRecTracks();
286 recTracksArray = muonData->RecTracks();
288 Int_t nRecTracks = 0;
290 nRecTracks = (Int_t) recTracksArray->GetEntriesFast(); //
293 for (Int_t iRecTracks = 0; iRecTracks < nRecTracks; iRecTracks++) {
295 // reading info from tracks
296 recTrack = (AliMUONTrack*) recTracksArray->At(iRecTracks);
298 trackParam = (AliMUONTrackParam*) (recTrack->GetTrackParamAtHit())->First();
299 trackParam->ExtrapToVertex(vertex[0],vertex[1],vertex[2]);
301 bendingSlope = trackParam->GetBendingSlope();
302 nonBendingSlope = trackParam->GetNonBendingSlope();
303 inverseBendingMomentum = trackParam->GetInverseBendingMomentum();
304 xRec = trackParam->GetNonBendingCoor();
305 yRec = trackParam->GetBendingCoor();
306 zRec = trackParam->GetZ();
308 nTrackHits = recTrack->GetNTrackHits();
309 fitFmin = recTrack->GetFitFMin();
310 matchTrigger = recTrack->GetMatchTrigger();
311 chi2MatchTrigger = recTrack->GetChi2MatchTrigger();
313 // setting data member of ESD MUON
314 theESDTrack->SetInverseBendingMomentum(inverseBendingMomentum);
315 theESDTrack->SetThetaX(TMath::ATan(nonBendingSlope));
316 theESDTrack->SetThetaY(TMath::ATan(bendingSlope));
317 theESDTrack->SetZ(zRec);
318 theESDTrack->SetBendingCoor(yRec); // calculate vertex at ESD or Tracking level ?
319 theESDTrack->SetNonBendingCoor(xRec);
320 theESDTrack->SetChi2(fitFmin);
321 theESDTrack->SetNHit(nTrackHits);
322 theESDTrack->SetMatchTrigger(matchTrigger);
323 theESDTrack->SetChi2MatchTrigger(chi2MatchTrigger);
325 // storing ESD MUON Track into ESD Event
327 esd->AddMuonTrack(theESDTrack);
330 // add global trigger pattern
332 esd->SetTrigger(trigPat);
335 muonData->ResetRecTracks();
336 muonData->ResetRecTriggerTracks();
338 //} // end loop on event
339 loader->UnloadTracks();
341 runLoader->UnloadKinematics();