ASTUnit.h revision 01b6e31a62e2265849f4388b9be6be0a5d13348d
14b562cf889bc59e1914dd2c5d9fbd7e7bfa1ad77Argyrios Kyrtzidis//===--- ASTUnit.h - ASTUnit utility ----------------------------*- C++ -*-===//
24b562cf889bc59e1914dd2c5d9fbd7e7bfa1ad77Argyrios Kyrtzidis//
34b562cf889bc59e1914dd2c5d9fbd7e7bfa1ad77Argyrios Kyrtzidis//                     The LLVM Compiler Infrastructure
44b562cf889bc59e1914dd2c5d9fbd7e7bfa1ad77Argyrios Kyrtzidis//
54b562cf889bc59e1914dd2c5d9fbd7e7bfa1ad77Argyrios Kyrtzidis// This file is distributed under the University of Illinois Open Source
64b562cf889bc59e1914dd2c5d9fbd7e7bfa1ad77Argyrios Kyrtzidis// License. See LICENSE.TXT for details.
74b562cf889bc59e1914dd2c5d9fbd7e7bfa1ad77Argyrios Kyrtzidis//
84b562cf889bc59e1914dd2c5d9fbd7e7bfa1ad77Argyrios Kyrtzidis//===----------------------------------------------------------------------===//
94b562cf889bc59e1914dd2c5d9fbd7e7bfa1ad77Argyrios Kyrtzidis//
104b562cf889bc59e1914dd2c5d9fbd7e7bfa1ad77Argyrios Kyrtzidis// ASTUnit utility class.
114b562cf889bc59e1914dd2c5d9fbd7e7bfa1ad77Argyrios Kyrtzidis//
124b562cf889bc59e1914dd2c5d9fbd7e7bfa1ad77Argyrios Kyrtzidis//===----------------------------------------------------------------------===//
134b562cf889bc59e1914dd2c5d9fbd7e7bfa1ad77Argyrios Kyrtzidis
140853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis#ifndef LLVM_CLANG_FRONTEND_ASTUNIT_H
150853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis#define LLVM_CLANG_FRONTEND_ASTUNIT_H
160853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis
17eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor#include "clang/Index/ASTLocation.h"
1805a07605322dfef2b017781042043a261c5a89cdSebastian Redl#include "clang/Serialization/ASTBitCodes.h"
19914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor#include "clang/Sema/Sema.h"
200a2c5e256abb4dc031c21fe4dc92c4f3afe9947cJohn McCall#include "clang/Sema/CodeCompleteConsumer.h"
214ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor#include "clang/Lex/PreprocessingRecord.h"
2231b87d8006d4863dd9b17e515ac720941efc38e3Daniel Dunbar#include "clang/Basic/SourceManager.h"
23eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor#include "clang/Basic/FileManager.h"
24389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis#include "clang/Basic/FileSystemOptions.h"
2587c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor#include "clang-c/Index.h"
2628019772db70d4547be05a042eb950bc910f134fDouglas Gregor#include "llvm/ADT/IntrusiveRefCntPtr.h"
270853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis#include "llvm/ADT/OwningPtr.h"
28a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor#include "llvm/ADT/SmallVector.h"
29cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor#include "llvm/ADT/StringMap.h"
3003013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Path.h"
31788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor#include <map>
320853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis#include <string>
33f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar#include <vector>
34f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar#include <cassert>
354db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor#include <utility>
36cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor#include <sys/types.h>
374db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor
384db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregornamespace llvm {
394db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  class MemoryBuffer;
404db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor}
410853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis
420853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidisnamespace clang {
43521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass ASTContext;
441abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregorclass CodeCompleteConsumer;
45521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass CompilerInvocation;
46521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass Decl;
47521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass Diagnostic;
48521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass FileEntry;
49521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass FileManager;
50521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass HeaderSearch;
51521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass Preprocessor;
52521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass SourceManager;
53521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass TargetInfo;
54d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidisclass ASTFrontendAction;
550853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis
56f96b524306ccfa623235d375deee79637bd38f29Steve Naroffusing namespace idx;
5744c181aec37789f25f6c15543c164416f72e562aDouglas Gregor
5848601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor/// \brief Allocator for a cached set of global code completions.
5948601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregorclass GlobalCodeCompletionAllocator
6048601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor  : public CodeCompletionAllocator,
6148601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor    public llvm::RefCountedBase<GlobalCodeCompletionAllocator>
6248601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor{
6348601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor
6448601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor};
6548601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor
663c7f4134603d04b44f997b43c0a9def270f25386Sebastian Redl/// \brief Utility class for loading a ASTContext from an AST file.
670853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis///
680853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidisclass ASTUnit {
69788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregorpublic:
70788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  typedef std::map<FileID, std::vector<PreprocessedEntity *> >
71788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    PreprocessedEntitiesByFileMap;
72bdbb004f38978da0c4a75af3294d1c7b5ff84af1Douglas Gregor
73788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregorprivate:
7428019772db70d4547be05a042eb950bc910f134fDouglas Gregor  llvm::IntrusiveRefCntPtr<Diagnostic> Diagnostics;
754f32786ac45210143654390177105eb749b614e9Ted Kremenek  llvm::IntrusiveRefCntPtr<FileManager>      FileMgr;
764f32786ac45210143654390177105eb749b614e9Ted Kremenek  llvm::IntrusiveRefCntPtr<SourceManager>    SourceMgr;
770853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis  llvm::OwningPtr<HeaderSearch>     HeaderInfo;
784f32786ac45210143654390177105eb749b614e9Ted Kremenek  llvm::IntrusiveRefCntPtr<TargetInfo>       Target;
794f32786ac45210143654390177105eb749b614e9Ted Kremenek  llvm::IntrusiveRefCntPtr<Preprocessor>     PP;
804f32786ac45210143654390177105eb749b614e9Ted Kremenek  llvm::IntrusiveRefCntPtr<ASTContext>       Ctx;
81389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis
82389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis  FileSystemOptions FileSystemOpts;
83389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis
84914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor  /// \brief The AST consumer that received information about the translation
85914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor  /// unit as it was parsed or loaded.
86914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor  llvm::OwningPtr<ASTConsumer> Consumer;
87914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor
88914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor  /// \brief The semantic analysis object used to type-check the translation
89914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor  /// unit.
90914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor  llvm::OwningPtr<Sema> TheSema;
91914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor
92807b06157a1a5c050520fc194d32f16d22d423a8Daniel Dunbar  /// Optional owned invocation, just used to make the invocation used in
93807b06157a1a5c050520fc194d32f16d22d423a8Daniel Dunbar  /// LoadFromCommandLine available.
944f32786ac45210143654390177105eb749b614e9Ted Kremenek  llvm::IntrusiveRefCntPtr<CompilerInvocation> Invocation;
95914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor
961aa27307c462baaa9e5fda14ff6797dd39fe8b84Douglas Gregor  /// \brief The set of target features.
971aa27307c462baaa9e5fda14ff6797dd39fe8b84Douglas Gregor  ///
981aa27307c462baaa9e5fda14ff6797dd39fe8b84Douglas Gregor  /// FIXME: each time we reparse, we need to restore the set of target
991aa27307c462baaa9e5fda14ff6797dd39fe8b84Douglas Gregor  /// features from this vector, because TargetInfo::CreateTargetInfo()
1001aa27307c462baaa9e5fda14ff6797dd39fe8b84Douglas Gregor  /// mangles the target options in place. Yuck!
1011aa27307c462baaa9e5fda14ff6797dd39fe8b84Douglas Gregor  std::vector<std::string> TargetFeatures;
1021aa27307c462baaa9e5fda14ff6797dd39fe8b84Douglas Gregor
1037d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor  // OnlyLocalDecls - when true, walking this AST should only visit declarations
1047d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor  // that come from the AST itself, not from included precompiled headers.
1057d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor  // FIXME: This is temporary; eventually, CIndex will always do this.
1067d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor  bool                              OnlyLocalDecls;
107c7822dbf3c01a2a5f837cff82ba7889ea755dacaDaniel Dunbar
108abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  /// \brief Whether to capture any diagnostics produced.
109abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  bool CaptureDiagnostics;
110e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor
111df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor  /// \brief Track whether the main file was loaded from an AST or not.
112c7822dbf3c01a2a5f837cff82ba7889ea755dacaDaniel Dunbar  bool MainFileIsAST;
113c7822dbf3c01a2a5f837cff82ba7889ea755dacaDaniel Dunbar
114df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor  /// \brief Whether this AST represents a complete translation unit.
115df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor  bool CompleteTranslationUnit;
116df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor
117213f18b3d654de7d1c7cf4a329ea9d3db1c50b6aDouglas Gregor  /// \brief Whether we should time each operation.
118213f18b3d654de7d1c7cf4a329ea9d3db1c50b6aDouglas Gregor  bool WantTiming;
11915727ddb11405c45372150b5bfb07dbfa4c9960bArgyrios Kyrtzidis
12015727ddb11405c45372150b5bfb07dbfa4c9960bArgyrios Kyrtzidis  /// \brief Whether the ASTUnit should delete the remapped buffers.
12115727ddb11405c45372150b5bfb07dbfa4c9960bArgyrios Kyrtzidis  bool OwnsRemappedFileBuffers;
122213f18b3d654de7d1c7cf4a329ea9d3db1c50b6aDouglas Gregor
123f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar  /// Track the top-level decls which appeared in an ASTUnit which was loaded
124f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar  /// from a source file.
125f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar  //
126f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar  // FIXME: This is just an optimization hack to avoid deserializing large parts
127f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar  // of a PCH file when using the Index library on an ASTUnit loaded from
128f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar  // source. In the long term we should make the Index library use efficient and
129f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar  // more scalable search mechanisms.
130f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar  std::vector<Decl*> TopLevelDecls;
131f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar
13289d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  /// \brief The list of preprocessed entities which appeared when the ASTUnit
13389d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  /// was loaded.
13489d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  ///
13589d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  /// FIXME: This is just an optimization hack to avoid deserializing large
13689d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  /// parts of a PCH file while performing a walk or search. In the long term,
13789d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  /// we should provide more scalable search mechanisms.
13889d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  std::vector<PreprocessedEntity *> PreprocessedEntities;
13989d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor
14068d40e2d16b9fadba386853d6bbb60089291fdc5Daniel Dunbar  /// The name of the original source file used to generate this ASTUnit.
14168d40e2d16b9fadba386853d6bbb60089291fdc5Daniel Dunbar  std::string OriginalSourceFile;
14268d40e2d16b9fadba386853d6bbb60089291fdc5Daniel Dunbar
143f96b524306ccfa623235d375deee79637bd38f29Steve Naroff  // Critical optimization when using clang_getCursor().
144f96b524306ccfa623235d375deee79637bd38f29Steve Naroff  ASTLocation LastLoc;
145f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar
146a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor  /// \brief The set of diagnostics produced when creating this
147a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor  /// translation unit.
148405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor  llvm::SmallVector<StoredDiagnostic, 4> StoredDiagnostics;
149a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor
1504cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor  /// \brief The number of stored diagnostics that come from the driver
1514cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor  /// itself.
1524cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor  ///
1534cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor  /// Diagnostics that come from the driver are retained from one parse to
1544cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor  /// the next.
1554cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor  unsigned NumStoredDiagnosticsFromDriver;
1564cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor
157313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor  /// \brief Temporary files that should be removed when the ASTUnit is
158313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor  /// destroyed.
159313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor  llvm::SmallVector<llvm::sys::Path, 4> TemporaryFiles;
160bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor
161788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  /// \brief A mapping from file IDs to the set of preprocessed entities
162788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  /// stored in that file.
163788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  ///
164788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  /// FIXME: This is just an optimization hack to avoid searching through
165788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  /// many preprocessed entities during cursor traversal in the CIndex library.
166788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  /// Ideally, we would just be able to perform a binary search within the
167788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  /// list of preprocessed entities.
168788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  PreprocessedEntitiesByFileMap PreprocessedEntitiesByFile;
169788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
170bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  /// \brief Simple hack to allow us to assert that ASTUnit is not being
171bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  /// used concurrently, which is not supported.
172bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  ///
173bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  /// Clients should create instances of the ConcurrencyCheck class whenever
174bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  /// using the ASTUnit in a way that isn't intended to be concurrent, which is
175bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  /// just about any usage.
176bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  unsigned int ConcurrencyCheckValue;
177bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  static const unsigned int CheckLocked = 28573289;
178bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  static const unsigned int CheckUnlocked = 9803453;
179175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor
180eababfbddb74d186f78783a9731a78ad371c9800Douglas Gregor  /// \brief Counter that determines when we want to try building a
181eababfbddb74d186f78783a9731a78ad371c9800Douglas Gregor  /// precompiled preamble.
182eababfbddb74d186f78783a9731a78ad371c9800Douglas Gregor  ///
183eababfbddb74d186f78783a9731a78ad371c9800Douglas Gregor  /// If zero, we will never build a precompiled preamble. Otherwise,
184eababfbddb74d186f78783a9731a78ad371c9800Douglas Gregor  /// it's treated as a counter that decrements each time we reparse
185eababfbddb74d186f78783a9731a78ad371c9800Douglas Gregor  /// without the benefit of a precompiled preamble. When it hits 1,
186eababfbddb74d186f78783a9731a78ad371c9800Douglas Gregor  /// we'll attempt to rebuild the precompiled header. This way, if
187eababfbddb74d186f78783a9731a78ad371c9800Douglas Gregor  /// building the precompiled preamble fails, we won't try again for
188eababfbddb74d186f78783a9731a78ad371c9800Douglas Gregor  /// some number of calls.
189eababfbddb74d186f78783a9731a78ad371c9800Douglas Gregor  unsigned PreambleRebuildCounter;
190eababfbddb74d186f78783a9731a78ad371c9800Douglas Gregor
191175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor  /// \brief The file in which the precompiled preamble is stored.
192385103b79c5338a2be5da0ca70652400bc267371Douglas Gregor  std::string PreambleFile;
193175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor
194175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor  /// \brief The contents of the preamble that has been precompiled to
195175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor  /// \c PreambleFile.
196175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor  std::vector<char> Preamble;
197175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor
198f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor  /// \brief Whether the preamble ends at the start of a new line.
199f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor  ///
200f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor  /// Used to inform the lexer as to whether it's starting at the beginning of
201f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor  /// a line after skipping the preamble.
202f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor  bool PreambleEndsAtStartOfLine;
203f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor
204175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor  /// \brief The size of the source buffer that we've reserved for the main
205175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor  /// file within the precompiled preamble.
206175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor  unsigned PreambleReservedSize;
207cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor
208cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor  /// \brief Keeps track of the files that were used when computing the
209cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor  /// preamble, with both their buffer size and their modification time.
210cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor  ///
211cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor  /// If any of the files have changed from one compile to the next,
212cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor  /// the preamble must be thrown away.
213cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor  llvm::StringMap<std::pair<off_t, time_t> > FilesInPreamble;
214cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor
21528233428da1ebec20c893d6297ae3191318940ddDouglas Gregor  /// \brief When non-NULL, this is the buffer used to store the contents of
21628233428da1ebec20c893d6297ae3191318940ddDouglas Gregor  /// the main file when it has been padded for use with the precompiled
21728233428da1ebec20c893d6297ae3191318940ddDouglas Gregor  /// preamble.
21828233428da1ebec20c893d6297ae3191318940ddDouglas Gregor  llvm::MemoryBuffer *SavedMainFileBuffer;
219c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor
220671947b18dba342f9aba022ee992babef325a833Douglas Gregor  /// \brief When non-NULL, this is the buffer used to store the
221671947b18dba342f9aba022ee992babef325a833Douglas Gregor  /// contents of the preamble when it has been padded to build the
222671947b18dba342f9aba022ee992babef325a833Douglas Gregor  /// precompiled preamble.
223671947b18dba342f9aba022ee992babef325a833Douglas Gregor  llvm::MemoryBuffer *PreambleBuffer;
224671947b18dba342f9aba022ee992babef325a833Douglas Gregor
225c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor  /// \brief The number of warnings that occurred while parsing the preamble.
226c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor  ///
227c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor  /// This value will be used to restore the state of the \c Diagnostic object
228c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor  /// when re-using the precompiled preamble. Note that only the
229c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor  /// number of warnings matters, since we will not save the preamble
230c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor  /// when any errors are present.
231c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor  unsigned NumWarningsInPreamble;
232c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor
233c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor  /// \brief The number of diagnostics that were stored when parsing
234c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor  /// the precompiled preamble.
235c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor  ///
236c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor  /// This value is used to determine how many of the stored
237c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor  /// diagnostics should be retained when reparsing in the presence of
238c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor  /// a precompiled preamble.
239c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor  unsigned NumStoredDiagnosticsInPreamble;
240c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor
2413c7f4134603d04b44f997b43c0a9def270f25386Sebastian Redl  /// \brief A list of the serialization ID numbers for each of the top-level
242eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor  /// declarations parsed within the precompiled preamble.
2438538e8d43a3a9bd439c987c0de37bcbf035dd391Sebastian Redl  std::vector<serialization::DeclID> TopLevelDeclsInPreamble;
244eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor
24589d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  /// \brief A list of the offsets into the precompiled preamble which
24689d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  /// correspond to preprocessed entities.
24789d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  std::vector<uint64_t> PreprocessedEntitiesInPreamble;
24889d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor
24987c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  /// \brief Whether we should be caching code-completion results.
25087c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  bool ShouldCacheCodeCompletionResults;
25187c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor
252dca8ee8b7bc86076916a3a80f553f7a4e98c14afDouglas Gregor  /// \brief Whether we want to include nested macro instantiations in the
253dca8ee8b7bc86076916a3a80f553f7a4e98c14afDouglas Gregor  /// detailed preprocessing record.
254dca8ee8b7bc86076916a3a80f553f7a4e98c14afDouglas Gregor  bool NestedMacroInstantiations;
255dca8ee8b7bc86076916a3a80f553f7a4e98c14afDouglas Gregor
256e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor  static void ConfigureDiags(llvm::IntrusiveRefCntPtr<Diagnostic> &Diags,
2570b53cf834346d78985aaa9e7300445a39c245614Douglas Gregor                             const char **ArgBegin, const char **ArgEnd,
258e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor                             ASTUnit &AST, bool CaptureDiagnostics);
259e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor
26087c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregorpublic:
26187c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  /// \brief A cached code-completion result, which may be introduced in one of
26287c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  /// many different contexts.
26387c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  struct CachedCodeCompletionResult {
26487c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    /// \brief The code-completion string corresponding to this completion
26587c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    /// result.
26687c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    CodeCompletionString *Completion;
26787c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor
26887c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    /// \brief A bitmask that indicates which code-completion contexts should
26987c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    /// contain this completion result.
27087c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    ///
27187c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    /// The bits in the bitmask correspond to the values of
27287c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    /// CodeCompleteContext::Kind. To map from a completion context kind to a
27387c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    /// bit, subtract one from the completion context kind and shift 1 by that
27487c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    /// number of bits. Many completions can occur in several different
27587c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    /// contexts.
27687c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    unsigned ShowInContexts;
27787c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor
27887c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    /// \brief The priority given to this code-completion result.
27987c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    unsigned Priority;
28087c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor
28187c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    /// \brief The libclang cursor kind corresponding to this code-completion
28287c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    /// result.
28387c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    CXCursorKind Kind;
2841827e10051638770ad9ccf3e285caf95f995afd1Douglas Gregor
28558ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor    /// \brief The availability of this code-completion result.
28658ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor    CXAvailabilityKind Availability;
28758ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
2881827e10051638770ad9ccf3e285caf95f995afd1Douglas Gregor    /// \brief The simplified type class for a non-macro completion result.
2891827e10051638770ad9ccf3e285caf95f995afd1Douglas Gregor    SimplifiedTypeClass TypeClass;
290f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor
291f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor    /// \brief The type of a non-macro completion result, stored as a unique
292f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor    /// integer used by the string map of cached completion types.
293f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor    ///
294f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor    /// This value will be zero if the type is not known, or a unique value
295f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor    /// determined by the formatted type string. Se \c CachedCompletionTypes
296f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor    /// for more information.
297f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor    unsigned Type;
29887c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  };
29987c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor
300f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor  /// \brief Retrieve the mapping from formatted type names to unique type
301f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor  /// identifiers.
302f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor  llvm::StringMap<unsigned> &getCachedCompletionTypes() {
303f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor    return CachedCompletionTypes;
304f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor  }
305f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor
30648601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor  /// \brief Retrieve the allocator used to cache global code completions.
30748601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor  llvm::IntrusiveRefCntPtr<GlobalCodeCompletionAllocator>
30848601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor  getCachedCompletionAllocator() {
30948601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor    return CachedCompletionAllocator;
31048601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor  }
31148601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor
31287c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregorprivate:
313218937c13ef5b0625a70aad41ca7a92da9278bd2Douglas Gregor  /// \brief Allocator used to store cached code completions.
31448601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor  llvm::IntrusiveRefCntPtr<GlobalCodeCompletionAllocator>
31548601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor    CachedCompletionAllocator;
316218937c13ef5b0625a70aad41ca7a92da9278bd2Douglas Gregor
31787c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  /// \brief The set of cached code-completion results.
31887c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  std::vector<CachedCodeCompletionResult> CachedCompletionResults;
31987c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor
320f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor  /// \brief A mapping from the formatted type name to a unique number for that
321f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor  /// type, which is used for type equality comparisons.
322f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor  llvm::StringMap<unsigned> CachedCompletionTypes;
323f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor
3249b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor  /// \brief A string hash of the top-level declaration and macro definition
3259b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor  /// names processed the last time that we reparsed the file.
326727d93ef49e18147149354fadd10e86b13bc4ab0Douglas Gregor  ///
3279b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor  /// This hash value is used to determine when we need to refresh the
3289b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor  /// global code-completion cache.
3299b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor  unsigned CompletionCacheTopLevelHashValue;
330727d93ef49e18147149354fadd10e86b13bc4ab0Douglas Gregor
3319b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor  /// \brief A string hash of the top-level declaration and macro definition
3329b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor  /// names processed the last time that we reparsed the precompiled preamble.
333727d93ef49e18147149354fadd10e86b13bc4ab0Douglas Gregor  ///
3349b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor  /// This hash value is used to determine when we need to refresh the
3359b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor  /// global code-completion cache after a rebuild of the precompiled preamble.
3369b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor  unsigned PreambleTopLevelHashValue;
337ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
3389b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor  /// \brief The current hash value for the top-level declaration and macro
3399b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor  /// definition names
3409b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor  unsigned CurrentTopLevelHashValue;
3419b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor
342ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  /// \brief Bit used by CIndex to mark when a translation unit may be in an
343ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  /// inconsistent state, and is not safe to free.
344ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned UnsafeToFree : 1;
345ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
346c4421e966d77a18f815284175b3fcbb46f36fa39Douglas Gregor  /// \brief Cache any "global" code-completion results, so that we can avoid
347c4421e966d77a18f815284175b3fcbb46f36fa39Douglas Gregor  /// recomputing them with each completion.
34887c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  void CacheCodeCompletionResults();
34987c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor
35087c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  /// \brief Clear out and deallocate
35187c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  void ClearCachedCompletionResults();
35287c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor
35331b87d8006d4863dd9b17e515ac720941efc38e3Daniel Dunbar  ASTUnit(const ASTUnit&); // DO NOT IMPLEMENT
35431b87d8006d4863dd9b17e515ac720941efc38e3Daniel Dunbar  ASTUnit &operator=(const ASTUnit &); // DO NOT IMPLEMENT
355bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor
3563687e9d3a5dbfa9963af02a49a2b139d91310813Douglas Gregor  explicit ASTUnit(bool MainFileIsAST);
3578b96253907c47141af0b7b2a44a368748d006a87Douglas Gregor
358abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  void CleanTemporaryFiles();
359754f3490c5b0f5d83361f001bc87944f23644abbDouglas Gregor  bool Parse(llvm::MemoryBuffer *OverrideMainBuffer);
360175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor
361f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor  std::pair<llvm::MemoryBuffer *, std::pair<unsigned, bool> >
362df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor  ComputePreamble(CompilerInvocation &Invocation,
363df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor                  unsigned MaxLines, bool &CreatedBuffer);
364175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor
365df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor  llvm::MemoryBuffer *getMainBufferWithPrecompiledPreamble(
36601b6e31a62e2265849f4388b9be6be0a5d13348dDouglas Gregor                               const CompilerInvocation &PreambleInvocationIn,
367df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor                                                     bool AllowRebuild = true,
368df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor                                                        unsigned MaxLines = 0);
369eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor  void RealizeTopLevelDeclsFromPreamble();
37089d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  void RealizePreprocessedEntitiesFromPreamble();
37189d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor
3720853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidispublic:
373bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  class ConcurrencyCheck {
374bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor    volatile ASTUnit &Self;
375bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor
376bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  public:
377bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor    explicit ConcurrencyCheck(ASTUnit &Self)
378bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor      : Self(Self)
379bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor    {
380bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor      assert(Self.ConcurrencyCheckValue == CheckUnlocked &&
381bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor             "Concurrent access to ASTUnit!");
382bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor      Self.ConcurrencyCheckValue = CheckLocked;
383bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor    }
384bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor
385bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor    ~ConcurrencyCheck() {
386bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor      Self.ConcurrencyCheckValue = CheckUnlocked;
387bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor    }
388bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  };
389bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  friend class ConcurrencyCheck;
390bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor
3910853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis  ~ASTUnit();
3920853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis
393c7822dbf3c01a2a5f837cff82ba7889ea755dacaDaniel Dunbar  bool isMainFileAST() const { return MainFileIsAST; }
394c7822dbf3c01a2a5f837cff82ba7889ea755dacaDaniel Dunbar
395ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  bool isUnsafeToFree() const { return UnsafeToFree; }
396ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  void setUnsafeToFree(bool Value) { UnsafeToFree = Value; }
397ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
3983687e9d3a5dbfa9963af02a49a2b139d91310813Douglas Gregor  const Diagnostic &getDiagnostics() const { return *Diagnostics; }
3993687e9d3a5dbfa9963af02a49a2b139d91310813Douglas Gregor  Diagnostic &getDiagnostics()             { return *Diagnostics; }
4003687e9d3a5dbfa9963af02a49a2b139d91310813Douglas Gregor
401405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor  const SourceManager &getSourceManager() const { return *SourceMgr; }
402405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor        SourceManager &getSourceManager()       { return *SourceMgr; }
4030853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis
4044f32786ac45210143654390177105eb749b614e9Ted Kremenek  const Preprocessor &getPreprocessor() const { return *PP; }
4054f32786ac45210143654390177105eb749b614e9Ted Kremenek        Preprocessor &getPreprocessor()       { return *PP; }
4061eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4074f32786ac45210143654390177105eb749b614e9Ted Kremenek  const ASTContext &getASTContext() const { return *Ctx; }
4084f32786ac45210143654390177105eb749b614e9Ted Kremenek        ASTContext &getASTContext()       { return *Ctx; }
4090853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis
410914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor  bool hasSema() const { return TheSema; }
411914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor  Sema &getSema() const {
412914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor    assert(TheSema && "ASTUnit does not have a Sema object!");
413914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor    return *TheSema;
414914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor  }
415914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor
416405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor  const FileManager &getFileManager() const { return *FileMgr; }
417405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor        FileManager &getFileManager()       { return *FileMgr; }
418f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar
419389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis  const FileSystemOptions &getFileSystemOpts() const { return FileSystemOpts; }
420389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis
42177accc11f04ed4ff9afd4e27d430144d4714be56Steve Naroff  const std::string &getOriginalSourceFileName();
4223c7f4134603d04b44f997b43c0a9def270f25386Sebastian Redl  const std::string &getASTFileName();
423b85bca2676b433ae555db09de4dd2823ff13b856Zhongxing Xu
424313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor  /// \brief Add a temporary file that the ASTUnit depends on.
425313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor  ///
426313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor  /// This file will be erased when the ASTUnit is destroyed.
427313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor  void addTemporaryFile(const llvm::sys::Path &TempFile) {
428313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor    TemporaryFiles.push_back(TempFile);
429313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor  }
430313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor
4317d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor  bool getOnlyLocalDecls() const { return OnlyLocalDecls; }
432f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar
43315727ddb11405c45372150b5bfb07dbfa4c9960bArgyrios Kyrtzidis  bool getOwnsRemappedFileBuffers() const { return OwnsRemappedFileBuffers; }
43415727ddb11405c45372150b5bfb07dbfa4c9960bArgyrios Kyrtzidis  void setOwnsRemappedFileBuffers(bool val) { OwnsRemappedFileBuffers = val; }
43515727ddb11405c45372150b5bfb07dbfa4c9960bArgyrios Kyrtzidis
436eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor  /// \brief Retrieve the maximum PCH level of declarations that a
437eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor  /// traversal of the translation unit should consider.
438eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor  unsigned getMaxPCHLevel() const;
439eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor
440f96b524306ccfa623235d375deee79637bd38f29Steve Naroff  void setLastASTLocation(ASTLocation ALoc) { LastLoc = ALoc; }
441f96b524306ccfa623235d375deee79637bd38f29Steve Naroff  ASTLocation getLastASTLocation() const { return LastLoc; }
442f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar
443213f18b3d654de7d1c7cf4a329ea9d3db1c50b6aDouglas Gregor
444213f18b3d654de7d1c7cf4a329ea9d3db1c50b6aDouglas Gregor  llvm::StringRef getMainFileName() const;
445213f18b3d654de7d1c7cf4a329ea9d3db1c50b6aDouglas Gregor
446eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor  typedef std::vector<Decl *>::iterator top_level_iterator;
447eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor
448eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor  top_level_iterator top_level_begin() {
449f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar    assert(!isMainFileAST() && "Invalid call for AST based ASTUnit!");
450eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor    if (!TopLevelDeclsInPreamble.empty())
451eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor      RealizeTopLevelDeclsFromPreamble();
452eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor    return TopLevelDecls.begin();
453f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar  }
454eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor
455eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor  top_level_iterator top_level_end() {
456f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar    assert(!isMainFileAST() && "Invalid call for AST based ASTUnit!");
457eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor    if (!TopLevelDeclsInPreamble.empty())
458eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor      RealizeTopLevelDeclsFromPreamble();
459eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor    return TopLevelDecls.end();
460eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor  }
461eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor
462eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor  std::size_t top_level_size() const {
463eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor    assert(!isMainFileAST() && "Invalid call for AST based ASTUnit!");
464eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor    return TopLevelDeclsInPreamble.size() + TopLevelDecls.size();
465eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor  }
466eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor
467eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor  bool top_level_empty() const {
468eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor    assert(!isMainFileAST() && "Invalid call for AST based ASTUnit!");
469eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor    return TopLevelDeclsInPreamble.empty() && TopLevelDecls.empty();
470eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor  }
471eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor
472eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor  /// \brief Add a new top-level declaration.
473eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor  void addTopLevelDecl(Decl *D) {
474eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor    TopLevelDecls.push_back(D);
475eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor  }
476eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor
477eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor  /// \brief Add a new top-level declaration, identified by its ID in
478eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor  /// the precompiled preamble.
4798538e8d43a3a9bd439c987c0de37bcbf035dd391Sebastian Redl  void addTopLevelDeclFromPreamble(serialization::DeclID D) {
480eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor    TopLevelDeclsInPreamble.push_back(D);
481f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar  }
482f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar
4839b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor  /// \brief Retrieve a reference to the current top-level name hash value.
4849b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor  ///
4859b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor  /// Note: This is used internally by the top-level tracking action
4869b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor  unsigned &getCurrentTopLevelHashValue() { return CurrentTopLevelHashValue; }
4879b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor
48889d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  typedef std::vector<PreprocessedEntity *>::iterator pp_entity_iterator;
48989d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor
49089d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  pp_entity_iterator pp_entity_begin();
49189d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  pp_entity_iterator pp_entity_end();
49289d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor
49389d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  /// \brief Add a new preprocessed entity that's stored at the given offset
49489d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  /// in the precompiled preamble.
49589d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  void addPreprocessedEntityFromPreamble(uint64_t Offset) {
49689d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor    PreprocessedEntitiesInPreamble.push_back(Offset);
49789d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  }
49889d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor
499788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  /// \brief Retrieve the mapping from File IDs to the preprocessed entities
500788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  /// within that file.
501788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  PreprocessedEntitiesByFileMap &getPreprocessedEntitiesByFile() {
502788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    return PreprocessedEntitiesByFile;
503788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  }
504788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
505a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor  // Retrieve the diagnostics associated with this AST
506405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor  typedef const StoredDiagnostic *stored_diag_iterator;
507405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor  stored_diag_iterator stored_diag_begin() const {
508405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor    return StoredDiagnostics.begin();
509405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor  }
510405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor  stored_diag_iterator stored_diag_end() const {
511405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor    return StoredDiagnostics.end();
512405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor  }
513405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor  unsigned stored_diag_size() const { return StoredDiagnostics.size(); }
514405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor
515405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor  llvm::SmallVector<StoredDiagnostic, 4> &getStoredDiagnostics() {
516405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor    return StoredDiagnostics;
517a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor  }
518a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor
51987c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  typedef std::vector<CachedCodeCompletionResult>::iterator
52087c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    cached_completion_iterator;
52187c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor
52287c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  cached_completion_iterator cached_completion_begin() {
52387c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    return CachedCompletionResults.begin();
52487c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  }
52587c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor
52687c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  cached_completion_iterator cached_completion_end() {
52787c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    return CachedCompletionResults.end();
52887c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  }
52987c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor
53087c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  unsigned cached_completion_size() const {
53187c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    return CachedCompletionResults.size();
53287c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  }
533389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis
534389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis  llvm::MemoryBuffer *getBufferForFile(llvm::StringRef Filename,
53575dfb65c38d51772df9a00ce2d2feeefd55667adChris Lattner                                       std::string *ErrorStr = 0);
536389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis
537df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor  /// \brief Whether this AST represents a complete translation unit.
538df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor  ///
539df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor  /// If false, this AST is only a partial translation unit, e.g., one
540df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor  /// that might still be used as a precompiled header or preamble.
541df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor  bool isCompleteTranslationUnit() const { return CompleteTranslationUnit; }
542df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor
543b1c86492f9a9bef01a4567408c22f961bbd604feArgyrios Kyrtzidis  typedef llvm::PointerUnion<const char *, const llvm::MemoryBuffer *>
544b1c86492f9a9bef01a4567408c22f961bbd604feArgyrios Kyrtzidis      FilenameOrMemBuf;
5454db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  /// \brief A mapping from a file name to the memory buffer that stores the
5464db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  /// remapped contents of that file.
547b1c86492f9a9bef01a4567408c22f961bbd604feArgyrios Kyrtzidis  typedef std::pair<std::string, FilenameOrMemBuf> RemappedFile;
548b0f4b9a558933b307073f7cd7753602f94354ae9Argyrios Kyrtzidis
549b0f4b9a558933b307073f7cd7753602f94354ae9Argyrios Kyrtzidis  /// \brief Create a ASTUnit. Gets ownership of the passed CompilerInvocation.
550b0f4b9a558933b307073f7cd7753602f94354ae9Argyrios Kyrtzidis  static ASTUnit *create(CompilerInvocation *CI,
551b0f4b9a558933b307073f7cd7753602f94354ae9Argyrios Kyrtzidis                         llvm::IntrusiveRefCntPtr<Diagnostic> Diags);
552b0f4b9a558933b307073f7cd7753602f94354ae9Argyrios Kyrtzidis
5533c7f4134603d04b44f997b43c0a9def270f25386Sebastian Redl  /// \brief Create a ASTUnit from an AST file.
5540853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis  ///
5553c7f4134603d04b44f997b43c0a9def270f25386Sebastian Redl  /// \param Filename - The AST file to load.
5560853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis  ///
5575262fda30b876c8aae95f2eb92e349418d6b14bbDaniel Dunbar  /// \param Diags - The diagnostics engine to use for reporting errors; its
5585262fda30b876c8aae95f2eb92e349418d6b14bbDaniel Dunbar  /// lifetime is expected to extend past that of the returned ASTUnit.
55931b87d8006d4863dd9b17e515ac720941efc38e3Daniel Dunbar  ///
5603c7f4134603d04b44f997b43c0a9def270f25386Sebastian Redl  /// \returns - The initialized ASTUnit or null if the AST failed to load.
5613c7f4134603d04b44f997b43c0a9def270f25386Sebastian Redl  static ASTUnit *LoadFromASTFile(const std::string &Filename,
56228019772db70d4547be05a042eb950bc910f134fDouglas Gregor                                  llvm::IntrusiveRefCntPtr<Diagnostic> Diags,
563389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis                                  const FileSystemOptions &FileSystemOpts,
5645cf48766d626ff6b223acc9d4b7e415ca8480836Ted Kremenek                                  bool OnlyLocalDecls = false,
5654db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor                                  RemappedFile *RemappedFiles = 0,
566a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                  unsigned NumRemappedFiles = 0,
567a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                  bool CaptureDiagnostics = false);
568521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar
5694cd912aa94656697a44c3ebb159f05060300524eDouglas Gregorprivate:
5704cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor  /// \brief Helper function for \c LoadFromCompilerInvocation() and
5714cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor  /// \c LoadFromCommandLine(), which loads an AST from a compiler invocation.
5724cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor  ///
5734cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor  /// \param PrecompilePreamble Whether to precompile the preamble of this
5744cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor  /// translation unit, to improve the performance of reparsing.
5754cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor  ///
5764cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor  /// \returns \c true if a catastrophic failure occurred (which means that the
5774cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor  /// \c ASTUnit itself is invalid), or \c false otherwise.
5784cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor  bool LoadFromCompilerInvocation(bool PrecompilePreamble);
5794cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor
5804cd912aa94656697a44c3ebb159f05060300524eDouglas Gregorpublic:
5814cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor
582d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis  /// \brief Create an ASTUnit from a source file, via a CompilerInvocation
583d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis  /// object, by invoking the optionally provided ASTFrontendAction.
584d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis  ///
585d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis  /// \param CI - The compiler invocation to use; it must have exactly one input
586d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis  /// source file. The ASTUnit takes ownership of the CompilerInvocation object.
587d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis  ///
588d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis  /// \param Diags - The diagnostics engine to use for reporting errors; its
589d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis  /// lifetime is expected to extend past that of the returned ASTUnit.
590d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis  ///
591d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis  /// \param Action - The ASTFrontendAction to invoke. Its ownership is not
592d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis  /// transfered.
593d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis  static ASTUnit *LoadFromCompilerInvocationAction(CompilerInvocation *CI,
594d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis                                     llvm::IntrusiveRefCntPtr<Diagnostic> Diags,
595d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis                                             ASTFrontendAction *Action = 0);
596d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis
597521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar  /// LoadFromCompilerInvocation - Create an ASTUnit from a source file, via a
598521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar  /// CompilerInvocation object.
599521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar  ///
600521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar  /// \param CI - The compiler invocation to use; it must have exactly one input
601f7acc37450d59ef751df73acb91de73850cc6517Daniel Dunbar  /// source file. The ASTUnit takes ownership of the CompilerInvocation object.
602521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar  ///
6035262fda30b876c8aae95f2eb92e349418d6b14bbDaniel Dunbar  /// \param Diags - The diagnostics engine to use for reporting errors; its
6045262fda30b876c8aae95f2eb92e349418d6b14bbDaniel Dunbar  /// lifetime is expected to extend past that of the returned ASTUnit.
605521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar  //
606521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar  // FIXME: Move OnlyLocalDecls, UseBumpAllocator to setters on the ASTUnit, we
607521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar  // shouldn't need to specify them at construction time.
608f7acc37450d59ef751df73acb91de73850cc6517Daniel Dunbar  static ASTUnit *LoadFromCompilerInvocation(CompilerInvocation *CI,
60928019772db70d4547be05a042eb950bc910f134fDouglas Gregor                                     llvm::IntrusiveRefCntPtr<Diagnostic> Diags,
610a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                             bool OnlyLocalDecls = false,
61144c181aec37789f25f6c15543c164416f72e562aDouglas Gregor                                             bool CaptureDiagnostics = false,
612df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor                                             bool PrecompilePreamble = false,
61387c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor                                          bool CompleteTranslationUnit = true,
614dca8ee8b7bc86076916a3a80f553f7a4e98c14afDouglas Gregor                                       bool CacheCodeCompletionResults = false,
615dca8ee8b7bc86076916a3a80f553f7a4e98c14afDouglas Gregor                                       bool NestedMacroInstantiations = true);
616521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar
6177b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar  /// LoadFromCommandLine - Create an ASTUnit from a vector of command line
6187b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar  /// arguments, which must specify exactly one source file.
6197b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar  ///
6207b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar  /// \param ArgBegin - The beginning of the argument vector.
6217b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar  ///
6227b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar  /// \param ArgEnd - The end of the argument vector.
6237b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar  ///
6245262fda30b876c8aae95f2eb92e349418d6b14bbDaniel Dunbar  /// \param Diags - The diagnostics engine to use for reporting errors; its
6255262fda30b876c8aae95f2eb92e349418d6b14bbDaniel Dunbar  /// lifetime is expected to extend past that of the returned ASTUnit.
6267b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar  ///
627869824e87940f97b87064db2df2861e82e08a8c6Daniel Dunbar  /// \param ResourceFilesPath - The path to the compiler resource files.
6287b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar  //
6297b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar  // FIXME: Move OnlyLocalDecls, UseBumpAllocator to setters on the ASTUnit, we
6307b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar  // shouldn't need to specify them at construction time.
6317b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar  static ASTUnit *LoadFromCommandLine(const char **ArgBegin,
6327b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar                                      const char **ArgEnd,
63328019772db70d4547be05a042eb950bc910f134fDouglas Gregor                                    llvm::IntrusiveRefCntPtr<Diagnostic> Diags,
634869824e87940f97b87064db2df2861e82e08a8c6Daniel Dunbar                                      llvm::StringRef ResourceFilesPath,
6357b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar                                      bool OnlyLocalDecls = false,
636e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor                                      bool CaptureDiagnostics = false,
6374db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor                                      RemappedFile *RemappedFiles = 0,
638a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                      unsigned NumRemappedFiles = 0,
639299a4a967b02c9f0d0d94ad8560e3ced893f9116Argyrios Kyrtzidis                                      bool RemappedFilesKeepOriginalName = true,
640df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor                                      bool PrecompilePreamble = false,
64187c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor                                      bool CompleteTranslationUnit = true,
64299ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor                                      bool CacheCodeCompletionResults = false,
64399ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor                                      bool CXXPrecompilePreamble = false,
644dca8ee8b7bc86076916a3a80f553f7a4e98c14afDouglas Gregor                                      bool CXXChainedPCH = false,
645dca8ee8b7bc86076916a3a80f553f7a4e98c14afDouglas Gregor                                      bool NestedMacroInstantiations = true);
646abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor
647abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  /// \brief Reparse the source files using the same command-line options that
648abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  /// were originally used to produce this translation unit.
649abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  ///
650abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  /// \returns True if a failure occurred that causes the ASTUnit not to
651abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  /// contain any translation-unit information, false otherwise.
652abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  bool Reparse(RemappedFile *RemappedFiles = 0,
653abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor               unsigned NumRemappedFiles = 0);
6541abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor
6551abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor  /// \brief Perform code completion at the given file, line, and
6561abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor  /// column within this translation unit.
6571abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor  ///
658cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor  /// \param File The file in which code completion will occur.
659cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor  ///
660cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor  /// \param Line The line at which code completion will occur.
661cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor  ///
662cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor  /// \param Column The column at which code completion will occur.
663cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor  ///
664cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor  /// \param IncludeMacros Whether to include macros in the code-completion
665cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor  /// results.
666cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor  ///
667cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor  /// \param IncludeCodePatterns Whether to include code patterns (such as a
668cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor  /// for loop) in the code-completion results.
6691abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor  ///
6702283d79155a3e82442fce124ce5fd704ca138801Douglas Gregor  /// FIXME: The Diag, LangOpts, SourceMgr, FileMgr, StoredDiagnostics, and
6712283d79155a3e82442fce124ce5fd704ca138801Douglas Gregor  /// OwnedBuffers parameters are all disgusting hacks. They will go away.
6721abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor  void CodeComplete(llvm::StringRef File, unsigned Line, unsigned Column,
6731abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor                    RemappedFile *RemappedFiles, unsigned NumRemappedFiles,
674cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor                    bool IncludeMacros, bool IncludeCodePatterns,
6751abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor                    CodeCompleteConsumer &Consumer,
6761abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor                    Diagnostic &Diag, LangOptions &LangOpts,
6771abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor                    SourceManager &SourceMgr, FileManager &FileMgr,
6782283d79155a3e82442fce124ce5fd704ca138801Douglas Gregor                    llvm::SmallVectorImpl<StoredDiagnostic> &StoredDiagnostics,
6792283d79155a3e82442fce124ce5fd704ca138801Douglas Gregor              llvm::SmallVectorImpl<const llvm::MemoryBuffer *> &OwnedBuffers);
6807ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor
6817ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor  /// \brief Save this translation unit to a file with the given name.
6827ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor  ///
6837ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor  /// \returns True if an error occurred, false otherwise.
6847ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor  bool Save(llvm::StringRef File);
685b0f4b9a558933b307073f7cd7753602f94354ae9Argyrios Kyrtzidis
686b0f4b9a558933b307073f7cd7753602f94354ae9Argyrios Kyrtzidis  /// \brief Serialize this translation unit with the given output stream.
687b0f4b9a558933b307073f7cd7753602f94354ae9Argyrios Kyrtzidis  ///
688b0f4b9a558933b307073f7cd7753602f94354ae9Argyrios Kyrtzidis  /// \returns True if an error occurred, false otherwise.
689b0f4b9a558933b307073f7cd7753602f94354ae9Argyrios Kyrtzidis  bool serialize(llvm::raw_ostream &OS);
6900853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis};
6910853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis
6920853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis} // namespace clang
6930853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis
6940853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis#endif
695