Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
14 | pmbaty | 1 | //===- ConstantPools.h - Keep track of assembler-generated ------*- 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 | // This file declares the ConstantPool and AssemblerConstantPools classes. |
||
10 | // |
||
11 | //===----------------------------------------------------------------------===// |
||
12 | |||
13 | #ifndef LLVM_MC_CONSTANTPOOLS_H |
||
14 | #define LLVM_MC_CONSTANTPOOLS_H |
||
15 | |||
16 | #include "llvm/ADT/MapVector.h" |
||
17 | #include "llvm/ADT/SmallVector.h" |
||
18 | #include "llvm/Support/SMLoc.h" |
||
19 | #include <cstdint> |
||
20 | #include <map> |
||
21 | |||
22 | namespace llvm { |
||
23 | |||
24 | class MCContext; |
||
25 | class MCExpr; |
||
26 | class MCSection; |
||
27 | class MCStreamer; |
||
28 | class MCSymbol; |
||
29 | class MCSymbolRefExpr; |
||
30 | |||
31 | struct ConstantPoolEntry { |
||
32 | ConstantPoolEntry(MCSymbol *L, const MCExpr *Val, unsigned Sz, SMLoc Loc_) |
||
33 | : Label(L), Value(Val), Size(Sz), Loc(Loc_) {} |
||
34 | |||
35 | MCSymbol *Label; |
||
36 | const MCExpr *Value; |
||
37 | unsigned Size; |
||
38 | SMLoc Loc; |
||
39 | }; |
||
40 | |||
41 | // A class to keep track of assembler-generated constant pools that are use to |
||
42 | // implement the ldr-pseudo. |
||
43 | class ConstantPool { |
||
44 | using EntryVecTy = SmallVector<ConstantPoolEntry, 4>; |
||
45 | EntryVecTy Entries; |
||
46 | std::map<int64_t, const MCSymbolRefExpr *> CachedConstantEntries; |
||
47 | DenseMap<const MCSymbol *, const MCSymbolRefExpr *> CachedSymbolEntries; |
||
48 | |||
49 | public: |
||
50 | // Initialize a new empty constant pool |
||
51 | ConstantPool() = default; |
||
52 | |||
53 | // Add a new entry to the constant pool in the next slot. |
||
54 | // \param Value is the new entry to put in the constant pool. |
||
55 | // \param Size is the size in bytes of the entry |
||
56 | // |
||
57 | // \returns a MCExpr that references the newly inserted value |
||
58 | const MCExpr *addEntry(const MCExpr *Value, MCContext &Context, |
||
59 | unsigned Size, SMLoc Loc); |
||
60 | |||
61 | // Emit the contents of the constant pool using the provided streamer. |
||
62 | void emitEntries(MCStreamer &Streamer); |
||
63 | |||
64 | // Return true if the constant pool is empty |
||
65 | bool empty(); |
||
66 | |||
67 | void clearCache(); |
||
68 | }; |
||
69 | |||
70 | class AssemblerConstantPools { |
||
71 | // Map type used to keep track of per-Section constant pools used by the |
||
72 | // ldr-pseudo opcode. The map associates a section to its constant pool. The |
||
73 | // constant pool is a vector of (label, value) pairs. When the ldr |
||
74 | // pseudo is parsed we insert a new (label, value) pair into the constant pool |
||
75 | // for the current section and add MCSymbolRefExpr to the new label as |
||
76 | // an opcode to the ldr. After we have parsed all the user input we |
||
77 | // output the (label, value) pairs in each constant pool at the end of the |
||
78 | // section. |
||
79 | // |
||
80 | // We use the MapVector for the map type to ensure stable iteration of |
||
81 | // the sections at the end of the parse. We need to iterate over the |
||
82 | // sections in a stable order to ensure that we have print the |
||
83 | // constant pools in a deterministic order when printing an assembly |
||
84 | // file. |
||
85 | using ConstantPoolMapTy = MapVector<MCSection *, ConstantPool>; |
||
86 | ConstantPoolMapTy ConstantPools; |
||
87 | |||
88 | public: |
||
89 | void emitAll(MCStreamer &Streamer); |
||
90 | void emitForCurrentSection(MCStreamer &Streamer); |
||
91 | void clearCacheForCurrentSection(MCStreamer &Streamer); |
||
92 | const MCExpr *addEntry(MCStreamer &Streamer, const MCExpr *Expr, |
||
93 | unsigned Size, SMLoc Loc); |
||
94 | |||
95 | private: |
||
96 | ConstantPool *getConstantPool(MCSection *Section); |
||
97 | ConstantPool &getOrCreateConstantPool(MCSection *Section); |
||
98 | }; |
||
99 | |||
100 | } // end namespace llvm |
||
101 | |||
102 | #endif // LLVM_MC_CONSTANTPOOLS_H |