ASTUnit.h revision f62d43d2afe1960755a1b5813cae1e5983bcac1b
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; 44f62d43d2afe1960755a1b5813cae1e5983bcac1bDouglas Gregorclass ASTReader; 451abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregorclass CodeCompleteConsumer; 46521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass CompilerInvocation; 47521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass Decl; 48521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass Diagnostic; 49521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass FileEntry; 50521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass FileManager; 51521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass HeaderSearch; 52521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass Preprocessor; 53521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass SourceManager; 54521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass TargetInfo; 55d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidisclass ASTFrontendAction; 560853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis 57f96b524306ccfa623235d375deee79637bd38f29Steve Naroffusing namespace idx; 5844c181aec37789f25f6c15543c164416f72e562aDouglas Gregor 5948601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor/// \brief Allocator for a cached set of global code completions. 6048601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregorclass GlobalCodeCompletionAllocator 6148601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor : public CodeCompletionAllocator, 6248601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor public llvm::RefCountedBase<GlobalCodeCompletionAllocator> 6348601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor{ 6448601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor 6548601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor}; 6648601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor 673c7f4134603d04b44f997b43c0a9def270f25386Sebastian Redl/// \brief Utility class for loading a ASTContext from an AST file. 680853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis/// 690853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidisclass ASTUnit { 70788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregorpublic: 71788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor typedef std::map<FileID, std::vector<PreprocessedEntity *> > 72788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor PreprocessedEntitiesByFileMap; 73bdbb004f38978da0c4a75af3294d1c7b5ff84af1Douglas Gregor 74788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregorprivate: 7528019772db70d4547be05a042eb950bc910f134fDouglas Gregor llvm::IntrusiveRefCntPtr<Diagnostic> Diagnostics; 764f32786ac45210143654390177105eb749b614e9Ted Kremenek llvm::IntrusiveRefCntPtr<FileManager> FileMgr; 774f32786ac45210143654390177105eb749b614e9Ted Kremenek llvm::IntrusiveRefCntPtr<SourceManager> SourceMgr; 780853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis llvm::OwningPtr<HeaderSearch> HeaderInfo; 794f32786ac45210143654390177105eb749b614e9Ted Kremenek llvm::IntrusiveRefCntPtr<TargetInfo> Target; 804f32786ac45210143654390177105eb749b614e9Ted Kremenek llvm::IntrusiveRefCntPtr<Preprocessor> PP; 814f32786ac45210143654390177105eb749b614e9Ted Kremenek llvm::IntrusiveRefCntPtr<ASTContext> Ctx; 82389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis 83389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis FileSystemOptions FileSystemOpts; 84389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis 85914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor /// \brief The AST consumer that received information about the translation 86914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor /// unit as it was parsed or loaded. 87914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor llvm::OwningPtr<ASTConsumer> Consumer; 88914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor 89914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor /// \brief The semantic analysis object used to type-check the translation 90914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor /// unit. 91914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor llvm::OwningPtr<Sema> TheSema; 92914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor 93807b06157a1a5c050520fc194d32f16d22d423a8Daniel Dunbar /// Optional owned invocation, just used to make the invocation used in 94807b06157a1a5c050520fc194d32f16d22d423a8Daniel Dunbar /// LoadFromCommandLine available. 954f32786ac45210143654390177105eb749b614e9Ted Kremenek llvm::IntrusiveRefCntPtr<CompilerInvocation> Invocation; 96914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor 971aa27307c462baaa9e5fda14ff6797dd39fe8b84Douglas Gregor /// \brief The set of target features. 981aa27307c462baaa9e5fda14ff6797dd39fe8b84Douglas Gregor /// 991aa27307c462baaa9e5fda14ff6797dd39fe8b84Douglas Gregor /// FIXME: each time we reparse, we need to restore the set of target 1001aa27307c462baaa9e5fda14ff6797dd39fe8b84Douglas Gregor /// features from this vector, because TargetInfo::CreateTargetInfo() 1011aa27307c462baaa9e5fda14ff6797dd39fe8b84Douglas Gregor /// mangles the target options in place. Yuck! 1021aa27307c462baaa9e5fda14ff6797dd39fe8b84Douglas Gregor std::vector<std::string> TargetFeatures; 1031aa27307c462baaa9e5fda14ff6797dd39fe8b84Douglas Gregor 1047d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor // OnlyLocalDecls - when true, walking this AST should only visit declarations 1057d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor // that come from the AST itself, not from included precompiled headers. 1067d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor // FIXME: This is temporary; eventually, CIndex will always do this. 1077d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor bool OnlyLocalDecls; 108c7822dbf3c01a2a5f837cff82ba7889ea755dacaDaniel Dunbar 109abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor /// \brief Whether to capture any diagnostics produced. 110abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor bool CaptureDiagnostics; 111e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor 112df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor /// \brief Track whether the main file was loaded from an AST or not. 113c7822dbf3c01a2a5f837cff82ba7889ea755dacaDaniel Dunbar bool MainFileIsAST; 114c7822dbf3c01a2a5f837cff82ba7889ea755dacaDaniel Dunbar 115df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor /// \brief Whether this AST represents a complete translation unit. 116df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor bool CompleteTranslationUnit; 117df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor 118213f18b3d654de7d1c7cf4a329ea9d3db1c50b6aDouglas Gregor /// \brief Whether we should time each operation. 119213f18b3d654de7d1c7cf4a329ea9d3db1c50b6aDouglas Gregor bool WantTiming; 12015727ddb11405c45372150b5bfb07dbfa4c9960bArgyrios Kyrtzidis 12115727ddb11405c45372150b5bfb07dbfa4c9960bArgyrios Kyrtzidis /// \brief Whether the ASTUnit should delete the remapped buffers. 12215727ddb11405c45372150b5bfb07dbfa4c9960bArgyrios Kyrtzidis bool OwnsRemappedFileBuffers; 123213f18b3d654de7d1c7cf4a329ea9d3db1c50b6aDouglas Gregor 124f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar /// Track the top-level decls which appeared in an ASTUnit which was loaded 125f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar /// from a source file. 126f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar // 127f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar // FIXME: This is just an optimization hack to avoid deserializing large parts 128f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar // of a PCH file when using the Index library on an ASTUnit loaded from 129f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar // source. In the long term we should make the Index library use efficient and 130f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar // more scalable search mechanisms. 131f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar std::vector<Decl*> TopLevelDecls; 132f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar 13389d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor /// \brief The list of preprocessed entities which appeared when the ASTUnit 13489d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor /// was loaded. 13589d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor /// 13689d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor /// FIXME: This is just an optimization hack to avoid deserializing large 13789d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor /// parts of a PCH file while performing a walk or search. In the long term, 13889d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor /// we should provide more scalable search mechanisms. 13989d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor std::vector<PreprocessedEntity *> PreprocessedEntities; 14089d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor 14168d40e2d16b9fadba386853d6bbb60089291fdc5Daniel Dunbar /// The name of the original source file used to generate this ASTUnit. 14268d40e2d16b9fadba386853d6bbb60089291fdc5Daniel Dunbar std::string OriginalSourceFile; 14368d40e2d16b9fadba386853d6bbb60089291fdc5Daniel Dunbar 144f96b524306ccfa623235d375deee79637bd38f29Steve Naroff // Critical optimization when using clang_getCursor(). 145f96b524306ccfa623235d375deee79637bd38f29Steve Naroff ASTLocation LastLoc; 146f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar 147f62d43d2afe1960755a1b5813cae1e5983bcac1bDouglas Gregor /// \brief The set of diagnostics produced when creating the preamble. 148f62d43d2afe1960755a1b5813cae1e5983bcac1bDouglas Gregor llvm::SmallVector<StoredDiagnostic, 4> PreambleDiagnostics; 149f62d43d2afe1960755a1b5813cae1e5983bcac1bDouglas Gregor 150a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor /// \brief The set of diagnostics produced when creating this 151a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor /// translation unit. 152405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor llvm::SmallVector<StoredDiagnostic, 4> StoredDiagnostics; 153a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor 1544cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor /// \brief The number of stored diagnostics that come from the driver 1554cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor /// itself. 1564cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor /// 1574cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor /// Diagnostics that come from the driver are retained from one parse to 1584cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor /// the next. 1594cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor unsigned NumStoredDiagnosticsFromDriver; 1604cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor 161313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor /// \brief Temporary files that should be removed when the ASTUnit is 162313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor /// destroyed. 163313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor llvm::SmallVector<llvm::sys::Path, 4> TemporaryFiles; 164bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor 165788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor /// \brief A mapping from file IDs to the set of preprocessed entities 166788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor /// stored in that file. 167788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor /// 168788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor /// FIXME: This is just an optimization hack to avoid searching through 169788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor /// many preprocessed entities during cursor traversal in the CIndex library. 170788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor /// Ideally, we would just be able to perform a binary search within the 171788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor /// list of preprocessed entities. 172788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor PreprocessedEntitiesByFileMap PreprocessedEntitiesByFile; 173788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor 174bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor /// \brief Simple hack to allow us to assert that ASTUnit is not being 175bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor /// used concurrently, which is not supported. 176bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor /// 177bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor /// Clients should create instances of the ConcurrencyCheck class whenever 178bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor /// using the ASTUnit in a way that isn't intended to be concurrent, which is 179bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor /// just about any usage. 180bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor unsigned int ConcurrencyCheckValue; 181bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor static const unsigned int CheckLocked = 28573289; 182bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor static const unsigned int CheckUnlocked = 9803453; 183175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor 184eababfbddb74d186f78783a9731a78ad371c9800Douglas Gregor /// \brief Counter that determines when we want to try building a 185eababfbddb74d186f78783a9731a78ad371c9800Douglas Gregor /// precompiled preamble. 186eababfbddb74d186f78783a9731a78ad371c9800Douglas Gregor /// 187eababfbddb74d186f78783a9731a78ad371c9800Douglas Gregor /// If zero, we will never build a precompiled preamble. Otherwise, 188eababfbddb74d186f78783a9731a78ad371c9800Douglas Gregor /// it's treated as a counter that decrements each time we reparse 189eababfbddb74d186f78783a9731a78ad371c9800Douglas Gregor /// without the benefit of a precompiled preamble. When it hits 1, 190eababfbddb74d186f78783a9731a78ad371c9800Douglas Gregor /// we'll attempt to rebuild the precompiled header. This way, if 191eababfbddb74d186f78783a9731a78ad371c9800Douglas Gregor /// building the precompiled preamble fails, we won't try again for 192eababfbddb74d186f78783a9731a78ad371c9800Douglas Gregor /// some number of calls. 193eababfbddb74d186f78783a9731a78ad371c9800Douglas Gregor unsigned PreambleRebuildCounter; 194eababfbddb74d186f78783a9731a78ad371c9800Douglas Gregor 195175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor /// \brief The file in which the precompiled preamble is stored. 196385103b79c5338a2be5da0ca70652400bc267371Douglas Gregor std::string PreambleFile; 197175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor 198175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor /// \brief The contents of the preamble that has been precompiled to 199175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor /// \c PreambleFile. 200175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor std::vector<char> Preamble; 201175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor 202f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor /// \brief Whether the preamble ends at the start of a new line. 203f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor /// 204f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor /// Used to inform the lexer as to whether it's starting at the beginning of 205f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor /// a line after skipping the preamble. 206f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor bool PreambleEndsAtStartOfLine; 207f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor 208175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor /// \brief The size of the source buffer that we've reserved for the main 209175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor /// file within the precompiled preamble. 210175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor unsigned PreambleReservedSize; 211cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor 212cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor /// \brief Keeps track of the files that were used when computing the 213cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor /// preamble, with both their buffer size and their modification time. 214cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor /// 215cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor /// If any of the files have changed from one compile to the next, 216cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor /// the preamble must be thrown away. 217cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor llvm::StringMap<std::pair<off_t, time_t> > FilesInPreamble; 218cc5888d833caf90ebda37f24da40d2cd06b4d820Douglas Gregor 21928233428da1ebec20c893d6297ae3191318940ddDouglas Gregor /// \brief When non-NULL, this is the buffer used to store the contents of 22028233428da1ebec20c893d6297ae3191318940ddDouglas Gregor /// the main file when it has been padded for use with the precompiled 22128233428da1ebec20c893d6297ae3191318940ddDouglas Gregor /// preamble. 22228233428da1ebec20c893d6297ae3191318940ddDouglas Gregor llvm::MemoryBuffer *SavedMainFileBuffer; 223c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor 224671947b18dba342f9aba022ee992babef325a833Douglas Gregor /// \brief When non-NULL, this is the buffer used to store the 225671947b18dba342f9aba022ee992babef325a833Douglas Gregor /// contents of the preamble when it has been padded to build the 226671947b18dba342f9aba022ee992babef325a833Douglas Gregor /// precompiled preamble. 227671947b18dba342f9aba022ee992babef325a833Douglas Gregor llvm::MemoryBuffer *PreambleBuffer; 228671947b18dba342f9aba022ee992babef325a833Douglas Gregor 229c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor /// \brief The number of warnings that occurred while parsing the preamble. 230c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor /// 231c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor /// This value will be used to restore the state of the \c Diagnostic object 232c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor /// when re-using the precompiled preamble. Note that only the 233c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor /// number of warnings matters, since we will not save the preamble 234c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor /// when any errors are present. 235c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor unsigned NumWarningsInPreamble; 236c0659ec614c428c7d15746fcad15d50a2703751dDouglas Gregor 2373c7f4134603d04b44f997b43c0a9def270f25386Sebastian Redl /// \brief A list of the serialization ID numbers for each of the top-level 238eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor /// declarations parsed within the precompiled preamble. 2398538e8d43a3a9bd439c987c0de37bcbf035dd391Sebastian Redl std::vector<serialization::DeclID> TopLevelDeclsInPreamble; 240eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor 24189d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor /// \brief A list of the offsets into the precompiled preamble which 24289d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor /// correspond to preprocessed entities. 24389d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor std::vector<uint64_t> PreprocessedEntitiesInPreamble; 24489d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor 24587c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor /// \brief Whether we should be caching code-completion results. 24687c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor bool ShouldCacheCodeCompletionResults; 24787c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor 248ba7537febdf1bc1cc617e1f1746f2644feba6274Chandler Carruth /// \brief Whether we want to include nested macro expansions in the 249dca8ee8b7bc86076916a3a80f553f7a4e98c14afDouglas Gregor /// detailed preprocessing record. 250ba7537febdf1bc1cc617e1f1746f2644feba6274Chandler Carruth bool NestedMacroExpansions; 251dca8ee8b7bc86076916a3a80f553f7a4e98c14afDouglas Gregor 252e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor static void ConfigureDiags(llvm::IntrusiveRefCntPtr<Diagnostic> &Diags, 2530b53cf834346d78985aaa9e7300445a39c245614Douglas Gregor const char **ArgBegin, const char **ArgEnd, 254e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor ASTUnit &AST, bool CaptureDiagnostics); 255e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor 256f62d43d2afe1960755a1b5813cae1e5983bcac1bDouglas Gregor void TranslateStoredDiagnostics(ASTReader *MMan, llvm::StringRef ModName, 257f62d43d2afe1960755a1b5813cae1e5983bcac1bDouglas Gregor SourceManager &SrcMan, 258f62d43d2afe1960755a1b5813cae1e5983bcac1bDouglas Gregor const llvm::SmallVectorImpl<StoredDiagnostic> &Diags, 259f62d43d2afe1960755a1b5813cae1e5983bcac1bDouglas Gregor llvm::SmallVectorImpl<StoredDiagnostic> &Out); 260f62d43d2afe1960755a1b5813cae1e5983bcac1bDouglas Gregor 26187c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregorpublic: 26287c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor /// \brief A cached code-completion result, which may be introduced in one of 26387c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor /// many different contexts. 26487c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor struct CachedCodeCompletionResult { 26587c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor /// \brief The code-completion string corresponding to this completion 26687c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor /// result. 26787c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor CodeCompletionString *Completion; 26887c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor 26987c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor /// \brief A bitmask that indicates which code-completion contexts should 27087c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor /// contain this completion result. 27187c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor /// 27287c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor /// The bits in the bitmask correspond to the values of 27387c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor /// CodeCompleteContext::Kind. To map from a completion context kind to a 27487c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor /// bit, subtract one from the completion context kind and shift 1 by that 27587c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor /// number of bits. Many completions can occur in several different 27687c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor /// contexts. 27787c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor unsigned ShowInContexts; 27887c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor 27987c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor /// \brief The priority given to this code-completion result. 28087c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor unsigned Priority; 28187c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor 28287c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor /// \brief The libclang cursor kind corresponding to this code-completion 28387c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor /// result. 28487c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor CXCursorKind Kind; 2851827e10051638770ad9ccf3e285caf95f995afd1Douglas Gregor 28658ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor /// \brief The availability of this code-completion result. 28758ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor CXAvailabilityKind Availability; 28858ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor 2891827e10051638770ad9ccf3e285caf95f995afd1Douglas Gregor /// \brief The simplified type class for a non-macro completion result. 2901827e10051638770ad9ccf3e285caf95f995afd1Douglas Gregor SimplifiedTypeClass TypeClass; 291f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor 292f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor /// \brief The type of a non-macro completion result, stored as a unique 293f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor /// integer used by the string map of cached completion types. 294f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor /// 295f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor /// This value will be zero if the type is not known, or a unique value 296f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor /// determined by the formatted type string. Se \c CachedCompletionTypes 297f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor /// for more information. 298f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor unsigned Type; 29987c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor }; 30087c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor 301f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor /// \brief Retrieve the mapping from formatted type names to unique type 302f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor /// identifiers. 303f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor llvm::StringMap<unsigned> &getCachedCompletionTypes() { 304f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor return CachedCompletionTypes; 305f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor } 306f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor 30748601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor /// \brief Retrieve the allocator used to cache global code completions. 30848601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor llvm::IntrusiveRefCntPtr<GlobalCodeCompletionAllocator> 30948601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor getCachedCompletionAllocator() { 31048601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor return CachedCompletionAllocator; 31148601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor } 31248601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor 31387c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregorprivate: 314218937c13ef5b0625a70aad41ca7a92da9278bd2Douglas Gregor /// \brief Allocator used to store cached code completions. 31548601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor llvm::IntrusiveRefCntPtr<GlobalCodeCompletionAllocator> 31648601b32321496b07a18fb6631f8563275d8c5fbDouglas Gregor CachedCompletionAllocator; 317218937c13ef5b0625a70aad41ca7a92da9278bd2Douglas Gregor 31887c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor /// \brief The set of cached code-completion results. 31987c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor std::vector<CachedCodeCompletionResult> CachedCompletionResults; 32087c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor 321f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor /// \brief A mapping from the formatted type name to a unique number for that 322f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor /// type, which is used for type equality comparisons. 323f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor llvm::StringMap<unsigned> CachedCompletionTypes; 324f5586f6b311c98e1022a8fe0609053849b70d323Douglas Gregor 3259b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor /// \brief A string hash of the top-level declaration and macro definition 3269b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor /// names processed the last time that we reparsed the file. 327727d93ef49e18147149354fadd10e86b13bc4ab0Douglas Gregor /// 3289b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor /// This hash value is used to determine when we need to refresh the 3299b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor /// global code-completion cache. 3309b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor unsigned CompletionCacheTopLevelHashValue; 331727d93ef49e18147149354fadd10e86b13bc4ab0Douglas Gregor 3329b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor /// \brief A string hash of the top-level declaration and macro definition 3339b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor /// names processed the last time that we reparsed the precompiled preamble. 334727d93ef49e18147149354fadd10e86b13bc4ab0Douglas Gregor /// 3359b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor /// This hash value is used to determine when we need to refresh the 3369b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor /// global code-completion cache after a rebuild of the precompiled preamble. 3379b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor unsigned PreambleTopLevelHashValue; 338ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar 3399b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor /// \brief The current hash value for the top-level declaration and macro 3409b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor /// definition names 3419b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor unsigned CurrentTopLevelHashValue; 3429b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor 343ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar /// \brief Bit used by CIndex to mark when a translation unit may be in an 344ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar /// inconsistent state, and is not safe to free. 345ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar unsigned UnsafeToFree : 1; 346ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar 347c4421e966d77a18f815284175b3fcbb46f36fa39Douglas Gregor /// \brief Cache any "global" code-completion results, so that we can avoid 348c4421e966d77a18f815284175b3fcbb46f36fa39Douglas Gregor /// recomputing them with each completion. 34987c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor void CacheCodeCompletionResults(); 35087c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor 35187c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor /// \brief Clear out and deallocate 35287c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor void ClearCachedCompletionResults(); 35387c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor 35431b87d8006d4863dd9b17e515ac720941efc38e3Daniel Dunbar ASTUnit(const ASTUnit&); // DO NOT IMPLEMENT 35531b87d8006d4863dd9b17e515ac720941efc38e3Daniel Dunbar ASTUnit &operator=(const ASTUnit &); // DO NOT IMPLEMENT 356bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor 3573687e9d3a5dbfa9963af02a49a2b139d91310813Douglas Gregor explicit ASTUnit(bool MainFileIsAST); 3588b96253907c47141af0b7b2a44a368748d006a87Douglas Gregor 359abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor void CleanTemporaryFiles(); 360754f3490c5b0f5d83361f001bc87944f23644abbDouglas Gregor bool Parse(llvm::MemoryBuffer *OverrideMainBuffer); 361175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor 362f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor std::pair<llvm::MemoryBuffer *, std::pair<unsigned, bool> > 363df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor ComputePreamble(CompilerInvocation &Invocation, 364df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor unsigned MaxLines, bool &CreatedBuffer); 365175c4a9aa61f4449f27b729737e4438684ac6d92Douglas Gregor 366df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor llvm::MemoryBuffer *getMainBufferWithPrecompiledPreamble( 36701b6e31a62e2265849f4388b9be6be0a5d13348dDouglas Gregor const CompilerInvocation &PreambleInvocationIn, 368df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor bool AllowRebuild = true, 369df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor unsigned MaxLines = 0); 370eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor void RealizeTopLevelDeclsFromPreamble(); 37189d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor void RealizePreprocessedEntitiesFromPreamble(); 37289d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor 3730853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidispublic: 374bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor class ConcurrencyCheck { 375bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor volatile ASTUnit &Self; 376bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor 377bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor public: 378bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor explicit ConcurrencyCheck(ASTUnit &Self) 379bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor : Self(Self) 380bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor { 381bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor assert(Self.ConcurrencyCheckValue == CheckUnlocked && 382bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor "Concurrent access to ASTUnit!"); 383bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor Self.ConcurrencyCheckValue = CheckLocked; 384bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor } 385bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor 386bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor ~ConcurrencyCheck() { 387bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor Self.ConcurrencyCheckValue = CheckUnlocked; 388bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor } 389bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor }; 390bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor friend class ConcurrencyCheck; 391bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor 3920853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis ~ASTUnit(); 3930853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis 394c7822dbf3c01a2a5f837cff82ba7889ea755dacaDaniel Dunbar bool isMainFileAST() const { return MainFileIsAST; } 395c7822dbf3c01a2a5f837cff82ba7889ea755dacaDaniel Dunbar 396ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar bool isUnsafeToFree() const { return UnsafeToFree; } 397ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar void setUnsafeToFree(bool Value) { UnsafeToFree = Value; } 398ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar 3993687e9d3a5dbfa9963af02a49a2b139d91310813Douglas Gregor const Diagnostic &getDiagnostics() const { return *Diagnostics; } 4003687e9d3a5dbfa9963af02a49a2b139d91310813Douglas Gregor Diagnostic &getDiagnostics() { return *Diagnostics; } 4013687e9d3a5dbfa9963af02a49a2b139d91310813Douglas Gregor 402405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor const SourceManager &getSourceManager() const { return *SourceMgr; } 403405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor SourceManager &getSourceManager() { return *SourceMgr; } 4040853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis 4054f32786ac45210143654390177105eb749b614e9Ted Kremenek const Preprocessor &getPreprocessor() const { return *PP; } 4064f32786ac45210143654390177105eb749b614e9Ted Kremenek Preprocessor &getPreprocessor() { return *PP; } 4071eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 4084f32786ac45210143654390177105eb749b614e9Ted Kremenek const ASTContext &getASTContext() const { return *Ctx; } 4094f32786ac45210143654390177105eb749b614e9Ted Kremenek ASTContext &getASTContext() { return *Ctx; } 4100853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis 411914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor bool hasSema() const { return TheSema; } 412914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor Sema &getSema() const { 413914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor assert(TheSema && "ASTUnit does not have a Sema object!"); 414914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor return *TheSema; 415914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor } 416914ed9d30e9abf829a62aa996b083b1e47c19ff6Douglas Gregor 417405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor const FileManager &getFileManager() const { return *FileMgr; } 418405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor FileManager &getFileManager() { return *FileMgr; } 419f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar 420389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis const FileSystemOptions &getFileSystemOpts() const { return FileSystemOpts; } 421389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis 42277accc11f04ed4ff9afd4e27d430144d4714be56Steve Naroff const std::string &getOriginalSourceFileName(); 4233c7f4134603d04b44f997b43c0a9def270f25386Sebastian Redl const std::string &getASTFileName(); 424b85bca2676b433ae555db09de4dd2823ff13b856Zhongxing Xu 425313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor /// \brief Add a temporary file that the ASTUnit depends on. 426313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor /// 427313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor /// This file will be erased when the ASTUnit is destroyed. 428313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor void addTemporaryFile(const llvm::sys::Path &TempFile) { 429313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor TemporaryFiles.push_back(TempFile); 430313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor } 431313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor 4327d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor bool getOnlyLocalDecls() const { return OnlyLocalDecls; } 433f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar 43415727ddb11405c45372150b5bfb07dbfa4c9960bArgyrios Kyrtzidis bool getOwnsRemappedFileBuffers() const { return OwnsRemappedFileBuffers; } 43515727ddb11405c45372150b5bfb07dbfa4c9960bArgyrios Kyrtzidis void setOwnsRemappedFileBuffers(bool val) { OwnsRemappedFileBuffers = val; } 43615727ddb11405c45372150b5bfb07dbfa4c9960bArgyrios Kyrtzidis 437eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor /// \brief Retrieve the maximum PCH level of declarations that a 438eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor /// traversal of the translation unit should consider. 439eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor unsigned getMaxPCHLevel() const; 440eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor 441f96b524306ccfa623235d375deee79637bd38f29Steve Naroff void setLastASTLocation(ASTLocation ALoc) { LastLoc = ALoc; } 442f96b524306ccfa623235d375deee79637bd38f29Steve Naroff ASTLocation getLastASTLocation() const { return LastLoc; } 443f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar 444213f18b3d654de7d1c7cf4a329ea9d3db1c50b6aDouglas Gregor 445213f18b3d654de7d1c7cf4a329ea9d3db1c50b6aDouglas Gregor llvm::StringRef getMainFileName() const; 446213f18b3d654de7d1c7cf4a329ea9d3db1c50b6aDouglas Gregor 447eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor typedef std::vector<Decl *>::iterator top_level_iterator; 448eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor 449eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor top_level_iterator top_level_begin() { 450f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar assert(!isMainFileAST() && "Invalid call for AST based ASTUnit!"); 451eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor if (!TopLevelDeclsInPreamble.empty()) 452eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor RealizeTopLevelDeclsFromPreamble(); 453eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor return TopLevelDecls.begin(); 454f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar } 455eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor 456eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor top_level_iterator top_level_end() { 457f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar assert(!isMainFileAST() && "Invalid call for AST based ASTUnit!"); 458eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor if (!TopLevelDeclsInPreamble.empty()) 459eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor RealizeTopLevelDeclsFromPreamble(); 460eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor return TopLevelDecls.end(); 461eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor } 462eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor 463eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor std::size_t top_level_size() const { 464eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor assert(!isMainFileAST() && "Invalid call for AST based ASTUnit!"); 465eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor return TopLevelDeclsInPreamble.size() + TopLevelDecls.size(); 466eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor } 467eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor 468eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor bool top_level_empty() const { 469eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor assert(!isMainFileAST() && "Invalid call for AST based ASTUnit!"); 470eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor return TopLevelDeclsInPreamble.empty() && TopLevelDecls.empty(); 471eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor } 472eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor 473eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor /// \brief Add a new top-level declaration. 474eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor void addTopLevelDecl(Decl *D) { 475eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor TopLevelDecls.push_back(D); 476eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor } 477eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor 478eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor /// \brief Add a new top-level declaration, identified by its ID in 479eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor /// the precompiled preamble. 4808538e8d43a3a9bd439c987c0de37bcbf035dd391Sebastian Redl void addTopLevelDeclFromPreamble(serialization::DeclID D) { 481eb8837b88c18631c69ac75f64ab1853762063180Douglas Gregor TopLevelDeclsInPreamble.push_back(D); 482f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar } 483f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar 4849b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor /// \brief Retrieve a reference to the current top-level name hash value. 4859b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor /// 4869b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor /// Note: This is used internally by the top-level tracking action 4879b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor unsigned &getCurrentTopLevelHashValue() { return CurrentTopLevelHashValue; } 4889b7db6200d366e4964d63ae1f33c7b9d7b9831cbDouglas Gregor 48989d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor typedef std::vector<PreprocessedEntity *>::iterator pp_entity_iterator; 49089d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor 49189d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor pp_entity_iterator pp_entity_begin(); 49289d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor pp_entity_iterator pp_entity_end(); 49389d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor 49489d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor /// \brief Add a new preprocessed entity that's stored at the given offset 49589d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor /// in the precompiled preamble. 49689d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor void addPreprocessedEntityFromPreamble(uint64_t Offset) { 49789d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor PreprocessedEntitiesInPreamble.push_back(Offset); 49889d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor } 49989d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor 500788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor /// \brief Retrieve the mapping from File IDs to the preprocessed entities 501788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor /// within that file. 502788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor PreprocessedEntitiesByFileMap &getPreprocessedEntitiesByFile() { 503788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor return PreprocessedEntitiesByFile; 504788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor } 505788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor 506a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor // Retrieve the diagnostics associated with this AST 507405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor typedef const StoredDiagnostic *stored_diag_iterator; 508405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor stored_diag_iterator stored_diag_begin() const { 509405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor return StoredDiagnostics.begin(); 510405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor } 511405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor stored_diag_iterator stored_diag_end() const { 512405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor return StoredDiagnostics.end(); 513405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor } 514405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor unsigned stored_diag_size() const { return StoredDiagnostics.size(); } 515405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor 516405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor llvm::SmallVector<StoredDiagnostic, 4> &getStoredDiagnostics() { 517405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor return StoredDiagnostics; 518a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor } 519a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor 52087c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor typedef std::vector<CachedCodeCompletionResult>::iterator 52187c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor cached_completion_iterator; 52287c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor 52387c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor cached_completion_iterator cached_completion_begin() { 52487c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor return CachedCompletionResults.begin(); 52587c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor } 52687c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor 52787c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor cached_completion_iterator cached_completion_end() { 52887c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor return CachedCompletionResults.end(); 52987c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor } 53087c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor 53187c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor unsigned cached_completion_size() const { 53287c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor return CachedCompletionResults.size(); 53387c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor } 534389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis 535389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis llvm::MemoryBuffer *getBufferForFile(llvm::StringRef Filename, 53675dfb65c38d51772df9a00ce2d2feeefd55667adChris Lattner std::string *ErrorStr = 0); 537389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis 538df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor /// \brief Whether this AST represents a complete translation unit. 539df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor /// 540df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor /// If false, this AST is only a partial translation unit, e.g., one 541df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor /// that might still be used as a precompiled header or preamble. 542df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor bool isCompleteTranslationUnit() const { return CompleteTranslationUnit; } 543df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor 544b1c86492f9a9bef01a4567408c22f961bbd604feArgyrios Kyrtzidis typedef llvm::PointerUnion<const char *, const llvm::MemoryBuffer *> 545b1c86492f9a9bef01a4567408c22f961bbd604feArgyrios Kyrtzidis FilenameOrMemBuf; 5464db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor /// \brief A mapping from a file name to the memory buffer that stores the 5474db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor /// remapped contents of that file. 548b1c86492f9a9bef01a4567408c22f961bbd604feArgyrios Kyrtzidis typedef std::pair<std::string, FilenameOrMemBuf> RemappedFile; 549b0f4b9a558933b307073f7cd7753602f94354ae9Argyrios Kyrtzidis 550b0f4b9a558933b307073f7cd7753602f94354ae9Argyrios Kyrtzidis /// \brief Create a ASTUnit. Gets ownership of the passed CompilerInvocation. 551b0f4b9a558933b307073f7cd7753602f94354ae9Argyrios Kyrtzidis static ASTUnit *create(CompilerInvocation *CI, 552b0f4b9a558933b307073f7cd7753602f94354ae9Argyrios Kyrtzidis llvm::IntrusiveRefCntPtr<Diagnostic> Diags); 553b0f4b9a558933b307073f7cd7753602f94354ae9Argyrios Kyrtzidis 5543c7f4134603d04b44f997b43c0a9def270f25386Sebastian Redl /// \brief Create a ASTUnit from an AST file. 5550853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis /// 5563c7f4134603d04b44f997b43c0a9def270f25386Sebastian Redl /// \param Filename - The AST file to load. 5570853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis /// 5585262fda30b876c8aae95f2eb92e349418d6b14bbDaniel Dunbar /// \param Diags - The diagnostics engine to use for reporting errors; its 5595262fda30b876c8aae95f2eb92e349418d6b14bbDaniel Dunbar /// lifetime is expected to extend past that of the returned ASTUnit. 56031b87d8006d4863dd9b17e515ac720941efc38e3Daniel Dunbar /// 5613c7f4134603d04b44f997b43c0a9def270f25386Sebastian Redl /// \returns - The initialized ASTUnit or null if the AST failed to load. 5623c7f4134603d04b44f997b43c0a9def270f25386Sebastian Redl static ASTUnit *LoadFromASTFile(const std::string &Filename, 56328019772db70d4547be05a042eb950bc910f134fDouglas Gregor llvm::IntrusiveRefCntPtr<Diagnostic> Diags, 564389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis const FileSystemOptions &FileSystemOpts, 5655cf48766d626ff6b223acc9d4b7e415ca8480836Ted Kremenek bool OnlyLocalDecls = false, 5664db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor RemappedFile *RemappedFiles = 0, 567a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor unsigned NumRemappedFiles = 0, 568a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor bool CaptureDiagnostics = false); 569521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar 5704cd912aa94656697a44c3ebb159f05060300524eDouglas Gregorprivate: 5714cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor /// \brief Helper function for \c LoadFromCompilerInvocation() and 5724cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor /// \c LoadFromCommandLine(), which loads an AST from a compiler invocation. 5734cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor /// 5744cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor /// \param PrecompilePreamble Whether to precompile the preamble of this 5754cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor /// translation unit, to improve the performance of reparsing. 5764cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor /// 5774cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor /// \returns \c true if a catastrophic failure occurred (which means that the 5784cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor /// \c ASTUnit itself is invalid), or \c false otherwise. 5794cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor bool LoadFromCompilerInvocation(bool PrecompilePreamble); 5804cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor 5814cd912aa94656697a44c3ebb159f05060300524eDouglas Gregorpublic: 5824cd912aa94656697a44c3ebb159f05060300524eDouglas Gregor 583d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis /// \brief Create an ASTUnit from a source file, via a CompilerInvocation 584d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis /// object, by invoking the optionally provided ASTFrontendAction. 585d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis /// 586d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis /// \param CI - The compiler invocation to use; it must have exactly one input 587d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis /// source file. The ASTUnit takes ownership of the CompilerInvocation object. 588d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis /// 589d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis /// \param Diags - The diagnostics engine to use for reporting errors; its 590d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis /// lifetime is expected to extend past that of the returned ASTUnit. 591d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis /// 592d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis /// \param Action - The ASTFrontendAction to invoke. Its ownership is not 593d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis /// transfered. 594d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis static ASTUnit *LoadFromCompilerInvocationAction(CompilerInvocation *CI, 595d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis llvm::IntrusiveRefCntPtr<Diagnostic> Diags, 596d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis ASTFrontendAction *Action = 0); 597d808bd2978bd4ac95a92b309b038452b533fd7a0Argyrios Kyrtzidis 598521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar /// LoadFromCompilerInvocation - Create an ASTUnit from a source file, via a 599521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar /// CompilerInvocation object. 600521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar /// 601521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar /// \param CI - The compiler invocation to use; it must have exactly one input 602f7acc37450d59ef751df73acb91de73850cc6517Daniel Dunbar /// source file. The ASTUnit takes ownership of the CompilerInvocation object. 603521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar /// 6045262fda30b876c8aae95f2eb92e349418d6b14bbDaniel Dunbar /// \param Diags - The diagnostics engine to use for reporting errors; its 6055262fda30b876c8aae95f2eb92e349418d6b14bbDaniel Dunbar /// lifetime is expected to extend past that of the returned ASTUnit. 606521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar // 607521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar // FIXME: Move OnlyLocalDecls, UseBumpAllocator to setters on the ASTUnit, we 608521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar // shouldn't need to specify them at construction time. 609f7acc37450d59ef751df73acb91de73850cc6517Daniel Dunbar static ASTUnit *LoadFromCompilerInvocation(CompilerInvocation *CI, 61028019772db70d4547be05a042eb950bc910f134fDouglas Gregor llvm::IntrusiveRefCntPtr<Diagnostic> Diags, 611a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor bool OnlyLocalDecls = false, 61244c181aec37789f25f6c15543c164416f72e562aDouglas Gregor bool CaptureDiagnostics = false, 613df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor bool PrecompilePreamble = false, 61487c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor bool CompleteTranslationUnit = true, 615dca8ee8b7bc86076916a3a80f553f7a4e98c14afDouglas Gregor bool CacheCodeCompletionResults = false, 616ba7537febdf1bc1cc617e1f1746f2644feba6274Chandler Carruth bool NestedMacroExpansions = true); 617521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar 6187b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar /// LoadFromCommandLine - Create an ASTUnit from a vector of command line 6197b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar /// arguments, which must specify exactly one source file. 6207b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar /// 6217b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar /// \param ArgBegin - The beginning of the argument vector. 6227b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar /// 6237b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar /// \param ArgEnd - The end of the argument vector. 6247b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar /// 6255262fda30b876c8aae95f2eb92e349418d6b14bbDaniel Dunbar /// \param Diags - The diagnostics engine to use for reporting errors; its 6265262fda30b876c8aae95f2eb92e349418d6b14bbDaniel Dunbar /// lifetime is expected to extend past that of the returned ASTUnit. 6277b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar /// 628869824e87940f97b87064db2df2861e82e08a8c6Daniel Dunbar /// \param ResourceFilesPath - The path to the compiler resource files. 6297b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar // 6307b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar // FIXME: Move OnlyLocalDecls, UseBumpAllocator to setters on the ASTUnit, we 6317b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar // shouldn't need to specify them at construction time. 6327b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar static ASTUnit *LoadFromCommandLine(const char **ArgBegin, 6337b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar const char **ArgEnd, 63428019772db70d4547be05a042eb950bc910f134fDouglas Gregor llvm::IntrusiveRefCntPtr<Diagnostic> Diags, 635869824e87940f97b87064db2df2861e82e08a8c6Daniel Dunbar llvm::StringRef ResourceFilesPath, 6367b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar bool OnlyLocalDecls = false, 637e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor bool CaptureDiagnostics = false, 6384db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor RemappedFile *RemappedFiles = 0, 639a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor unsigned NumRemappedFiles = 0, 640299a4a967b02c9f0d0d94ad8560e3ced893f9116Argyrios Kyrtzidis bool RemappedFilesKeepOriginalName = true, 641df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor bool PrecompilePreamble = false, 64287c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor bool CompleteTranslationUnit = true, 64399ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor bool CacheCodeCompletionResults = false, 64499ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor bool CXXPrecompilePreamble = false, 645dca8ee8b7bc86076916a3a80f553f7a4e98c14afDouglas Gregor bool CXXChainedPCH = false, 646ba7537febdf1bc1cc617e1f1746f2644feba6274Chandler Carruth bool NestedMacroExpansions = true); 647abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor 648abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor /// \brief Reparse the source files using the same command-line options that 649abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor /// were originally used to produce this translation unit. 650abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor /// 651abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor /// \returns True if a failure occurred that causes the ASTUnit not to 652abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor /// contain any translation-unit information, false otherwise. 653abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor bool Reparse(RemappedFile *RemappedFiles = 0, 654abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor unsigned NumRemappedFiles = 0); 6551abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor 6561abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor /// \brief Perform code completion at the given file, line, and 6571abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor /// column within this translation unit. 6581abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor /// 659cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor /// \param File The file in which code completion will occur. 660cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor /// 661cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor /// \param Line The line at which code completion will occur. 662cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor /// 663cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor /// \param Column The column at which code completion will occur. 664cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor /// 665cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor /// \param IncludeMacros Whether to include macros in the code-completion 666cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor /// results. 667cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor /// 668cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor /// \param IncludeCodePatterns Whether to include code patterns (such as a 669cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor /// for loop) in the code-completion results. 6701abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor /// 6712283d79155a3e82442fce124ce5fd704ca138801Douglas Gregor /// FIXME: The Diag, LangOpts, SourceMgr, FileMgr, StoredDiagnostics, and 6722283d79155a3e82442fce124ce5fd704ca138801Douglas Gregor /// OwnedBuffers parameters are all disgusting hacks. They will go away. 6731abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor void CodeComplete(llvm::StringRef File, unsigned Line, unsigned Column, 6741abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor RemappedFile *RemappedFiles, unsigned NumRemappedFiles, 675cee235cdf0b8047761ffac598c4c3a32ab7411a2Douglas Gregor bool IncludeMacros, bool IncludeCodePatterns, 6761abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor CodeCompleteConsumer &Consumer, 6771abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor Diagnostic &Diag, LangOptions &LangOpts, 6781abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor SourceManager &SourceMgr, FileManager &FileMgr, 6792283d79155a3e82442fce124ce5fd704ca138801Douglas Gregor llvm::SmallVectorImpl<StoredDiagnostic> &StoredDiagnostics, 6802283d79155a3e82442fce124ce5fd704ca138801Douglas Gregor llvm::SmallVectorImpl<const llvm::MemoryBuffer *> &OwnedBuffers); 6817ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor 6827ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor /// \brief Save this translation unit to a file with the given name. 6837ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor /// 68439c411fa229b2a6747b92f945d1702ee674d3470Douglas Gregor /// \returns An indication of whether the save was successful or not. 68539c411fa229b2a6747b92f945d1702ee674d3470Douglas Gregor CXSaveError Save(llvm::StringRef File); 686b0f4b9a558933b307073f7cd7753602f94354ae9Argyrios Kyrtzidis 687b0f4b9a558933b307073f7cd7753602f94354ae9Argyrios Kyrtzidis /// \brief Serialize this translation unit with the given output stream. 688b0f4b9a558933b307073f7cd7753602f94354ae9Argyrios Kyrtzidis /// 689b0f4b9a558933b307073f7cd7753602f94354ae9Argyrios Kyrtzidis /// \returns True if an error occurred, false otherwise. 690b0f4b9a558933b307073f7cd7753602f94354ae9Argyrios Kyrtzidis bool serialize(llvm::raw_ostream &OS); 6910853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis}; 6920853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis 6930853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis} // namespace clang 6940853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis 6950853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis#endif 696