csv_utility.hpp

#pragma once
#include "common.hpp"
#include "csv_format.hpp"
#include "csv_reader.hpp"
#include "data_type.hpp"
#include "string_view_stream.hpp"
 
#include <memory>
#include <sstream>
#include <string>
#include <type_traits>
#include <unordered_map>
 
namespace csv {
    struct CSVFileInfo {
        std::string filename;               
        std::vector<std::string> col_names; 
        char delim;                         
        size_t n_rows;                      
        size_t n_cols;                      
    };
 
 
    inline CSVReader parse(csv::string_view in, const CSVFormat& format = CSVFormat::guess_csv()) {
        std::unique_ptr<std::istream> ss(new std::stringstream(std::string(in)));
        return CSVReader(std::move(ss), format);
    }
 
    inline CSVReader parse_unsafe(csv::string_view in, CSVFormat format = CSVFormat::guess_csv()) {
        std::unique_ptr<std::istream> stream(new internals::StringViewStream(in));
        return CSVReader(std::move(stream), format);
    }
 
    inline CSVReader parse_no_header(csv::string_view in) {
        CSVFormat format;
        format.header_row(-1);
        return parse(in, format);
    }
 
    inline CSVReader operator ""_csv(const char* in, size_t n) {
        return parse_unsafe(csv::string_view(in, n));
    }
 
    inline CSVReader operator ""_csv_no_header(const char* in, size_t n) {
        CSVFormat format;
        format.header_row(-1);
        return parse_unsafe(csv::string_view(in, n), format);
    }
 
    std::unordered_map<std::string, DataType> csv_data_types(const std::string&);
 
    inline CSVFileInfo get_file_info(const std::string& filename) {
        CSVReader reader(filename);
        CSVFormat format = reader.get_format();
        for (auto it = reader.begin(); it != reader.end(); ++it);
 
        return {
            filename,
            reader.get_col_names(),
            format.get_delim(),
            reader.n_rows(),
            reader.get_col_names().size()
        };
    }
 
    inline std::vector<std::string> get_col_names(
        csv::string_view filename,
        const CSVFormat& format = CSVFormat::guess_csv()) {
        auto head = internals::get_csv_head(filename);
        return parse_unsafe(head, format).get_col_names();
    }
 
    inline long long get_col_pos(csv::string_view filename, csv::string_view col_name,
        const CSVFormat& format = CSVFormat::guess_csv()) {
        auto col_names = get_col_names(filename, format);
        return col_names.empty() ? CSV_NOT_FOUND :
            std::distance(col_names.begin(), std::find(col_names.begin(), col_names.end(), col_name));
    }
}