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 |