IfStream.cpp

#define BIORBD_API_EXPORTS
#include "Utils/IfStream.h"
 
#include <clocale>
#include <fstream>
#include "Utils/Error.h"
#include "Utils/Equation.h"
 
// Constructor
biorbd::utils::IfStream::IfStream() :
    m_isOpen(std::make_shared<bool>(false)),
    m_ifs(std::make_shared<std::ifstream>()),
    m_path(std::make_shared<biorbd::utils::Path>())
{
    setlocale(LC_ALL, "C");
}
biorbd::utils::IfStream::IfStream(
        const biorbd::utils::Path& path,
        std::ios_base::openmode mode = std::ios_base::in ) :
    m_isOpen(std::make_shared<bool>(false)),
    m_ifs(std::make_shared<std::ifstream>()),
    m_path(std::make_shared<biorbd::utils::Path>(path))
{
    open(m_path->absolutePath().c_str(), mode);
    setlocale(LC_ALL, "C");
}
biorbd::utils::IfStream::IfStream(
        const char* path,
        std::ios_base::openmode mode = std::ios_base::in ) :
    m_isOpen(std::make_shared<bool>(false)),
    m_ifs(std::make_shared<std::ifstream>()),
    m_path(std::make_shared<biorbd::utils::Path>(path))
{
    open(m_path->absolutePath().c_str(), mode);
    setlocale(LC_ALL, "C");
}
 
 
// Open a file
bool biorbd::utils::IfStream::open(
        const biorbd::utils::Path& path,
        std::ios_base::openmode mode = std::ios_base::in )
{
    biorbd::utils::Error::check(path.isFileExist(), path.absolutePath() + " could not be loaded");
    *m_ifs = std::ifstream(path.absolutePath().c_str(), mode);
    *m_isOpen = true;
    return *m_isOpen;
}
 
// Read a file
bool biorbd::utils::IfStream::readSpecificTag(
        const biorbd::utils::String& tag,
        biorbd::utils::String& text){
    reachSpecificTag(tag);
    read(text);
    return true;
}
bool biorbd::utils::IfStream::reachSpecificTag(const biorbd::utils::String& tag){
    biorbd::utils::String text;
    while (read(text))
        if (!text.tolower().compare(tag))
            return true;
 
    biorbd::utils::String outMessage(tag + " parameter could not be found in Data file..");
    biorbd::utils::Error::raise(outMessage);
#ifdef _WIN32
    // It is impossible to get here, but it's better to have a return for the compiler
    return false;
#endif
}
 
int biorbd::utils::IfStream::countTagsInAConsecutiveLines(const biorbd::utils::String &tag)
{
    // Remember where we were in the file
    std::streamoff positionInFile(m_ifs->tellg());
    biorbd::utils::String text;
    int nMarkers(0);
 
    // Read the first word of the line
    while (read(text))
        // If same as tag, skip to the other line and restart
        if (!text.compare(tag)){
            getline(text);
            ++nMarkers;
        }
        else
            break;
 
    // Reset in file to the origine point
    m_ifs->seekg(positionInFile);
    return nMarkers;
}
 
bool biorbd::utils::IfStream::read(biorbd::utils::String& text){
    bool out(*m_ifs >> text);
 
    if (out == 0)
        return out;
 
    if (!text(0,1).compare("//")){ // If it's a comment by //
        getline(text);
        read(text);
    }
    else if (!text(0,1).compare("/*")){ // If it's a comment by / *
        while (readAWord(text)){
            if (!text(0,1).compare("*/") || (text.length()>=2 && !text(static_cast<unsigned int>(text.length()-2),static_cast<unsigned int>(text.length()-1)).compare("*/")))
                break;
        }
        read(text);
    }
    return out;
}
bool biorbd::utils::IfStream::readAWord(biorbd::utils::String& text){
    bool out(*m_ifs >> text);
    return out;
}
bool biorbd::utils::IfStream::read(
        double& val){
    std::map<biorbd::utils::Equation, double> dumb;
    return read(val, dumb);
}
#ifdef BIORBD_USE_CASADI_MATH
bool biorbd::utils::IfStream::read(
        RBDLCasadiMath::MX_Xd_SubMatrix val){
    std::map<biorbd::utils::Equation, double> dumb;
    return read(val, dumb);
}
#endif
bool biorbd::utils::IfStream::read(
        double& result,
        const std::map<biorbd::utils::Equation, double> &variables){
    biorbd::utils::Equation tp;
    bool out(read(tp));
    // Manage in case of an equation
    try {
        result = biorbd::utils::Equation::evaluateEquation(tp, variables);
    } catch (std::runtime_error) {
        biorbd::utils::Error::raise("The following expression cannot be parsed properly: \"" + tp + "\"");
    }
    return out;
}
#ifdef BIORBD_USE_CASADI_MATH
bool biorbd::utils::IfStream::read(
        RBDLCasadiMath::MX_Xd_SubMatrix result,
        const std::map<biorbd::utils::Equation, double> &variables){
    biorbd::utils::Equation tp;
    bool out(read(tp));
    // Manage in case of an equation
    try {
        result = biorbd::utils::Equation::evaluateEquation(tp, variables);
    } catch (std::runtime_error) {
        biorbd::utils::Error::raise("The following expression cannot be parsed properly: \"" + tp + "\"");
    }
    return out;
}
#endif
bool biorbd::utils::IfStream::read(
        int& val){
    biorbd::utils::String tp;
    bool out(read(tp));
    val = std::stoi(tp);
    return out;
}
bool biorbd::utils::IfStream::read(
        unsigned int& val){
    biorbd::utils::String tp;
    bool out(read(tp));
    val = static_cast<unsigned int>(std::stoul(tp));
    return out;
}
bool biorbd::utils::IfStream::read(
        bool& val){
    biorbd::utils::String tp;
    bool out(read(tp));
    val = std::stoi(tp) != 0;
    return out;
}
// Read the entire line
void biorbd::utils::IfStream::getline(biorbd::utils::String& text){
    std::getline(*m_ifs, text);
}
 
 
// Close the file
bool biorbd::utils::IfStream::close(){
    m_ifs->close();
    return 1;
}
 
bool biorbd::utils::IfStream::eof()
{
    return m_ifs->eof();
}