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, typename ItemParentClass>
28 template<typename TPtr>
29 void SymbolTableListTraits<ValueSubClass,ItemParentClass>
setSymTabObject(TPtr * Dest,TPtr Src)30 ::setSymTabObject(TPtr *Dest, TPtr Src) {
31 // Get the old symtab and value list before doing the assignment.
32 ValueSymbolTable *OldST = TraitsClass::getSymTab(getListOwner());
33
34 // Do it.
35 *Dest = Src;
36
37 // Get the new SymTab object.
38 ValueSymbolTable *NewST = TraitsClass::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 iplist<ValueSubClass> &ItemList = TraitsClass::getList(getListOwner());
45 if (ItemList.empty()) return;
46
47 if (OldST) {
48 // Remove all entries from the previous symtab.
49 for (typename iplist<ValueSubClass>::iterator I = ItemList.begin();
50 I != ItemList.end(); ++I)
51 if (I->hasName())
52 OldST->removeValueName(I->getValueName());
53 }
54
55 if (NewST) {
56 // Add all of the items to the new symtab.
57 for (typename iplist<ValueSubClass>::iterator I = ItemList.begin();
58 I != ItemList.end(); ++I)
59 if (I->hasName())
60 NewST->reinsertValue(I);
61 }
62
63 }
64
65 template<typename ValueSubClass, typename ItemParentClass>
66 void SymbolTableListTraits<ValueSubClass,ItemParentClass>
addNodeToList(ValueSubClass * V)67 ::addNodeToList(ValueSubClass *V) {
68 assert(!V->getParent() && "Value already in a container!!");
69 ItemParentClass *Owner = getListOwner();
70 V->setParent(Owner);
71 if (V->hasName())
72 if (ValueSymbolTable *ST = TraitsClass::getSymTab(Owner))
73 ST->reinsertValue(V);
74 }
75
76 template<typename ValueSubClass, typename ItemParentClass>
77 void SymbolTableListTraits<ValueSubClass,ItemParentClass>
removeNodeFromList(ValueSubClass * V)78 ::removeNodeFromList(ValueSubClass *V) {
79 V->setParent(nullptr);
80 if (V->hasName())
81 if (ValueSymbolTable *ST = TraitsClass::getSymTab(getListOwner()))
82 ST->removeValueName(V->getValueName());
83 }
84
85 template<typename ValueSubClass, typename ItemParentClass>
86 void SymbolTableListTraits<ValueSubClass,ItemParentClass>
transferNodesFromList(ilist_traits<ValueSubClass> & L2,ilist_iterator<ValueSubClass> first,ilist_iterator<ValueSubClass> last)87 ::transferNodesFromList(ilist_traits<ValueSubClass> &L2,
88 ilist_iterator<ValueSubClass> first,
89 ilist_iterator<ValueSubClass> last) {
90 // We only have to do work here if transferring instructions between BBs
91 ItemParentClass *NewIP = getListOwner(), *OldIP = L2.getListOwner();
92 if (NewIP == OldIP) return; // No work to do at all...
93
94 // We only have to update symbol table entries if we are transferring the
95 // instructions to a different symtab object...
96 ValueSymbolTable *NewST = TraitsClass::getSymTab(NewIP);
97 ValueSymbolTable *OldST = TraitsClass::getSymTab(OldIP);
98 if (NewST != OldST) {
99 for (; first != last; ++first) {
100 ValueSubClass &V = *first;
101 bool HasName = V.hasName();
102 if (OldST && HasName)
103 OldST->removeValueName(V.getValueName());
104 V.setParent(NewIP);
105 if (NewST && HasName)
106 NewST->reinsertValue(&V);
107 }
108 } else {
109 // Just transferring between blocks in the same function, simply update the
110 // parent fields in the instructions...
111 for (; first != last; ++first)
112 first->setParent(NewIP);
113 }
114 }
115
116 } // End llvm namespace
117
118 #endif
119