mirror of
https://github.com/cosmo-sims/MUSIC.git
synced 2024-09-19 17:03:46 +02:00
added support for velocity transfer functions added with the latest (early 2015) CAMB release
This commit is contained in:
Oliver Hahn
parent
86cd42f41f
commit
2b9bc68fac
1 changed files with 266 additions and 189 deletions
|
|
@ -1,8 +1,7 @@
|
|||
/*
|
||||
|
||||
transfer_camb.cc - This file is part of MUSIC -
|
||||
a code to generate multi-scale initial conditions
|
||||
for cosmological simulations
|
||||
a code to generate multi-scale initial conditions for cosmological simulations
|
||||
|
||||
Copyright (C) 2010 Oliver Hahn
|
||||
|
||||
|
|
@ -10,207 +9,285 @@
|
|||
|
||||
#include "transfer_function.hh"
|
||||
|
||||
const double tiny = 1e-30;
|
||||
|
||||
class transfer_CAMB_plugin : public transfer_function_plugin
|
||||
{
|
||||
|
||||
|
||||
private:
|
||||
std::string m_filename_Pk, m_filename_Tk;
|
||||
std::vector<double> m_tab_k, m_tab_Tk_tot, m_tab_Tk_cdm, m_tab_Tk_baryon;
|
||||
gsl_interp_accel *acc_tot, *acc_cdm, *acc_baryon;
|
||||
gsl_spline *spline_tot, *spline_cdm, *spline_baryon;
|
||||
|
||||
double m_kmin, m_kmax;
|
||||
unsigned m_nlines;
|
||||
|
||||
void read_table( void ){
|
||||
std::string m_filename_Pk, m_filename_Tk;
|
||||
std::vector<double> m_tab_k, m_tab_Tk_tot, m_tab_Tk_cdm, m_tab_Tk_baryon;
|
||||
std::vector<double> m_tab_Tvk_cdm, m_tab_Tvk_baryon; //>[150609SH: add]
|
||||
gsl_interp_accel *acc_tot, *acc_cdm, *acc_baryon;
|
||||
gsl_interp_accel *acc_vcdm, *acc_vbaryon; //>[150609SH: add]
|
||||
gsl_spline *spline_tot, *spline_cdm, *spline_baryon;
|
||||
gsl_spline *spline_vcdm, *spline_vbaryon; //>[150609SH: add]
|
||||
|
||||
double m_kmin, m_kmax;
|
||||
unsigned m_nlines;
|
||||
|
||||
bool m_linbaryoninterp;
|
||||
|
||||
double m_velunits;
|
||||
|
||||
void read_table( void ){
|
||||
#ifdef WITH_MPI
|
||||
if( MPI::COMM_WORLD.Get_rank() == 0 ){
|
||||
#endif
|
||||
std::cerr
|
||||
<< " - reading tabulated transfer function data from file \n"
|
||||
<< " \'" << m_filename_Tk << "\'\n";
|
||||
|
||||
std::string line;
|
||||
std::ifstream ifs( m_filename_Tk.c_str() );
|
||||
|
||||
if(! ifs.good() )
|
||||
throw std::runtime_error("Could not find transfer function file \'"+m_filename_Tk+"\'");
|
||||
|
||||
m_tab_k.clear();
|
||||
m_tab_Tk_tot.clear();
|
||||
m_tab_Tk_cdm.clear();
|
||||
m_tab_Tk_baryon.clear();
|
||||
|
||||
m_kmin = 1e30;
|
||||
m_kmax = -1e30;
|
||||
m_nlines = 0;
|
||||
|
||||
while( !ifs.eof() ){
|
||||
getline(ifs,line);
|
||||
|
||||
if(ifs.eof()) break;
|
||||
|
||||
std::stringstream ss(line);
|
||||
|
||||
double k, Tkc, Tkb, Tkg, Tkr, Tknu, Tktot;
|
||||
ss >> k;
|
||||
ss >> Tkc;
|
||||
ss >> Tkb;
|
||||
ss >> Tkg;
|
||||
ss >> Tkr;
|
||||
ss >> Tknu;
|
||||
ss >> Tktot;
|
||||
|
||||
if( k < m_kmin ) m_kmin = k;
|
||||
if( k > m_kmax ) m_kmax = k;
|
||||
|
||||
m_tab_k.push_back( log10(k) );
|
||||
|
||||
m_tab_Tk_tot.push_back( log10(Tktot) );
|
||||
m_tab_Tk_baryon.push_back( log10(Tkb) );
|
||||
m_tab_Tk_cdm.push_back( log10(Tkc) );
|
||||
++m_nlines;
|
||||
}
|
||||
|
||||
ifs.close();
|
||||
|
||||
|
||||
|
||||
|
||||
#ifdef WITH_MPI
|
||||
}
|
||||
|
||||
unsigned n=m_tab_k.size();
|
||||
MPI::COMM_WORLD.Bcast( &n, 1, MPI_UNSIGNED, 0 );
|
||||
|
||||
if( MPI::COMM_WORLD.Get_rank() > 0 ){
|
||||
m_tab_k.assign(n,0);
|
||||
m_tab_Tk_tot.assign(n,0);
|
||||
m_tab_Tk_cdm.assign(n,0);
|
||||
m_tab_Tk_baryon.assign(n,0);
|
||||
|
||||
}
|
||||
|
||||
MPI::COMM_WORLD.Bcast( &m_tab_k[0], n, MPI_DOUBLE, 0 );
|
||||
MPI::COMM_WORLD.Bcast( &m_tab_Tk_tot[0], n, MPI_DOUBLE, 0 );
|
||||
MPI::COMM_WORLD.Bcast( &m_tab_Tk_cdm[0], n, MPI_DOUBLE, 0 );
|
||||
MPI::COMM_WORLD.Bcast( &m_tab_Tk_baryon[0], n, MPI_DOUBLE, 0 );
|
||||
|
||||
#endif
|
||||
|
||||
}
|
||||
|
||||
public:
|
||||
transfer_CAMB_plugin( config_file& cf )
|
||||
: transfer_function_plugin( cf )
|
||||
{
|
||||
m_filename_Tk = pcf_->getValue<std::string>("cosmology","transfer_file");
|
||||
|
||||
read_table( );
|
||||
|
||||
acc_tot = gsl_interp_accel_alloc();
|
||||
acc_cdm = gsl_interp_accel_alloc();
|
||||
acc_baryon = gsl_interp_accel_alloc();
|
||||
|
||||
|
||||
spline_tot = gsl_spline_alloc( gsl_interp_cspline, m_tab_k.size() );
|
||||
spline_cdm = gsl_spline_alloc( gsl_interp_cspline, m_tab_k.size() );
|
||||
spline_baryon = gsl_spline_alloc( gsl_interp_cspline, m_tab_k.size() );
|
||||
|
||||
gsl_spline_init (spline_tot, &m_tab_k[0], &m_tab_Tk_tot[0], m_tab_k.size() );
|
||||
gsl_spline_init (spline_cdm, &m_tab_k[0], &m_tab_Tk_cdm[0], m_tab_k.size() );
|
||||
gsl_spline_init (spline_baryon, &m_tab_k[0], &m_tab_Tk_baryon[0], m_tab_k.size() );
|
||||
|
||||
tf_distinct_ = true;
|
||||
tf_withvel_ = false;
|
||||
}
|
||||
|
||||
~transfer_CAMB_plugin()
|
||||
{
|
||||
gsl_spline_free (spline_tot);
|
||||
gsl_spline_free (spline_cdm);
|
||||
gsl_spline_free (spline_baryon);
|
||||
|
||||
gsl_interp_accel_free (acc_tot);
|
||||
gsl_interp_accel_free (acc_cdm);
|
||||
gsl_interp_accel_free (acc_baryon);
|
||||
}
|
||||
|
||||
// linear interpolation in log-log
|
||||
inline double extrap_right( double k, const tf_type& type )
|
||||
{
|
||||
double v1(1.0), v2(1.0);
|
||||
if( MPI::COMM_WORLD.Get_rank() == 0 ){
|
||||
#endif
|
||||
LOGINFO("Reading tabulated transfer function data from file \n \'%s\'",m_filename_Tk.c_str() );
|
||||
|
||||
int n=m_tab_k.size()-1, n1=n-1;
|
||||
switch( type )
|
||||
{
|
||||
case cdm:
|
||||
v1 = m_tab_Tk_cdm[n1];
|
||||
v2 = m_tab_Tk_cdm[n];
|
||||
break;
|
||||
case baryon:
|
||||
v1 = m_tab_Tk_baryon[n1];
|
||||
v2 = m_tab_Tk_baryon[n];
|
||||
break;
|
||||
case vcdm:
|
||||
case vbaryon:
|
||||
case total:
|
||||
v1 = m_tab_Tk_tot[n1];
|
||||
v2 = m_tab_Tk_tot[n];
|
||||
break;
|
||||
|
||||
default:
|
||||
throw std::runtime_error("Invalid type requested in transfer function evaluation");
|
||||
std::string line;
|
||||
std::ifstream ifs( m_filename_Tk.c_str() );
|
||||
|
||||
if(! ifs.good() )
|
||||
throw std::runtime_error("Could not find transfer function file \'"+m_filename_Tk+"\'");
|
||||
|
||||
m_tab_k.clear();
|
||||
m_tab_Tk_tot.clear();
|
||||
m_tab_Tk_cdm.clear();
|
||||
m_tab_Tk_baryon.clear();
|
||||
m_tab_Tvk_cdm.clear(); //>[150609SH: add]
|
||||
m_tab_Tvk_baryon.clear(); //>[150609SH: add]
|
||||
|
||||
|
||||
//>[150609SH: add]
|
||||
m_kmin = 1e30;
|
||||
m_kmax = -1e30;
|
||||
|
||||
while( !ifs.eof() ){
|
||||
getline(ifs,line);
|
||||
|
||||
if(ifs.eof()) break;
|
||||
|
||||
std::stringstream ss(line);
|
||||
|
||||
double k, Tkc, Tkb, Tktot, Tkvc, Tkvb, dummy;
|
||||
|
||||
ss >> k;
|
||||
ss >> Tkc;
|
||||
ss >> Tkb;
|
||||
ss >> dummy;
|
||||
ss >> dummy;
|
||||
ss >> dummy;
|
||||
ss >> Tktot;
|
||||
ss >> dummy;
|
||||
ss >> dummy;
|
||||
ss >> dummy;
|
||||
ss >> Tkvc; //>[150609SH: add]
|
||||
ss >> Tkvb; //>[150609SH: add]
|
||||
ss >> dummy;
|
||||
|
||||
m_linbaryoninterp |= Tkb < 0.0 || Tkvb < 0.0;
|
||||
|
||||
m_tab_k.push_back( log10(k) );
|
||||
|
||||
m_tab_Tk_tot.push_back( Tktot );
|
||||
m_tab_Tk_baryon.push_back( Tkb );
|
||||
m_tab_Tk_cdm.push_back( Tkc );
|
||||
m_tab_Tvk_baryon.push_back( Tkvb );
|
||||
m_tab_Tvk_cdm.push_back( Tkvc );
|
||||
|
||||
++m_nlines;
|
||||
|
||||
if( k < m_kmin ) m_kmin = k;
|
||||
if( k > m_kmax ) m_kmax = k;
|
||||
}
|
||||
|
||||
double lk = log10(k);
|
||||
double dk = m_tab_k[n]-m_tab_k[n1];
|
||||
double delk = lk-m_tab_k[n];
|
||||
for( size_t i=0; i<m_tab_k.size(); ++i ){
|
||||
m_tab_Tk_tot[i] = log10( m_tab_Tk_tot[i] );
|
||||
m_tab_Tk_cdm[i] = log10( m_tab_Tk_cdm[i] );
|
||||
m_tab_Tvk_cdm[i] = log10( m_tab_Tvk_cdm[i] );
|
||||
|
||||
if( !m_linbaryoninterp )
|
||||
{
|
||||
m_tab_Tk_baryon[i] = log10( m_tab_Tk_baryon[i] );
|
||||
m_tab_Tvk_baryon[i] = log10( m_tab_Tvk_baryon[i] );
|
||||
}
|
||||
}
|
||||
|
||||
return pow(10.0,(v2-v1)/dk*(delk)+v2);
|
||||
ifs.close();
|
||||
|
||||
LOGINFO("Read CAMB transfer function table with %d rows", m_nlines);
|
||||
|
||||
if( m_linbaryoninterp )
|
||||
LOGINFO("Using log-lin interpolation for baryons\n (TF is not positive definite)");
|
||||
|
||||
#ifdef WITH_MPI
|
||||
}
|
||||
|
||||
inline double compute( double k, tf_type type )
|
||||
{
|
||||
// use constant interpolation on the left side of the tabulated values
|
||||
if( k < m_kmin )
|
||||
{
|
||||
if( type == cdm )
|
||||
return pow(10.0,m_tab_Tk_cdm[0]);
|
||||
|
||||
else if( type == baryon )
|
||||
return pow(10.0,m_tab_Tk_baryon[0]);
|
||||
|
||||
return pow(10.0,m_tab_Tk_tot[0]);
|
||||
|
||||
}
|
||||
// use linear interpolation on the right side of the tabulated values
|
||||
else if( k>m_kmax )
|
||||
return extrap_right( k, type );
|
||||
|
||||
|
||||
double lk = log10(k);
|
||||
if( type == cdm )
|
||||
return pow(10.0, gsl_spline_eval (spline_cdm, lk, acc_cdm) );
|
||||
|
||||
if( type == baryon )
|
||||
return pow(10.0, gsl_spline_eval (spline_baryon, lk, acc_baryon) );
|
||||
unsigned n=m_tab_k.size();
|
||||
MPI::COMM_WORLD.Bcast( &n, 1, MPI_UNSIGNED, 0 );
|
||||
|
||||
if( MPI::COMM_WORLD.Get_rank() > 0 ){
|
||||
m_tab_k.assign(n,0);
|
||||
m_tab_Tk_tot.assign(n,0);
|
||||
m_tab_Tk_cdm.assign(n,0);
|
||||
m_tab_Tk_baryon.assign(n,0);
|
||||
m_tab_Tvk_cdm.assign(n,0); //>[150609SH: add]
|
||||
m_tab_Tvk_baryon.assign(n,0); //>[150609SH: add]
|
||||
}
|
||||
|
||||
MPI::COMM_WORLD.Bcast( &m_tab_k[0], n, MPI_DOUBLE, 0 );
|
||||
MPI::COMM_WORLD.Bcast( &m_tab_Tk_tot[0], n, MPI_DOUBLE, 0 );
|
||||
MPI::COMM_WORLD.Bcast( &m_tab_Tk_cdm[0], n, MPI_DOUBLE, 0 );
|
||||
MPI::COMM_WORLD.Bcast( &m_tab_Tk_baryon[0], n, MPI_DOUBLE, 0 );
|
||||
MPI::COMM_WORLD.Bcast( &m_tab_Tvk_cdm[0], n, MPI_DOUBLE, 0 ); //>[150609SH: add]
|
||||
MPI::COMM_WORLD.Bcast( &m_tab_Tvk_baryon[0], n, MPI_DOUBLE, 0 ); //>[150609SH: add]
|
||||
|
||||
#endif
|
||||
|
||||
return pow(10.0, gsl_spline_eval (spline_tot, lk, acc_tot) );
|
||||
}
|
||||
|
||||
inline double get_kmin( void ){
|
||||
return pow(10.0,m_tab_k[1]);
|
||||
}
|
||||
|
||||
inline double get_kmax( void ){
|
||||
return pow(10.0,m_tab_k[m_tab_k.size()-2]);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
public:
|
||||
transfer_CAMB_plugin( config_file& cf )
|
||||
: transfer_function_plugin( cf )
|
||||
{
|
||||
m_filename_Tk = pcf_->getValue<std::string>("cosmology","transfer_file");
|
||||
|
||||
read_table( );
|
||||
|
||||
acc_tot = gsl_interp_accel_alloc();
|
||||
acc_cdm = gsl_interp_accel_alloc();
|
||||
acc_baryon = gsl_interp_accel_alloc();
|
||||
acc_vcdm = gsl_interp_accel_alloc(); //>[150609SH: add]
|
||||
acc_vbaryon = gsl_interp_accel_alloc(); //>[150609SH: add]
|
||||
|
||||
spline_tot = gsl_spline_alloc( gsl_interp_cspline, m_tab_k.size() );
|
||||
spline_cdm = gsl_spline_alloc( gsl_interp_cspline, m_tab_k.size() );
|
||||
spline_baryon = gsl_spline_alloc( gsl_interp_cspline, m_tab_k.size() );
|
||||
spline_vcdm = gsl_spline_alloc( gsl_interp_cspline, m_tab_k.size() ); //>[150609SH: add]
|
||||
spline_vbaryon = gsl_spline_alloc( gsl_interp_cspline, m_tab_k.size() ); //>[150609SH: add]
|
||||
|
||||
gsl_spline_init (spline_tot, &m_tab_k[0], &m_tab_Tk_tot[0], m_tab_k.size() );
|
||||
gsl_spline_init (spline_cdm, &m_tab_k[0], &m_tab_Tk_cdm[0], m_tab_k.size() );
|
||||
gsl_spline_init (spline_baryon, &m_tab_k[0], &m_tab_Tk_baryon[0], m_tab_k.size() );
|
||||
gsl_spline_init (spline_vcdm, &m_tab_k[0], &m_tab_Tvk_cdm[0], m_tab_k.size() ); //>[150609SH: add]
|
||||
gsl_spline_init (spline_vbaryon, &m_tab_k[0], &m_tab_Tvk_baryon[0], m_tab_k.size() ); //>[150609SH: add]
|
||||
|
||||
tf_distinct_ = true; // [150612SH: different density between CDM v.s. Baryon]
|
||||
tf_withvel_ = true; // [150612SH: using velocity transfer function]
|
||||
}
|
||||
|
||||
~transfer_CAMB_plugin()
|
||||
{
|
||||
gsl_spline_free (spline_tot);
|
||||
gsl_spline_free (spline_cdm);
|
||||
gsl_spline_free (spline_baryon);
|
||||
gsl_spline_free (spline_vcdm); //>[150609SH: add]
|
||||
gsl_spline_free (spline_vbaryon); //>[150609SH: add]
|
||||
|
||||
gsl_interp_accel_free (acc_tot);
|
||||
gsl_interp_accel_free (acc_cdm);
|
||||
gsl_interp_accel_free (acc_baryon);
|
||||
gsl_interp_accel_free (acc_vcdm); //>[150609SH: add]
|
||||
gsl_interp_accel_free (acc_vbaryon); //>[150609SH: add]
|
||||
}
|
||||
|
||||
// linear interpolation in log-log
|
||||
inline double extrap_right( double k, const tf_type& type )
|
||||
{
|
||||
int n=m_tab_k.size()-1, n1=n-1;
|
||||
|
||||
double v1(1.0), v2(1.0);
|
||||
|
||||
double lk = log10(k);
|
||||
double dk = m_tab_k[n]-m_tab_k[n1];
|
||||
double delk = lk-m_tab_k[n];
|
||||
|
||||
switch( type )
|
||||
{
|
||||
case cdm:
|
||||
v1 = m_tab_Tk_cdm[n1];
|
||||
v2 = m_tab_Tk_cdm[n];
|
||||
return pow(10.0,(v2-v1)/dk*(delk)+v2);
|
||||
case baryon:
|
||||
v1 = m_tab_Tk_baryon[n1];
|
||||
v2 = m_tab_Tk_baryon[n];
|
||||
if( m_linbaryoninterp)
|
||||
return std::max((v2-v1)/dk*(delk)+v2,tiny);
|
||||
return pow(10.0,(v2-v1)/dk*(delk)+v2);
|
||||
case vcdm: //>[150609SH: add]
|
||||
v1 = m_tab_Tvk_baryon[n1];
|
||||
v2 = m_tab_Tvk_baryon[n];
|
||||
return pow(10.0,(v2-v1)/dk*(delk)+v2);
|
||||
case vbaryon: //>[150609SH: add]
|
||||
v1 = m_tab_Tvk_baryon[n1];
|
||||
v2 = m_tab_Tvk_baryon[n];
|
||||
if( m_linbaryoninterp)
|
||||
return std::max((v2-v1)/dk*(delk)+v2,tiny);
|
||||
return pow(10.0,(v2-v1)/dk*(delk)+v2);
|
||||
case total:
|
||||
v1 = m_tab_Tk_tot[n1];
|
||||
v2 = m_tab_Tk_tot[n];
|
||||
return pow(10.0,(v2-v1)/dk*(delk)+v2);
|
||||
default:
|
||||
throw std::runtime_error("Invalid type requested in transfer function evaluation");
|
||||
}
|
||||
|
||||
return 0.0;
|
||||
}
|
||||
|
||||
inline double compute( double k, tf_type type )
|
||||
{
|
||||
// use constant interpolation on the left side of the tabulated values
|
||||
if( k < m_kmin ){
|
||||
switch( type ){
|
||||
case cdm:
|
||||
return pow(10.0,m_tab_Tk_cdm[0]);
|
||||
case baryon:
|
||||
if( m_linbaryoninterp )
|
||||
return m_tab_Tk_baryon[0];
|
||||
return pow(10.0,m_tab_Tk_baryon[0]);
|
||||
case vcdm:
|
||||
return pow(10.0,m_tab_Tvk_cdm[0]);
|
||||
case vbaryon:
|
||||
if( m_linbaryoninterp )
|
||||
return m_tab_Tvk_baryon[0];
|
||||
return pow(10.0,m_tab_Tvk_baryon[0]);
|
||||
case total:
|
||||
return pow(10.0,m_tab_Tk_tot[0]);
|
||||
default:
|
||||
throw std::runtime_error("Invalid type requested in transfer function evaluation");
|
||||
}
|
||||
}
|
||||
// use linear interpolation on the right side of the tabulated values
|
||||
else if( k>m_kmax )
|
||||
return extrap_right( k, type );
|
||||
|
||||
|
||||
double lk = log10(k);
|
||||
switch( type ){
|
||||
case cdm:
|
||||
return pow(10.0, gsl_spline_eval (spline_cdm, lk, acc_cdm) );
|
||||
case baryon:
|
||||
if( m_linbaryoninterp )
|
||||
return gsl_spline_eval (spline_baryon, lk, acc_baryon);
|
||||
return pow(10.0, gsl_spline_eval (spline_baryon, lk, acc_baryon) );
|
||||
case vcdm:
|
||||
return pow(10.0, gsl_spline_eval (spline_vcdm, lk, acc_vcdm) );
|
||||
case vbaryon:
|
||||
if( m_linbaryoninterp )
|
||||
return gsl_spline_eval (spline_vbaryon, lk, acc_vbaryon);
|
||||
return pow(10.0, gsl_spline_eval (spline_vbaryon, lk, acc_vbaryon) );
|
||||
case total:
|
||||
return pow(10.0, gsl_spline_eval (spline_tot, lk, acc_tot) );
|
||||
default:
|
||||
throw std::runtime_error("Invalid type requested in transfer function evaluation");
|
||||
}
|
||||
}
|
||||
|
||||
inline double get_kmin( void ){
|
||||
return pow(10.0,m_tab_k[1]);
|
||||
}
|
||||
|
||||
inline double get_kmax( void ){
|
||||
return pow(10.0,m_tab_k[m_tab_k.size()-2]);
|
||||
}
|
||||
|
||||
};
|
||||
|
||||
namespace{
|
||||
transfer_function_plugin_creator_concrete< transfer_CAMB_plugin > creator("camb_file");
|
||||
transfer_function_plugin_creator_concrete< transfer_CAMB_plugin > creator("camb_file");
|
||||
}
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue