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 **************************************************************************/
16 /* $Id: AliTRDgtuSim.cxx 28397 2008-09-02 09:33:00Z cblume $ */
18 ////////////////////////////////////////////////////////////////////////////
22 // Authors: J. Klein (Jochen.Klein@cern.ch) //
24 ////////////////////////////////////////////////////////////////////////////
32 #include "TClonesArray.h"
34 #include "AliTRDgtuSim.h"
35 #include "AliTRDmcmTracklet.h"
36 #include "AliTRDgtuTMU.h"
37 #include "AliTRDtrackGTU.h"
38 #include "AliTRDtrackletWord.h"
39 #include "AliTRDtrackletMCM.h"
40 #include "AliESDEvent.h"
43 #include "AliRunLoader.h"
44 #include "AliLoader.h"
47 ClassImp(AliTRDgtuSim)
49 AliTRDgtuSim::AliTRDgtuSim(AliRunLoader *rl)
57 fTrackletTree = new TTree("gtutracklets", "Tree with GTU tracklets");
60 AliTRDgtuSim::~AliTRDgtuSim()
63 fTrackletArray->Delete();
64 delete fTrackletArray;
68 Bool_t AliTRDgtuSim::RunGTUFromTrackletFile(TString filename, Int_t event, Int_t noev)
70 AliInfo(Form("Running the GTU simulation on file: %s", filename.Data()));
71 ifstream input(filename.Data());
77 Int_t iEventPrev = -1;
78 Int_t iStackPrev = -1;
88 AliInfo("--------- Reading from file ----------");
89 while (getline(input, str)) {
92 AliInfo(Form("Line %i : %s", lineno, string.Data()));
94 TObjArray *tokens = string.Tokenize(" ");
95 if (tokens->GetEntriesFast() < 7) {
96 AliWarning(Form("Invalid input in line %i, too few parameters", lineno));
100 if ( ((TObjString*) tokens->At(0))->GetString().Atoi() < event)
103 iEvent = ((TObjString*) tokens->At(0))->GetString().Atoi();
104 iSec = ((TObjString*) tokens->At(1))->GetString().Atoi();
105 iStack = ((TObjString*) tokens->At(2))->GetString().Atoi();
106 iLink = 2 * ((TObjString*) tokens->At(3))->GetString().Atoi() + ((TObjString*) tokens->At(4))->GetString().Atoi();
108 if (iEvent != iEventPrev || iStack != iStackPrev || iSec != iSecPrev) {
110 TList *ListOfTracks = new TList();
111 fTMU->SetStack(iStackPrev);
112 fTMU->SetSector(iSecPrev);
113 fTMU->RunTMU(ListOfTracks);
114 AliInfo(Form("--- There are %i tracks. Writing ...", ListOfTracks->GetEntries()));
115 WriteTracksToTree(ListOfTracks);
116 fTMU->WriteTrackletsToTree(fTrackletTree);
117 WriteTracksToDataFile(ListOfTracks, iEventPrev);
118 if (ListOfTracks->GetEntries() > 0)
119 AliInfo(Form(" %d GeV/c", ((AliTRDtrackGTU*) ListOfTracks->At(0))->GetPt() ));
121 fTMU = new AliTRDgtuTMU();
125 fTMU = new AliTRDgtuTMU();
134 for (Int_t i = 5; i < tokens->GetEntriesFast(); i++) {
135 UInt_t trackletWord = 0;
136 sscanf(((TObjString*) tokens->At(i))->GetString().Data(), "%x", &trackletWord);
137 if (trackletWord == 0x10001000)
139 AliInfo(Form("%i. tracklet: %s -> 0x%08x", i-4, ((TObjString*) tokens->At(i))->GetString().Data(), trackletWord));
140 AliTRDtrackletWord *trkl = new AliTRDtrackletWord(trackletWord);
141 fTMU->AddTracklet(trkl, iLink);
145 if (fTMU && evcnt < noev) {
146 TList *ListOfTracks = new TList();
147 fTMU->SetStack(iStackPrev);
148 fTMU->SetSector(iSecPrev);
149 fTMU->RunTMU(ListOfTracks);
150 WriteTracksToTree(ListOfTracks);
151 fTMU->WriteTrackletsToTree(fTrackletTree);
152 WriteTracksToDataFile(ListOfTracks, iEventPrev);
158 AliInfo(Form("Analyzed %i events", evcnt));
162 Bool_t AliTRDgtuSim::RunGTU(AliLoader *loader, AliESDEvent *esd)
164 if (!LoadTracklets(loader)) {
165 AliError("Could not load the tracklets. Aborting ...");
169 Int_t iStackPrev = -1;
180 TList *ListOfTracks = new TList();
182 TIter next(fTrackletArray);
183 AliTRDtrackletBase *trkl;
185 while ((trkl = (AliTRDtrackletBase*) next())) {
186 iSec = trkl->GetDetector() / 30;
187 iStack = (trkl->GetDetector() % 30) / 6;
188 iLink = 2 * (trkl->GetDetector() % 6) + (trkl->GetYbin() < 0 ? 0 : 1);
190 if (iStack != iStackPrev || iSec != iSecPrev) {
192 fTMU->SetStack(iStackPrev);
193 fTMU->SetSector(iSecPrev);
194 fTMU->RunTMU(ListOfTracks);
195 WriteTracksToTree(ListOfTracks);
196 fTMU->WriteTrackletsToTree(fTrackletTree);
197 WriteTracksToESD(ListOfTracks, esd);
199 ListOfTracks->Delete();
201 fTMU = new AliTRDgtuTMU();
206 fTMU->AddTracklet(trkl, iLink);
210 fTMU->SetStack(iStackPrev);
211 fTMU->SetSector(iSecPrev);
212 fTMU->RunTMU(ListOfTracks);
213 WriteTracksToTree(ListOfTracks);
214 fTMU->WriteTrackletsToTree(fTrackletTree);
215 WriteTracksToESD(ListOfTracks, esd);
218 ListOfTracks->Delete();
226 Bool_t AliTRDgtuSim::LoadTracklets(AliLoader *loader)
228 AliInfo("Loading tracklets ...");
231 AliError("No loader given!");
235 AliDataLoader *trackletLoader = loader->GetDataLoader("tracklets");
236 if (!trackletLoader) {
237 AliError("No tracklet loader found!");
241 trackletLoader->Load();
243 TTree *trackletTree = trackletLoader->Tree();
245 AliError("No tracklet tree found");
250 TBranch *trklbranch = trackletTree->GetBranch("mcmtrklbranch");
253 fTrackletArray = new TClonesArray("AliTRDtrackletMCM", 1000);
254 else if ((TClass::GetClass("AliTRDtrackletMCM"))->InheritsFrom(fTrackletArray->Class()))
255 fTrackletArray->Delete();
257 fTrackletArray->Delete();
258 delete fTrackletArray;
259 fTrackletArray = new TClonesArray("AliTRDtrackletMCM", 1000);
262 AliTRDtrackletMCM *trkl = new AliTRDtrackletMCM;
263 trklbranch->SetAddress(&trkl);
264 for (Int_t iTracklet = 0; iTracklet < trklbranch->GetEntries(); iTracklet++) {
265 trklbranch->GetEntry(iTracklet);
266 new ((*fTrackletArray)[fTrackletArray->GetEntries()]) AliTRDtrackletMCM(*trkl);
271 trklbranch = trackletTree->GetBranch("trkbranch");
274 AliError("Could not get trkbranch");
279 fTrackletArray = new TClonesArray("AliTRDtrackletWord", 1000);
280 else if ((TClass::GetClass("AliTRDtrackletWord"))->InheritsFrom(fTrackletArray->Class()))
281 fTrackletArray->Delete();
283 fTrackletArray->Delete();
284 delete fTrackletArray;
285 fTrackletArray = new TClonesArray("AliTRDtrackletWord", 1000);
289 UInt_t *leaves = new UInt_t[258];
290 AliInfo(Form("No. of entries: %i", trklbranch->GetEntries()));
292 for (Int_t iEntry = 0; iEntry < trklbranch->GetEntries(); iEntry++) {
293 trklbranch->SetAddress(leaves);
294 trklbranch->GetEntry(iEntry);
295 for (Int_t iTracklet = 0; iTracklet < 256; iTracklet++) {
296 if (leaves[2 + iTracklet] == 0)
298 new((*fTrackletArray)[notrkl]) AliTRDtrackletWord(leaves[2 + iTracklet], leaves[0] + leaves[1]);
301 AliInfo(Form("Entry: %3i: Det: %3i, side: %i, 1st tracklet: 0x%08x, no: %i", iEntry, leaves[0], leaves[1], leaves[2], notrkl));
307 Bool_t AliTRDgtuSim::WriteTracksToDataFile(TList *ListOfTracks, Int_t event)
313 out = fopen("test.data", "a");
315 AliInfo(Form("%i tracks found in event %i", ListOfTracks->GetSize(), event));
316 fprintf(out, "0 %5i %2i %i 00000000\n", event, sm, stack);
317 for (Int_t i = 0; i < ListOfTracks->GetSize(); i++) {
318 AliTRDtrackGTU *trk = (AliTRDtrackGTU*) ListOfTracks->At(i);
319 sm = trk->GetSector();
320 stack = trk->GetStack();
321 fprintf(out, "1 %5i %2i %2i %3i %3i %3i %3i %3i %3i %3i %4i %f\n", event, sm, stack, trk->GetTrackletMask(),
322 trk->GetTrackletIndex(5),
323 trk->GetTrackletIndex(4),
324 trk->GetTrackletIndex(3),
325 trk->GetTrackletIndex(2),
326 trk->GetTrackletIndex(1),
327 trk->GetTrackletIndex(0),
335 Bool_t AliTRDgtuSim::WriteTracksToTree(TList *ListOfTracks, Int_t /*event*/)
337 AliInfo(Form("Writing %i tracks to the tree...", ListOfTracks->GetEntries()));
342 if (ListOfTracks->GetEntries() <= 0)
346 fTrackTree = new TTree("gtutracks", "GTU tracks");
348 AliTRDtrackGTU *trk = 0x0;
349 TBranch *branch = fTrackTree->GetBranch("TRDgtuTrack");
351 branch = fTrackTree->Branch("TRDgtuTrack", "AliTRDtrackGTU", &trk, 32000, 99);
354 TIter next(ListOfTracks);
355 while ((trk = (AliTRDtrackGTU*) next())) {
357 branch->SetAddress(&trk);
360 fTrackTree->ResetBranchAddress(branch);
365 Bool_t AliTRDgtuSim::WriteTreesToFile() {
366 TFile *f = TFile::Open("TRD.GtuTracking.root", "RECREATE");
369 f->WriteTObject(fTrackTree);
371 f->WriteTObject(fTrackletTree);
376 Bool_t AliTRDgtuSim::WriteTracksToESD(TList *ListOfTracks, AliESDEvent *esd)
379 TIter next(ListOfTracks);
380 while (AliTRDtrackGTU *trk = (AliTRDtrackGTU*) next()) {
381 AliESDTrdTrack *trdtrack = trk->CreateTrdTrack();
382 esd->AddTrdTrack(trdtrack);