EventVertexGenerators/interface/GaussEvtVtxGenerator.h

0001 #ifndef IOMC_GaussEvtVtxGenerator_H
0002 #define IOMC_GaussEvtVtxGenerator_H
0003
0004 /**
0005  * Generate event vertices according to a Gauss distribution.
0006  * Attention: All values are assumed to be cm!
0007  *
0008  */
0009
0010 #include "IOMC/EventVertexGenerators/interface/BaseEvtVtxGenerator.h"
0011 #include "FWCore/Framework/interface/ESWatcher.h"
0012 #include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
0013 #include "FWCore/Utilities/interface/ESGetToken.h"
0014 #include "CondFormats/DataRecord/interface/SimBeamSpotObjectsRcd.h"
0015 #include "CondFormats/BeamSpotObjects/interface/SimBeamSpotObjects.h"
0016
0017 namespace CLHEP {
0018   class HepRandomEngine;
0019 }
0020
0021 class GaussEvtVtxGenerator : public BaseEvtVtxGenerator {
0022 public:
0023   GaussEvtVtxGenerator(const edm::ParameterSet& p);
0024   /** Copy constructor */
0025   GaussEvtVtxGenerator(const GaussEvtVtxGenerator& p) = delete;
0026   /** Copy assignment operator */
0027   GaussEvtVtxGenerator& operator=(const GaussEvtVtxGenerator& rhs) = delete;
0028   ~GaussEvtVtxGenerator() override = default;
0029
0030   static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
0031
0032   void beginLuminosityBlock(edm::LuminosityBlock const&, edm::EventSetup const&) override;
0033
0034   /// return a new event vertex
0035   ROOT::Math::XYZTVector vertexShift(CLHEP::HepRandomEngine*) const override;
0036
0037   TMatrixD const* GetInvLorentzBoost() const override { return nullptr; }
0038
0039   /// set resolution in X in cm
0040   void sigmaX(double s = 1.0);
0041   /// set resolution in Y in cm
0042   void sigmaY(double s = 1.0);
0043   /// set resolution in Z in cm
0044   void sigmaZ(double s = 1.0);
0045
0046   /// set mean in X in cm
0047   void meanX(double m = 0) { fMeanX = m; }
0048   /// set mean in Y in cm
0049   void meanY(double m = 0) { fMeanY = m; }
0050   /// set mean in Z in cm
0051   void meanZ(double m = 0) { fMeanZ = m; }
0052
0053 private:
0054   bool readDB_;
0055
0056   double fSigmaX, fSigmaY, fSigmaZ;
0057   double fMeanX, fMeanY, fMeanZ;
0058   double fTimeOffset;
0059
0060   void update(const edm::EventSetup& iEventSetup);
0061   edm::ESWatcher<SimBeamSpotObjectsRcd> parameterWatcher_;
0062   edm::ESGetToken<SimBeamSpotObjects, SimBeamSpotObjectsRcd> beamToken_;
0063 };
0064
0065 #endif