BCHistogramFitter.h

Go to the documentation of this file.

#ifndef __BCMODELHISTOGRAMFITTER__H
#define __BCMODELHISTOGRAMFITTER__H

/*
 * Copyright (C) 2008-2012, Daniel Kollar and Kevin Kroeninger.
 * All rights reserved.
 *
 * For the licensing terms see doc/COPYING.
 */

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

#include <vector>

#include "../../BAT/BCModel.h"

class BCHistogramFitterToyDataInterface;

// ROOT classes
class TH1D;
class TF1;

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

class BCHistogramFitter : public BCModel
{
   public:

      /* @{ */

      BCHistogramFitter();

      BCHistogramFitter(const char * name);

      BCHistogramFitter(TH1D * hist, TF1 * func);

      BCHistogramFitter(const char * name, TH1D * hist, TF1 * func);

      ~BCHistogramFitter();

      /* @} */

      /* @{ */

      TH1D * GetHistogram()
         { return fHistogram; };

      TH1D * GetHistogramExpected()
         { return fHistogramExpected; };


      TF1 * GetFitFunction()
         { return fFitFunction; };

      TGraph * GetErrorBand()
         { return fErrorBand; };

      TGraph * GetGraphFitFunction()
         { return fGraphFitFunction; };

      /* @} */

      /* @{ */

      int SetHistogram(TH1D * hist);

      int SetHistogramExpected(const std::vector<double>& parameters);

      int SetFitFunction(TF1 * func);

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

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

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

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

//      void PrintHistogram(const char * options = "", const char * filename = "");

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

      int Fit();

      int Fit(TH1D * hist, TF1 * func);

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

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

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

      int CalculatePValueLikelihood(const std::vector<double> &par, double &pvalue);

      int CalculatePValueLeastSquares(const std::vector<double> &par, double &pvalue, bool weightExpect=true);

      int CalculatePValueKolmogorov(const std::vector<double> &par, double &pvalue);


      double CDF(const std::vector<double>& parameters, int index, bool lower=false);

      /* @} */

   private:

      TH1D * fHistogram;

      TF1 * fFitFunction;

      bool fFlagIntegration;

      TGraph * fErrorBand;

      TGraph * fGraphFitFunction;

      TH1D * fHistogramExpected;

};

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

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

   virtual ~BCHistogramFitterToyDataInterface(){}
};

#endif