LHEGenericFilter.cc

Go to the documentation of this file.

#include "GeneratorInterface/GenFilters/interface/LHEGenericFilter.h"

using namespace edm;
using namespace std;

LHEGenericFilter::LHEGenericFilter(const edm::ParameterSet& iConfig) :
numRequired_(iConfig.getParameter<int>("NumRequired")),
acceptLogic_(iConfig.getParameter<std::string>("AcceptLogic")),
particleID_(iConfig.getParameter< std::vector<int> >("ParticleID")),
totalEvents_(0), passedEvents_(0)
{
  //here do whatever other initialization is needed
  src_ = consumes<LHEEventProduct>(iConfig.getParameter<edm::InputTag>("src"));
  
  if(acceptLogic_.compare("LT")==0) whichlogic = LT;
  else if(acceptLogic_.compare("GT")==0) whichlogic = GT;
  else if(acceptLogic_.compare("EQ")==0) whichlogic = EQ;
  else if(acceptLogic_.compare("NE")==0) whichlogic = NE;
  else edm::LogError  ("cat_A")   << "wrong input for AcceptLogic string";

  
}

LHEGenericFilter::~LHEGenericFilter()
{

  // do anything here that needs to be done at destruction time
  // (e.g. close files, deallocate resources etc.)

}


// ------------ method called to skim the data  ------------
bool LHEGenericFilter::filter(edm::Event& iEvent, const edm::EventSetup& iSetup)
{
  edm::Handle< LHEEventProduct > EvtHandle ;
  iEvent.getByToken( src_ , EvtHandle ) ;
  
  totalEvents_++;
  int nFound = 0;
  
  for (int i = 0; i < EvtHandle->hepeup().NUP; ++i) {
    if (EvtHandle->hepeup().ISTUP[i] != 1) { // keep only outgoing particles
      continue;
    }
    for (unsigned int j = 0; j < particleID_.size(); ++j) {
      if (particleID_[j] == 0 || abs(particleID_[j]) == abs(EvtHandle->hepeup().IDUP[i]) ) {
        nFound++;
        break; // only match a given particle once!
      }
    } // loop over targets
    
  } // loop over particles
  
  // event accept/reject logic
  if (
         (whichlogic==LT && nFound < numRequired_)
      || (whichlogic==GT && nFound > numRequired_)
      || (whichlogic==EQ && nFound == numRequired_)
      || (whichlogic==NE && nFound != numRequired_)
      ) {
    passedEvents_++;
    return true;
  } else {
    return false;
  }
  
}

// ------------ method called once each job just after ending the event loop  ------------
void LHEGenericFilter::endJob() {
  edm::LogInfo("LHEGenericFilter") << "=== Results of LHEGenericFilter: passed "
  << passedEvents_ << "/" << totalEvents_ << " events" << std::endl;
}

//define this as a plug-in
DEFINE_FWK_MODULE(LHEGenericFilter);