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
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	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	produceHitsRelativeToBeam_
bool	useBeamPositionFromLHCInfo_
bool	verbosity_

Detailed Description

Definition at line 17 of file BaseProtonTransport.h.

Member Enumeration Documentation

enum BaseProtonTransport::TransportMode

strongprotected

Enumerator
HECTOR
TOTEM
OPTICALFUNCTIONS

Definition at line 41 of file BaseProtonTransport.h.

{ HECTOR, TOTEM, OPTICALFUNCTIONS };

Constructor & Destructor Documentation

BaseProtonTransport::BaseProtonTransport

(

const edm::ParameterSet &

iConfig

)

Definition at line 8 of file BaseProtonTransport.cc.

References beamEnergy_, and setBeamEnergy().

    : 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")),
      beamEnergy_(iConfig.getParameter<double>("BeamEnergy")),
      etaCut_(iConfig.getParameter<double>("EtaCut")),
      momentumCut_(iConfig.getParameter<double>("MomentumCut")) {
  setBeamEnergy(beamEnergy_);
}

virtual BaseProtonTransport::~BaseProtonTransport ( )

inlinevirtual

Definition at line 20 of file BaseProtonTransport.h.

References clear().

{ this->clear(); };

Member Function Documentation

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

Definition at line 58 of file BaseProtonTransport.cc.

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

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

void BaseProtonTransport::ApplyBeamCorrection

(

HepMC::GenParticle *

p

)

Definition at line 20 of file BaseProtonTransport.cc.

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

void BaseProtonTransport::ApplyBeamCorrection

(

TLorentzVector &

p

)

Definition at line 29 of file BaseProtonTransport.cc.

References funct::cos(), alignCSCRings::e, relval_parameters_module::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.

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

double BaseProtonTransport::beamEnergy ( )

inline

Definition at line 37 of file BaseProtonTransport.h.

References beamEnergy_.

{ return beamEnergy_; };

double BaseProtonTransport::beamMomentum ( )

inline

Definition at line 38 of file BaseProtonTransport.h.

References beamMomentum_.

Referenced by OpticalFunctionsTransport::transportProton().

{ return beamMomentum_; };

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

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

inline

Definition at line 22 of file BaseProtonTransport.h.

References m_CorrespondenceMap.

{ return m_CorrespondenceMap; }

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

pure virtual

Implemented in HectorTransport, OpticalFunctionsTransport, and TotemTransport.

void BaseProtonTransport::setBeamEnergy ( double  e )

inline

Definition at line 32 of file BaseProtonTransport.h.

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

Referenced by BaseProtonTransport().

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

Member Data Documentation

bool BaseProtonTransport::bApplyZShift

protected

Definition at line 52 of file BaseProtonTransport.h.

std::string BaseProtonTransport::beam1Filename_

protected

Definition at line 56 of file BaseProtonTransport.h.

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

std::string BaseProtonTransport::beam2Filename_

protected

Definition at line 57 of file BaseProtonTransport.h.

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

double BaseProtonTransport::beamEnergy_

protected

Definition at line 67 of file BaseProtonTransport.h.

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

double BaseProtonTransport::beamMomentum_

protected

Definition at line 66 of file BaseProtonTransport.h.

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

CLHEP::HepRandomEngine* BaseProtonTransport::engine_

protected

Definition at line 45 of file BaseProtonTransport.h.

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

double BaseProtonTransport::etaCut_

protected

Definition at line 68 of file BaseProtonTransport.h.

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

double BaseProtonTransport::fCrossingAngleX_45

protected

Definition at line 61 of file BaseProtonTransport.h.

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

double BaseProtonTransport::fCrossingAngleX_56

protected

Definition at line 62 of file BaseProtonTransport.h.

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

double BaseProtonTransport::fCrossingAngleY_45

protected

Definition at line 63 of file BaseProtonTransport.h.

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

double BaseProtonTransport::fCrossingAngleY_56

protected

Definition at line 64 of file BaseProtonTransport.h.

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

double BaseProtonTransport::fPPSRegionStart_45

protected

Definition at line 59 of file BaseProtonTransport.h.

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

double BaseProtonTransport::fPPSRegionStart_56

protected

Definition at line 60 of file BaseProtonTransport.h.

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

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(), TotemTransport::transportProton(), OpticalFunctionsTransport::transportProton(), and HectorTransport::transportProton().

std::vector<LHCTransportLink> BaseProtonTransport::m_CorrespondenceMap

protected

Definition at line 46 of file BaseProtonTransport.h.

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

double BaseProtonTransport::m_sig_E

protected

Definition at line 75 of file BaseProtonTransport.h.

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

double BaseProtonTransport::m_sigmaSTX

protected

Definition at line 71 of file BaseProtonTransport.h.

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

double BaseProtonTransport::m_sigmaSTY

protected

Definition at line 72 of file BaseProtonTransport.h.

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

double BaseProtonTransport::m_sigmaSX

protected

Definition at line 73 of file BaseProtonTransport.h.

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

double BaseProtonTransport::m_sigmaSY

protected

Definition at line 74 of file BaseProtonTransport.h.

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

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().

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().

TransportMode BaseProtonTransport::MODE

protected

Definition at line 42 of file BaseProtonTransport.h.

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

double BaseProtonTransport::momentumCut_

protected

Definition at line 69 of file BaseProtonTransport.h.

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

int BaseProtonTransport::NEvent

protected

Definition at line 44 of file BaseProtonTransport.h.

Referenced by addPartToHepMC().

bool BaseProtonTransport::produceHitsRelativeToBeam_

protected

Definition at line 54 of file BaseProtonTransport.h.

Referenced by HectorTransport::transportProton().

bool BaseProtonTransport::useBeamPositionFromLHCInfo_

protected

Definition at line 53 of file BaseProtonTransport.h.

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

bool BaseProtonTransport::verbosity_

protected

Definition at line 51 of file BaseProtonTransport.h.

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