49 lines · cpp
1//===-- DataCollection.cpp --------------------------------------*- C++ -*-===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8 9#include "clang/AST/DataCollection.h"10#include "clang/Basic/SourceManager.h"11#include "clang/Lex/Lexer.h"12 13namespace clang {14namespace data_collection {15 16/// Prints the macro name that contains the given SourceLocation into the given17/// raw_string_ostream.18static void printMacroName(llvm::raw_string_ostream &MacroStack,19 ASTContext &Context, SourceLocation Loc) {20 MacroStack << Lexer::getImmediateMacroName(Loc, Context.getSourceManager(),21 Context.getLangOpts());22 23 // Add an empty space at the end as a padding to prevent24 // that macro names concatenate to the names of other macros.25 MacroStack << " ";26}27 28/// Returns a string that represents all macro expansions that expanded into the29/// given SourceLocation.30///31/// If 'getMacroStack(A) == getMacroStack(B)' is true, then the SourceLocations32/// A and B are expanded from the same macros in the same order.33std::string getMacroStack(SourceLocation Loc, ASTContext &Context) {34 std::string MacroStack;35 llvm::raw_string_ostream MacroStackStream(MacroStack);36 SourceManager &SM = Context.getSourceManager();37 38 // Iterate over all macros that expanded into the given SourceLocation.39 while (Loc.isMacroID()) {40 // Add the macro name to the stream.41 printMacroName(MacroStackStream, Context, Loc);42 Loc = SM.getImmediateMacroCallerLoc(Loc);43 }44 return MacroStack;45}46 47} // end namespace data_collection48} // end namespace clang49