stream.h 1.82 KB
Newer Older
1
2
#pragma once

3
4
5
6
#ifndef STREAM_H_62B23520_7C8E_11DE_8A39_0800200C9A66
#define STREAM_H_62B23520_7C8E_11DE_8A39_0800200C9A66


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

namespace YAML
{
18
	static const size_t MAX_PARSER_PUSHBACK = 8;
19

20
	class Stream: private noncopyable
21
	{
22
	public:
23
24
		friend class StreamCharSource;
		
25
26
		Stream(std::istream& input);
		~Stream();
27

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

31
		char peek() const;
32
33
34
35
		char get();
		std::string get(int n);
		void eat(int n = 1);

36
		static char eof() { return 0x04; }
37
38
39
40
41
42
		
		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; }
43

44
	private:
45
46
47
		enum CharacterSet {utf8, utf16le, utf16be, utf32le, utf32be};

		std::istream& m_input;
48
49
		Mark m_mark;
		
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
		CharacterSet m_charSet;
		unsigned char m_bufPushback[MAX_PARSER_PUSHBACK];
		mutable size_t m_nPushedBack;
		mutable std::deque<char> m_readahead;
		unsigned char* const m_pPrefetched;
		mutable size_t m_nPrefetchedAvailable;
		mutable size_t m_nPrefetchedUsed;
		
		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;
66
	};
67
68
69
70
71
72
73
74
75
76
77
78

	// 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);
	}	
79
}
80
81

#endif // STREAM_H_62B23520_7C8E_11DE_8A39_0800200C9A66