KinematicConstrainedVertexFitter.h

Go to the documentation of this file.

#ifndef KinematicConstrainedVertexFitter_H
#define KinematicConstrainedVertexFitter_H
 
#include "RecoVertex/KinematicFitPrimitives/interface/RefCountedKinematicTree.h"
#include "RecoVertex/KinematicFitPrimitives/interface/MultiTrackKinematicConstraint.h"
#include "RecoVertex/VertexTools/interface/LinearizationPointFinder.h"
#include "RecoVertex/KinematicFit/interface/KinematicConstrainedVertexUpdator.h"
#include "RecoVertex/KinematicFit/interface/VertexKinematicConstraint.h"
#include "RecoVertex/KinematicFit/interface/ConstrainedTreeBuilder.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
 
class KinematicConstrainedVertexFitter {
public:
  KinematicConstrainedVertexFitter();
 
  KinematicConstrainedVertexFitter(const LinearizationPointFinder &fnd);
 
  ~KinematicConstrainedVertexFitter();
 
  void setParameters(const edm::ParameterSet &pSet);
 
  RefCountedKinematicTree fit(const std::vector<RefCountedKinematicParticle> &part) {
    return fit(part, nullptr, nullptr);
  }
 
  RefCountedKinematicTree fit(const std::vector<RefCountedKinematicParticle> &part, MultiTrackKinematicConstraint *cs) {
    return fit(part, cs, nullptr);
  };
 
  RefCountedKinematicTree fit(const std::vector<RefCountedKinematicParticle> &part,
                              MultiTrackKinematicConstraint *cs,
                              GlobalPoint *pt);
 
  //return the number of iterations
  int getNit() const;
  //return the value of the constraint equation
  float getCSum() const;
 
private:
  void defaultParameters();
 
  float theMaxDelta;  //maximum (delta parameter)^2/(sigma parameter)^2 per iteration for convergence
  int theMaxStep;
  float theMaxReducedChiSq;  //max of initial (after 2 iterations) chisq/dof value
  float theMinChiSqImprovement;  //minimum required improvement in chisq to avoid fit termination for cases exceeding theMaxReducedChiSq
  LinearizationPointFinder *finder;
  KinematicConstrainedVertexUpdator *updator;
  VertexKinematicConstraint *vCons;
  ConstrainedTreeBuilder *tBuilder;
  int iterations;
  float csum;
};
 
#endif