Add optional terse flag to PrintableDecl

Make terse printing the default.

2 files changed, 10 insertions, 4 deletions

diff --git a/iwyu_ast_util.cc b/iwyu_ast_util.cc
index 8ff8420..23b9b05 100644
--- a/iwyu_ast_util.cc
+++ b/iwyu_ast_util.cc
@@ -402,10 +402,16 @@ string PrintableSourceRange(SourceRange range) {
   return PrintableLoc(range.getBegin()) + " - " + PrintableLoc(range.getEnd());
 }
 
-string PrintableDecl(const Decl* decl) {
-  std::string buffer;    // llvm wants regular string, not our versa-string
+string PrintableDecl(const Decl* decl, bool terse/*=true*/) {
+  // Use the terse flag to limit the level of output to one line.
+  clang::PrintingPolicy policy = decl->getASTContext().getPrintingPolicy();
+  policy.TerseOutput = terse;
+  policy.SuppressInitializers = terse;
+  policy.PolishForDeclaration = terse;
+
+  std::string buffer;
   raw_string_ostream ostream(buffer);
-  decl->print(ostream);  // Note: can also set indentation and printingpolicy
+  decl->print(ostream, policy);
   return ostream.str();
 }
 
diff --git a/iwyu_ast_util.h b/iwyu_ast_util.h
index 1bcf4a2..dd3f5e0 100644
--- a/iwyu_ast_util.h
+++ b/iwyu_ast_util.h
@@ -424,7 +424,7 @@ const clang::DeclContext* GetDeclContext(const ASTNode* ast_node);
 
 string PrintableLoc(clang::SourceLocation loc);
 string PrintableSourceRange(clang::SourceRange range);
-string PrintableDecl(const clang::Decl* decl);
+string PrintableDecl(const clang::Decl* decl, bool terse=true);
 string PrintableStmt(const clang::Stmt* stmt);
 string PrintableType(const clang::Type* type);
 string PrintableTypeLoc(const clang::TypeLoc& typeloc);