update track selection QA (M. Verweij)

[u/mrichter/AliRoot.git] / PWG4 / JetTasks / AliPWG4HighPtSpectra.cxx

diff --git a/PWG4/JetTasks/AliPWG4HighPtSpectra.cxx b/PWG4/JetTasks/AliPWG4HighPtSpectra.cxx
index fd7cafc149674be7abdc246b5826e934019bac80..da5822d5ecfc1be303659a16cb563f0b3a791473 100644 (file)
--- a/PWG4/JetTasks/AliPWG4HighPtSpectra.cxx
+++ b/PWG4/JetTasks/AliPWG4HighPtSpectra.cxx
@@ -75,6 +75,7 @@ AliPWG4HighPtSpectra::AliPWG4HighPtSpectra() : AliAnalysisTask("AliPWG4HighPtSpe
   fCentClass(10),
   fTrackType(0),
   fTrackCuts(0x0),
+  fTrackCutsReject(0x0),
   fSigmaConstrainedMax(5.),
   fAvgTrials(1),
   fHistList(0),
@@ -105,6 +106,7 @@ AliPWG4HighPtSpectra::AliPWG4HighPtSpectra(const Char_t* name) :
   fCentClass(10),
   fTrackType(0),
   fTrackCuts(0x0),
+  fTrackCutsReject(0x0),
   fSigmaConstrainedMax(5.),
   fAvgTrials(1),
   fHistList(0),
@@ -397,17 +399,20 @@ void AliPWG4HighPtSpectra::Exec(Option_t *)
        }
       }
 
-
-   
       if (fTrackCuts->AcceptTrack(track)) {
 
        if(fTrackType==7) {
+         if(fTrackCutsReject ) {
+           if(fTrackCutsReject->AcceptTrack(track) )
+             continue;
+         }
+         
          if(esdtrack->GetConstrainedParam()) 
            track->Set(esdtrack->GetConstrainedParam()->GetX(),esdtrack->GetConstrainedParam()->GetAlpha(),esdtrack->GetConstrainedParam()->GetParameter(),esdtrack->GetConstrainedParam()->GetCovariance());
-         else
-           continue;
        }
 
+       if(!track) continue;
+
        //fill the container
        containerInputRec[0] = track->Pt();
        containerInputRec[1] = track->Phi();