averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-21 12:14:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

Updated the playground transform:

Browse Source 
- Fixed a bug where it did not instrument a "for-in" loop.
    
  - Made sure nested function declarations' bodies are
    instrumented.
   
  - Added instrumentation for return statements.
  
<rdar://problem/16516131>


Swift SVN r15960
This commit is contained in:
Sean Callanan
2014-04-04 22:24:53 +00:00
parent 627534144d
commit 00d6c54241
1 changed files with 43 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
lib/Sema/PlaygroundTransform.cpp
View File

@@ -56,6 +56,8 @@ public:
return transformDoWhileStmt(llvm::cast<DoWhileStmt>(S));
case StmtKind::For:
return transformForStmt(llvm::cast<ForStmt>(S));
case StmtKind::ForEach:
return transformForEachStmt(llvm::cast<ForEachStmt>(S));
}
}
@@ -212,10 +214,39 @@ public:
}
}
} else if (Stmt *S = Element.dyn_cast<Stmt*>()) {
Stmt *NS = transformStmt(S);
if (NS != S) {
Modified = true;
Elements[EI] = NS;
if (ReturnStmt *RS = llvm::dyn_cast<ReturnStmt>(S)) {
if (RS->hasResult()) {
std::pair<PatternBindingDecl *, VarDecl *> PV =
buildPatternAndVariable(RS->getResult());
DeclRefExpr *DRE =
new (Context) DeclRefExpr(ConcreteDeclRef(PV.second),
SourceLoc(),
true, // implicit
false, // uses direct property access
RS->getResult()->getType());
ReturnStmt *NRS = new (Context) ReturnStmt(SourceLoc(),
DRE,
true); // implicit
Expr *Log = buildLoggerCall(
new (Context) DeclRefExpr(ConcreteDeclRef(PV.second),
SourceLoc(),
true, // implicit
false, // uses direct property access
RS->getResult()->getType()),
RS->getResult()->getSourceRange());
Modified = true;
Elements[EI] = PV.first;
Elements.insert(Elements.begin() + (EI + 1), PV.second);
Elements.insert(Elements.begin() + (EI + 2), Log);
Elements.insert(Elements.begin() + (EI + 3), NRS);
EI += 3;
}
} else {
Stmt *NS = transformStmt(S);
if (NS != S) {
Modified = true;
Elements[EI] = NS;
}
}
} else if (Decl *D = Element.dyn_cast<Decl*>()) {
if (VarDecl *VD = llvm::dyn_cast<VarDecl>(D)) {
@@ -226,6 +257,14 @@ public:
++EI;
}
}
else if (FuncDecl *FD = llvm::dyn_cast<FuncDecl>(D)) {
if (BraceStmt *B = FD->getBody()) {
BraceStmt *NB = transformBraceStmt(B);
if (NB != B) {
FD->setBody(NB);
}
}
}
}
}