BCGraphFitter.h

Go to the documentation of this file.

#ifndef __BCMODELGRAPHFITTER__H
#define __BCMODELGRAPHFITTER__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"

class TGraphErrors;
class TF1;

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

class BCGraphFitter : public BCModel
{
   public:

      /* @{ */

      BCGraphFitter();

      BCGraphFitter(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(std::vector <double> parameters);

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

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

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

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

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

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

      /* @} */

   private:

      TGraphErrors * fGraph;

      TF1 * fFitFunction;

      TGraph * fErrorBand;

      TGraph * fGraphFitFunction;

};

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

#endif

---