// $Id$
//-----------------------------------------------------------------------
//       The GSI Online Offline Object Oriented (Go4) Project
//         Experiment Data Processing at EE department, GSI
//-----------------------------------------------------------------------
// Copyright (C) 2000- GSI Helmholtzzentrum fuer Schwerionenforschung GmbH
//                     Planckstr. 1, 64291 Darmstadt, Germany
// Contact:            http://go4.gsi.de
//-----------------------------------------------------------------------
// This software can be used under the license agreements as stated
// in Go4License.txt file which is part of the distribution.
//-----------------------------------------------------------------------

#include "TGo4FitDataHistogram.h"

#include <iostream>

#include "TH1.h"

TGo4FitDataHistogram::TGo4FitDataHistogram() : TGo4FitData(), fxHistogram(this, TH1::Class()) {}

TGo4FitDataHistogram::TGo4FitDataHistogram(const char *iName, TH1 *iHistogram, Bool_t iHistogramOwned, Bool_t AddAmpl)
   : TGo4FitData(iName, "TH1 and derived classes", dtHistogram, AddAmpl),
     fxHistogram("Histogram", "Histogram object of TH1 class ", this, TH1::Class(), kTRUE, iHistogram, iHistogramOwned)
{
}

TGo4FitDataHistogram::~TGo4FitDataHistogram() {}

void TGo4FitDataHistogram::SetHistogram(TH1 *iHistogram, Bool_t iHistogramOwned)
{
   fxHistogram.ClearObject();
   fxHistogram.SetObject(iHistogram, iHistogramOwned);
}

std::unique_ptr<TGo4FitDataIter> TGo4FitDataHistogram::MakeIter()
{
   return std::make_unique<TGo4FitDataHistogramIter>(this);
}

void TGo4FitDataHistogram::FillSlotList(TSeqCollection *list)
{
   TGo4FitData::FillSlotList(list);
   list->Add(&fxHistogram);
}

void TGo4FitDataHistogram::Print(Option_t *option) const
{
   TGo4FitData::Print(option);
   std::cout << "   ";
   fxHistogram.Print(option);
}

// *********************************************************************************

TGo4FitDataHistogramIter::TGo4FitDataHistogramIter()
   : TGo4FitDataIter(), fxData(nullptr), fxLimits(), fxOwnScales(), fxOwnWidths()
{
}

TGo4FitDataHistogramIter::TGo4FitDataHistogramIter(TGo4FitDataHistogram *data)
   : TGo4FitDataIter(), fxData(data), fxLimits(), fxOwnScales(), fxOwnWidths()
{
}

TGo4FitDataHistogramIter::~TGo4FitDataHistogramIter() {}

Bool_t TGo4FitDataHistogramIter::StartReset()
{
   if (!fxData)
      return kFALSE;
   TH1 *histo = fxData->GetHistogram();
   if (!histo)
      return kFALSE;

   Int_t NumDimen = histo->GetDimension();
   fxLimits.Set(NumDimen);
   fxLimits[0] = histo->GetNbinsX();
   if (NumDimen > 1)
      fxLimits[1] = histo->GetNbinsY();
   if (NumDimen > 2)
      fxLimits[2] = histo->GetNbinsZ();

   fxOwnScales.Set(NumDimen);
   fxOwnScales.Reset(0.);
   fxOwnWidths.Set(NumDimen);
   fxOwnWidths.Reset(0.);

   return ReserveArrays(NumDimen, NumDimen, kTRUE);
}

Bool_t TGo4FitDataHistogramIter::ReadCurrentPoint()
{
   if (!fxData)
      return kFALSE;
   TH1 *histo = fxData->GetHistogram();
   if (!histo)
      return kFALSE;

   switch (histo->GetDimension()) {
   case 1: fdValue = histo->GetBinContent(fxIndexes[0] + 1); break;
   case 2: fdValue = histo->GetBinContent(fxIndexes[0] + 1, fxIndexes[1] + 1); break;
   case 3: fdValue = histo->GetBinContent(fxIndexes[0] + 1, fxIndexes[1] + 1, fxIndexes[2] + 1); break;
   default: fdValue = 0;
   }

   if (!GetDeviation())
      fdStandardDeviation = fdValue;

   for (Int_t n = 0; n < fxOwnScales.GetSize(); n++)
      switch (n) {
      case 0: fxOwnScales[0] = histo->GetXaxis()->GetBinCenter(fxIndexes[0] + 1); break;
      case 1: fxOwnScales[1] = histo->GetYaxis()->GetBinCenter(fxIndexes[1] + 1); break;
      case 2: fxOwnScales[2] = histo->GetZaxis()->GetBinCenter(fxIndexes[2] + 1); break;
      }

   for (Int_t n = 0; n < fxOwnWidths.GetSize(); n++)
      switch (n) {
      case 0: fxOwnWidths[0] = histo->GetXaxis()->GetBinWidth(fxIndexes[0] + 1); break;
      case 1: fxOwnWidths[1] = histo->GetYaxis()->GetBinWidth(fxIndexes[1] + 1); break;
      case 2: fxOwnWidths[2] = histo->GetZaxis()->GetBinWidth(fxIndexes[2] + 1); break;
      }

   return ProduceScales(fxIndexes.GetArray(), fxOwnScales.GetArray(), fxOwnWidths.GetArray());
}

Bool_t TGo4FitDataHistogramIter::ShiftToNextPoint()
{
   return NextIndex(fxIndexes, fxLimits);
}