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