Skip to content

Commit 78f0259

Browse files
AnthonyLatsisAnthonyLatsis
authored
Merge pull request swiftlang#33283 from AnthonyLatsis/type-locura-2
ASTWalker: Offload TypeLoc
2 parents 9f8ba0a + 17f12da commit 78f0259

16 files changed

+34
-83
lines changed

‎include/swift/AST/ASTWalker.h

Lines changed: 0 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,6 @@ class ModuleDecl;
2525
class Stmt;
2626
class Pattern;
2727
class TypeRepr;
28-
class TypeLoc;
2928
class ParameterList;
3029
enum class AccessKind: unsigned char;
3130

@@ -177,19 +176,6 @@ class ASTWalker {
177176
/// returns failure.
178177
virtual bool walkToDeclPost(Decl *D) { return true; }
179178

180-
/// This method is called when first visiting a TypeLoc, before
181-
/// walking into its TypeRepr children. If it returns false, the subtree is
182-
/// skipped.
183-
///
184-
/// \param TL The TypeLoc to check.
185-
virtual bool walkToTypeLocPre(TypeLoc &TL) { return true; }
186-
187-
/// This method is called after visiting the children of a TypeLoc.
188-
/// If it returns false, the remaining traversal is terminated and returns
189-
/// failure.
190-
virtual bool walkToTypeLocPost(TypeLoc &TL) { return true; }
191-
192-
193179
/// This method is called when first visiting a TypeRepr, before
194180
/// walking into its children. If it returns false, the subtree is skipped.
195181
///

‎include/swift/AST/Pattern.h

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,6 @@ namespace swift {
3434
class Expr;
3535
enum class CheckedCastKind : unsigned;
3636
class TypeExpr;
37-
class TypeLoc;
3837

3938
/// PatternKind - The classification of different kinds of
4039
/// value-matching pattern.
@@ -447,7 +446,6 @@ class TypedPattern : public Pattern {
447446

448447
TypeRepr *getTypeRepr() const { return PatTypeRepr; }
449448

450-
TypeLoc getTypeLoc() const;
451449
SourceLoc getLoc() const;
452450
SourceRange getSourceRange() const;
453451

‎include/swift/IDE/Utils.h

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -280,8 +280,6 @@ class NameMatcher: public ASTWalker {
280280
bool walkToDeclPost(Decl *D) override;
281281
std::pair<bool, Stmt*> walkToStmtPre(Stmt *S) override;
282282
Stmt* walkToStmtPost(Stmt *S) override;
283-
bool walkToTypeLocPre(TypeLoc &TL) override;
284-
bool walkToTypeLocPost(TypeLoc &TL) override;
285283
bool walkToTypeReprPre(TypeRepr *T) override;
286284
bool walkToTypeReprPost(TypeRepr *T) override;
287285
std::pair<bool, Pattern*> walkToPatternPre(Pattern *P) override;

‎lib/AST/ASTPrinter.cpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1079,7 +1079,9 @@ void PrintAST::printTypedPattern(const TypedPattern *TP) {
10791079
if (auto decl = named->getDecl())
10801080
isIUO = decl->isImplicitlyUnwrappedOptional();
10811081

1082-
printTypeLocForImplicitlyUnwrappedOptional(TP->getTypeLoc(), isIUO);
1082+
const auto TyLoc = TypeLoc(TP->getTypeRepr(),
1083+
TP->hasType() ? TP->getType() : Type());
1084+
printTypeLocForImplicitlyUnwrappedOptional(TyLoc, isIUO);
10831085
}
10841086

10851087
/// Determines if we are required to print the name of a property declaration,

‎lib/AST/ASTScopeCreation.cpp

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1783,7 +1783,6 @@ void ScopeCreator::forEachClosureIn(
17831783
return {false, P};
17841784
}
17851785
bool walkToDeclPre(Decl *D) override { return false; }
1786-
bool walkToTypeLocPre(TypeLoc &TL) override { return false; }
17871786
bool walkToTypeReprPre(TypeRepr *T) override { return false; }
17881787
bool walkToParameterListPre(ParameterList *PL) override { return false; }
17891788
};

‎lib/AST/ASTWalker.cpp

Lines changed: 19 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -155,8 +155,9 @@ class Traversal : public ASTVisitor<Traversal, Expr*, Stmt*,
155155
if (doIt(typeRepr))
156156
return true;
157157
for (auto &Inherit : ED->getInherited()) {
158-
if (doIt(Inherit))
159-
return true;
158+
if (auto *const TyR = Inherit.getTypeRepr())
159+
if (doIt(TyR))
160+
return true;
160161
}
161162
if (visitTrailingRequirements(ED))
162163
return true;
@@ -258,9 +259,10 @@ class Traversal : public ASTVisitor<Traversal, Expr*, Stmt*,
258259
}
259260

260261
bool visitAbstractTypeParamDecl(AbstractTypeParamDecl *TPD) {
261-
for (auto Inherit: TPD->getInherited()) {
262-
if (doIt(Inherit))
263-
return true;
262+
for (const auto &Inherit: TPD->getInherited()) {
263+
if (auto *const TyR = Inherit.getTypeRepr())
264+
if (doIt(TyR))
265+
return true;
264266
}
265267

266268
if (const auto ATD = dyn_cast<AssociatedTypeDecl>(TPD)) {
@@ -282,9 +284,10 @@ class Traversal : public ASTVisitor<Traversal, Expr*, Stmt*,
282284

283285
bool WalkGenerics = visitGenericParamListIfNeeded(NTD);
284286

285-
for (auto &Inherit : NTD->getInherited()) {
286-
if (doIt(Inherit))
287-
return true;
287+
for (const auto &Inherit : NTD->getInherited()) {
288+
if (auto *const TyR = Inherit.getTypeRepr())
289+
if (doIt(Inherit.getTypeRepr()))
290+
return true;
288291
}
289292

290293
// Visit requirements
@@ -355,8 +358,9 @@ class Traversal : public ASTVisitor<Traversal, Expr*, Stmt*,
355358
bool WalkGenerics = visitGenericParamListIfNeeded(SD);
356359

357360
visit(SD->getIndices());
358-
if (doIt(SD->getElementTypeLoc()))
359-
return true;
361+
if (auto *const TyR = SD->getElementTypeLoc().getTypeRepr())
362+
if (doIt(TyR))
363+
return true;
360364

361365
// Visit trailing requirements
362366
if (WalkGenerics && visitTrailingRequirements(SD))
@@ -388,10 +392,12 @@ class Traversal : public ASTVisitor<Traversal, Expr*, Stmt*,
388392
visit(PD);
389393
visit(AFD->getParameters());
390394

391-
if (auto *FD = dyn_cast<FuncDecl>(AFD))
395+
if (auto *FD = dyn_cast<FuncDecl>(AFD)) {
392396
if (!isa<AccessorDecl>(FD))
393-
if (doIt(FD->getBodyResultTypeLoc()))
394-
return true;
397+
if (auto *const TyR = FD->getBodyResultTypeLoc().getTypeRepr())
398+
if (doIt(TyR))
399+
return true;
400+
}
395401

396402
// Visit trailing requirements
397403
if (WalkGenerics && visitTrailingRequirements(AFD))
@@ -1320,21 +1326,6 @@ class Traversal : public ASTVisitor<Traversal, Expr*, Stmt*,
13201326
return false;
13211327
}
13221328

1323-
bool doIt(TypeLoc &TL) {
1324-
if (!Walker.walkToTypeLocPre(TL))
1325-
return false;
1326-
1327-
// No "visit" since TypeLocs are not a class hierarchy. Clients can do what
1328-
// they want in walkToTypeLocPre.
1329-
1330-
if (auto typerepr = TL.getTypeRepr())
1331-
if (doIt(typerepr))
1332-
return true;
1333-
1334-
// If we didn't bail out, do post-order visitation.
1335-
return !Walker.walkToTypeLocPost(TL);
1336-
}
1337-
13381329
/// Returns true on failure.
13391330
bool doIt(TypeRepr *T) {
13401331
// Do the pre-order visitation. If it returns false, we just

‎lib/AST/Expr.cpp

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -423,7 +423,6 @@ forEachImmediateChildExpr(llvm::function_ref<Expr *(Expr *)> callback) {
423423
}
424424
bool walkToDeclPre(Decl *D) override { return false; }
425425
bool walkToTypeReprPre(TypeRepr *T) override { return false; }
426-
bool walkToTypeLocPre(TypeLoc &TL) override { return false; }
427426
};
428427

429428
this->walk(ChildWalker(callback, this));
@@ -453,7 +452,6 @@ void Expr::forEachChildExpr(llvm::function_ref<Expr *(Expr *)> callback) {
453452
}
454453
bool walkToDeclPre(Decl *D) override { return false; }
455454
bool walkToTypeReprPre(TypeRepr *T) override { return false; }
456-
bool walkToTypeLocPre(TypeLoc &TL) override { return false; }
457455
};
458456

459457
this->walk(ChildWalker(callback));

‎lib/AST/Pattern.cpp

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -187,7 +187,6 @@ namespace {
187187
std::pair<bool, Stmt *> walkToStmtPre(Stmt *S) override {
188188
return { false, S };
189189
}
190-
bool walkToTypeLocPre(TypeLoc &TL) override { return false; }
191190
bool walkToTypeReprPre(TypeRepr *T) override { return false; }
192191
bool walkToParameterListPre(ParameterList *PL) override { return false; }
193192
bool walkToDeclPre(Decl *D) override { return false; }
@@ -407,15 +406,6 @@ TypedPattern::TypedPattern(Pattern *pattern, TypeRepr *tr)
407406
Bits.TypedPattern.IsPropagatedType = false;
408407
}
409408

410-
TypeLoc TypedPattern::getTypeLoc() const {
411-
TypeLoc loc = TypeLoc(PatTypeRepr);
412-
413-
if (hasType())
414-
loc.setType(getType());
415-
416-
return loc;
417-
}
418-
419409
SourceLoc TypedPattern::getLoc() const {
420410
if (SubPattern->isImplicit() && PatTypeRepr)
421411
return PatTypeRepr->getSourceRange().Start;

‎lib/IDE/ExprContextAnalysis.cpp

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,6 @@ class ExprFinder : public ASTWalker {
177177
return {isInterstingRange(S), S};
178178
}
179179

180-
bool walkToTypeLocPre(TypeLoc &TL) override { return false; }
181180
bool walkToTypeReprPre(TypeRepr *T) override { return false; }
182181
};
183182
} // anonymous namespace

‎lib/IDE/SwiftSourceDocInfo.cpp

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -468,16 +468,6 @@ Expr *NameMatcher::walkToExprPost(Expr *E) {
468468
return E;
469469
}
470470

471-
bool NameMatcher::walkToTypeLocPre(TypeLoc &TL) {
472-
if (isDone() || shouldSkip(TL.getSourceRange()))
473-
return false;
474-
return true;
475-
}
476-
477-
bool NameMatcher::walkToTypeLocPost(TypeLoc &TL) {
478-
return !isDone();
479-
}
480-
481471
bool NameMatcher::walkToTypeReprPre(TypeRepr *T) {
482472
if (isDone() || shouldSkip(T->getSourceRange()))
483473
return false;

0 commit comments

Comments
 (0)