CsvLogger.cpp

Go to the documentation of this file.

#include "mvsim/CsvLogger.h"

CSVLogger::CSVLogger()
{
        file_ = std::make_shared<std::ofstream>(std::ofstream());
}

CSVLogger::~CSVLogger() { close(); }
void CSVLogger::addColumn(std::string name) { columns_[name] = 0.0; }
void CSVLogger::updateColumn(std::string name, double value)
{
        columns_[name] = value;
}

bool CSVLogger::writeHeader()
{
        columns_type::iterator it;
        for (it = columns_.begin(); it != columns_.end();)
        {
                *file_ << it->first;

                if (++it != columns_.end())
                {
                        *file_ << ", ";
                }
        }

        *file_ << "\n";  // most CSV readers don't use \r\n

        return !!file_;
}

bool CSVLogger::writeRow()
{
        if (!isRecording) return true;

        if (!isOpen()) clear();

        columns_type::iterator it;
        for (it = columns_.begin(); it != columns_.end();)
        {
                *file_ << it->second;

                if (++it != columns_.end())
                {
                        *file_ << ", ";
                }
        }

        *file_ << "\n";

        return !!file_;
}

bool CSVLogger::open()
{
        if (file_)
        {
                file_->open((std::string("session") + std::to_string(currentSession) +
                                         std::string("-") + filepath_)
                                                .c_str());
                return isOpen();
        }
        return false;
}

bool CSVLogger::isOpen() { return file_->is_open(); }
bool CSVLogger::close()
{
        if (file_)
        {
                file_->close();
                return !isOpen();
        }
        return false;
}

bool CSVLogger::clear()
{
        if (isOpen()) close();

        if (open())
        {
                return writeHeader();
        }

        return false;
}

void CSVLogger::newSession()
{
        currentSession++;
        close();
}