Update from Leonardo:

[u/mrichter/AliRoot.git] / PWGCF / Correlations / Base / AliAnalyseLeadingTrackUE.h

diff --git a/PWGCF/Correlations/Base/AliAnalyseLeadingTrackUE.h b/PWGCF/Correlations/Base/AliAnalyseLeadingTrackUE.h
index b44bcac1794be168f7b03d7dd8683704e6568f51..335d8b9170aaa7904737cf2748010d0d90771ba7 100644 (file)
--- a/PWGCF/Correlations/Base/AliAnalyseLeadingTrackUE.h
+++ b/PWGCF/Correlations/Base/AliAnalyseLeadingTrackUE.h
@@ -29,6 +29,8 @@ class TClonesArray;
 class TObject;
 class TROOT;
 class TVector3;
+class AliVTrack;
+class AliHelperPID;
 
 class AliAnalyseLeadingTrackUE : public TObject {
 
@@ -44,6 +46,7 @@ class AliAnalyseLeadingTrackUE : public TObject {
                                                                                             fOnlyHadrons = onlyhadrons;
                                                                                             fTrackEtaCut = etacut;     
                                                                                             fTrackPtMin = ptMin;}
+  void SetEventSelection(UInt_t bits) { fEventSelection = bits; }
   void  SetDebug(Int_t debug) { fDebug = debug; } 
   Bool_t         ApplyCuts(TObject* track);                       // Reproduces the cuts of the corresponding bit in the ESD->AOD filtering
   void           DefineESDCuts(Int_t filterbit);                                 // Emulate filterbit
@@ -54,18 +57,26 @@ class AliAnalyseLeadingTrackUE : public TObject {
   void          QSortTracks(TObjArray &a, Int_t first, Int_t last);               // Sort by pT an array of AliVParticles 
   TObjArray*     SortRegions(const AliVParticle* leading, TObject* obj, TObject* arrayMC, Bool_t onlyprimaries = kTRUE); // Assign particles to towards, away or transverse regions
   TObjArray*     GetAcceptedParticles(TObject* obj, TObject* arrayMC, Bool_t onlyprimaries = kTRUE, Int_t particleSpecies = -1, Bool_t useEtaPtCuts = kFALSE); 
+  TObjArray*     GetFakeParticles(TObject* obj, TObject* arrayMC, Bool_t onlyprimaries, Int_t particleSpecies, Bool_t useEtaPtCuts);
   Bool_t         TriggerSelection(const TObject* obj);                                   // Select good triggers with AliPhysicsSelection class
   Bool_t         VertexSelection(const TObject* obj, Int_t ntracks, Double_t zed);       // Vertex selection: see implementation
-
- private:
+  void                  RemoveInjectedSignals(TObjArray* tracks, TObject* arrayMC, Int_t maxLabel);
+  void                  RemoveWeakDecays(TObjArray* tracks, TObject* mcObj);
+  AliHelperPID*  GetHelperPID() { return fHelperPID; }
+  void                  SetHelperPID(AliHelperPID* pid) { fHelperPID = pid; }
+  
+private:
   Int_t          fDebug;             // debug flag
   Int_t          fFilterBit;         // track selection cuts
   Bool_t         fOnlyHadrons;       // consider only charged Pions, Protons and Kaons 
   Double_t       fTrackEtaCut;       // pseudo-rapidity limit of transverse regions     
   Double_t       fTrackPtMin;        // pt limit for selecting particles
+  UInt_t         fEventSelection;    // bit for physics selection
   AliESDtrackCuts *fEsdTrackCuts;    // set of cuts when reading ESD
-  AliESDtrackCuts *fEsdTrackCutsSPD;    // set of cuts when reading ESD
-  AliESDtrackCuts *fEsdTrackCutsSDD;    // set of cuts when reading ESD
+  AliESDtrackCuts *fEsdTrackCutsExtra1;    // set of cuts when reading ESD
+  AliESDtrackCuts *fEsdTrackCutsExtra2;    // set of cuts when reading ESD
+  AliHelperPID   *fHelperPID;    // PID Helper object
+  
   ClassDef(AliAnalyseLeadingTrackUE,0)
 };
 #endif