ASTUnit.h revision 4db64a461cb3442934afe43c83ed3f17f7c11c1d
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 1731b87d8006d4863dd9b17e515ac720941efc38e3Daniel Dunbar#include "clang/Basic/SourceManager.h" 180853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis#include "llvm/ADT/OwningPtr.h" 1936c4464ba6cfc2a63dc67c493ef2f5ab2aea09ccSteve Naroff#include "clang/Basic/FileManager.h" 20f96b524306ccfa623235d375deee79637bd38f29Steve Naroff#include "clang/Index/ASTLocation.h" 210853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis#include <string> 22f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar#include <vector> 23f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar#include <cassert> 244db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor#include <utility> 254db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 264db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregornamespace llvm { 274db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor class MemoryBuffer; 284db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor} 290853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis 300853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidisnamespace clang { 31521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass ASTContext; 32521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass CompilerInvocation; 33521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass Decl; 34521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass Diagnostic; 35521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass FileEntry; 36521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass FileManager; 37521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass HeaderSearch; 38521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass Preprocessor; 39521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass SourceManager; 40521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbarclass TargetInfo; 410853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis 42f96b524306ccfa623235d375deee79637bd38f29Steve Naroffusing namespace idx; 43f96b524306ccfa623235d375deee79637bd38f29Steve Naroff 440853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis/// \brief Utility class for loading a ASTContext from a PCH file. 450853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis/// 460853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidisclass ASTUnit { 4736c4464ba6cfc2a63dc67c493ef2f5ab2aea09ccSteve Naroff FileManager FileMgr; 4836c4464ba6cfc2a63dc67c493ef2f5ab2aea09ccSteve Naroff 4931b87d8006d4863dd9b17e515ac720941efc38e3Daniel Dunbar SourceManager SourceMgr; 500853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis llvm::OwningPtr<HeaderSearch> HeaderInfo; 510853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis llvm::OwningPtr<TargetInfo> Target; 520853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis llvm::OwningPtr<Preprocessor> PP; 530853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis llvm::OwningPtr<ASTContext> Ctx; 54e19944c93961b7618f4f3f3185f698f46369ea54Steve Naroff bool tempFile; 55f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar 567d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor // OnlyLocalDecls - when true, walking this AST should only visit declarations 577d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor // that come from the AST itself, not from included precompiled headers. 587d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor // FIXME: This is temporary; eventually, CIndex will always do this. 597d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor bool OnlyLocalDecls; 60c7822dbf3c01a2a5f837cff82ba7889ea755dacaDaniel Dunbar 61f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar /// Track whether the main file was loaded from an AST or not. 62c7822dbf3c01a2a5f837cff82ba7889ea755dacaDaniel Dunbar bool MainFileIsAST; 63c7822dbf3c01a2a5f837cff82ba7889ea755dacaDaniel Dunbar 64f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar /// Track the top-level decls which appeared in an ASTUnit which was loaded 65f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar /// from a source file. 66f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar // 67f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar // FIXME: This is just an optimization hack to avoid deserializing large parts 68f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar // of a PCH file when using the Index library on an ASTUnit loaded from 69f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar // source. In the long term we should make the Index library use efficient and 70f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar // more scalable search mechanisms. 71f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar std::vector<Decl*> TopLevelDecls; 72f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar 7368d40e2d16b9fadba386853d6bbb60089291fdc5Daniel Dunbar /// The name of the original source file used to generate this ASTUnit. 7468d40e2d16b9fadba386853d6bbb60089291fdc5Daniel Dunbar std::string OriginalSourceFile; 7568d40e2d16b9fadba386853d6bbb60089291fdc5Daniel Dunbar 76f96b524306ccfa623235d375deee79637bd38f29Steve Naroff // Critical optimization when using clang_getCursor(). 77f96b524306ccfa623235d375deee79637bd38f29Steve Naroff ASTLocation LastLoc; 78f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar 7931b87d8006d4863dd9b17e515ac720941efc38e3Daniel Dunbar ASTUnit(const ASTUnit&); // DO NOT IMPLEMENT 8031b87d8006d4863dd9b17e515ac720941efc38e3Daniel Dunbar ASTUnit &operator=(const ASTUnit &); // DO NOT IMPLEMENT 811eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 820853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidispublic: 835262fda30b876c8aae95f2eb92e349418d6b14bbDaniel Dunbar ASTUnit(bool MainFileIsAST); 840853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis ~ASTUnit(); 850853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis 86c7822dbf3c01a2a5f837cff82ba7889ea755dacaDaniel Dunbar bool isMainFileAST() const { return MainFileIsAST; } 87c7822dbf3c01a2a5f837cff82ba7889ea755dacaDaniel Dunbar 8831b87d8006d4863dd9b17e515ac720941efc38e3Daniel Dunbar const SourceManager &getSourceManager() const { return SourceMgr; } 8931b87d8006d4863dd9b17e515ac720941efc38e3Daniel Dunbar SourceManager &getSourceManager() { return SourceMgr; } 900853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis 910853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis const Preprocessor &getPreprocessor() const { return *PP.get(); } 920853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis Preprocessor &getPreprocessor() { return *PP.get(); } 931eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 940853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis const ASTContext &getASTContext() const { return *Ctx.get(); } 950853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis ASTContext &getASTContext() { return *Ctx.get(); } 960853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis 9736c4464ba6cfc2a63dc67c493ef2f5ab2aea09ccSteve Naroff const FileManager &getFileManager() const { return FileMgr; } 9836c4464ba6cfc2a63dc67c493ef2f5ab2aea09ccSteve Naroff FileManager &getFileManager() { return FileMgr; } 99f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar 10077accc11f04ed4ff9afd4e27d430144d4714be56Steve Naroff const std::string &getOriginalSourceFileName(); 101e19944c93961b7618f4f3f3185f698f46369ea54Steve Naroff const std::string &getPCHFileName(); 102b85bca2676b433ae555db09de4dd2823ff13b856Zhongxing Xu 103e19944c93961b7618f4f3f3185f698f46369ea54Steve Naroff void unlinkTemporaryFile() { tempFile = true; } 104f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar 1057d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor bool getOnlyLocalDecls() const { return OnlyLocalDecls; } 106f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar 107f96b524306ccfa623235d375deee79637bd38f29Steve Naroff void setLastASTLocation(ASTLocation ALoc) { LastLoc = ALoc; } 108f96b524306ccfa623235d375deee79637bd38f29Steve Naroff ASTLocation getLastASTLocation() const { return LastLoc; } 109f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar 110f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar std::vector<Decl*> &getTopLevelDecls() { 111f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar assert(!isMainFileAST() && "Invalid call for AST based ASTUnit!"); 112f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar return TopLevelDecls; 113f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar } 114f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar const std::vector<Decl*> &getTopLevelDecls() const { 115f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar assert(!isMainFileAST() && "Invalid call for AST based ASTUnit!"); 116f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar return TopLevelDecls; 117f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar } 118f772d1e2a5688572d07f42896a50ac57a4a41fe8Daniel Dunbar 1194db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor /// \brief A mapping from a file name to the memory buffer that stores the 1204db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor /// remapped contents of that file. 1214db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor typedef std::pair<std::string, const llvm::MemoryBuffer *> RemappedFile; 1224db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 1230853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis /// \brief Create a ASTUnit from a PCH file. 1240853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis /// 12531b87d8006d4863dd9b17e515ac720941efc38e3Daniel Dunbar /// \param Filename - The PCH file to load. 1260853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis /// 1275262fda30b876c8aae95f2eb92e349418d6b14bbDaniel Dunbar /// \param Diags - The diagnostics engine to use for reporting errors; its 1285262fda30b876c8aae95f2eb92e349418d6b14bbDaniel Dunbar /// lifetime is expected to extend past that of the returned ASTUnit. 12931b87d8006d4863dd9b17e515ac720941efc38e3Daniel Dunbar /// 13031b87d8006d4863dd9b17e515ac720941efc38e3Daniel Dunbar /// \returns - The initialized ASTUnit or null if the PCH failed to load. 1310853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis static ASTUnit *LoadFromPCHFile(const std::string &Filename, 1325262fda30b876c8aae95f2eb92e349418d6b14bbDaniel Dunbar Diagnostic &Diags, 1335cf48766d626ff6b223acc9d4b7e415ca8480836Ted Kremenek bool OnlyLocalDecls = false, 1344db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor bool UseBumpAllocator = false, 1354db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor RemappedFile *RemappedFiles = 0, 1364db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor unsigned NumRemappedFiles = 0); 137521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar 138521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar /// LoadFromCompilerInvocation - Create an ASTUnit from a source file, via a 139521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar /// CompilerInvocation object. 140521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar /// 141521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar /// \param CI - The compiler invocation to use; it must have exactly one input 142521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar /// source file. 143521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar /// 1445262fda30b876c8aae95f2eb92e349418d6b14bbDaniel Dunbar /// \param Diags - The diagnostics engine to use for reporting errors; its 1455262fda30b876c8aae95f2eb92e349418d6b14bbDaniel Dunbar /// lifetime is expected to extend past that of the returned ASTUnit. 146521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar // 147521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar // FIXME: Move OnlyLocalDecls, UseBumpAllocator to setters on the ASTUnit, we 148521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar // shouldn't need to specify them at construction time. 149521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar static ASTUnit *LoadFromCompilerInvocation(const CompilerInvocation &CI, 150521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar Diagnostic &Diags, 15168ea2ac7fd98a5a44c7a5b04c54076cf794531cbDaniel Dunbar bool OnlyLocalDecls = false); 152521bf9c529e653ab28896d027352d3e16e2672d5Daniel Dunbar 1537b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar /// LoadFromCommandLine - Create an ASTUnit from a vector of command line 1547b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar /// arguments, which must specify exactly one source file. 1557b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar /// 1567b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar /// \param ArgBegin - The beginning of the argument vector. 1577b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar /// 1587b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar /// \param ArgEnd - The end of the argument vector. 1597b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar /// 1605262fda30b876c8aae95f2eb92e349418d6b14bbDaniel Dunbar /// \param Diags - The diagnostics engine to use for reporting errors; its 1615262fda30b876c8aae95f2eb92e349418d6b14bbDaniel Dunbar /// lifetime is expected to extend past that of the returned ASTUnit. 1627b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar /// 163869824e87940f97b87064db2df2861e82e08a8c6Daniel Dunbar /// \param ResourceFilesPath - The path to the compiler resource files. 1647b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar // 1657b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar // FIXME: Move OnlyLocalDecls, UseBumpAllocator to setters on the ASTUnit, we 1667b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar // shouldn't need to specify them at construction time. 1677b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar static ASTUnit *LoadFromCommandLine(const char **ArgBegin, 1687b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar const char **ArgEnd, 1697b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar Diagnostic &Diags, 170869824e87940f97b87064db2df2861e82e08a8c6Daniel Dunbar llvm::StringRef ResourceFilesPath, 1717b55668db7618334cc40011d3c1e128524d89462Daniel Dunbar bool OnlyLocalDecls = false, 1724db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor bool UseBumpAllocator = false, 1734db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor RemappedFile *RemappedFiles = 0, 1744db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor unsigned NumRemappedFiles = 0); 1750853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis}; 1760853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis 1770853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis} // namespace clang 1780853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis 1790853a02c3b04d96a3c432b883e403175c954cd81Argyrios Kyrtzidis#endif 180