119 lines
		
	
	
		
			4.1 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			119 lines
		
	
	
		
			4.1 KiB
		
	
	
	
		
			C++
		
	
	
	
| //===-- llvm/SymbolTableListTraitsImpl.h - Implementation ------*- C++ -*--===//
 | |
| //
 | |
| //                     The LLVM Compiler Infrastructure
 | |
| //
 | |
| // This file is distributed under the University of Illinois Open Source
 | |
| // License. See LICENSE.TXT for details.
 | |
| //
 | |
| //===----------------------------------------------------------------------===//
 | |
| //
 | |
| // This file implements the stickier parts of the SymbolTableListTraits class,
 | |
| // and is explicitly instantiated where needed to avoid defining all this code
 | |
| // in a widely used header.
 | |
| //
 | |
| //===----------------------------------------------------------------------===//
 | |
| 
 | |
| #ifndef LLVM_SYMBOLTABLELISTTRAITS_IMPL_H
 | |
| #define LLVM_SYMBOLTABLELISTTRAITS_IMPL_H
 | |
| 
 | |
| #include "llvm/SymbolTableListTraits.h"
 | |
| #include "llvm/ValueSymbolTable.h"
 | |
| 
 | |
| namespace llvm {
 | |
| 
 | |
| /// setSymTabObject - This is called when (f.e.) the parent of a basic block
 | |
| /// changes.  This requires us to remove all the instruction symtab entries from
 | |
| /// the current function and reinsert them into the new function.
 | |
| template<typename ValueSubClass, typename ItemParentClass>
 | |
| template<typename TPtr>
 | |
| void SymbolTableListTraits<ValueSubClass,ItemParentClass>
 | |
| ::setSymTabObject(TPtr *Dest, TPtr Src) {
 | |
|   // Get the old symtab and value list before doing the assignment.
 | |
|   ValueSymbolTable *OldST = TraitsClass::getSymTab(getListOwner());
 | |
| 
 | |
|   // Do it.
 | |
|   *Dest = Src;
 | |
|   
 | |
|   // Get the new SymTab object.
 | |
|   ValueSymbolTable *NewST = TraitsClass::getSymTab(getListOwner());
 | |
|   
 | |
|   // If there is nothing to do, quick exit.
 | |
|   if (OldST == NewST) return;
 | |
|   
 | |
|   // Move all the elements from the old symtab to the new one.
 | |
|   iplist<ValueSubClass> &ItemList = TraitsClass::getList(getListOwner());
 | |
|   if (ItemList.empty()) return;
 | |
|   
 | |
|   if (OldST) {
 | |
|     // Remove all entries from the previous symtab.
 | |
|     for (typename iplist<ValueSubClass>::iterator I = ItemList.begin();
 | |
|          I != ItemList.end(); ++I)
 | |
|       if (I->hasName())
 | |
|         OldST->removeValueName(I->getValueName());
 | |
|   }
 | |
| 
 | |
|   if (NewST) {
 | |
|     // Add all of the items to the new symtab.
 | |
|     for (typename iplist<ValueSubClass>::iterator I = ItemList.begin();
 | |
|          I != ItemList.end(); ++I)
 | |
|       if (I->hasName())
 | |
|         NewST->reinsertValue(I);
 | |
|   }
 | |
|   
 | |
| }
 | |
| 
 | |
| template<typename ValueSubClass, typename ItemParentClass>
 | |
| void SymbolTableListTraits<ValueSubClass,ItemParentClass>
 | |
| ::addNodeToList(ValueSubClass *V) {
 | |
|   assert(V->getParent() == 0 && "Value already in a container!!");
 | |
|   ItemParentClass *Owner = getListOwner();
 | |
|   V->setParent(Owner);
 | |
|   if (V->hasName())
 | |
|     if (ValueSymbolTable *ST = TraitsClass::getSymTab(Owner))
 | |
|       ST->reinsertValue(V);
 | |
| }
 | |
| 
 | |
| template<typename ValueSubClass, typename ItemParentClass>
 | |
| void SymbolTableListTraits<ValueSubClass,ItemParentClass>
 | |
| ::removeNodeFromList(ValueSubClass *V) {
 | |
|   V->setParent(0);
 | |
|   if (V->hasName())
 | |
|     if (ValueSymbolTable *ST = TraitsClass::getSymTab(getListOwner()))
 | |
|       ST->removeValueName(V->getValueName());
 | |
| }
 | |
| 
 | |
| template<typename ValueSubClass, typename ItemParentClass>
 | |
| void SymbolTableListTraits<ValueSubClass,ItemParentClass>
 | |
| ::transferNodesFromList(iplist<ValueSubClass, ilist_traits<ValueSubClass> > &L2,
 | |
|                         ilist_iterator<ValueSubClass> first,
 | |
|                         ilist_iterator<ValueSubClass> last) {
 | |
|   // We only have to do work here if transferring instructions between BBs
 | |
|   ItemParentClass *NewIP = getListOwner(), *OldIP = L2.getListOwner();
 | |
|   if (NewIP == OldIP) return;  // No work to do at all...
 | |
| 
 | |
|   // We only have to update symbol table entries if we are transferring the
 | |
|   // instructions to a different symtab object...
 | |
|   ValueSymbolTable *NewST = TraitsClass::getSymTab(NewIP);
 | |
|   ValueSymbolTable *OldST = TraitsClass::getSymTab(OldIP);
 | |
|   if (NewST != OldST) {
 | |
|     for (; first != last; ++first) {
 | |
|       ValueSubClass &V = *first;
 | |
|       bool HasName = V.hasName();
 | |
|       if (OldST && HasName)
 | |
|         OldST->removeValueName(V.getValueName());
 | |
|       V.setParent(NewIP);
 | |
|       if (NewST && HasName)
 | |
|         NewST->reinsertValue(&V);
 | |
|     }
 | |
|   } else {
 | |
|     // Just transferring between blocks in the same function, simply update the
 | |
|     // parent fields in the instructions...
 | |
|     for (; first != last; ++first)
 | |
|       first->setParent(NewIP);
 | |
|   }
 | |
| }
 | |
| 
 | |
| } // End llvm namespace
 | |
| 
 | |
| #endif
 |