ignore already seen locations in compiler plugins

Change-Id: Icba8dfa61aee7237af569b2414b1cf4ce93ca476
This commit is contained in:
Luboš Luňák
2013-05-03 16:40:51 +02:00
parent 6f16cfb5a6
commit a0da672521
2 changed files with 16 additions and 6 deletions

View File

@@ -56,12 +56,21 @@ bool Plugin::ignoreLocation( SourceLocation loc )
const char* bufferName = compiler.getSourceManager().getPresumedLoc( expansionLoc ).getFilename();
if( bufferName == NULL )
return true;
if( strncmp( bufferName, OUTDIR, strlen( OUTDIR )) == 0
|| strncmp( bufferName, WORKDIR, strlen( WORKDIR )) == 0
|| strncmp( bufferName, BUILDDIR, strlen( BUILDDIR )) == 0
|| strncmp( bufferName, SRCDIR, strlen( SRCDIR )) == 0 )
return false; // ok
return true;
if( strncmp( bufferName, OUTDIR, strlen( OUTDIR )) != 0
&& strncmp( bufferName, WORKDIR, strlen( WORKDIR )) != 0
&& strncmp( bufferName, BUILDDIR, strlen( BUILDDIR )) != 0
&& strncmp( bufferName, SRCDIR, strlen( SRCDIR )) != 0 )
return true; // not in LO sources
// Sometimes a VisitXXX function may be called more than once for one source location.
// This can happen e.g. with a default argument for a function, if it involves constructing
// a temporary, then this temporary will be constructed whenever the function is called
// and the default argument is needed. As this would mean processing the same piece of code
// more than once, and thus possibly modifying the source code more than once, just
// ignore an already seen location.
if( alreadySeen.find( loc ) != alreadySeen.end())
return true;
alreadySeen.insert( loc );
return false;
}
void Plugin::registerPlugin( Plugin* (*create)( CompilerInstance&, Rewriter& ), const char* optionName, bool isRewriter )

View File

@@ -53,6 +53,7 @@ class Plugin
bool ignoreLocation( const Decl* decl );
bool ignoreLocation( const Stmt* stmt );
CompilerInstance& compiler;
set< SourceLocation > alreadySeen;
private:
static void registerPlugin( Plugin* (*create)( CompilerInstance&, Rewriter& ), const char* optionName, bool isRewriter );
template< typename T > static Plugin* createHelper( CompilerInstance& compiler, Rewriter& rewriter );