TopKinFitter.h

Go to the documentation of this file.

#ifndef TopQuarkAnalysis_TopKinFitter_TopKinFitter_h
#define TopQuarkAnalysis_TopKinFitter_TopKinFitter_h

#include <memory>

#include "TMath.h"

#include "PhysicsTools/KinFitter/interface/TKinFitter.h"

/*
  \class   TopKinFitter TopKinFitter.h "TopQuarkAnalysis/TopKinFitter/interface/TopKinFitter.h"
  
  \brief   one line description to be added here...

  text to be added here...
  
**/

class TopKinFitter {
public:
  enum Param { kEMom, kEtEtaPhi, kEtThetaPhi };

public:
  explicit TopKinFitter(const int maxNrIter = 200,
                        const double maxDeltaS = 5e-5,
                        const double maxF = 1e-4,
                        const double mW = 80.4,
                        const double mTop = 173.);
  ~TopKinFitter();

  double fitS() const { return fitter_->getS(); };
  int fitNrIter() const { return fitter_->getNbIter(); };
  double fitProb() const { return TMath::Prob(fitter_->getS(), fitter_->getNDF()); };
  void setVerbosity(const int verbosityLevel) { fitter_->setVerbosity(verbosityLevel); };

protected:
  std::string param(const Param& param) const;

protected:
  std::unique_ptr<TKinFitter> fitter_;
  int maxNrIter_;
  double maxDeltaS_;
  double maxF_;
  double mW_;
  double mTop_;
};

#endif