include/openvic-dataloader/csv/Parser.hpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

#pragma once

#include <openvic-dataloader/csv/LineObject.hpp>
#include <openvic-dataloader/detail/BasicParser.hpp>

namespace ovdl::csv {
   enum class EncodingType {
      Windows1252,
      Utf8
   };

   template<EncodingType Encoding = EncodingType::Windows1252>
   class Parser final : public detail::BasicParser {
   public:
      Parser();

      static Parser from_buffer(const char* data, std::size_t size);
      static Parser from_buffer(const char* start, const char* end);
      static Parser from_string(const std::string_view string);
      static Parser from_file(const char* path);
      static Parser from_file(const std::filesystem::path& path);

      constexpr Parser& load_from_buffer(const char* data, std::size_t size);
      constexpr Parser& load_from_buffer(const char* start, const char* end);
      constexpr Parser& load_from_string(const std::string_view string);
      constexpr Parser& load_from_file(const char* path);
      Parser& load_from_file(const std::filesystem::path& path);

      constexpr Parser& load_from_file(const detail::Has_c_str auto& path);

      bool parse_csv();

      const std::vector<csv::LineObject>& get_lines() const;

      Parser(Parser&&);
      Parser& operator=(Parser&&);

      ~Parser();

   private:
      class BufferHandler;
      std::unique_ptr<BufferHandler> _buffer_handler;
      std::vector<csv::LineObject> _lines;

      template<typename... Args>
      constexpr void _run_load_func(detail::LoadCallback<BufferHandler, Args...> auto func, Args... args);
   };

   using Windows1252Parser = Parser<EncodingType::Windows1252>;
   using Utf8Parser = Parser<EncodingType::Utf8>;
}