read_theta.hpp

Go to the documentation of this file.

// SPDX-License-Identifier: BSD-2-Clause
 
#ifndef cMHN_UTILITY_READ_THETA_H
#define cMHN_UTILITY_READ_THETA_H
 
#include <fstream>
#include <iomanip>
#include <string>
 
#include <prc.hpp>
 
namespace cMHN
{
    template<class T, pRC::Size D>
    static inline auto readTheta(std::string const &filename)
    {
        std::ifstream file(filename);
 
        if(!file.is_open())
        {
            pRC::Logging::warning(
                "Unable to open file containing previous theta!");
            pRC::Logging::info("Returning pRC::zero()");
            return eval(pRC::zero<pRC::Tensor<T, D, D>>());
        }
 
        pRC::Tensor<T, D, D> theta = pRC::zero();
        pRC::Tensor<pRC::Bool, D, D> check = pRC::zero();
 
        std::string line;
 
        // First line contains header
        std::getline(file, line);
 
        pRC::Index i = 0;
        while(std::getline(file, line))
        {
            std::istringstream iss(line);
            pRC::Index j = 0;
            T v;
            while(iss >> v())
            {
                if(j >= D)
                {
                    pRC::Logging::warning(
                        "Number of events in file and binary do not match "
                        "(#File > #Binary)");
                    pRC::Logging::info("Returning pRC::zero()");
                    return eval(pRC::zero<decltype(theta)>());
                }
                theta(i, j) = v;
                check(i, j) = true;
                ++j;
            }
            ++i;
        }
 
        if(check)
        {
            return theta;
        }
        else
        {
            pRC::Logging::warning("Failed to read thetas from:", filename);
            pRC::Logging::warning("Probably file had fewer events than binary");
            pRC::Logging::info("Returning pRC::zero()");
            return eval(pRC::zero<decltype(theta)>());
        }
    }
}
 
#endif // CMHN_UTILITY_READ_THETA_H