Add track propagation (Marta)

[u/mrichter/AliRoot.git] / PWG / EMCAL / AliEmcalEsdTpcTrackTask.h

diff --git a/PWG/EMCAL/AliEmcalEsdTpcTrackTask.h b/PWG/EMCAL/AliEmcalEsdTpcTrackTask.h
index 184e60e5d8331d89ba110b0314da50a90a884c6c..c5ba40583b3150d5bbf3d243d625425dd660501c 100644 (file)
--- a/PWG/EMCAL/AliEmcalEsdTpcTrackTask.h
+++ b/PWG/EMCAL/AliEmcalEsdTpcTrackTask.h
@@ -25,12 +25,17 @@ class AliEmcalEsdTpcTrackTask : public AliAnalysisTaskSE {
   void SetTracksName(const char *name)           { fTracksName       = name; }
   void SetIncludeNoITS(Bool_t f)                 { fIncludeNoITS     = f;    }
 
+  void SetDoPropagation(Bool_t b)                { fDoPropagation    = b;    }
+
  protected:
+  void PropagateTrackToEMCal(AliESDtrack *esdTrack);
+
   AliESDtrackCuts   *fEsdTrackCuts;      // esd track cuts
   Bool_t             fDoSpdVtxCon;       // if true then do vertex constraint
   AliESDtrackCuts   *fHybridTrackCuts;   // hybrid track cuts
   TString            fTracksName;        // name of tracks 
   Bool_t             fIncludeNoITS;      // includes tracks with failed ITS refit
+  Bool_t             fDoPropagation;     // propagate all hybrid tracks to EMCal surface
   AliESDEvent       *fEsdEv;             //!esd event
   TClonesArray      *fTracks;            //!track array