FitControl.h

Go to the documentation of this file.

#pragma once

#include <goofit/Error.h>
#include <goofit/GlobalCudaDefines.h>

#include <string>
#include <vector>

namespace GooFit {

class PdfBase;

class FitControl {
  public:
    FitControl(bool bin, std::string mn)
        : binned(bin)
        , metricName(mn) {}

    inline bool binnedFit() const { return binned; }
    inline bool binErrors() const { return errorsOnBins; }
    inline bool metricIsPdf() const { return !errorsOnBins; }
    inline std::string getMetric() const { return metricName; }

  protected:
    bool errorsOnBins{false};

  private:
    bool binned;
    std::string metricName;
};

class UnbinnedNllFit : public FitControl {
  public:
    UnbinnedNllFit()
        : FitControl(false, "ptr_to_NLL") {}
};

class BinnedNllFit : public FitControl {
  public:
    BinnedNllFit()
        : FitControl(true, "ptr_to_BinAvg") {}
};

class BinnedErrorFit : public FitControl {
  public:
    BinnedErrorFit()
        : FitControl(true, "ptr_to_BinWithError") {
        errorsOnBins = true;
    }
};

class BinnedChisqFit : public FitControl {
  public:
    BinnedChisqFit()
        : FitControl(true, "ptr_to_Chisq") {}
};

} // namespace GooFit