QA histograms filling for missing steps

[u/mrichter/AliRoot.git] / CORRFW / test / AliCFSingleTrackTask.cxx

diff --git a/CORRFW/test/AliCFSingleTrackTask.cxx b/CORRFW/test/AliCFSingleTrackTask.cxx
index 12828d263eaa69f1f007093e66438da1e3226336..895d12e19fc0271a1cb2740c6509ad29dc9b8746 100644 (file)
--- a/CORRFW/test/AliCFSingleTrackTask.cxx
+++ b/CORRFW/test/AliCFSingleTrackTask.cxx
@@ -156,7 +156,9 @@ void AliCFSingleTrackTask::Exec(Option_t *)
     AliMCParticle *mcPart  = mcEvent->GetTrack(ipart);
 
     //check the MC-level cuts
+    fCFManager->FillQABeforeParticleCuts(AliCFManager::kPartGenCuts,mcPart);  
     if (!fCFManager->CheckParticleCuts(AliCFManager::kPartGenCuts,mcPart)) continue;
+    fCFManager->FillQAAfterParticleCuts(AliCFManager::kPartGenCuts,mcPart);  
 
     containerInput[0] = (Float_t)mcPart->Pt();
     containerInput[1] = mcPart->Eta() ;
@@ -164,7 +166,9 @@ void AliCFSingleTrackTask::Exec(Option_t *)
     fCFManager->GetParticleContainer()->Fill(containerInput,kStepGenerated);
     
     //check the Acceptance-level cuts
+    fCFManager->FillQABeforeParticleCuts(AliCFManager::kPartAccCuts,mcPart);  
     if (!fCFManager->CheckParticleCuts(AliCFManager::kPartAccCuts,mcPart)) continue;
+    fCFManager->FillQAAfterParticleCuts(AliCFManager::kPartAccCuts,mcPart);  
     //fill the container for Acceptance-level selection
     fCFManager->GetParticleContainer()->Fill(containerInput,kStepReconstructible);
   }    
@@ -192,7 +196,12 @@ void AliCFSingleTrackTask::Exec(Option_t *)
     containerInput[0] = pt ;
     containerInput[1] = track->Eta();
     fCFManager->GetParticleContainer()->Fill(containerInput,kStepReconstructed) ;   
+
+
+    fCFManager->FillQABeforeParticleCuts(AliCFManager::kPartSelCuts,track);  
     if (!fCFManager->CheckParticleCuts(AliCFManager::kPartSelCuts,track)) continue ;
+    fCFManager->FillQAAfterParticleCuts(AliCFManager::kPartSelCuts,track);  
+
     fCFManager->GetParticleContainer()->Fill(containerInput,kStepSelected);
   }