tools/libclang/CIndexInclusionStack.cpp

   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"
  22 using namespace clang;
  23
  24 extern "C" {
  25 void 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()->Entry)
  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()->Entry,
  65        InclusionStack.data(), InclusionStack.size(), clientData);
  66   }
  67 }
  68 } // end extern C