mirror of
https://github.com/Gericom/teak-llvm.git
synced 2025-06-20 03:55:48 -04:00
1a2676924a
Summary: This moves the data collection macro calls for Stmt nodes to lib/AST/StmtDataCollectors.inc Users can subclass ConstStmtVisitor and include StmtDataCollectors.inc to define visitor methods for each Stmt subclass. This makes it also possible to customize the visit methods as exemplified in lib/Analysis/CloneDetection.cpp. Move helper methods for data collection to a new module, AST/DataCollection. Add data collection for DeclRefExpr, MemberExpr and some literals. Reviewers: arphaman, teemperor! Subscribers: mgorny, xazax.hun, cfe-commits Differential Revision: https://reviews.llvm.org/D36664 llvm-svn: 311569
51 lines
1.8 KiB
C++
51 lines
1.8 KiB
C++
//===-- DataCollection.cpp --------------------------------------*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "clang/AST/DataCollection.h"
|
|
|
|
#include "clang/Lex/Lexer.h"
|
|
|
|
namespace clang {
|
|
namespace data_collection {
|
|
|
|
/// Prints the macro name that contains the given SourceLocation into the given
|
|
/// raw_string_ostream.
|
|
static void printMacroName(llvm::raw_string_ostream &MacroStack,
|
|
ASTContext &Context, SourceLocation Loc) {
|
|
MacroStack << Lexer::getImmediateMacroName(Loc, Context.getSourceManager(),
|
|
Context.getLangOpts());
|
|
|
|
// Add an empty space at the end as a padding to prevent
|
|
// that macro names concatenate to the names of other macros.
|
|
MacroStack << " ";
|
|
}
|
|
|
|
/// Returns a string that represents all macro expansions that expanded into the
|
|
/// given SourceLocation.
|
|
///
|
|
/// If 'getMacroStack(A) == getMacroStack(B)' is true, then the SourceLocations
|
|
/// A and B are expanded from the same macros in the same order.
|
|
std::string getMacroStack(SourceLocation Loc, ASTContext &Context) {
|
|
std::string MacroStack;
|
|
llvm::raw_string_ostream MacroStackStream(MacroStack);
|
|
SourceManager &SM = Context.getSourceManager();
|
|
|
|
// Iterate over all macros that expanded into the given SourceLocation.
|
|
while (Loc.isMacroID()) {
|
|
// Add the macro name to the stream.
|
|
printMacroName(MacroStackStream, Context, Loc);
|
|
Loc = SM.getImmediateMacroCallerLoc(Loc);
|
|
}
|
|
MacroStackStream.flush();
|
|
return MacroStack;
|
|
}
|
|
|
|
} // end namespace data_collection
|
|
} // end namespace clang
|