Add a new Rewriter::getRangeSize method.
Rename SourceRange::Begin()/End() to getBegin()/getEnd() for
consistency with other code.
Start building the rewriter towards handling @encode.
git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@43047 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/Driver/RewriteTest.cpp b/Driver/RewriteTest.cpp
index 5b27f55..731d6fe 100644
--- a/Driver/RewriteTest.cpp
+++ b/Driver/RewriteTest.cpp
@@ -36,6 +36,9 @@
void HandleDeclInMainFile(Decl *D);
void RewriteInclude(SourceLocation Loc);
+
+ void RewriteFunctionBody(Stmt *S);
+ void RewriteAtEncode(ObjCEncodeExpr *Exp);
~RewriteTest();
};
@@ -86,10 +89,36 @@
/// HandleDeclInMainFile - This is called for each top-level decl defined in the
/// main file of the input.
void RewriteTest::HandleDeclInMainFile(Decl *D) {
+ if (FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
+ if (Stmt *Body = FD->getBody())
+ RewriteFunctionBody(Body);
// Nothing yet.
}
+void RewriteTest::RewriteFunctionBody(Stmt *S) {
+ // Handle specific things.
+ if (ObjCEncodeExpr *AtEncode = dyn_cast<ObjCEncodeExpr>(S))
+ return RewriteAtEncode(AtEncode);
+
+ // Otherwise, just rewrite all children.
+ for (Stmt::child_iterator CI = S->child_begin(), E = S->child_end();
+ CI != E; ++CI)
+ RewriteFunctionBody(*CI);
+}
+
+void RewriteTest::RewriteAtEncode(ObjCEncodeExpr *Exp) {
+#if 0
+ int Size = Rewrite.getRangeSize(Exp->getSourceRange());
+ if (Size == -1) {
+ printf("BLAH!");
+ }
+
+ Rewrite.RemoveText(Exp->getEncLoc(), Size);
+#endif
+}
+
+
RewriteTest::~RewriteTest() {
// Get the top-level buffer that this corresponds to.
std::pair<const char*, const char*> MainBuf = SM->getBufferData(MainFileID);