Format.

Author: VitaNuo

clang/lib/Frontend/FrontendAction.cpp

@@ -51,27 +51,32 @@ LLVM_INSTANTIATE_REGISTRY(FrontendPluginRegistry)
 
 namespace {
 
-/// DeserializedDeclsLineRangePrinter dumps ranges of deserialized declarations to aid debugging and bug minimization.
-/// It implements ASTConsumer and ASTDeserializationListener, so that an object of DeserializedDeclsLineRangePrinter registers
-/// as its own listener.
-/// The ASTDeserializationListener interface provides the DeclRead callback that we use to collect the deserialized Decls.
-/// Note that printing or otherwise processing them as this point is dangerous, since that could trigger additional
-/// deserialization and crash compilation.
-/// Therefore, we process the collected Decls in HandleTranslationUnit method of ASTConsumer.
-/// This is a safe point, since we know that by this point all the Decls needed by the compiler frontend have been
-/// deserialized. In case our processing causes further deserialization, DeclRead from the listener might be called again.
+/// DeserializedDeclsLineRangePrinter dumps ranges of deserialized declarations
+/// to aid debugging and bug minimization. It implements ASTConsumer and
+/// ASTDeserializationListener, so that an object of
+/// DeserializedDeclsLineRangePrinter registers as its own listener. The
+/// ASTDeserializationListener interface provides the DeclRead callback that we
+/// use to collect the deserialized Decls. Note that printing or otherwise
+/// processing them as this point is dangerous, since that could trigger
+/// additional deserialization and crash compilation. Therefore, we process the
+/// collected Decls in HandleTranslationUnit method of ASTConsumer. This is a
+/// safe point, since we know that by this point all the Decls needed by the
+/// compiler frontend have been deserialized. In case our processing causes
+/// further deserialization, DeclRead from the listener might be called again.
 /// However, at that point we don't accept any more Decls for processing.
-class DeserializedDeclsLineRangePrinter : public ASTDeserializationListener, public ASTConsumer {
+class DeserializedDeclsLineRangePrinter : public ASTDeserializationListener,
+                                          public ASTConsumer {
 public:
-  explicit DeserializedDeclsLineRangePrinter(SourceManager &SM, std::unique_ptr<llvm::raw_fd_ostream> OS)
+  explicit DeserializedDeclsLineRangePrinter(
+      SourceManager &SM, std::unique_ptr<llvm::raw_fd_ostream> OS)
       : ASTDeserializationListener(), SM(SM), OS(std::move(OS)) {}
 
   void DeclRead(GlobalDeclID ID, const Decl *D) override {
     if (!IsCollectingDecls) {
       return;
     }
     if (!D || isa<TranslationUnitDecl>(D) || isa<LinkageSpecDecl>(D) ||
-          isa<NamespaceDecl>(D))
+        isa<NamespaceDecl>(D))
       return;
     if (auto *DC = D->getDeclContext(); !DC || !DC->isFileContext())
       return;
@@ -81,15 +86,15 @@ class DeserializedDeclsLineRangePrinter : public ASTDeserializationListener, pub
 
   using Position = std::pair<unsigned, unsigned>;
   struct RequiredRanges {
-      StringRef Filename;
-      std::vector<std::pair<Position, Position>> FromTo;
+    StringRef Filename;
+    std::vector<std::pair<Position, Position>> FromTo;
   };
   void HandleTranslationUnit(ASTContext &Context) override {
     IsCollectingDecls = false;
     std::vector<const Decl *> Decls = std::move(PendingDecls);
     if (!PendingDecls.empty()) {
       llvm::errs() << "Deserialized more decls while printing, total of "
-                    << PendingDecls.size() << "\n";
+                   << PendingDecls.size() << "\n";
       PendingDecls.clear();
     }
 
@@ -112,18 +117,20 @@ class DeserializedDeclsLineRangePrinter : public ASTDeserializationListener, pub
 
       auto &Data = FileToLines[F];
       if (!Data.Ref)
-        Data.Ref =
-            SM.getFileEntryRefForID(SM.getFileID(R.getBegin()));
-      Data.FromTo.push_back({{SM.getSpellingLineNumber(R.getBegin()), SM.getSpellingColumnNumber(R.getBegin())},
-                            {SM.getSpellingLineNumber(R.getEnd()), SM.getSpellingColumnNumber(R.getEnd())}});
+        Data.Ref = SM.getFileEntryRefForID(SM.getFileID(R.getBegin()));
+      Data.FromTo.push_back({{SM.getSpellingLineNumber(R.getBegin()),
+                              SM.getSpellingColumnNumber(R.getBegin())},
+                             {SM.getSpellingLineNumber(R.getEnd()),
+                              SM.getSpellingColumnNumber(R.getEnd())}});
     }
 
     std::vector<RequiredRanges> Result;
     for (auto &[F, Data] : FileToLines) {
-      auto& FromTo = Data.FromTo;
+      auto &FromTo = Data.FromTo;
       assert(!FromTo.empty());
 
-      if (!Data.Ref) continue;
+      if (!Data.Ref)
+        continue;
 
       llvm::sort(FromTo);
 
@@ -142,7 +149,7 @@ class DeserializedDeclsLineRangePrinter : public ASTDeserializationListener, pub
     printJson(Result);
   }
 
-  void printJson(const std::vector<RequiredRanges>& Result) {
+  void printJson(const std::vector<RequiredRanges> &Result) {
     *OS << "{\n";
     *OS << "  \"required_ranges\": [\n";
     for (size_t i = 0; i < Result.size(); ++i) {
@@ -182,13 +189,12 @@ class DeserializedDeclsLineRangePrinter : public ASTDeserializationListener, pub
   }
 
 private:
-std::vector<const Decl *> PendingDecls;
-bool IsCollectingDecls = true;
-const SourceManager &SM;
-std::unique_ptr<llvm::raw_ostream> OS;
+  std::vector<const Decl *> PendingDecls;
+  bool IsCollectingDecls = true;
+  const SourceManager &SM;
+  std::unique_ptr<llvm::raw_ostream> OS;
 };
 
-
 /// Dumps deserialized declarations.
 class DeserializedDeclsDumper : public DelegatingDeserializationListener {
 public:
@@ -262,15 +268,22 @@ FrontendAction::CreateWrappedASTConsumer(CompilerInstance &CI,
     return nullptr;
 
   std::vector<std::unique_ptr<ASTConsumer>> Consumers;
-  llvm::StringRef DumpDeserializedDeclarationRangesPath = CI.getFrontendOpts().DumpDeserializedDeclarationRangesPath;
+  llvm::StringRef DumpDeserializedDeclarationRangesPath =
+      CI.getFrontendOpts().DumpDeserializedDeclarationRangesPath;
   if (!DumpDeserializedDeclarationRangesPath.empty()) {
     std::error_code ErrorCode;
-    auto FileStream = std::make_unique<llvm::raw_fd_ostream>(DumpDeserializedDeclarationRangesPath, ErrorCode, llvm::sys::fs::OF_None);
+    auto FileStream = std::make_unique<llvm::raw_fd_ostream>(
+        DumpDeserializedDeclarationRangesPath, ErrorCode,
+        llvm::sys::fs::OF_None);
     if (!ErrorCode) {
-      auto Printer = std::make_unique<DeserializedDeclsLineRangePrinter>(CI.getSourceManager(), std::move(FileStream));
+      auto Printer = std::make_unique<DeserializedDeclsLineRangePrinter>(
+          CI.getSourceManager(), std::move(FileStream));
       Consumers.push_back(std::move(Printer));
     } else {
-      llvm::errs() << "Failed to create output file for -dump-deserialized-declaration-ranges flag, file path: " << DumpDeserializedDeclarationRangesPath << ", error: " << ErrorCode.message() << "\n";
+      llvm::errs() << "Failed to create output file for "
+                      "-dump-deserialized-declaration-ranges flag, file path: "
+                   << DumpDeserializedDeclarationRangesPath
+                   << ", error: " << ErrorCode.message() << "\n";
     }
   }
 
@@ -340,7 +353,7 @@ FrontendAction::CreateWrappedASTConsumer(CompilerInstance &CI,
   }
 
   assert(Consumers.size() >= 1 && "should have added the main consumer");
-  if (Consumers.size() == 1) 
+  if (Consumers.size() == 1)
     return std::move(Consumers.front());
   return std::make_unique<MultiplexConsumer>(std::move(Consumers));
 }