Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
14 | pmbaty | 1 | //===- CommentVisitor.h - Visitor for Comment subclasses --------*- 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 | #ifndef LLVM_CLANG_AST_COMMENTVISITOR_H |
||
10 | #define LLVM_CLANG_AST_COMMENTVISITOR_H |
||
11 | |||
12 | #include "clang/AST/Comment.h" |
||
13 | #include "llvm/ADT/STLExtras.h" |
||
14 | #include "llvm/Support/ErrorHandling.h" |
||
15 | |||
16 | namespace clang { |
||
17 | namespace comments { |
||
18 | template <template <typename> class Ptr, typename ImplClass, |
||
19 | typename RetTy = void, class... ParamTys> |
||
20 | class CommentVisitorBase { |
||
21 | public: |
||
22 | #define PTR(CLASS) typename Ptr<CLASS>::type |
||
23 | #define DISPATCH(NAME, CLASS) \ |
||
24 | return static_cast<ImplClass *>(this)->visit##NAME( \ |
||
25 | static_cast<PTR(CLASS)>(C), std::forward<ParamTys>(P)...) |
||
26 | |||
27 | RetTy visit(PTR(Comment) C, ParamTys... P) { |
||
28 | if (!C) |
||
29 | return RetTy(); |
||
30 | |||
31 | switch (C->getCommentKind()) { |
||
32 | default: llvm_unreachable("Unknown comment kind!"); |
||
33 | #define ABSTRACT_COMMENT(COMMENT) |
||
34 | #define COMMENT(CLASS, PARENT) \ |
||
35 | case Comment::CLASS##Kind: DISPATCH(CLASS, CLASS); |
||
36 | #include "clang/AST/CommentNodes.inc" |
||
37 | #undef ABSTRACT_COMMENT |
||
38 | #undef COMMENT |
||
39 | } |
||
40 | } |
||
41 | |||
42 | // If the derived class does not implement a certain Visit* method, fall back |
||
43 | // on Visit* method for the superclass. |
||
44 | #define ABSTRACT_COMMENT(COMMENT) COMMENT |
||
45 | #define COMMENT(CLASS, PARENT) \ |
||
46 | RetTy visit##CLASS(PTR(CLASS) C, ParamTys... P) { DISPATCH(PARENT, PARENT); } |
||
47 | #include "clang/AST/CommentNodes.inc" |
||
48 | #undef ABSTRACT_COMMENT |
||
49 | #undef COMMENT |
||
50 | |||
51 | RetTy visitComment(PTR(Comment) C, ParamTys... P) { return RetTy(); } |
||
52 | |||
53 | #undef PTR |
||
54 | #undef DISPATCH |
||
55 | }; |
||
56 | |||
57 | template <typename ImplClass, typename RetTy = void, class... ParamTys> |
||
58 | class CommentVisitor : public CommentVisitorBase<std::add_pointer, ImplClass, |
||
59 | RetTy, ParamTys...> {}; |
||
60 | |||
61 | template <typename ImplClass, typename RetTy = void, class... ParamTys> |
||
62 | class ConstCommentVisitor |
||
63 | : public CommentVisitorBase<llvm::make_const_ptr, ImplClass, RetTy, |
||
64 | ParamTys...> {}; |
||
65 | |||
66 | } // namespace comments |
||
67 | } // namespace clang |
||
68 | |||
69 | #endif // LLVM_CLANG_AST_COMMENTVISITOR_H |