HectorTransport Class Reference

#include <HectorTransport.h>

Inheritance diagram for HectorTransport:

Public Member Functions
	HectorTransport (const edm::ParameterSet &ps, edm::ConsumesCollector iC)
void	process (const HepMC::GenEvent *ev, const edm::EventSetup &es, CLHEP::HepRandomEngine *engine) override
	~HectorTransport () override
Public Member Functions inherited from BaseProtonTransport
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 ()
void	setBeamEnergy (double e)
virtual	~BaseProtonTransport ()

Private Member Functions
bool	setBeamLine ()
bool	transportProton (const HepMC::GenParticle *)
	propagate the particles through a beamline to PPS More...

Private Attributes
const CTPPSBeamParameters *	beamParameters_
edm::ESGetToken < CTPPSBeamParameters, CTPPSBeamParametersRcd >	beamParametersToken_
const BeamSpotObjects *	beamspot_
edm::ESGetToken < BeamSpotObjects, BeamSpotObjectsRcd >	beamspotToken_
std::unique_ptr< H_BeamLine >	m_beamline45
std::unique_ptr< H_BeamLine >	m_beamline56
double	m_fEtacut
double	m_fMomentumMin
bool	m_verbosity

Static Private Attributes
static constexpr double	fPPSBeamLineLength_ = 250.

Additional Inherited Members
Protected Types inherited from BaseProtonTransport
enum	TransportMode { TransportMode::HECTOR, TransportMode::TOTEM, TransportMode::OPTICALFUNCTIONS }
Protected Attributes inherited from BaseProtonTransport
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 39 of file HectorTransport.h.

Constructor & Destructor Documentation

HectorTransport::HectorTransport	(	const edm::ParameterSet &	ps,
		edm::ConsumesCollector	iC
	)

Definition at line 14 of file HectorTransport.cc.

References BaseProtonTransport::beam1Filename_, BaseProtonTransport::beam2Filename_, BaseProtonTransport::fCrossingAngleX_45, BaseProtonTransport::fCrossingAngleX_56, BaseProtonTransport::fCrossingAngleY_45, BaseProtonTransport::fCrossingAngleY_56, edm::ParameterSet::getParameter(), BaseProtonTransport::HECTOR, BaseProtonTransport::m_sig_E, BaseProtonTransport::m_sigmaSTX, BaseProtonTransport::m_sigmaSTY, BaseProtonTransport::m_sigmaSX, BaseProtonTransport::m_sigmaSY, BaseProtonTransport::MODE, setBeamLine(), and AlCaHLTBitMon_QueryRunRegistry::string.

    : BaseProtonTransport(iConfig),
      m_beamline45(nullptr),
      m_beamline56(nullptr),
      beamParametersToken_(iC.esConsumes()),
      beamspotToken_(iC.esConsumes()) {
  // Create LHC beam line

  MODE = TransportMode::HECTOR;  // defines the MODE for the transport
  beam1Filename_ = iConfig.getParameter<std::string>("Beam1Filename");
  beam2Filename_ = iConfig.getParameter<std::string>("Beam2Filename");
  fCrossingAngleX_45 = iConfig.getParameter<double>("halfCrossingAngleXSector45");
  fCrossingAngleX_56 = iConfig.getParameter<double>("halfCrossingAngleXSector56");
  fCrossingAngleY_45 = iConfig.getParameter<double>("halfCrossingAngleYSector45");
  fCrossingAngleY_56 = iConfig.getParameter<double>("halfCrossingAngleYSector56");
  m_sigmaSTX = iConfig.getParameter<double>("BeamDivergenceX");
  m_sigmaSTY = iConfig.getParameter<double>("BeamDivergenceY");
  m_sigmaSX = iConfig.getParameter<double>("BeamSigmaX");
  m_sigmaSY = iConfig.getParameter<double>("BeamSigmaY");
  m_sig_E = iConfig.getParameter<double>("BeamEnergyDispersion");

  //PPS
  edm::LogInfo("ProtonTransport") << "=============================================================================\n"
                                  << "             Bulding LHC Proton transporter based on HECTOR model\n"
                                  << "=============================================================================\n";
  setBeamLine();
}

HectorTransport::~HectorTransport ( )

override

Definition at line 12 of file HectorTransport.cc.

{};

Member Function Documentation

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

overridevirtual

Implements BaseProtonTransport.

Definition at line 45 of file HectorTransport.cc.

References beamParameters_, beamParametersToken_, beamspot_, beamspotToken_, BaseProtonTransport::clear(), BaseProtonTransport::engine_, BaseProtonTransport::etaCut_, GenParticle::GenParticle, edm::EventSetup::getData(), geometryCSVtoXML::line, BaseProtonTransport::m_beamPart, BaseProtonTransport::momentumCut_, and transportProton().

                                                             {
  this->clear();

  engine_ = _engine;  // the engine needs to be updated for each event

  beamspot_ = &iSetup.getData(beamspotToken_);
  beamParameters_ = &iSetup.getData(beamParametersToken_);

  for (HepMC::GenEvent::particle_const_iterator eventParticle = evt->particles_begin();
       eventParticle != evt->particles_end();
       ++eventParticle) {
    if (!((*eventParticle)->status() == 1 && (*eventParticle)->pdg_id() == 2212))
      continue;

    if (!(fabs((*eventParticle)->momentum().eta()) > etaCut_ && fabs((*eventParticle)->momentum().pz()) > momentumCut_))
      continue;  // discard protons outside kinematic acceptance

    unsigned int line = (*eventParticle)->barcode();
    HepMC::GenParticle* gpart = (*eventParticle);

    if (gpart->pdg_id() != 2212)
      continue;  // only transport stable protons
    if (gpart->status() != 1)
      continue;
    if (m_beamPart.find(line) != m_beamPart.end())
      continue;  // assures this protons has not been already propagated

    transportProton(gpart);
  }
}

bool HectorTransport::setBeamLine ( )

private

Definition at line 193 of file HectorTransport.cc.

References b1, b2, BaseProtonTransport::beam1Filename_, BaseProtonTransport::beam2Filename_, fPPSBeamLineLength_, BaseProtonTransport::fPPSRegionStart_45, BaseProtonTransport::fPPSRegionStart_56, LogDebug, m_beamline45, m_beamline56, and BaseProtonTransport::verbosity_.

Referenced by HectorTransport().

                                  {
  edm::FileInPath b1(beam1Filename_.c_str());
  edm::FileInPath b2(beam2Filename_.c_str());

  // construct beam line for PPS (forward 1 backward 2):
  if (fPPSBeamLineLength_ > 0.) {
    m_beamline45 = std::make_unique<H_BeamLine>(-1, fPPSBeamLineLength_ + 0.1);
    m_beamline45->fill(b2.fullPath(), 1, "IP5");
    m_beamline56 = std::make_unique<H_BeamLine>(1, fPPSBeamLineLength_ + 0.1);
    m_beamline56->fill(b1.fullPath(), 1, "IP5");
  } else {
    if (verbosity_)
      LogDebug("HectorTransportSetup") << "HectorTransport: WARNING: lengthctpps=  " << fPPSBeamLineLength_;
    return false;
  }
  if (verbosity_) {
    edm::LogInfo("HectorTransportSetup") << "===================================================================\n"
                                         << " * * * * * * * * * * * * * * * * * * * * * * * * * * * *           \n"
                                         << " *                                                         *       \n"
                                         << " *                   --<--<--  A fast simulator --<--<--     *     \n"
                                         << " *                 | --<--<--     of particle   --<--<--     *     \n"
                                         << " *  ----HECTOR----<                                          *     \n"
                                         << " *                 | -->-->-- transport through-->-->--      *     \n"
                                         << " *                   -->-->-- generic beamlines -->-->--     *     \n"
                                         << " *                                                           *     \n"
                                         << " * JINST 2:P09005 (2007)                                     *     \n"
                                         << " *      X Rouby, J de Favereau, K Piotrzkowski (CP3)         *     \n"
                                         << " *       http://www.fynu.ucl.ac.be/hector.html               *     \n"
                                         << " *                                                           *     \n"
                                         << " * Center for Cosmology, Particle Physics and Phenomenology  *     \n"
                                         << " *              Universite catholique de Louvain             *     \n"
                                         << " *                 Louvain-la-Neuve, Belgium                 *     \n"
                                         << " *                                                         *       \n"
                                         << " * * * * * * * * * * * * * * * * * * * * * * * * * * * *           \n"
                                         << " HectorTransport configuration: \n"
                                         << " Beam line length      = " << fPPSBeamLineLength_ << "\n"
                                         << " PPS Region Start 44   =  " << fPPSRegionStart_45 << "\n"
                                         << " PPS Region Start 56   =  " << fPPSRegionStart_56 << "\n"
                                         << "===================================================================\n";
  }
  if (verbosity_) {
    edm::LogInfo("HectorTransportSetup") << "====================================================================\n"
                                         << "                  Forward beam line elements \n";
    m_beamline45->showElements();
    edm::LogInfo("HectorTransportSetup") << "====================================================================\n"
                                         << "                 Backward beam line elements \n";
    m_beamline56->showElements();
  }
  return true;
}

bool HectorTransport::transportProton ( const HepMC::GenParticle *  gpart )

private

propagate the particles through a beamline to PPS

Definition at line 77 of file HectorTransport.cc.

References BaseProtonTransport::beamEnergy_, BaseProtonTransport::beamMomentum_, beamParameters_, beamspot_, RecoTauCleanerPlugins::charge, cm_to_mm, funct::cos(), alignCSCRings::e, BaseProtonTransport::fCrossingAngleX_45, BaseProtonTransport::fCrossingAngleX_56, BaseProtonTransport::fCrossingAngleY_45, BaseProtonTransport::fCrossingAngleY_56, BaseProtonTransport::fPPSRegionStart_45, BaseProtonTransport::fPPSRegionStart_56, CTPPSBeamParameters::getVtxOffsetX45(), CTPPSBeamParameters::getVtxOffsetY45(), CTPPSBeamParameters::getVtxOffsetZ45(), BeamSpotObjects::GetX(), BeamSpotObjects::GetY(), BeamSpotObjects::GetZ(), geometryCSVtoXML::line, LogDebug, m_beamline45, m_beamline56, BaseProtonTransport::m_beamPart, BaseProtonTransport::m_xAtTrPoint, BaseProtonTransport::m_yAtTrPoint, ResonanceBuilder::mass, mm_to_um, funct::pow(), BaseProtonTransport::produceHitsRelativeToBeam_, ProtonMassSQ, mathSSE::sqrt(), funct::tan(), theta(), um_to_mm, urad, BaseProtonTransport::useBeamPositionFromLHCInfo_, and BaseProtonTransport::verbosity_.

Referenced by process().

                                                                   {
  edm::LogInfo("ProtonTransport") << "Starting proton transport using HECTOR method\n";

  double px, py, pz, e;
  unsigned int line = (gpart)->barcode();

  double mass = gpart->generatedMass();
  double charge = 1.;

  // remove the CMS vertex shift
  double vtxXoffset_;
  double vtxYoffset_;
  double vtxZoffset_;
  if (useBeamPositionFromLHCInfo_) {
    vtxXoffset_ = beamParameters_->getVtxOffsetX45() * cm_to_mm;
    vtxYoffset_ = beamParameters_->getVtxOffsetY45() * cm_to_mm;
    vtxZoffset_ = beamParameters_->getVtxOffsetZ45() * cm_to_mm;
  } else {
    vtxXoffset_ = beamspot_->GetX() * cm_to_mm;
    vtxYoffset_ = beamspot_->GetY() * cm_to_mm;
    vtxZoffset_ = beamspot_->GetZ() * cm_to_mm;
  }
  vtxZoffset_ *= 1;  // to avoid compilation error, should be used in the future
  // Momentum in LHC ref. frame
  px = -gpart->momentum().px();
  py = gpart->momentum().py();
  pz = -gpart->momentum().pz();
  e = gpart->momentum().e();

  int direction = (pz > 0) ? 1 : -1;  // in relation to LHC ref frame

  double XforPosition = -gpart->production_vertex()->position().x();  //mm in the ref. frame of LHC
  double YforPosition = gpart->production_vertex()->position().y();   //mm
  double ZforPosition = -gpart->production_vertex()->position().z();  //mm
  double fCrossingAngleX = (pz < 0) ? fCrossingAngleX_45 : fCrossingAngleX_56;
  double fCrossingAngleY = (pz < 0) ? fCrossingAngleY_45 : fCrossingAngleY_56;
  //
  H_BeamParticle h_p(mass, charge);
  h_p.set4Momentum(px, py, pz, e);
  //
  // shift the starting position of the track to Z=0 if configured so (all the variables below are still in cm)
  XforPosition = XforPosition - ZforPosition * (px / pz + fCrossingAngleX * urad);  // theta ~=tan(theta)
  YforPosition = YforPosition - ZforPosition * (py / pz + fCrossingAngleY * urad);
  XforPosition -= (-vtxXoffset_);  // X was already in the LHC ref. frame
  YforPosition -= vtxYoffset_;
  ZforPosition = 0.;

  // set position, but do not invert the coordinate for the angles (TX,TY) because it is done by set4Momentum
  h_p.setPosition(XforPosition * mm_to_um, YforPosition * mm_to_um, h_p.getTX(), h_p.getTY(), 0.);

  bool is_stop;
  float x1_ctpps;
  float y1_ctpps;

  H_BeamLine* _beamline = nullptr;
  double _targetZ = 0;
  switch (direction) {
    case 1:
      _beamline = &*m_beamline56;  // negative side propagation
      _targetZ = fPPSRegionStart_56;
      break;
    case -1:
      _beamline = &*m_beamline45;
      _targetZ = fPPSRegionStart_45;
      break;
  }
  // insert protection for NULL beamlines here
  h_p.computePath(&*_beamline);
  is_stop = h_p.stopped(&*_beamline);
  if (verbosity_)
    LogDebug("HectorTransportEventProcessing")
        << "HectorTransport:filterPPS: barcode = " << line << " is_stop=  " << is_stop;
  if (is_stop) {
    return false;
  }
  //
  //propagating
  //
  h_p.propagate(_targetZ);
  x1_ctpps = h_p.getX();
  y1_ctpps = h_p.getY();

  double thx = h_p.getTX();
  double thy = h_p.getTY();

  if (produceHitsRelativeToBeam_) {
    H_BeamParticle p_beam(mass, charge);
    p_beam.set4Momentum(0., 0., beamMomentum_, beamEnergy_);
    p_beam.setPosition(0., 0., fCrossingAngleX * urad, fCrossingAngleY * urad, 0.);
    p_beam.computePath(&*_beamline);
    thx -= p_beam.getTX();
    thy -= p_beam.getTY();
    x1_ctpps -= p_beam.getX();
    y1_ctpps -= p_beam.getY();
    edm::LogInfo("HectorTransportEventProcessing")
        << "Shifting the hit relative to beam :  X = " << p_beam.getX() << "(mm)      Y = " << p_beam.getY() << "(mm)";
  }

  double partP = sqrt(pow(h_p.getE(), 2) - ProtonMassSQ);
  double theta = sqrt(thx * thx + thy * thy) * urad;

  // copy the kinematic changing to CMS ref. frame, only the negative Pz needs to be changed
  TLorentzVector p_out(-tan(thx * urad) * partP * cos(theta),
                       tan(thy * urad) * partP * cos(theta),
                       -direction * partP * cos(theta),
                       h_p.getE());

  m_beamPart[line] = p_out;
  m_xAtTrPoint[line] = -x1_ctpps * um_to_mm;  // move to CMS ref. frame
  m_yAtTrPoint[line] = y1_ctpps * um_to_mm;
  if (verbosity_)
    LogDebug("HectorTransportEventProcessing")
        << "HectorTransport:filterPPS: barcode = " << line << " x=  " << x1_ctpps << " y= " << y1_ctpps;

  return true;
}

Member Data Documentation

const CTPPSBeamParameters* HectorTransport::beamParameters_

private

Definition at line 69 of file HectorTransport.h.

Referenced by process(), and transportProton().

edm::ESGetToken<CTPPSBeamParameters, CTPPSBeamParametersRcd> HectorTransport::beamParametersToken_

private

Definition at line 66 of file HectorTransport.h.

Referenced by process().

const BeamSpotObjects* HectorTransport::beamspot_

private

Definition at line 70 of file HectorTransport.h.

Referenced by process(), and transportProton().

edm::ESGetToken<BeamSpotObjects, BeamSpotObjectsRcd> HectorTransport::beamspotToken_

private

Definition at line 67 of file HectorTransport.h.

Referenced by process().

constexpr double HectorTransport::fPPSBeamLineLength_ = 250.

staticprivate

Definition at line 49 of file HectorTransport.h.

Referenced by setBeamLine().

std::unique_ptr<H_BeamLine> HectorTransport::m_beamline45

private

Definition at line 63 of file HectorTransport.h.

Referenced by setBeamLine(), and transportProton().

std::unique_ptr<H_BeamLine> HectorTransport::m_beamline56

private

Definition at line 64 of file HectorTransport.h.

Referenced by setBeamLine(), and transportProton().

double HectorTransport::m_fEtacut

private

Definition at line 59 of file HectorTransport.h.

double HectorTransport::m_fMomentumMin

private

Definition at line 60 of file HectorTransport.h.

bool HectorTransport::m_verbosity

private

Definition at line 47 of file HectorTransport.h.