#include <ProtonTransport.h>

Inheritance diagram for ProtonTransport:

Public Member Functions
void	addPartToHepMC (HepMC::GenEvent *)

void	ApplyBeamCorrection (HepMC::GenParticle *p)

void	ApplyBeamCorrection (TLorentzVector &p)

void	clear ()

std::vector< LHCTransportLink > &	getCorrespondenceMap ()

virtual void	process (const HepMC::GenEvent ev, const edm::EventSetup &es, CLHEP::HepRandomEngine engine)=0

	ProtonTransport ()

virtual	~ProtonTransport ()

Protected Types
enum	TransportMode { TransportMode::HECTOR, TransportMode::TOTEM }

Protected Attributes
bool	bApplyZShift

CLHEP::HepRandomEngine *	engine

double	fBeamEnergy

double	fBeamMomentum

double	fBeamXatIP

double	fBeamYatIP

double	fCrossingAngle_45

double	fCrossingAngle_56

double	fPPSRegionStart_45

double	fPPSRegionStart_56

double	fVtxMeanX

double	fVtxMeanY

double	fVtxMeanZ

std::map< unsigned int, TLorentzVector >	m_beamPart

std::vector< LHCTransportLink >	m_CorrespondenceMap

double	m_sig_E

double	m_sigmaSTX

double	m_sigmaSTY

double	m_sigmaSX

double	m_sigmaSY

bool	m_verbosity

std::map< unsigned int, double >	m_xAtTrPoint

std::map< unsigned int, double >	m_yAtTrPoint

TransportMode	MODE

int	NEvent

Detailed Description

Definition at line 16 of file ProtonTransport.h.

Member Enumeration Documentation

enum ProtonTransport::TransportMode

strongprotected

Enumerator
HECTOR
TOTEM

Definition at line 28 of file ProtonTransport.h.

28 {HECTOR,TOTEM};

Constructor & Destructor Documentation

ProtonTransport::ProtonTransport ( )

Definition at line 7 of file ProtonTransport.cc.

7 {};

ProtonTransport::~ProtonTransport ( )

virtual

Definition at line 8 of file ProtonTransport.cc.

8 {};

Member Function Documentation

void ProtonTransport::addPartToHepMC ( HepMC::GenEvent * evt )

Definition at line 14 of file ProtonTransport.cc.

References plotBeamSpotDB::first, fPPSRegionStart_45, fPPSRegionStart_56, GenParticle::GenParticle, mps_splice::line, LogDebug, m_beamPart, m_CorrespondenceMap, m_to_mm, m_verbosity, m_xAtTrPoint, m_yAtTrPoint, NEvent, edm::second(), and ntuplemaker::time.

Referenced by TotemTransport::process(), and HectorTransport::process().

 {
     NEvent++;
     m_CorrespondenceMap.clear();
 
     int direction=0;
     HepMC::GenParticle * gpart;
 
     unsigned int line;
 
     for( auto const& it : m_beamPart ) {
         line = (it ).first;
         gpart = evt->barcode_to_particle( line );
 
         direction=(gpart->momentum().pz()>0)?1:-1;
 
         double ddd=(direction>0)?fPPSRegionStart_45:fabs(fPPSRegionStart_56); // Totem uses negative Z for sector 56 while Hector uses always positive distance
 
         double time = ( ddd*meter - gpart->production_vertex()->position().z()*mm ); // mm
 
 //
 // ATTENTION: at this point, the vertex at PPS is already in mm
 //
         if(ddd == 0.) continue;
         if(m_verbosity) {
                 LogDebug("ProtonTransportEventProcessing") <<"ProtonTransport:: x= "<< (*(m_xAtTrPoint.find(line))).second<< "\n"
                         <<"ProtonTransport:: y= "<< (*(m_yAtTrPoint.find(line))).second<< "\n"
                         <<"ProtonTransport:: z= "<< ddd * direction*m_to_mm << "\n"
                         <<"ProtonTransport:: t= "<< time;
         }
         TLorentzVector const& p_out = (it).second;
 
         HepMC::GenVertex * vert = new HepMC::GenVertex(
                         HepMC::FourVector( (*(m_xAtTrPoint.find(line))).second,
                                 (*(m_yAtTrPoint.find(line))).second,
                                 ddd * direction *m_to_mm, time + time*0.001 ) );
 
         gpart->set_status( 2 );
         vert->add_particle_in( gpart );
         vert->add_particle_out( new HepMC::GenParticle( HepMC::FourVector(p_out.Px(),p_out.Py(),p_out.Pz(),p_out.E()), gpart->pdg_id(), 1, gpart->flow() )) ;
         evt->add_vertex( vert );
 
         int ingoing = (*vert->particles_in_const_begin())->barcode();
         int outgoing = (*vert->particles_out_const_begin())->barcode();
 
         LHCTransportLink theLink(ingoing,outgoing);
         if (m_verbosity) LogDebug("ProtonTransportEventProcessing") << "ProtonTransport:addPartToHepMC: LHCTransportLink " << theLink;
         m_CorrespondenceMap.push_back(theLink);
     }
 }

void ProtonTransport::ApplyBeamCorrection ( HepMC::GenParticle * p )

Definition at line 64 of file ProtonTransport.cc.

Referenced by TotemTransport::transportProton(), and HectorTransport::transportProton().

 {
      TLorentzVector p_out;
      p_out.SetPx(p->momentum().px());
      p_out.SetPy(p->momentum().py());
      p_out.SetPz(p->momentum().pz());
      p_out.SetE(p->momentum().e());
      ApplyBeamCorrection(p_out);
      p->set_momentum(HepMC::FourVector(p_out.Px(),p_out.Py(),p_out.Pz(),p_out.E()));
 }

void ProtonTransport::ApplyBeamCorrection ( TLorentzVector & p )

Definition at line 74 of file ProtonTransport.cc.

References funct::cos(), MillePedeFileConverter_cfg::e, randomXiThetaGunProducer_cfi::energy, engine, fCrossingAngle_45, fCrossingAngle_56, m_sig_E, m_sigmaSTX, m_sigmaSTY, MODE, AlCaHLTBitMon_ParallelJobs::p, Pi, funct::pow(), ProtonMassSQ, funct::sin(), mathSSE::sqrt(), theta(), TOTEM, and urad.

 {
     double theta  = p_out.Theta();
     double thetax = atan(p_out.Px()/fabs(p_out.Pz()));
     double thetay = atan(p_out.Py()/fabs(p_out.Pz()));
     double energy = p_out.E();
     double urad = 1e-6;
 
     int direction = (p_out.Pz()>0)?1:-1;
 
     if (p_out.Pz()<0) theta=TMath::Pi()-theta;
 
     if (MODE==TransportMode::TOTEM) thetax+=(p_out.Pz()>0)?fCrossingAngle_45*urad:fCrossingAngle_56*urad;
 
     double dtheta_x = (double)CLHEP::RandGauss::shoot(engine,0.,m_sigmaSTX);
     double dtheta_y = (double)CLHEP::RandGauss::shoot(engine,0.,m_sigmaSTY);
     double denergy  = (double)CLHEP::RandGauss::shoot(engine,0.,m_sig_E);
 
     double s_theta = sqrt(pow(thetax+dtheta_x*urad,2)+pow(thetay+dtheta_y*urad,2));
     double s_phi = atan2(thetay+dtheta_y*urad,thetax+dtheta_x*urad);
     energy+=denergy;
     double p = sqrt(pow(energy,2)-ProtonMassSQ);
 
     p_out.SetPx((double)p*sin(s_theta)*cos(s_phi));
     p_out.SetPy((double)p*sin(s_theta)*sin(s_phi));
     p_out.SetPz((double)p*(cos(s_theta))*direction);
     p_out.SetE(energy);
 }

void ProtonTransport::clear ( void )

Definition at line 9 of file ProtonTransport.cc.

References m_beamPart, m_xAtTrPoint, and m_yAtTrPoint.

Referenced by PPSSimTrackProducer::produce(), HectorTransport::~HectorTransport(), and TotemTransport::~TotemTransport().

 {
      m_beamPart.clear(); m_xAtTrPoint.clear(); m_yAtTrPoint.clear();
 };

std::vector<LHCTransportLink>& ProtonTransport::getCorrespondenceMap ( )

inline

Definition at line 20 of file ProtonTransport.h.

References hitfit::clear(), ev, GenParticle::GenParticle, AlCaHLTBitMon_ParallelJobs::p, and LaserDQM_cfg::process.

Referenced by PPSSimTrackProducer::produce().

20 { return m_CorrespondenceMap; }

ProtonTransport::m_CorrespondenceMap

std::vector< LHCTransportLink > m_CorrespondenceMap

Definition: ProtonTransport.h:40

virtual void ProtonTransport::process	(	const HepMC::GenEvent *	ev,
		const edm::EventSetup &	es,
		CLHEP::HepRandomEngine *	engine
	)

pure virtual

Implemented in HectorTransport, and TotemTransport.

Referenced by PPSSimTrackProducer::produce().

Member Data Documentation

bool ProtonTransport::bApplyZShift

protected

Definition at line 34 of file ProtonTransport.h.

Referenced by HectorTransport::HectorTransport(), TotemTransport::TotemTransport(), and HectorTransport::transportProton().

CLHEP::HepRandomEngine* ProtonTransport::engine

protected

Definition at line 32 of file ProtonTransport.h.

Referenced by ApplyBeamCorrection(), TotemTransport::process(), and HectorTransport::process().

double ProtonTransport::fBeamEnergy

protected

Definition at line 56 of file ProtonTransport.h.

Referenced by HectorTransport::HectorTransport(), TotemTransport::TotemTransport(), and HectorTransport::transportProton().

double ProtonTransport::fBeamMomentum

protected

Definition at line 55 of file ProtonTransport.h.

Referenced by HectorTransport::HectorTransport(), TotemTransport::TotemTransport(), and HectorTransport::transportProton().

double ProtonTransport::fBeamXatIP

protected

Definition at line 53 of file ProtonTransport.h.

Referenced by HectorTransport::HectorTransport(), TotemTransport::TotemTransport(), TotemTransport::transportProton(), and HectorTransport::transportProton().

double ProtonTransport::fBeamYatIP

protected

Definition at line 54 of file ProtonTransport.h.

Referenced by HectorTransport::HectorTransport(), TotemTransport::TotemTransport(), TotemTransport::transportProton(), and HectorTransport::transportProton().

double ProtonTransport::fCrossingAngle_45

protected

Definition at line 37 of file ProtonTransport.h.

Referenced by ApplyBeamCorrection(), HectorTransport::HectorTransport(), TotemTransport::TotemTransport(), TotemTransport::transportProton(), and HectorTransport::transportProton().

double ProtonTransport::fCrossingAngle_56

protected

Definition at line 38 of file ProtonTransport.h.

Referenced by ApplyBeamCorrection(), HectorTransport::HectorTransport(), TotemTransport::TotemTransport(), TotemTransport::transportProton(), and HectorTransport::transportProton().

double ProtonTransport::fPPSRegionStart_45

protected

Definition at line 36 of file ProtonTransport.h.

Referenced by addPartToHepMC(), HectorTransport::HectorTransport(), and TotemTransport::TotemTransport().

double ProtonTransport::fPPSRegionStart_56

protected

Definition at line 35 of file ProtonTransport.h.

Referenced by addPartToHepMC(), HectorTransport::HectorTransport(), and TotemTransport::TotemTransport().

double ProtonTransport::fVtxMeanX

protected

Definition at line 50 of file ProtonTransport.h.

Referenced by HectorTransport::HectorTransport(), TotemTransport::TotemTransport(), TotemTransport::transportProton(), and HectorTransport::transportProton().

double ProtonTransport::fVtxMeanY

protected

Definition at line 51 of file ProtonTransport.h.

Referenced by HectorTransport::HectorTransport(), TotemTransport::TotemTransport(), TotemTransport::transportProton(), and HectorTransport::transportProton().

double ProtonTransport::fVtxMeanZ

protected

Definition at line 52 of file ProtonTransport.h.

Referenced by HectorTransport::HectorTransport(), TotemTransport::TotemTransport(), and TotemTransport::transportProton().

std::map<unsigned int, TLorentzVector> ProtonTransport::m_beamPart

protected

Definition at line 41 of file ProtonTransport.h.

Referenced by addPartToHepMC(), clear(), HectorTransport::genProtonsLoop(), TotemTransport::process(), TotemTransport::transportProton(), and HectorTransport::transportProton().

std::vector<LHCTransportLink> ProtonTransport::m_CorrespondenceMap

protected

Definition at line 40 of file ProtonTransport.h.

Referenced by addPartToHepMC().

double ProtonTransport::m_sig_E

protected

Definition at line 49 of file ProtonTransport.h.

Referenced by ApplyBeamCorrection(), HectorTransport::HectorTransport(), and TotemTransport::TotemTransport().

double ProtonTransport::m_sigmaSTX

protected

Definition at line 47 of file ProtonTransport.h.

Referenced by ApplyBeamCorrection(), HectorTransport::HectorTransport(), and TotemTransport::TotemTransport().

double ProtonTransport::m_sigmaSTY

protected

Definition at line 48 of file ProtonTransport.h.

Referenced by ApplyBeamCorrection(), HectorTransport::HectorTransport(), and TotemTransport::TotemTransport().

double ProtonTransport::m_sigmaSX

protected

Definition at line 45 of file ProtonTransport.h.

Referenced by HectorTransport::HectorTransport().

double ProtonTransport::m_sigmaSY

protected

Definition at line 46 of file ProtonTransport.h.

Referenced by HectorTransport::HectorTransport().

bool ProtonTransport::m_verbosity

protected

Definition at line 31 of file ProtonTransport.h.

Referenced by addPartToHepMC(), HectorTransport::HectorTransport(), HectorTransport::setBeamLine(), and HectorTransport::transportProton().

std::map<unsigned int, double> ProtonTransport::m_xAtTrPoint

protected

Definition at line 42 of file ProtonTransport.h.

Referenced by addPartToHepMC(), clear(), TotemTransport::transportProton(), and HectorTransport::transportProton().

std::map<unsigned int, double> ProtonTransport::m_yAtTrPoint

protected

Definition at line 43 of file ProtonTransport.h.

Referenced by addPartToHepMC(), clear(), TotemTransport::transportProton(), and HectorTransport::transportProton().

TransportMode ProtonTransport::MODE

protected

Definition at line 29 of file ProtonTransport.h.

Referenced by ApplyBeamCorrection(), HectorTransport::HectorTransport(), and TotemTransport::TotemTransport().

int ProtonTransport::NEvent

protected

Definition at line 30 of file ProtonTransport.h.

Referenced by addPartToHepMC().

Public Member Functions

Protected Types

Protected Attributes

Detailed Description

Member Enumeration Documentation

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation