1 //===--- ASTLambda.h - Lambda Helper Functions --------------*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
11 /// This file provides some common utility functions for processing
12 /// Lambda related AST Constructs.
14 //===----------------------------------------------------------------------===//
16 #ifndef LLVM_CLANG_AST_ASTLAMBDA_H
17 #define LLVM_CLANG_AST_ASTLAMBDA_H
19 #include "clang/AST/DeclCXX.h"
20 #include "clang/AST/DeclTemplate.h"
23 inline StringRef getLambdaStaticInvokerName() {
26 // This function returns true if M is a specialization, a template,
27 // or a non-generic lambda call operator.
28 inline bool isLambdaCallOperator(const CXXMethodDecl *MD) {
29 const CXXRecordDecl *LambdaClass = MD->getParent();
30 if (!LambdaClass || !LambdaClass->isLambda()) return false;
31 return MD->getOverloadedOperator() == OO_Call;
34 inline bool isLambdaCallOperator(const DeclContext *DC) {
35 if (!DC || !isa<CXXMethodDecl>(DC)) return false;
36 return isLambdaCallOperator(cast<CXXMethodDecl>(DC));
39 inline bool isGenericLambdaCallOperatorSpecialization(const CXXMethodDecl *MD) {
40 if (!MD) return false;
41 const CXXRecordDecl *LambdaClass = MD->getParent();
42 if (LambdaClass && LambdaClass->isGenericLambda())
43 return isLambdaCallOperator(MD) &&
44 MD->isFunctionTemplateSpecialization();
48 inline bool isLambdaConversionOperator(CXXConversionDecl *C) {
49 return C ? C->getParent()->isLambda() : false;
52 inline bool isLambdaConversionOperator(Decl *D) {
54 if (CXXConversionDecl *Conv = dyn_cast<CXXConversionDecl>(D))
55 return isLambdaConversionOperator(Conv);
56 if (FunctionTemplateDecl *F = dyn_cast<FunctionTemplateDecl>(D))
57 if (CXXConversionDecl *Conv =
58 dyn_cast_or_null<CXXConversionDecl>(F->getTemplatedDecl()))
59 return isLambdaConversionOperator(Conv);
63 inline bool isGenericLambdaCallOperatorSpecialization(DeclContext *DC) {
64 return isGenericLambdaCallOperatorSpecialization(
65 dyn_cast<CXXMethodDecl>(DC));
69 // This returns the parent DeclContext ensuring that the correct
70 // parent DeclContext is returned for Lambdas
71 inline DeclContext *getLambdaAwareParentOfDeclContext(DeclContext *DC) {
72 if (isLambdaCallOperator(DC))
73 return DC->getParent()->getParent();
75 return DC->getParent();