summaryrefslogtreecommitdiffstats
path: root/include/clang/Frontend/LayoutOverrideSource.h
blob: 167b4f2a60274181a9b58ba380bb349ccc9f77ba (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
//===--- LayoutOverrideSource.h --Override Record Layouts -----------------===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//

#ifndef LLVM_CLANG_FRONTEND_LAYOUTOVERRIDESOURCE_H
#define LLVM_CLANG_FRONTEND_LAYOUTOVERRIDESOURCE_H

#include "clang/AST/ExternalASTSource.h"
#include "clang/Basic/LLVM.h"
#include "llvm/ADT/StringMap.h"
#include "llvm/ADT/StringRef.h"

namespace clang {
  /// \brief An external AST source that overrides the layout of
  /// a specified set of record types.
  ///
  /// This class is used only for testing the ability of external AST sources
  /// to override the layout of record types. Its input is the output format
  /// of the command-line argument -fdump-record-layouts.
  class LayoutOverrideSource : public ExternalASTSource {
    /// \brief The layout of a given record.
    struct Layout {
      /// \brief The size of the record.
      uint64_t Size;
      
      /// \brief The alignment of the record.
      uint64_t Align;
      
      /// \brief The offsets of the fields, in source order.
      SmallVector<uint64_t, 8> FieldOffsets;
    };
    
    /// \brief The set of layouts that will be overridden.
    llvm::StringMap<Layout> Layouts;
    
  public:
    /// \brief Create a new AST source that overrides the layout of some
    /// set of record types.
    ///
    /// The file is the result of passing -fdump-record-layouts to a file.
    explicit LayoutOverrideSource(StringRef Filename);
    
    /// \brief If this particular record type has an overridden layout,
    /// return that layout.
    bool
    layoutRecordType(const RecordDecl *Record,
       uint64_t &Size, uint64_t &Alignment,
       llvm::DenseMap<const FieldDecl *, uint64_t> &FieldOffsets,
       llvm::DenseMap<const CXXRecordDecl *, CharUnits> &BaseOffsets,
       llvm::DenseMap<const CXXRecordDecl *,
                      CharUnits> &VirtualBaseOffsets) override;
    
    /// \brief Dump the overridden layouts.
    void dump();
  };
} // namespace clang

#endif