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

3
#include "noncopyable.h"
4
#include "mark.h"
5
#include <deque>
6
7
#include <ios>
#include <string>
8
9
#include <iostream>
#include <set>
10
11
12

namespace YAML
{
13
	static const size_t MAX_PARSER_PUSHBACK = 8;
14

15
	class Stream: private noncopyable
16
	{
17
	public:
18
19
		friend class StreamCharSource;
		
20
21
		Stream(std::istream& input);
		~Stream();
22

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

26
		char peek() const;
27
28
29
30
		char get();
		std::string get(int n);
		void eat(int n = 1);

31
		static char eof() { return 0x04; }
32
33
34
35
36
37
		
		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; }
38

39
	private:
40
41
42
		enum CharacterSet {utf8, utf16le, utf16be, utf32le, utf32be};

		std::istream& m_input;
43
44
		Mark m_mark;
		
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
		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;
61
	};
62
63
64
65
66
67
68
69
70
71
72
73

	// 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);
	}	
74
}