//   Read the documentation to learn more about C++ code generator
//   versioning.
//	  %X% %Q% %Z% %W%

// DataContainer
#include <XSModel/GlobalContainer/DataContainer.h>
// ModelContainer
#include <XSModel/GlobalContainer/ModelContainer.h>
// PlotChiSq
#include <XSPlot/Commands/PlotChiSq.h>
#include <XSFit/Fit/Fit.h>
#include <XSFit/Fit/StatMethod.h>
#include <XSPlot/Plot/PlotGroup.h>
#include <XSContainer.h>
#include <XSstreams.h>


// Class PlotChiSq 

PlotChiSq::PlotChiSq (Plot* plot)
   :PlotCommand(plot,"chisq")
{
  modelRequired(Plot::FOLDED);
}


void PlotChiSq::manipulate ()
{
  PlotGroupContainer::iterator g    = plot()->groups().begin();      
  PlotGroupContainer::iterator gEnd = plot()->groups().end();   

  StatMethod* stat = XSContainer::fit->statMethod();

  while ( g != gEnd )
  {
     // first bin.
     int arraySize = (*g)->n;
     const std::vector<Real>& X      = (*g)->xAxis.data;
     std::vector<Real>& yData  = (*g)->yData.data;
     const std::vector<Real>& yError = (*g)->yData.errors[0];
     const std::vector<Real>& model  = (*g)->model[0].data;

     (*g)->yData.lineStep = true;
     for (int j = 0; j < arraySize; ++j)
     {
        if ( X[j] != Plot::NODATA())
        {
           Real obs (yData[j]);
           Real mod (model[j]);
           Real err (yError[j]);
           Real areaTime = (*g)->saveData[j];
           yData[j]  = stat->plotChi(obs,mod,err,areaTime,Plot::NODATA() );
        }
     }

     ++g;
  }


}

void PlotChiSq::primaryGraph (const string& value)
{

  Plot* pPlot = plot();
  StatMethod* stat = XSContainer::fit->statMethod();
  if (stat->fullName() == string("C-Statistic") ||
      stat->fullName() == string("L-statistic"))
  {
     tcout <<"Note: When using cstat or lstat, \"plot chisq\" is only valid\n"
           <<"      for data sets with no background files." << std::endl;
  }

  pPlot->makeBinnedPlotArrays();

  pPlot->xAxisLabel();
  pPlot->setSelectedLabel(1,"ys",stat->scriptName());
  pPlot->setSelectedLabel(2,"ts",stat->fullName());


  pPlot->xLine(std::pair<bool,Real>(true, 0.0));  
  pPlot->rangeDefaultFlags(12);


  manipulate();
  fillDataAttributes(0,pPlot->modelStyle(),2);
  turnModelsOff();


  pPlot->bundlePlotVectors();
  pPlot->prepareTheGraph();
  pPlot->makeTheGraph();   
}

// Additional Declarations