KinematicParticleFitter.h

Go to the documentation of this file.

#ifndef KinematicParticleFitter_H
#define KinematicParticleFitter_H

#include "RecoVertex/KinematicFitPrimitives/interface/RefCountedKinematicParticle.h"
#include "RecoVertex/KinematicFitPrimitives/interface/RefCountedKinematicVertex.h"
#include "RecoVertex/KinematicFitPrimitives/interface/KinematicConstraint.h"
#include "RecoVertex/KinematicFitPrimitives/interface/RefCountedKinematicTree.h"
#include "RecoVertex/KinematicFit/interface/ParentParticleFitter.h"
#include "RecoVertex/KinematicFit/interface/ChildUpdator.h"

class KinematicParticleFitter {
public:
  KinematicParticleFitter();

  KinematicParticleFitter(const ParentParticleFitter& fitter, const ChildUpdator& updator);

  ~KinematicParticleFitter();

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

  RefCountedKinematicTree fit(KinematicConstraint* cs, RefCountedKinematicTree tree) const;

private:
  ParentParticleFitter* parentFitter;
  ChildUpdator* cUpdator;
};

#endif