KVIsoscaling.h

#ifndef __KVISOSCALING_H
#define __KVISOSCALING_H
 
#include <vector>
#include <utility>
#include <unordered_map>
#include <map>
#include <string>
 
#include "KVNumberList.h"
#include "KVList.h"
#include "KVHashList.h"
 
#include "TMultiGraph.h"
#include "TGraphErrors.h"
 
/*** Yields struct and map ***/
struct IsotopeYield_t
{
    Float_t yield;
    Float_t yield_err;
};
using Element_t    = std::map<Int_t, IsotopeYield_t>; // Type alias with "using" keyword (similar to typdef)
using YieldData_t  = std::map<Int_t, Element_t>;
using YieldGraph_t = std::map<Int_t, TGraphErrors*>;
using YieldGauss_t = std::map<Int_t, TF1*>;
 
struct System_t {
    YieldData_t yields;    // all yields
    YieldGraph_t graphs;   // all yields in the form of a TGraphErrors
    YieldGraph_t lngraphs; // all Ln(yields) in the form of a TGraphErrors
    YieldGauss_t gauss;    // all gaussian fits applied to the TGraphErrors
    std::string file_path; // input file path
 
};
 
/*** Isoscaling struct and map ***/
using IsoGraph_t          = std::map<Int_t, TGraphErrors*>; // isoscaling lnR21 graphs
using IsoFit_t            = std::map<Int_t, TF1*>; // isoscaling lnR21 fits
using IsoGaussRatio       = std::map<Int_t, TF1*>; // ratio of the gaussian fits
 
struct Isoscaling_t {
    IsoGraph_t graphs;
    IsoFit_t fits;
    IsoGaussRatio gauss_ratios;
    IsoGaussRatio gauss_ratiosln;
    TGraphErrors* gr_alpha;
    TGraphErrors* gr_delta;
    TGraphErrors* gr_alpha_delta;
    TGraphErrors* gr_csymT;
};
 
class KVIsoscaling : public TObject {
public:
 
   KVIsoscaling()
   {
      ftol_  = 2.5;
      fdebug_= kFALSE;
   }
 
   void ReadYieldsFile(const std::string& system_name, const Char_t* file_path);
 
   void TestGaussianApprox(const std::string& system_name1, const std::string& system_name2, Int_t zz, Double_t tol); //use to test the rms tolerance for the given system combination and charge, return reduced chisquare
   void BuildLnR21vsNPlots(const std::string& system_name1, const std::string& system_name2);
 
   void FitLnR21vsNPlots(const std::string& system_name1, const std::string& system_name2, Option_t* option = "MNVR", Option_t* gooption = "goff");
 
   void DrawAlphavsNFits(const std::string& system_name1, const std::string& system_name2);
   void DrawLnR21vsNFits(const std::string& system_name1, const std::string& system_name2);
   void BuildIsoscalingPlots(const std::string& system_name1, const std::string& system_name2, Int_t mcolor, Int_t mstyle, Bool_t draw = kFALSE);
   void CreateCsymOverTMultiGraph(TMultiGraph* mgr);
   void SaveResultsROOT(const Char_t* file_name = "./isoscaling_output_file.root");
   void SaveResultsASCII(const Char_t* file_name = "./isoscaling_output_file.txt");
 
   Bool_t  GetAMean(const std::string& system_name, Int_t zz, Float_t& meanA, Float_t& meanA_err);
   Bool_t  GetAlpha(const std::string& system_name1, const std::string& system_name2, Int_t zz, Float_t& alpha, Float_t& alpha_err);
   Bool_t  GetDeltaZA2(const std::string& system_name1, const std::string& system_name2, Int_t zz, Float_t& denum, Float_t& denum_err, Bool_t debug);
   Bool_t  GetCsymOverT(const std::string& system_name1, const std::string& system_name2, Int_t zz, Float_t& csymT, Float_t& csymT_err, Bool_t debug);
   Int_t   FindZFromAmean(const std::string& system_name, Int_t aa);
   Double_t GetRMSTolerance()
   {
      return ftol_;
   }
 
   KVNumberList GetZNumberList(const std::string& system_name);
   KVNumberList GetANumberList(const std::string& system_name, Int_t zz);
   KVNumberList GetSharedZNumberList(const std::string& system_name1, const std::string& system_name2);
   KVNumberList GetSharedANumberList(const std::string& system_name1, const std::string& system_name2, Int_t zz);
 
   void PrintYieldsList();
   void PrintSystemsList();
 
   inline void SetVerbose(Bool_t debug)
   {
      fdebug_ = debug;
   }
 
   inline void SetRMSTolerance(Double_t tol)
   {
      ftol_ = tol;
   }
 
protected:
    std::unordered_map<std::string, System_t> fSystemList_;
    std::unordered_map<std::string, Isoscaling_t> fIsoscalingList_;
 
    Bool_t fdebug_;
    Double_t ftol_;
 
   void BuildGaussianPlots(const std::string& system_name);
 
   ClassDef(KVIsoscaling, 2) //KVClass
};
 
#endif