MCMultiParticleFilter.cc

Go to the documentation of this file.

#include "GeneratorInterface/GenFilters/interface/MCMultiParticleFilter.h"
#include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"

MCMultiParticleFilter::MCMultiParticleFilter(const edm::ParameterSet& iConfig) :
  src_(iConfig.getUntrackedParameter<edm::InputTag>("src",edm::InputTag(std::string("generator"),"unsmeared"))),
  token_(consumes<edm::HepMCProduct>(src_)),
  numRequired_(iConfig.getParameter<int>("NumRequired")),
  acceptMore_(iConfig.getParameter<bool>("AcceptMore")),
  particleID_(iConfig.getParameter< std::vector<int> >("ParticleID")),
  ptMin_(iConfig.getParameter< std::vector<double> >("PtMin")),
  etaMax_(iConfig.getParameter< std::vector<double> >("EtaMax")),
  status_(iConfig.getParameter< std::vector<int> >("Status")),
  totalEvents_(0), passedEvents_(0)
{
  //here do whatever other initialization is needed

  // default pt, eta, status cuts to "don't care"
  std::vector<double> defptmin(1, 0);
  std::vector<double> defetamax(1, 999.0);
  std::vector<int> defstat(1, 0);
  std::vector<int> defmother;
  defmother.push_back(0);
  motherID_ = iConfig.getUntrackedParameter< std::vector<int> >("MotherID", defmother);
    
  std::vector<double> defDecayRadiusmin;
  defDecayRadiusmin.push_back(-1.);
  decayRadiusMin = iConfig.getUntrackedParameter<std::vector<double> >("MinDecayRadius", defDecayRadiusmin);

  std::vector<double> defDecayRadiusmax;
  defDecayRadiusmax.push_back(1.e5);
  decayRadiusMax = iConfig.getUntrackedParameter<std::vector<double> >("MaxDecayRadius", defDecayRadiusmax);

  std::vector<double> defDecayZmin;
  defDecayZmin.push_back(-1.e5);
  decayZMin = iConfig.getUntrackedParameter<std::vector<double> >("MinDecayZ", defDecayZmin);

  std::vector<double> defDecayZmax;
  defDecayZmax.push_back(1.e5);
  decayZMax = iConfig.getUntrackedParameter<std::vector<double> >("MaxDecayZ", defDecayZmax);

  // check for same size
  if ( (ptMin_.size() > 1 &&  particleID_.size() != ptMin_.size()) 
       ||  (etaMax_.size() > 1 && particleID_.size() != etaMax_.size()) 
       ||  (status_.size() > 1 && particleID_.size() != status_.size()) 
       ||  (motherID_.size() > 1 && particleID_.size() != motherID_.size())
       ||  (decayRadiusMin.size() > 1 && particleID_.size() != decayRadiusMin.size())
       ||  (decayRadiusMax.size() > 1 && particleID_.size() != decayRadiusMax.size())
       ||  (decayZMin.size() > 1 && particleID_.size() != decayZMin.size())
       ||  (decayZMax.size() > 1 && particleID_.size() != decayZMax.size())
       ) {
    edm::LogWarning("MCMultiParticleFilter") << "WARNING: MCMultiParticleFilter: size of PtMin, EtaMax, motherID, decayRadiusMin, decayRadiusMax, decayZMin, decayZMax and/or Status does not match ParticleID size!" << std::endl;
  }
  
  // Fill arrays with defaults if necessary
  while (ptMin_.size() < particleID_.size())
    ptMin_.push_back(defptmin[0]);
  while (etaMax_.size() < particleID_.size())
    etaMax_.push_back(defetamax[0]);
  while (status_.size() < particleID_.size())
    status_.push_back(defstat[0]);
  while (motherID_.size() < particleID_.size())
    motherID_.push_back(defmother[0]);
    
  // if decayRadiusMin size smaller than particleID , fill up further with defaults
  if (particleID_.size() > decayRadiusMin.size()) {
    for (unsigned int i = decayRadiusMin.size(); i < particleID_.size(); i++) {
      decayRadiusMin.push_back(-10.);
    }
  }
  // if decayRadiusMax size smaller than particleID , fill up further with defaults
  if (particleID_.size() > decayRadiusMax.size()) {
    for (unsigned int i = decayRadiusMax.size(); i < particleID_.size(); i++) {
      decayRadiusMax.push_back(1.e5);
    }
  }

  // if decayZMin size smaller than particleID , fill up further with defaults
  if (particleID_.size() > decayZMin.size()) {
    for (unsigned int i = decayZMin.size(); i < particleID_.size(); i++) {
      decayZMin.push_back(-1.e5);
    }
  }
  // if decayZMax size smaller than particleID , fill up further with defaults
  if (particleID_.size() > decayZMax.size()) {
    for (unsigned int i = decayZMax.size(); i < particleID_.size(); i++) {
      decayZMax.push_back(1.e5);
    }
  }
}

MCMultiParticleFilter::~MCMultiParticleFilter()
{
 
   // 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 MCMultiParticleFilter::filter(edm::Event& iEvent, const edm::EventSetup& iSetup)
{
  edm::Handle<edm::HepMCProduct> evt;
  iEvent.getByToken(token_, evt);
  
  totalEvents_++;
  int nFound = 0;
  
  const HepMC::GenEvent * myGenEvent = evt->GetEvent();
  
  for ( HepMC::GenEvent::particle_const_iterator p = myGenEvent->particles_begin();
    p != myGenEvent->particles_end(); ++p ) {
    
    for (unsigned int i = 0; i < particleID_.size(); ++i) {
      if ((particleID_[i] == 0 || abs(particleID_[i]) == abs((*p)->pdg_id())) &&
      (*p)->momentum().perp() > ptMin_[i] &&
      fabs((*p)->momentum().eta()) < etaMax_[i] &&
      (status_[i] == 0 || (*p)->status() == status_[i])) {
    if (!((*p)->production_vertex()))
      continue;

    double decx = (*p)->production_vertex()->position().x();
    double decy = (*p)->production_vertex()->position().y();
    double decrad = sqrt(decx * decx + decy * decy);
    if (decrad < decayRadiusMin[i])
      continue;
    if (decrad > decayRadiusMax[i])
      continue;

    double decz = (*p)->production_vertex()->position().z();
    if (decz < decayZMin[i])
      continue;
    if (decz > decayZMax[i])
      continue;
          
    if(motherID_[i] == 0 ){ // do not check for mother ID if not sepcified
      nFound++;
      break; // only match a given particle once!
    }
    else{
      bool hascorrectmother=false;
      for ( HepMC::GenVertex::particles_in_const_iterator mo = (*p)->production_vertex()->particles_in_const_begin(); mo != (*p)->production_vertex()->particles_in_const_end(); ++mo){
        if( (*mo)->pdg_id() == motherID_[i]){
          hascorrectmother = true;
          break;
        }
      }
      if(hascorrectmother){
        nFound++;
        break; // only match a given particle once! 
      }
    }
      }
    } // loop over targets
    
    if (acceptMore_ && nFound == numRequired_) break; // stop looking if we don't mind having more
  } // loop over particles
  
  if (nFound == numRequired_) {
    passedEvents_++;
    return true;
  } else {
    return false;
  }
  
}

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

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