1//===--- ASTLambda.h - Lambda Helper Functions --------------*- 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/// This file provides some common utility functions for processing
11/// Lambda related AST Constructs.
12///
13//===----------------------------------------------------------------------===//
14
15#ifndef LLVM_CLANG_AST_ASTLAMBDA_H
16#define LLVM_CLANG_AST_ASTLAMBDA_H
17
18#include "clang/AST/DeclCXX.h"
19#include "clang/AST/DeclTemplate.h"
20
21namespace clang {
22inline StringRef getLambdaStaticInvokerName() {
23 return "__invoke";
24}
25// This function returns true if M is a specialization, a template,
26// or a non-generic lambda call operator.
27inline bool isLambdaCallOperator(const CXXMethodDecl *MD) {
28 const CXXRecordDecl *LambdaClass = MD->getParent();
29 if (!LambdaClass || !LambdaClass->isLambda()) return false;
30 return MD->getOverloadedOperator() == OO_Call;
31}
32
33inline bool isLambdaCallOperator(const DeclContext *DC) {
34 if (!DC || !isa<CXXMethodDecl>(Val: DC)) return false;
35 return isLambdaCallOperator(MD: cast<CXXMethodDecl>(Val: DC));
36}
37
38inline bool isLambdaCallWithExplicitObjectParameter(const DeclContext *DC) {
39 return isLambdaCallOperator(DC) &&
40 cast<CXXMethodDecl>(Val: DC)->isExplicitObjectMemberFunction();
41}
42
43inline bool isLambdaCallWithImplicitObjectParameter(const DeclContext *DC) {
44 return isLambdaCallOperator(DC) &&
45 // FIXME: Checking for a null type is not great
46 // but lambdas with invalid captures or whose closure parameter list
47 // have not fully been parsed may have a call operator whose type is
48 // null.
49 !cast<CXXMethodDecl>(Val: DC)->getType().isNull() &&
50 !cast<CXXMethodDecl>(Val: DC)->isExplicitObjectMemberFunction();
51}
52
53inline bool isGenericLambdaCallOperatorSpecialization(const CXXMethodDecl *MD) {
54 if (!MD) return false;
55 const CXXRecordDecl *LambdaClass = MD->getParent();
56 if (LambdaClass && LambdaClass->isGenericLambda())
57 return isLambdaCallOperator(MD) &&
58 MD->isFunctionTemplateSpecialization();
59 return false;
60}
61
62inline bool isLambdaConversionOperator(CXXConversionDecl *C) {
63 return C ? C->getParent()->isLambda() : false;
64}
65
66inline bool isLambdaConversionOperator(Decl *D) {
67 if (!D) return false;
68 if (CXXConversionDecl *Conv = dyn_cast<CXXConversionDecl>(Val: D))
69 return isLambdaConversionOperator(C: Conv);
70 if (FunctionTemplateDecl *F = dyn_cast<FunctionTemplateDecl>(Val: D))
71 if (CXXConversionDecl *Conv =
72 dyn_cast_or_null<CXXConversionDecl>(Val: F->getTemplatedDecl()))
73 return isLambdaConversionOperator(C: Conv);
74 return false;
75}
76
77inline bool isGenericLambdaCallOperatorSpecialization(DeclContext *DC) {
78 return isGenericLambdaCallOperatorSpecialization(
79 MD: dyn_cast<CXXMethodDecl>(Val: DC));
80}
81
82inline bool isGenericLambdaCallOperatorOrStaticInvokerSpecialization(
83 const DeclContext *DC) {
84 const auto *MD = dyn_cast<CXXMethodDecl>(Val: DC);
85 if (!MD) return false;
86 const CXXRecordDecl *LambdaClass = MD->getParent();
87 if (LambdaClass && LambdaClass->isGenericLambda())
88 return (isLambdaCallOperator(MD) || MD->isLambdaStaticInvoker()) &&
89 MD->isFunctionTemplateSpecialization();
90 return false;
91}
92
93// This returns the parent DeclContext ensuring that the correct
94// parent DeclContext is returned for Lambdas
95inline DeclContext *getLambdaAwareParentOfDeclContext(DeclContext *DC) {
96 if (isLambdaCallOperator(DC))
97 return DC->getParent()->getParent();
98 else
99 return DC->getParent();
100}
101
102} // clang
103
104#endif
105

source code of clang/include/clang/AST/ASTLambda.h