CandCommonVertexFitterBase Class Reference

#include <CandCommonVertexFitter.h>

Inheritance diagram for CandCommonVertexFitterBase:

Public Types
typedef reco::Vertex::CovarianceMatrix	CovarianceMatrix

Public Member Functions
	CandCommonVertexFitterBase (const edm::ParameterSet &)
void	set (const MagneticField *bField)
void	set (reco::VertexCompositeCandidate &) const
virtual	~CandCommonVertexFitterBase ()

Protected Member Functions
void	fill (std::vector< reco::TransientTrack > &, std::vector< reco::Candidate * > &, std::vector< reco::RecoCandidate::TrackType > &, reco::Candidate &) const
virtual bool	fit (TransientVertex &, const std::vector< reco::TransientTrack > &) const =0

Protected Attributes
const MagneticField *	bField_
double	chi2_
	chi-sqared More...
CovarianceMatrix	cov_
	covariance matrix (3x3) More...
double	ndof_
	number of degrees of freedom More...

Detailed Description

Definition at line 19 of file CandCommonVertexFitter.h.

Member Typedef Documentation

typedef reco::Vertex::CovarianceMatrix CandCommonVertexFitterBase::CovarianceMatrix

Definition at line 21 of file CandCommonVertexFitter.h.

Constructor & Destructor Documentation

CandCommonVertexFitterBase::CandCommonVertexFitterBase ( const edm::ParameterSet &  )

inline

Definition at line 22 of file CandCommonVertexFitter.h.

: bField_(nullptr) {}

virtual CandCommonVertexFitterBase::~CandCommonVertexFitterBase ( )

inlinevirtual

Definition at line 23 of file CandCommonVertexFitter.h.

{}

Member Function Documentation

void CandCommonVertexFitterBase::fill ( std::vector< reco::TransientTrack > &  tracks, std::vector< reco::Candidate * > &  daughters, std::vector< reco::RecoCandidate::TrackType > &  trackTypes, reco::Candidate &  c  ) const

protected

Definition at line 60 of file CandCommonVertexFitter.cc.

References EcnaPython_AdcPeg12_S1_10_R170298_1_0_150_Dee0::cerr, ztail::d, d1, reco::Candidate::daughter(), Exception, PVValHelper::fill(), reco::Candidate::get(), edm::errors::InvalidReference, dqmiolumiharvest::j, reco::Candidate::numberOfDaughters(), and reco::Candidate::pdgId().

                                                          {
  size_t nDau = c.numberOfDaughters();
  for (unsigned int j = 0; j < nDau; ++j) {
    Candidate *d = c.daughter(j);
    if (d == nullptr) {
      ostringstream message;
      message << "Can't access in write mode candidate daughters. "
              << "pdgId = " << c.pdgId() << ".\n";
      const Candidate *d1 = c.daughter(j);
      if (d1 == nullptr)
        message << "Null daughter also found in read-only mode\n";
      else
        message << "Daughter found in read-only mode with id: " << d1->pdgId() << "\n";
      throw edm::Exception(edm::errors::InvalidReference) << message.str();
    }
    if (d->numberOfDaughters() > 0)
      fill(tracks, daughters, trackTypes, *d);
    else {
      const Track *trk = d->get<const Track *>();
      RecoCandidate::TrackType type = d->get<RecoCandidate::TrackType>();
      if (trk != nullptr) {
        tracks.push_back(TransientTrack(*trk, bField_));
        daughters.push_back(d);
        trackTypes.push_back(type);
      } else {
        cerr << ">>> warning: candidate of type " << d->pdgId() << " has no track reference." << endl;
      }
    }
  }
}

virtual bool CandCommonVertexFitterBase::fit ( TransientVertex &  , const std::vector< reco::TransientTrack > &    ) const

protectedpure virtual

Implemented in CandCommonVertexFitter< Fitter >.

Referenced by trackingPlots.Iteration::modules().

void CandCommonVertexFitterBase::set ( const MagneticField *  bField )

inline

Definition at line 24 of file CandCommonVertexFitter.h.

References ecalTB2006H4_GenSimDigiReco_cfg::bField, and bField_.

Referenced by reco::modules::CandVertexFitterEventSetupInit< Fitter >::init().

{ bField_ = bField; }

void CandCommonVertexFitterBase::set

(

reco::VertexCompositeCandidate &

c

)

const

Definition at line 9 of file CandCommonVertexFitter.cc.

References cms::cuda::assert(), reco::Vertex::chi2(), relval_parameters_module::energy, reco::Candidate::energy(), Exception, PVValHelper::fill(), reco::Vertex::fill(), gpuVertexFinder::fit, edm::errors::InvalidReference, ResonanceBuilder::mass, reco::Candidate::mass(), reco::Vertex::ndof(), AlCaHLTBitMon_ParallelJobs::p, reco::TrackBase::p(), reco::TrackBase::px(), reco::TrackBase::py(), reco::TrackBase::pz(), reco::RecoCandidate::recoTrackType, TransientVertex::refittedTracks(), pileupReCalc_HLTpaths::scale, reco::VertexCompositeCandidate::setChi2AndNdof(), reco::VertexCompositeCandidate::setCovariance(), reco::Candidate::setP4(), reco::LeafCandidate::setP4(), reco::Candidate::setVertex(), reco::LeafCandidate::setVertex(), mathSSE::sqrt(), HLT_FULL_cff::track, tracks, and findQualityFiles::v.

                                                                      {
  if (bField_ == nullptr)
    throw edm::Exception(edm::errors::InvalidReference)
        << "B-Field was not set up CandCommonVertexFitter.\n"
        << "the following method must be called before fitting a candidate:\n"
        << " CandCommonVertexFitter:.set( const MagneticField * )" << endl;
  vector<TransientTrack> tracks;
  vector<Candidate *> daughters;
  vector<RecoCandidate::TrackType> trackTypes;
  fill(tracks, daughters, trackTypes, c);
  assert(tracks.size() == daughters.size());
  TransientVertex vertex;
  if (fit(vertex, tracks)) {
    tracks = vertex.refittedTracks();
    Candidate::Point vtx(vertex.position());
    c.setVertex(vtx);
    vector<TransientTrack>::const_iterator trackIt = tracks.begin(), tracksEnd = tracks.end();
    vector<Candidate *>::const_iterator daughterIt = daughters.begin();
    vector<RecoCandidate::TrackType>::const_iterator trackTypeIt = trackTypes.begin();
    Candidate::LorentzVector mp4(0, 0, 0, 0);
    for (; trackIt != tracksEnd; ++trackIt, ++daughterIt, ++trackTypeIt) {
      const Track &track = trackIt->track();
      Candidate &daughter = **daughterIt;
      double px = track.px(), py = track.py(), pz = track.pz(), p = track.p();
      double energy;
      daughter.setVertex(vtx);
      if (*trackTypeIt == RecoCandidate::recoTrackType) {
        double mass = daughter.mass();
        energy = sqrt(p * p + mass * mass);
      } else {
        energy = daughter.energy();
        double scale = energy / p;
        px *= scale;
        py *= scale;
        pz *= scale;
      }
      Candidate::LorentzVector dp4(px, py, pz, energy);
      daughter.setP4(dp4);
      mp4 += dp4;
    }
    c.setP4(mp4);
    Vertex v = vertex;
    c.setChi2AndNdof(chi2_ = v.chi2(), ndof_ = v.ndof());
    v.fill(cov_);
    c.setCovariance(cov_);
  } else {
    c.setChi2AndNdof(chi2_ = -1, ndof_ = 0);
    c.setCovariance(cov_ = CovarianceMatrix(ROOT::Math::SMatrixIdentity()));
  }
}

Member Data Documentation

const MagneticField* CandCommonVertexFitterBase::bField_

protected

Definition at line 28 of file CandCommonVertexFitter.h.

Referenced by set().

double CandCommonVertexFitterBase::chi2_

mutableprotected

chi-sqared

Definition at line 35 of file CandCommonVertexFitter.h.

CovarianceMatrix CandCommonVertexFitterBase::cov_

mutableprotected

covariance matrix (3x3)

Definition at line 39 of file CandCommonVertexFitter.h.

double CandCommonVertexFitterBase::ndof_

mutableprotected

number of degrees of freedom

Definition at line 37 of file CandCommonVertexFitter.h.