TotemTransport Class Reference

#include <TotemTransport.h>

Inheritance diagram for TotemTransport:

Public Member Functions
void	process (const HepMC::GenEvent *ev, const edm::EventSetup &es, CLHEP::HepRandomEngine *engine) override
	TotemTransport (const edm::ParameterSet &ps)
	~TotemTransport () override
Public Member Functions inherited from BaseProtonTransport
void	addPartToHepMC (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
LHCOpticsApproximator *	ReadParameterization (const std::string &, const std::string &)
bool	transportProton (const HepMC::GenParticle *)

Private Attributes
LHCOpticsApproximator *	m_aprox_ip_150_l = nullptr
LHCOpticsApproximator *	m_aprox_ip_150_r = nullptr
double	m_beampipe_aperture_radius
double	m_fEtacut
double	m_fMomentumMin
std::string	m_model_ip_150_l_name
std::string	m_model_ip_150_r_name

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	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
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 23 of file TotemTransport.h.

Constructor & Destructor Documentation

TotemTransport()

TotemTransport::TotemTransport

(

const edm::ParameterSet &

ps

)

Definition at line 10 of file TotemTransport.cc.

    : BaseProtonTransport(iConfig),
      m_model_ip_150_r_name(iConfig.getParameter<std::string>("Model_IP_150_R_Name")),
      m_model_ip_150_l_name(iConfig.getParameter<std::string>("Model_IP_150_L_Name")),
      m_beampipe_aperture_radius(iConfig.getParameter<double>("BeampipeApertureRadius")) {
  MODE = TransportMode::TOTEM;
  beam1Filename_ = iConfig.getParameter<std::string>("Beam1Filename");
  beam2Filename_ = iConfig.getParameter<std::string>("Beam2Filename");
  fCrossingAngle_45 = iConfig.getParameter<double>("halfCrossingAngleSector45");
  fCrossingAngle_56 = iConfig.getParameter<double>("halfCrossingAngleSector56");
  beamEnergy_ = iConfig.getParameter<double>("BeamEnergy");
  fVtxMeanX = iConfig.getParameter<double>("VtxMeanX");
  fVtxMeanY = iConfig.getParameter<double>("VtxMeanY");
  fVtxMeanZ = iConfig.getParameter<double>("VtxMeanZ");
  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");
  fBeamXatIP = iConfig.getUntrackedParameter<double>("BeamXatIP", fVtxMeanX);
  fBeamYatIP = iConfig.getUntrackedParameter<double>("BeamYatIP", fVtxMeanY);
 
  if (fPPSRegionStart_56 > 0)
    fPPSRegionStart_56 *= -1;  // make sure sector 56 has negative position, as TOTEM convention
 
  edm::LogInfo("TotemTransport") << "=============================================================================\n"
                                 << "             Bulding LHC Proton transporter based on TOTEM model\n"
                                 << "=============================================================================\n";
 
  m_aprox_ip_150_r = ReadParameterization(m_model_ip_150_r_name, beam1Filename_);
  m_aprox_ip_150_l = ReadParameterization(m_model_ip_150_l_name, beam2Filename_);
 
  if (m_aprox_ip_150_r == nullptr || m_aprox_ip_150_l == nullptr) {
    edm::LogError("TotemTransport") << "Parameterisation " << m_model_ip_150_r_name << " or " << m_model_ip_150_l_name
                                    << " missing in file. Cannot proceed. ";
    exit(1);
  }
  edm::LogInfo("TotemTransport") << "Parameterizations read from file, pointers:" << m_aprox_ip_150_r << " "
                                 << m_aprox_ip_150_l << " ";
}

References BaseProtonTransport::beam1Filename_, BaseProtonTransport::beam2Filename_, BaseProtonTransport::beamEnergy_, beamvalidation::exit(), BaseProtonTransport::fBeamXatIP, BaseProtonTransport::fBeamYatIP, BaseProtonTransport::fCrossingAngle_45, BaseProtonTransport::fCrossingAngle_56, BaseProtonTransport::fPPSRegionStart_56, BaseProtonTransport::fVtxMeanX, BaseProtonTransport::fVtxMeanY, BaseProtonTransport::fVtxMeanZ, edm::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), m_aprox_ip_150_l, m_aprox_ip_150_r, m_model_ip_150_l_name, m_model_ip_150_r_name, BaseProtonTransport::m_sig_E, BaseProtonTransport::m_sigmaSTX, BaseProtonTransport::m_sigmaSTY, BaseProtonTransport::m_sigmaSX, BaseProtonTransport::m_sigmaSY, BaseProtonTransport::MODE, ReadParameterization(), AlCaHLTBitMon_QueryRunRegistry::string, and BaseProtonTransport::TOTEM.

~TotemTransport()

TotemTransport::~TotemTransport ( )

inlineoverride

Definition at line 26 of file TotemTransport.h.

{};

Member Function Documentation

process()

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

overridevirtual

Implements BaseProtonTransport.

Definition at line 54 of file TotemTransport.cc.

                                                            {
  this->clear();
 
  engine_ = _engine;  // the engine needs to be updated for each event
 
  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())  // assures this protons has not been already propagated
      continue;
 
    transportProton(gpart);
  }
}

References BaseProtonTransport::clear(), BaseProtonTransport::engine_, BaseProtonTransport::etaCut_, GenParticle::GenParticle, mps_splice::line, BaseProtonTransport::m_beamPart, BaseProtonTransport::momentumCut_, and transportProton().

ReadParameterization()

LHCOpticsApproximator * TotemTransport::ReadParameterization ( const std::string &  m_model_name, const std::string &  rootfile  )

private

Definition at line 184 of file TotemTransport.cc.

                                                                                       {
  edm::FileInPath fileName(rootfile.c_str());
  TFile* f = TFile::Open(fileName.fullPath().c_str(), "read");
  if (!f) {
    edm::LogError("TotemTransport") << "File " << fileName << " not found. Exiting.";
    return nullptr;
  }
  edm::LogInfo("TotemTransport") << "Root file opened, pointer:" << f;
 
  // read parametrization
  LHCOpticsApproximator* aprox = (LHCOpticsApproximator*)f->Get(m_model_name.c_str());
  f->Close();
  return aprox;
}

References f, MillePedeFileConverter_cfg::fileName, and doHarvest::rootfile.

Referenced by TotemTransport().

transportProton()

bool TotemTransport::transportProton ( const HepMC::GenParticle *  in_trk )

private

Definition at line 87 of file TotemTransport.cc.

                                                                   {
  //
  //
  edm::LogInfo("TotemTransport") << "Starting proton transport using TOTEM method\n";
  //
  ApplyBeamCorrection(const_cast<HepMC::GenParticle*>(in_trk));
 
  const HepMC::GenVertex* in_pos = in_trk->production_vertex();
  const HepMC::FourVector in_mom = in_trk->momentum();
  //
  // ATTENTION: HepMC uses mm, vertex config of CMS uses cm and SimTransport uses mm
  //
  double in_position[3] = {(in_pos->position().x() - fVtxMeanX * cm) / meter + fBeamXatIP * mm / meter,
                           (in_pos->position().y() - fVtxMeanY * cm) / meter + fBeamYatIP * mm / meter,
                           (in_pos->position().z() - fVtxMeanZ * cm) / meter};  // move to z=0 if configured below
 
  // (bApplyZShift) -- The TOTEM parameterization requires the shift to z=0
  double fCrossingAngle = (in_mom.z() > 0) ? fCrossingAngle_45 : -fCrossingAngle_56;
  in_position[0] = in_position[0] +
                   (tan((long double)fCrossingAngle * urad) - ((long double)in_mom.x()) / ((long double)in_mom.z())) *
                       in_position[2];
  in_position[1] = in_position[1] - ((long double)in_mom.y()) / ((long double)in_mom.z()) * in_position[2];
  in_position[2] = 0.;
  //
  double in_momentum[3] = {in_mom.x(), in_mom.y(), in_mom.z()};
  double out_position[3];
  double out_momentum[3];
  edm::LogInfo("TotemTransport") << "before transport ->"
                                 << " position: " << in_position[0] << ", " << in_position[1] << ", " << in_position[2]
                                 << " momentum: " << in_momentum[0] << ", " << in_momentum[1] << ", " << in_momentum[2];
 
  LHCOpticsApproximator* approximator_ = nullptr;
  double m_Zin_;
  double m_Zout_;
  if (in_mom.z() > 0) {
    approximator_ = m_aprox_ip_150_l;
    m_Zin_ = 0.0;  // Totem propagations assumes the starting point at 0 (zero)
    m_Zout_ = fPPSRegionStart_45;
  } else {
    approximator_ = m_aprox_ip_150_r;
    m_Zin_ = 0.0;  // Totem propagations assumes the starting point at 0 (zero)
    m_Zout_ = fPPSRegionStart_56;
  }
 
  bool invert_beam_coord_system =
      true;  // it doesn't matter the option here, it is hard coded as TRUE inside LHCOpticsApproximator!
 
  bool tracked = approximator_->Transport_m_GeV(
      in_position, in_momentum, out_position, out_momentum, invert_beam_coord_system, m_Zout_ - m_Zin_);
 
  if (!tracked)
    return false;
 
  edm::LogInfo("TotemTransport") << "after transport -> "
                                 << "position: " << out_position[0] << ", " << out_position[1] << ", "
                                 << out_position[2] << "momentum: " << out_momentum[0] << ", " << out_momentum[1]
                                 << ", " << out_momentum[2];
 
  if (out_position[0] * out_position[0] + out_position[1] * out_position[1] >
      m_beampipe_aperture_radius * m_beampipe_aperture_radius) {
    edm::LogInfo("TotemTransport") << "Proton ouside beampipe";
    edm::LogInfo("TotemTransport") << "===== END Transport "
                                   << "====================";
    return false;
  }
 
  TVector3 out_pos(out_position[0] * meter, out_position[1] * meter, out_position[2] * meter);
  TVector3 out_mom(out_momentum[0], out_momentum[1], out_momentum[2]);
 
  if (verbosity_) {
    LogDebug("TotemTransport") << "output -> position: ";
    out_pos.Print();
    LogDebug("TotemTransport") << " momentum: ";
    out_mom.Print();
  }
 
  double px = -out_momentum[0];  // tote calculates px by means of TH_X, which is in the LHC ref. frame.
  double py = out_momentum[1];   // this need to be checked again, since it seems an invertion is occuring in  the prop.
  double pz =
      out_momentum[2];  // totem calculates output pz already in the CMS ref. frame, it doesn't need to be converted
  double e = sqrt(px * px + py * py + pz * pz + ProtonMassSQ);
  TLorentzVector p_out(px, py, pz, e);
  double x1_ctpps = -out_position[0] * meter;  // Totem parameterization uses meter, one need it in millimeter
  double y1_ctpps = out_position[1] * meter;
 
  unsigned int line = in_trk->barcode();
 
  if (verbosity_)
    LogDebug("TotemTransport") << "TotemTransport:filterPPS: barcode = " << line << " x=  " << x1_ctpps
                               << " y= " << y1_ctpps;
 
  m_beamPart[line] = p_out;
  m_xAtTrPoint[line] = x1_ctpps;
  m_yAtTrPoint[line] = y1_ctpps;
  return true;
}

References BaseProtonTransport::ApplyBeamCorrection(), MillePedeFileConverter_cfg::e, BaseProtonTransport::fBeamXatIP, BaseProtonTransport::fBeamYatIP, BaseProtonTransport::fCrossingAngle_45, BaseProtonTransport::fCrossingAngle_56, BaseProtonTransport::fPPSRegionStart_45, BaseProtonTransport::fPPSRegionStart_56, BaseProtonTransport::fVtxMeanX, BaseProtonTransport::fVtxMeanY, BaseProtonTransport::fVtxMeanZ, mps_splice::line, LogDebug, m_aprox_ip_150_l, m_aprox_ip_150_r, BaseProtonTransport::m_beamPart, m_beampipe_aperture_radius, BaseProtonTransport::m_xAtTrPoint, BaseProtonTransport::m_yAtTrPoint, ProtonMassSQ, multPhiCorr_741_25nsDY_cfi::px, multPhiCorr_741_25nsDY_cfi::py, mathSSE::sqrt(), funct::tan(), LHCOpticsApproximator::Transport_m_GeV(), urad, and BaseProtonTransport::verbosity_.

Referenced by process().

Member Data Documentation

m_aprox_ip_150_l

LHCOpticsApproximator* TotemTransport::m_aprox_ip_150_l = nullptr

private

Definition at line 35 of file TotemTransport.h.

Referenced by TotemTransport(), and transportProton().

m_aprox_ip_150_r

LHCOpticsApproximator* TotemTransport::m_aprox_ip_150_r = nullptr

private

Definition at line 34 of file TotemTransport.h.

Referenced by TotemTransport(), and transportProton().

m_beampipe_aperture_radius

double TotemTransport::m_beampipe_aperture_radius

private

Definition at line 40 of file TotemTransport.h.

Referenced by transportProton().

m_fEtacut

double TotemTransport::m_fEtacut

private

Definition at line 41 of file TotemTransport.h.

m_fMomentumMin

double TotemTransport::m_fMomentumMin

private

Definition at line 42 of file TotemTransport.h.

m_model_ip_150_l_name

std::string TotemTransport::m_model_ip_150_l_name

private

Definition at line 38 of file TotemTransport.h.

Referenced by TotemTransport().

m_model_ip_150_r_name

std::string TotemTransport::m_model_ip_150_r_name

private

Definition at line 37 of file TotemTransport.h.

Referenced by TotemTransport().