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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
|
#ifndef _CHILLAST_DEF_H_
#define _CHILLAST_DEF_H_
#include "chilldebug.h"
#define CHILL_INDENT_AMOUNT 2
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <assert.h>
#include <vector> // std::vector
#include <ir_enums.hh> // for IR_CONDITION_*
#define CHILLAST_NODE_FORSTMT CHILLAST_NODE_LOOP
#define CHILLAST_NODE_TRANSLATIONUNIT CHILLAST_NODE_SOURCEFILE
enum CHILLAST_NODE_TYPE {
CHILLAST_NODE_UNKNOWN = 0,
CHILLAST_NODE_SOURCEFILE,
CHILLAST_NODE_TYPEDEFDECL,
CHILLAST_NODE_VARDECL,
// CHILLAST_NODE_PARMVARDECL, not used any more
CHILLAST_NODE_FUNCTIONDECL,
CHILLAST_NODE_RECORDDECL, // struct or union (or class)
CHILLAST_NODE_MACRODEFINITION,
CHILLAST_NODE_COMPOUNDSTMT,
CHILLAST_NODE_LOOP, // AKA ForStmt
CHILLAST_NODE_TERNARYOPERATOR,
CHILLAST_NODE_BINARYOPERATOR,
CHILLAST_NODE_UNARYOPERATOR,
CHILLAST_NODE_ARRAYSUBSCRIPTEXPR,
CHILLAST_NODE_MEMBEREXPR, // structs/unions
CHILLAST_NODE_DECLREFEXPR,
CHILLAST_NODE_INTEGERLITERAL,
CHILLAST_NODE_FLOATINGLITERAL,
CHILLAST_NODE_IMPLICITCASTEXPR,
CHILLAST_NODE_RETURNSTMT,
CHILLAST_NODE_CALLEXPR,
CHILLAST_NODE_DECLSTMT,
CHILLAST_NODE_PARENEXPR,
CHILLAST_NODE_CSTYLECASTEXPR,
CHILLAST_NODE_CSTYLEADDRESSOF,
CHILLAST_NODE_IFSTMT,
CHILLAST_NODE_SIZEOF,
CHILLAST_NODE_MALLOC,
CHILLAST_NODE_FREE,
CHILLAST_NODE_PREPROCESSING, // comments, #define, #include, whatever else works
CHILLAST_NODE_NOOP, // NO OP
// CUDA specific
CHILLAST_NODE_CUDAMALLOC,
CHILLAST_NODE_CUDAFREE,
CHILLAST_NODE_CUDAMEMCPY,
CHILLAST_NODE_CUDAKERNELCALL,
CHILLAST_NODE_CUDASYNCTHREADS,
CHILLAST_NODE_NULL // explicit non-statement
// TODO
};
enum CHILLAST_FUNCTION_TYPE {
CHILLAST_FUNCTION_CPU = 0,
CHILLAST_FUNCTION_GPU
};
enum CHILLAST_MEMBER_EXP_TYPE {
CHILLAST_MEMBER_EXP_DOT = 0,
CHILLAST_MEMBER_EXP_ARROW
};
enum CHILLAST_PREPROCESSING_TYPE {
CHILLAST_PREPROCESSING_TYPEUNKNOWN = 0,
CHILLAST_PREPROCESSING_COMMENT,
CHILLAST_PREPROCESSING_POUNDDEFINE,
CHILLAST_PREPROCESSING_POUNDINCLUDE,
CHILLAST_PREPROCESSING_PRAGMA // unused so far
};
enum CHILLAST_PREPROCESSING_POSITION { // when tied to another statement
CHILLAST_PREPROCESSING_POSITIONUNKNOWN = 0,
CHILLAST_PREPROCESSING_LINEBEFORE, // previous line
CHILLAST_PREPROCESSING_LINEAFTER, // next line
CHILLAST_PREPROCESSING_TOTHERIGHT, // for this kind of comment, on same line
CHILLAST_PREPROCESSING_IMMEDIATELYBEFORE // on same line
};
char *parseUnderlyingType(char *sometype);
char *parseArrayParts(char *sometype);
bool isRestrict(const char *sometype);
char *splitTypeInfo(char *underlyingtype);
//! change "1024UL" to "1024"
char *ulhack(char *brackets);
//! remove __restrict__ , MODIFIES the argument!
char *restricthack(char *typeinfo);
// TODO phase out this one unnecessary as we can write a virtual function that does this
extern const char *ChillAST_Node_Names[]; // WARNING MUST BE KEPT IN SYNC WITH BELOW LIST
// fwd declarations
//! the generic node, who specific types are derived from
class chillAST_Node;
//! empty node
class chillAST_NULL;
//! ast for an entire source file (translationunit)
class chillAST_SourceFile;
class chillAST_TypedefDecl;
class chillAST_VarDecl;
//class chillAST_ParmVarDecl;
class chillAST_FunctionDecl;
//! structs and unions (and classes?)
class chillAST_RecordDecl;
class chillAST_MacroDefinition;
//! A sequence of statements
class chillAST_CompoundStmt;
//! a For LOOP
class chillAST_ForStmt;
class chillAST_UnaryOperator;
class chillAST_BinaryOperator;
class chillAST_TernaryOperator;
class chillAST_ArraySubscriptExpr;
class chillAST_MemberExpr;
class chillAST_DeclRefExpr;
class chillAST_IntegerLiteral;
class chillAST_FloatingLiteral;
class chillAST_ImplicitCastExpr;
class chillAST_CStyleCastExpr;
class chillAST_CStyleAddressOf;
class chillAST_ReturnStmt;
class chillAST_CallExpr;
class chillAST_ParenExpr;
class chillAST_Sizeof;
class chillAST_Malloc;
class chillAST_Free;
class chillAST_NoOp;
class chillAST_CudaMalloc;
class chillAST_CudaFree;
class chillAST_CudaMemcpy;
class chillAST_CudaKernelCall;
class chillAST_CudaSyncthreads;
class chillAST_Preprocessing;
typedef std::vector<chillAST_Node *> chillAST_NodeList;
typedef std::vector<chillAST_VarDecl *> chillAST_SymbolTable; // typedef
typedef std::vector<chillAST_TypedefDecl *> chillAST_TypedefTable; // typedef
chillAST_VarDecl *symbolTableFindName(chillAST_SymbolTable *table, const char *name); // fwd decl
chillAST_VarDecl *symbolTableFindVariableNamed(chillAST_SymbolTable *table,
const char *name); // fwd decl TODO too many similar named functions
void printSymbolTable(chillAST_SymbolTable *st); // fwd decl
void printSymbolTableMoreInfo(chillAST_SymbolTable *st); // fwd decl
chillAST_Node *lessthanmacro(chillAST_Node *left, chillAST_Node *right); // fwd declaration
chillAST_SymbolTable *addSymbolToTable(chillAST_SymbolTable *st, chillAST_VarDecl *vd); // fwd decl
chillAST_TypedefTable *addTypedefToTable(chillAST_TypedefTable *tt, chillAST_TypedefDecl *td); // fwd decl
bool streq(const char *a, const char *b); // fwd decl
void chillindent(int i, FILE *fp); // fwd declaration
void insertNewDeclAtLocationOfOldIfNeeded(chillAST_VarDecl *newdecl, chillAST_VarDecl *olddecl);
chillAST_DeclRefExpr *buildDeclRefExpr(chillAST_VarDecl *);
chillAST_FunctionDecl *findFunctionDecl(chillAST_Node *node, const char *procname);
#endif
|
