BCHistogramFitter.h

Go to the documentation of this file.

#ifndef __BCMODELHISTOGRAMFITTER__H
#define __BCMODELHISTOGRAMFITTER__H

/*
 * Copyright (C) 2007-2013, 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 BCHistogramFitter : public BCFitter
{
   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, unsigned nIterations = 100000);

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

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

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

      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;
};

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

#endif