Adding new macros in loadmacros.C
[u/mrichter/AliRoot.git] / MUON / MUONRecoCheck.C
CommitLineData
b8dc484b 1/**************************************************************************
48217459 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**************************************************************************/
b8dc484b 15
e54bf126 16// $Id$
17
18/// \ingroup macros
19/// \file MUONRecoCheck.C
20/// \brief Utility macro to check the muon reconstruction.
21///
22/// Reconstructed tracks are compared to reference tracks. The reference tracks
23/// are built from AliTrackReference for the hit in chamber (0..9) and from
24/// kinematics (TreeK) for the vertex parameters.
25///
26/// \author Jean-Pierre Cussonneau, Subatech
27
b8dc484b 28// ROOT includes
29#include "TClonesArray.h"
30#include "TH1.h"
b8dc484b 31#include "TFile.h"
a0bfad0e 32#include <TGeoManager.h>
b8dc484b 33
34// STEER includes
35#include "AliRun.h"
36#include "AliHeader.h"
37#include "AliMC.h"
38#include "AliStack.h"
f7a1cc68 39#include "AliMagF.h"
6b092dfc 40#include "AliTracker.h"
b8dc484b 41
42// MUON includes
b8dc484b 43#include "AliMUONConstants.h"
44#include "AliMUONTrack.h"
45#include "AliMUONRecoCheck.h"
46#include "AliMUONTrackParam.h"
8cde4af5 47#include "AliMUONTrackExtrap.h"
48217459 48#include "AliMUONVTrackStore.h"
b8dc484b 49
50Int_t TrackCheck( Bool_t *compTrack);
51
edcc5b56 52void MUONRecoCheck (Int_t nEvent = -1, char* geoFilename = "geometry.root",
e54bf126 53 char * pathSim="./generated/", char * esdFileName="AliESDs.root")
54{
b8dc484b 55
c59f70b9 56 Bool_t compTrack[10];
b8dc484b 57 Bool_t compTrackOK[10];
96ebe67e 58 Int_t nClusterOk = 0;
b8dc484b 59 Int_t testTrack = 0;
60 Int_t iTrack = 0;
48217459 61 AliMUONTrack* trackOK(0x0);
b8dc484b 62 Int_t trackID = 0;
61fed964 63 Double_t sigmaCut = 4.; // 4 sigmas cut
64 Double_t maxChi2 = 999.;
b8dc484b 65 AliMUONTrackParam *trackParam;
b8dc484b 66 Double_t x1,y1,z1,pX1,pY1,pZ1,p1;
67 Double_t x2,y2,z2,pX2,pY2,pZ2,p2;
48217459 68
b8dc484b 69 // File for histograms and histogram booking
70 TFile *histoFile = new TFile("MUONRecoCheck.root", "RECREATE");
48217459 71
b8dc484b 72 TH1F *hReconstructible = new TH1F("hReconstructible"," Nb of reconstructible tracks ",15,-0.5,14.5);
73 TH1F *hReco = new TH1F("hReco"," Nb of reconstructed tracks / evt",15,-0.5,14.5);
96ebe67e 74 TH1F *hNClusterComp = new TH1F("hNClusterComp"," Nb of compatible clusters / track ",15,-0.5,14.5);
b8dc484b 75 TH1F *hTestTrack = new TH1F("hTestTrack"," Reconstruction requirement / track",15,-0.5,14.5);
76 TH1F *hTrackRefID = new TH1F("hTrackRefID"," track reference ID ",100,-0.5,99.5);
77
96ebe67e 78 TH1F *hResMomVertex = new TH1F("hResMomVertex"," delta P vertex (GeV/c)",100,-10.,10);
79 TH1F *hResMomFirstCluster = new TH1F("hResMomFirstCluster"," delta P first cluster (GeV/c)",100,-10.,10);
48217459 80
a0bfad0e 81 // Import TGeo geometry (needed by AliMUONTrackExtrap::ExtrapToVertex)
82 if (!gGeoManager) {
83 TGeoManager::Import(geoFilename);
84 if (!gGeoManager) {
85 Error("MUONmass_ESD", "getting geometry from file %s failed", geoFilename);
86 return;
87 }
88 }
89
45a05f07 90 // set mag field
91 // waiting for mag field in CDB
f7a1cc68 92 if (!TGeoGlobalMagField::Instance()->GetField()) {
93 printf("Loading field map...\n");
4642ac4b 94 AliMagF* field = new AliMagF("Maps","Maps",1.,1.,AliMagF::k5kG);
f7a1cc68 95 TGeoGlobalMagField::Instance()->SetField(field);
96 }
a0bfad0e 97 // set the magnetic field for track extrapolations
f7a1cc68 98 AliMUONTrackExtrap::SetField();
99
61fed964 100 AliMUONRecoCheck rc(esdFileName, pathSim);
48217459 101
102 Int_t nevents = rc.NumberOfEvents();
b8dc484b 103
edcc5b56 104 if (nevents < nEvent || nEvent < 0) nEvent = nevents;
b8dc484b 105
106 Int_t ievent;
107 Int_t nReconstructibleTracks = 0;
108 Int_t nReconstructedTracks = 0;
109 Int_t nReconstructibleTracksCheck = 0;
48217459 110
111 for (ievent=0; ievent<nEvent; ievent++)
112 {
b8dc484b 113 if (!(ievent%10)) printf(" **** event # %d \n",ievent);
b8dc484b 114
48217459 115 AliMUONVTrackStore* trackStore = rc.ReconstructedTracks(ievent);
116 AliMUONVTrackStore* trackRefStore = rc.ReconstructibleTracks(ievent);
b8dc484b 117
48217459 118 hReconstructible->Fill(trackRefStore->GetSize());
119 hReco->Fill(trackStore->GetSize());
120
121 nReconstructibleTracks += trackRefStore->GetSize();
122 nReconstructedTracks += trackStore->GetSize();
123
124 TIter next(trackRefStore->CreateIterator());
125 AliMUONTrack* trackRef;
126
127 while ( ( trackRef = static_cast<AliMUONTrack*>(next()) ) )
128 {
61fed964 129 maxChi2 = 999.;
b8dc484b 130 testTrack = 0;
48217459 131 trackOK = 0x0;
b8dc484b 132 for (Int_t ch = 0; ch < AliMUONConstants::NTrackingCh(); ch++)
48217459 133 {
134 compTrackOK[ch] = kFALSE;
b8dc484b 135 }
48217459 136
137 TIter next2(trackStore->CreateIterator());
138 AliMUONTrack* trackReco;
139
140 while ( ( trackReco = static_cast<AliMUONTrack*>(next2()) ) )
141 {
61fed964 142 // check if trackRef is compatible with trackReco
143 if (trackReco->GetNClusters() > 1) {
144
145 // check cluster by cluster if trackReco contain info at each cluster
c59f70b9 146 trackRef->CompatibleTrack(trackReco,sigmaCut,compTrack);
61fed964 147
148 iTrack = TrackCheck(compTrack);
149
150 if (iTrack > testTrack)
151 {
152 nClusterOk = 0;
153 for (Int_t ch = 0; ch < AliMUONConstants::NTrackingCh(); ch++)
154 {
155 if (compTrack[ch]) nClusterOk++;
156 compTrackOK[ch] = compTrack[ch];
157 }
158 testTrack = iTrack;
159 trackOK = trackReco;
160 }
161
162 } else {
163
164 // check only parameters at the z position of the first trackRef
165 AliMUONTrackParam *refParam = (AliMUONTrackParam*) trackRef->GetTrackParamAtCluster()->First();
166 AliMUONTrackParam recoParam(*((AliMUONTrackParam*) trackReco->GetTrackParamAtCluster()->First()));
61fed964 167 Double_t chi2;
4ea3f013 168 if (AliMUONTrackExtrap::ExtrapToZCov(&recoParam, refParam->GetZ()) &&
169 refParam->CompatibleTrackParam(recoParam, sigmaCut, chi2) && chi2 < maxChi2)
170 {
61fed964 171 maxChi2 = chi2;
172 trackOK = trackReco;
61fed964 173 }
174
175 }
48217459 176
48217459 177 }
178
b8dc484b 179 hTestTrack->Fill(testTrack);
edcc5b56 180 trackID = trackRef->GetMCLabel();
b8dc484b 181 hTrackRefID->Fill(trackID);
48217459 182
61fed964 183 if (testTrack == 4 || maxChi2 < 5.*sigmaCut*sigmaCut) { // tracking requirements verified, track is found
48217459 184 nReconstructibleTracksCheck++;
96ebe67e 185 hNClusterComp->Fill(nClusterOk);
48217459 186 trackParam = trackRef->GetTrackParamAtVertex();
187 x1 = trackParam->GetNonBendingCoor();
188 y1 = trackParam->GetBendingCoor();
189 z1 = trackParam->GetZ();
190 pX1 = trackParam->Px();
191 pY1 = trackParam->Py();
192 pZ1 = trackParam->Pz();
193 p1 = trackParam->P();
194
195 // printf(" Ref. track at vertex: x,y,z: %f %f %f px,py,pz,p: %f %f %f %f \n",x1,y1,z1,pX1,pY1,pZ1,p1);
196 trackReco = trackOK;
fa5e35be 197 trackParam = new AliMUONTrackParam(*((AliMUONTrackParam*)(trackReco->GetTrackParamAtVertex())));
48217459 198 x2 = trackParam->GetNonBendingCoor();
199 y2 = trackParam->GetBendingCoor();
200 z2 = trackParam->GetZ();
201 pX2 = trackParam->Px();
202 pY2 = trackParam->Py();
203 pZ2 = trackParam->Pz();
204 p2 = trackParam->P();
205 delete trackParam;
206 // printf(" Reconst. track at vertex: x,y,z: %f %f %f px,py,pz: %f %f %f %f \n",x2,y2,z2,pX2,pY2,pZ2,p2);
207
208 hResMomVertex->Fill(p2-p1);
209
96ebe67e 210 trackParam = (AliMUONTrackParam*) trackRef->GetTrackParamAtCluster()->First();
48217459 211 x1 = trackParam->GetNonBendingCoor();
212 y1 = trackParam->GetBendingCoor();
213 z1 = trackParam->GetZ();
214 pX1 = trackParam->Px();
215 pY1 = trackParam->Py();
216 pZ1 = trackParam->Pz();
217 p1 = trackParam->P();
96ebe67e 218 // printf(" Ref. track at 1st cluster: x,y,z: %f %f %f px,py,pz: %f %f %f \n",x1,y1,z1,pX1,pY1,pZ1);
219 trackParam = (AliMUONTrackParam*) trackOK->GetTrackParamAtCluster()->First();
48217459 220 x2 = trackParam->GetNonBendingCoor();
221 y2 = trackParam->GetBendingCoor();
222 z2 = trackParam->GetZ();
223 pX2 = trackParam->Px();
224 pY2 = trackParam->Py();
225 pZ2 = trackParam->Pz();
226 p2 = trackParam->P();
96ebe67e 227 // printf(" Reconst. track at 1st cluster: x,y,z: %f %f %f px,py,pz: %f %f %f \n",x2,y2,z2,pX2,pY2,pZ2);
48217459 228
96ebe67e 229 hResMomFirstCluster->Fill(p2-p1);
b8dc484b 230
231 }
232 } // end loop track ref.
b8dc484b 233
48217459 234 } // end loop on event
235
b8dc484b 236 printf(" nb of reconstructible tracks: %d \n", nReconstructibleTracks);
237 printf(" nb of reconstructed tracks: %d \n", nReconstructedTracks);
238 printf(" nb of reconstructible tracks which are reconstructed: %d \n", nReconstructibleTracksCheck);
239
240 histoFile->Write();
241 histoFile->Close();
242}
243
244
245Int_t TrackCheck( Bool_t *compTrack)
246{
247 // Apply reconstruction requirements
248 // Return number of validated conditions
249 // If all the tests are verified then TrackCheck = 4 (good track)
250 Int_t iTrack = 0;
96ebe67e 251 Int_t nCompClustersInLastStations = 0;
b8dc484b 252
253 // apply reconstruction requirements
96ebe67e 254 if (compTrack[0] || compTrack[1]) iTrack++; // at least one compatible cluster in st. 0
255 if (compTrack[2] || compTrack[3]) iTrack++; // at least one compatible cluster in st. 1
256 if (compTrack[4] || compTrack[5]) iTrack++; // at least one compatible cluster in st. 2
b8dc484b 257 for (Int_t ch = 6; ch < AliMUONConstants::NTrackingCh(); ch++) {
96ebe67e 258 if (compTrack[ch]) nCompClustersInLastStations++;
b8dc484b 259 }
96ebe67e 260 if (nCompClustersInLastStations > 2) iTrack++; // at least 3 compatible clusters in st. 3 & 4
b8dc484b 261
262 return iTrack;
48217459 263
b8dc484b 264}
265