FitManagerMinuit2.h

Go to the documentation of this file.

#pragma once

#include <Minuit2/FunctionMinimum.h>

#include <memory>

#include <goofit/GlobalCudaDefines.h>
#include <goofit/fitting/FCN.h>

namespace GooFit {

class PdfBase;

enum class FitErrors : int { Valid = 0, NotRun = 50, InValid };

class FitManagerMinuit2 {
  public:
    FitManagerMinuit2(PdfBase *dat);

    ROOT::Minuit2::FunctionMinimum fit();

    void setMaxCalls(unsigned int max_calls = 0) { maxfcn_ = max_calls; }

    Params *getParams() { return &upar_; }

    FCN *getFCN() { return &fcn_; }

    operator bool() const { return retval_ == FitErrors::Valid; }

    operator int() const { return static_cast<int>(retval_); }

    void setVerbosity(int value) { verbosity = value; }

    int getVerbosity() const { return verbosity; }

  private:
    Params upar_;
    FCN fcn_;
    unsigned int maxfcn_{0};
    FitErrors retval_{FitErrors::NotRun};
    int verbosity{3};
};
} // namespace GooFit