MixBoostEvtVtxGenerator Class Reference

Inheritance diagram for MixBoostEvtVtxGenerator:

Public Member Functions
void	Alpha (double m=0)
	angle between crossing plane and horizontal plane More...
void	Beta (double m=0)
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 *	getRecVertex (edm::Event &)
virtual HepMC::FourVector *	getVertex (edm::Event &)
	MixBoostEvtVtxGenerator (const edm::ParameterSet &p)
	MixBoostEvtVtxGenerator (const MixBoostEvtVtxGenerator &p)=delete
MixBoostEvtVtxGenerator &	operator= (const MixBoostEvtVtxGenerator &rhs)=delete
void	Phi (double m=0)
	set half crossing angle More...
void	produce (edm::Event &, const edm::EventSetup &) override
	return a new event vertex More...
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...
	~MixBoostEvtVtxGenerator () override
Public Member Functions inherited from edm::EDProducer
	EDProducer ()
SerialTaskQueue *	globalLuminosityBlocksQueue ()
SerialTaskQueue *	globalRunsQueue ()
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)
	~ProducerBase () noexcept(false) override
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
ESProxyIndex const *	esGetTokenIndices (edm::Transition iTrans) const
std::vector< ESProxyIndex > const &	esGetTokenIndicesVector (edm::Transition iTrans) const
std::vector< ESRecordIndex > const &	esGetTokenRecordIndicesVector (edm::Transition iTrans) const
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::array< std::vector< ModuleDescription const * > *, NumBranchTypes > &modulesAll, std::vector< ModuleProcessName > &modulesInPreviousProcesses, 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	selectInputProcessBlocks (ProductRegistry const &productRegistry, ProcessBlockHelperBase const &processBlockHelperBase)
ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const
void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
void	updateLookup (eventsetup::ESRecordsToProxyIndices const &)
virtual	~EDConsumerBase () noexcept(false)

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
edm::EDGetTokenT < CrossingFrame< HepMCProduct > >	mixLabel
double	phi_
edm::EDGetTokenT< HepMCProduct >	signalLabel
bool	useRecVertex
edm::EDGetTokenT < reco::VertexCollection >	vtxLabel
std::vector< double >	vtxOffset

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)
static bool	wantsGlobalLuminosityBlocks ()
static bool	wantsGlobalRuns ()
static bool	wantsInputProcessBlocks ()
static bool	wantsProcessBlocks ()
static bool	wantsStreamLuminosityBlocks ()
static bool	wantsStreamRuns ()
Protected Member Functions inherited from edm::ProducerBase
ProducesCollector	producesCollector ()
Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)
template<BranchType B = InEvent>
EDConsumerBaseAdaptor< B >	consumes (edm::InputTag tag) noexcept
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 ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes ()
template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag const &tag)
template<Transition Tr = Transition::Event>
constexpr auto	esConsumes ()
template<Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag tag)
template<Transition Tr = Transition::Event>
ESGetTokenGeneric	esConsumes (eventsetup::EventSetupRecordKey const &iRecord, eventsetup::DataKey const &iKey)
	Used with EventSetupRecord::doGet. More...
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)
void	resetItemsToGetFrom (BranchType iType)

Detailed Description

Definition at line 50 of file MixBoostEvtVtxGenerator.cc.

Constructor & Destructor Documentation

MixBoostEvtVtxGenerator::MixBoostEvtVtxGenerator

(

const edm::ParameterSet &

p

)

Definition at line 110 of file MixBoostEvtVtxGenerator.cc.

References alpha_, beta_, edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), phi_, and vtxOffset.

    : fVertex(nullptr),
      boost_(nullptr),
      fTimeOffset(0),
      vtxLabel(mayConsume<reco::VertexCollection>(pset.getParameter<edm::InputTag>("vtxLabel"))),
      signalLabel(consumes<HepMCProduct>(pset.getParameter<edm::InputTag>("signalLabel"))),
      mixLabel(consumes<CrossingFrame<HepMCProduct> >(pset.getParameter<edm::InputTag>("mixLabel"))),
      useRecVertex(pset.exists("useRecVertex") ? pset.getParameter<bool>("useRecVertex") : false) {
  beta_ = pset.getParameter<double>("Beta");
  alpha_ = 0;
  phi_ = 0;
  if (pset.exists("Alpha")) {
    alpha_ = pset.getParameter<double>("Alpha") * radian;
    phi_ = pset.getParameter<double>("Phi") * radian;
  }

  vtxOffset.resize(3);
  if (pset.exists("vtxOffset"))
    vtxOffset = pset.getParameter<std::vector<double> >("vtxOffset");

  produces<edm::HepMCProduct>();
}

MixBoostEvtVtxGenerator::MixBoostEvtVtxGenerator ( const MixBoostEvtVtxGenerator &  p )

delete

Copy constructor

MixBoostEvtVtxGenerator::~MixBoostEvtVtxGenerator ( )

override

Definition at line 133 of file MixBoostEvtVtxGenerator.cc.

References boost_, and fVertex.

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

Member Function Documentation

void MixBoostEvtVtxGenerator::Alpha ( double  m = 0 )

inline

angle between crossing plane and horizontal plane

Definition at line 78 of file MixBoostEvtVtxGenerator.cc.

References visualization-live-secondInstance_cfg::m.

{ alpha_ = m; }

void MixBoostEvtVtxGenerator::Beta ( double  m = 0 )

inline

Definition at line 79 of file MixBoostEvtVtxGenerator.cc.

References visualization-live-secondInstance_cfg::m.

{ beta_ = m; }

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

beta function

Definition at line 140 of file MixBoostEvtVtxGenerator.cc.

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

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

void MixBoostEvtVtxGenerator::betastar ( double  m = 0 )

inline

set beta_star

Definition at line 82 of file MixBoostEvtVtxGenerator.cc.

References visualization-live-secondInstance_cfg::m.

{ fbetastar = m; }

void MixBoostEvtVtxGenerator::emittance ( double  m = 0 )

inline

emittance (no the normalized)

Definition at line 84 of file MixBoostEvtVtxGenerator.cc.

References visualization-live-secondInstance_cfg::m.

{ femittance = m; }

TMatrixD * MixBoostEvtVtxGenerator::GetInvLorentzBoost ( )

virtual

Definition at line 153 of file MixBoostEvtVtxGenerator.cc.

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

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();

  //cout<<"Boosting with beta : "<<beta_<<endl;

  boost_ = new TMatrixD(tmpboostXYZ);
  boost_->Print();

  return boost_;
}

HepMC::FourVector * MixBoostEvtVtxGenerator::getRecVertex ( edm::Event &  evt )

virtual

Definition at line 258 of file MixBoostEvtVtxGenerator.cc.

References fVertex, edm::Event::getByToken(), input, vtxLabel, and vtxOffset.

Referenced by produce().

                                                                 {
  Handle<reco::VertexCollection> input;
  evt.getByToken(vtxLabel, input);

  double aX, aY, aZ;

  aX = input->begin()->position().x() + vtxOffset[0];
  aY = input->begin()->position().y() + vtxOffset[1];
  aZ = input->begin()->position().z() + vtxOffset[2];

  if (!fVertex)
    fVertex = new HepMC::FourVector();
  fVertex->set(10.0 * aX, 10.0 * aY, 10.0 * aZ, 0.0);  // HepMC positions in mm (RECO in cm)

  return fVertex;
}

HepMC::FourVector * MixBoostEvtVtxGenerator::getVertex ( edm::Event &  evt )

virtual

Definition at line 216 of file MixBoostEvtVtxGenerator.cc.

References gather_cfg::cout, Exception, fVertex, edm::Event::getByToken(), reco_application_tbsim_DetSim-Digi_cfg::mix, mixLabel, edm::Handle< T >::product(), and DiDispStaMuonMonitor_cfi::pt.

Referenced by produce().

                                                              {
  const HepMC::GenEvent* inev = nullptr;

  Handle<CrossingFrame<HepMCProduct> > cf;
  evt.getByToken(mixLabel, cf);
  MixCollection<HepMCProduct> mix(cf.product());

  const HepMCProduct& bkg = mix.getObject(1);
  if (!(bkg.isVtxGenApplied())) {
    throw cms::Exception("MatchVtx") << "Input background does not have smeared vertex!" << endl;
  } else {
    inev = bkg.GetEvent();
  }

  HepMC::GenVertex* genvtx = inev->signal_process_vertex();
  if (!genvtx) {
    cout << "No Signal Process Vertex!" << endl;
    HepMC::GenEvent::particle_const_iterator pt = inev->particles_begin();
    HepMC::GenEvent::particle_const_iterator ptend = inev->particles_end();
    while (!genvtx || (genvtx->particles_in_size() == 1 && pt != ptend)) {
      if (!genvtx)
        cout << "No Gen Vertex!" << endl;
      if (pt == ptend)
        cout << "End reached!" << endl;
      genvtx = (*pt)->production_vertex();
      ++pt;
    }
  }
  double aX, aY, aZ, aT;

  aX = genvtx->position().x();
  aY = genvtx->position().y();
  aZ = genvtx->position().z();
  aT = genvtx->position().t();

  if (!fVertex)
    fVertex = new HepMC::FourVector();
  fVertex->set(aX, aY, aZ, aT);

  return fVertex;
}

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

delete

Copy assignment operator

void MixBoostEvtVtxGenerator::Phi ( double  m = 0 )

inline

set half crossing angle

Definition at line 76 of file MixBoostEvtVtxGenerator.cc.

References visualization-live-secondInstance_cfg::m.

{ phi_ = m; }

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

overridevirtual

return a new event vertex

Implements edm::EDProducer.

Definition at line 275 of file MixBoostEvtVtxGenerator.cc.

References edm::Event::getByToken(), GetInvLorentzBoost(), getRecVertex(), getVertex(), eostools::move(), edm::Event::put(), signalLabel, and useRecVertex.

                                                                   {
  Handle<HepMCProduct> HepUnsmearedMCEvt;
  evt.getByToken(signalLabel, 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->boostToLab(GetInvLorentzBoost(), "vertex");
  HepMCEvt->boostToLab(GetInvLorentzBoost(), "momentum");

  HepMCEvt->applyVtxGen(useRecVertex ? getRecVertex(evt) : getVertex(evt));

  evt.put(std::move(HepMCEvt));
  return;
}

void MixBoostEvtVtxGenerator::sigmaZ

(

double

s = 1.0

)

set resolution in Z in cm

Definition at line 144 of file MixBoostEvtVtxGenerator.cc.

References Exception, fSigmaZ, and alignCSCRings::s.

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

void MixBoostEvtVtxGenerator::X0 ( double  m = 0 )

inline

set mean in X in cm

Definition at line 69 of file MixBoostEvtVtxGenerator.cc.

References visualization-live-secondInstance_cfg::m.

{ fX0 = m; }

void MixBoostEvtVtxGenerator::Y0 ( double  m = 0 )

inline

set mean in Y in cm

Definition at line 71 of file MixBoostEvtVtxGenerator.cc.

References visualization-live-secondInstance_cfg::m.

{ fY0 = m; }

void MixBoostEvtVtxGenerator::Z0 ( double  m = 0 )

inline

set mean in Z in cm

Definition at line 73 of file MixBoostEvtVtxGenerator.cc.

References visualization-live-secondInstance_cfg::m.

{ fZ0 = m; }

Member Data Documentation

double MixBoostEvtVtxGenerator::alpha_

private

Definition at line 90 of file MixBoostEvtVtxGenerator.cc.

Referenced by GetInvLorentzBoost(), and MixBoostEvtVtxGenerator().

double MixBoostEvtVtxGenerator::beta_

private

Definition at line 92 of file MixBoostEvtVtxGenerator.cc.

Referenced by GetInvLorentzBoost(), and MixBoostEvtVtxGenerator().

TMatrixD* MixBoostEvtVtxGenerator::boost_

private

Definition at line 100 of file MixBoostEvtVtxGenerator.cc.

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

double MixBoostEvtVtxGenerator::falpha

private

Definition at line 97 of file MixBoostEvtVtxGenerator.cc.

double MixBoostEvtVtxGenerator::fbetastar

private

Definition at line 96 of file MixBoostEvtVtxGenerator.cc.

Referenced by BetaFunction().

double MixBoostEvtVtxGenerator::femittance

private

Definition at line 96 of file MixBoostEvtVtxGenerator.cc.

Referenced by BetaFunction().

double MixBoostEvtVtxGenerator::fSigmaZ

private

Definition at line 94 of file MixBoostEvtVtxGenerator.cc.

Referenced by sigmaZ().

double MixBoostEvtVtxGenerator::fTimeOffset

private

Definition at line 101 of file MixBoostEvtVtxGenerator.cc.

HepMC::FourVector* MixBoostEvtVtxGenerator::fVertex

private

Definition at line 99 of file MixBoostEvtVtxGenerator.cc.

Referenced by getRecVertex(), getVertex(), and ~MixBoostEvtVtxGenerator().

double MixBoostEvtVtxGenerator::fX0

private

Definition at line 93 of file MixBoostEvtVtxGenerator.cc.

double MixBoostEvtVtxGenerator::fY0

private

Definition at line 93 of file MixBoostEvtVtxGenerator.cc.

double MixBoostEvtVtxGenerator::fZ0

private

Definition at line 93 of file MixBoostEvtVtxGenerator.cc.

edm::EDGetTokenT<CrossingFrame<HepMCProduct> > MixBoostEvtVtxGenerator::mixLabel

private

Definition at line 105 of file MixBoostEvtVtxGenerator.cc.

Referenced by getVertex().

double MixBoostEvtVtxGenerator::phi_

private

Definition at line 90 of file MixBoostEvtVtxGenerator.cc.

Referenced by GetInvLorentzBoost(), and MixBoostEvtVtxGenerator().

edm::EDGetTokenT<HepMCProduct> MixBoostEvtVtxGenerator::signalLabel

private

Definition at line 104 of file MixBoostEvtVtxGenerator.cc.

Referenced by produce().

bool MixBoostEvtVtxGenerator::useRecVertex

private

Definition at line 106 of file MixBoostEvtVtxGenerator.cc.

Referenced by produce().

edm::EDGetTokenT<reco::VertexCollection> MixBoostEvtVtxGenerator::vtxLabel

private

Definition at line 103 of file MixBoostEvtVtxGenerator.cc.

Referenced by getRecVertex().

std::vector<double> MixBoostEvtVtxGenerator::vtxOffset

private

Definition at line 107 of file MixBoostEvtVtxGenerator.cc.

Referenced by getRecVertex(), and MixBoostEvtVtxGenerator().