ALIROOT-5488 Remove build/include from the include directories
[u/mrichter/AliRoot.git] / TRD / AliTRDgtuTMU.h
index 93fd2b2..9875eb6 100644 (file)
 //--------------------------------------------------------------------
 
 #include "TObject.h"
-#include "TList.h"
 
 #include "AliTRDtrackletGTU.h"
 #include "AliTRDgtuParam.h"
 
 class TTree;
+class TList;
+class TClonesArray;
 class TBranch;
 class AliTRDtrackGTU;
 class AliESDEvent;
@@ -30,25 +31,28 @@ 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 RunTMU(TList *ListOfTracks = 0x0, AliESDEvent *esd = 0x0, Int_t outLabel = -1);
   Bool_t Reset();
 
   // ----- successive stages of the processing in the TMU -----
   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
+  TClonesArray *fTrackArray; // array of tracks
   TList **fTracks; // lists of tracks
   AliTRDgtuParam *fGtuParam; // pointer to the instance of the GtuParam class