GaussEvtVtxGenerator.h

Go to the documentation of this file.

#ifndef IOMC_GaussEvtVtxGenerator_H
#define IOMC_GaussEvtVtxGenerator_H

#include "IOMC/EventVertexGenerators/interface/BaseEvtVtxGenerator.h"
#include "FWCore/Framework/interface/ESWatcher.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/Utilities/interface/ESGetToken.h"
#include "CondFormats/DataRecord/interface/SimBeamSpotObjectsRcd.h"
#include "CondFormats/BeamSpotObjects/interface/SimBeamSpotObjects.h"

namespace CLHEP {
  class HepRandomEngine;
}

class GaussEvtVtxGenerator : public BaseEvtVtxGenerator {
public:
  GaussEvtVtxGenerator(const edm::ParameterSet& p);
  GaussEvtVtxGenerator(const GaussEvtVtxGenerator& p) = delete;
  GaussEvtVtxGenerator& operator=(const GaussEvtVtxGenerator& rhs) = delete;
  ~GaussEvtVtxGenerator() override = default;

  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);

  void beginLuminosityBlock(edm::LuminosityBlock const&, edm::EventSetup const&) override;

  //virtual CLHEP::Hep3Vector* newVertex();
  HepMC::FourVector newVertex(CLHEP::HepRandomEngine*) const override;

  TMatrixD const* GetInvLorentzBoost() const override { return nullptr; }

  void sigmaX(double s = 1.0);
  void sigmaY(double s = 1.0);
  void sigmaZ(double s = 1.0);

  void meanX(double m = 0) { fMeanX = m; }
  void meanY(double m = 0) { fMeanY = m; }
  void meanZ(double m = 0) { fMeanZ = m; }

private:
  bool readDB_;

  double fSigmaX, fSigmaY, fSigmaZ;
  double fMeanX, fMeanY, fMeanZ;
  double fTimeOffset;

  void update(const edm::EventSetup& iEventSetup);
  edm::ESWatcher<SimBeamSpotObjectsRcd> parameterWatcher_;
  edm::ESGetToken<SimBeamSpotObjects, SimBeamSpotObjectsRcd> beamToken_;
};

#endif