#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 15 of file ProtonTransport.h.

Member Enumeration Documentation

enum ProtonTransport::TransportMode

strongprotected

Enumerator
HECTOR
TOTEM

Definition at line 27 of file ProtonTransport.h.

27 { HECTOR, TOTEM };

Constructor & Destructor Documentation

ProtonTransport::ProtonTransport ( )

Definition at line 6 of file ProtonTransport.cc.

6 {};

ProtonTransport::~ProtonTransport ( )

virtual

Definition at line 7 of file ProtonTransport.cc.

7 {};

Member Function Documentation

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

Definition at line 14 of file ProtonTransport.cc.

References dqmdumpme::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 71 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 80 of file ProtonTransport.cc.

References funct::cos(), MillePedeFileConverter_cfg::e, HCALHighEnergyHPDFilter_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 8 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 19 of file ProtonTransport.h.

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

Referenced by PPSSimTrackProducer::produce().

19 { return m_CorrespondenceMap; }

ProtonTransport::m_CorrespondenceMap

std::vector< LHCTransportLink > m_CorrespondenceMap

Definition: ProtonTransport.h:39

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 33 of file ProtonTransport.h.

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

CLHEP::HepRandomEngine* ProtonTransport::engine

protected

Definition at line 31 of file ProtonTransport.h.

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

double ProtonTransport::fBeamEnergy

protected

Definition at line 55 of file ProtonTransport.h.

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

double ProtonTransport::fBeamMomentum

protected

Definition at line 54 of file ProtonTransport.h.

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

double ProtonTransport::fBeamXatIP

protected

Definition at line 52 of file ProtonTransport.h.

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

double ProtonTransport::fBeamYatIP

protected

Definition at line 53 of file ProtonTransport.h.

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

double ProtonTransport::fCrossingAngle_45

protected

Definition at line 36 of file ProtonTransport.h.

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

double ProtonTransport::fCrossingAngle_56

protected

Definition at line 37 of file ProtonTransport.h.

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

double ProtonTransport::fPPSRegionStart_45

protected

Definition at line 35 of file ProtonTransport.h.

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

double ProtonTransport::fPPSRegionStart_56

protected

Definition at line 34 of file ProtonTransport.h.

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

double ProtonTransport::fVtxMeanX

protected

Definition at line 49 of file ProtonTransport.h.

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

double ProtonTransport::fVtxMeanY

protected

Definition at line 50 of file ProtonTransport.h.

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

double ProtonTransport::fVtxMeanZ

protected

Definition at line 51 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 40 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 39 of file ProtonTransport.h.

Referenced by addPartToHepMC().

double ProtonTransport::m_sig_E

protected

Definition at line 48 of file ProtonTransport.h.

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

double ProtonTransport::m_sigmaSTX

protected

Definition at line 46 of file ProtonTransport.h.

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

double ProtonTransport::m_sigmaSTY

protected

Definition at line 47 of file ProtonTransport.h.

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

double ProtonTransport::m_sigmaSX

protected

Definition at line 44 of file ProtonTransport.h.

Referenced by HectorTransport::HectorTransport().

double ProtonTransport::m_sigmaSY

protected

Definition at line 45 of file ProtonTransport.h.

Referenced by HectorTransport::HectorTransport().

bool ProtonTransport::m_verbosity

protected

Definition at line 30 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 41 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 42 of file ProtonTransport.h.

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

TransportMode ProtonTransport::MODE

protected

Definition at line 28 of file ProtonTransport.h.

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

int ProtonTransport::NEvent

protected

Definition at line 29 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