Chris Lattner | 57698e2 | 2002-03-26 18:01:55 +0000 | [diff] [blame] | 1 | //===-- Function.cpp - Implement the Global object classes -------*- C++ -*--=// |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 2 | // |
Chris Lattner | 57698e2 | 2002-03-26 18:01:55 +0000 | [diff] [blame] | 3 | // This file implements the Function & GlobalVariable classes for the VMCore |
Chris Lattner | da97550 | 2001-09-10 07:58:01 +0000 | [diff] [blame] | 4 | // library. |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 5 | // |
| 6 | //===----------------------------------------------------------------------===// |
| 7 | |
| 8 | #include "llvm/ValueHolderImpl.h" |
| 9 | #include "llvm/DerivedTypes.h" |
| 10 | #include "llvm/SymbolTable.h" |
| 11 | #include "llvm/Module.h" |
Chris Lattner | 57698e2 | 2002-03-26 18:01:55 +0000 | [diff] [blame] | 12 | #include "llvm/Function.h" |
Chris Lattner | da97550 | 2001-09-10 07:58:01 +0000 | [diff] [blame] | 13 | #include "llvm/GlobalVariable.h" |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 14 | #include "llvm/BasicBlock.h" |
| 15 | #include "llvm/iOther.h" |
| 16 | |
Chris Lattner | da97550 | 2001-09-10 07:58:01 +0000 | [diff] [blame] | 17 | //===----------------------------------------------------------------------===// |
Chris Lattner | 57698e2 | 2002-03-26 18:01:55 +0000 | [diff] [blame] | 18 | // Function Implementation |
Chris Lattner | da97550 | 2001-09-10 07:58:01 +0000 | [diff] [blame] | 19 | //===----------------------------------------------------------------------===// |
| 20 | |
| 21 | |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 22 | // Instantiate Templates - This ugliness is the price we have to pay |
| 23 | // for having a ValueHolderImpl.h file seperate from ValueHolder.h! :( |
| 24 | // |
Chris Lattner | 57698e2 | 2002-03-26 18:01:55 +0000 | [diff] [blame] | 25 | template class ValueHolder<FunctionArgument, Function, Function>; |
| 26 | template class ValueHolder<BasicBlock , Function, Function>; |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 27 | |
Chris Lattner | 91db582 | 2002-03-29 03:44:36 +0000 | [diff] [blame^] | 28 | Function::Function(const FunctionType *Ty, bool isInternal, |
Chris Lattner | 4e8c487 | 2002-03-23 22:51:58 +0000 | [diff] [blame] | 29 | const std::string &name) |
Chris Lattner | 57698e2 | 2002-03-26 18:01:55 +0000 | [diff] [blame] | 30 | : GlobalValue(PointerType::get(Ty), Value::FunctionVal, isInternal, name), |
Vikram S. Adve | b375b89 | 2001-11-08 04:38:58 +0000 | [diff] [blame] | 31 | SymTabValue(this), BasicBlocks(this), ArgumentList(this, this) { |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 32 | } |
| 33 | |
Chris Lattner | 4e8c487 | 2002-03-23 22:51:58 +0000 | [diff] [blame] | 34 | Function::~Function() { |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 35 | dropAllReferences(); // After this it is safe to delete instructions. |
| 36 | |
| 37 | // TODO: Should remove from the end, not the beginning of vector! |
Chris Lattner | 4cee8d8 | 2001-06-27 23:41:11 +0000 | [diff] [blame] | 38 | iterator BI = begin(); |
| 39 | while ((BI = begin()) != end()) |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 40 | delete BasicBlocks.remove(BI); |
| 41 | |
| 42 | // Delete all of the method arguments and unlink from symbol table... |
| 43 | ArgumentList.delete_all(); |
| 44 | ArgumentList.setParent(0); |
| 45 | } |
| 46 | |
| 47 | // Specialize setName to take care of symbol table majik |
Chris Lattner | 4e8c487 | 2002-03-23 22:51:58 +0000 | [diff] [blame] | 48 | void Function::setName(const std::string &name, SymbolTable *ST) { |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 49 | Module *P; |
Chris Lattner | 5c764a5 | 2001-09-07 16:47:18 +0000 | [diff] [blame] | 50 | assert((ST == 0 || (!getParent() || ST == getParent()->getSymbolTable())) && |
| 51 | "Invalid symtab argument!"); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 52 | if ((P = getParent()) && hasName()) P->getSymbolTable()->remove(this); |
| 53 | Value::setName(name); |
| 54 | if (P && getName() != "") P->getSymbolTableSure()->insert(this); |
| 55 | } |
| 56 | |
Chris Lattner | 4e8c487 | 2002-03-23 22:51:58 +0000 | [diff] [blame] | 57 | void Function::setParent(Module *parent) { |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 58 | Parent = parent; |
| 59 | |
| 60 | // Relink symbol tables together... |
| 61 | setParentSymTab(Parent ? Parent->getSymbolTableSure() : 0); |
| 62 | } |
| 63 | |
Chris Lattner | 91db582 | 2002-03-29 03:44:36 +0000 | [diff] [blame^] | 64 | const FunctionType *Function::getFunctionType() const { |
| 65 | return cast<FunctionType>(getType()->getElementType()); |
Chris Lattner | 7fac070 | 2001-10-03 14:53:21 +0000 | [diff] [blame] | 66 | } |
| 67 | |
Chris Lattner | 4e8c487 | 2002-03-23 22:51:58 +0000 | [diff] [blame] | 68 | const Type *Function::getReturnType() const { |
Chris Lattner | 91db582 | 2002-03-29 03:44:36 +0000 | [diff] [blame^] | 69 | return getFunctionType()->getReturnType(); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 70 | } |
| 71 | |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 72 | // dropAllReferences() - This function causes all the subinstructions to "let |
| 73 | // go" of all references that they are maintaining. This allows one to |
| 74 | // 'delete' a whole class at a time, even though there may be circular |
| 75 | // references... first all references are dropped, and all use counts go to |
| 76 | // zero. Then everything is delete'd for real. Note that no operations are |
| 77 | // valid on an object that has "dropped all references", except operator |
| 78 | // delete. |
| 79 | // |
Chris Lattner | 4e8c487 | 2002-03-23 22:51:58 +0000 | [diff] [blame] | 80 | void Function::dropAllReferences() { |
Chris Lattner | 4cee8d8 | 2001-06-27 23:41:11 +0000 | [diff] [blame] | 81 | for_each(begin(), end(), std::mem_fun(&BasicBlock::dropAllReferences)); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 82 | } |
Chris Lattner | da97550 | 2001-09-10 07:58:01 +0000 | [diff] [blame] | 83 | |
| 84 | //===----------------------------------------------------------------------===// |
| 85 | // GlobalVariable Implementation |
| 86 | //===----------------------------------------------------------------------===// |
| 87 | |
Chris Lattner | 3462ae3 | 2001-12-03 22:26:30 +0000 | [diff] [blame] | 88 | GlobalVariable::GlobalVariable(const Type *Ty, bool constant, bool isIntern, |
| 89 | Constant *Initializer = 0, |
Chris Lattner | 7f74a56 | 2002-01-20 22:54:45 +0000 | [diff] [blame] | 90 | const std::string &Name = "") |
Chris Lattner | 7a78722 | 2001-11-26 19:14:56 +0000 | [diff] [blame] | 91 | : GlobalValue(PointerType::get(Ty), Value::GlobalVariableVal, isIntern, Name), |
Chris Lattner | 3462ae3 | 2001-12-03 22:26:30 +0000 | [diff] [blame] | 92 | isConstantGlobal(constant) { |
Chris Lattner | 3779864 | 2001-09-18 04:01:05 +0000 | [diff] [blame] | 93 | if (Initializer) Operands.push_back(Use((Value*)Initializer, this)); |
Chris Lattner | da97550 | 2001-09-10 07:58:01 +0000 | [diff] [blame] | 94 | } |
| 95 | |
| 96 | // Specialize setName to take care of symbol table majik |
Chris Lattner | 7f74a56 | 2002-01-20 22:54:45 +0000 | [diff] [blame] | 97 | void GlobalVariable::setName(const std::string &name, SymbolTable *ST) { |
Chris Lattner | da97550 | 2001-09-10 07:58:01 +0000 | [diff] [blame] | 98 | Module *P; |
| 99 | assert((ST == 0 || (!getParent() || ST == getParent()->getSymbolTable())) && |
| 100 | "Invalid symtab argument!"); |
| 101 | if ((P = getParent()) && hasName()) P->getSymbolTable()->remove(this); |
| 102 | Value::setName(name); |
| 103 | if (P && getName() != "") P->getSymbolTableSure()->insert(this); |
| 104 | } |