blob: 6bc4f2e776a35bab51081345c54cf17e9faaf3b6 [file] [log] [blame]
Ted Kremenek16b55a72010-01-26 19:31:51 +00001//===- 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"
Ted Kremenek0a90d322010-11-17 23:24:11 +000016#include "CXTranslationUnit.h"
Ted Kremenek16b55a72010-01-26 19:31:51 +000017#include "CXSourceLocation.h"
18#include "clang/AST/DeclVisitor.h"
Benjamin Kramerb846deb2010-04-12 19:45:50 +000019#include "clang/Frontend/ASTUnit.h"
Ted Kremenek16b55a72010-01-26 19:31:51 +000020#include "llvm/ADT/SmallString.h"
21#include "llvm/Support/raw_ostream.h"
Benjamin Kramerb846deb2010-04-12 19:45:50 +000022using namespace clang;
Ted Kremenek16b55a72010-01-26 19:31:51 +000023
24extern "C" {
25void clang_getInclusions(CXTranslationUnit TU, CXInclusionVisitor CB,
26 CXClientData clientData) {
27
Ted Kremeneka60ed472010-11-16 08:15:36 +000028 ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
Ted Kremenek16b55a72010-01-26 19:31:51 +000029 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) {
Douglas Gregore23ac652011-04-20 00:21:03 +000043 bool Invalid = false;
44 const SrcMgr::SLocEntry &SL = SM.getSLocEntry(i, &Invalid);
Ted Kremenek16b55a72010-01-26 19:31:51 +000045
Douglas Gregore23ac652011-04-20 00:21:03 +000046 if (!SL.isFile() || Invalid)
Ted Kremenek16b55a72010-01-26 19:31:51 +000047 continue;
48
49 const SrcMgr::FileInfo &FI = SL.getFile();
Argyrios Kyrtzidisb1c86492011-03-05 01:03:53 +000050 if (!FI.getContentCache()->OrigEntry)
Ted Kremenek16b55a72010-01-26 19:31:51 +000051 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));
Douglas Gregorcb7b1e12010-11-12 07:15:47 +000059 L = PLoc.isValid()? PLoc.getIncludeLoc() : SourceLocation();
Ted Kremenek16b55a72010-01-26 19:31:51 +000060 }
61
62 // Callback to the client.
63 // FIXME: We should have a function to construct CXFiles.
Argyrios Kyrtzidisb1c86492011-03-05 01:03:53 +000064 CB((CXFile) FI.getContentCache()->OrigEntry,
Ted Kremenek16b55a72010-01-26 19:31:51 +000065 InclusionStack.data(), InclusionStack.size(), clientData);
66 }
67}
68} // end extern C