#ifndef PLFIT_H
#define PLFIT_H

#include <iostream>
#include <string>
#include <vector>
#include <TH1D.h>
#include <TF1.h>
#include <TRandom.h>
#include <TCanvas.h>
#include "PLUtil.h"

using namespace std;

class PLFit {

  public: 

  PLFit(double xMin, double xMax, int nBin, vector<double> truePdfPar,
        double nuTot, vector<double> trueScalePar, int fitOption);

    TH1D*    dataHist(){ return m_dataHist; }
    TH1D*    trueHist(){ return m_trueHist; }
    TH1D*    modelHist(){ return m_modelHist; }
    double   xMin(){ return m_xMin; }
    double   xMax(){ return m_xMax; }
    int      nBin(){ return m_nBin; }
    double   nuTot(){ return m_nuTot; }
    double   dx(){ return (m_xMax - m_xMin)/(double)m_nBin; }
    double   modelPdf (double, vector<double>);
    double   scaleFactor (double, vector<double>);
    vector<double> truePdfPar(){ return m_truePdfPar; }
    vector<double> trueScalePar(){ return m_trueScalePar; }
    void     setModelHist(vector<double> pdfPar, vector<double> scalePar);
    void     integrateModel(int a, int b, double& y, double& sigma,
		      vector<double> thetaHat, vector<vector<double> > V);

    void     checkHist(TH1D* h);
    void     plotSetup(TCanvas* canvas);
    // void     plotFit(double par[], TCanvas* canvas, bool logY);
    void     printCanvas(TCanvas* canvas, string fileName);
    int      fitOption(){ return m_fitOption; }
    void     setFitOption(int fitOption) { m_fitOption = fitOption; }
    void     generateData(TRandom* ran);

 private:

    TH1D*    m_dataHist;
    TH1D*    m_trueHist;
    TH1D*    m_trueBernPar;
    TH1D*    m_modelHist;
    int      m_fitOption;
    double   m_xMin;
    double   m_xMax;
    int      m_nBin;
    double   m_nuTot;
    vector<double> m_truePdfPar;
    vector<double> m_trueScalePar;

};

// TMinuit needs non-member fcn

void fcn(int& n, double* d, double& f, double par[], int flag);

#endif