LagrangeParentParticleFitter.h

Go to the documentation of this file.

#ifndef LagrangeParentParticleFitter_H
#define LagrangeParentParticleFitter_H

#include "RecoVertex/KinematicFit/interface/ParentParticleFitter.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

class LagrangeParentParticleFitter : public ParentParticleFitter {
public:
  LagrangeParentParticleFitter();

  ~LagrangeParentParticleFitter() override {}

  void setParameters(const edm::ParameterSet& pSet);

  std::vector<RefCountedKinematicTree> fit(const std::vector<RefCountedKinematicTree>& trees,
                                           KinematicConstraint* cs) const override;

  LagrangeParentParticleFitter* clone() const override { return new LagrangeParentParticleFitter(*this); }

private:
  void defaultParameters();

  float theMaxDiff;
  int theMaxStep;
};
#endif