BCHistogramFitter.h

Go to the documentation of this file.

#ifndef __BCMODELHISTOGRAMFITTER__H
#define __BCMODELHISTOGRAMFITTER__H

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

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

#include <vector>

#include "BAT/BCModel.h"

// ROOT classes
class TH1D;
class TF1;

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

class BCHistogramFitter : public BCModel
{
   public:

      /* @{ */

      BCHistogramFitter();

      BCHistogramFitter(TH1D * hist, TF1 * func);

      ~BCHistogramFitter();

      /* @} */

      /* @{ */

      TH1D * GetHistogram()
         { return fHistogram; };

      TF1 * GetFitFunction()
         { return fFitFunction; };

      TGraph * GetErrorBand()
         { return fErrorBand; };

      TGraph * GetGraphFitFunction()
         { return fGraphFitFunction; };

      /* @} */

      /* @{ */

      int SetHistogram(TH1D * hist);

      int SetFitFunction(TF1 * func);

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

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

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

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

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

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

      int Fit()
         { return this -> Fit(fHistogram, fFitFunction); };

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

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

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

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

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

      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

---