Added "Decl::getCodyBody()", a virtual method that returns the root AST node (Stmt*) that the Decl wraps (if any). Currently this only returns a non-null value for FunctionDecl and ObjCMethodDecl.
git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@52552 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/include/clang/AST/Decl.h b/include/clang/AST/Decl.h
index bf7efe8..a3713d5 100644
--- a/include/clang/AST/Decl.h
+++ b/include/clang/AST/Decl.h
@@ -438,6 +438,10 @@
return getBody(Definition);
}
+ virtual Stmt* getCodeBody() const {
+ return getBody();
+ }
+
/// isThisDeclarationADefinition - Returns whether this specific
/// declaration of the function is also a definition. This does not
/// determine whether the function has been defined (e.g., in a
diff --git a/include/clang/AST/DeclBase.h b/include/clang/AST/DeclBase.h
index 55c2e61..69db09c 100644
--- a/include/clang/AST/DeclBase.h
+++ b/include/clang/AST/DeclBase.h
@@ -195,6 +195,12 @@
return IdentifierNamespace(IDNS_Tag | IDNS_Ordinary);
}
}
+
+ // getCodeBody - If this Decl represents a declaration for a body of code,
+ // such as a function or method definition, this method returns the top-level
+ // Stmt* of that body. Otherwise this method returns null.
+ virtual Stmt* getCodeBody() const { return 0; }
+
// global temp stats (until we have a per-module visitor)
static void addDeclKind(Kind k);
static bool CollectingStats(bool Enable = false);
diff --git a/include/clang/AST/DeclObjC.h b/include/clang/AST/DeclObjC.h
index 093a5cf..ed09eac 100644
--- a/include/clang/AST/DeclObjC.h
+++ b/include/clang/AST/DeclObjC.h
@@ -185,9 +185,11 @@
ImplementationControl getImplementationControl() const {
return ImplementationControl(DeclImplementation);
}
- Stmt *getBody() { return Body; }
- const Stmt *getBody() const { return Body; }
+
+ Stmt *getBody() const { return Body; }
void setBody(Stmt *B) { Body = B; }
+
+ virtual Stmt* getCodeBody() const { return getBody(); }
// Implement isa/cast/dyncast/etc.
static bool classof(const Decl *D) { return D->getKind() == ObjCMethod; }