BCEfficiencyFitter.h

#ifndef __BCEFFICIENCYFITTER__H
#define __BCEFFICIENCYFITTER__H

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

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

#include <vector>

#include "BCFitter.h"

// ROOT classes
class TH1D;
class TF1;
class TGraphAsymmErrors;

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

class BCEfficiencyFitter : public BCFitter
{
   public:

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

         virtual ~ToyDataInterface(){}
      };

      /* @{ */

      BCEfficiencyFitter();

      BCEfficiencyFitter(const char * name);

      BCEfficiencyFitter(TH1D * hist1, TH1D * hist2, TF1 * func);

      BCEfficiencyFitter(const char * name, TH1D * hist1, TH1D * hist2, TF1 * func);

      ~BCEfficiencyFitter();

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

      TH1D * GetHistogram1()
         { return fHistogram1; };

      TH1D * GetHistogram2()
         { return fHistogram2; };

      TGraphAsymmErrors * GetHistogramRatio()
         { return fHistogramRatio; };

      TF1 * GetFitFunction()
         { return fFitFunction; };

      TGraph * GetErrorBand()
         { return fErrorBand; };

      TGraph * GetGraphFitFunction()
         { return fGraphFitFunction; };

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

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

      int SetHistograms(TH1D * hist1, TH1D * hist2);

      int SetFitFunction(TF1 * func);

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

      void SetDataPointType(int type);

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

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

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

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

      int Fit();

      int Fit(TH1D * hist1, TH1D * hist2, TF1 * func);

      void DrawFit(const char * options = "", bool flaglegend = false);

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

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

      /* @} */

   private:

      TH1D * fHistogram1;

      TH1D * fHistogram2;

      TGraphAsymmErrors * fHistogramRatio;

      TF1 * fFitFunction;

      bool fFlagIntegration;

      TGraph * fErrorBand;

      TGraph * fGraphFitFunction;

      TH1D * fHistogramBinomial;

      unsigned int fDataPointType;

};

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

#endif