csvDataRecorder.ipp

Go to the documentation of this file.

 
// Note this is a template implementation file and should not be included directly.
// The typical header protection macro is not added to avoid it showing up in Doxygen API
// documentation.
#pragma once
 
#include <rtctk/componentFramework/csvDataRecorder.hpp>
 
#include <fmt/core.h>
#include <iomanip>
#include <iostream>
#include <ostream>
#include <sstream>
#include <string_view>
#include <type_traits>
 
namespace rtctk::componentFramework {
 
template <class... T>
CsvDataRecorder<T...>::CsvDataRecorder(const ColumnDescription& columns,
                                       const std::array<bool, sizeof...(T)>& disabled)
    : DataRecorder<T...>(columns, disabled) {
}
 
template <class... T>
void CsvDataRecorder<T...>::Open(const std::filesystem::path& file) {
    m_output.open(file, std::ios::out);
    bool first = true;
    for (const auto& column : this->m_column_description) {
        if (!first) {
            m_output << ",";
        } else {
            first = false;
        }
        if (column.unit != "") {
            m_output << std::quoted(fmt::format("{} [{}]", column.name, column.unit));
        } else {
            m_output << std::quoted(column.name);
        }
    }
    m_output << "\n";
}
 
template <class... T>
void CsvDataRecorder<T...>::Close() {
    m_output.close();
}
 
template <class... T>
void CsvDataRecorder<T...>::Write(const std::tuple<T...>& data) {
    bool first = true;
    size_t column_index = 0;
 
    std::apply(
        [this, &first, &column_index](auto&&... args) mutable {
            auto printer = [this, &first, &column_index](auto&& arg) mutable {
                if (this->m_disabled_fields[column_index]) {
                    column_index++;
                    return;
                }
                if (first) {
                    if constexpr (std::is_same_v<std::decay_t<decltype(arg)>, std::string>) {
                        m_output << std::quoted(arg);
                    } else {
                        m_output << "\"" << arg << "\"";
                    }
                    first = false;
                } else {
                    if constexpr (std::is_same_v<std::decay_t<decltype(arg)>, std::string>) {
                        m_output << "," << std::quoted(arg);
                    } else {
                        m_output << ",\"" << arg << "\"";
                    }
                }
            };
            (printer(args), ...);
        },
        data);
    m_output << "\n";
}
 
}  // namespace rtctk::componentFramework