stream.h 1.99 KB
Newer Older
1
2
3
#ifndef STREAM_H_62B23520_7C8E_11DE_8A39_0800200C9A66
#define STREAM_H_62B23520_7C8E_11DE_8A39_0800200C9A66

Jesse Beder's avatar
Jesse Beder committed
4
5
6
#if defined(_MSC_VER) ||                                            \
    (defined(__GNUC__) && (__GNUC__ == 3 && __GNUC_MINOR__ >= 4) || \
     (__GNUC__ >= 4))  // GCC supports "pragma once" correctly since 3.4
7
8
9
#pragma once
#endif

10
#include "yaml-cpp/mark.h"
Jesse Beder's avatar
Jesse Beder committed
11
#include <cstddef>
12
#include <deque>
13
#include <ios>
14
15
#include <iostream>
#include <set>
Jesse Beder's avatar
Jesse Beder committed
16
#include <string>
17

Jesse Beder's avatar
Jesse Beder committed
18
namespace YAML {
19
class Stream {
Jesse Beder's avatar
Jesse Beder committed
20
21
22
23
 public:
  friend class StreamCharSource;

  Stream(std::istream& input);
24
25
26
27
  Stream(const Stream&) = delete;
  Stream(Stream&&) = delete;
  Stream& operator=(const Stream&) = delete;
  Stream& operator=(Stream&&) = delete;
Jesse Beder's avatar
Jesse Beder committed
28
29
30
31
32
33
34
35
36
  ~Stream();

  operator bool() const;
  bool operator!() const { return !static_cast<bool>(*this); }

  char peek() const;
  char get();
  std::string get(int n);
  void eat(int n = 1);
37

Jesse Beder's avatar
Jesse Beder committed
38
  static char eof() { return 0x04; }
39

Jesse Beder's avatar
Jesse Beder committed
40
41
42
43
44
  const Mark mark() const { return m_mark; }
  int pos() const { return m_mark.pos; }
  int line() const { return m_mark.line; }
  int column() const { return m_mark.column; }
  void ResetColumn() { m_mark.column = 0; }
45

Jesse Beder's avatar
Jesse Beder committed
46
 private:
Jesse Beder's avatar
Jesse Beder committed
47
  enum CharacterSet { utf8, utf16le, utf16be, utf32le, utf32be };
48

Jesse Beder's avatar
Jesse Beder committed
49
50
  std::istream& m_input;
  Mark m_mark;
51

Jesse Beder's avatar
Jesse Beder committed
52
53
54
55
56
  CharacterSet m_charSet;
  mutable std::deque<char> m_readahead;
  unsigned char* const m_pPrefetched;
  mutable size_t m_nPrefetchedAvailable;
  mutable size_t m_nPrefetchedUsed;
57

Jesse Beder's avatar
Jesse Beder committed
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
  void AdvanceCurrent();
  char CharAt(size_t i) const;
  bool ReadAheadTo(size_t i) const;
  bool _ReadAheadTo(size_t i) const;
  void StreamInUtf8() const;
  void StreamInUtf16() const;
  void StreamInUtf32() const;
  unsigned char GetNextByte() const;
};

// CharAt
// . Unchecked access
inline char Stream::CharAt(size_t i) const { return m_readahead[i]; }

inline bool Stream::ReadAheadTo(size_t i) const {
  if (m_readahead.size() > i)
    return true;
  return _ReadAheadTo(i);
}
77
}  // namespace YAML
78

Jesse Beder's avatar
Jesse Beder committed
79
#endif  // STREAM_H_62B23520_7C8E_11DE_8A39_0800200C9A66