Updates

[u/mrichter/AliRoot.git] / ANALYSIS / AliKineTrackCuts.cxx

diff --git a/ANALYSIS/AliKineTrackCuts.cxx b/ANALYSIS/AliKineTrackCuts.cxx
index 6112621f392ac91f407bb5f8118ce4624e424614..60298ab4df912b0a31e6def8862cd7b4a0e5c167 100644 (file)
--- a/ANALYSIS/AliKineTrackCuts.cxx
+++ b/ANALYSIS/AliKineTrackCuts.cxx
@@ -32,6 +32,8 @@ ClassImp(AliKineTrackCuts)
 //____________________________________________________________________
 AliKineTrackCuts::AliKineTrackCuts(const Char_t* name, const Char_t* title) : 
   AliAnalysisCuts(name,title),
+  fOnlyFinalParticles(kFALSE),
+  fOnlyPrimary(kFALSE), 
   fPMin(0),
   fPMax(0),
   fPtMin(0),
@@ -68,6 +70,10 @@ Bool_t  AliKineTrackCuts::IsSelected(TObject* obj)
 
   TParticle * part = (TParticle *)obj;
   
+  // only final particles
+  if( fOnlyFinalParticles && part->GetStatusCode() !=1 ) return kFALSE;
+  if( fOnlyPrimary && part->IsPrimary() !=1 ) return kFALSE;
+  
   // getting the kinematic variables of the track
   Float_t momentum = part->P();
   Float_t pt       = part->Pt();