]>
Commit | Line | Data |
---|---|---|
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 | /////////////////////////////////////////////////////////////////////////////// | |
19 | // // | |
20 | // class for MUON reconstruction // | |
21 | // // | |
22 | /////////////////////////////////////////////////////////////////////////////// | |
23 | #include <TParticle.h> | |
24 | #include <TArrayF.h> | |
25 | ||
26 | #include "AliRunLoader.h" | |
27 | #include "AliHeader.h" | |
28 | #include "AliGenEventHeader.h" | |
29 | #include "AliESD.h" | |
30 | ||
31 | #include "AliMUONData.h" | |
32 | #include "AliMUONEventReconstructor.h" | |
33 | #include "AliMUONClusterReconstructor.h" | |
34 | #include "AliMUONTriggerDecision.h" | |
35 | #include "AliMUONClusterFinderVS.h" | |
36 | #include "AliMUONTrack.h" | |
37 | #include "AliMUONTrackParam.h" | |
38 | #include "AliMUONTriggerTrack.h" | |
39 | #include "AliESDMuonTrack.h" | |
40 | #include "AliMUONReconstructor.h" | |
41 | ||
42 | ClassImp(AliMUONReconstructor) | |
43 | //_____________________________________________________________________________ | |
44 | AliMUONReconstructor::AliMUONReconstructor() | |
45 | : AliReconstructor() | |
46 | { | |
47 | } | |
48 | //_____________________________________________________________________________ | |
49 | AliMUONReconstructor::~AliMUONReconstructor() | |
50 | { | |
51 | } | |
52 | //_____________________________________________________________________________ | |
53 | void AliMUONReconstructor::Reconstruct(AliRunLoader* runLoader) const | |
54 | { | |
55 | // AliLoader | |
56 | AliLoader* loader = runLoader->GetLoader("MUONLoader"); | |
57 | Int_t nEvents = runLoader->GetNumberOfEvents(); | |
58 | ||
59 | // used local container for each method | |
60 | // passing fLoader as argument, could be avoided ??? | |
61 | AliMUONEventReconstructor* recoEvent = new AliMUONEventReconstructor(loader); | |
62 | AliMUONData* dataEvent = recoEvent->GetMUONData(); | |
63 | ||
64 | AliMUONClusterReconstructor* recoCluster = new AliMUONClusterReconstructor(loader); | |
65 | AliMUONData* dataCluster = recoCluster->GetMUONData(); | |
66 | ||
67 | AliMUONTriggerDecision* trigDec = new AliMUONTriggerDecision(loader,0,dataCluster); | |
68 | // AliMUONData* dataTrig = trigDec->GetMUONData(); | |
69 | ||
70 | ||
71 | for (Int_t i = 0; i < 10; i++) { | |
72 | AliMUONClusterFinderVS *recModel = new AliMUONClusterFinderVS(); | |
73 | recModel->SetGhostChi2Cut(10); | |
74 | recoCluster->SetReconstructionModel(i,recModel); | |
75 | } | |
76 | ||
77 | loader->LoadDigits("READ"); | |
78 | loader->LoadRecPoints("RECREATE"); | |
79 | loader->LoadTracks("RECREATE"); | |
80 | ||
81 | // Loop over events | |
82 | for(Int_t ievent = 0; ievent < nEvents; ievent++) { | |
83 | printf("Event %d\n",ievent); | |
84 | runLoader->GetEvent(ievent); | |
85 | ||
86 | //----------------------- digit2cluster & Digits2Trigger ------------------- | |
87 | if (!loader->TreeR()) loader->MakeRecPointsContainer(); | |
88 | ||
89 | // tracking branch | |
90 | dataCluster->MakeBranch("RC"); | |
91 | dataCluster->SetTreeAddress("D,RC"); | |
92 | recoCluster->Digits2Clusters(); | |
93 | dataCluster->Fill("RC"); | |
94 | ||
95 | // trigger branch | |
96 | dataCluster->MakeBranch("TC"); | |
97 | dataCluster->SetTreeAddress("TC"); | |
98 | trigDec->Trigger2Trigger(); | |
99 | dataCluster->Fill("TC"); | |
100 | ||
101 | loader->WriteRecPoints("OVERWRITE"); | |
102 | ||
103 | //---------------------------- Track & TriggerTrack --------------------- | |
104 | if (!loader->TreeT()) loader->MakeTracksContainer(); | |
105 | ||
106 | // trigger branch | |
107 | dataEvent->MakeBranch("RL"); //trigger track | |
108 | dataEvent->SetTreeAddress("RL"); | |
109 | recoEvent->EventReconstructTrigger(); | |
110 | dataEvent->Fill("RL"); | |
111 | ||
112 | // tracking branch | |
113 | dataEvent->MakeBranch("RT"); //track | |
114 | dataEvent->SetTreeAddress("RT"); | |
115 | recoEvent->EventReconstruct(); | |
116 | dataEvent->Fill("RT"); | |
117 | ||
118 | loader->WriteTracks("OVERWRITE"); | |
119 | ||
120 | //--------------------------- Resetting branches ----------------------- | |
121 | dataCluster->ResetDigits(); | |
122 | dataCluster->ResetRawClusters(); | |
123 | dataCluster->ResetTrigger(); | |
124 | ||
125 | dataEvent->ResetRawClusters(); | |
126 | dataEvent->ResetTrigger(); | |
127 | dataEvent->ResetRecTracks(); | |
128 | dataEvent->ResetRecTriggerTracks(); | |
129 | ||
130 | } | |
131 | loader->UnloadDigits(); | |
132 | loader->UnloadRecPoints(); | |
133 | loader->UnloadTracks(); | |
134 | ||
135 | delete recoCluster; | |
136 | delete recoEvent; | |
137 | delete trigDec; | |
138 | } | |
139 | //_____________________________________________________________________________ | |
140 | void AliMUONReconstructor::FillESD(AliRunLoader* runLoader, AliESD* esd) const | |
141 | { | |
142 | TClonesArray* recTracksArray = 0; | |
143 | TClonesArray* recTrigTracksArray = 0; | |
144 | ||
145 | AliLoader* loader = runLoader->GetLoader("MUONLoader"); | |
146 | loader->LoadTracks("READ"); | |
147 | AliMUONData* muonData = new AliMUONData(loader,"MUON","MUON"); | |
148 | ||
149 | // declaration | |
150 | Int_t iEvent, nPart; | |
151 | Int_t nTrackHits, nPrimary; | |
152 | Double_t fitFmin; | |
153 | TArrayF vertex(3); | |
154 | ||
155 | Double_t bendingSlope, nonBendingSlope, inverseBendingMomentum; | |
156 | Double_t xRec, yRec, zRec, chi2MatchTrigger; | |
157 | Bool_t matchTrigger; | |
158 | ||
159 | // setting pointer for tracks, triggertracks & trackparam at vertex | |
160 | AliMUONTrack* recTrack = 0; | |
161 | AliMUONTrackParam* trackParam = 0; | |
162 | AliMUONTriggerTrack* recTriggerTrack = 0; | |
163 | TParticle* particle = new TParticle(); | |
164 | AliGenEventHeader* header = 0; | |
165 | iEvent = runLoader->GetEventNumber(); | |
166 | runLoader->GetEvent(iEvent); | |
167 | ||
168 | // vertex calculation (maybe it exists already somewhere else) | |
169 | vertex[0] = vertex[1] = vertex[2] = 0.; | |
170 | nPrimary = 0; | |
171 | if ( (header = runLoader->GetHeader()->GenEventHeader()) ) { | |
172 | header->PrimaryVertex(vertex); | |
173 | } else { | |
174 | runLoader->LoadKinematics("READ"); | |
175 | runLoader->TreeK()->GetBranch("Particles")->SetAddress(&particle); | |
176 | nPart = (Int_t)runLoader->TreeK()->GetEntries(); | |
177 | for(Int_t iPart = 0; iPart < nPart; iPart++) { | |
178 | runLoader->TreeK()->GetEvent(iPart); | |
179 | if (particle->GetFirstMother() == -1) { | |
180 | vertex[0] += particle->Vx(); | |
181 | vertex[1] += particle->Vy(); | |
182 | vertex[2] += particle->Vz(); | |
183 | nPrimary++; | |
184 | } | |
185 | if (nPrimary) { | |
186 | vertex[0] /= (double)nPrimary; | |
187 | vertex[1] /= (double)nPrimary; | |
188 | vertex[2] /= (double)nPrimary; | |
189 | } | |
190 | } | |
191 | } | |
192 | // setting ESD MUON class | |
193 | AliESDMuonTrack* theESDTrack = new AliESDMuonTrack() ; | |
194 | ||
195 | //-------------------- trigger tracks------------- | |
196 | Long_t trigPat = 0; | |
197 | muonData->SetTreeAddress("RL"); | |
198 | muonData->GetRecTriggerTracks(); | |
199 | recTrigTracksArray = muonData->RecTriggerTracks(); | |
200 | ||
201 | // ready global trigger pattern from first track | |
202 | if (recTrigTracksArray) | |
203 | recTriggerTrack = (AliMUONTriggerTrack*) recTrigTracksArray->First(); | |
204 | if (recTriggerTrack) trigPat = recTriggerTrack->GetGTPattern(); | |
205 | ||
206 | //printf(">>> Event %d Number of Recconstructed tracks %d \n",iEvent, nrectracks); | |
207 | ||
208 | // -------------------- tracks------------- | |
209 | muonData->SetTreeAddress("RT"); | |
210 | muonData->GetRecTracks(); | |
211 | recTracksArray = muonData->RecTracks(); | |
212 | ||
213 | Int_t nRecTracks = 0; | |
214 | if (recTracksArray) | |
215 | nRecTracks = (Int_t) recTracksArray->GetEntriesFast(); // | |
216 | ||
217 | // loop over tracks | |
218 | for (Int_t iRecTracks = 0; iRecTracks < nRecTracks; iRecTracks++) { | |
219 | ||
220 | // reading info from tracks | |
221 | recTrack = (AliMUONTrack*) recTracksArray->At(iRecTracks); | |
222 | ||
223 | trackParam = recTrack->GetTrackParamAtVertex(); | |
224 | ||
225 | bendingSlope = trackParam->GetBendingSlope(); | |
226 | nonBendingSlope = trackParam->GetNonBendingSlope(); | |
227 | inverseBendingMomentum = trackParam->GetInverseBendingMomentum(); | |
228 | xRec = trackParam->GetNonBendingCoor(); | |
229 | yRec = trackParam->GetBendingCoor(); | |
230 | zRec = trackParam->GetZ(); | |
231 | ||
232 | nTrackHits = recTrack->GetNTrackHits(); | |
233 | fitFmin = recTrack->GetFitFMin(); | |
234 | matchTrigger = recTrack->GetMatchTrigger(); | |
235 | chi2MatchTrigger = recTrack->GetChi2MatchTrigger(); | |
236 | ||
237 | // setting data member of ESD MUON | |
238 | theESDTrack->SetInverseBendingMomentum(inverseBendingMomentum); | |
239 | theESDTrack->SetThetaX(TMath::ATan(nonBendingSlope)); | |
240 | theESDTrack->SetThetaY(TMath::ATan(bendingSlope)); | |
241 | theESDTrack->SetZ(vertex[2]); | |
242 | theESDTrack->SetBendingCoor(vertex[1]); // calculate vertex at ESD or Tracking level ? | |
243 | theESDTrack->SetNonBendingCoor(vertex[0]); | |
244 | theESDTrack->SetChi2(fitFmin); | |
245 | theESDTrack->SetNHit(nTrackHits); | |
246 | theESDTrack->SetMatchTrigger(matchTrigger); | |
247 | theESDTrack->SetChi2MatchTrigger(chi2MatchTrigger); | |
248 | ||
249 | // storing ESD MUON Track into ESD Event | |
250 | if (nRecTracks != 0) | |
251 | esd->AddMuonTrack(theESDTrack); | |
252 | } // end loop tracks | |
253 | ||
254 | // add global trigger pattern | |
255 | if (nRecTracks != 0) | |
256 | esd->SetTrigger(trigPat); | |
257 | ||
258 | // reset muondata | |
259 | muonData->ResetRecTracks(); | |
260 | muonData->ResetRecTriggerTracks(); | |
261 | ||
262 | //} // end loop on event | |
263 | loader->UnloadTracks(); | |
264 | if (!header) | |
265 | runLoader->UnloadKinematics(); | |
266 | delete theESDTrack; | |
267 | delete muonData; | |
268 | // delete particle; | |
269 | } |