X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=PWG%2FEMCAL%2FAliEmcalAodTrackFilterTask.h;h=467d35c897096c88067d659237415dc233f61b1f;hb=322f926b339b38b3d10e017c886be48b00d2fe1b;hp=27e0bacaa98e3801c40d8499fde9b2516ed38448;hpb=fe75ca27678973b28a76cf2b5900928bf7185122;p=u%2Fmrichter%2FAliRoot.git

diff --git a/PWG/EMCAL/AliEmcalAodTrackFilterTask.h b/PWG/EMCAL/AliEmcalAodTrackFilterTask.h
index 27e0bacaa98..467d35c8970 100644
--- a/PWG/EMCAL/AliEmcalAodTrackFilterTask.h
+++ b/PWG/EMCAL/AliEmcalAodTrackFilterTask.h
@@ -4,8 +4,6 @@
 // $Id$
 
 class TClonesArray;
-class AliVEvent;
-class AliVTrack;
 
 #include "AliAnalysisTaskSE.h"
 
@@ -15,42 +13,35 @@ class AliEmcalAodTrackFilterTask : public AliAnalysisTaskSE {
   AliEmcalAodTrackFilterTask(const char *name);
   virtual ~AliEmcalAodTrackFilterTask();
 
-  void UserCreateOutputObjects();
-  void UserExec(Option_t *option);
-
-  void SetAODfilterBits(Int_t b0 = 0, Int_t b1 = 0)         { fAODfilterBits[0]  = b0  ; fAODfilterBits[1] = b1  ; }
-  void SetCutMaxFractionSharedTPCClusters(Double_t c = 0.4) { fCutMaxFrShTPCClus = c   ; }
-  void SetDist(Double_t d)                                  { fDist              = d   ; }
-  void SetDoPropagation(Bool_t b)                           { fDoPropagation     = b   ; }
-  void SetIncludeNoITS(Bool_t f)                            { fIncludeNoITS      = f   ; }
-  void SetMC(Bool_t a)                                      { fIsMC              = a   ; }
-  void SetModifyTrack(Bool_t b)                             { fModifyTrack       = b;    }
-  void SetTrackEfficiency(Double_t eff = 0.95)              { fTrackEfficiency   = eff ; }
-  void SetTrackEtaLimits(Double_t min, Double_t max)        { fMaxTrackEta       = max ; fMinTrackEta      = min ; }
-  void SetTrackPhiLimits(Double_t min, Double_t max)        { fMaxTrackPhi       = max ; fMinTrackPhi      = min ; }
-  void SetTrackPtLimits(Double_t min, Double_t max)         { fMaxTrackPt        = max ; fMinTrackPt       = min ; }
-  void SetTracksInName(const char *name)                    { fTracksInName      = name; }
-  void SetTracksOutName(const char *name)                   { fTracksOutName     = name; }
-  void SetUseNegativeLabels(Bool_t f)                       { fUseNegativeLabels = f   ; }
+  void               SetAODfilterBits(Int_t b0 = 0, Int_t b1 = 0)         { fAODfilterBits[0]  = b0  ; fAODfilterBits[1] = b1  ; }
+  void               SetAttemptProp(Bool_t b)                             { fAttemptProp       = b   ; }
+  void               SetAttemptPropMatch(Bool_t b)                        { fAttemptPropMatch  = b   ; }
+  void               SetCutMaxFractionSharedTPCClusters(Double_t c = 0.4) { fCutMaxFrShTPCClus = c   ; }
+  void               SetDist(Double_t d)                                  { fDist              = d   ; }
+  void               SetDoPropagation(Bool_t b)                           { fDoPropagation     = b   ; }
+  void               SetIncludeNoITS(Bool_t f)                            { fIncludeNoITS      = f   ; }
+  void               SetMC(Bool_t b)                                      { fIsMC              = b   ; }
+  void               SetTracksInName(const char *name)                    { fTracksInName      = name; }
+  void               SetTracksOutName(const char *name)                   { fTracksOutName     = name; }
+  void               SetUseNegativeLabels(Bool_t f)                       { fUseNegativeLabels = f   ; }
+  void               SetTrackEfficiency(Double_t eff = 0.95)              { fTrackEfficiency   = eff ; }
 
  protected:
+  void               UserCreateOutputObjects();
+  void               UserExec(Option_t *option);
+
   Int_t              fAODfilterBits[2];     // AOD track filter bit map
   TString            fTracksOutName;        // name of output track array
   TString            fTracksInName;         // name of input track array
-  Double_t           fMinTrackPt;           // mix pt of tracks
-  Double_t           fMaxTrackPt;           // max pt of tracks
-  Double_t           fMinTrackEta;          // cut on track eta
-  Double_t           fMaxTrackEta;          // cut on track eta
-  Double_t           fMinTrackPhi;          // cut on track phi
-  Double_t           fMaxTrackPhi;          // cut on track phi
-  Double_t           fTrackEfficiency;      // track efficiency
   Bool_t             fIncludeNoITS;         // includes tracks with failed ITS refit
+  Double_t           fCutMaxFrShTPCClus;    // max fraction of shared TPC clusters
   Bool_t             fUseNegativeLabels;    // whether or not should use negative MC labels
   Bool_t             fIsMC;                 // whether it is a MC event or not
-  Double_t           fCutMaxFrShTPCClus;    // max fraction of shared TPC clusters
-  Bool_t             fModifyTrack;          // if true then overwrite some fields in AodTrack
   Bool_t             fDoPropagation;        // if true then propagate all hybrid tracks to EMCal surface
+  Bool_t             fAttemptProp;          // if true then attempt to propagate if not done yet
+  Bool_t             fAttemptPropMatch;     // if true then attempt to propagate if not done yet but IsEMCAL is true
   Double_t           fDist;                 // distance to surface (440cm default)
+  Double_t           fTrackEfficiency;      // track efficiency
   TClonesArray      *fTracksIn;             //!track array in
   TClonesArray      *fTracksOut;            //!track array out
 
@@ -58,6 +49,6 @@ class AliEmcalAodTrackFilterTask : public AliAnalysisTaskSE {
   AliEmcalAodTrackFilterTask(const AliEmcalAodTrackFilterTask&);            // not implemented
   AliEmcalAodTrackFilterTask &operator=(const AliEmcalAodTrackFilterTask&); // not implemented
 
-  ClassDef(AliEmcalAodTrackFilterTask, 1); // Task to filter Aod tracks
+  ClassDef(AliEmcalAodTrackFilterTask, 3); // Task to filter Aod tracks
 };
 #endif