CSV.hpp

Go to the documentation of this file.

/*
 * Copyright (c) 2015 - 2024 Intel Corporation
 * SPDX-License-Identifier: BSD-3-Clause
 */
 
#ifndef CSV_HPP_INCLUDE
#define CSV_HPP_INCLUDE
 
#include <vector>
#include <functional>
#include <map>
#include <string>
#include <fstream>
#include <sstream>
 
namespace geopm
{
    class CSV
    {
        public:
            CSV() = default;
            virtual ~CSV() = default;
            virtual void add_column(const std::string &name) = 0;
            virtual void add_column(const std::string &name,
                                    const std::string &format) = 0;
            virtual void add_column(const std::string &name,
                                    std::function<std::string(double)> format) = 0;
            virtual void activate(void) = 0;
            virtual void update(const std::vector<double> &sample) = 0;
            virtual void flush(void) = 0;
    };
 
    class CSVImp : public CSV
    {
        public:
            CSVImp(const std::string &file_path,
                   const std::string &host_name,
                   const std::string &start_time,
                   size_t buffer_size);
            CSVImp(const CSVImp &other) = delete;
            CSVImp & operator=(const CSVImp &other) = delete;
            virtual ~CSVImp();
            void add_column(const std::string &name) override;
            void add_column(const std::string &name,
                            const std::string &format) override;
            void add_column(const std::string &name,
                            std::function<std::string(double)> format) override;
            void activate(void) override;
            void update(const std::vector<double> &sample) override;
            void flush(void) override;
        private:
            void write_header(const std::string &host_name, const std::string &start_time);
            void write_names(void);
 
            const std::map<std::string, std::function<std::string(double)> > M_NAME_FORMAT_MAP;
            const char M_SEPARATOR;
            std::string m_file_path;
            std::vector<std::string> m_column_name;
            std::vector<std::function<std::string(double)> > m_column_format;
            std::ofstream m_stream;
            std::ostringstream m_buffer;
            off_t m_buffer_limit;
            bool m_is_active;
    };
}
 
#endif