BetaBoostEvtVtxGenerator Class Reference

Inheritance diagram for BetaBoostEvtVtxGenerator:

Public Member Functions
void	Alpha (double m=0)
	angle between crossing plane and horizontal plane More...
void	Beta (double m=0)
	BetaBoostEvtVtxGenerator (const edm::ParameterSet &p)
double	BetaFunction (double z, double z0)
	beta function More...
void	betastar (double m=0)
	set beta_star More...
void	emittance (double m=0)
	emittance (no the normalized) More...
virtual TMatrixD *	GetInvLorentzBoost ()
virtual HepMC::FourVector *	newVertex (CLHEP::HepRandomEngine *)
	return a new event vertex More...
void	Phi (double m=0)
	set half crossing angle More...
virtual void	produce (edm::Event &, const edm::EventSetup &) override
void	sigmaZ (double s=1.0)
	set resolution in Z in cm More...
void	X0 (double m=0)
	set mean in X in cm More...
void	Y0 (double m=0)
	set mean in Y in cm More...
void	Z0 (double m=0)
	set mean in Z in cm More...
virtual	~BetaBoostEvtVtxGenerator ()
Public Member Functions inherited from edm::EDProducer
	EDProducer ()
ModuleDescription const &	moduleDescription () const
	~EDProducer () override
Public Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
std::vector< edm::ProductResolverIndex > const &	indiciesForPutProducts (BranchType iBranchType) const
	ProducerBase ()
std::vector< edm::ProductResolverIndex > const &	putTokenIndexToProductResolverIndex () const
void	registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...
void	resolvePutIndicies (BranchType iBranchType, ModuleToResolverIndicies const &iIndicies, std::string const &moduleLabel)
virtual	~ProducerBase () noexcept(false)
Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const
void	convertCurrentProcessAlias (std::string const &processName)
	Convert "@currentProcess" in InputTag process names to the actual current process name. More...
	EDConsumerBase ()
	EDConsumerBase (EDConsumerBase const &)=delete
	EDConsumerBase (EDConsumerBase &&)=default
ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const
void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
std::vector< ProductResolverIndexAndSkipBit > const &	itemsToGetFrom (BranchType iType) const
void	labelsForToken (EDGetToken iToken, Labels &oLabels) const
void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const
EDConsumerBase const &	operator= (EDConsumerBase const &)=delete
EDConsumerBase &	operator= (EDConsumerBase &&)=default
bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const
bool	registeredToConsumeMany (TypeID const &, BranchType) const
void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
virtual	~EDConsumerBase () noexcept(false)

Private Member Functions
	BetaBoostEvtVtxGenerator (const BetaBoostEvtVtxGenerator &p)
BetaBoostEvtVtxGenerator &	operator= (const BetaBoostEvtVtxGenerator &rhs)

Private Attributes
double	alpha_
double	beta_
TMatrixD *	boost_
double	falpha
double	fbetastar
double	femittance
double	fSigmaZ
double	fTimeOffset
HepMC::FourVector *	fVertex
double	fX0
double	fY0
double	fZ0
double	phi_
edm::EDGetTokenT< HepMCProduct >	sourceLabel
bool	verbosity_

Additional Inherited Members
Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType
Public Types inherited from edm::ProducerBase
using	ModuleToResolverIndicies = std::unordered_multimap< std::string, std::tuple< edm::TypeID const *, const char *, edm::ProductResolverIndex >>
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList
Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels
Static Public Member Functions inherited from edm::EDProducer
static const std::string &	baseType ()
static void	fillDescriptions (ConfigurationDescriptions &descriptions)
static void	prevalidate (ConfigurationDescriptions &descriptions)
Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)
EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)
template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)
ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...
template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()
void	consumesMany (const TypeToGet &id)
template<BranchType B>
void	consumesMany (const TypeToGet &id)
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)
template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

Definition at line 52 of file BetaBoostEvtVtxGenerator.cc.

Constructor & Destructor Documentation

BetaBoostEvtVtxGenerator::BetaBoostEvtVtxGenerator

(

const edm::ParameterSet &

p

)

Definition at line 113 of file BetaBoostEvtVtxGenerator.cc.

References alpha_, beta_, Exception, fbetastar, femittance, fSigmaZ, fTimeOffset, fX0, fY0, fZ0, edm::ParameterSet::getParameter(), and phi_.

                                                                           :
  fVertex(0), boost_(0), fTimeOffset(0),
  sourceLabel(consumes<HepMCProduct>(p.getParameter<edm::InputTag>("src"))),
  verbosity_(p.getUntrackedParameter<bool>("verbosity",false))
{
  fX0 =        p.getParameter<double>("X0")*cm;
  fY0 =        p.getParameter<double>("Y0")*cm;
  fZ0 =        p.getParameter<double>("Z0")*cm;
  fSigmaZ =    p.getParameter<double>("SigmaZ")*cm;
  alpha_ =     p.getParameter<double>("Alpha")*radian;
  phi_ =       p.getParameter<double>("Phi")*radian;
  fbetastar =  p.getParameter<double>("BetaStar")*cm;
  femittance = p.getParameter<double>("Emittance")*cm; // this is not the normalized emittance
  fTimeOffset = p.getParameter<double>("TimeOffset")*ns*c_light; // HepMC time units are mm
  beta_=p.getParameter<double>("Beta"); 
  if (fSigmaZ <= 0) {
    throw cms::Exception("Configuration")
      << "Error in BetaBoostEvtVtxGenerator: "
      << "Illegal resolution in Z (SigmaZ is negative)";
  }

  produces<edm::HepMCProduct>();
  
}

BetaBoostEvtVtxGenerator::~BetaBoostEvtVtxGenerator ( )

virtual

Definition at line 138 of file BetaBoostEvtVtxGenerator.cc.

References boost_, and fVertex.

{
  delete fVertex ;
  if (boost_ != 0 ) delete boost_;
}

BetaBoostEvtVtxGenerator::BetaBoostEvtVtxGenerator ( const BetaBoostEvtVtxGenerator &  p )

private

Copy constructor

Member Function Documentation

void BetaBoostEvtVtxGenerator::Alpha ( double  m = 0 )

inline

angle between crossing plane and horizontal plane

Definition at line 77 of file BetaBoostEvtVtxGenerator.cc.

References funct::m.

{ alpha_=m; }

void BetaBoostEvtVtxGenerator::Beta ( double  m = 0 )

inline

Definition at line 78 of file BetaBoostEvtVtxGenerator.cc.

References funct::m.

{ beta_=m; }

double BetaBoostEvtVtxGenerator::BetaFunction	(	double	z,
		double	z0
	)

beta function

Definition at line 172 of file BetaBoostEvtVtxGenerator.cc.

References fbetastar, femittance, and mathSSE::sqrt().

Referenced by newVertex().

{
  return sqrt(femittance*(fbetastar+(((z-z0)*(z-z0))/fbetastar)));

}

void BetaBoostEvtVtxGenerator::betastar ( double  m = 0 )

inline

set beta_star

Definition at line 81 of file BetaBoostEvtVtxGenerator.cc.

References funct::m.

{ fbetastar=m; }

void BetaBoostEvtVtxGenerator::emittance ( double  m = 0 )

inline

emittance (no the normalized)

Definition at line 83 of file BetaBoostEvtVtxGenerator.cc.

References funct::m.

{ femittance=m; }

TMatrixD * BetaBoostEvtVtxGenerator::GetInvLorentzBoost ( )

virtual

Definition at line 191 of file BetaBoostEvtVtxGenerator.cc.

References alpha_, beta_, boost_, funct::cos(), phi_, funct::sin(), mathSSE::sqrt(), funct::tan(), and verbosity_.

Referenced by produce().

                                                       {

  //alpha_ = 0;
  //phi_ = 142.e-6;
  //    if (boost_ != 0 ) return boost_;
    
  //boost_.ResizeTo(4,4);
  //boost_ = new TMatrixD(4,4);
  TMatrixD tmpboost(4,4);
  TMatrixD tmpboostZ(4,4);
  TMatrixD tmpboostXYZ(4,4);

  //if ( (alpha_ == 0) && (phi_==0) ) { boost_->Zero(); return boost_; }
    
  // Lorentz boost to frame where the collision is head-on
  // phi is the half crossing angle in the plane ZS
  // alpha is the angle to the S axis from the X axis in the XY plane
    
  tmpboost(0,0) = 1./cos(phi_);
  tmpboost(0,1) = - cos(alpha_)*sin(phi_);
  tmpboost(0,2) = - tan(phi_)*sin(phi_);
  tmpboost(0,3) = - sin(alpha_)*sin(phi_);
  tmpboost(1,0) = - cos(alpha_)*tan(phi_);
  tmpboost(1,1) = 1.;
  tmpboost(1,2) = cos(alpha_)*tan(phi_);
  tmpboost(1,3) = 0.;
  tmpboost(2,0) = 0.;
  tmpboost(2,1) = - cos(alpha_)*sin(phi_);
  tmpboost(2,2) = cos(phi_);
  tmpboost(2,3) = - sin(alpha_)*sin(phi_);
  tmpboost(3,0) = - sin(alpha_)*tan(phi_);
  tmpboost(3,1) = 0.;
  tmpboost(3,2) = sin(alpha_)*tan(phi_);
  tmpboost(3,3) = 1.;
  //cout<<"beta "<<beta_;
  double gama=1.0/sqrt(1-beta_*beta_);
  tmpboostZ(0,0)=gama;
  tmpboostZ(0,1)=0.;
  tmpboostZ(0,2)=-1.0*beta_*gama;
  tmpboostZ(0,3)=0.;
  tmpboostZ(1,0)=0.;
  tmpboostZ(1,1) = 1.;
  tmpboostZ(1,2)=0.;
  tmpboostZ(1,3)=0.;
  tmpboostZ(2,0)=-1.0*beta_*gama;
  tmpboostZ(2,1) = 0.;
  tmpboostZ(2,2)=gama;
  tmpboostZ(2,3) = 0.;
  tmpboostZ(3,0)=0.;
  tmpboostZ(3,1)=0.;
  tmpboostZ(3,2)=0.;
  tmpboostZ(3,3) = 1.;

  tmpboostXYZ=tmpboostZ*tmpboost;
  tmpboostXYZ.Invert();



  boost_ = new TMatrixD(tmpboostXYZ);
  if ( verbosity_ ) { boost_->Print(); }
    
  return boost_;
}

HepMC::FourVector * BetaBoostEvtVtxGenerator::newVertex ( CLHEP::HepRandomEngine *  engine )

virtual

return a new event vertex

Definition at line 145 of file BetaBoostEvtVtxGenerator.cc.

References BetaFunction(), fSigmaZ, fTimeOffset, fVertex, fX0, fY0, fZ0, mathSSE::sqrt(), X, DOFs::Y, and DOFs::Z.

Referenced by produce().

                                                                                 {

    
  double X,Y,Z;

  double tmp_sigz = CLHEP::RandGaussQ::shoot(engine, 0.0, fSigmaZ);
  Z = tmp_sigz + fZ0;

  double tmp_sigx = BetaFunction(Z,fZ0); 
  // need sqrt(2) for beamspot width relative to single beam width
  tmp_sigx /= sqrt(2.0);
  X = CLHEP::RandGaussQ::shoot(engine, 0.0, tmp_sigx) + fX0; // + Z*fdxdz ;

  double tmp_sigy = BetaFunction(Z,fZ0);
  // need sqrt(2) for beamspot width relative to single beam width
  tmp_sigy /= sqrt(2.0);
  Y = CLHEP::RandGaussQ::shoot(engine, 0.0, tmp_sigy) + fY0; // + Z*fdydz;

  double tmp_sigt = CLHEP::RandGaussQ::shoot(engine, 0.0, fSigmaZ);
  double T = tmp_sigt + fTimeOffset; 

  if ( fVertex == 0 ) fVertex = new HepMC::FourVector();
  fVertex->set(X,Y,Z,T);
        
  return fVertex;
}

BetaBoostEvtVtxGenerator& BetaBoostEvtVtxGenerator::operator= ( const BetaBoostEvtVtxGenerator &  rhs )

private

Copy assignment operator

void BetaBoostEvtVtxGenerator::Phi ( double  m = 0 )

inline

set half crossing angle

Definition at line 75 of file BetaBoostEvtVtxGenerator.cc.

References funct::m.

{ phi_=m; }

void BetaBoostEvtVtxGenerator::produce ( edm::Event &  evt, const edm::EventSetup &    )

overridevirtual

Definition at line 255 of file BetaBoostEvtVtxGenerator.cc.

References DEFINE_FWK_MODULE, Exception, edm::Event::getByToken(), edm::RandomNumberGenerator::getEngine(), edm::HepMCProduct::GetEvent(), GetInvLorentzBoost(), edm::Service< T >::isAvailable(), eostools::move(), newVertex(), edm::Event::put(), reco::return(), sourceLabel, and edm::Event::streamID().

{
  edm::Service<RandomNumberGenerator> rng;
  if (!rng.isAvailable()) {
    throw cms::Exception("Configuration")
      << "Attempt to get a random engine when the RandomNumberGeneratorService is not configured.\n"
         "You must configure the service if you want an engine.\n";
  }
  CLHEP::HepRandomEngine* engine = &rng->getEngine(evt.streamID());
  
  Handle<HepMCProduct> HepUnsmearedMCEvt;
  evt.getByToken(sourceLabel, HepUnsmearedMCEvt);

  // Copy the HepMC::GenEvent
  HepMC::GenEvent* genevt = new HepMC::GenEvent(*HepUnsmearedMCEvt->GetEvent());
  std::unique_ptr<edm::HepMCProduct> HepMCEvt(new edm::HepMCProduct(genevt));

  // generate new vertex & apply the shift 
  //
  HepMCEvt->applyVtxGen( newVertex(engine) ) ;
 
  //HepMCEvt->LorentzBoost( 0., 142.e-6 );
  HepMCEvt->boostToLab( GetInvLorentzBoost(), "vertex" );
  HepMCEvt->boostToLab( GetInvLorentzBoost(), "momentum" );
  evt.put(std::move(HepMCEvt));

  return ;
}

void BetaBoostEvtVtxGenerator::sigmaZ

(

double

s = 1.0

)

set resolution in Z in cm

Definition at line 179 of file BetaBoostEvtVtxGenerator.cc.

References Exception, fSigmaZ, and alignCSCRings::s.

{ 
  if (s>=0 ) {
    fSigmaZ=s; 
  }
  else {
    throw cms::Exception("LogicError")
      << "Error in BetaBoostEvtVtxGenerator::sigmaZ: "
      << "Illegal resolution in Z (negative)";
  }
}

void BetaBoostEvtVtxGenerator::X0 ( double  m = 0 )

inline

set mean in X in cm

Definition at line 68 of file BetaBoostEvtVtxGenerator.cc.

References funct::m.

{ fX0=m; }

void BetaBoostEvtVtxGenerator::Y0 ( double  m = 0 )

inline

set mean in Y in cm

Definition at line 70 of file BetaBoostEvtVtxGenerator.cc.

References funct::m.

{ fY0=m; }

void BetaBoostEvtVtxGenerator::Z0 ( double  m = 0 )

inline

set mean in Z in cm

Definition at line 72 of file BetaBoostEvtVtxGenerator.cc.

References funct::m.

{ fZ0=m; }

Member Data Documentation

double BetaBoostEvtVtxGenerator::alpha_

private

Definition at line 94 of file BetaBoostEvtVtxGenerator.cc.

Referenced by BetaBoostEvtVtxGenerator(), and GetInvLorentzBoost().

double BetaBoostEvtVtxGenerator::beta_

private

Definition at line 96 of file BetaBoostEvtVtxGenerator.cc.

Referenced by BetaBoostEvtVtxGenerator(), and GetInvLorentzBoost().

TMatrixD* BetaBoostEvtVtxGenerator::boost_

private

Definition at line 104 of file BetaBoostEvtVtxGenerator.cc.

Referenced by GetInvLorentzBoost(), and ~BetaBoostEvtVtxGenerator().

double BetaBoostEvtVtxGenerator::falpha

private

Definition at line 101 of file BetaBoostEvtVtxGenerator.cc.

double BetaBoostEvtVtxGenerator::fbetastar

private

Definition at line 100 of file BetaBoostEvtVtxGenerator.cc.

Referenced by BetaBoostEvtVtxGenerator(), and BetaFunction().

double BetaBoostEvtVtxGenerator::femittance

private

Definition at line 100 of file BetaBoostEvtVtxGenerator.cc.

Referenced by BetaBoostEvtVtxGenerator(), and BetaFunction().

double BetaBoostEvtVtxGenerator::fSigmaZ

private

Definition at line 98 of file BetaBoostEvtVtxGenerator.cc.

Referenced by BetaBoostEvtVtxGenerator(), newVertex(), and sigmaZ().

double BetaBoostEvtVtxGenerator::fTimeOffset

private

Definition at line 105 of file BetaBoostEvtVtxGenerator.cc.

Referenced by BetaBoostEvtVtxGenerator(), and newVertex().

HepMC::FourVector* BetaBoostEvtVtxGenerator::fVertex

private

Definition at line 103 of file BetaBoostEvtVtxGenerator.cc.

Referenced by newVertex(), and ~BetaBoostEvtVtxGenerator().

double BetaBoostEvtVtxGenerator::fX0

private

Definition at line 97 of file BetaBoostEvtVtxGenerator.cc.

Referenced by BetaBoostEvtVtxGenerator(), and newVertex().

double BetaBoostEvtVtxGenerator::fY0

private

Definition at line 97 of file BetaBoostEvtVtxGenerator.cc.

Referenced by BetaBoostEvtVtxGenerator(), and newVertex().

double BetaBoostEvtVtxGenerator::fZ0

private

Definition at line 97 of file BetaBoostEvtVtxGenerator.cc.

Referenced by BetaBoostEvtVtxGenerator(), and newVertex().

double BetaBoostEvtVtxGenerator::phi_

private

Definition at line 94 of file BetaBoostEvtVtxGenerator.cc.

Referenced by BetaBoostEvtVtxGenerator(), and GetInvLorentzBoost().

edm::EDGetTokenT<HepMCProduct> BetaBoostEvtVtxGenerator::sourceLabel

private

Definition at line 107 of file BetaBoostEvtVtxGenerator.cc.

Referenced by produce().

bool BetaBoostEvtVtxGenerator::verbosity_

private

Definition at line 109 of file BetaBoostEvtVtxGenerator.cc.

Referenced by GetInvLorentzBoost().