Possibility to keep only D mesons that have a c or b quark as a grandmother (Francesc...
[u/mrichter/AliRoot.git] / PWG3 / hfe / AliHFEcutStep.cxx
index bfed772..9eeb68a 100644 (file)
@@ -12,6 +12,9 @@
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/
+
+/* $Id$ */
+
 //
 // Cut step class
 // Select all tracks surviving cuts in one special cut step
@@ -91,8 +94,10 @@ Bool_t AliHFEcutStep::IsSelected(TObject *o){
   AliDebug(1, Form("Cut Step %s: Number of cut objects: %d", GetName(), fCuts->GetEntriesFast()));
   if(!fCuts->GetEntriesFast()) return kTRUE;
   Bool_t isSelected = kTRUE;
+  AliAnalysisCuts *cuts = NULL;
   for(Int_t iCut = 0; iCut < fCuts->GetEntriesFast(); iCut++){
-    if(!(dynamic_cast<AliAnalysisCuts *>(fCuts->UncheckedAt(iCut)))->IsSelected(o)) isSelected = kFALSE;
+    cuts = dynamic_cast<AliAnalysisCuts *>(fCuts->UncheckedAt(iCut));
+    if(cuts && (!cuts->IsSelected(o))) isSelected = kFALSE;
   }
   AliDebug(1, Form("Accepted: %s", isSelected ? "yes" : "no"));
   return isSelected;