ParseAST.cpp revision 467dc88512b4ba4bb16e274ea3771dc1415d31da 
1//===--- ParseAST.cpp - Provide the clang::ParseAST method ----------------===//
2//
3//                     The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file implements the clang::ParseAST method.
11//
12//===----------------------------------------------------------------------===//
13
14#include "clang/Parse/ParseAST.h"
15#include "clang/Sema/Sema.h"
16#include "clang/Sema/CodeCompleteConsumer.h"
17#include "clang/Sema/SemaConsumer.h"
18#include "clang/Sema/ExternalSemaSource.h"
19#include "clang/AST/ASTConsumer.h"
20#include "clang/AST/DeclCXX.h"
21#include "clang/AST/ExternalASTSource.h"
22#include "clang/AST/Stmt.h"
23#include "clang/Parse/Parser.h"
24#include "llvm/ADT/OwningPtr.h"
25#include "llvm/Support/CrashRecoveryContext.h"
26#include <cstdio>
27
28using namespace clang;
29
30//===----------------------------------------------------------------------===//
31// Public interface to the file
32//===----------------------------------------------------------------------===//
33
34/// ParseAST - Parse the entire file specified, notifying the ASTConsumer as
35/// the file is parsed.  This inserts the parsed decls into the translation unit
36/// held by Ctx.
37///
38void clang::ParseAST(Preprocessor &PP, ASTConsumer *Consumer,
39                     ASTContext &Ctx, bool PrintStats,
40                     TranslationUnitKind TUKind,
41                     CodeCompleteConsumer *CompletionConsumer) {
42
43  llvm::OwningPtr<Sema> S(new Sema(PP, Ctx, *Consumer,
44                                   TUKind,
45                                   CompletionConsumer));
46
47  // Recover resources if we crash before exiting this method.
48  llvm::CrashRecoveryContextCleanupRegistrar<Sema> CleaupSema(S.get());
49
50  ParseAST(*S.get(), PrintStats);
51}
52
53void clang::ParseAST(Sema &S, bool PrintStats) {
54  // Collect global stats on Decls/Stmts (until we have a module streamer).
55  if (PrintStats) {
56    Decl::CollectingStats(true);
57    Stmt::CollectingStats(true);
58  }
59
60  // Also turn on collection of stats inside of the Sema object.
61  bool OldCollectStats = PrintStats;
62  std::swap(OldCollectStats, S.CollectStats);
63
64  ASTConsumer *Consumer = &S.getASTConsumer();
65
66  llvm::OwningPtr<Parser> ParseOP(new Parser(S.getPreprocessor(), S));
67  Parser &P = *ParseOP.get();
68
69  PrettyStackTraceParserEntry CrashInfo(P);
70
71  // Recover resources if we crash before exiting this method.
72  llvm::CrashRecoveryContextCleanupRegistrar<Parser>
73    CleaupParser(ParseOP.get());
74
75  S.getPreprocessor().EnterMainSourceFile();
76  P.Initialize();
77  S.Initialize();
78
79  if (ExternalASTSource *External = S.getASTContext().getExternalSource())
80    External->StartTranslationUnit(Consumer);
81
82  Parser::DeclGroupPtrTy ADecl;
83
84  while (!P.ParseTopLevelDecl(ADecl)) {  // Not end of file.
85    // If we got a null return and something *was* parsed, ignore it.  This
86    // is due to a top-level semicolon, an action override, or a parse error
87    // skipping something.
88    if (ADecl)
89      Consumer->HandleTopLevelDecl(ADecl.get());
90  };
91  // Check for any pending objective-c implementation decl.
92  while ((ADecl = P.FinishPendingObjCActions()))
93    Consumer->HandleTopLevelDecl(ADecl.get());
94
95  // Process any TopLevelDecls generated by #pragma weak.
96  for (SmallVector<Decl*,2>::iterator
97       I = S.WeakTopLevelDecls().begin(),
98       E = S.WeakTopLevelDecls().end(); I != E; ++I)
99    Consumer->HandleTopLevelDecl(DeclGroupRef(*I));
100
101  Consumer->HandleTranslationUnit(S.getASTContext());
102
103  std::swap(OldCollectStats, S.CollectStats);
104  if (PrintStats) {
105    llvm::errs() << "\nSTATISTICS:\n";
106    P.getActions().PrintStats();
107    S.getASTContext().PrintStats();
108    Decl::PrintStats();
109    Stmt::PrintStats();
110    Consumer->PrintStats();
111  }
112}
113