BCGraphFitter.h

Go to the documentation of this file.

#ifndef __BCMODELGRAPHFITTER__H
#define __BCMODELGRAPHFITTER__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 TGraphErrors;
class TF1;

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

class BCGraphFitter : public BCModel
{
   public:

      /* @{ */

      BCGraphFitter();

      BCGraphFitter(const char * name);

      BCGraphFitter(TGraphErrors * graph, TF1 * func);

      BCGraphFitter(const char * name, TGraphErrors * graph, TF1 * func);

      ~BCGraphFitter();

      /* @} */

      /* @{ */

      TGraphErrors * GetGraph()
         { return fGraph; };

      TF1 * GetFitFunction()
         { return fFitFunction; };

      TGraph * GetErrorBand()
         { return fErrorBand; };

      TGraph * GetGraphFitFunction()
         { return fGraphFitFunction; };

      /* @} */

      /* @{ */

      int SetGraph(TGraphErrors * graph);

      int SetFitFunction(TF1 * func);

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

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

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

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

      int Fit();

      int Fit(TGraphErrors * graph, TF1 * func);

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

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

      /* @} */

   private:

      TGraphErrors * fGraph;

      TF1 * fFitFunction;

      TGraph * fErrorBand;

      TGraph * fGraphFitFunction;

};

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

#endif