Reverting GetEntriesFast to GetEntries for TList only

[u/mrichter/AliRoot.git] / TRD / AliTRDgtuSim.cxx

diff --git a/TRD/AliTRDgtuSim.cxx b/TRD/AliTRDgtuSim.cxx
index 94058d4e62a66062f00c27138dce964440a3e1d2..f22efe25f2d0a568369638ed1bfba3f2d11616a0 100644 (file)
--- a/TRD/AliTRDgtuSim.cxx
+++ b/TRD/AliTRDgtuSim.cxx
@@ -48,6 +48,7 @@
 #include "AliTRDtrackletMCM.h"
 #include "AliESDEvent.h"
 
+using std::ifstream;
 ClassImp(AliTRDgtuSim)
 
 AliTRDgtuSim::AliTRDgtuSim(AliRunLoader *rl)
@@ -168,7 +169,7 @@ Bool_t AliTRDgtuSim::RunGTUFromTrackletFile(TString filename, Int_t event, Int_t
   return kTRUE;
 }
 
-Bool_t AliTRDgtuSim::RunGTU(AliLoader *loader, AliESDEvent *esd)
+Bool_t AliTRDgtuSim::RunGTU(AliLoader *loader, AliESDEvent *esd, Int_t label)
 {
   // run the GTU on tracklets taken from the loader
   // if specified the GTU tracks are written to the ESD event
@@ -186,7 +187,7 @@ Bool_t AliTRDgtuSim::RunGTU(AliLoader *loader, AliESDEvent *esd)
     }
   }
   else {
-    LoadTracklets(esd);
+    LoadTracklets(esd, label);
   }
 
     AliDebug(1, Form("running on %i tracklets", fTrackletArray->GetEntriesFast()));
@@ -250,7 +251,7 @@ Bool_t AliTRDgtuSim::RunGTU(AliLoader *loader, AliESDEvent *esd)
     return kTRUE;
 }
 
-Bool_t AliTRDgtuSim::LoadTracklets(const AliESDEvent *const esd)
+Bool_t AliTRDgtuSim::LoadTracklets(const AliESDEvent *const esd, Int_t label)
 {
   AliDebug(1,"Loading tracklets from ESD event ...");
 
@@ -259,7 +260,13 @@ Bool_t AliTRDgtuSim::LoadTracklets(const AliESDEvent *const esd)
 
   for (Int_t iTracklet = 0; iTracklet < esd->GetNumberOfTrdTracklets(); iTracklet++) {
     AliESDTrdTracklet *trkl = esd->GetTrdTracklet(iTracklet);
-    new ((*fTrackletArray)[fTrackletArray->GetEntries()]) AliTRDtrackletGTU(trkl);
+    if (label < -1) {
+      if (trkl->GetLabel() == label)
+       new ((*fTrackletArray)[fTrackletArray->GetEntries()]) AliTRDtrackletGTU(trkl);
+    }
+    else
+      if (trkl->GetLabel() >= -1)
+       new ((*fTrackletArray)[fTrackletArray->GetEntries()]) AliTRDtrackletGTU(trkl);
   }
 
   return kTRUE;
@@ -398,8 +405,6 @@ Bool_t AliTRDgtuSim::WriteTracksToESD(const TList * const listOfTracks, AliESDEv
        TIter next(listOfTracks);
        while (AliTRDtrackGTU *trk = (AliTRDtrackGTU*) next()) {
            AliESDTrdTrack *trdtrack = trk->CreateTrdTrack();
-           if (trdtrack->GetLabel() < 0)
-             trdtrack->SetLabel(-2);
            esd->AddTrdTrack(trdtrack);
            delete trdtrack;
        }