col_names.hpp

#pragma once
#include <memory>
#include <stdexcept>
#include <unordered_map>
#include <string>
#include <vector>
 
#include "common.hpp"
#include "csv_format.hpp"
 
namespace csv {
    namespace internals {
        struct ColNames;
        using ColNamesPtr = std::shared_ptr<ColNames>;
 
        struct ColNames {
        public:
            ColNames() = default;
            ColNames(const std::vector<std::string>& names) {
                set_col_names(names);
            }
 
            std::vector<std::string> get_col_names() const;
            void set_col_names(const std::vector<std::string>&);
            int index_of(csv::string_view) const;
 
            void set_policy(csv::ColumnNamePolicy policy);
 
            bool empty() const noexcept { return this->col_names.empty(); }
            size_t size() const noexcept;
 
            const std::string& operator[](size_t i) const;
 
        private:
            std::vector<std::string> col_names;
            std::unordered_map<std::string, size_t> col_pos;
            csv::ColumnNamePolicy _policy = csv::ColumnNamePolicy::EXACT;
        };
    }
}