BCFitter.h

#ifndef __BCFITTER__H
#define __BCFITTER__H

/*
 * Copyright (C) 2007-2014, the BAT core developer team
 * All rights reserved.
 *
 * For the licensing terms see doc/COPYING.
 * For documentation see http://mpp.mpg.de/bat
 */

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

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

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

class BCFitter : public BCModel
{
 public:

  /* @{ */

  BCFitter();

  BCFitter(const char * name);

  ~BCFitter();

  /* @} */

  /* @{ */

  TGraph * GetErrorBand()
  { return fErrorBand; };

  TH2D * GetErrorBandXY() const
  { return fErrorBandXY; }

  TH2D * GetErrorBandXY_yellow(double level=.68, int nsmooth=0) const;

  std::vector<double> GetErrorBand(double level) const;

  TGraph * GetErrorBandGraph(double level1, double level2) const;

  TGraph * GetFitFunctionGraph(const std::vector<double> &parameters);

  TGraph * GetFitFunctionGraph()
  { return GetFitFunctionGraph(GetBestFitParameters()); }

  TGraph * GetFitFunctionGraph(const std::vector<double> &parameters, double xmin, double xmax, int n=1000);

  void FixDataAxis(unsigned int index, bool fixed);

  bool GetFixedDataAxis(unsigned int index) const;

  /* @} */

  /* @{ */

  void SetFillErrorBand(bool flag = true)
  { fFlagFillErrorBand=flag; }

  void SetErrorBandHisto(TH2D * h)
  { fErrorBandXY = h; }

  void UnsetFillErrorBand()
  { fFlagFillErrorBand=false; }

  void SetFitFunctionIndexX(int index)
  { fFitFunctionIndexX = index; }

  void SetFitFunctionIndexY(int index)
  { fFitFunctionIndexY = index; }

  void SetFitFunctionIndices(int indexx, int indexy)
  { SetFitFunctionIndexX(indexx);
    SetFitFunctionIndexY(indexy); }

  void SetErrorBandContinuous(bool flag);

  virtual double FitFunction(const std::vector<double> &/*x*/, const std::vector<double> &/*parameters*/)
  { return 0; }

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

  int ReadErrorBandFromFile(const char * file);

  virtual int Fit() = 0;

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

  void MCMCIterationInterface();

  void MarginalizePreprocess();

  void MarginalizePostprocess()
  {;};

  void FillErrorBand();

  /* @} */

 private:

  TGraph * fErrorBand;

  bool fFlagFillErrorBand;

  int fFitFunctionIndexX;
  int fFitFunctionIndexY;

  bool fErrorBandContinuous;
  std::vector<double> fErrorBandX;

  unsigned fErrorBandNbinsX;

  unsigned fErrorBandNbinsY;

 protected:

  TH2D * fErrorBandXY;

};

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

#endif