BCEfficiencyFitter.h

#ifndef __BCEFFICIENCYFITTER__H
#define __BCEFFICIENCYFITTER__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 "BCFitter.h"

#include <TH1D.h>

#include <vector>

// ROOT classes
class TGraphAsymmErrors;

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

class BCEfficiencyFitter : public BCFitter
{
public:

    enum DataPointType {
        kDataPointRMS = 0,              
        kDataPointSmallestInterval = 1, 
        kDataPointCentralInterval = 2   
    };

    class ToyDataInterface
    {
    public:
        virtual void operator()(const std::vector<double>& expectation, const std::vector<unsigned>& toyData) = 0;

        virtual ~ToyDataInterface()
        { }
    };

    /* @{ */

    BCEfficiencyFitter(const TH1& trials, const TH1& successes, const TF1& func, const std::string& name = "efficiency_fitter_model");

    virtual ~BCEfficiencyFitter();

    /* @} */
    /* @{ */

    TH1& GetTrials()
    { return fTrials; };

    TH1& GetSuccesses()
    { return fSuccesses; };

    bool GetUncertainties(int n, int k, double p, double& xexp, double& xmin, double& xmax);

    /* @} */
    /* @{ */

    void SetDataPointType(DataPointType type)
    { fDataPointType = type; }

    /* @} */
    /* @{ */

    //      virtual double LogAPrioriProbability(const std::vector<double> & parameters);

    virtual double LogLikelihood(const std::vector<double>& parameters);

    virtual void Fit();

    virtual void DrawData(bool flaglegend = false);

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

    double CalculatePValueFast(const std::vector<double>& par, BCEfficiencyFitter::ToyDataInterface* callback, unsigned nIterations = 100000);

    double CalculatePValueFast(const std::vector<double>& par, unsigned nIterations = 100000);

    /* @} */

private:
    TH1D fTrials;

    TH1D fSuccesses;

    TH1D fHistogramBinomial;

    DataPointType fDataPointType;
};

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

#endif