llvm-project

1
//===-- ClangDoc.cpp - ClangDoc ---------------------------------*- C++ -*-===//
2
//
3
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4
// See https://llvm.org/LICENSE.txt for license information.
5
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6
//
7
//===----------------------------------------------------------------------===//
8
//
9
// This file implements the main entry point for the clang-doc tool. It runs
10
// the clang-doc mapper on a given set of source code files using a
11
// FrontendActionFactory.
12
//
13
//===----------------------------------------------------------------------===//
14

15
#include "ClangDoc.h"
16
#include "Mapper.h"
17
#include "Representation.h"
18
#include "clang/AST/AST.h"
19
#include "clang/AST/ASTConsumer.h"
20
#include "clang/AST/ASTContext.h"
21
#include "clang/AST/RecursiveASTVisitor.h"
22
#include "clang/Frontend/ASTConsumers.h"
23
#include "clang/Frontend/CompilerInstance.h"
24
#include "clang/Frontend/FrontendActions.h"
25

26
namespace clang {
27
namespace doc {
28

29
class MapperActionFactory : public tooling::FrontendActionFactory {
30
public:
31
  MapperActionFactory(ClangDocContext CDCtx) : CDCtx(CDCtx) {}
32
  std::unique_ptr<FrontendAction> create() override;
33

34
private:
35
  ClangDocContext CDCtx;
36
};
37

38
std::unique_ptr<FrontendAction> MapperActionFactory::create() {
39
  class ClangDocAction : public clang::ASTFrontendAction {
40
  public:
41
    ClangDocAction(ClangDocContext CDCtx) : CDCtx(CDCtx) {}
42

43
    std::unique_ptr<clang::ASTConsumer>
44
    CreateASTConsumer(clang::CompilerInstance &Compiler,
45
                      llvm::StringRef InFile) override {
46
      return std::make_unique<MapASTVisitor>(&Compiler.getASTContext(), CDCtx);
47
    }
48

49
  private:
50
    ClangDocContext CDCtx;
51
  };
52
  return std::make_unique<ClangDocAction>(CDCtx);
53
}
54

55
std::unique_ptr<tooling::FrontendActionFactory>
56
newMapperActionFactory(ClangDocContext CDCtx) {
57
  return std::make_unique<MapperActionFactory>(CDCtx);
58
}
59

60
} // namespace doc
61
} // namespace clang
62