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 = 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   iplist<ValueSubClass> &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, typename ItemParentClass>
  64 void SymbolTableListTraits<ValueSubClass,ItemParentClass>
  65 ::addNodeToList(ValueSubClass *V) {
  66   assert(!V->getParent() && "Value already in a container!!");
  67   ItemParentClass *Owner = getListOwner();
  68   V->setParent(Owner);
  69   if (V->hasName())
  70     if (ValueSymbolTable *ST = getSymTab(Owner))
  71       ST->reinsertValue(V);
  72 }
  73
  74 template<typename ValueSubClass, typename ItemParentClass>
  75 void SymbolTableListTraits<ValueSubClass,ItemParentClass>
  76 ::removeNodeFromList(ValueSubClass *V) {
  77   V->setParent(nullptr);
  78   if (V->hasName())
  79     if (ValueSymbolTable *ST = getSymTab(getListOwner()))
  80       ST->removeValueName(V->getValueName());
  81 }
  82
  83 template<typename ValueSubClass, typename ItemParentClass>
  84 void SymbolTableListTraits<ValueSubClass,ItemParentClass>
  85 ::transferNodesFromList(ilist_traits<ValueSubClass> &L2,
  86                         ilist_iterator<ValueSubClass> first,
  87                         ilist_iterator<ValueSubClass> last) {
  88   // We only have to do work here if transferring instructions between BBs
  89   ItemParentClass *NewIP = getListOwner(), *OldIP = L2.getListOwner();
  90   if (NewIP == OldIP) return;  // No work to do at all...
  91
  92   // We only have to update symbol table entries if we are transferring the
  93   // instructions to a different symtab object...
  94   ValueSymbolTable *NewST = getSymTab(NewIP);
  95   ValueSymbolTable *OldST = getSymTab(OldIP);
  96   if (NewST != OldST) {
  97     for (; first != last; ++first) {
  98       ValueSubClass &V = *first;
  99       bool HasName = V.hasName();
 100       if (OldST && HasName)
 101         OldST->removeValueName(V.getValueName());
 102       V.setParent(NewIP);
 103       if (NewST && HasName)
 104         NewST->reinsertValue(&V);
 105     }
 106   } else {
 107     // Just transferring between blocks in the same function, simply update the
 108     // parent fields in the instructions...
 109     for (; first != last; ++first)
 110       first->setParent(NewIP);
 111   }
 112 }
 113
 114 } // End llvm namespace
 115
 116 #endif