InputGenJetsParticleSelector Class Reference

#include <InputGenJetsParticleSelector.h>

Inheritance diagram for InputGenJetsParticleSelector:

Public Types
typedef std::vector< bool >	ParticleBitmap
typedef std::vector< const reco::GenParticle * >	ParticleVector
enum	ResonanceState { kNo = 0, kDirect, kIndirect }
Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType
typedef WorkerT< EDProducer >	WorkerType
Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Public Member Functions
ResonanceState	fromResonance (ParticleBitmap &invalid, const ParticleVector &p, const reco::GenParticle *particle) const
bool	getExcludeResonances () const
const std::vector< unsigned int > &	getIgnoredParticles () const
bool	getPartonicFinalState () const
double	getPtMin () const
bool	getTausAndJets () const
bool	hasPartonChildren (ParticleBitmap &invalid, const ParticleVector &p, const reco::GenParticle *particle) const
	InputGenJetsParticleSelector (const edm::ParameterSet &)
bool	isIgnored (int pdgId) const
bool	isParton (int pdgId) const
virtual void	produce (edm::Event &evt, const edm::EventSetup &evtSetup)
void	setExcludeResonances (bool flag=true)
void	setPartonicFinalState (bool flag=true)
void	setPtMin (double ptMin)
void	setTausAsJets (bool flag=true)
	~InputGenJetsParticleSelector ()
Public Member Functions inherited from edm::EDProducer
	EDProducer ()
virtual	~EDProducer ()
Public Member Functions inherited from edm::ProducerBase
	ProducerBase ()
void	registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
boost::function< void(const BranchDescription &)>	registrationCallback () const
	used by the fwk to register list of products More...
virtual	~ProducerBase ()

Static Public Member Functions
static bool	isHadron (int pdgId)
static bool	isResonance (int pdgId)
Static Public Member Functions inherited from edm::EDProducer
static const std::string &	baseType ()
static void	fillDescriptions (ConfigurationDescriptions &descriptions)
static void	prevalidate (ConfigurationDescriptions &descriptions)

Private Member Functions
	InputGenJetsParticleSelector ()
bool	isExcludedFromResonance (int pdgId) const
void	setExcludeFromResonancePids (const std::vector< unsigned int > &particleIDs)
void	setIgnoredParticles (const std::vector< unsigned int > &particleIDs)
int	testPartonChildren (ParticleBitmap &invalid, const ParticleVector &p, const reco::GenParticle *particle) const

Private Attributes
std::vector< unsigned int >	excludeFromResonancePids
bool	excludeResonances
std::vector< unsigned int >	ignoreParticleIDs
edm::InputTag	inTag
bool	partonicFinalState
double	ptMin
bool	tausAsJets

Additional Inherited Members
Protected Member Functions inherited from edm::EDProducer
CurrentProcessingContext const *	currentContext () const
Protected Member Functions inherited from edm::ProducerBase
template<class TProducer , class TMethod >
void	callWhenNewProductsRegistered (TProducer *iProd, TMethod iMethod)

Detailed Description

Definition at line 28 of file InputGenJetsParticleSelector.h.

Member Typedef Documentation

typedef std::vector<bool> InputGenJetsParticleSelector::ParticleBitmap

Definition at line 31 of file InputGenJetsParticleSelector.h.

typedef std::vector<const reco::GenParticle*> InputGenJetsParticleSelector::ParticleVector

Definition at line 32 of file InputGenJetsParticleSelector.h.

Member Enumeration Documentation

enum InputGenJetsParticleSelector::ResonanceState

Enumerator
kNo
kDirect
kIndirect

Definition at line 62 of file InputGenJetsParticleSelector.h.

                      {
    kNo = 0,
    kDirect,
    kIndirect
  };

Constructor & Destructor Documentation

InputGenJetsParticleSelector::InputGenJetsParticleSelector

(

const edm::ParameterSet &

params

)

Definition at line 42 of file InputGenJetsParticleSelector.cc.

References edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), setExcludeFromResonancePids(), and setIgnoredParticles().

                                                                                        :
  inTag(params.getParameter<edm::InputTag>("src")),
  partonicFinalState(params.getParameter<bool>("partonicFinalState")),
  excludeResonances(params.getParameter<bool>("excludeResonances")),
  tausAsJets(params.getParameter<bool>("tausAsJets")),
  ptMin(0.0){
  if (params.exists("ignoreParticleIDs"))
    setIgnoredParticles(params.getParameter<std::vector<unsigned int> >
            ("ignoreParticleIDs"));
  setExcludeFromResonancePids(params.getParameter<std::vector<unsigned int> >
            ("excludeFromResonancePids"));

  produces <reco::GenParticleRefVector> ();
      
}

InputGenJetsParticleSelector::~InputGenJetsParticleSelector

(

)

Definition at line 58 of file InputGenJetsParticleSelector.cc.

{}

InputGenJetsParticleSelector::InputGenJetsParticleSelector ( )

inlineprivate

Definition at line 76 of file InputGenJetsParticleSelector.h.

{} //should not be used!

Member Function Documentation

InputGenJetsParticleSelector::ResonanceState InputGenJetsParticleSelector::fromResonance	(	ParticleBitmap &	invalid,
		const ParticleVector &	p,
		const reco::GenParticle *	particle
	)		const

Definition at line 172 of file InputGenJetsParticleSelector.cc.

References i, isExcludedFromResonance(), isIgnored(), isParton(), isResonance(), kDirect, kIndirect, kNo, reco::CompositeRefCandidateT< D >::mother(), reco::CompositeRefCandidateT< D >::numberOfMothers(), partIdx(), benchmark_cfg::pdgId, reco::LeafCandidate::pdgId(), query::result, and reco::LeafCandidate::status().

Referenced by produce().

{
  unsigned int idx = partIdx(p, particle);
  int id = particle->pdgId();

  if (invalid[idx]) return kIndirect;
      
  if (isResonance(id) && particle->status() == 3){
    return kDirect;
  }

  
  if (!isIgnored(id) && (isParton(id)))
    return kNo;
  
  
  
  unsigned int nMo=particle->numberOfMothers();
  if (!nMo)
    return kNo;
  

  for(unsigned int i=0;i<nMo;++i){
    ResonanceState result = fromResonance(invalid,p,dynamic_cast<const reco::GenParticle*>(particle->mother(i)));
    switch(result) {
    case kNo:
      break;
    case kDirect:
      if (dynamic_cast<const reco::GenParticle*>(particle->mother(i))->pdgId()==id || isResonance(id))
    return kDirect;
      if(!isExcludedFromResonance(id))
    break;
    case kIndirect:
      return kIndirect;
    }
  }
return kNo;
}

bool InputGenJetsParticleSelector::getExcludeResonances ( ) const

inline

Definition at line 41 of file InputGenJetsParticleSelector.h.

References excludeResonances.

{ return excludeResonances; }

const std::vector<unsigned int>& InputGenJetsParticleSelector::getIgnoredParticles ( ) const

inline

Definition at line 44 of file InputGenJetsParticleSelector.h.

References ignoreParticleIDs.

    { return ignoreParticleIDs; }

bool InputGenJetsParticleSelector::getPartonicFinalState ( ) const

inline

Definition at line 40 of file InputGenJetsParticleSelector.h.

References partonicFinalState.

{ return partonicFinalState; }

double InputGenJetsParticleSelector::getPtMin ( ) const

inline

Definition at line 43 of file InputGenJetsParticleSelector.h.

References ptMin.

{ return ptMin; }

bool InputGenJetsParticleSelector::getTausAndJets ( ) const

inline

Definition at line 42 of file InputGenJetsParticleSelector.h.

References tausAsJets.

{ return tausAsJets; }

bool InputGenJetsParticleSelector::hasPartonChildren	(	ParticleBitmap &	invalid,
		const ParticleVector &	p,
		const reco::GenParticle *	particle
	)		const

Definition at line 214 of file InputGenJetsParticleSelector.cc.

References testPartonChildren().

Referenced by produce().

                                                                    {
  return testPartonChildren(invalid, p, particle) > 0;
}

bool InputGenJetsParticleSelector::isExcludedFromResonance ( int  pdgId ) const

private

Definition at line 104 of file InputGenJetsParticleSelector.cc.

References excludeFromResonancePids, benchmark_cfg::pdgId, and pos.

Referenced by fromResonance().

{
  pdgId = pdgId > 0 ? pdgId : -pdgId;
  std::vector<unsigned int>::const_iterator pos =
    std::lower_bound(excludeFromResonancePids.begin(),
             excludeFromResonancePids.end(),
             (unsigned int)pdgId);
  return pos != excludeFromResonancePids.end() && *pos == (unsigned int)pdgId;
 
}

bool InputGenJetsParticleSelector::isHadron ( int  pdgId )

static

Definition at line 80 of file InputGenJetsParticleSelector.cc.

References benchmark_cfg::pdgId.

{
  pdgId = (pdgId > 0 ? pdgId : -pdgId) % 10000;
  return (pdgId > 100 && pdgId < 900) ||
    (pdgId > 1000 && pdgId < 9000);
}

bool InputGenJetsParticleSelector::isIgnored

(

int

pdgId

)

const

Definition at line 94 of file InputGenJetsParticleSelector.cc.

References ignoreParticleIDs, benchmark_cfg::pdgId, and pos.

Referenced by fromResonance(), and produce().

{
  pdgId = pdgId > 0 ? pdgId : -pdgId;
  std::vector<unsigned int>::const_iterator pos =
    std::lower_bound(ignoreParticleIDs.begin(),
             ignoreParticleIDs.end(),
             (unsigned int)pdgId);
  return pos != ignoreParticleIDs.end() && *pos == (unsigned int)pdgId;
}

bool InputGenJetsParticleSelector::isParton

(

int

pdgId

)

const

Definition at line 72 of file InputGenJetsParticleSelector.cc.

References benchmark_cfg::pdgId, and tausAsJets.

Referenced by fromResonance(), and produce().

                                                          {
  pdgId = (pdgId > 0 ? pdgId : -pdgId) % 10000;
  return (pdgId > 0 && pdgId < 6) || pdgId == 7 ||
    pdgId == 9 || (tausAsJets && pdgId == 15) || pdgId == 21;
  // tops are not considered "regular" partons
  // but taus eventually are (since they may hadronize later)
}

bool InputGenJetsParticleSelector::isResonance ( int  pdgId )

static

Definition at line 87 of file InputGenJetsParticleSelector.cc.

References benchmark_cfg::pdgId.

Referenced by fromResonance().

{
  // gauge bosons and tops
  pdgId = (pdgId > 0 ? pdgId : -pdgId) % 10000;
  return (pdgId > 21 && pdgId <= 42) || pdgId == 6 || pdgId == 8 ;  //BUG! war 21. 22=gamma..
}

void InputGenJetsParticleSelector::produce ( edm::Event &  evt, const edm::EventSetup &  evtSetup  )

virtual

Implements edm::EDProducer.

Definition at line 222 of file InputGenJetsParticleSelector.cc.

References prof2calltree::count, excludeResonances, fromResonance(), genParticleCandidates2GenParticles_cfi::genParticles, edm::Event::getByLabel(), hasPartonChildren(), i, inTag, align::invalid, invalidateTree(), isIgnored(), isParton(), reco::CompositeRefCandidateT< D >::numberOfDaughters(), partonicFinalState, reco::LeafCandidate::pdgId(), reco::LeafCandidate::pt(), ptMin, edm::Event::put(), findQualityFiles::size, python.multivaluedict::sort(), and reco::LeafCandidate::status().

                                                                                       {

 
  std::auto_ptr<reco::GenParticleRefVector> selected_ (new reco::GenParticleRefVector);
    
  edm::Handle<reco::GenParticleCollection> genParticles;
  //  evt.getByLabel("genParticles", genParticles );
  evt.getByLabel(inTag, genParticles );
    
    
  ParticleVector particles;
  for (reco::GenParticleCollection::const_iterator iter=genParticles->begin();iter!=genParticles->end();++iter){
    particles.push_back(&*iter); 
  }
  
  std::sort(particles.begin(), particles.end());
  unsigned int size = particles.size();
  
  ParticleBitmap selected(size, false);
  ParticleBitmap invalid(size, false);

  for(unsigned int i = 0; i < size; i++) {
    const reco::GenParticle *particle = particles[i];
    if (invalid[i])
      continue;
    if (particle->status() == 1)
      selected[i] = true;
    if (partonicFinalState && isParton(particle->pdgId())) {
      
      if (particle->numberOfDaughters()==0 &&
      particle->status() != 1) {
    // some brokenness in event...
    invalid[i] = true;
      }
      else if (!hasPartonChildren(invalid, particles,
                  particle)) {
    selected[i] = true;
    invalidateTree(invalid, particles,particle); //this?!?
      }
    }
    
  }
 unsigned int count=0;
  for(size_t idx=0;idx<genParticles->size();++idx){ 
    const reco::GenParticle *particle = particles[idx];
    if (!selected[idx] || invalid[idx]){
      continue;
    }
    
    if (excludeResonances &&
    fromResonance(invalid, particles, particle)) {
      invalid[idx] = true;
      //cout<<"[INPUTSELECTOR] Invalidates FROM RESONANCE!: ["<<setw(4)<<idx<<"] "<<particle->pdgId()<<" "<<particle->pt()<<endl;
      continue;
    }
    
    if (isIgnored(particle->pdgId())){
      continue;
    }

   
    if (particle->pt() >= ptMin){
      edm::Ref<reco::GenParticleCollection> particleRef(genParticles,idx);
      selected_->push_back(particleRef);
      //cout<<"Finally we have: ["<<setw(4)<<idx<<"] "<<setw(4)<<particle->pdgId()<<" "<<particle->pt()<<endl;
      count++;
    }
  }
  evt.put(selected_);
}

void InputGenJetsParticleSelector::setExcludeFromResonancePids ( const std::vector< unsigned int > &  particleIDs )

private

Definition at line 66 of file InputGenJetsParticleSelector.cc.

References excludeFromResonancePids, and python.multivaluedict::sort().

Referenced by InputGenJetsParticleSelector().

{
  excludeFromResonancePids = particleIDs;
  std::sort( excludeFromResonancePids.begin(), excludeFromResonancePids.end());
}

void InputGenJetsParticleSelector::setExcludeResonances ( bool  flag = true )

inline

Definition at line 49 of file InputGenJetsParticleSelector.h.

References excludeResonances.

  { excludeResonances = flag; }

void InputGenJetsParticleSelector::setIgnoredParticles ( const std::vector< unsigned int > &  particleIDs )

private

Definition at line 60 of file InputGenJetsParticleSelector.cc.

References ignoreParticleIDs, and python.multivaluedict::sort().

Referenced by InputGenJetsParticleSelector().

{
  ignoreParticleIDs = particleIDs;
  std::sort(ignoreParticleIDs.begin(), ignoreParticleIDs.end());
}

void InputGenJetsParticleSelector::setPartonicFinalState ( bool  flag = true )

inline

Definition at line 47 of file InputGenJetsParticleSelector.h.

References partonicFinalState.

  { partonicFinalState = flag; }

void InputGenJetsParticleSelector::setPtMin ( double  ptMin )

inline

Definition at line 52 of file InputGenJetsParticleSelector.h.

References ptMin.

{ this->ptMin = ptMin; }

void InputGenJetsParticleSelector::setTausAsJets ( bool  flag = true )

inline

Definition at line 51 of file InputGenJetsParticleSelector.h.

References tausAsJets.

{ tausAsJets = flag; }

int InputGenJetsParticleSelector::testPartonChildren ( InputGenJetsParticleSelector::ParticleBitmap &  invalid, const ParticleVector &  p, const reco::GenParticle *  particle  ) const

private

Definition at line 148 of file InputGenJetsParticleSelector.cc.

References reco::CompositeRefCandidateT< D >::daughter(), i, CandMCTagUtils::isParton(), npart, reco::CompositeRefCandidateT< D >::numberOfDaughters(), partIdx(), reco::Candidate::pdgId(), and query::result.

Referenced by hasPartonChildren().

{
  unsigned int npart=particle->numberOfDaughters();
  if (!npart) {return 0;}

  for (unsigned int i=0;i<npart;++i){
    unsigned int idx = partIdx(p,dynamic_cast<const reco::GenParticle*>(particle->daughter(i)));
    if (invalid[idx])
      continue;
    if (isParton((particle->daughter(i)->pdgId()))){
      return 1;
    }
    if (isHadron((particle->daughter(i)->pdgId()))){
      return -1;
    }
    int result = testPartonChildren(invalid,p,dynamic_cast<const reco::GenParticle*>(particle->daughter(i)));
    if (result) return result;
  }
  return 0;
}

Member Data Documentation

std::vector<unsigned int> InputGenJetsParticleSelector::excludeFromResonancePids

private

Definition at line 84 of file InputGenJetsParticleSelector.h.

Referenced by isExcludedFromResonance(), and setExcludeFromResonancePids().

bool InputGenJetsParticleSelector::excludeResonances

private

Definition at line 90 of file InputGenJetsParticleSelector.h.

Referenced by getExcludeResonances(), produce(), and setExcludeResonances().

std::vector<unsigned int> InputGenJetsParticleSelector::ignoreParticleIDs

private

Definition at line 83 of file InputGenJetsParticleSelector.h.

Referenced by getIgnoredParticles(), isIgnored(), and setIgnoredParticles().

edm::InputTag InputGenJetsParticleSelector::inTag

private

Definition at line 78 of file InputGenJetsParticleSelector.h.

Referenced by produce().

bool InputGenJetsParticleSelector::partonicFinalState

private

Definition at line 89 of file InputGenJetsParticleSelector.h.

Referenced by getPartonicFinalState(), produce(), and setPartonicFinalState().

double InputGenJetsParticleSelector::ptMin

private

Definition at line 92 of file InputGenJetsParticleSelector.h.

Referenced by getPtMin(), produce(), and setPtMin().

bool InputGenJetsParticleSelector::tausAsJets

private

Definition at line 91 of file InputGenJetsParticleSelector.h.

Referenced by getTausAndJets(), isParton(), and setTausAsJets().