scc/PreprocessState.h at master · Vector35/scc · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
#ifndef __PREPROCESSSTATE_H__
#define __PREPROCESSSTATE_H__

#include <string>
#include <map>
#include <vector>
#include <stack>
#include <inttypes.h>
#include "Expr.h"
#include "Struct.h"
#include "Token.h"
#include "Variable.h"
#include "Settings.h"


struct Macro
{
	std::string name;
	bool hasParams;
	std::vector<std::string> params;
	std::vector< Ref<Token> > tokens;
	Location location;
};

struct MacroExpansion
{
	Macro macro;
	size_t parens;
	std::vector< std::vector< Ref<Token> > > params;
	std::vector< Ref<Token> > curParam;
};

class PreprocessState
{
	std::string m_fileName;
	void* m_scanner;
	void* m_lvalue;
	int m_errors;

	std::map<std::string, Macro> m_macros;

	bool m_expansionInProgress;
	MacroExpansion m_expansion;
	std::stack<MacroExpansion> m_expansionStack;

	std::stack<bool> m_ifStack;
	size_t m_ifFailCount;
	size_t m_startingIfStackSize;

	std::string m_output;
	bool m_locationRequest;

	bool m_paste;

	Settings m_settings;

public:
	PreprocessState(const std::string& name, void* scanner, const Settings& settings);
	PreprocessState(PreprocessState& parent, const std::string& name, void* scanner);
	~PreprocessState();

	const std::string& GetFileName() { return m_fileName; }
	int GetLineNumber();
	Location GetLocation();

	const std::string& GetOutput() { return m_output; }
	void Append(const std::string& str);
	void Append(Token* token);
	void AppendLocation();
	void RequestLocation() { m_locationRequest = true; }

	void* GetScanner() { return m_scanner; }
	void* GetLValue() const { return m_lvalue; }
	void SetLValue(void* lvalue) { m_lvalue = lvalue; }

	void Error() { m_errors++; }
	bool HasErrors() const { return m_errors != 0; }

	void IncludeSource(const std::string& source);
	void IncludeFile(const std::string& name);

	void Define(const std::string& name, const std::vector<std::string>& params,
		const std::vector< Ref<Token> >& tokens, bool hasParams);
	void Undefine(const std::string& name);
	bool IsDefined(const std::string& name);

	void BeginMacroExpansion(const std::string& name);
	void FinishMacroExpansion();
	void Finalize();

	void BeginIf(bool result);
	void Else();
	void EndIf();

	void Serialize(OutputBlock* output);
	bool Deserialize(InputBlock* input);

	static bool PreprocessSource(const Settings& settings, const std::string& source, const std::string& fileName,
		std::string& output, PreprocessState* parent = NULL);
};


#endif