xref: /llvm-project/clang/lib/AST/ByteCode/Record.cpp (revision 82ce8296b8024f1fb549c4b2dfcf736f809e19b7)
1 //===--- Record.cpp - struct and class metadata for the VM ------*- 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 #include "Record.h"
10 #include "clang/AST/ASTContext.h"
11 
12 using namespace clang;
13 using namespace clang::interp;
14 
15 Record::Record(const RecordDecl *Decl, BaseList &&SrcBases,
16                FieldList &&SrcFields, VirtualBaseList &&SrcVirtualBases,
17                unsigned VirtualSize, unsigned BaseSize)
18     : Decl(Decl), Bases(std::move(SrcBases)), Fields(std::move(SrcFields)),
19       BaseSize(BaseSize), VirtualSize(VirtualSize), IsUnion(Decl->isUnion()),
20       IsAnonymousUnion(IsUnion && Decl->isAnonymousStructOrUnion()) {
21   for (Base &V : SrcVirtualBases)
22     VirtualBases.push_back({V.Decl, V.Offset + BaseSize, V.Desc, V.R});
23 
24   for (Base &B : Bases)
25     BaseMap[B.Decl] = &B;
26   for (Field &F : Fields)
27     FieldMap[F.Decl] = &F;
28   for (Base &V : VirtualBases)
29     VirtualBaseMap[V.Decl] = &V;
30 }
31 
32 const std::string Record::getName() const {
33   std::string Ret;
34   llvm::raw_string_ostream OS(Ret);
35   Decl->getNameForDiagnostic(OS, Decl->getASTContext().getPrintingPolicy(),
36                              /*Qualified=*/true);
37   return Ret;
38 }
39 
40 const Record::Field *Record::getField(const FieldDecl *FD) const {
41   auto It = FieldMap.find(FD->getFirstDecl());
42   assert(It != FieldMap.end() && "Missing field");
43   return It->second;
44 }
45 
46 const Record::Base *Record::getBase(const RecordDecl *FD) const {
47   auto It = BaseMap.find(FD);
48   assert(It != BaseMap.end() && "Missing base");
49   return It->second;
50 }
51 
52 const Record::Base *Record::getBase(QualType T) const {
53   if (auto *RT = T->getAs<RecordType>()) {
54     const RecordDecl *RD = RT->getDecl();
55     return BaseMap.lookup(RD);
56   }
57   return nullptr;
58 }
59 
60 const Record::Base *Record::getVirtualBase(const RecordDecl *FD) const {
61   auto It = VirtualBaseMap.find(FD);
62   assert(It != VirtualBaseMap.end() && "Missing virtual base");
63   return It->second;
64 }
65