source: src/Common/CodeLocationTools.cpp@ e9e9f56

ADT ast-experimental pthread-emulation
Last change on this file since e9e9f56 was b0d9ff7, checked in by JiadaL <j82liang@…>, 3 years ago

Fix up the QualifiedNameExpr. It should now work on both old AST and new AST. There are some known bugs to fix so make all-tests will fail.
  • Property mode set to 100644
File size: 8.0 KB
Line 
1//
2// Cforall Version 1.0.0 Copyright (C) 2015 University of Waterloo
3//
4// The contents of this file are covered under the licence agreement in the
5// file "LICENCE" distributed with Cforall.
6//
7// CodeLocationTools.cpp -- Additional tools for code locations.
8//
9// Author : Andrew Beach
10// Created On : Fri Dec 4 15:42:00 2020
11// Last Modified By : Andrew Beach
12// Last Modified On : Wed May 11 16:16:00 2022
13// Update Count : 5
14//
15
16#include "CodeLocationTools.hpp"
17
18#include <type_traits>
19
20#include "AST/Pass.hpp"
21#include "AST/TranslationUnit.hpp"
22#include "Common/CodeLocation.h"
23
24namespace {
25
26// Fill every location with a nearby (parent) location.
27class FillCore : public ast::WithGuards {
28 CodeLocation const * parent;
29
30 template<typename node_t>
31 node_t const * parse_visit( node_t const * node ) {
32 if ( node->location.isUnset() ) {
33 assert( parent );
34 node_t * newNode = ast::mutate( node );
35 newNode->location = *parent;
36 return newNode;
37 }
38 GuardValue( parent ) = &node->location;
39 return node;
40 }
41
42 bool hasUnsetLabels( const ast::Stmt * stmt ) {
43 for ( const ast::Label& label : stmt->labels ) {
44 if ( label.location.isUnset() ) {
45 return true;
46 }
47 }
48 return false;
49 }
50
51 template<typename node_t>
52 node_t const * stmt_visit( node_t const * node ) {
53 assert( node->location.isSet() );
54
55 if ( hasUnsetLabels( node ) ) {
56 node_t * newNode = ast::mutate( node );
57 for ( ast::Label& label : newNode->labels ) {
58 if ( label.location.isUnset() ) {
59 label.location = newNode->location;
60 }
61 }
62 return newNode;
63 }
64 return node;
65 }
66
67 template<typename node_t>
68 auto visit( node_t const * node, long ) {
69 return node;
70 }
71
72 template<typename node_t>
73 auto visit( node_t const * node, int ) -> typename
74 std::remove_reference< decltype( node->location, node ) >::type {
75 return parse_visit( node );
76 }
77
78 template<typename node_t>
79 auto visit( node_t const * node, char ) -> typename
80 std::remove_reference< decltype( node->labels, node ) >::type {
81 return stmt_visit( parse_visit( node ) );
82 }
83
84public:
85 FillCore() : parent( nullptr ) {}
86 FillCore( const CodeLocation& location ) : parent( &location ) {
87 assert( location.isSet() );
88 }
89
90 template<typename node_t>
91 node_t const * previsit( node_t const * node ) {
92 return visit( node, '\0' );
93 }
94};
95
96// ALL_VISITS(macro)
97// Expands `macro(node_type, return_type)` for every visit method of the
98// ast::Visitor class where node_type is the name of the parameter and
99// return_type is the name of the return type; not including the namespace,
100// pointer or const qualifiers.
101#define ALL_VISITS(macro) \
102 macro(ObjectDecl, DeclWithType) \
103 macro(FunctionDecl, DeclWithType) \
104 macro(StructDecl, Decl) \
105 macro(UnionDecl, Decl) \
106 macro(EnumDecl, Decl) \
107 macro(TraitDecl, Decl) \
108 macro(TypeDecl, Decl) \
109 macro(TypedefDecl, Decl) \
110 macro(AsmDecl, AsmDecl) \
111 macro(DirectiveDecl, DirectiveDecl) \
112 macro(StaticAssertDecl, StaticAssertDecl) \
113 macro(CompoundStmt, CompoundStmt) \
114 macro(ExprStmt, Stmt) \
115 macro(AsmStmt, Stmt) \
116 macro(DirectiveStmt, Stmt) \
117 macro(IfStmt, Stmt) \
118 macro(WhileDoStmt, Stmt) \
119 macro(ForStmt, Stmt) \
120 macro(SwitchStmt, Stmt) \
121 macro(CaseClause, CaseClause) \
122 macro(BranchStmt, Stmt) \
123 macro(ReturnStmt, Stmt) \
124 macro(ThrowStmt, Stmt) \
125 macro(TryStmt, Stmt) \
126 macro(CatchClause, CatchClause) \
127 macro(FinallyClause, FinallyClause) \
128 macro(SuspendStmt, Stmt) \
129 macro(WaitForStmt, Stmt) \
130 macro(WaitForClause, WaitForClause) \
131 macro(WithStmt, Decl) \
132 macro(NullStmt, NullStmt) \
133 macro(DeclStmt, Stmt) \
134 macro(ImplicitCtorDtorStmt, Stmt) \
135 macro(MutexStmt, Stmt) \
136 macro(ApplicationExpr, Expr) \
137 macro(UntypedExpr, Expr) \
138 macro(NameExpr, Expr) \
139 macro(QualifiedNameExpr, Expr) \
140 macro(AddressExpr, Expr) \
141 macro(LabelAddressExpr, Expr) \
142 macro(CastExpr, Expr) \
143 macro(KeywordCastExpr, Expr) \
144 macro(VirtualCastExpr, Expr) \
145 macro(UntypedMemberExpr, Expr) \
146 macro(MemberExpr, Expr) \
147 macro(VariableExpr, Expr) \
148 macro(ConstantExpr, Expr) \
149 macro(SizeofExpr, Expr) \
150 macro(AlignofExpr, Expr) \
151 macro(UntypedOffsetofExpr, Expr) \
152 macro(OffsetofExpr, Expr) \
153 macro(OffsetPackExpr, Expr) \
154 macro(LogicalExpr, Expr) \
155 macro(ConditionalExpr, Expr) \
156 macro(CommaExpr, Expr) \
157 macro(TypeExpr, Expr) \
158 macro(DimensionExpr, Expr) \
159 macro(AsmExpr, Expr) \
160 macro(ImplicitCopyCtorExpr, Expr) \
161 macro(ConstructorExpr, Expr) \
162 macro(CompoundLiteralExpr, Expr) \
163 macro(RangeExpr, Expr) \
164 macro(UntypedTupleExpr, Expr) \
165 macro(TupleExpr, Expr) \
166 macro(TupleIndexExpr, Expr) \
167 macro(TupleAssignExpr, Expr) \
168 macro(StmtExpr, Expr) \
169 macro(UniqueExpr, Expr) \
170 macro(UntypedInitExpr, Expr) \
171 macro(InitExpr, Expr) \
172 macro(DeletedExpr, Expr) \
173 macro(DefaultArgExpr, Expr) \
174 macro(GenericExpr, Expr) \
175 macro(VoidType, Type) \
176 macro(BasicType, Type) \
177 macro(PointerType, Type) \
178 macro(ArrayType, Type) \
179 macro(ReferenceType, Type) \
180 macro(QualifiedType, Type) \
181 macro(FunctionType, Type) \
182 macro(StructInstType, Type) \
183 macro(UnionInstType, Type) \
184 macro(EnumInstType, Type) \
185 macro(TraitInstType, Type) \
186 macro(TypeInstType, Type) \
187 macro(TupleType, Type) \
188 macro(TypeofType, Type) \
189 macro(VTableType, Type) \
190 macro(VarArgsType, Type) \
191 macro(ZeroType, Type) \
192 macro(OneType, Type) \
193 macro(GlobalScopeType, Type) \
194 macro(Designation, Designation) \
195 macro(SingleInit, Init) \
196 macro(ListInit, Init) \
197 macro(ConstructorInit, Init) \
198 macro(Attribute, Attribute) \
199 macro(TypeSubstitution, TypeSubstitution)
200
201// These could even go into the ast namespace.
202enum class LeafKind {
203#define VISIT(node_type, return_type) node_type,
204 ALL_VISITS(VISIT)
205#undef VISIT
206};
207
208struct LeafKindVisitor : public ast::Visitor {
209 LeafKind kind;
210
211#define VISIT(node_type, return_type) \
212 const ast::return_type * visit( const ast::node_type * ) final { \
213 kind = LeafKind::node_type; \
214 return nullptr; \
215 }
216 ALL_VISITS(VISIT)
217#undef VISIT
218};
219
220constexpr size_t leaf_kind_count = (1 + (size_t)LeafKind::TypeSubstitution);
221
222LeafKind get_leaf_kind( ast::Node const * node ) {
223 LeafKindVisitor visitor;
224 node->accept( visitor );
225 return visitor.kind;
226}
227
228const char * leaf_kind_names[leaf_kind_count] = {
229#define VISIT(node_type, return_type) #node_type,
230 ALL_VISITS(VISIT)
231#undef VISIT
232};
233
234// Collect pointers to all the nodes with unset code locations.
235class CollectCore {
236 std::list< ast::ptr< ast::Node > > & unset;
237public:
238 CollectCore( std::list< ast::ptr< ast::Node > > & unset ) :
239 unset( unset )
240 {}
241
242 template<typename node_t>
243 auto previsit( node_t const * node ) -> decltype( node->location, void() ) {
244 if ( node->location.isUnset() ) {
245 unset.push_back( node );
246 }
247 }
248};
249
250} // namespace
251
252void checkAllCodeLocations( ast::TranslationUnit const & unit ) {
253 checkAllCodeLocations( "unknown location", unit );
254}
255
256void checkAllCodeLocations( char const * label, ast::TranslationUnit const & unit ) {
257 std::list< ast::ptr< ast::Node > > unsetNodes;
258 {
259 ast::Pass<CollectCore> collector( unsetNodes );
260 for ( auto node : unit.decls ) {
261 node->accept( collector );
262 }
263 }
264 if ( unsetNodes.empty() ) {
265 return;
266 }
267
268 std::cerr << "Code Location check at " << label << " failed." << std::endl;
269 std::cerr << "Total nodes without a set code location: "
270 << unsetNodes.size() << std::endl;
271
272 size_t node_counts[leaf_kind_count] = {0};
273 for ( auto unset : unsetNodes ) {
274 node_counts[(size_t)get_leaf_kind(unset)] += 1;
275 }
276 for ( size_t i = 0 ; i < leaf_kind_count ; ++i ) {
277 if ( node_counts[i] ) {
278 std::cerr << '\t' << node_counts[i]
279 << " of type " << leaf_kind_names[i] << std::endl;
280 }
281 }
282
283 assert( unsetNodes.empty() );
284}
285
286void forceFillCodeLocations( ast::TranslationUnit & unit ) {
287 ast::Pass<FillCore>::run( unit );
288}
289
290ast::Node const * localFillCodeLocations(
291 CodeLocation const & location , ast::Node const * node ) {
292 ast::Pass<FillCore> visitor( location );
293 return node->accept( visitor );
294}
Note: See TracBrowser for help on using the repository browser.