lambda_nanjing.h

Go to the documentation of this file.

//
// Created by iorio on 12/14/21.
//
 
#ifndef SEVN_LAMBDA_NANJING_H
#define SEVN_LAMBDA_NANJING_H
 
#include <vector>
#include <sevnlog.h>
#include <utilities.h>
#include <lambda_base.h>
 
class Star;
class IO;
 
class Lambda_Nanjing : public Lambda_Base {
 
public:
    Lambda_Nanjing(){};
    Lambda_Nanjing(_UNUSED const Star *s){};
    Lambda_Nanjing(_UNUSED const IO *io){};
    virtual ~Lambda_Nanjing() = default;
 
    double operator() (const Star* s) override;
 
protected:
 
 
    virtual double lambda_Giant(const Star* s);
    virtual double lambda_Cheb(const Star* s);
    virtual double lambda_AGB(const Star* s);
    virtual double lambda_pureHe(const Star* s);
 
    //Properly estimated the two lambdas, they call set_coeff** to set the coefficients of the fittin equations
    virtual double lambda_Giant(_UNUSED double Mzams, _UNUSED  double Z, _UNUSED  double Radius, _UNUSED double Menv, _UNUSED double lambda_th);
    virtual double lambda_Cheb(_UNUSED double Mzams, _UNUSED  double Z, _UNUSED  double Radius, _UNUSED double Menv, _UNUSED double lambda_th);
    virtual double lambda_AGB(_UNUSED double Mzams, _UNUSED  double Z, _UNUSED  double Radius, _UNUSED double Menv, _UNUSED double lambda_th);
    virtual double lambda_pureHe(_UNUSED double Mzams, _UNUSED  double Z, _UNUSED  double Radius, _UNUSED double Menv, _UNUSED double lambda_th);
 
    //Set the coefficient of the lambda fitting equations
    typedef unsigned int coeff_status;
    virtual coeff_status set_coeff_Giant(double Mzams, double Z, double Radius);  //From HG::CalculateLambdaNanjing in HG.cpp in COMPAS
    virtual coeff_status set_coeff_Cheb(double Mzams, double Z, double Radius);   //From CHeB::CalculateLambdaNanjing in CHeb.cpp in COMPAS
    virtual coeff_status set_coeff_AGB(double Mzams, double Z, double Radius);    //From EAGB::CalculateLambdaNanjing in EAGB.cpp in COMPAS
 
    std::vector<double> fitting_equation(double x){
        double x2 = x * x;
        double x3 = x2 * x;
        double x4 = x2 * x2;
        double x5 = x3 * x2;
 
        double y1 = coeff_lambdab[0] + (coeff_lambdab[1] * x) + (coeff_lambdab[2] * x2) + (coeff_lambdab[3] * x3) + (coeff_lambdab[4] * x4) + (coeff_lambdab[5] * x5);
        double y2 = coeff_lambdag[0] + (coeff_lambdag[1] * x) + (coeff_lambdag[2] * x2) + (coeff_lambdag[3] * x3) + (coeff_lambdag[4] * x4) + (coeff_lambdag[5] * x5);
 
        std::vector<double> y_vec = {y1,y2};
 
        return y_vec;
    }
 
    sevnstd::SevnLogging svlog;
    std::vector<double> coeff_lambdab; 
    std::vector<double> coeff_lambdag; 
    std::vector<double> maxBG = {};  
    std::vector<double> lambdaBG = {}; 
    inline int reset(){
         coeff_lambdab.clear();
         coeff_lambdag.clear();
         maxBG.clear();
         lambdaBG.clear();
         return EXIT_SUCCESS;
     }
 
private:
 
    const double ZpopI = 0.0085; 
    constexpr static unsigned int COEF_SET = 1;
    constexpr static unsigned int COEF_NOT_SET = 0;
 
    inline double lambda_estimate(double lambda_th){
        return lambda_th*lambdaBG[0] + (1-lambda_th)*lambdaBG[1];
    }
};
 
 
class Lambda_Nanjing_interpolator : public Lambda_Nanjing{
 
public:
    Lambda_Nanjing_interpolator(){};
    Lambda_Nanjing_interpolator(_UNUSED const Star *s){};
    Lambda_Nanjing_interpolator(_UNUSED const IO *io){};
 
    double operator() (const Star* s) override;
 
protected:
 
    double operator() (unsigned int Phase, double Mzams, double Z, double Radius, double Menv, double lambda_th);
 
 
    static std::vector<double> Mzams_list;  
    double Mzams_cache{0}, Z_cache{0}; 
    void find_interpolators(double Mzams);
 
private:
 
    double Mzams_interpolators [2] = {0.,0.}; 
    double wM [2] = {0., 0.}; 
};
 
#endif //SEVN_LAMBDA_NANJING_H