CandKinematicVertexFitter.h

Go to the documentation of this file.

#ifndef RecoCandUtils_CandKinematicVertexFitter_h
#define RecoCandUtils_CandKinematicVertexFitter_h
/* \class CandKinematicVertexFitter
 *
 * \author Luca Lista, INFN
 *
 */

#include "FWCore/ParameterSet/interface/ParameterSetfwd.h"
#include "TrackingTools/TransientTrack/interface/TransientTrack.h"
#include "RecoVertex/VertexPrimitives/interface/TransientVertex.h"
#include "DataFormats/RecoCandidate/interface/RecoCandidate.h"
#include "RecoVertex/KinematicFit/interface/KinematicParticleVertexFitter.h"
#include "PhysicsTools/UtilAlgos/interface/ParameterAdapter.h"
#include "RecoVertex/KinematicFitPrimitives/interface/KinematicParticleFactoryFromTransientTrack.h"
#include "SimGeneral/HepPDTRecord/interface/ParticleDataTable.h"
#include <vector>
#include "boost/shared_ptr.hpp"

class MagneticField;
namespace reco { class VertexCompositeCandidate; }

class CandKinematicVertexFitter {
public:
  typedef reco::Vertex::CovarianceMatrix CovarianceMatrix;
  CandKinematicVertexFitter(const edm::ParameterSet & cfg) :  
    bField_(nullptr), pdt_(nullptr), fitter_(), fitters_(new std::vector<CandKinematicVertexFitter>) { 
  }
  CandKinematicVertexFitter(const CandKinematicVertexFitter& o) :
    bField_(o.bField_), pdt_(o.pdt_), fitter_(), fitters_(new std::vector<CandKinematicVertexFitter>) {
  }
  void set(const MagneticField * bField) { bField_ = bField; }
  void set(const ParticleDataTable * pdt) { pdt_ = pdt; }
  void set(reco::VertexCompositeCandidate &) const;
  bool fit(const std::vector<RefCountedKinematicParticle> & tracks) const;
  RefCountedKinematicParticle currentParticle() const {
    tree_->movePointerToTheTop();
    return tree_->currentParticle();
  }
private:
  const MagneticField * bField_;
  const ParticleDataTable * pdt_;
  void fill(std::vector<RefCountedKinematicParticle> &,
        std::vector<reco::Candidate *> &,
        std::vector<reco::RecoCandidate::TrackType> &,
        reco::Candidate &) const;
  KinematicParticleVertexFitter fitter_;
  mutable RefCountedKinematicTree tree_;
  KinematicParticleFactoryFromTransientTrack factory_;
  mutable double chi2_;
  mutable double ndof_;
  mutable CovarianceMatrix cov_;
  boost::shared_ptr<std::vector<CandKinematicVertexFitter> > fitters_;
};

#endif