ParticleReplacerParticleGun.h

Go to the documentation of this file.

#ifndef TauAnalysis_MCEmbeddingTools_ParticleReplacerParticleGun_h
#define TauAnalysis_MCEmbeddingTools_ParticleReplacerParticleGun_h

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

#include "GeneratorInterface/Pythia6Interface/interface/Pythia6Service.h"
#include "GeneratorInterface/TauolaInterface/interface/TauolaInterfaceBase.h"
#include "GeneratorInterface/TauolaInterface/interface/TauolaFactory.h"

#include<string>
#include<vector>

class ParticleReplacerParticleGun: public ParticleReplacerBase 
{
 public:
  explicit ParticleReplacerParticleGun(const edm::ParameterSet&);
  virtual ~ParticleReplacerParticleGun() {}

  virtual void beginJob();
  virtual void endJob();

  std::auto_ptr<HepMC::GenEvent> produce(const std::vector<reco::Particle>&, const reco::Vertex* = 0, const HepMC::GenEvent* = 0, MCParticleReplacer* = 0);

 private:
  void correctTauMass(const std::vector<reco::Particle>&, std::vector<HepMC::FourVector>&);
  void forceTauolaTauDecays();
  void tauola_forParticleGun(int, int, const HepMC::FourVector&);
  float tauHelicity(int);
  float randomPolarization();

  gen::TauolaInterfaceBase* tauola_;
  gen::Pythia6Service pythia_;

  std::string particleOrigin_;
  std::string forceTauPolarization_;
  std::string forceTauDecay_;
  std::string generatorMode_;
  int gunParticle_;
  int forceTauPlusHelicity_;
  int forceTauMinusHelicity_;

  float pol1_[4];
  float pol2_[4];
};

#endif