replacing calls to AliCDBManager::GetId after this has become privat in r57556

[u/mrichter/AliRoot.git] / HLT / JET / AliHLTJETTrackCuts.cxx

diff --git a/HLT/JET/AliHLTJETTrackCuts.cxx b/HLT/JET/AliHLTJETTrackCuts.cxx
index 0c4acbefe22a64dfdfa67bbb55512ec7da83e648..7f048a6c8c98afe68b5b7d0d121d59713adf8afa 100644 (file)
--- a/HLT/JET/AliHLTJETTrackCuts.cxx
+++ b/HLT/JET/AliHLTJETTrackCuts.cxx
@@ -33,6 +33,7 @@ using namespace std;
 #endif
 
 #include "AliHLTJETTrackCuts.h"
+#include <TDatabasePDG.h>
 
 /** ROOT macro for the implementation of ROOT specific class methods */
 ClassImp(AliHLTJETTrackCuts)
@@ -47,7 +48,12 @@ ClassImp(AliHLTJETTrackCuts)
 AliHLTJETTrackCuts::AliHLTJETTrackCuts(const Char_t* name, const Char_t* title )
   : 
   AliAnalysisCuts(name, title),
-  fPtMin(0.) {
+  fChargedOnly(kFALSE),
+  fPtMin(0.0),
+  fEtaMin(-0.9),
+  fEtaMax(0.9),
+  fPhiMin(0.0),
+  fPhiMax(6.3) {
   // see header file for class documentation
   // or
   // refer to README to build package
@@ -73,11 +79,11 @@ Bool_t AliHLTJETTrackCuts::IsSelected( TObject *obj ) {
   // see header file for class documentation
 
   Bool_t bResult = kTRUE;
-
-  if ( ! strcmp(obj->ClassName(),"TParticle") )
-    bResult = IsSelected( dynamic_cast<TParticle*> (obj));
-  else if ( ! strcmp(obj->ClassName(),"AliESDtrack") )
-    bResult = IsSelected( dynamic_cast<AliESDtrack*> (obj));
+  
+  if ( obj->IsA() == TParticle::Class() )
+    bResult = IsSelected( static_cast<TParticle*> (obj));
+  else if ( obj->IsA() == AliESDtrack::Class() )
+    bResult = IsSelected( static_cast<AliESDtrack*> (obj));
   else {
     HLTError("Unknown object type %s", obj->ClassName() );
     bResult = kFALSE;
@@ -92,31 +98,63 @@ Bool_t AliHLTJETTrackCuts::IsSelected( TObject *obj ) {
 Bool_t AliHLTJETTrackCuts::IsSelected( TParticle *particle ) {
   // see header file for class documentation
 
+  // ----------------------------------
+  //  Applied before in AliHLTMCEvent:
+  //  - Is Physical Primary  
+  //   stack->IsPhysicalPrimary(iter)
+  //  - final state
+  //   particle->GetNDaughters() == 0
+  // ----------------------------------
+
   Bool_t bResult = kTRUE;
 
   Int_t   status  = particle->GetStatusCode();
-  Int_t   pdg     = TMath::Abs( particle->GetPdgCode() );
+  Int_t   pdgCode = TMath::Abs( particle->GetPdgCode() );
   
   // -- Skip non-final state particles (status != 1), neutrinos (12,14,16)
-  if ( (status != 1) || (pdg == 12 || pdg == 14 || pdg == 16) )
+  if ( (status != 1) || (pdgCode == 12 || pdgCode == 14 || pdgCode == 16) )
+    bResult = kFALSE;
+
+  // -- Charged particles only
+  if ( ! particle->GetPDG() )
+    bResult = kFALSE;
+  else {
+    if ( fChargedOnly && !particle->GetPDG()->Charge() )
+      bResult = kFALSE;
+  }
+
+  // -- cut on min Pt
+  if ( particle->Pt() < fPtMin )
+    bResult = kFALSE;
+
+  // -- cut on eta acceptance
+  if ( ( particle->Eta() < fEtaMin ) || ( particle->Eta() > fEtaMax ) )
+    bResult = kFALSE;
+
+  // -- cut on phi acceptance
+  if ( ( particle->Phi() < fPhiMin ) || ( particle->Phi() > fPhiMax ) )
     bResult = kFALSE;
-  else
-    HLTInfo("Is selected !");
 
   return bResult;
 }
 
 // #################################################################################
-Bool_t AliHLTJETTrackCuts::IsSelected( AliESDtrack */*esdTrack*/ ) {
+Bool_t AliHLTJETTrackCuts::IsSelected( AliESDtrack *esdTrack ) {
   // see header file for class documentation
 
   Bool_t bResult = kTRUE;
 
+  // -- cut on min Pt
+  if ( esdTrack->Pt() < fPtMin )
+    bResult = kFALSE;
+
+  // -- cut on eta acceptance
+  if ( ( esdTrack->Eta() < fEtaMin ) || ( esdTrack->Eta() > fEtaMax ) )
+    bResult = kFALSE;
+
+  // -- cut on phi acceptance
+  if ( ( esdTrack->Phi() < fPhiMin ) || ( esdTrack->Phi() > fPhiMax ) )
+    bResult = kFALSE;
+
   return bResult;
 }
-
-/*
- * ---------------------------------------------------------------------------------
- *                                     Setter
- * ---------------------------------------------------------------------------------
- */