StringCutEventSelector.h

Go to the documentation of this file.

#ifndef _StringCutEventSelector
#define _StringCutEventSelector

#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "CommonTools/UtilAlgos/interface/EventSelector.h"
#include "CommonTools/UtilAlgos/interface/InputTagDistributor.h"
#include "CommonTools/Utils/interface/StringCutObjectSelector.h"
#include "FWCore/ServiceRegistry/interface/Service.h"

template<typename Object>
class  StringCutEventSelector : public EventSelector {
 public:
  StringCutEventSelector(const edm::ParameterSet& pset, edm::ConsumesCollector && iC) :
    StringCutEventSelector(pset, iC) {}
  StringCutEventSelector(const edm::ParameterSet& pset, edm::ConsumesCollector & iC) :
    EventSelector(pset, iC),
    src_(edm::Service<InputTagDistributorService>()->retrieve("src",pset)),
    srcToken_(iC.consumes<edm::View<Object> >(src_)),
    f_(pset.getParameter<std::string>("cut")),
    //put this guy to 0 to do the check on "all" object in the collection
    nFirst_(pset.getParameter<unsigned int>("nFirst"))
      {
      std::stringstream ss;
      ss<<"string cut based selection on collection: "<<src_;
      description_.push_back(ss.str());
      ss.str("");
      description_.push_back(std::string("selection cut is: ")+pset.getParameter<std::string>("cut"));
      }

    bool select (const edm::Event& e) const{
      edm::Handle<edm::View<Object> > oH;
      e.getByToken(srcToken_, oH);
      //reject events if not enough object in collection
      //      if ((nFirst_!=0) && (oH->size()<nFirst_)) return false;
      unsigned int i=0;
      for (;i!=oH->size();i++)
    {
      //stop doing the check if reaching too far in the collection
      if ((nFirst_!=0) && (i>=nFirst_)) break;
      const Object & o = (*oH)[i];
      if (!f_(o)) return false;
    }
      return true;
    }

 private:
    edm::InputTag src_;
    edm::EDGetTokenT<edm::View<Object> > srcToken_;
    StringCutObjectSelector<Object> f_;
    unsigned int nFirst_;
};


template<typename Object, bool existenceMatter=true>
class  StringCutsEventSelector : public EventSelector {
 public:
  StringCutsEventSelector(const edm::ParameterSet& pset, edm::ConsumesCollector && iC) :
    StringCutsEventSelector(pset, iC) {}
  StringCutsEventSelector(const edm::ParameterSet& pset, edm::ConsumesCollector & iC) :
    EventSelector(pset, iC),
    src_(edm::Service<InputTagDistributorService>()->retrieve("src",pset)),
    srcToken_(iC.consumes<edm::View<Object> >(src_))
      {
    std::vector<std::string> selection=pset.getParameter<std::vector<std::string > >("cut");
    std::stringstream ss;
    ss<<"string cut based selection on collection: "<<src_;
    description_.push_back(ss.str());       ss.str("");
    description_.push_back("selection cuts are:");
    for (unsigned int i=0;i!=selection.size();i++)
      if (selection[i]!="-"){
        f_.push_back( new StringCutObjectSelector<Object>(selection[i]));
        ss<<"["<<i<<"]: "<<selection[i];
        description_.push_back(ss.str());           ss.str("");
      }
      else
        {
          f_.push_back(0);
          ss<<"["<<i<<"]: no selection";
          description_.push_back(ss.str());           ss.str("");
        }
      }
 ~StringCutsEventSelector(){unsigned int i=0; for (;i!=f_.size();i++) if (f_[i]){ delete f_[i];f_[i]=0;}}

    bool select (const edm::Event& e) const{
      edm::Handle<edm::View<Object> > oH;
      e.getByToken(srcToken_, oH);
      unsigned int i=0;
      if (existenceMatter && oH->size()<f_.size()) return false;
      for (;i!=f_.size();i++)
    {
      if (!existenceMatter && i==oH->size()) break;
      if (!f_[i]) continue;
      const Object & o = (*oH)[i];
      if (!(*f_[i])(o)) return false;
    }
      return true;
    }

 private:
    edm::InputTag src_;
    edm::EDGetTokenT<edm::View<Object> > srcToken_;
    std::vector<StringCutObjectSelector<Object> *> f_;
};

#endif