BCFitter.h

#ifndef __BCFITTER__H
#define __BCFITTER__H

/*
 * Copyright (C) 2007-2018, the BAT core developer team
 * All rights reserved.
 *
 * For the licensing terms see doc/COPYING.
 * For documentation see http://mpp.mpg.de/bat
 */

// ---------------------------------------------------------

#include <BAT/BCAux.h>
#include <BAT/BCDataSet.h>
#include <BAT/BCModel.h>

#include <TF1.h>
#include <TH2D.h>

#include <string>

// ---------------------------------------------------------

class BCFitter : public BCModel
{
public:

    /* @{ */

    BCFitter(const TF1& f, const std::string& name = "fitter_model");

    virtual ~BCFitter() = 0;

    /* @} */

    /* @{ */

    TF1& GetFitFunction()
    {
        return fFitFunction.at(GetCurrentChain());
    }

    TH2* GetGraphicalErrorBandXY(double level = .68, int nsmooth = 0, bool overcoverage = true) const;

    const TH2& GetErrorBandXY()const
    { return fErrorBandXY; };

    std::vector<double> GetErrorBand(double level) const;

    TGraph* GetErrorBandGraph(double level1, double level2) const;

    TGraph* GetFitFunctionGraph(const std::vector<double>& parameters);

    TGraph* GetFitFunctionGraph()
    { return GetFitFunctionGraph(GetBestFitParameters()); }

    TGraph* GetFitFunctionGraph(const std::vector<double>& parameters, double xmin, double xmax, int n = 1000);

    void FixDataAxis(unsigned int index, bool fixed);

    bool GetFixedDataAxis(unsigned int index) const;

    double GetPValue() const
    {
        return fPValue;
    }

    /* @} */

    /* @{ */

    void SetErrorBandContinuous(bool flag);

    void SetErrorBandExtensionLowEdgeX(double extension)
    {
        fErrorBandExtensionLowEdgeX = extension;
    }

    void SetErrorBandExtensionUpEdgeX(double extension)
    {
        fErrorBandExtensionUpEdgeX = extension;
    }

    void SetErrorBandExtensionLowEdgeY(double extension)
    {
        fErrorBandExtensionLowEdgeY = extension;
    }

    void SetErrorBandExtensionUpEdgeY(double extension)
    {
        fErrorBandExtensionUpEdgeY = extension;
    }
    void SetFillErrorBand(bool flag = true)
    {
        fFlagFillErrorBand = flag;
    }

    void UnsetFillErrorBand()
    {
        fFlagFillErrorBand = false;
    }

    void SetFitFunctionIndexX(int index)
    {
        fFitFunctionIndexX = index;
    }

    void SetFitFunctionIndexY(int index)
    {
        fFitFunctionIndexY = index;
    }

    void SetFitFunctionIndices(int indexx, int indexy)
    {
        SetFitFunctionIndexX(indexx);
        SetFitFunctionIndexY(indexy);
    }

    void SetFlagIntegration(bool flag)
    {
        fFlagIntegration = flag;
    };

    virtual double FitFunction(const std::vector<double>& x, const std::vector<double>& parameters);

    double Integral(const std::vector<double>& parameters, double xmin, double xmax);


    /* @} */
    /* @{ */
    virtual void Fit() = 0;

    virtual void DrawFit(const std::string& options, bool flaglegend = false) = 0;

    virtual void MCMCUserIterationInterface();

    virtual void MarginalizePreprocess();

    void FillErrorBand();

    void PrintShortFitSummary();

    /* @} */

    virtual void MCMCUserInitialize();

private:
    std::vector<TF1> fFitFunction;

protected:
    bool fFlagFillErrorBand;

    int fFitFunctionIndexX;
    int fFitFunctionIndexY;

    bool fErrorBandContinuous;

    std::vector<double> fErrorBandX;

    unsigned fErrorBandNbinsX;

    unsigned fErrorBandNbinsY;

    double fPValue;

    TH2D fErrorBandXY;

    BCDataSet fFitterDataSet;

    bool fFlagIntegration;

    double fErrorBandExtensionLowEdgeX;
    double fErrorBandExtensionUpEdgeX;

    double fErrorBandExtensionLowEdgeY;

    double fErrorBandExtensionUpEdgeY;

    using BCModel::SetDataSet;
    using BCModel::GetDataSet;

    virtual double GraphCorrection(unsigned /* ibin */) const
    {
        return 1.0;
    }
};

// ---------------------------------------------------------

#endif