File indexing completed on 2023-03-17 11:10:12
0001 #ifndef IOMC_BetafuncEvtVtxGenerator_H
0002 #define IOMC_BetafuncEvtVtxGenerator_H
0003
0004
0005
0006
0007
0008
0009
0010
0011
0012
0013
0014
0015
0016
0017
0018
0019
0020
0021 #include "IOMC/EventVertexGenerators/interface/BaseEvtVtxGenerator.h"
0022 #include "FWCore/Framework/interface/ESWatcher.h"
0023 #include "FWCore/Utilities/interface/ESGetToken.h"
0024 #include "CondFormats/DataRecord/interface/SimBeamSpotObjectsRcd.h"
0025 #include "CondFormats/BeamSpotObjects/interface/SimBeamSpotObjects.h"
0026
0027 namespace CLHEP {
0028 class HepRandomEngine;
0029 }
0030
0031 class BetafuncEvtVtxGenerator : public BaseEvtVtxGenerator {
0032 public:
0033 BetafuncEvtVtxGenerator(const edm::ParameterSet& p);
0034
0035 BetafuncEvtVtxGenerator(const BetafuncEvtVtxGenerator& p) = delete;
0036
0037 BetafuncEvtVtxGenerator& operator=(const BetafuncEvtVtxGenerator& rhs) = delete;
0038 ~BetafuncEvtVtxGenerator() override;
0039
0040 void beginLuminosityBlock(edm::LuminosityBlock const&, edm::EventSetup const&) override;
0041
0042
0043
0044 HepMC::FourVector newVertex(CLHEP::HepRandomEngine*) const override;
0045
0046 TMatrixD const* GetInvLorentzBoost() const override;
0047
0048
0049 void sigmaZ(double s = 1.0);
0050
0051
0052 void X0(double m = 0) { fX0 = m; }
0053
0054 void Y0(double m = 0) { fY0 = m; }
0055
0056 void Z0(double m = 0) { fZ0 = m; }
0057
0058
0059 void betastar(double m = 0) { fbetastar = m; }
0060
0061 void emittance(double m = 0) { femittance = m; }
0062
0063
0064 double BetaFunction(double z, double z0) const;
0065
0066 private:
0067 void setBoost(double alpha, double phi);
0068
0069 private:
0070 bool readDB_;
0071
0072 double fX0, fY0, fZ0;
0073 double fSigmaZ;
0074
0075 double fbetastar, femittance;
0076
0077 double fTimeOffset;
0078
0079 TMatrixD boost_;
0080
0081 void update(const edm::EventSetup& iEventSetup);
0082 edm::ESWatcher<SimBeamSpotObjectsRcd> parameterWatcher_;
0083 edm::ESGetToken<SimBeamSpotObjects, SimBeamSpotObjectsRcd> beamToken_;
0084 };
0085
0086 #endif