adding charge independent correlation output (for comparison with other correlation...

[u/mrichter/AliRoot.git] / CORRFW / AliCFPairIsPrimaryCuts.cxx

diff --git a/CORRFW/AliCFPairIsPrimaryCuts.cxx b/CORRFW/AliCFPairIsPrimaryCuts.cxx
index e7db8213d8ad130f7c02e38879e28799190cde34..fbded3dc20e675720b1fa8f3fb36c519df6eb6cb 100755 (executable)
--- a/CORRFW/AliCFPairIsPrimaryCuts.cxx
+++ b/CORRFW/AliCFPairIsPrimaryCuts.cxx
@@ -76,17 +76,16 @@ Bool_t AliCFPairIsPrimaryCuts::IsSelected(TObject* obj) {
   // loops over decisions of single cuts and returns if the track is accepted
   //
 
-  if (!obj) return kFALSE ;
-  TString className(obj->ClassName());
+  AliCFPair* pair = dynamic_cast<AliCFPair*>(obj);
+  if (!pair) return kFALSE ;
+  TString className(pair->ClassName());
   if (className.CompareTo("AliCFPair") != 0) {
     Error("IsSelected","obj must point to a AliCFPair !");
     return kFALSE ;
   }
 
-  AliCFPair* pair = dynamic_cast<AliCFPair*>(obj);
-
-  AliESDtrack* tneg = pair->GetNeg();
-  AliESDtrack* tpos = pair->GetPos();
+  AliVParticle* tneg = pair->GetNeg();
+  AliVParticle* tpos = pair->GetPos();
   if (!tneg || !tpos) return kFALSE ;
 
   if ( ! fCutNeg->IsSelected((TObject*)tneg) || ! fCutPos->IsSelected((TObject*)tpos) ) return kFALSE ;