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_SYMBOLTABLELISTTRAITS_IMPL_H
17#define LLVM_SYMBOLTABLELISTTRAITS_IMPL_H
18
19#include "llvm/IR/SymbolTableListTraits.h"
20#include "llvm/IR/ValueSymbolTable.h"
21
22namespace 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.
27template<typename ValueSubClass, typename ItemParentClass>
28template<typename TPtr>
29void 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(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
65template<typename ValueSubClass, typename ItemParentClass>
66void SymbolTableListTraits<ValueSubClass,ItemParentClass>
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
76template<typename ValueSubClass, typename ItemParentClass>
77void SymbolTableListTraits<ValueSubClass,ItemParentClass>
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
85template<typename ValueSubClass, typename ItemParentClass>
86void SymbolTableListTraits<ValueSubClass,ItemParentClass>
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