ParentBuilder needs support for ObjCMethodDecl

Change-Id: I8883649da31eabbc5a0f9780fb953ad58f154e1e
This commit is contained in:
Stephan Bergmann
2014-02-21 23:38:54 +01:00
parent 278409c3f1
commit 411fcfff58

View File

@@ -60,6 +60,7 @@ const Stmt* Plugin::parentStmt( const Stmt* stmt )
{ {
if( parents.empty()) if( parents.empty())
buildParents( compiler ); buildParents( compiler );
if(parents.count(stmt)!=1)stmt->dump();
assert( parents.count( stmt ) == 1 ); assert( parents.count( stmt ) == 1 );
return parents[ stmt ]; return parents[ stmt ];
} }
@@ -79,6 +80,7 @@ class ParentBuilder
{ {
public: public:
bool VisitFunctionDecl( const FunctionDecl* function ); bool VisitFunctionDecl( const FunctionDecl* function );
bool VisitObjCMethodDecl( const ObjCMethodDecl* method );
void walk( const Stmt* stmt ); void walk( const Stmt* stmt );
unordered_map< const Stmt*, const Stmt* >* parents; unordered_map< const Stmt*, const Stmt* >* parents;
}; };
@@ -107,6 +109,19 @@ bool ParentBuilder::VisitFunctionDecl( const FunctionDecl* function )
return true; 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 ) void ParentBuilder::walk( const Stmt* stmt )
{ {
for( ConstStmtIterator it = stmt->child_begin(); for( ConstStmtIterator it = stmt->child_begin();