Refactor SymbolTableListTraits to only have a single pointer in it, instead
[oota-llvm.git] / lib / VMCore / SymbolTableListTraitsImpl.h
1 //===-- llvm/SymbolTableListTraitsImpl.h - Implementation ------*- C++ -*--===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file was developed by the LLVM research group and is distributed under
6 // the University of Illinois Open Source 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_SYMBOLTABLELISTTRAITS_IMPL_H
17 #define LLVM_SYMBOLTABLELISTTRAITS_IMPL_H
18
19 #include "llvm/SymbolTableListTraits.h"
20 #include "llvm/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>
30 ::setSymTabObject(TPtr *Dest, TPtr Src) {
31   // Get the old symtab and value list before doing the assignment.
32   ValueSymbolTable *OldST = TraitsClass::getSymTab(ItemParent);
33
34   // Do it.
35   *Dest = Src;
36   
37   // Get the new SymTab object.
38   ValueSymbolTable *NewST = TraitsClass::getSymTab(ItemParent);
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(ItemParent);
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>
67 ::addNodeToList(ValueSubClass *V) {
68   assert(V->getParent() == 0 && "Value already in a container!!");
69   V->setParent(ItemParent);
70   if (V->hasName())
71     if (ValueSymbolTable *ST = TraitsClass::getSymTab(ItemParent))
72       ST->reinsertValue(V);
73 }
74
75 template<typename ValueSubClass, typename ItemParentClass>
76 void SymbolTableListTraits<ValueSubClass,ItemParentClass>
77 ::removeNodeFromList(ValueSubClass *V) {
78   V->setParent(0);
79   if (V->hasName())
80     if (ValueSymbolTable *ST = TraitsClass::getSymTab(ItemParent))
81       ST->removeValueName(V->getValueName());
82 }
83
84 template<typename ValueSubClass, typename ItemParentClass>
85 void SymbolTableListTraits<ValueSubClass,ItemParentClass>
86 ::transferNodesFromList(iplist<ValueSubClass, ilist_traits<ValueSubClass> > &L2,
87                         ilist_iterator<ValueSubClass> first,
88                         ilist_iterator<ValueSubClass> last) {
89   // We only have to do work here if transferring instructions between BBs
90   ItemParentClass *NewIP = ItemParent, *OldIP = L2.ItemParent;
91   if (NewIP == OldIP) return;  // No work to do at all...
92
93   // We only have to update symbol table entries if we are transferring the
94   // instructions to a different symtab object...
95   ValueSymbolTable *NewST = TraitsClass::getSymTab(ItemParent);
96   ValueSymbolTable *OldST = TraitsClass::getSymTab(OldIP);
97   if (NewST != OldST) {
98     for (; first != last; ++first) {
99       ValueSubClass &V = *first;
100       bool HasName = V.hasName();
101       if (OldST && HasName)
102         OldST->removeValueName(V.getValueName());
103       V.setParent(NewIP);
104       if (NewST && HasName)
105         NewST->reinsertValue(&V);
106     }
107   } else {
108     // Just transferring between blocks in the same function, simply update the
109     // parent fields in the instructions...
110     for (; first != last; ++first)
111       first->setParent(NewIP);
112   }
113 }
114
115 } // End llvm namespace
116
117 #endif