stream.h 1.91 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
11
#include "yaml-cpp/noncopyable.h"
#include "yaml-cpp/mark.h"
Jesse Beder's avatar
Jesse Beder committed
12
#include <cstddef>
13
#include <deque>
14
#include <ios>
15
16
#include <iostream>
#include <set>
Jesse Beder's avatar
Jesse Beder committed
17
#include <string>
18

Jesse Beder's avatar
Jesse Beder committed
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
namespace YAML {
class Stream : private noncopyable {
 public:
  friend class StreamCharSource;

  Stream(std::istream& input);
  ~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);
34

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

Jesse Beder's avatar
Jesse Beder committed
37
38
39
40
41
  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; }
42

Jesse Beder's avatar
Jesse Beder committed
43
44
45
46
47
48
49
50
 private:
  enum CharacterSet {
    utf8,
    utf16le,
    utf16be,
    utf32le,
    utf32be
  };
51

Jesse Beder's avatar
Jesse Beder committed
52
53
  std::istream& m_input;
  Mark m_mark;
54

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

Jesse Beder's avatar
Jesse Beder committed
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
  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);
}
80
}
81

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