node.h 2.23 KB
Newer Older
Jesse Beder's avatar
Jesse Beder committed
1
2
3
#pragma once

#include <string>
4
#include <ios>
5
6
#include <vector>
#include <map>
7
#include "parserstate.h"
8
#include "exceptions.h"
9
#include "iterator.h"
Jesse Beder's avatar
Jesse Beder committed
10
11
12

namespace YAML
{
13
	class Content;
14
	class Scanner;
Jesse Beder's avatar
Jesse Beder committed
15

Jesse Beder's avatar
Jesse Beder committed
16
17
	enum CONTENT_TYPE { CT_NONE, CT_SCALAR, CT_SEQUENCE, CT_MAP };

Jesse Beder's avatar
Jesse Beder committed
18
19
20
21
22
23
	class Node
	{
	public:
		Node();
		~Node();

24
		void Clear();
25
		void Parse(Scanner *pScanner, const ParserState& state);
26
		void Write(std::ostream& out, int indent, bool startedLine, bool onlyOneCharOnLine) const;
27

Jesse Beder's avatar
Jesse Beder committed
28
29
		CONTENT_TYPE GetType() const;

30
		// accessors
31
32
		Iterator begin() const;
		Iterator end() const;
33
		unsigned size() const;
34
35

		template <typename T>
36
		const Node& GetValue(const T& key) const {
37
38
39
40
41
42
43
44
45
			if(!m_pContent)
				throw BadDereference();

			for(Iterator it=begin();it!=end();++it) {
				T t;
				try {
					it.first() >> t;
					if(key == t)
						return it.second();
Jesse Beder's avatar
Jesse Beder committed
46
				} catch(RepresentationException&) {
47
48
49
50
51
52
				}
			}

			throw BadDereference();
		}

53
54
55
56
57
		template <typename T>
		const Node& operator [] (const T& key) const {
			return GetValue(key);
		}

58
		const Node& operator [] (const char *key) const {
59
			return GetValue(std::string(key));
60
61
		}

62
63
64
		const Node& operator [] (unsigned u) const;
		const Node& operator [] (int i) const;

65
66
67
68
69
70
71
72
73
		// extraction
		friend void operator >> (const Node& node, std::string& s);
		friend void operator >> (const Node& node, int& i);
		friend void operator >> (const Node& node, unsigned& u);
		friend void operator >> (const Node& node, long& l);
		friend void operator >> (const Node& node, float& f);
		friend void operator >> (const Node& node, double& d);
		friend void operator >> (const Node& node, char& c);

74
75
76
		// insertion
		friend std::ostream& operator << (std::ostream& out, const Node& node);

77
78
79
80
		// ordering
		int Compare(const Node& rhs) const;
		friend bool operator < (const Node& n1, const Node& n2);

81
	private:
82
83
84
85
		void ParseHeader(Scanner *pScanner, const ParserState& state);
		void ParseTag(Scanner *pScanner, const ParserState& state);
		void ParseAnchor(Scanner *pScanner, const ParserState& state);
		void ParseAlias(Scanner *pScanner, const ParserState& state);
86

Jesse Beder's avatar
Jesse Beder committed
87
	private:
88
		std::string m_anchor, m_tag;
89
		Content *m_pContent;
Jesse Beder's avatar
Jesse Beder committed
90
		bool m_alias;
Jesse Beder's avatar
Jesse Beder committed
91
92
	};
}