KVEvaporationSpectrum.h

#pragma once
#include "TF1.h"
#include "TMultiGraph.h"
 
class KVEvaporationSpectrum : public TF1
{
   struct spectrum_calculator
   {
      bool con_sig;
      double norm_const;
      spectrum_calculator(bool sig_constrain = true)
         : con_sig{sig_constrain}
      {
         if(con_sig)
            norm_const = TMath::E()/TMath::Sqrt(TMath::TwoPi());
         else
            norm_const = TMath::Sqrt(TMath::TwoPi())/TMath::E();
      }
      double operator()(double*,double*) const;
      bool is_sigma_constrained() const { return con_sig; }
   };
   struct multi_comp_spectrum_calculator
   {
      int Nc;
      spectrum_calculator SC;
      multi_comp_spectrum_calculator(int ncomp, bool sig_constrain = true)
         : Nc{ncomp}, SC{sig_constrain}
      {}
      double operator()(double*,double*) const;
      bool is_sigma_constrained() const { return SC.is_sigma_constrained(); }
   };
 
public:
   multi_comp_spectrum_calculator spe_calc;
   KVEvaporationSpectrum(int ncomponents=1, bool constrain_sigma = true)
      : KVEvaporationSpectrum("KVEvaporationSpectrum", ncomponents, constrain_sigma)
   {}
   KVEvaporationSpectrum(const TString& name, int ncomponents=1, bool constrain_sigma = true);
   void SetInitialParameters(const TH1*);
   auto GetBarrier(int comp = 0) const
   {
      return GetParameter(get_npar()*comp+1);
   }
   auto GetBarrierUncertainty(int comp = 0) const
   {
      return GetParError(get_npar()*comp+1);
   }
   auto GetTemperature(int comp = 0) const
   {
      return GetParameter(get_npar()*comp+2);
   }
   auto GetTemperatureUncertainty(int comp = 0) const
   {
      return GetParError(get_npar()*comp+2);
   }
 
   Int_t GetNComponents() const
   {
      return GetNpar()/get_npar();
   }
   TMultiGraph* GetComponents(const std::vector<Color_t>& line_colors={kGray+3, kGray+2, kGray+1, kGray}) const;
 
   ClassDefOverride(KVEvaporationSpectrum,1)
 
   private:
   int get_npar() const
   {
      return spe_calc.is_sigma_constrained() ? 3 : 4;
   }
};