CIndexInclusionStack.cpp revision cb7b1e17b63967317ab5cc55682168cf0380519a 
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 "CXSourceLocation.h"
17#include "clang/AST/DeclVisitor.h"
18#include "clang/Frontend/ASTUnit.h"
19#include "llvm/ADT/SmallString.h"
20#include "llvm/Support/raw_ostream.h"
21using namespace clang;
22
23extern "C" {
24void clang_getInclusions(CXTranslationUnit TU, CXInclusionVisitor CB,
25                         CXClientData clientData) {
26
27  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU);
28  SourceManager &SM = CXXUnit->getSourceManager();
29  ASTContext &Ctx = CXXUnit->getASTContext();
30
31  llvm::SmallVector<CXSourceLocation, 10> InclusionStack;
32  unsigned i = SM.sloc_loaded_entry_size();
33  unsigned n =  SM.sloc_entry_size();
34
35  // In the case where all the SLocEntries are in an external source, traverse
36  // those SLocEntries as well.  This is the case where we are looking
37  // at the inclusion stack of an AST/PCH file.
38  if (i >= n)
39    i = 0;
40
41  for ( ; i < n ; ++i) {
42
43    const SrcMgr::SLocEntry &SL = SM.getSLocEntry(i);
44
45    if (!SL.isFile())
46      continue;
47
48    const SrcMgr::FileInfo &FI = SL.getFile();
49    if (!FI.getContentCache()->Entry)
50      continue;
51
52    // Build the inclusion stack.
53    SourceLocation L = FI.getIncludeLoc();
54    InclusionStack.clear();
55    while (L.isValid()) {
56      PresumedLoc PLoc = SM.getPresumedLoc(L);
57      InclusionStack.push_back(cxloc::translateSourceLocation(Ctx, L));
58      L = PLoc.isValid()? PLoc.getIncludeLoc() : SourceLocation();
59    }
60
61    // Callback to the client.
62    // FIXME: We should have a function to construct CXFiles.
63    CB((CXFile) FI.getContentCache()->Entry,
64       InclusionStack.data(), InclusionStack.size(), clientData);
65  }
66}
67} // end extern C
68