1*7330f729Sjoerg //===- DeclFriend.cpp - C++ Friend Declaration AST Node Implementation ----===//
2*7330f729Sjoerg //
3*7330f729Sjoerg // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4*7330f729Sjoerg // See https://llvm.org/LICENSE.txt for license information.
5*7330f729Sjoerg // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6*7330f729Sjoerg //
7*7330f729Sjoerg //===----------------------------------------------------------------------===//
8*7330f729Sjoerg //
9*7330f729Sjoerg // This file implements the AST classes related to C++ friend
10*7330f729Sjoerg // declarations.
11*7330f729Sjoerg //
12*7330f729Sjoerg //===----------------------------------------------------------------------===//
13*7330f729Sjoerg
14*7330f729Sjoerg #include "clang/AST/DeclFriend.h"
15*7330f729Sjoerg #include "clang/AST/Decl.h"
16*7330f729Sjoerg #include "clang/AST/DeclBase.h"
17*7330f729Sjoerg #include "clang/AST/DeclCXX.h"
18*7330f729Sjoerg #include "clang/AST/ASTContext.h"
19*7330f729Sjoerg #include "clang/AST/DeclTemplate.h"
20*7330f729Sjoerg #include "clang/Basic/LLVM.h"
21*7330f729Sjoerg #include "llvm/Support/Casting.h"
22*7330f729Sjoerg #include <cassert>
23*7330f729Sjoerg #include <cstddef>
24*7330f729Sjoerg
25*7330f729Sjoerg using namespace clang;
26*7330f729Sjoerg
anchor()27*7330f729Sjoerg void FriendDecl::anchor() {}
28*7330f729Sjoerg
getNextFriendSlowCase()29*7330f729Sjoerg FriendDecl *FriendDecl::getNextFriendSlowCase() {
30*7330f729Sjoerg return cast_or_null<FriendDecl>(
31*7330f729Sjoerg NextFriend.get(getASTContext().getExternalSource()));
32*7330f729Sjoerg }
33*7330f729Sjoerg
Create(ASTContext & C,DeclContext * DC,SourceLocation L,FriendUnion Friend,SourceLocation FriendL,ArrayRef<TemplateParameterList * > FriendTypeTPLists)34*7330f729Sjoerg FriendDecl *FriendDecl::Create(ASTContext &C, DeclContext *DC,
35*7330f729Sjoerg SourceLocation L,
36*7330f729Sjoerg FriendUnion Friend,
37*7330f729Sjoerg SourceLocation FriendL,
38*7330f729Sjoerg ArrayRef<TemplateParameterList *> FriendTypeTPLists) {
39*7330f729Sjoerg #ifndef NDEBUG
40*7330f729Sjoerg if (Friend.is<NamedDecl *>()) {
41*7330f729Sjoerg const auto *D = Friend.get<NamedDecl*>();
42*7330f729Sjoerg assert(isa<FunctionDecl>(D) ||
43*7330f729Sjoerg isa<CXXRecordDecl>(D) ||
44*7330f729Sjoerg isa<FunctionTemplateDecl>(D) ||
45*7330f729Sjoerg isa<ClassTemplateDecl>(D));
46*7330f729Sjoerg
47*7330f729Sjoerg // As a temporary hack, we permit template instantiation to point
48*7330f729Sjoerg // to the original declaration when instantiating members.
49*7330f729Sjoerg assert(D->getFriendObjectKind() ||
50*7330f729Sjoerg (cast<CXXRecordDecl>(DC)->getTemplateSpecializationKind()));
51*7330f729Sjoerg // These template parameters are for friend types only.
52*7330f729Sjoerg assert(FriendTypeTPLists.empty());
53*7330f729Sjoerg }
54*7330f729Sjoerg #endif
55*7330f729Sjoerg
56*7330f729Sjoerg std::size_t Extra =
57*7330f729Sjoerg FriendDecl::additionalSizeToAlloc<TemplateParameterList *>(
58*7330f729Sjoerg FriendTypeTPLists.size());
59*7330f729Sjoerg auto *FD = new (C, DC, Extra) FriendDecl(DC, L, Friend, FriendL,
60*7330f729Sjoerg FriendTypeTPLists);
61*7330f729Sjoerg cast<CXXRecordDecl>(DC)->pushFriendDecl(FD);
62*7330f729Sjoerg return FD;
63*7330f729Sjoerg }
64*7330f729Sjoerg
CreateDeserialized(ASTContext & C,unsigned ID,unsigned FriendTypeNumTPLists)65*7330f729Sjoerg FriendDecl *FriendDecl::CreateDeserialized(ASTContext &C, unsigned ID,
66*7330f729Sjoerg unsigned FriendTypeNumTPLists) {
67*7330f729Sjoerg std::size_t Extra =
68*7330f729Sjoerg additionalSizeToAlloc<TemplateParameterList *>(FriendTypeNumTPLists);
69*7330f729Sjoerg return new (C, ID, Extra) FriendDecl(EmptyShell(), FriendTypeNumTPLists);
70*7330f729Sjoerg }
71*7330f729Sjoerg
getFirstFriend() const72*7330f729Sjoerg FriendDecl *CXXRecordDecl::getFirstFriend() const {
73*7330f729Sjoerg ExternalASTSource *Source = getParentASTContext().getExternalSource();
74*7330f729Sjoerg Decl *First = data().FirstFriend.get(Source);
75*7330f729Sjoerg return First ? cast<FriendDecl>(First) : nullptr;
76*7330f729Sjoerg }
77