ProtonReconstructionAlgorithm Class Reference

#include <ProtonReconstructionAlgorithm.h>

Classes
class	ChiSquareCalculator
	class for calculation of chi^2 More...
struct	RPOpticsData
	optics data associated with 1 RP More...

Public Member Functions
void	init (const LHCInterpolatedOpticalFunctionsSetCollection &opticalFunctions)
	ProtonReconstructionAlgorithm (bool fit_vtx_y, bool improved_estimate, unsigned int verbosity)
reco::ForwardProton	reconstructFromMultiRP (const CTPPSLocalTrackLiteRefVector &tracks, const LHCInfo &lhcInfo, std::ostream &os) const
	run proton reconstruction using multiple-RP strategy More...
reco::ForwardProton	reconstructFromSingleRP (const CTPPSLocalTrackLiteRef &track, const LHCInfo &lhcInfo, std::ostream &os) const
	run proton reconstruction using single-RP strategy More...
void	release ()
	~ProtonReconstructionAlgorithm ()=default

Static Private Member Functions
static void	doLinearFit (const std::vector< double > &vx, const std::vector< double > &vy, double &b, double &a)

Private Attributes
std::unique_ptr< ChiSquareCalculator >	chiSquareCalculator_
	object to calculate chi^2 More...
std::unique_ptr< ROOT::Fit::Fitter >	fitter_
	fitter object More...
bool	fitVtxY_
bool	initialized_
std::map< unsigned int, RPOpticsData >	m_rp_optics_
	map: RP id –> optics data More...
bool	useImprovedInitialEstimate_
unsigned int	verbosity_

Detailed Description

Definition at line 26 of file ProtonReconstructionAlgorithm.h.

Constructor & Destructor Documentation

ProtonReconstructionAlgorithm::ProtonReconstructionAlgorithm	(	bool	fit_vtx_y,
		bool	improved_estimate,
		unsigned int	verbosity
	)

Definition at line 22 of file ProtonReconstructionAlgorithm.cc.

References chiSquareCalculator_, and fitter_.

                                                                                                                           :
  verbosity_(verbosity),
  fitVtxY_(fit_vtx_y),
  useImprovedInitialEstimate_(improved_estimate),
  initialized_(false),
  fitter_(new ROOT::Fit::Fitter), chiSquareCalculator_(new ChiSquareCalculator)
{
  // needed for thread safety
  TMinuitMinimizer::UseStaticMinuit(false);

  // initialise fitter
  double pStart[] = { 0, 0, 0, 0 };
  fitter_->SetFCN(4, *chiSquareCalculator_, pStart, 0, true);
}

ProtonReconstructionAlgorithm::~ProtonReconstructionAlgorithm ( )

default

Member Function Documentation

void ProtonReconstructionAlgorithm::doLinearFit ( const std::vector< double > &  vx, const std::vector< double > &  vy, double &  b, double &  a  )

staticprivate

Definition at line 83 of file ProtonReconstructionAlgorithm.cc.

References edmIntegrityCheck::d, and mps_fire::i.

Referenced by init().

{
  double s_1=0., s_x=0., s_xx=0., s_y=0., s_xy=0.;
  for (unsigned int i = 0; i < vx.size(); ++i)
  {
    s_1 += 1.;
    s_x += vx[i];
    s_xx += vx[i] * vx[i];
    s_y += vy[i];
    s_xy += vx[i] * vy[i];
  }

  const double d = s_xx * s_1 - s_x * s_x;
  a = ( s_1 * s_xy -  s_x * s_y) / d;
  b = (-s_x * s_xy + s_xx * s_y) / d;
}

void ProtonReconstructionAlgorithm::init

(

const LHCInterpolatedOpticalFunctionsSetCollection &

opticalFunctions

)

Definition at line 39 of file ProtonReconstructionAlgorithm.cc.

References ProtonReconstructionAlgorithm::RPOpticsData::ch0, ProtonReconstructionAlgorithm::RPOpticsData::ch1, doLinearFit(), LHCOpticalFunctionsSet::eLx, LHCOpticalFunctionsSet::eLy, LHCOpticalFunctionsSet::evy, LHCOpticalFunctionsSet::exd, LHCOpticalFunctionsSet::eyd, LHCOpticalFunctionsSet::getFcnValues(), LHCOpticalFunctionsSet::getXiValues(), initialized_, ProtonReconstructionAlgorithm::RPOpticsData::la0, ProtonReconstructionAlgorithm::RPOpticsData::la1, m_rp_optics_, eostools::move(), ProtonReconstructionAlgorithm::RPOpticsData::optics, AlCaHLTBitMon_ParallelJobs::p, release(), year_2016_postTS2_cff::rpId, ProtonReconstructionAlgorithm::RPOpticsData::s_L_y_vs_xi, ProtonReconstructionAlgorithm::RPOpticsData::s_v_y_vs_xi, ProtonReconstructionAlgorithm::RPOpticsData::s_xi_vs_x_d, ProtonReconstructionAlgorithm::RPOpticsData::s_y_d_vs_xi, LHCInterpolatedOpticalFunctionsSet::splines(), LHCInterpolatedOpticalFunctionsSet::transport(), LHCInterpolatedOpticalFunctionsSet::Kinematics::x, ProtonReconstructionAlgorithm::RPOpticsData::x0, LHCInterpolatedOpticalFunctionsSet::Kinematics::y, and ProtonReconstructionAlgorithm::RPOpticsData::y0.

Referenced by CTPPSProtonProducer::produce().

{
  // reset cache
  release();

  // build optics data for each object
  for (const auto &p : opticalFunctions)
  {
    const LHCInterpolatedOpticalFunctionsSet &ofs = p.second;

    // make record
    RPOpticsData rpod;
    rpod.optics = &p.second;
    rpod.s_y_d_vs_xi = ofs.splines()[LHCOpticalFunctionsSet::eyd];
    rpod.s_v_y_vs_xi = ofs.splines()[LHCOpticalFunctionsSet::evy];
    rpod.s_L_y_vs_xi = ofs.splines()[LHCOpticalFunctionsSet::eLy];

    vector<double> xiValues = ofs.getXiValues();  // local copy made since the TSpline constructor needs non-const parameters
    vector<double> xDValues = ofs.getFcnValues()[LHCOpticalFunctionsSet::exd];
    rpod.s_xi_vs_x_d = make_shared<TSpline3>("", xDValues.data(), xiValues.data(), xiValues.size());

    // calculate auxiliary data
    LHCInterpolatedOpticalFunctionsSet::Kinematics k_in = { 0., 0., 0., 0., 0. };
    LHCInterpolatedOpticalFunctionsSet::Kinematics k_out;
    rpod.optics->transport(k_in, k_out);
    rpod.x0 = k_out.x;
    rpod.y0 = k_out.y;

    doLinearFit(ofs.getXiValues(), ofs.getFcnValues()[LHCOpticalFunctionsSet::exd], rpod.ch0, rpod.ch1);
    rpod.ch0 -= rpod.x0;

    doLinearFit(ofs.getXiValues(), ofs.getFcnValues()[LHCOpticalFunctionsSet::eLx], rpod.la0, rpod.la1);

    // insert record
    const CTPPSDetId rpId(p.first);
    m_rp_optics_.emplace(rpId, std::move(rpod));
  }

  // update settings
  initialized_ = true;
}

reco::ForwardProton ProtonReconstructionAlgorithm::reconstructFromMultiRP	(	const CTPPSLocalTrackLiteRefVector &	tracks,
		const LHCInfo &	lhcInfo,
		std::ostream &	os
	)		const

run proton reconstruction using multiple-RP strategy

Definition at line 144 of file ProtonReconstructionAlgorithm.cc.

References a, protons_cff::arm, b, EnergyCorrector::c, ProtonReconstructionAlgorithm::RPOpticsData::ch0, ProtonReconstructionAlgorithm::RPOpticsData::ch1, chiSquareCalculator_, LHCInfo::energy(), fitter_, fitVtxY_, mps_fire::i, createfilelist::int, ProtonReconstructionAlgorithm::RPOpticsData::la0, ProtonReconstructionAlgorithm::RPOpticsData::la1, m_rp_optics_, AlCaHLTBitMon_ParallelJobs::p, mps_fire::result, mathSSE::sqrt(), HiIsolationCommonParameters_cff::track, l1t::tracks, useImprovedInitialEstimate_, verbosity_, protons_cff::xi, and y.

Referenced by CTPPSProtonProducer::produce().

{
  // make sure optics is available for all tracks
  for (const auto &it : tracks) {
    auto oit = m_rp_optics_.find(it->getRPId());
    if (oit == m_rp_optics_.end())
      throw cms::Exception("ProtonReconstructionAlgorithm") << "Optics data not available for RP " <<
        it->getRPId() << ", i.e. " << CTPPSDetId(it->getRPId()) << ".";
  }

  // initial estimate of xi and th_x
  double xi_init = 0., th_x_init = 0.;

  if (useImprovedInitialEstimate_) {
    double x_N = tracks[0]->getX()*1E-1,  // conversion: mm --> cm
      x_F = tracks[1]->getX()*1E-1;

    const RPOpticsData &i_N = m_rp_optics_.find(tracks[0]->getRPId())->second,
      &i_F = m_rp_optics_.find(tracks[1]->getRPId())->second;

    const double a = i_F.ch1*i_N.la1 - i_N.ch1*i_F.la1;
    const double b = i_F.ch0*i_N.la1 - i_N.ch0*i_F.la1 + i_F.ch1*i_N.la0 - i_N.ch1*i_F.la0 + x_N*i_F.la1 - x_F*i_N.la1;
    const double c = x_N*i_F.la0 - x_F*i_N.la0 + i_F.ch0*i_N.la0 - i_N.ch0*i_F.la0;
    const double D = b*b - 4.*a*c;
    const double sqrt_D = (D >= 0.) ? sqrt(D) : 0.;

    xi_init = (-b + sqrt_D) / 2. / a;
    th_x_init = (x_N - i_N.ch0 - i_N.ch1 * xi_init) / (i_N.la0 + i_N.la1 * xi_init);
  }
  else {
    double s_xi0 = 0., s_1 = 0.;
    for (const auto &track : tracks) {
      auto oit = m_rp_optics_.find(track->getRPId());
      double xi = oit->second.s_xi_vs_x_d->Eval(track->getX()*1E-1 + oit->second.x0);  // conversion: mm --> cm

      s_1 += 1.;
      s_xi0 += xi;
    }

    xi_init = s_xi0 / s_1;
  }

  if (!std::isfinite(xi_init))
    xi_init = 0.;
  if (!std::isfinite(th_x_init))
    th_x_init = 0.;

  // initial estimate of th_y and vtx_y
  double y[2]={0}, v_y[2]={0}, L_y[2]={0};
  unsigned int y_idx = 0;
  for (const auto &track : tracks) {
    if (y_idx >= 2) break;

    auto oit = m_rp_optics_.find(track->getRPId());

    y[y_idx] = track->getY()*1E-1 - oit->second.s_y_d_vs_xi->Eval(xi_init); // track y: mm --> cm
    v_y[y_idx] = oit->second.s_v_y_vs_xi->Eval(xi_init);
    L_y[y_idx] = oit->second.s_L_y_vs_xi->Eval(xi_init);

    y_idx++;
  }

  double vtx_y_init = 0.;
  double th_y_init = 0.;

  if (fitVtxY_) {
    const double det_y = v_y[0] * L_y[1] - L_y[0] * v_y[1];
    vtx_y_init = (det_y != 0.) ? (L_y[1] * y[0] - L_y[0] * y[1]) / det_y : 0.;
    th_y_init = (det_y != 0.) ? (v_y[0] * y[1] - v_y[1] * y[0]) / det_y : 0.;
  }
  else {
    vtx_y_init = 0.;
    th_y_init = (y[1]/L_y[1] + y[0]/L_y[0]) / 2.;
  }

  if (!std::isfinite(vtx_y_init))
    vtx_y_init = 0.;
  if (!std::isfinite(th_y_init))
    th_y_init = 0.;

  unsigned int armId = CTPPSDetId((*tracks.begin())->getRPId()).arm();

  if (verbosity_)
    os
      << "ProtonReconstructionAlgorithm::reconstructFromMultiRP(" << armId << ")" << std::endl
      << "    initial estimate: xi_init = " << xi_init << ", th_x_init = " << th_x_init
      << ", th_y_init = " << th_y_init << ", vtx_y_init = " << vtx_y_init << "." << std::endl;

  // minimisation
  fitter_->Config().ParSettings(0).Set("xi", xi_init, 0.005);
  fitter_->Config().ParSettings(1).Set("th_x", th_x_init, 2E-6);
  fitter_->Config().ParSettings(2).Set("th_y", th_y_init, 2E-6);
  fitter_->Config().ParSettings(3).Set("vtx_y", vtx_y_init, 10E-6);

  if (!fitVtxY_)
    fitter_->Config().ParSettings(3).Fix();

  chiSquareCalculator_->tracks = &tracks;
  chiSquareCalculator_->m_rp_optics = &m_rp_optics_;

  fitter_->FitFCN();
  fitter_->FitFCN();  // second minimisation in case the first one had troubles

  // extract proton parameters
  const ROOT::Fit::FitResult& result = fitter_->Result();
  const double *params = result.GetParams();

  if (verbosity_)
    os
      << "    fit valid=" << result.IsValid()
      << std::endl
      << "    xi=" << params[0] << " +- " << result.Error(0)
      << ", th_x=" << params[1] << " +-" << result.Error(1)
      << ", th_y=" << params[2] << " +-" << result.Error(2)
      << ", vtx_y=" << params[3] << " +-" << result.Error(3)
      << ", chiSq = " << result.Chi2() << std::endl;

  // save reco candidate
  using FP = reco::ForwardProton;

  const double sign_z = (armId == 0) ? +1. : -1.;  // CMS convention
  const FP::Point vertex(0., params[3], 0.);
  const double xi = params[0];
  const double th_x = params[1];
  const double th_y = params[2];
  const double cos_th = sqrt(1. - th_x*th_x - th_y*th_y);
  const double p = lhcInfo.energy() * (1. - xi);
  const FP::Vector momentum(
    - p * th_x,   // the signs reflect change LHC --> CMS convention
    + p * th_y,
    sign_z * p * cos_th
  );
  signed int ndf = 2.*tracks.size() - ((fitVtxY_) ? 4. : 3.);

  map<unsigned int, signed int> index_map = {
    {(unsigned int) FP::Index::xi, 0},
    {(unsigned int) FP::Index::th_x, 1},
    {(unsigned int) FP::Index::th_y, 2},
    {(unsigned int) FP::Index::vtx_y, ((fitVtxY_) ? 3 : -1)},
    {(unsigned int) FP::Index::vtx_x, -1},
  };

  FP::CovarianceMatrix cm;
  for (unsigned int i = 0; i < (unsigned int) FP::Index::num_indices; ++i) {
    signed int fit_i = index_map[i];

    for (unsigned int j = 0; j < (unsigned int) FP::Index::num_indices; ++j) {
      signed int fit_j = index_map[j];

      cm(i, j) = (fit_i >= 0 && fit_j >= 0) ? result.CovMatrix(fit_i, fit_j) : 0.;
    }
  }

  return reco::ForwardProton(result.Chi2(), ndf, vertex, momentum, xi, cm,
    FP::ReconstructionMethod::multiRP, tracks, result.IsValid());
}

reco::ForwardProton ProtonReconstructionAlgorithm::reconstructFromSingleRP	(	const CTPPSLocalTrackLiteRef &	track,
		const LHCInfo &	lhcInfo,
		std::ostream &	os
	)		const

run proton reconstruction using single-RP strategy

Definition at line 304 of file ProtonReconstructionAlgorithm.cc.

References funct::abs(), protons_cff::arm, LHCInfo::energy(), m_rp_optics_, AlCaHLTBitMon_ParallelJobs::p, funct::pow(), edm::RefVector< C, T, F >::push_back(), year_2016_postTS2_cff::rpId, mathSSE::sqrt(), verbosity_, and protons_cff::xi.

Referenced by CTPPSProtonProducer::produce().

{
  CTPPSDetId rpId(track->getRPId());

  if (verbosity_)
    os << "reconstructFromSingleRP(" << rpId.arm()*100 + rpId.station()*10 + rpId.rp() << ")" << std::endl;

  // make sure optics is available for the track
  auto oit = m_rp_optics_.find(track->getRPId());
  if (oit == m_rp_optics_.end())
    throw cms::Exception("ProtonReconstructionAlgorithm") << "Optics data not available for RP " << track->getRPId()
      << ", i.e. " << rpId << ".";

  // rough estimate of xi and th_y from each track
  const double x_full = track->getX()*1E-1 + oit->second.x0; // conversion mm --> cm
  const double xi = oit->second.s_xi_vs_x_d->Eval(x_full);
  const double L_y = oit->second.s_L_y_vs_xi->Eval(xi);
  const double th_y = track->getY()*1E-1 / L_y; // conversion mm --> cm

  const double ep_x = 1E-6;
  const double dxi_dx = (oit->second.s_xi_vs_x_d->Eval(x_full + ep_x) - xi) / ep_x;
  const double xi_unc = abs(dxi_dx) * track->getXUnc() * 1E-1; // conversion mm --> cm

  const double ep_xi = 1E-4;
  const double dL_y_dxi = ( oit->second.s_L_y_vs_xi->Eval(xi + ep_xi) - L_y ) / ep_xi;
  const double th_y_unc = th_y * sqrt( pow(track->getYUnc() / track->getY(), 2.) + pow(dL_y_dxi * xi_unc / L_y, 2.) );

  if (verbosity_)
    os << "    xi = " << xi << " +- " << xi_unc << ", th_y = " << th_y << " +- " << th_y_unc << "." << std::endl;

  using FP = reco::ForwardProton;

  // save proton candidate
  const double sign_z = (CTPPSDetId(track->getRPId()).arm() == 0) ? +1. : -1.;  // CMS convention
  const FP::Point vertex(0., 0., 0.);
  const double cos_th = sqrt(1. - th_y*th_y);
  const double p = lhcInfo.energy() * (1. - xi);
  const FP::Vector momentum(0., p * th_y, sign_z * p * cos_th);

  FP::CovarianceMatrix cm;
  cm((int)FP::Index::xi, (int)FP::Index::xi) = xi_unc * xi_unc;
  cm((int)FP::Index::th_y, (int)FP::Index::th_y) = th_y_unc * th_y_unc;

  CTPPSLocalTrackLiteRefVector trk;
  trk.push_back( track );

  return reco::ForwardProton(0., 0, vertex, momentum, xi, cm, FP::ReconstructionMethod::singleRP, trk, true);
}

void ProtonReconstructionAlgorithm::release

(

)

Definition at line 102 of file ProtonReconstructionAlgorithm.cc.

References initialized_, and m_rp_optics_.

Referenced by init(), and CTPPSProtonProducer::produce().

{
  initialized_ = false;

  m_rp_optics_.clear();
}

Member Data Documentation

std::unique_ptr<ChiSquareCalculator> ProtonReconstructionAlgorithm::chiSquareCalculator_

private

object to calculate chi^2

Definition at line 81 of file ProtonReconstructionAlgorithm.h.

Referenced by ProtonReconstructionAlgorithm(), and reconstructFromMultiRP().

std::unique_ptr<ROOT::Fit::Fitter> ProtonReconstructionAlgorithm::fitter_

private

fitter object

Definition at line 78 of file ProtonReconstructionAlgorithm.h.

Referenced by ProtonReconstructionAlgorithm(), and reconstructFromMultiRP().

bool ProtonReconstructionAlgorithm::fitVtxY_

private

Definition at line 45 of file ProtonReconstructionAlgorithm.h.

Referenced by reconstructFromMultiRP().

bool ProtonReconstructionAlgorithm::initialized_

private

Definition at line 47 of file ProtonReconstructionAlgorithm.h.

Referenced by init(), and release().

std::map<unsigned int, RPOpticsData> ProtonReconstructionAlgorithm::m_rp_optics_

private

map: RP id –> optics data

Definition at line 63 of file ProtonReconstructionAlgorithm.h.

Referenced by init(), reconstructFromMultiRP(), reconstructFromSingleRP(), and release().

bool ProtonReconstructionAlgorithm::useImprovedInitialEstimate_

private

Definition at line 46 of file ProtonReconstructionAlgorithm.h.

Referenced by reconstructFromMultiRP().

unsigned int ProtonReconstructionAlgorithm::verbosity_

private

Definition at line 44 of file ProtonReconstructionAlgorithm.h.

Referenced by reconstructFromMultiRP(), and reconstructFromSingleRP().