AnalysisSelection.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ANALYSISUTILS_ANALYSISSELECTION_H
#define ANALYSISUTILS_ANALYSISSELECTION_H

 
#include <vector>
 
namespace AnalysisUtils {
 
  template <class CRITERIA> struct Selection : public CRITERIA
  {
    
    template <class COLL> void getObjects(const COLL *coll, std::vector<typename COLL::value_type> &out)
    {
      out.clear();
 
      // loop over all elements
      typename COLL::const_iterator it  = coll->begin();
      typename COLL::const_iterator itE = coll->end();
      for (; it != itE; ++it)
    // selection
    if (CRITERIA::isAccepted(*it))
      out.push_back(*it);
    }

    template <class COLL> void getIndices(const COLL *coll, std::vector<int> &out)
    {
      out.clear();
 
      // loop over all elements
      int index = 0;
      typename COLL::const_iterator it  = coll->begin();
      typename COLL::const_iterator itE = coll->end();
      for (; it != itE; ++it)
    {
      // selection
      if (CRITERIA::isAccepted(*it))
        out.push_back(index);
      
      ++index;
    }
    }
  };
 
} // end of AnalysisUtils namespace
 
#endif