CorrFit3DCFMultiDim.cxx

/*
 * CorrFit3DCFMultiDim.cxx
 *
 *  Created on: 18 maj 2020
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
 
#include "CorrFit3DCFMultiDim.h"
 
#include "CorrFit.h"
#include "CorrFitFunctor.h"
#include "CorrFitInfo.h"
#include "Cout.h"
#include "Femto3DCF.h"
#include "Minimizer.h"
 
#include <Rtypes.h>
#include <RtypesCore.h>
#include <TH1.h>
#include <TString.h>
#include <iostream>
#include <string>
 
//#define _MULTIDIM_DEBUG_
 
 
namespace Hal {
  void CorrFit3DCFMultiDim::RecalculateFunction() const {
    Double_t pars[3];
    pars[0] = fTempParamsEval[RoutID()];
    pars[1] = fTempParamsEval[RsideID()];
    pars[2] = fTempParamsEval[RlongID()];
    fData   = fFunctor->GetData(pars);
  }
 
  Double_t CorrFit3DCFMultiDim::CalculateCF(const Double_t* /*x*/, const Double_t* params) const {
    Double_t bin = (fBinX - 1) * fFunctorXbins * fFunctorYbins + (fBinY - 1) * fFunctorYbins + fBinZ - 1;
    Double_t num = fData->Get(bin) - 1.0;
    if (TMath::IsNaN(num)) num = 0;
    return params[NormID()] * (1.0 + num * params[LambdaID()]);
  }
 
  CorrFit3DCFMultiDim::CorrFit3DCFMultiDim(Int_t params) : CorrFit3DCF(params) { fMinAlgo = EMinAlgo::kHalScan; }
 
  void CorrFit3DCFMultiDim::SetFunctor(CorrFitFunctor* functor) {
    fFunctor = functor;
    SetAutoLimits();
  }
 
  void CorrFit3DCFMultiDim::SetFunctorFromMap(TString filename) {
    fFunctor = new CorrFitFunctor(filename);
    SetAutoLimits();
  }
 
  void CorrFit3DCFMultiDim::SetAutoLimits() {
    TString name;
    Double_t min, max;
    Int_t points;
    // get out
#ifdef _MULTIDIM_DEBUG_
    std::cout << "SET AUTO LIMITS" << std::endl;
#endif
    for (int i = 0; i < fFunctor->GetNParams(); i++) {
      name = fFunctor->GetParameterName(i);
      fFunctor->GetParameterConfig(name, min, max, points);
      fParameters[i].SetIsDiscrete(kTRUE);
      fParameters[i].SetMapRange(min, max, points);
      SetParLimits(i, min, max);
    }
#ifdef _MULTIDIM_DEBUG_
    std::cout << "AUTO DONE" << std::endl;
#endif
  }
 
  CorrFit3DCFMultiDim::~CorrFit3DCFMultiDim() {
    if (fFunctor) delete fFunctor;
  }
 
  void CorrFit3DCFMultiDim::SetMinimizer(EMinAlgo alg) {
    if (alg == kHalScan || alg == kHalAnt) {
      if (alg == kHalAnt) { std::cout << "Smart scan" << std::endl; }
      CorrFit3DCF::SetMinimizer(alg);
    } else {
      Cout::Text("This class supports correctly only Minimizer", "L", kYellow);
    }
  }
 
  void CorrFit3DCFMultiDim::PrepareHalMinimizer() const {
    std::cout << "PREPARING MINIMIZER" << GetParametersNo() << std::endl;
    Minimizer* min = Minimizer::Instance();
    for (int i = 0; i < GetParametersNo(); i++) {
      std::string par_name = GetParameterName(i).Data();
      if (!IsParFixed(i)) {  // fixed parameters will be configured later
        TString name = par_name;
        Double_t Min, Max;
        Int_t points;
        if (fFunctor->GetParameterConfig(name, Min, Max, points)) {
          Double_t step_size = (Max - Min) / (Double_t(points - 1));
          if (points == 0) step_size = 0;
#ifdef _MULTIDIM_DEBUG_
          std::cout << "SET MAP STEPS " << par_name << "\t" << Min << " " << Max << " " << points << " "
                    << " " << step_size << std::endl;
#endif
          std::string Name = name.Data();
          min->SetLimitedVariable(i, Name, 0.5 * (Min + Max), step_size, Min, Max);
        } else {
#ifdef _MULTIDIM_DEBUG_
          std::cout << "SET FREE STEPS " << par_name << "\t" << GetParMin(i) << " " << GetParMax(i) << " " << points << " "
                    << " " << 0.01 << std::endl;
#endif
          min->SetLimitedVariable(i, par_name, 0.5 * (GetParMin(i) + GetParMax(i)), 0.01, GetParMin(i), GetParMax(i));
        }
        min->SetVariableLimits(i, GetParMin(i), GetParMax(i));
      } else {
#ifdef _MULTIDIM_DEBUG_
        std::cout << "SET FIXED " << par_name << "\t" << GetParMin(i) << " " << GetParMax(i) << std::endl;
#endif
        min->SetLimitedVariable(i, par_name, GetParMin(i), 1, GetParMin(i), GetParMin(i));
        min->SetFixedVariable(i, par_name, GetParMin(i));
      }
    }
    std::cout << "PREPARING MINIMIZER DONE !" << std::endl;
  }
 
  void CorrFit3DCFMultiDim::Check() {
    CorrFit3DCF::Check();
    CorrFitInfo* info = fFunctor->GetInfo();
    Femto3DCF* cf     = dynamic_cast<Femto3DCF*>(info->GetCf());
    if (cf == nullptr) { Cout::PrintInfo("Cannot find 3DCF in CorrFitInfo !", EInfo::kCriticalError); }
    Femto3DCF* thisCf = (Femto3DCF*) fCF;
    if (thisCf->GetFrame() != cf->GetFrame()) { Cout::PrintInfo("CorrFitInfo incompatible frames !", EInfo::kCriticalError); }
    TH3* testThis = (TH3*) thisCf->GetNum();
    TH3* testMap  = (TH3*) cf->GetNum();
    if (!Hal::Std::AreSimilar(testThis, testMap, kFALSE)) {
      Cout::PrintInfo("CorrFitInfo incompatible histograms in CorrFitInfo and Femto3DCF !", EInfo::kError);
    }
    fFunctorXbins = testThis->GetXaxis()->GetNbins();
    fFunctorYbins = testThis->GetYaxis()->GetNbins();
  }
}  // namespace Hal