From 411fcfff58bc49699b525a0e43d75b9f6121da6f Mon Sep 17 00:00:00 2001
From: Stephan Bergmann <sbergman@redhat.com>
Date: Fri, 21 Feb 2014 23:38:54 +0100
Subject: ParentBuilder needs support for ObjCMethodDecl

Change-Id: I8883649da31eabbc5a0f9780fb953ad58f154e1e
---
 compilerplugins/clang/plugin.cxx | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/compilerplugins/clang/plugin.cxx b/compilerplugins/clang/plugin.cxx
index 88841ea308a3..48c98430d082 100644
--- a/compilerplugins/clang/plugin.cxx
+++ b/compilerplugins/clang/plugin.cxx
@@ -60,6 +60,7 @@ const Stmt* Plugin::parentStmt( const Stmt* stmt )
     {
     if( parents.empty())
         buildParents( compiler );
+if(parents.count(stmt)!=1)stmt->dump();
     assert( parents.count( stmt ) == 1 );
     return parents[ stmt ];
     }
@@ -79,6 +80,7 @@ class ParentBuilder
     {
     public:
         bool VisitFunctionDecl( const FunctionDecl* function );
+        bool VisitObjCMethodDecl( const ObjCMethodDecl* method );
         void walk( const Stmt* stmt );
         unordered_map< const Stmt*, const Stmt* >* parents;
     };
@@ -107,6 +109,19 @@ bool ParentBuilder::VisitFunctionDecl( const FunctionDecl* function )
     return true;
     }
 
+bool ParentBuilder::VisitObjCMethodDecl( const ObjCMethodDecl* method )
+    {
+//    if( ignoreLocation( declaration ))
+//        return true; ???
+    if( method->hasBody())
+        {
+        const Stmt* body = method->getBody();
+        (*parents)[ body ] = NULL; // no parent
+        walk( body );
+        }
+    return true;
+    }
+
 void ParentBuilder::walk( const Stmt* stmt )
     {
     for( ConstStmtIterator it = stmt->child_begin();
-- 
cgit