aboutsummaryrefslogtreecommitdiff
path: root/include/clang/Index/Indexer.h
blob: a7c29581a02b41c7e9a8234d0d44315c26ca25c3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
//===--- Indexer.h - IndexProvider implementation ---------------*- C++ -*-===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
//  IndexProvider implementation.
//
//===----------------------------------------------------------------------===//

#ifndef LLVM_CLANG_INDEX_INDEXER_H
#define LLVM_CLANG_INDEX_INDEXER_H

#include "clang/Index/IndexProvider.h"
#include "llvm/ADT/SmallPtrSet.h"
#include <map>

namespace clang {

namespace idx {
  class Program;
  class TranslationUnit;

/// \brief Maps information to TranslationUnits.
class Indexer : public IndexProvider {
public:
  typedef llvm::SmallPtrSet<TranslationUnit *, 4> TUSetTy;
  typedef std::map<Entity, TUSetTy> MapTy;

  explicit Indexer(Program &prog) : Prog(prog) { }

  Program &getProgram() const { return Prog; }

  /// \brief Find all Entities and map them to the given translation unit.
  void IndexAST(TranslationUnit *TU);

  virtual void GetTranslationUnitsFor(Entity Ent,
                                      TranslationUnitHandler *Handler);

  typedef TUSetTy::iterator translation_unit_iterator;

  translation_unit_iterator translation_units_begin(Entity Ent) const;
  translation_unit_iterator translation_units_end(Entity Ent) const;
  bool translation_units_empty(Entity Ent) const;

private:
  Program &Prog;
  MapTy Map;
  CtxTUMapTy CtxTUMap;
};

} // namespace idx

} // namespace clang

#endif