BaseProtonTransport Class Reference

#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
virtual	~BaseProtonTransport ()

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

Protected Member Functions
void	setBeamFileNames (const std::string &nam1, const std::string &nam2)
void	setBeamParameters (double stx, double sty, double sx, double sy, double se)
void	setCrossingAngles (double cx45, double cx56, double cy45, double cy56)

Protected Attributes
bool	bApplyZShift_
std::string	beam1Filename_ {""}
std::string	beam2Filename_ {""}
CLHEP::HepRandomEngine *	engine_ {nullptr}
double	etaCut_
double	fCrossingAngleX_45_ {0.0}
double	fCrossingAngleX_56_ {0.0}
double	fCrossingAngleY_45_ {0.0}
double	fCrossingAngleY_56_ {0.0}
double	fPPSRegionStart_45_
double	fPPSRegionStart_56_
std::map< unsigned int, TLorentzVector >	m_beamPart
std::vector< LHCTransportLink >	m_CorrespondenceMap
std::map< unsigned int, double >	m_xAtTrPoint
std::map< unsigned int, double >	m_yAtTrPoint
TransportMode	MODE
double	momentumCut_
bool	produceHitsRelativeToBeam_
bool	useBeamPositionFromLHCInfo_
bool	verbosity_

Private Attributes
double	beamEnergy_
double	beamMomentum_
double	m_sig_E {0.0}
double	m_sigmaSTX {0.0}
double	m_sigmaSTY {0.0}
double	m_sigmaSX {0.0}
double	m_sigmaSY {0.0}

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 56 of file BaseProtonTransport.h.

{ HECTOR, TOTEM, OPTICALFUNCTIONS };

Constructor & Destructor Documentation

BaseProtonTransport()

BaseProtonTransport::BaseProtonTransport

(

const edm::ParameterSet &

iConfig

)

Definition at line 7 of file BaseProtonTransport.cc.

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

    : verbosity_(iConfig.getParameter<bool>("Verbosity")),
      bApplyZShift_(iConfig.getParameter<bool>("ApplyZShift")),
      useBeamPositionFromLHCInfo_(iConfig.getParameter<bool>("useBeamPositionFromLHCInfo")),
      produceHitsRelativeToBeam_(iConfig.getParameter<bool>("produceHitsRelativeToBeam")),
      fPPSRegionStart_45_(iConfig.getParameter<double>("PPSRegionStart_45")),
      fPPSRegionStart_56_(iConfig.getParameter<double>("PPSRegionStart_56")),
      etaCut_(iConfig.getParameter<double>("EtaCut")),
      momentumCut_(iConfig.getParameter<double>("MomentumCut")),
      beamEnergy_(iConfig.getParameter<double>("BeamEnergy")) {
  beamMomentum_ = std::sqrt(beamEnergy_ * beamEnergy_ - ProtonMassSQ);
}

~BaseProtonTransport()

BaseProtonTransport::~BaseProtonTransport ( )

virtual

Definition at line 20 of file BaseProtonTransport.cc.

References clear().

{ clear(); }

Member Function Documentation

addPartToHepMC()

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

Definition at line 59 of file BaseProtonTransport.cc.

References dqmdumpme::first, fPPSRegionStart_45_, fPPSRegionStart_56_, GenParticle::GenParticle, ALPAKA_ACCELERATOR_NAMESPACE::vertexFinder::it, mps_splice::line, LogDebug, m_beamPart, m_CorrespondenceMap, m_to_mm, m_xAtTrPoint, m_yAtTrPoint, edm::second(), hcalRecHitTable_cff::time, and verbosity_.

                                                                                        {
  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 * CLHEP::meter - gpart->production_vertex()->position().z() * CLHEP::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);
  }
}

ApplyBeamCorrection() [1/2]

void BaseProtonTransport::ApplyBeamCorrection

(

HepMC::GenParticle *

p

)

Definition at line 22 of file BaseProtonTransport.cc.

References AlCaHLTBitMon_ParallelJobs::p.

Referenced by TotemTransport::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()));
}

ApplyBeamCorrection() [2/2]

void BaseProtonTransport::ApplyBeamCorrection

(

TLorentzVector &

p

)

Definition at line 32 of file BaseProtonTransport.cc.

References funct::cos(), MillePedeFileConverter_cfg::e, hcalRecHitTable_cff::energy, engine_, fCrossingAngleX_45_, fCrossingAngleX_56_, m_sig_E, m_sigmaSTX, m_sigmaSTY, MODE, AlCaHLTBitMon_ParallelJobs::p, funct::pow(), ProtonMassSQ, funct::sin(), mathSSE::sqrt(), TOTEM, and urad.

                                                                   {
  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 (MODE == TransportMode::TOTEM)
    thetax += (p_out.Pz() > 0) ? fCrossingAngleX_45_ * urad : fCrossingAngleX_56_ * urad;

  double dtheta_x = (m_sigmaSTX <= 0.0) ? 0.0 : CLHEP::RandGauss::shoot(engine_, 0., m_sigmaSTX);
  double dtheta_y = (m_sigmaSTY <= 0.0) ? 0.0 : CLHEP::RandGauss::shoot(engine_, 0., m_sigmaSTY);
  double denergy = (m_sig_E <= 0.0) ? 0.0 : CLHEP::RandGauss::shoot(engine_, 0., m_sig_E);

  double s_theta = std::sqrt(pow(thetax + dtheta_x * urad, 2) + std::pow(thetay + dtheta_y * urad, 2));
  double s_phi = std::atan2(thetay + dtheta_y * urad, thetax + dtheta_x * urad);
  energy += denergy;
  double p = std::sqrt(std::pow(energy, 2) - ProtonMassSQ);
  double sint = std::sin(s_theta);

  p_out.SetPx(p * sint * std::cos(s_phi));
  p_out.SetPy(p * sint * std::sin(s_phi));
  p_out.SetPz(p * std::cos(s_theta) * direction);
  p_out.SetE(energy);
}

beamEnergy()

double BaseProtonTransport::beamEnergy ( )

inline

Definition at line 32 of file BaseProtonTransport.h.

References beamEnergy_.

Referenced by HectorTransport::transportProton().

{ return beamEnergy_; };

beamMomentum()

double BaseProtonTransport::beamMomentum ( )

inline

Definition at line 33 of file BaseProtonTransport.h.

References beamMomentum_.

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

{ return beamMomentum_; };

clear()

void BaseProtonTransport::clear

(

void

)

Definition at line 112 of file BaseProtonTransport.cc.

References m_beamPart, m_CorrespondenceMap, m_xAtTrPoint, and m_yAtTrPoint.

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

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

getCorrespondenceMap()

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

inline

Definition at line 22 of file BaseProtonTransport.h.

References m_CorrespondenceMap.

{ return m_CorrespondenceMap; }

process()

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

pure virtual

Implemented in HectorTransport, OpticalFunctionsTransport, and TotemTransport.

setBeamFileNames()

void BaseProtonTransport::setBeamFileNames ( const std::string &  nam1, const std::string &  nam2  )

inlineprotected

Definition at line 36 of file BaseProtonTransport.h.

References beam1Filename_, and beam2Filename_.

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

                                                                      {
    beam1Filename_ = nam1;
    beam2Filename_ = nam2;
  };

setBeamParameters()

void BaseProtonTransport::setBeamParameters ( double  stx, double  sty, double  sx, double  sy, double  se  )

inlineprotected

Definition at line 41 of file BaseProtonTransport.h.

References m_sig_E, m_sigmaSTX, m_sigmaSTY, m_sigmaSX, m_sigmaSY, fftjetcommon_cfi::sx, and fftjetcommon_cfi::sy.

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

                                                                                  {
    m_sigmaSTX = stx;
    m_sigmaSTY = sty;
    m_sigmaSX = sx;
    m_sigmaSY = sy;
    m_sig_E = se;
  };

setCrossingAngles()

void BaseProtonTransport::setCrossingAngles ( double  cx45, double  cx56, double  cy45, double  cy56  )

inlineprotected

Definition at line 49 of file BaseProtonTransport.h.

References fCrossingAngleX_45_, fCrossingAngleX_56_, fCrossingAngleY_45_, and fCrossingAngleY_56_.

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

                                                                             {
    fCrossingAngleX_45_ = cx45;
    fCrossingAngleX_56_ = cx56;
    fCrossingAngleY_45_ = cy45;
    fCrossingAngleY_56_ = cy56;
  };

Member Data Documentation

bApplyZShift_

bool BaseProtonTransport::bApplyZShift_

protected

Definition at line 66 of file BaseProtonTransport.h.

beam1Filename_

std::string BaseProtonTransport::beam1Filename_ {""}

protected

Definition at line 70 of file BaseProtonTransport.h.

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

beam2Filename_

std::string BaseProtonTransport::beam2Filename_ {""}

protected

Definition at line 71 of file BaseProtonTransport.h.

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

beamEnergy_

double BaseProtonTransport::beamEnergy_

private

Definition at line 85 of file BaseProtonTransport.h.

Referenced by BaseProtonTransport(), and beamEnergy().

beamMomentum_

double BaseProtonTransport::beamMomentum_

private

Definition at line 84 of file BaseProtonTransport.h.

Referenced by BaseProtonTransport(), and beamMomentum().

engine_

CLHEP::HepRandomEngine* BaseProtonTransport::engine_ {nullptr}

protected

Definition at line 59 of file BaseProtonTransport.h.

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

etaCut_

double BaseProtonTransport::etaCut_

protected

Definition at line 80 of file BaseProtonTransport.h.

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

fCrossingAngleX_45_

double BaseProtonTransport::fCrossingAngleX_45_ {0.0}

protected

Definition at line 75 of file BaseProtonTransport.h.

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

fCrossingAngleX_56_

double BaseProtonTransport::fCrossingAngleX_56_ {0.0}

protected

Definition at line 76 of file BaseProtonTransport.h.

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

fCrossingAngleY_45_

double BaseProtonTransport::fCrossingAngleY_45_ {0.0}

protected

Definition at line 77 of file BaseProtonTransport.h.

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

fCrossingAngleY_56_

double BaseProtonTransport::fCrossingAngleY_56_ {0.0}

protected

Definition at line 78 of file BaseProtonTransport.h.

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

fPPSRegionStart_45_

double BaseProtonTransport::fPPSRegionStart_45_

protected

Definition at line 73 of file BaseProtonTransport.h.

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

fPPSRegionStart_56_

double BaseProtonTransport::fPPSRegionStart_56_

protected

Definition at line 74 of file BaseProtonTransport.h.

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

m_beamPart

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

protected

Definition at line 61 of file BaseProtonTransport.h.

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

m_CorrespondenceMap

std::vector<LHCTransportLink> BaseProtonTransport::m_CorrespondenceMap

protected

Definition at line 60 of file BaseProtonTransport.h.

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

m_sig_E

double BaseProtonTransport::m_sig_E {0.0}

private

Definition at line 91 of file BaseProtonTransport.h.

Referenced by ApplyBeamCorrection(), and setBeamParameters().

m_sigmaSTX

double BaseProtonTransport::m_sigmaSTX {0.0}

private

Definition at line 87 of file BaseProtonTransport.h.

Referenced by ApplyBeamCorrection(), and setBeamParameters().

m_sigmaSTY

double BaseProtonTransport::m_sigmaSTY {0.0}

private

Definition at line 88 of file BaseProtonTransport.h.

Referenced by ApplyBeamCorrection(), and setBeamParameters().

m_sigmaSX

double BaseProtonTransport::m_sigmaSX {0.0}

private

Definition at line 89 of file BaseProtonTransport.h.

Referenced by setBeamParameters().

m_sigmaSY

double BaseProtonTransport::m_sigmaSY {0.0}

private

Definition at line 90 of file BaseProtonTransport.h.

Referenced by setBeamParameters().

m_xAtTrPoint

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

protected

Definition at line 62 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 63 of file BaseProtonTransport.h.

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

MODE

TransportMode BaseProtonTransport::MODE

protected

Definition at line 57 of file BaseProtonTransport.h.

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

momentumCut_

double BaseProtonTransport::momentumCut_

protected

Definition at line 81 of file BaseProtonTransport.h.

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

produceHitsRelativeToBeam_

bool BaseProtonTransport::produceHitsRelativeToBeam_

protected

Definition at line 68 of file BaseProtonTransport.h.

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

useBeamPositionFromLHCInfo_

bool BaseProtonTransport::useBeamPositionFromLHCInfo_

protected

Definition at line 67 of file BaseProtonTransport.h.

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

verbosity_

bool BaseProtonTransport::verbosity_

protected

Definition at line 65 of file BaseProtonTransport.h.

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