AliTPCcalibDB.cxx - Calculate the distance to the closest measurement
[u/mrichter/AliRoot.git] / MUON / DecodeRecoCocktail.C
CommitLineData
fa6e7866 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
e54bf126 18/// \ingroup macros
19/// \file DecodeRecoCocktail.C
20/// \brief add brief description
21///
22/// \author A. De Falco, H. Woehri, INFN Cagliari, July 2006
23///
24/// This macro reads the generation/reconstruction files in the
25/// input directories (recodir and simdir),
26/// builds a tree that contains, for each event, an array of muons and dimuons
27/// (TClonesArrays of AliMUONTrackLight and AliMUONPairLight objects)
28/// and writes the tree in an output file (outFileName)
29/// Note that if the path for the output file is not explicitly specified,
30/// it will be written in the directory containing the generation/reconstruction
31/// 27-Nov-2006: modified by in order to loop on files
32///
33/// 13 Nov 2007:
34/// Updated this macro to work with new version of AliMUONRecoCheck. Also, we are
35/// now fetching reconstructed track data from ESD and not the track tree, because
36/// the AliMUONTrack objects for reconstructed tracks are no longer stored on disk.
37/// - Artur Szostak <artursz@iafrica.com>
9bdda5f6 38
4d309390 39#if !defined(__CINT__) || defined(__MAKECINT__)
40#include <Riostream.h>
fa6e7866 41#include <TClonesArray.h>
42#include <TFile.h>
43#include <TTree.h>
44#include <TMath.h>
45#include <TLorentzVector.h>
46#include <TParticle.h>
47#include <TSystem.h>
59830a87 48#include <TGeoManager.h>
a99c3449 49#include "AliMUONCDB.h"
4d309390 50#include "AliMUONRecoCheck.h"
51#include "AliMUONTrack.h"
fa6e7866 52#include "AliMUONTrackLight.h"
53#include "AliMUONPairLight.h"
9bdda5f6 54#include "AliMUONVTrackStore.h"
22ccc301 55#include "AliMUONTrackExtrap.h"
9bdda5f6 56#include "AliESDEvent.h"
57#include "AliESDVertex.h"
58#include "AliMCEventHandler.h"
59#include "AliMCEvent.h"
a99c3449 60#include "AliStack.h"
61#include "AliCDBManager.h"
9bdda5f6 62/*TODO: need to update this with changes made to ITS
4d309390 63#include "AliITSVertexerPPZ.h"
64#include "AliITSLoader.h"
9bdda5f6 65*/
4d309390 66#endif
fa6e7866 67
4d309390 68
9bdda5f6 69void DecodeRecoCocktail(
70 char* recodir=".", // The directory containing galice.root for reconstructed data.
71 char* simdir="generated/", // The directory containing galice.root for simulated data.
72 char* outFileName = "MuonLight.root", // The output filename containing AliMUONTrackLight and AliMUONPairLight objects.
61fed964 73 char* geoFilename = "geometry.root" // The filename containing the geometry.
9bdda5f6 74 )
75{
e54bf126 76/// \param recodir The directory containing galice.root for reconstructed data.
77/// \param simdir The directory containing galice.root for simulated data.
78/// \param outFileName The output filename containing AliMUONTrackLight and AliMUONPairLight objects.
79/// \param geoFilename The filename containing the geometry.
80
4d309390 81 char startingDir[200];
82 sprintf (startingDir,"%s",gSystem->pwd());
9bdda5f6 83 gSystem->cd(recodir);
4d309390 84
85 TClonesArray *muonArray = new TClonesArray("AliMUONTrackLight",10);
86 TClonesArray *dimuonArray = new TClonesArray("AliMUONPairLight",10);
87 TTree *treeOut = new TTree("tree","tree");
88
fa6e7866 89 TFile *fout = new TFile(outFileName,"recreate");
4d309390 90 fout->cd();
fa6e7866 91
4d309390 92 treeOut->Branch("muons",&muonArray);
93 treeOut->Branch("dimuons",&dimuonArray);
94
59830a87 95 // Import TGeo geometry (needed by AliMUONTrackExtrap::ExtrapToVertex)
96 if (!gGeoManager) {
97 TGeoManager::Import(geoFilename);
98 if (!gGeoManager) {
99 Error("MUONmass_ESD", "getting geometry from file %s failed", geoFilename);
100 return;
101 }
102 }
103
a99c3449 104 // load necessary data from OCDB
105 AliCDBManager::Instance()->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
106 AliCDBManager::Instance()->SetRun(0);
107 if (!AliMUONCDB::LoadField()) return;
59830a87 108 // set the magnetic field for track extrapolations
f7a1cc68 109 AliMUONTrackExtrap::SetField();
59830a87 110
61fed964 111 AliMUONRecoCheck *rc = new AliMUONRecoCheck("AliESDs.root", simdir);
9bdda5f6 112 Int_t nev = rc->NumberOfEvents();
fa6e7866 113
9bdda5f6 114 /*TODO: need to update this with changes made to ITS
a4ee7ab9 115 AliITSLoader* ITSloader = (AliITSLoader*) runLoaderSim->GetLoader("ITSLoader");
4d309390 116 AliITSVertexerPPZ *dovert = 0;
117 if (ITSloader) {
118 dovert = new AliITSVertexerPPZ("default",0,0);
21939432 119 // dovert->SetDebug(0);
4d309390 120 dovert->SetDiffPhiMax(0.05);
121 dovert->SetWindow(3.);
122 }
123 AliESDVertex *vert = 0;
9bdda5f6 124 */
fa6e7866 125
126 TLorentzVector v;
9bdda5f6 127
fa6e7866 128 for(Int_t ievent = 0; ievent < nev; ievent++){ // loop over events
9bdda5f6 129
130 /*TODO: need to update this with changes made to ITS
a4ee7ab9 131 runLoaderSim->GetHeader();
4d309390 132 if (ITSloader) {
133 vert = dovert->FindVertexForCurrentEvent(ievent);
134 }
9bdda5f6 135 */
136
a4ee7ab9 137 //printf ("Event %d of %d\n",ievent,nev);
fa6e7866 138 muonArray->Clear(); // clean muon and dimuon arrays
139 dimuonArray->Clear();
fa6e7866 140
a99c3449 141 AliMUONVTrackStore* recoTracks = rc->ReconstructedTracks(ievent, kFALSE); // Use tracks from actual reconstruction.
9bdda5f6 142 //AliMUONVTrackStore* trackRefs = rc->ReconstructibleTracks(ievent); // Only use reconstructible reference tracks.
143 AliMUONVTrackStore* trackRefs = rc->TrackRefs(ievent);
61fed964 144 AliStack* pstack = (const_cast<AliMCEventHandler*>(rc->GetMCEventHandler()))->MCEvent()->Stack();
9bdda5f6 145
146 TIter next(recoTracks->CreateIterator());
147 AliMUONTrack* trackReco = NULL;
148
149 Int_t nTrackReco = recoTracks->GetSize();
61fed964 150 Int_t nTracksESD = rc->GetESDEvent()->GetNumberOfMuonTracks();
9bdda5f6 151 if (nTrackReco != nTracksESD) printf ("Tracks in recoTracks (%d) and in ESD (%d) do not match!\n", nTrackReco, nTracksESD);
152 Int_t nreftracks = 0;
153 Int_t itrRec = 0;
154 while ( (trackReco = static_cast<AliMUONTrack*>(next())) != NULL )
155 {
fa6e7866 156 // assign parameters concerning the reconstructed tracks
4d309390 157 AliMUONTrackLight muLight;
a4ee7ab9 158
61fed964 159 muLight.FillFromESD(rc->GetESDEvent()->GetMuonTrack(itrRec));
9bdda5f6 160 // muLight.FillFromAliMUONTrack(trackReco);
4d309390 161
162 // find the reference track and store further information
61fed964 163 TParticle *part = muLight.FindRefTrack(trackReco, trackRefs, pstack);
fa6e7866 164 if (part) {
165 v.SetPxPyPzE(part->Px(), part->Py(), part->Pz(), part->Energy());
82724a77 166 muLight.SetPGen(v);
9bdda5f6 167 muLight.FillMuonHistory(pstack, part);
4d309390 168 // muLight.PrintInfo("A");
fa6e7866 169 //store the referenced track in the muonArray:
170 TClonesArray &muons = *muonArray;
82724a77 171 new (muons[nreftracks++]) AliMUONTrackLight(muLight);
4d309390 172 }
9bdda5f6 173
174 itrRec++;
4d309390 175 } // end reco track
fa6e7866 176
177 // now loop over muon pairs to build dimuons
178 Int_t nmuons = muonArray->GetEntriesFast();
179 Int_t ndimuons = 0;
180 for(Int_t itrRec1 = 0; itrRec1 < nmuons-1; itrRec1++) {
181 AliMUONTrackLight* mu1 = (AliMUONTrackLight*) muonArray->At(itrRec1);
182 for(Int_t itrRec2 = itrRec1+1; itrRec2 < nmuons; itrRec2++){
183 AliMUONTrackLight* mu2 = (AliMUONTrackLight*) muonArray->At(itrRec2);
4d309390 184 AliMUONPairLight dimuLight;
82724a77 185 dimuLight.SetMuons(*mu1, *mu2);
fa6e7866 186 TClonesArray &dimuons = *dimuonArray;
82724a77 187 new (dimuons[ndimuons++]) AliMUONPairLight(dimuLight);
fa6e7866 188 }
189 }
9bdda5f6 190
fa6e7866 191 treeOut->Fill();
4d309390 192 }
193
fa6e7866 194 fout->cd();
195 treeOut->Write();
196 gSystem->cd(startingDir);
82724a77 197 fout->Close();
82724a77 198 delete muonArray;
199 delete dimuonArray;
4d309390 200 delete treeOut;
201 delete fout;
202 delete rc;
fa6e7866 203}
4d309390 204