PixelNtupletsFitter Class Reference

#include <PixelNtupletsFitter.h>

Inheritance diagram for PixelNtupletsFitter:

Public Member Functions
	PixelNtupletsFitter (float nominalB, const MagneticField *field, bool useRiemannFit)
std::unique_ptr< reco::Track >	run (const std::vector< const TrackingRecHit * > &hits, const TrackingRegion &region) const override
	~PixelNtupletsFitter () override=default
Public Member Functions inherited from PixelFitterBase
virtual	~PixelFitterBase ()

Private Attributes
const MagneticField *	field_
float	nominalB_
bool	useRiemannFit_

Detailed Description

Definition at line 12 of file PixelNtupletsFitter.h.

Constructor & Destructor Documentation

PixelNtupletsFitter::PixelNtupletsFitter ( float  nominalB, const MagneticField *  field, bool  useRiemannFit  )

explicit

Definition at line 22 of file PixelNtupletsFitter.cc.

    : nominalB_(nominalB), field_(field), useRiemannFit_(useRiemannFit) {}

PixelNtupletsFitter::~PixelNtupletsFitter ( )

overridedefault

Member Function Documentation

std::unique_ptr< reco::Track > PixelNtupletsFitter::run ( const std::vector< const TrackingRecHit * > &  hits, const TrackingRegion &  region  ) const

overridevirtual

Implements PixelFitterBase.

Definition at line 25 of file PixelNtupletsFitter.cc.

References cms::cuda::assert(), PV3DBase< T, PVType, FrameType >::basicVector(), HLT_FULL_cff::chi2, declareDynArray, field_, brokenline::helixFit(), riemannFit::helixFit(), mps_fire::i, PixelPluginsPhase0_cfi::isBarrel, nhits, nominalB_, TrackingRegion::origin(), phi, DiDispStaMuonMonitor_cfi::pt, runTheMatrix::ret, mathSSE::sqrt(), HLT_FULL_cff::tip, useRiemannFit_, and ComparisonHelper::zip().

                                                                                          {
  using namespace riemannFit;

  std::unique_ptr<reco::Track> ret;

  unsigned int nhits = hits.size();

  if (nhits < 2)
    return ret;

  declareDynArray(GlobalPoint, nhits, points);
  declareDynArray(GlobalError, nhits, errors);
  declareDynArray(bool, nhits, isBarrel);

  for (unsigned int i = 0; i != nhits; ++i) {
    auto const& recHit = hits[i];
    points[i] = GlobalPoint(recHit->globalPosition().basicVector() - region.origin().basicVector());
    errors[i] = recHit->globalPositionError();
    isBarrel[i] = recHit->detUnit()->type().isBarrel();
  }

  assert(nhits == 4);
  riemannFit::Matrix3xNd<4> hits_gp;

  Eigen::Matrix<float, 6, 4> hits_ge = Eigen::Matrix<float, 6, 4>::Zero();

  for (unsigned int i = 0; i < nhits; ++i) {
    hits_gp.col(i) << points[i].x(), points[i].y(), points[i].z();

    hits_ge.col(i) << errors[i].cxx(), errors[i].cyx(), errors[i].cyy(), errors[i].czx(), errors[i].czy(),
        errors[i].czz();
  }

  HelixFit fittedTrack = useRiemannFit_ ? riemannFit::helixFit(hits_gp, hits_ge, nominalB_, true)
                                        : brokenline::helixFit(hits_gp, hits_ge, nominalB_);

  int iCharge = fittedTrack.qCharge;

  // parameters are:
  // 0: phi
  // 1: tip
  // 2: curvature
  // 3: cottheta
  // 4: zip
  float valPhi = fittedTrack.par(0);

  float valTip = fittedTrack.par(1);

  float valCotTheta = fittedTrack.par(3);

  float valZip = fittedTrack.par(4);
  float valPt = fittedTrack.par(2);
  //
  //  PixelTrackErrorParam param(valEta, valPt);
  float errValPhi = std::sqrt(fittedTrack.cov(0, 0));
  float errValTip = std::sqrt(fittedTrack.cov(1, 1));

  float errValPt = std::sqrt(fittedTrack.cov(2, 2));

  float errValCotTheta = std::sqrt(fittedTrack.cov(3, 3));
  float errValZip = std::sqrt(fittedTrack.cov(4, 4));

  float chi2 = fittedTrack.chi2_line + fittedTrack.chi2_circle;

  PixelTrackBuilder builder;
  Measurement1D phi(valPhi, errValPhi);
  Measurement1D tip(valTip, errValTip);

  Measurement1D pt(valPt, errValPt);
  Measurement1D cotTheta(valCotTheta, errValCotTheta);
  Measurement1D zip(valZip, errValZip);

  ret.reset(builder.build(pt, phi, cotTheta, tip, zip, chi2, iCharge, hits, field_, region.origin()));
  return ret;
}

Member Data Documentation

const MagneticField* PixelNtupletsFitter::field_

private

Definition at line 21 of file PixelNtupletsFitter.h.

Referenced by run().

float PixelNtupletsFitter::nominalB_

private

Definition at line 20 of file PixelNtupletsFitter.h.

Referenced by run().

bool PixelNtupletsFitter::useRiemannFit_

private

Definition at line 22 of file PixelNtupletsFitter.h.

Referenced by run().