MCParticleReplacer.h

Go to the documentation of this file.

#ifndef TauAnalysis_MCEmbeddingTools_MCParticleReplacer_h
#define TauAnalysis_MCEmbeddingTools_MCParticleReplacer_h

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

#include "TauAnalysis/MCEmbeddingTools/interface/ParticleReplacerBase.h"

#include<string>

#include<boost/shared_ptr.hpp>

class MCParticleReplacer : public edm::EDProducer
{
 public:
  explicit MCParticleReplacer(const edm::ParameterSet&);
  ~MCParticleReplacer();

  virtual void produce(edm::Event&, const edm::EventSetup&);
  virtual void beginRun(edm::Run&,const edm::EventSetup&);
  virtual void endRun();
  virtual void beginJob();
  virtual void endJob();
    
  template <typename T>
  void call_produces(const std::string& instanceName)
  {
    produces<T>(instanceName);
  }

  template <typename T>
  void call_put(T& product, const std::string& instanceName)
  {
    evt_->put(product, instanceName);
  }

  edm::StreamID getStreamID() const { assert(evt_ != nullptr); return evt_->streamID(); }

 private:
  enum HepMcMode { kInvalid = 0, kNew, kReplace };
  static HepMcMode stringToHepMcMode(const std::string& name);

  edm::InputTag src_;
  edm::InputTag srcHepMC_;
  HepMcMode hepMcMode_;
  ParticleReplacerBase* replacer_;
  edm::Event* evt_;

  int verbosity_;
};

#endif