// $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 "TGo4FitModelPolynom.h"

#include "TMath.h"

#include "TGo4FitParameter.h"

TGo4FitModelPolynom::TGo4FitModelPolynom() : TGo4FitModel(), fxAllOrders(nullptr)
{
}

TGo4FitModelPolynom::TGo4FitModelPolynom(const char *iName, Double_t iOrderX) :
   TGo4FitModel(iName,"polynomial function",kTRUE), fxAllOrders(nullptr)
{
     CreateOrdersPars(&iOrderX, 0, 0);
     SetBackgroundGroupIndex();
}

TGo4FitModelPolynom::TGo4FitModelPolynom(const char *iName, Double_t iOrderX, Double_t iOrderY) :
   TGo4FitModel(iName,"polynomial function",kTRUE), fxAllOrders(nullptr)
{
    Double_t arr[2] = { iOrderX, iOrderY };
    CreateOrdersPars(arr, 0, 1);
    SetBackgroundGroupIndex();
}

TGo4FitModelPolynom::TGo4FitModelPolynom(const char *iName, Double_t iOrderX, Double_t iOrderY, Double_t iOrderZ) :
   TGo4FitModel(iName,"polynomial function",kTRUE), fxAllOrders(nullptr)
{
    Double_t arr[3] = { iOrderX, iOrderY, iOrderZ };
    CreateOrdersPars(arr, 0, 2);
    SetBackgroundGroupIndex();
}

TGo4FitModelPolynom::TGo4FitModelPolynom(const char *iName, const TArrayD& iPolynomOrders) :
  TGo4FitModel(iName,"polynomial function",kTRUE), fxAllOrders(nullptr)
{
   CreateOrdersPars(iPolynomOrders.GetArray(),0,iPolynomOrders.GetSize()-1);
   SetBackgroundGroupIndex();
}

TGo4FitModelPolynom::~TGo4FitModelPolynom()
{
  if (fxAllOrders) delete fxAllOrders;
}

void TGo4FitModelPolynom::CreateOrdersPars(const Double_t *orders, Int_t startaxis, Int_t maxaxis, Int_t AtIndx)
{
   for(Int_t n=startaxis;n<=maxaxis;n++) {
      TString infostr;
      infostr.Form("Polynom order for axis %d",n);
      if (orders) NewParameter(GetOrderParName(n),infostr.Data(),orders[n],kTRUE, AtIndx);
            else  NewParameter(GetOrderParName(n),infostr.Data(),0.,kTRUE, AtIndx);
   }
}

TString TGo4FitModelPolynom::GetOrderParName(Int_t naxis) const
{
   TString res;
   res.Form("Order%d",naxis);
   return res;
}

Double_t TGo4FitModelPolynom::GetPolynomOrder(Int_t naxis)
{
   TGo4FitParameter *par = FindPar(GetOrderParName(naxis));
   return par ? par->GetValue() : 0.;
}

Int_t TGo4FitModelPolynom::GetMaxNumAxis()
{
   Int_t naxis = 0;
   while (FindPar(GetOrderParName(naxis))) 
      naxis++;
   return naxis-1;
}

Bool_t TGo4FitModelPolynom::SetMaxNumAxis(Int_t numaxis)
{
   Int_t numaxisold = GetMaxNumAxis();
   if ((numaxis < 0) || (numaxis == numaxisold))
      return kFALSE;
   if (numaxis < numaxisold)
      for (Int_t n = numaxis + 1; n <= numaxisold; n++)
         RemovePar(GetOrderParName(n));
   else {
      Int_t indx = GetParIndex(FindPar(GetOrderParName(numaxisold)));
      CreateOrdersPars(nullptr, numaxisold + 1, numaxis, indx + 1);
   }

   return kTRUE;
}

Bool_t TGo4FitModelPolynom::BeforeEval(Int_t NDimension)
{
   if (fxAllOrders)
      delete fxAllOrders;
   Par_ndim = NDimension;
   fxAllOrders = new TArrayD(Par_ndim);
   fxAllOrders->Reset(0.);
   Par_orders = fxAllOrders->GetArray();
   for (Int_t i = 0; i < Par_ndim; i++)
      Par_orders[i] = GetPolynomOrder(i);
   return kTRUE;
}

Double_t TGo4FitModelPolynom::EvalN(const Double_t *v)
{
   Double_t zn = 1.;
   for (Int_t n = 0; n < Par_ndim; n++)
      zn *= TMath::Power(v[n], Par_orders[n]);
   return zn;
}

void TGo4FitModelPolynom::AfterEval()
{
   if (fxAllOrders) {
      delete fxAllOrders;
      fxAllOrders = nullptr;
   }
}

void TGo4FitModelPolynom::Print(Option_t *option) const
{
   TGo4FitModel::Print(option);
}