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)
virtual HepMC::FourVector *	newVertex ()
	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	~MixBoostEvtVtxGenerator ()
Public Member Functions inherited from edm::EDProducer
	EDProducer ()
ModuleDescription const &	moduleDescription () const
virtual	~EDProducer ()
Public Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
	ProducerBase ()
void	registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...
virtual	~ProducerBase ()
Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const
	EDConsumerBase ()
ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const
void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const
void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const
std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const
void	labelsForToken (EDGetToken iToken, Labels &oLabels) const
void	modulesDependentUpon (std::string const &iProcessName, std::string const &iModuleLabel, bool iPrint, std::vector< char const * > &oModuleLabels) const
void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const
bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const
bool	registeredToConsumeMany (TypeID const &, BranchType) const
void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)
virtual	~EDConsumerBase ()

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

Private Attributes
double	alpha_
double	beta_
TMatrixD *	boost_
double	falpha
double	fbetastar
double	femittance
CLHEP::RandGaussQ *	fRandom
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
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 51 of file MixBoostEvtVtxGenerator.cc.

Constructor & Destructor Documentation

MixBoostEvtVtxGenerator::MixBoostEvtVtxGenerator

(

const edm::ParameterSet &

p

)

Definition at line 117 of file MixBoostEvtVtxGenerator.cc.

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

                                                                             :
  fVertex(0), boost_(0), 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)
{ 

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

  produces<edm::HepMCProduct>();
  
}

MixBoostEvtVtxGenerator::~MixBoostEvtVtxGenerator ( )

virtual

Definition at line 132 of file MixBoostEvtVtxGenerator.cc.

References boost_, fRandom, and fVertex.

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

MixBoostEvtVtxGenerator::MixBoostEvtVtxGenerator ( const MixBoostEvtVtxGenerator &  p )

private

Copy constructor

Member Function Documentation

void MixBoostEvtVtxGenerator::Alpha ( double  m = 0 )

inline

angle between crossing plane and horizontal plane

Definition at line 77 of file MixBoostEvtVtxGenerator.cc.

References visualization-live-secondInstance_cfg::m.

{ alpha_=m; }

void MixBoostEvtVtxGenerator::Beta ( double  m = 0 )

inline

Definition at line 78 of file MixBoostEvtVtxGenerator.cc.

References visualization-live-secondInstance_cfg::m.

{ beta_=m; }

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

beta function

Definition at line 168 of file MixBoostEvtVtxGenerator.cc.

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

Referenced by newVertex().

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

}

void MixBoostEvtVtxGenerator::betastar ( double  m = 0 )

inline

set beta_star

Definition at line 81 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 83 of file MixBoostEvtVtxGenerator.cc.

References visualization-live-secondInstance_cfg::m.

{ femittance=m; }

TMatrixD * MixBoostEvtVtxGenerator::GetInvLorentzBoost ( )

virtual

Definition at line 187 of file MixBoostEvtVtxGenerator.cc.

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

                                                      {

    //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=tmpboost*tmpboostZ;
       tmpboost.Invert();



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

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

virtual

Definition at line 293 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 251 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 EnergyCorrector::pt.

Referenced by produce().

                                                              {
  
  const HepMC::GenEvent* inev = 0;

  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;
  
}

HepMC::FourVector * MixBoostEvtVtxGenerator::newVertex ( )

virtual

return a new event vertex

Definition at line 141 of file MixBoostEvtVtxGenerator.cc.

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

                                                    {

    
    double X,Y,Z;
    
    double tmp_sigz = fRandom->fire(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 = fRandom->fire(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 = fRandom->fire(0.,tmp_sigy) + fY0; // + Z*fdydz;

    double tmp_sigt = fRandom->fire(0., fSigmaZ);
    double T = tmp_sigt + fTimeOffset; 

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

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

private

Copy assignment operator

void MixBoostEvtVtxGenerator::Phi ( double  m = 0 )

inline

set half crossing angle

Definition at line 75 of file MixBoostEvtVtxGenerator.cc.

References visualization-live-secondInstance_cfg::m.

{ phi_=m; }

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

overridevirtual

Implements edm::EDProducer.

Definition at line 313 of file MixBoostEvtVtxGenerator.cc.

References edm::Event::getByToken(), getRecVertex(), getVertex(), eostools::move(), edm::Event::put(), reco::return(), 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->applyVtxGen( useRecVertex ? getRecVertex(evt) : getVertex(evt) ) ;
 
  //   HepMCEvt->boostToLab( GetInvLorentzBoost(), "vertex" );
  //   HepMCEvt->boostToLab( GetInvLorentzBoost(), "momentum" );
  
  evt.put(std::move(HepMCEvt));
  return ;
}

void MixBoostEvtVtxGenerator::sigmaZ

(

double

s = 1.0

)

set resolution in Z in cm

Definition at line 175 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 68 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 70 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 72 of file MixBoostEvtVtxGenerator.cc.

References visualization-live-secondInstance_cfg::m.

{ fZ0=m; }

Member Data Documentation

double MixBoostEvtVtxGenerator::alpha_

private

Definition at line 94 of file MixBoostEvtVtxGenerator.cc.

Referenced by GetInvLorentzBoost().

double MixBoostEvtVtxGenerator::beta_

private

Definition at line 96 of file MixBoostEvtVtxGenerator.cc.

Referenced by GetInvLorentzBoost().

TMatrixD* MixBoostEvtVtxGenerator::boost_

private

Definition at line 104 of file MixBoostEvtVtxGenerator.cc.

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

double MixBoostEvtVtxGenerator::falpha

private

Definition at line 101 of file MixBoostEvtVtxGenerator.cc.

double MixBoostEvtVtxGenerator::fbetastar

private

Definition at line 100 of file MixBoostEvtVtxGenerator.cc.

Referenced by BetaFunction().

double MixBoostEvtVtxGenerator::femittance

private

Definition at line 100 of file MixBoostEvtVtxGenerator.cc.

Referenced by BetaFunction().

CLHEP::RandGaussQ* MixBoostEvtVtxGenerator::fRandom

private

Definition at line 107 of file MixBoostEvtVtxGenerator.cc.

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

double MixBoostEvtVtxGenerator::fSigmaZ

private

Definition at line 98 of file MixBoostEvtVtxGenerator.cc.

Referenced by newVertex(), and sigmaZ().

double MixBoostEvtVtxGenerator::fTimeOffset

private

Definition at line 105 of file MixBoostEvtVtxGenerator.cc.

Referenced by newVertex().

HepMC::FourVector* MixBoostEvtVtxGenerator::fVertex

private

Definition at line 103 of file MixBoostEvtVtxGenerator.cc.

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

double MixBoostEvtVtxGenerator::fX0

private

Definition at line 97 of file MixBoostEvtVtxGenerator.cc.

Referenced by newVertex().

double MixBoostEvtVtxGenerator::fY0

private

Definition at line 97 of file MixBoostEvtVtxGenerator.cc.

Referenced by newVertex().

double MixBoostEvtVtxGenerator::fZ0

private

Definition at line 97 of file MixBoostEvtVtxGenerator.cc.

Referenced by newVertex().

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

private

Definition at line 111 of file MixBoostEvtVtxGenerator.cc.

Referenced by getVertex().

double MixBoostEvtVtxGenerator::phi_

private

Definition at line 94 of file MixBoostEvtVtxGenerator.cc.

Referenced by GetInvLorentzBoost().

edm::EDGetTokenT<HepMCProduct> MixBoostEvtVtxGenerator::signalLabel

private

Definition at line 110 of file MixBoostEvtVtxGenerator.cc.

Referenced by produce().

bool MixBoostEvtVtxGenerator::useRecVertex

private

Definition at line 112 of file MixBoostEvtVtxGenerator.cc.

Referenced by produce().

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

private

Definition at line 109 of file MixBoostEvtVtxGenerator.cc.

Referenced by getRecVertex().

std::vector<double> MixBoostEvtVtxGenerator::vtxOffset

private

Definition at line 113 of file MixBoostEvtVtxGenerator.cc.

Referenced by getRecVertex(), and MixBoostEvtVtxGenerator().