RawParticleTypeFilter.cc

Go to the documentation of this file.

// -----------------------------------------------------------------------------
//  Prototype for a particle class
// -----------------------------------------------------------------------------
//  $Date: 2007/05/18 09:12:46 $
//  $Revision: 1.4 $
// -----------------------------------------------------------------------------
//  Author: Stephan Wynhoff - RWTH-Aachen (Email: Stephan.Wynhoff@cern.ch)
// -----------------------------------------------------------------------------
#include "FastSimulation/Particle/interface/RawParticleTypeFilter.h"

RawParticleTypeFilter::RawParticleTypeFilter(const std::string& particleName) {
  XYZTLorentzVector one;
  RawParticle tmp(particleName,one);
//    cout << tmp.pid() << endl;
  myAcceptIDs.push_back(tmp.pid());
}

RawParticleTypeFilter::RawParticleTypeFilter(const std::string& particleName1, 
                         const std::string& particleName2) {
  XYZTLorentzVector one;
  RawParticle tmp1(particleName1,one),tmp2(particleName2,one);
//    cout << tmp1.pid() << endl;
//    cout << tmp2.pid() << endl;
  myAcceptIDs.push_back(tmp1.pid());
  myAcceptIDs.push_back(tmp2.pid());
}

RawParticleTypeFilter::RawParticleTypeFilter(const int pid) {
  myAcceptIDs.push_back(pid);
}

RawParticleTypeFilter::RawParticleTypeFilter(const int pid1, const int pid2) {
  myAcceptIDs.push_back(pid1);
  myAcceptIDs.push_back(pid2);
}

void RawParticleTypeFilter::addAccept(const int pid) {
  myAcceptIDs.push_back(pid);
  myRejectIDs.clear();
}

void RawParticleTypeFilter::addAccept(const std::string& name) {
  XYZTLorentzVector one;
  RawParticle tmp(name,one);
  this->addAccept(tmp.pid());
}

void RawParticleTypeFilter::addReject(const int pid) {
  myRejectIDs.push_back(pid);
  myAcceptIDs.clear();
}

void RawParticleTypeFilter::addReject(const std::string& name) {
  XYZTLorentzVector one;
  RawParticle tmp(name,one);
  this->addReject(tmp.pid());
}

bool RawParticleTypeFilter::isOKForMe(const RawParticle *p) const
{
  bool acceptThis = true;
  if (myAcceptIDs.size() > 0) {
    acceptThis = this->isAcceptable(p->pid());
  } 
  if (myRejectIDs.size() > 0) {
    acceptThis = ! this->isRejectable(p->pid());
  }
  return acceptThis;
}

bool RawParticleTypeFilter::isAcceptable(const int id) const
{
  bool acceptThis = false;

  std::vector<int>::const_iterator myAcceptIDsItr;
  for (myAcceptIDsItr = myAcceptIDs.begin(); myAcceptIDsItr != myAcceptIDs.end();
       myAcceptIDsItr++) {
    if (id == (*myAcceptIDsItr)) {
      acceptThis = true;
    }
  }
  return acceptThis;
}

bool RawParticleTypeFilter::isRejectable(const int id) const
{
  bool acceptThis = false;

  std::vector<int>::const_iterator myRejectIDsItr;
  for (myRejectIDsItr = myRejectIDs.begin(); myRejectIDsItr != myRejectIDs.end();
       myRejectIDsItr++) {
    if (id == (*myRejectIDsItr)) {
      acceptThis = true;
    }
  }
  return acceptThis;
}