- update label handling in GTU simulation

[u/mrichter/AliRoot.git] / TRD / AliTRDgtuTMU.h

diff --git a/TRD/AliTRDgtuTMU.h b/TRD/AliTRDgtuTMU.h
index 0aeb420835c795d04562f9a7f73771924e8e881c..8310b093c4204d5c83992c6f315d4b2aa6898e98 100644 (file)
--- a/TRD/AliTRDgtuTMU.h
+++ b/TRD/AliTRDgtuTMU.h
@@ -30,8 +30,7 @@ class AliTRDgtuTMU : public TObject {
   Bool_t SetSector(Int_t sector);
   Bool_t SetStack(Int_t stack);
 
-  Bool_t AddTracklet(AliTRDtrackletBase *tracklet, Int_t link);
-  Bool_t WriteTrackletsToTree(TTree *trklTree);
+  Bool_t AddTracklet(AliTRDtrackletGTU *tracklet, Int_t link);
 
   Bool_t RunTMU(TList *ListOfTracks = 0x0, AliESDEvent *esd = 0x0);
   Bool_t Reset();
@@ -40,16 +39,19 @@ class AliTRDgtuTMU : public TObject {
   Bool_t RunInputUnit(Int_t layer);
   Bool_t RunZChannelUnit(Int_t layer);
   Bool_t RunTrackFinder(Int_t zchannel, TList* ListOfTracks);
-  Bool_t RunTrackMerging(TList* ListOfTracks); 
+  Bool_t RunTrackMerging(TList* ListOfTracks);
   Bool_t RunTrackReconstruction(TList* ListOfTracks);
 
-  Bool_t CalculateTrackParams(AliTRDtrackGTU *track); 
-  Bool_t Uniquifier(TList* inlist, TList *outlist);
+  Bool_t CalculateTrackParams(AliTRDtrackGTU *track);
+  Bool_t Uniquifier(const TList* inlist, TList *outlist);
+  Bool_t CalculatePID(AliTRDtrackGTU *track);
 
 protected:
-  TObjArray **fTracklets; // holding all tracklets from one detector (i. e. one chamber)
+  TObjArray **fTracklets; // holding all tracklets per link
+  TObjArray **fTrackletsPostInput; // holding all tracklets of a layer
+                                  // after sorting/calculation in input units
   TList **fZChannelTracklets; // holding all tracklets for layer and z-channel
-  TList **fTracks;
+  TList **fTracks; // lists of tracks
   AliTRDgtuParam *fGtuParam; // pointer to the instance of the GtuParam class
 
   Int_t fStack;                        // Stack of this TMU