Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
14 | pmbaty | 1 | //===--- LambdaCapture.h - Types for C++ Lambda Captures --------*- 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 | /// \file |
||
10 | /// Defines the LambdaCapture class. |
||
11 | /// |
||
12 | //===----------------------------------------------------------------------===// |
||
13 | |||
14 | #ifndef LLVM_CLANG_AST_LAMBDACAPTURE_H |
||
15 | #define LLVM_CLANG_AST_LAMBDACAPTURE_H |
||
16 | |||
17 | #include "clang/AST/Decl.h" |
||
18 | #include "clang/Basic/Lambda.h" |
||
19 | #include "llvm/ADT/PointerIntPair.h" |
||
20 | |||
21 | namespace clang { |
||
22 | |||
23 | /// Describes the capture of a variable or of \c this, or of a |
||
24 | /// C++1y init-capture. |
||
25 | class LambdaCapture { |
||
26 | enum { |
||
27 | /// Flag used by the Capture class to indicate that the given |
||
28 | /// capture was implicit. |
||
29 | Capture_Implicit = 0x01, |
||
30 | |||
31 | /// Flag used by the Capture class to indicate that the |
||
32 | /// given capture was by-copy. |
||
33 | /// |
||
34 | /// This includes the case of a non-reference init-capture. |
||
35 | Capture_ByCopy = 0x02, |
||
36 | |||
37 | /// Flag used by the Capture class to distinguish between a capture |
||
38 | /// of '*this' and a capture of a VLA type. |
||
39 | Capture_This = 0x04 |
||
40 | }; |
||
41 | |||
42 | // Decl could represent: |
||
43 | // - a VarDecl* that represents the variable that was captured or the |
||
44 | // init-capture. |
||
45 | // - or, is a nullptr and Capture_This is set in Bits if this represents a |
||
46 | // capture of '*this' by value or reference. |
||
47 | // - or, is a nullptr and Capture_This is not set in Bits if this represents |
||
48 | // a capture of a VLA type. |
||
49 | llvm::PointerIntPair<Decl*, 3> DeclAndBits; |
||
50 | |||
51 | SourceLocation Loc; |
||
52 | SourceLocation EllipsisLoc; |
||
53 | |||
54 | friend class ASTStmtReader; |
||
55 | friend class ASTStmtWriter; |
||
56 | |||
57 | public: |
||
58 | /// Create a new capture of a variable or of \c this. |
||
59 | /// |
||
60 | /// \param Loc The source location associated with this capture. |
||
61 | /// |
||
62 | /// \param Kind The kind of capture (this, byref, bycopy), which must |
||
63 | /// not be init-capture. |
||
64 | /// |
||
65 | /// \param Implicit Whether the capture was implicit or explicit. |
||
66 | /// |
||
67 | /// \param Var The local variable being captured, or null if capturing |
||
68 | /// \c this. |
||
69 | /// |
||
70 | /// \param EllipsisLoc The location of the ellipsis (...) for a |
||
71 | /// capture that is a pack expansion, or an invalid source |
||
72 | /// location to indicate that this is not a pack expansion. |
||
73 | LambdaCapture(SourceLocation Loc, bool Implicit, LambdaCaptureKind Kind, |
||
74 | ValueDecl *Var = nullptr, |
||
75 | SourceLocation EllipsisLoc = SourceLocation()); |
||
76 | |||
77 | /// Determine the kind of capture. |
||
78 | LambdaCaptureKind getCaptureKind() const; |
||
79 | |||
80 | /// Determine whether this capture handles the C++ \c this |
||
81 | /// pointer. |
||
82 | bool capturesThis() const { |
||
83 | return DeclAndBits.getPointer() == nullptr && |
||
84 | (DeclAndBits.getInt() & Capture_This); |
||
85 | } |
||
86 | |||
87 | /// Determine whether this capture handles a variable. |
||
88 | bool capturesVariable() const { |
||
89 | return isa_and_nonnull<ValueDecl>(DeclAndBits.getPointer()); |
||
90 | } |
||
91 | |||
92 | /// Determine whether this captures a variable length array bound |
||
93 | /// expression. |
||
94 | bool capturesVLAType() const { |
||
95 | return DeclAndBits.getPointer() == nullptr && |
||
96 | !(DeclAndBits.getInt() & Capture_This); |
||
97 | } |
||
98 | |||
99 | /// Retrieve the declaration of the local variable being |
||
100 | /// captured. |
||
101 | /// |
||
102 | /// This operation is only valid if this capture is a variable capture |
||
103 | /// (other than a capture of \c this). |
||
104 | ValueDecl *getCapturedVar() const { |
||
105 | assert(capturesVariable() && "No variable available for capture"); |
||
106 | return static_cast<ValueDecl *>(DeclAndBits.getPointer()); |
||
107 | } |
||
108 | |||
109 | /// Determine whether this was an implicit capture (not |
||
110 | /// written between the square brackets introducing the lambda). |
||
111 | bool isImplicit() const { |
||
112 | return DeclAndBits.getInt() & Capture_Implicit; |
||
113 | } |
||
114 | |||
115 | /// Determine whether this was an explicit capture (written |
||
116 | /// between the square brackets introducing the lambda). |
||
117 | bool isExplicit() const { return !isImplicit(); } |
||
118 | |||
119 | /// Retrieve the source location of the capture. |
||
120 | /// |
||
121 | /// For an explicit capture, this returns the location of the |
||
122 | /// explicit capture in the source. For an implicit capture, this |
||
123 | /// returns the location at which the variable or \c this was first |
||
124 | /// used. |
||
125 | SourceLocation getLocation() const { return Loc; } |
||
126 | |||
127 | /// Determine whether this capture is a pack expansion, |
||
128 | /// which captures a function parameter pack. |
||
129 | bool isPackExpansion() const { return EllipsisLoc.isValid(); } |
||
130 | |||
131 | /// Retrieve the location of the ellipsis for a capture |
||
132 | /// that is a pack expansion. |
||
133 | SourceLocation getEllipsisLoc() const { |
||
134 | assert(isPackExpansion() && "No ellipsis location for a non-expansion"); |
||
135 | return EllipsisLoc; |
||
136 | } |
||
137 | }; |
||
138 | |||
139 | } // end namespace clang |
||
140 | |||
141 | #endif // LLVM_CLANG_AST_LAMBDACAPTURE_H |