write_theta.hpp

Go to the documentation of this file.

// SPDX-License-Identifier: BSD-2-Clause
 
#ifndef cMHN_UTILITY_WRITE_THETA_H
#define cMHN_UTILITY_WRITE_THETA_H
 
#include <fstream>
#include <iomanip>
#include <map>
#include <string>
 
namespace cMHN
{
    template<class T, pRC::Size D>
    static inline auto writeTheta(std::string const &filename,
        std::string const &header, pRC::Tensor<T, D, D> const &theta,
        std::map<std::string, std::string> const &logInfoNames = {},
        std::map<std::string, double> const &logInfoNumbers = {})
    {
        std::ofstream file(filename);
 
        if(!file.is_open())
        {
            pRC::Logging::error("Unable to open output file!");
        }
 
        // print header and theta matrix into file
        file << header << std::endl;
        file << std::fixed << std::showpos << std::setprecision(6);
 
        for(pRC::Index i = 0; i < D; ++i)
        {
            file << theta(i, 0)();
            for(pRC::Index j = 1; j < D; ++j)
            {
                file << " " << theta(i, j)();
            }
            file << std::endl;
        }
 
        // print additional logging information into file
        for(auto const &[k, v] : logInfoNames)
        {
            file << k << ": " << v << std::endl;
        }
        file << std::scientific;
        for(auto const &[k, v] : logInfoNumbers)
        {
            file << k << ": " << v << std::endl;
        }
    }
}
 
#endif // CMHN_UTILITY_WRITE_THETA_H