Add enclosing_class()

1 files changed, 7 insertions, 0 deletions

diff --git a/ast.c b/ast.c
index 96907e3..0418dd6 100644
--- a/ast.c
+++ b/ast.c
@@ -529,6 +529,13 @@ ast_enclosing_function (ast_t *ast)
 	ast, AST_DEFINITION, AST_FUNCTION_DEFINITION);
 }
 
+ast_class_definition_t *
+ast_enclosing_class (ast_t *ast)
+{
+    return (ast_class_definition_t *)enclosing (
+	ast, AST_DEFINITION, AST_CLASS_DEFINITION);
+}
+
 ast_while_statement_t *
 ast_enclosing_while (ast_t *ast)
 {