]>
Commit | Line | Data |
---|---|---|
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 | ||
18 | // A. De Falco, H. Woehri, INFN Cagliari, July 2006 | |
19 | // This macro reads the generation/reconstruction files in the | |
20 | // input directory (dirname), | |
21 | // builds a tree that contains, for each event, an array of muons and dimuons | |
22 | // (TClonesArrays of AliMUONTrackLight and AliMUONPairLight objects) | |
23 | // and writes the tree in an output file (outFileName) | |
24 | // Note that if the path for the output file is not explicitly specified, | |
25 | // it will be written in the directory containing the generation/reconstruction | |
26 | ||
27 | #include <iostream> | |
28 | #include <TClonesArray.h> | |
29 | #include <TFile.h> | |
30 | #include <TTree.h> | |
31 | #include <TMath.h> | |
32 | #include <TLorentzVector.h> | |
33 | #include <TParticle.h> | |
34 | #include <TSystem.h> | |
35 | #include <AliMUONRecoCheck.h> | |
36 | #include <AliMUONTrack.h> | |
37 | #include <AliMUONTrackParam.h> | |
38 | #include <AliRunLoader.h> | |
39 | #include "AliMUONTrackLight.h" | |
40 | #include "AliMUONPairLight.h" | |
22ccc301 | 41 | #include "AliMUONTrackExtrap.h" |
fa6e7866 | 42 | |
43 | void DecodeRecoCocktail(char* dirname=".", char* outFileName = "MuonLight.root"){ | |
44 | const char *startingDir = gSystem->pwd(); | |
45 | gSystem->cd(dirname); | |
46 | TFile *fout = new TFile(outFileName,"recreate"); | |
47 | ||
48 | AliMUONRecoCheck *rc = new AliMUONRecoCheck("galice.root"); | |
49 | AliRunLoader *runLoader = rc->GetRunLoader(); | |
50 | ||
51 | TClonesArray *muonArray = new TClonesArray("AliMUONTrackLight",10); | |
52 | TClonesArray *dimuonArray = new TClonesArray("AliMUONPairLight",10); | |
53 | TTree *treeOut = new TTree("tree","tree"); | |
54 | ||
55 | fout->cd(); | |
56 | treeOut->Branch("muons",&muonArray); | |
57 | treeOut->Branch("dimuons",&dimuonArray); | |
58 | ||
59 | runLoader->LoadKinematics("READ");; | |
60 | Int_t nev = runLoader->GetNumberOfEvents(); | |
61 | ||
62 | TLorentzVector v; | |
63 | for(Int_t ievent = 0; ievent < nev; ievent++){ // loop over events | |
64 | printf ("Event %d of %d\n",ievent,nev); | |
65 | muonArray->Clear(); // clean muon and dimuon arrays | |
66 | dimuonArray->Clear(); | |
67 | runLoader->GetEvent(ievent); | |
68 | rc->ResetTracks(); | |
69 | rc->MakeTrackRef(); // make reconstructable tracks | |
70 | ||
71 | TClonesArray * trackRecoArray = rc->GetTrackReco(); | |
72 | TClonesArray * trackRefArray = rc->GetMuonTrackRef(); | |
73 | Int_t nTrackReco = trackRecoArray->GetEntriesFast(); | |
74 | Int_t nreftracks = 0; | |
75 | for (Int_t itrRec = 0; itrRec<nTrackReco; itrRec++) { | |
76 | // assign parameters concerning the reconstructed tracks | |
77 | AliMUONTrack *trackReco = (AliMUONTrack *)trackRecoArray->At(itrRec); | |
82724a77 | 78 | AliMUONTrackLight muLight; |
22ccc301 | 79 | AliMUONTrackParam trPar(*((AliMUONTrackParam*) (trackReco->GetTrackParamAtHit()->First()))); |
80 | AliMUONTrackExtrap::ExtrapToVertex(&trPar,0.,0.,0.); | |
81 | muLight.SetCharge(Int_t(TMath::Sign(1.,trPar.GetInverseBendingMomentum()))); | |
82 | muLight.SetPxPyPz(trPar.Px(),trPar.Py(), trPar.Pz()); | |
82724a77 | 83 | muLight.SetTriggered(trackReco->GetMatchTrigger()); |
22ccc301 | 84 | Double_t xyz[3] = { trPar.GetNonBendingCoor(), |
85 | trPar.GetBendingCoor(), | |
86 | trPar.GetZ()}; | |
82724a77 | 87 | muLight.SetVertex(xyz); |
fa6e7866 | 88 | // find the reference track and store further information |
82724a77 | 89 | TParticle *part = muLight.FindRefTrack(trackReco,trackRefArray,runLoader); |
fa6e7866 | 90 | if (part) { |
91 | v.SetPxPyPzE(part->Px(), part->Py(), part->Pz(), part->Energy()); | |
82724a77 | 92 | muLight.SetPGen(v); |
93 | muLight.FillMuonHistory(runLoader, part); | |
94 | muLight.PrintInfo("A"); | |
fa6e7866 | 95 | //store the referenced track in the muonArray: |
96 | TClonesArray &muons = *muonArray; | |
82724a77 | 97 | new (muons[nreftracks++]) AliMUONTrackLight(muLight); |
fa6e7866 | 98 | } |
99 | } | |
100 | ||
101 | // now loop over muon pairs to build dimuons | |
102 | Int_t nmuons = muonArray->GetEntriesFast(); | |
103 | Int_t ndimuons = 0; | |
104 | for(Int_t itrRec1 = 0; itrRec1 < nmuons-1; itrRec1++) { | |
105 | AliMUONTrackLight* mu1 = (AliMUONTrackLight*) muonArray->At(itrRec1); | |
106 | for(Int_t itrRec2 = itrRec1+1; itrRec2 < nmuons; itrRec2++){ | |
107 | AliMUONTrackLight* mu2 = (AliMUONTrackLight*) muonArray->At(itrRec2); | |
82724a77 | 108 | AliMUONPairLight dimuLight; |
109 | dimuLight.SetMuons(*mu1, *mu2); | |
110 | // if(dimuLight.GetCreationProcess() == 2){ | |
fa6e7866 | 111 | // printf ("#dimuon = %d (%d, %d) \n", ndimuons, itrRec1, itrRec2); |
82724a77 | 112 | // dimuLight.PrintInfo("A"); |
fa6e7866 | 113 | // } |
114 | //store the referenced track in the dimuonArray: | |
115 | TClonesArray &dimuons = *dimuonArray; | |
82724a77 | 116 | new (dimuons[ndimuons++]) AliMUONPairLight(dimuLight); |
fa6e7866 | 117 | } |
118 | } | |
82724a77 | 119 | //Int_t ndimu2 = dimuonArray->GetEntriesFast(); |
fa6e7866 | 120 | // printf ("dimuonArray has %d entries\n",ndimu2); |
121 | // dimuonArray->Dump(); | |
122 | // printf ("filling tree\n"); | |
123 | // // fill the tree | |
124 | treeOut->Fill(); | |
125 | // printf ("done\n"); | |
126 | ||
127 | } | |
128 | fout->cd(); | |
129 | treeOut->Write(); | |
130 | gSystem->cd(startingDir); | |
82724a77 | 131 | fout->Close(); |
132 | delete fout; | |
133 | delete rc; | |
134 | delete muonArray; | |
135 | delete dimuonArray; | |
fa6e7866 | 136 | } |