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");
58 fTrackletTree->SetDirectory(0);
61 AliTRDgtuSim::~AliTRDgtuSim()
64 fTrackletArray->Delete();
65 delete fTrackletArray;
69 Bool_t AliTRDgtuSim::RunGTUFromTrackletFile(TString filename, Int_t event, Int_t noev)
71 AliInfo(Form("Running the GTU simulation on file: %s", filename.Data()));
72 ifstream input(filename.Data());
78 Int_t iEventPrev = -1;
79 Int_t iStackPrev = -1;
89 AliDebug(5,"--------- Reading from file ----------");
90 while (getline(input, str)) {
93 AliDebug(5,Form("Line %i : %s", lineno, string.Data()));
95 TObjArray *tokens = string.Tokenize(" ");
96 if (tokens->GetEntriesFast() < 7) {
97 AliWarning(Form("Invalid input in line %i, too few parameters", lineno));
101 if ( ((TObjString*) tokens->At(0))->GetString().Atoi() < event)
104 iEvent = ((TObjString*) tokens->At(0))->GetString().Atoi();
105 iSec = ((TObjString*) tokens->At(1))->GetString().Atoi();
106 iStack = ((TObjString*) tokens->At(2))->GetString().Atoi();
107 iLink = 2 * ((TObjString*) tokens->At(3))->GetString().Atoi() + ((TObjString*) tokens->At(4))->GetString().Atoi();
109 if (iEvent != iEventPrev || iStack != iStackPrev || iSec != iSecPrev) {
111 TList *ListOfTracks = new TList();
112 fTMU->SetStack(iStackPrev);
113 fTMU->SetSector(iSecPrev);
114 fTMU->RunTMU(ListOfTracks);
115 AliDebug(1,Form("--- There are %i tracks. Writing ...", ListOfTracks->GetEntries()));
116 WriteTracksToTree(ListOfTracks);
117 fTMU->WriteTrackletsToTree(fTrackletTree);
118 WriteTracksToDataFile(ListOfTracks, iEventPrev);
119 if (ListOfTracks->GetEntries() > 0)
120 AliDebug(2,Form(" %d GeV/c", ((AliTRDtrackGTU*) ListOfTracks->At(0))->GetPt() ));
122 fTMU = new AliTRDgtuTMU();
126 fTMU = new AliTRDgtuTMU();
135 for (Int_t i = 5; i < tokens->GetEntriesFast(); i++) {
136 UInt_t trackletWord = 0;
137 sscanf(((TObjString*) tokens->At(i))->GetString().Data(), "%x", &trackletWord);
138 if (trackletWord == 0x10001000)
140 AliDebug(2,Form("%i. tracklet: %s -> 0x%08x", i-4, ((TObjString*) tokens->At(i))->GetString().Data(), trackletWord));
141 AliTRDtrackletWord *trkl = new AliTRDtrackletWord(trackletWord);
142 fTMU->AddTracklet(trkl, iLink);
146 if (fTMU && evcnt < noev) {
147 TList *ListOfTracks = new TList();
148 fTMU->SetStack(iStackPrev);
149 fTMU->SetSector(iSecPrev);
150 fTMU->RunTMU(ListOfTracks);
151 WriteTracksToTree(ListOfTracks);
152 fTMU->WriteTrackletsToTree(fTrackletTree);
153 WriteTracksToDataFile(ListOfTracks, iEventPrev);
159 AliInfo(Form("Analyzed %i events", evcnt));
163 Bool_t AliTRDgtuSim::RunGTU(AliLoader *loader, AliESDEvent *esd)
165 if (!LoadTracklets(loader)) {
166 AliError("Could not load the tracklets. Aborting ...");
170 Int_t iStackPrev = -1;
181 TList *ListOfTracks = new TList();
183 TIter next(fTrackletArray);
184 AliTRDtrackletBase *trkl;
186 while ((trkl = (AliTRDtrackletBase*) next())) {
187 iSec = trkl->GetDetector() / 30;
188 iStack = (trkl->GetDetector() % 30) / 6;
189 iLink = 2 * (trkl->GetDetector() % 6) + (trkl->GetYbin() < 0 ? 0 : 1);
191 if (iStack != iStackPrev || iSec != iSecPrev) {
193 fTMU->SetStack(iStackPrev);
194 fTMU->SetSector(iSecPrev);
195 fTMU->RunTMU(ListOfTracks);
196 WriteTracksToTree(ListOfTracks);
197 fTMU->WriteTrackletsToTree(fTrackletTree);
198 WriteTracksToESD(ListOfTracks, esd);
200 ListOfTracks->Delete();
202 fTMU = new AliTRDgtuTMU();
207 fTMU->AddTracklet(trkl, iLink);
211 fTMU->SetStack(iStackPrev);
212 fTMU->SetSector(iSecPrev);
213 fTMU->RunTMU(ListOfTracks);
214 WriteTracksToTree(ListOfTracks);
215 fTMU->WriteTrackletsToTree(fTrackletTree);
216 WriteTracksToESD(ListOfTracks, esd);
219 ListOfTracks->Delete();
227 Bool_t AliTRDgtuSim::LoadTracklets(AliLoader *loader)
229 AliDebug(1,"Loading tracklets ...");
232 AliError("No loader given!");
236 AliDataLoader *trackletLoader = loader->GetDataLoader("tracklets");
237 if (!trackletLoader) {
238 AliError("No tracklet loader found!");
242 trackletLoader->Load();
244 TTree *trackletTree = trackletLoader->Tree();
246 AliError("No tracklet tree found");
251 TBranch *trklbranch = trackletTree->GetBranch("mcmtrklbranch");
254 fTrackletArray = new TClonesArray("AliTRDtrackletMCM", 1000);
255 else if ((TClass::GetClass("AliTRDtrackletMCM"))->InheritsFrom(fTrackletArray->Class()))
256 fTrackletArray->Delete();
258 fTrackletArray->Delete();
259 delete fTrackletArray;
260 fTrackletArray = new TClonesArray("AliTRDtrackletMCM", 1000);
263 AliTRDtrackletMCM *trkl = new AliTRDtrackletMCM;
264 trklbranch->SetAddress(&trkl);
265 for (Int_t iTracklet = 0; iTracklet < trklbranch->GetEntries(); iTracklet++) {
266 trklbranch->GetEntry(iTracklet);
267 new ((*fTrackletArray)[fTrackletArray->GetEntries()]) AliTRDtrackletMCM(*trkl);
272 trklbranch = trackletTree->GetBranch("trkbranch");
275 AliError("Could not get trkbranch");
280 fTrackletArray = new TClonesArray("AliTRDtrackletWord", 1000);
281 else if ((TClass::GetClass("AliTRDtrackletWord"))->InheritsFrom(fTrackletArray->Class()))
282 fTrackletArray->Delete();
284 fTrackletArray->Delete();
285 delete fTrackletArray;
286 fTrackletArray = new TClonesArray("AliTRDtrackletWord", 1000);
290 UInt_t *leaves = new UInt_t[258];
291 AliDebug(1,Form("No. of entries: %i", trklbranch->GetEntries()));
293 for (Int_t iEntry = 0; iEntry < trklbranch->GetEntries(); iEntry++) {
294 trklbranch->SetAddress(leaves);
295 trklbranch->GetEntry(iEntry);
296 for (Int_t iTracklet = 0; iTracklet < 256; iTracklet++) {
297 if (leaves[2 + iTracklet] == 0)
299 new((*fTrackletArray)[notrkl]) AliTRDtrackletWord(leaves[2 + iTracklet], leaves[0] + leaves[1]);
302 AliDebug(2,Form("Entry: %3i: Det: %3i, side: %i, 1st tracklet: 0x%08x, no: %i", iEntry, leaves[0], leaves[1], leaves[2], notrkl));
308 Bool_t AliTRDgtuSim::WriteTracksToDataFile(TList *ListOfTracks, Int_t event)
314 out = fopen("test.data", "a");
316 AliDebug(1,Form("%i tracks found in event %i", ListOfTracks->GetSize(), event));
317 fprintf(out, "0 %5i %2i %i 00000000\n", event, sm, stack);
318 for (Int_t i = 0; i < ListOfTracks->GetSize(); i++) {
319 AliTRDtrackGTU *trk = (AliTRDtrackGTU*) ListOfTracks->At(i);
320 sm = trk->GetSector();
321 stack = trk->GetStack();
322 fprintf(out, "1 %5i %2i %2i %3i %3i %3i %3i %3i %3i %3i %4i %f\n", event, sm, stack, trk->GetTrackletMask(),
323 trk->GetTrackletIndex(5),
324 trk->GetTrackletIndex(4),
325 trk->GetTrackletIndex(3),
326 trk->GetTrackletIndex(2),
327 trk->GetTrackletIndex(1),
328 trk->GetTrackletIndex(0),
336 Bool_t AliTRDgtuSim::WriteTracksToTree(TList *ListOfTracks, Int_t /*event*/)
338 AliDebug(1,Form("Writing %i tracks to the tree...", ListOfTracks->GetEntries()));
343 if (ListOfTracks->GetEntries() <= 0)
347 fTrackTree = new TTree("gtutracks", "GTU tracks");
348 fTrackTree->SetDirectory(0);
351 AliTRDtrackGTU *trk = 0x0;
352 TBranch *branch = fTrackTree->GetBranch("TRDgtuTrack");
354 branch = fTrackTree->Branch("TRDgtuTrack", "AliTRDtrackGTU", &trk, 32000, 99);
357 TIter next(ListOfTracks);
358 while ((trk = (AliTRDtrackGTU*) next())) {
360 branch->SetAddress(&trk);
363 fTrackTree->ResetBranchAddress(branch);
368 Bool_t AliTRDgtuSim::WriteTreesToFile() {
369 TFile *f = TFile::Open("TRD.GtuTracking.root", "RECREATE");
372 f->WriteTObject(fTrackTree);
374 f->WriteTObject(fTrackletTree);
379 Bool_t AliTRDgtuSim::WriteTracksToESD(TList *ListOfTracks, AliESDEvent *esd)
382 TIter next(ListOfTracks);
383 while (AliTRDtrackGTU *trk = (AliTRDtrackGTU*) next()) {
384 AliESDTrdTrack *trdtrack = trk->CreateTrdTrack();
385 esd->AddTrdTrack(trdtrack);
392 Bool_t AliTRDgtuSim::WriteTracksToLoader()
395 AliError("No track tree found!");
399 AliRunLoader *rl = AliRunLoader::Instance();
400 AliDataLoader *dl = 0x0;
402 dl = rl->GetLoader("TRDLoader")->GetDataLoader("gtutracks");
404 AliError("Could not get the GTU-track data loader!");
408 TTree *trackTree = dl->Tree();
411 trackTree = dl->Tree();
414 AliTRDtrackGTU *trk = 0x0;
415 TBranch *trkbranch = trackTree->GetBranch("TRDtrackGTU");
417 trkbranch = trackTree->Branch("TRDtrackGTU", "AliTRDtrackGTU", &trk, 32000);
419 TBranch *branch = fTrackTree->GetBranch("TRDgtuTrack");
423 AliDebug(1,Form("Found %i tracks", branch->GetEntries()));
425 for (Int_t iTrack = 0; iTrack < branch->GetEntries(); iTrack++) {
426 branch->SetAddress(&trk);
427 branch->GetEntry(iTrack);
428 trkbranch->SetAddress(&trk);
432 dl->WriteData("OVERWRITE");