1 //===-- llvm/SymbolTableListTraitsImpl.h - Implementation ------*- C++ -*--===//
2 //
3 // The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file implements the stickier parts of the SymbolTableListTraits class,
11 // and is explicitly instantiated where needed to avoid defining all this code
12 // in a widely used header.
13 //
14 //===----------------------------------------------------------------------===//
15
16 #ifndef LLVM_LIB_IR_SYMBOLTABLELISTTRAITSIMPL_H
17 #define LLVM_LIB_IR_SYMBOLTABLELISTTRAITSIMPL_H
18
19 #include "llvm/IR/SymbolTableListTraits.h"
20 #include "llvm/IR/ValueSymbolTable.h"
21
22 namespace llvm {
23
24 /// setSymTabObject - This is called when (f.e.) the parent of a basic block
25 /// changes. This requires us to remove all the instruction symtab entries from
26 /// the current function and reinsert them into the new function.
27 template <typename ValueSubClass>
28 template <typename TPtr>
setSymTabObject(TPtr * Dest,TPtr Src)29 void SymbolTableListTraits<ValueSubClass>::setSymTabObject(TPtr *Dest,
30 TPtr Src) {
31 // Get the old symtab and value list before doing the assignment.
32 ValueSymbolTable *OldST = getSymTab(getListOwner());
33
34 // Do it.
35 *Dest = Src;
36
37 // Get the new SymTab object.
38 ValueSymbolTable *NewST = getSymTab(getListOwner());
39
40 // If there is nothing to do, quick exit.
41 if (OldST == NewST) return;
42
43 // Move all the elements from the old symtab to the new one.
44 ListTy &ItemList = getList(getListOwner());
45 if (ItemList.empty()) return;
46
47 if (OldST) {
48 // Remove all entries from the previous symtab.
49 for (auto I = ItemList.begin(); I != ItemList.end(); ++I)
50 if (I->hasName())
51 OldST->removeValueName(I->getValueName());
52 }
53
54 if (NewST) {
55 // Add all of the items to the new symtab.
56 for (auto I = ItemList.begin(); I != ItemList.end(); ++I)
57 if (I->hasName())
58 NewST->reinsertValue(&*I);
59 }
60
61 }
62
63 template <typename ValueSubClass>
addNodeToList(ValueSubClass * V)64 void SymbolTableListTraits<ValueSubClass>::addNodeToList(ValueSubClass *V) {
65 assert(!V->getParent() && "Value already in a container!!");
66 ItemParentClass *Owner = getListOwner();
67 V->setParent(Owner);
68 if (V->hasName())
69 if (ValueSymbolTable *ST = getSymTab(Owner))
70 ST->reinsertValue(V);
71 }
72
73 template <typename ValueSubClass>
removeNodeFromList(ValueSubClass * V)74 void SymbolTableListTraits<ValueSubClass>::removeNodeFromList(
75 ValueSubClass *V) {
76 V->setParent(nullptr);
77 if (V->hasName())
78 if (ValueSymbolTable *ST = getSymTab(getListOwner()))
79 ST->removeValueName(V->getValueName());
80 }
81
82 template <typename ValueSubClass>
transferNodesFromList(SymbolTableListTraits & L2,ilist_iterator<ValueSubClass> first,ilist_iterator<ValueSubClass> last)83 void SymbolTableListTraits<ValueSubClass>::transferNodesFromList(
84 SymbolTableListTraits &L2, ilist_iterator<ValueSubClass> first,
85 ilist_iterator<ValueSubClass> last) {
86 // We only have to do work here if transferring instructions between BBs
87 ItemParentClass *NewIP = getListOwner(), *OldIP = L2.getListOwner();
88 if (NewIP == OldIP) return; // No work to do at all...
89
90 // We only have to update symbol table entries if we are transferring the
91 // instructions to a different symtab object...
92 ValueSymbolTable *NewST = getSymTab(NewIP);
93 ValueSymbolTable *OldST = getSymTab(OldIP);
94 if (NewST != OldST) {
95 for (; first != last; ++first) {
96 ValueSubClass &V = *first;
97 bool HasName = V.hasName();
98 if (OldST && HasName)
99 OldST->removeValueName(V.getValueName());
100 V.setParent(NewIP);
101 if (NewST && HasName)
102 NewST->reinsertValue(&V);
103 }
104 } else {
105 // Just transferring between blocks in the same function, simply update the
106 // parent fields in the instructions...
107 for (; first != last; ++first)
108 first->setParent(NewIP);
109 }
110 }
111
112 } // End llvm namespace
113
114 #endif
115