Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line |
|---|---|---|---|
| 14 | pmbaty | 1 | //===- llvm/SymbolTableListTraits.h - Traits for iplist ---------*- 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 defines a generic class that is used to implement the automatic |
||
| 10 | // symbol table manipulation that occurs when you put (for example) a named |
||
| 11 | // instruction into a basic block. |
||
| 12 | // |
||
| 13 | // The way that this is implemented is by using a special traits class with the |
||
| 14 | // intrusive list that makes up the list of instructions in a basic block. When |
||
| 15 | // a new element is added to the list of instructions, the traits class is |
||
| 16 | // notified, allowing the symbol table to be updated. |
||
| 17 | // |
||
| 18 | // This generic class implements the traits class. It must be generic so that |
||
| 19 | // it can work for all uses it, which include lists of instructions, basic |
||
| 20 | // blocks, arguments, functions, global variables, etc... |
||
| 21 | // |
||
| 22 | //===----------------------------------------------------------------------===// |
||
| 23 | |||
| 24 | #ifndef LLVM_IR_SYMBOLTABLELISTTRAITS_H |
||
| 25 | #define LLVM_IR_SYMBOLTABLELISTTRAITS_H |
||
| 26 | |||
| 27 | #include "llvm/ADT/ilist.h" |
||
| 28 | #include "llvm/ADT/simple_ilist.h" |
||
| 29 | #include <cstddef> |
||
| 30 | |||
| 31 | namespace llvm { |
||
| 32 | |||
| 33 | class Argument; |
||
| 34 | class BasicBlock; |
||
| 35 | class Function; |
||
| 36 | class GlobalAlias; |
||
| 37 | class GlobalIFunc; |
||
| 38 | class GlobalVariable; |
||
| 39 | class Instruction; |
||
| 40 | class Module; |
||
| 41 | class ValueSymbolTable; |
||
| 42 | |||
| 43 | /// Template metafunction to get the parent type for a symbol table list. |
||
| 44 | /// |
||
| 45 | /// Implementations create a typedef called \c type so that we only need a |
||
| 46 | /// single template parameter for the list and traits. |
||
| 47 | template <typename NodeTy> struct SymbolTableListParentType {}; |
||
| 48 | |||
| 49 | #define DEFINE_SYMBOL_TABLE_PARENT_TYPE(NODE, PARENT) \ |
||
| 50 | template <> struct SymbolTableListParentType<NODE> { using type = PARENT; }; |
||
| 51 | DEFINE_SYMBOL_TABLE_PARENT_TYPE(Instruction, BasicBlock) |
||
| 52 | DEFINE_SYMBOL_TABLE_PARENT_TYPE(BasicBlock, Function) |
||
| 53 | DEFINE_SYMBOL_TABLE_PARENT_TYPE(Argument, Function) |
||
| 54 | DEFINE_SYMBOL_TABLE_PARENT_TYPE(Function, Module) |
||
| 55 | DEFINE_SYMBOL_TABLE_PARENT_TYPE(GlobalVariable, Module) |
||
| 56 | DEFINE_SYMBOL_TABLE_PARENT_TYPE(GlobalAlias, Module) |
||
| 57 | DEFINE_SYMBOL_TABLE_PARENT_TYPE(GlobalIFunc, Module) |
||
| 58 | #undef DEFINE_SYMBOL_TABLE_PARENT_TYPE |
||
| 59 | |||
| 60 | template <typename NodeTy> class SymbolTableList; |
||
| 61 | |||
| 62 | // ValueSubClass - The type of objects that I hold, e.g. Instruction. |
||
| 63 | // ItemParentClass - The type of object that owns the list, e.g. BasicBlock. |
||
| 64 | // |
||
| 65 | template <typename ValueSubClass> |
||
| 66 | class SymbolTableListTraits : public ilist_alloc_traits<ValueSubClass> { |
||
| 67 | using ListTy = SymbolTableList<ValueSubClass>; |
||
| 68 | using iterator = typename simple_ilist<ValueSubClass>::iterator; |
||
| 69 | using ItemParentClass = |
||
| 70 | typename SymbolTableListParentType<ValueSubClass>::type; |
||
| 71 | |||
| 72 | public: |
||
| 73 | SymbolTableListTraits() = default; |
||
| 74 | |||
| 75 | private: |
||
| 76 | /// getListOwner - Return the object that owns this list. If this is a list |
||
| 77 | /// of instructions, it returns the BasicBlock that owns them. |
||
| 78 | ItemParentClass *getListOwner() { |
||
| 79 | size_t Offset = reinterpret_cast<size_t>( |
||
| 80 | &((ItemParentClass *)nullptr->*ItemParentClass::getSublistAccess( |
||
| 81 | static_cast<ValueSubClass *>( |
||
| 82 | nullptr)))); |
||
| 83 | ListTy *Anchor = static_cast<ListTy *>(this); |
||
| 84 | return reinterpret_cast<ItemParentClass*>(reinterpret_cast<char*>(Anchor)- |
||
| 85 | Offset); |
||
| 86 | } |
||
| 87 | |||
| 88 | static ListTy &getList(ItemParentClass *Par) { |
||
| 89 | return Par->*(Par->getSublistAccess((ValueSubClass*)nullptr)); |
||
| 90 | } |
||
| 91 | |||
| 92 | static ValueSymbolTable *getSymTab(ItemParentClass *Par) { |
||
| 93 | return Par ? toPtr(Par->getValueSymbolTable()) : nullptr; |
||
| 94 | } |
||
| 95 | |||
| 96 | public: |
||
| 97 | void addNodeToList(ValueSubClass *V); |
||
| 98 | void removeNodeFromList(ValueSubClass *V); |
||
| 99 | void transferNodesFromList(SymbolTableListTraits &L2, iterator first, |
||
| 100 | iterator last); |
||
| 101 | // private: |
||
| 102 | template<typename TPtr> |
||
| 103 | void setSymTabObject(TPtr *, TPtr); |
||
| 104 | static ValueSymbolTable *toPtr(ValueSymbolTable *P) { return P; } |
||
| 105 | static ValueSymbolTable *toPtr(ValueSymbolTable &R) { return &R; } |
||
| 106 | }; |
||
| 107 | |||
| 108 | /// List that automatically updates parent links and symbol tables. |
||
| 109 | /// |
||
| 110 | /// When nodes are inserted into and removed from this list, the associated |
||
| 111 | /// symbol table will be automatically updated. Similarly, parent links get |
||
| 112 | /// updated automatically. |
||
| 113 | template <class T> |
||
| 114 | class SymbolTableList |
||
| 115 | : public iplist_impl<simple_ilist<T>, SymbolTableListTraits<T>> {}; |
||
| 116 | |||
| 117 | } // end namespace llvm |
||
| 118 | |||
| 119 | #endif // LLVM_IR_SYMBOLTABLELISTTRAITS_H |