lib/IR/SymbolTableListTraitsImpl.h

   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>
  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>
  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>
  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>
  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