CIndexInclusionStack.cpp revision b1c86492f9a9bef01a4567408c22f961bbd604fe
1//===- CIndexInclusionStack.cpp - Clang-C Source Indexing Library ---------===// 2// 3// The LLVM Compiler Infrastructure 4// 5// This file is distributed under the University of Illinois Open Source 6// License. See LICENSE.TXT for details. 7// 8//===----------------------------------------------------------------------===// 9// 10// This file defines a callback mechanism for clients to get the inclusion 11// stack from a translation unit. 12// 13//===----------------------------------------------------------------------===// 14 15#include "CIndexer.h" 16#include "CXTranslationUnit.h" 17#include "CXSourceLocation.h" 18#include "clang/AST/DeclVisitor.h" 19#include "clang/Frontend/ASTUnit.h" 20#include "llvm/ADT/SmallString.h" 21#include "llvm/Support/raw_ostream.h" 22using namespace clang; 23 24extern "C" { 25void clang_getInclusions(CXTranslationUnit TU, CXInclusionVisitor CB, 26 CXClientData clientData) { 27 28 ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData); 29 SourceManager &SM = CXXUnit->getSourceManager(); 30 ASTContext &Ctx = CXXUnit->getASTContext(); 31 32 llvm::SmallVector<CXSourceLocation, 10> InclusionStack; 33 unsigned i = SM.sloc_loaded_entry_size(); 34 unsigned n = SM.sloc_entry_size(); 35 36 // In the case where all the SLocEntries are in an external source, traverse 37 // those SLocEntries as well. This is the case where we are looking 38 // at the inclusion stack of an AST/PCH file. 39 if (i >= n) 40 i = 0; 41 42 for ( ; i < n ; ++i) { 43 44 const SrcMgr::SLocEntry &SL = SM.getSLocEntry(i); 45 46 if (!SL.isFile()) 47 continue; 48 49 const SrcMgr::FileInfo &FI = SL.getFile(); 50 if (!FI.getContentCache()->OrigEntry) 51 continue; 52 53 // Build the inclusion stack. 54 SourceLocation L = FI.getIncludeLoc(); 55 InclusionStack.clear(); 56 while (L.isValid()) { 57 PresumedLoc PLoc = SM.getPresumedLoc(L); 58 InclusionStack.push_back(cxloc::translateSourceLocation(Ctx, L)); 59 L = PLoc.isValid()? PLoc.getIncludeLoc() : SourceLocation(); 60 } 61 62 // Callback to the client. 63 // FIXME: We should have a function to construct CXFiles. 64 CB((CXFile) FI.getContentCache()->OrigEntry, 65 InclusionStack.data(), InclusionStack.size(), clientData); 66 } 67} 68} // end extern C 69