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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68

#pragma once

#include <filesystem>

#include <openvic-dataloader/Error.hpp>
#include <openvic-dataloader/Parser.hpp>
#include <openvic-dataloader/csv/LineObject.hpp>
#include <openvic-dataloader/detail/utility/Concepts.hpp>
#include <openvic-dataloader/detail/utility/ErrorRange.hpp>

#include <dryad/node.hpp>

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

   template<EncodingType Encoding = EncodingType::Windows1252>
   class Parser final : public detail::BasicParser {
   public:
      Parser();
      Parser(std::basic_ostream<char>& error_stream);

      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::HasCstr auto& path) {
         return load_from_file(path.c_str());
      }

      bool parse_csv(bool handle_strings = false);

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

      using error_range = ovdl::detail::error_range;
      Parser::error_range get_errors() const;

      const FilePosition get_error_position(const error::Error* error) const;

      void print_errors_to(std::basic_ostream<char>& stream) const;

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

      ~Parser();

   private:
      class ParseHandler;
      std::unique_ptr<ParseHandler> _parse_handler;
      std::vector<csv::LineObject> _lines;

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

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