#include <BaseProtonTransport.h>

Inheritance diagram for BaseProtonTransport:

Public Member Functions
void	addPartToHepMC (const HepMC::GenEvent , HepMC::GenEvent )

void	ApplyBeamCorrection (HepMC::GenParticle *p)

void	ApplyBeamCorrection (TLorentzVector &p)

	BaseProtonTransport (const edm::ParameterSet &iConfig)

double	beamEnergy ()

double	beamMomentum ()

void	clear ()

std::vector< LHCTransportLink > &	getCorrespondenceMap ()

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

void	setBeamEnergy (double e)

virtual	~BaseProtonTransport ()

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

Protected Attributes
bool	bApplyZShift

std::string	beam1Filename_

std::string	beam2Filename_

double	beamEnergy_

double	beamMomentum_

CLHEP::HepRandomEngine *	engine_

double	etaCut_

double	fBeamXatIP

double	fBeamYatIP

double	fCrossingAngleX_45

double	fCrossingAngleX_56

double	fCrossingAngleY_45

double	fCrossingAngleY_56

double	fPPSRegionStart_45

double	fPPSRegionStart_56

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

std::map< unsigned int, double >	m_xAtTrPoint

std::map< unsigned int, double >	m_yAtTrPoint

TransportMode	MODE

double	momentumCut_

int	NEvent

bool	verbosity_

Detailed Description

Definition at line 17 of file BaseProtonTransport.h.

Member Enumeration Documentation

◆ TransportMode

enum BaseProtonTransport::TransportMode

strongprotected

Enumerator
HECTOR
TOTEM
OPTICALFUNCTIONS

Definition at line 41 of file BaseProtonTransport.h.

41 { HECTOR, TOTEM, OPTICALFUNCTIONS };

Constructor & Destructor Documentation

◆ BaseProtonTransport()

BaseProtonTransport::BaseProtonTransport ( const edm::ParameterSet & iConfig )

Definition at line 8 of file BaseProtonTransport.cc.

     : verbosity_(iConfig.getParameter<bool>("Verbosity")),
       bApplyZShift(iConfig.getParameter<bool>("ApplyZShift")),
       fPPSRegionStart_45(iConfig.getParameter<double>("PPSRegionStart_45")),
       fPPSRegionStart_56(iConfig.getParameter<double>("PPSRegionStart_56")),
       etaCut_(iConfig.getParameter<double>("EtaCut")),
       momentumCut_(iConfig.getParameter<double>("MomentumCut")) {
   beamMomentum_ = sqrt(beamEnergy_ * beamEnergy_ - ProtonMassSQ);
 }

References beamEnergy_, beamMomentum_, ProtonMassSQ, and mathSSE::sqrt().

◆ ~BaseProtonTransport()

virtual BaseProtonTransport::~BaseProtonTransport ( )

inlinevirtual

Definition at line 20 of file BaseProtonTransport.h.

20 { this->clear(); };

References clear().

Member Function Documentation

◆ addPartToHepMC()

void BaseProtonTransport::addPartToHepMC	(	const HepMC::GenEvent *	in_evt,
		HepMC::GenEvent *	evt
	)

Definition at line 55 of file BaseProtonTransport.cc.

                                                                                         {
   NEvent++;
   m_CorrespondenceMap.clear();
  
   int direction = 0;
   HepMC::GenParticle* gpart;
  
   unsigned int line;
  
   for (auto const& it : m_beamPart) {
     line = (it).first;
     gpart = in_evt->barcode_to_particle(line);
  
     direction = (gpart->momentum().pz() > 0) ? 1 : -1;
  
     // Totem uses negative Z for sector 56 while Hector uses always positive distance
     double ddd = (direction > 0) ? fPPSRegionStart_45 : fabs(fPPSRegionStart_56);
  
     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 (verbosity_) {
       LogDebug("BaseProtonTransportEventProcessing")
           << "BaseProtonTransport:: x= " << (*(m_xAtTrPoint.find(line))).second << "\n"
           << "BaseProtonTransport:: y= " << (*(m_yAtTrPoint.find(line))).second << "\n"
           << "BaseProtonTransport:: z= " << ddd * direction * m_to_mm << "\n"
           << "BaseProtonTransport:: 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));
  
     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 = 0;  //do not attach the incoming proton to this vertex to avoid duplicating data
     int outgoing = (*vert->particles_out_const_begin())->barcode();
  
     LHCTransportLink theLink(ingoing, outgoing);
     if (verbosity_)
       LogDebug("BaseProtonTransportEventProcessing")
           << "BaseProtonTransport:addPartToHepMC: LHCTransportLink " << theLink;
     m_CorrespondenceMap.push_back(theLink);
   }
 }

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

Referenced by ProtonTransport::addPartToHepMC().

◆ ApplyBeamCorrection() [1/2]

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

Definition at line 17 of file BaseProtonTransport.cc.

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

References AlCaHLTBitMon_ParallelJobs::p.

Referenced by TotemTransport::transportProton().

◆ ApplyBeamCorrection() [2/2]

void BaseProtonTransport::ApplyBeamCorrection ( TLorentzVector & p )

Definition at line 26 of file BaseProtonTransport.cc.

                                                                    {
   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) ? fCrossingAngleX_45 * urad : fCrossingAngleX_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);
 }

References funct::cos(), MillePedeFileConverter_cfg::e, HCALHighEnergyHPDFilter_cfi::energy, engine_, fCrossingAngleX_45, fCrossingAngleX_56, m_sig_E, m_sigmaSTX, m_sigmaSTY, MODE, AlCaHLTBitMon_ParallelJobs::p, Pi, funct::pow(), ProtonMassSQ, funct::sin(), mathSSE::sqrt(), theta(), TOTEM, and urad.

◆ beamEnergy()

double BaseProtonTransport::beamEnergy ( )

inline

Definition at line 37 of file BaseProtonTransport.h.

37 { return beamEnergy_; };

References beamEnergy_.

◆ beamMomentum()

double BaseProtonTransport::beamMomentum ( )

inline

Definition at line 38 of file BaseProtonTransport.h.

38 { return beamMomentum_; };

References beamMomentum_.

Referenced by OpticalFunctionsTransport::transportProton().

◆ clear()

void BaseProtonTransport::clear ( void )

Definition at line 109 of file BaseProtonTransport.cc.

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

References m_beamPart, m_CorrespondenceMap, m_xAtTrPoint, and m_yAtTrPoint.

Referenced by OpticalFunctionsTransport::process(), TotemTransport::process(), HectorTransport::process(), and ~BaseProtonTransport().

◆ getCorrespondenceMap()

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

inline

Definition at line 22 of file BaseProtonTransport.h.

22 { return m_CorrespondenceMap; }

References m_CorrespondenceMap.

Referenced by ProtonTransport::getCorrespondenceMap().

◆ process()

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

pure virtual

Implemented in HectorTransport, TotemTransport, and OpticalFunctionsTransport.

Referenced by ProtonTransport::process().

◆ setBeamEnergy()

void BaseProtonTransport::setBeamEnergy ( double e )

inline

Definition at line 32 of file BaseProtonTransport.h.

                                {
     beamEnergy_ = e;
     beamMomentum_ = sqrt(beamEnergy_ * beamEnergy_ - ProtonMassSQ);
   }

References beamEnergy_, beamMomentum_, MillePedeFileConverter_cfg::e, ProtonMassSQ, and mathSSE::sqrt().

Member Data Documentation

◆ bApplyZShift

bool BaseProtonTransport::bApplyZShift

protected

Definition at line 52 of file BaseProtonTransport.h.

◆ beam1Filename_

std::string BaseProtonTransport::beam1Filename_

protected

Definition at line 54 of file BaseProtonTransport.h.

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

◆ beam2Filename_

std::string BaseProtonTransport::beam2Filename_

protected

Definition at line 55 of file BaseProtonTransport.h.

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

◆ beamEnergy_

double BaseProtonTransport::beamEnergy_

protected

Definition at line 65 of file BaseProtonTransport.h.

Referenced by BaseProtonTransport(), beamEnergy(), HectorTransport::HectorTransport(), setBeamEnergy(), TotemTransport::TotemTransport(), and HectorTransport::transportProton().

◆ beamMomentum_

double BaseProtonTransport::beamMomentum_

protected

Definition at line 64 of file BaseProtonTransport.h.

Referenced by BaseProtonTransport(), beamMomentum(), setBeamEnergy(), and HectorTransport::transportProton().

◆ engine_

CLHEP::HepRandomEngine* BaseProtonTransport::engine_

protected

Definition at line 45 of file BaseProtonTransport.h.

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

◆ etaCut_

double BaseProtonTransport::etaCut_

protected

Definition at line 66 of file BaseProtonTransport.h.

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

◆ fBeamXatIP

double BaseProtonTransport::fBeamXatIP

protected

Definition at line 74 of file BaseProtonTransport.h.

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

◆ fBeamYatIP

double BaseProtonTransport::fBeamYatIP

protected

Definition at line 75 of file BaseProtonTransport.h.

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

◆ fCrossingAngleX_45

double BaseProtonTransport::fCrossingAngleX_45

protected

Definition at line 59 of file BaseProtonTransport.h.

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

◆ fCrossingAngleX_56

double BaseProtonTransport::fCrossingAngleX_56

protected

Definition at line 60 of file BaseProtonTransport.h.

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

◆ fCrossingAngleY_45

double BaseProtonTransport::fCrossingAngleY_45

protected

Definition at line 61 of file BaseProtonTransport.h.

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

◆ fCrossingAngleY_56

double BaseProtonTransport::fCrossingAngleY_56

protected

Definition at line 62 of file BaseProtonTransport.h.

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

◆ fPPSRegionStart_45

double BaseProtonTransport::fPPSRegionStart_45

protected

Definition at line 57 of file BaseProtonTransport.h.

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

◆ fPPSRegionStart_56

double BaseProtonTransport::fPPSRegionStart_56

protected

Definition at line 58 of file BaseProtonTransport.h.

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

◆ m_beamPart

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

protected

Definition at line 47 of file BaseProtonTransport.h.

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

◆ m_CorrespondenceMap

std::vector<LHCTransportLink> BaseProtonTransport::m_CorrespondenceMap

protected

Definition at line 46 of file BaseProtonTransport.h.

Referenced by addPartToHepMC(), clear(), and getCorrespondenceMap().

◆ m_sig_E

double BaseProtonTransport::m_sig_E

protected

Definition at line 73 of file BaseProtonTransport.h.

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

◆ m_sigmaSTX

double BaseProtonTransport::m_sigmaSTX

protected

Definition at line 69 of file BaseProtonTransport.h.

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

◆ m_sigmaSTY

double BaseProtonTransport::m_sigmaSTY

protected

Definition at line 70 of file BaseProtonTransport.h.

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

◆ m_sigmaSX

double BaseProtonTransport::m_sigmaSX

protected

Definition at line 71 of file BaseProtonTransport.h.

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

◆ m_sigmaSY

double BaseProtonTransport::m_sigmaSY

protected

Definition at line 72 of file BaseProtonTransport.h.

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

◆ m_xAtTrPoint

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

protected

Definition at line 48 of file BaseProtonTransport.h.

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

◆ m_yAtTrPoint

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

protected

Definition at line 49 of file BaseProtonTransport.h.

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

◆ MODE

TransportMode BaseProtonTransport::MODE

protected

Definition at line 42 of file BaseProtonTransport.h.

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

◆ momentumCut_

double BaseProtonTransport::momentumCut_

protected

Definition at line 67 of file BaseProtonTransport.h.

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

◆ NEvent

int BaseProtonTransport::NEvent

protected

Definition at line 44 of file BaseProtonTransport.h.

Referenced by addPartToHepMC().

◆ verbosity_

bool BaseProtonTransport::verbosity_

protected

Definition at line 51 of file BaseProtonTransport.h.

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

Public Member Functions

Protected Types

Protected Attributes

Detailed Description

Member Enumeration Documentation

◆ TransportMode

Constructor & Destructor Documentation

◆ BaseProtonTransport()

◆ ~BaseProtonTransport()

Member Function Documentation

◆ addPartToHepMC()

◆ ApplyBeamCorrection() [1/2]

◆ ApplyBeamCorrection() [2/2]

◆ beamEnergy()

◆ beamMomentum()

◆ clear()

◆ getCorrespondenceMap()

◆ process()

◆ setBeamEnergy()

Member Data Documentation

◆ bApplyZShift

◆ beam1Filename_

◆ beam2Filename_

◆ beamEnergy_

◆ beamMomentum_

◆ engine_

◆ etaCut_

◆ fBeamXatIP

◆ fBeamYatIP

◆ fCrossingAngleX_45

◆ fCrossingAngleX_56

◆ fCrossingAngleY_45

◆ fCrossingAngleY_56

◆ fPPSRegionStart_45

◆ fPPSRegionStart_56

◆ m_beamPart

◆ m_CorrespondenceMap

◆ m_sig_E

◆ m_sigmaSTX

◆ m_sigmaSTY

◆ m_sigmaSX

◆ m_sigmaSY

◆ m_xAtTrPoint

◆ m_yAtTrPoint

◆ MODE

◆ momentumCut_

◆ NEvent

◆ verbosity_