utils_mpi_io.h

#pragma once
 
#include "envs_io.h"
#include "envs_mpi.h"
 
namespace LIBRPA
{
 
namespace utils
{
 
template <typename... Args>
void lib_printf_root(const char* format, Args&&... args)
{
    if (envs::myid_global == 0)
    {
        envs::redirect_stdout ?
            fprintf(envs::pfile_redirect, format, std::forward<Args>(args)...) :
            printf(format, std::forward<Args>(args)...);
    }
}
 
template <typename... Args>
void lib_printf_coll(const char* format, Args&&... args)
{
    for (int i = 0; i < envs::size_global; i++)
    {
        if (envs::myid_global == i)
        {
            envs::redirect_stdout ?
                fprintf(envs::pfile_redirect, format, std::forward<Args>(args)...) :
                printf(format, std::forward<Args>(args)...);
        }
        MPI_Barrier(envs::mpi_comm_global);
    }
}
 
} /* end of name space utils */
 
} /* end of name space LIBRPA */