scanner.h 1.82 KB
Newer Older
Jesse Beder's avatar
Jesse Beder committed
1
2
3
4
5
6
#pragma once

#include <ios>
#include <string>
#include <queue>
#include <stack>
Jesse Beder's avatar
Jesse Beder committed
7
#include <set>
Jesse Beder's avatar
Jesse Beder committed
8
9
10

namespace YAML
{
11
	struct Token;
Jesse Beder's avatar
Jesse Beder committed
12
13
14
15
16
17
18

	class Scanner
	{
	public:
		Scanner(std::istream& in);
		~Scanner();

19
20
		Token *GetNextToken();

Jesse Beder's avatar
Jesse Beder committed
21
		void ScanNextToken();
Jesse Beder's avatar
Jesse Beder committed
22
		void ScanToNextToken();
Jesse Beder's avatar
Jesse Beder committed
23
		Token *PushIndentTo(int column, bool sequence);
Jesse Beder's avatar
Jesse Beder committed
24
		void PopIndentTo(int column);
25
26
		void IncreaseFlowLevel();
		void DecreaseFlowLevel();
Jesse Beder's avatar
Jesse Beder committed
27

Jesse Beder's avatar
Jesse Beder committed
28
29
30
31
		void InsertSimpleKey();
		bool ValidateSimpleKey();
		void ValidateAllSimpleKeys();

Jesse Beder's avatar
Jesse Beder committed
32
33
34
35
		void Scan();

	private:
		char GetChar();
36
		std::string GetChar(int n);
Jesse Beder's avatar
Jesse Beder committed
37
		void Eat(int n = 1);
Jesse Beder's avatar
Jesse Beder committed
38
39
		void EatLineBreak();

Jesse Beder's avatar
Jesse Beder committed
40
		bool IsWhitespaceToBeEaten(char ch);
Jesse Beder's avatar
Jesse Beder committed
41
42
		bool IsDocumentStart();
		bool IsDocumentEnd();
Jesse Beder's avatar
Jesse Beder committed
43
44
45
46
47
		bool IsBlockEntry();
		bool IsKey();
		bool IsValue();
		bool IsPlainScalar();

48
49
		struct WhitespaceInfo {
			WhitespaceInfo();
Jesse Beder's avatar
Jesse Beder committed
50

51
52
53
54
55
56
57
58
			void AddBlank(char ch);
			void AddBreak(const std::string& line);
			std::string Join();

			bool leadingBlanks;
			std::string whitespace, leadingBreaks, trailingBreaks;
		};

Jesse Beder's avatar
Jesse Beder committed
59
		struct SimpleKey {
60
			SimpleKey(int pos_, int line_, int column_, int flowLevel_);
Jesse Beder's avatar
Jesse Beder committed
61
62
63
64

			void Validate();
			void Invalidate();

65
			int pos, line, column, flowLevel;
Jesse Beder's avatar
Jesse Beder committed
66
67
68
69
			bool required;
			Token *pMapStart, *pKey;
		};

Jesse Beder's avatar
Jesse Beder committed
70
		template <typename T> void ScanAndEnqueue(T *pToken);
Jesse Beder's avatar
Jesse Beder committed
71
72
73
74
75
		template <typename T> T *ScanToken(T *pToken);

	private:
		// the stream
		std::istream& INPUT;
Jesse Beder's avatar
Jesse Beder committed
76
		int m_line, m_column;
Jesse Beder's avatar
Jesse Beder committed
77
78
79

		// the output (tokens)
		std::queue <Token *> m_tokens;
Jesse Beder's avatar
Jesse Beder committed
80
		std::set <Token *> m_limboTokens;
Jesse Beder's avatar
Jesse Beder committed
81
82

		// state info
83
		bool m_startedStream, m_endedStream;
Jesse Beder's avatar
Jesse Beder committed
84
85
		bool m_simpleKeyAllowed;
		int m_flowLevel;                // number of unclosed '[' and '{' indicators
86
		bool m_isLastKeyValid;
Jesse Beder's avatar
Jesse Beder committed
87
		std::stack <SimpleKey> m_simpleKeys;
Jesse Beder's avatar
Jesse Beder committed
88
89
90
		std::stack <int> m_indents;
	};
}