BaseFlatGunProducer.h

Go to the documentation of this file.

#ifndef BaseFlatGunProducer_H
#define BaseFlatGunProducer_H

#include <string>

#include "HepPDT/defs.h"
#include "HepPDT/TableBuilder.hh"
#include "HepPDT/ParticleDataTable.hh"

#include "HepMC/GenEvent.h"

#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/one/EDProducer.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/Run.h"

#include <memory>

namespace edm {

  class BaseFlatGunProducer : public one::EDProducer<one::WatchRuns, EndRunProducer> {
  public:
    BaseFlatGunProducer(const ParameterSet&);
    ~BaseFlatGunProducer() override;
    void beginRun(const edm::Run& r, const edm::EventSetup&) override;
    void endRun(edm::Run const& r, const edm::EventSetup&) override;
    void endRunProduce(edm::Run& r, const edm::EventSetup&) override;

  private:
    const ESGetToken<HepPDT::ParticleDataTable, edm::DefaultRecord> fPDGTableToken;

  protected:
    // non-virtuals ! this and only way !
    //
    // data members

    // gun particle(s) characteristics
    std::vector<int> fPartIDs;
    double fMinEta;
    double fMaxEta;
    double fMinPhi;
    double fMaxPhi;

    // the event format itself
    HepMC::GenEvent* fEvt;

    // HepMC/HepPDT related things
    // (for particle/event construction)
    //std::string      fPDGTablePath ;
    //std::string      fPDGTableName ;
    // DefaultConfig::ParticleDataTable* fPDGTable;
    // DefaultConfig::ParticleDataTable* fTestTable ;
    // ESHandle<DefaultConfig::ParticleDataTable> fPDGTable ;
    ESHandle<HepPDT::ParticleDataTable> fPDGTable;

    int fVerbosity;

    bool fAddAntiParticle;
  };
}  // namespace edm

#endif