TopDecaySubset.h

Go to the documentation of this file.

#include <memory>
#include <string>
#include <vector>
#include <map>

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EDProducer.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"

#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"

#include "SimDataFormats/GeneratorProducts/interface/GenEventInfoProduct.h"

class TopDecaySubset : public edm::EDProducer {
public:
  enum FillMode { kStable, kME };
  enum ShowerModel { kStart = -1, kNone, kPythia, kHerwig, kPythia8, kSherpa };
  enum RunMode { kRun1, kRun2 };

  explicit TopDecaySubset(const edm::ParameterSet& cfg);
  ~TopDecaySubset() override;
  void produce(edm::Event& event, const edm::EventSetup& setup) override;

private:
  std::vector<const reco::GenParticle*> findTops(const reco::GenParticleCollection& parts);
  std::vector<const reco::GenParticle*> findPrimalTops(const reco::GenParticleCollection& parts);
  std::vector<const reco::GenParticle*> findDecayingTops(const reco::GenParticleCollection& parts);
  const reco::GenParticle* findPrimalW(const reco::GenParticle* top);
  //  const reco::GenParticle* findDecayingW(const reco::GenParticle* top);
  const reco::GenParticle* findLastParticleInChain(const reco::GenParticle* p);
  ShowerModel checkShowerModel(const std::vector<const reco::GenParticle*>& tops) const;
  ShowerModel checkShowerModel(edm::Event& event);
  void checkWBosons(std::vector<const reco::GenParticle*>& tops) const;
  void fillListing(const std::vector<const reco::GenParticle*>& tops, reco::GenParticleCollection& target);
  void fillListing(const std::vector<const reco::GenParticle*>& primalTops,
                   const std::vector<const reco::GenParticle*>& decayingTops,
                   reco::GenParticleCollection& target);

  void clearReferences();
  void fillReferences(const reco::GenParticleRefProd& refProd, reco::GenParticleCollection& target);
  reco::Particle::LorentzVector p4(const std::vector<const reco::GenParticle*>::const_iterator top, int statusFlag);
  reco::Particle::LorentzVector p4(const reco::GenParticle::const_iterator part, int statusFlag);
  void addDaughters(int& idx,
                    const reco::GenParticle::const_iterator part,
                    reco::GenParticleCollection& target,
                    bool recursive = true);
  void addRadiation(int& idx, const reco::GenParticle::const_iterator part, reco::GenParticleCollection& target);
  void addRadiation(int& idx, const reco::GenParticle* part, reco::GenParticleCollection& target);

private:
  edm::EDGetTokenT<reco::GenParticleCollection> srcToken_;
  edm::EDGetTokenT<GenEventInfoProduct> genEventInfo_srcToken_;
  bool addRadiation_;
  FillMode fillMode_;
  ShowerModel showerModel_;
  RunMode runMode_;

  int motherPartIdx_;
  std::map<int, std::vector<int> > refs_;
};