BetafuncEvtVtxGenerator.h

Go to the documentation of this file.

#ifndef IOMC_BetafuncEvtVtxGenerator_H
#define IOMC_BetafuncEvtVtxGenerator_H

/*
________________________________________________________________________

 BetafuncEvtVtxGenerator

 Smear vertex according to the Beta function on the transverse plane
 and a Gaussian on the z axis. It allows the beam to have a crossing
 angle (dx/dz and dy/dz).

 Based on GaussEvtVtxGenerator.h
 implemented by Francisco Yumiceva (yumiceva@fnal.gov)

 FERMILAB
 2006
________________________________________________________________________
*/

#include "IOMC/EventVertexGenerators/interface/BaseEvtVtxGenerator.h"
#include "FWCore/Framework/interface/ESWatcher.h"
#include "CondFormats/DataRecord/interface/SimBeamSpotObjectsRcd.h"

namespace CLHEP {
  class HepRandomEngine;
}

class BetafuncEvtVtxGenerator : public BaseEvtVtxGenerator 
{
public:
  BetafuncEvtVtxGenerator(const edm::ParameterSet & p);
  virtual ~BetafuncEvtVtxGenerator();

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

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

  virtual TMatrixD const* GetInvLorentzBoost() const override;

    
  void sigmaZ(double s=1.0);

  void X0(double m=0) { fX0=m; }
  void Y0(double m=0) { fY0=m; }
  void Z0(double m=0) { fZ0=m; }

  void betastar(double m=0) { fbetastar=m; }
  void emittance(double m=0) { femittance=m; }

  double BetaFunction(double z, double z0) const;
    
private:
  BetafuncEvtVtxGenerator(const BetafuncEvtVtxGenerator &p);
  BetafuncEvtVtxGenerator&  operator = (const BetafuncEvtVtxGenerator & rhs );

  void setBoost(double alpha, double phi);
private:

  bool readDB_;

  double fX0, fY0, fZ0;
  double fSigmaZ;
  //double fdxdz, fdydz;
  double fbetastar, femittance;
  //  double falpha;
  double fTimeOffset;

  TMatrixD boost_;

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

#endif