InputGenJetsParticleSelector.h

Go to the documentation of this file.

#ifndef __InputGenJetsParticleSelector__
#define __InputGenJetsParticleSelector__

/* \class InputGenJetsParticleSelector
 * 
 * Selects particles for the GenJet input.
 * Deselect specified particles, also radiation from resoances.
 * Or only select partonic final state.
 * The algorithm is based on code of Christophe Saout. 
 *
 * \author: Andreas Oehler
 * 
 * 
 */

#include "DataFormats/Provenance/interface/Provenance.h"
#include "FWCore/Framework/interface/Event.h"
#include "DataFormats/Candidate/interface/CandidateFwd.h"
#include "DataFormats/Candidate/interface/Candidate.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "FWCore/Framework/interface/global/EDProducer.h"
#include "DataFormats/Common/interface/EDProductfwd.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"


class InputGenJetsParticleSelector : public edm::global::EDProducer<> {
  // collection type
 public:
  typedef std::vector<bool>     ParticleBitmap;
  typedef std::vector<const reco::Candidate*> ParticleVector;
      
  InputGenJetsParticleSelector(const edm::ParameterSet & ); 
  ~InputGenJetsParticleSelector(); 
  // select object from a collection and 
  // possibly event content
  virtual void produce (edm::StreamID, edm::Event &evt, const edm::EventSetup &evtSetup) const override;
      
  bool getPartonicFinalState() const { return partonicFinalState; }
  bool getExcludeResonances() const { return excludeResonances; }
  bool getTausAndJets() const { return tausAsJets; }
  double getPtMin() const { return ptMin; }
  const std::vector<unsigned int> &getIgnoredParticles() const
    { return ignoreParticleIDs; }
  
  void setPartonicFinalState(bool flag = true)
  { partonicFinalState = flag; }
  void setExcludeResonances(bool flag = true)
  { excludeResonances = flag; }
  void setTausAsJets(bool flag = true) { tausAsJets = flag; }
  void setPtMin(double ptMin) { this->ptMin = ptMin; }
  bool isParton(int pdgId) const;
  static bool isHadron(int pdgId);
  static bool isResonance(int pdgId);

  bool isIgnored(int pdgId) const;
  bool hasPartonChildren(ParticleBitmap &invalid,
             const ParticleVector &p,
             const reco::Candidate *particle) const;
  
  enum ResonanceState {
    kNo = 0,
    kDirect,
    kIndirect
  };
  ResonanceState fromResonance(ParticleBitmap &invalid,
                   const ParticleVector &p,
                   const reco::Candidate *particle) const;
  
  
  // iterators over selected objects: collection begin
  
 private:
  //container selected_;  //container required by selector
  InputGenJetsParticleSelector(){} //should not be used!
  
  edm::InputTag inTag;
  edm::InputTag prunedInTag;
  int testPartonChildren(ParticleBitmap &invalid,
             const ParticleVector &p,
             const reco::Candidate *particle) const;
  
  std::vector<unsigned int> ignoreParticleIDs;
  std::vector<unsigned int> excludeFromResonancePids;
  void setExcludeFromResonancePids(const std::vector<unsigned int> &particleIDs);
  void setIgnoredParticles(const std::vector<unsigned int> &particleIDs);
  bool isExcludedFromResonance(int pdgId) const;
  
  bool          partonicFinalState;
  bool          excludeResonances;
  bool          tausAsJets;
  bool          isMiniAOD;
  double        ptMin;
  
  edm::EDGetTokenT<reco::CandidateView> input_genpartcoll_token_;
  edm::EDGetTokenT<reco::CandidateView> input_prunedgenpartcoll_token_;
  
};

#endif