+Bool_t AliTRDclusterizer::ReadTracklets()
+{
+ //
+ // Reads simulated tracklets from the input aliroot file
+ //
+
+ AliRunLoader *runLoader = AliRunLoader::Instance();
+ if (!runLoader) {
+ AliError("No run loader available");
+ return kFALSE;
+ }
+
+ AliLoader* loader = runLoader->GetLoader("TRDLoader");
+
+ AliDataLoader *trackletLoader = loader->GetDataLoader("tracklets");
+ if (!trackletLoader) {
+ return kFALSE;
+ }
+ trackletLoader->Load();
+
+ Bool_t loaded = kFALSE;
+ // look for simulated tracklets
+ TTree *trackletTree = trackletLoader->Tree();
+
+ if (trackletTree) {
+ TBranch *trklbranch = trackletTree->GetBranch("mcmtrklbranch");
+ TClonesArray *trklArray = TrackletsArray("AliTRDtrackletMCM");
+ if (trklbranch && trklArray) {
+ AliTRDtrackletMCM *trkl = 0x0;
+ trklbranch->SetAddress(&trkl);
+ Int_t nTracklets = trklbranch->GetEntries();
+ for (Int_t iTracklet = 0; iTracklet < nTracklets; iTracklet++) {
+ trklbranch->GetEntry(iTracklet);
+ new ((*trklArray)[trklArray->GetEntries()]) AliTRDtrackletMCM(*trkl);
+ }
+ loaded = kTRUE;
+ }
+ }
+ else {
+ // if no simulated tracklets found, look for raw tracklets
+ AliTreeLoader *treeLoader = (AliTreeLoader*) trackletLoader->GetBaseLoader("tracklets-raw");
+ trackletTree = treeLoader ? treeLoader->Load(), treeLoader->Tree() : 0x0;
+
+ if (trackletTree) {
+ TClonesArray *trklArray = TrackletsArray("AliTRDtrackletWord");
+
+ Int_t hc;
+ TClonesArray *ar = 0x0;
+ trackletTree->SetBranchAddress("hc", &hc);
+ trackletTree->SetBranchAddress("trkl", &ar);
+
+ Int_t nEntries = trackletTree->GetEntries();
+ for (Int_t iEntry = 0; iEntry < nEntries; iEntry++) {
+ trackletTree->GetEntry(iEntry);
+ Int_t nTracklets = ar->GetEntriesFast();
+ AliDebug(2, Form("%i tracklets in HC %i", nTracklets, hc));
+ for (Int_t iTracklet = 0; iTracklet < nTracklets; iTracklet++) {
+ AliTRDtrackletWord *trklWord = (AliTRDtrackletWord*) (*ar)[iTracklet];
+ new ((*trklArray)[trklArray->GetEntries()]) AliTRDtrackletWord(trklWord->GetTrackletWord(), hc);
+ }
+ }
+ loaded = kTRUE;
+ }
+ }
+
+ trackletLoader->UnloadAll();
+ trackletLoader->CloseFile();
+
+ return loaded;
+}
+
+Bool_t AliTRDclusterizer::ReadTracks()
+{
+ //
+ // Reads simulated GTU tracks from the input aliroot file
+ //
+
+ AliRunLoader *runLoader = AliRunLoader::Instance();
+
+ if (!runLoader) {
+ AliError("No run loader available");
+ return kFALSE;
+ }
+
+ AliLoader* loader = runLoader->GetLoader("TRDLoader");
+ if (!loader) {
+ return kFALSE;
+ }
+
+ AliDataLoader *trackLoader = loader->GetDataLoader("gtutracks");
+ if (!trackLoader) {
+ return kFALSE;
+ }
+
+ Bool_t loaded = kFALSE;
+
+ trackLoader->Load();
+
+ TTree *trackTree = trackLoader->Tree();
+ if (trackTree) {
+ TClonesArray *trackArray = TracksArray();
+ AliTRDtrackGTU *trk = 0x0;
+ trackTree->SetBranchAddress("TRDtrackGTU", &trk);
+ for (Int_t iTrack = 0; iTrack < trackTree->GetEntries(); iTrack++) {
+ trackTree->GetEntry(iTrack);
+ new ((*trackArray)[trackArray->GetEntries()]) AliESDTrdTrack(*(trk->CreateTrdTrack()));
+ }
+ loaded = kTRUE;
+ }
+
+ trackLoader->UnloadAll();
+ trackLoader->CloseFile();
+
+ return loaded;
+}
+