BeamProfileVtxGenerator.h

Go to the documentation of this file.

#ifndef IOMC_BeamProfileVtxGenerator_H
#define IOMC_BeamProfileVtxGenerator_H

#include "IOMC/EventVertexGenerators/interface/BaseEvtVtxGenerator.h"
#include <vector>

namespace CLHEP {
  class HepRandomEngine;
}

class BeamProfileVtxGenerator : public BaseEvtVtxGenerator {
public:
  BeamProfileVtxGenerator(const edm::ParameterSet& p);
  BeamProfileVtxGenerator(const BeamProfileVtxGenerator& p) = delete;
  BeamProfileVtxGenerator& operator=(const BeamProfileVtxGenerator& rhs) = delete;
  ~BeamProfileVtxGenerator() 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 meanX(double m = 0) { fMeanX = m; }
  void meanY(double m = 0) { fMeanY = m; }
  void beamPos(double m = 0) { fMeanZ = m; }

  void eta(double m = 0);
  void phi(double m = 0) { fPhi = m; }
  void psi(double m = 999) { fPsi = m; }
  void setType(bool m = true);

private:
  double fSigmaX, fSigmaY;
  double fMeanX, fMeanY, fMeanZ;
  double fEta, fPhi, fTheta;

  double fPsi;

  bool fType, ffile;
  int nBinx, nBiny;
  std::vector<double> fdistn;
  double fTimeOffset;
};

#endif