X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=lib%2FVMCore%2FFunction.cpp;h=70569c0f67ce7539f0e878921347b0fd5260ee95;hb=627079d42a1340360f8699cd87865e20799cff21;hp=5d1132f0a7d6f2d78df77cb1bb28138527ca1fbb;hpb=7f5ef148b07e1ea547d2a9c01c29851247a47882;p=oota-llvm.git diff --git a/lib/VMCore/Function.cpp b/lib/VMCore/Function.cpp index 5d1132f0a7d..70569c0f67c 100644 --- a/lib/VMCore/Function.cpp +++ b/lib/VMCore/Function.cpp @@ -1,69 +1,140 @@ -//===-- Method.cpp - Implement the Method class ------------------*- C++ -*--=// +//===-- Function.cpp - Implement the Global object classes -------*- C++ -*--=// // -// This file implements the Method & GlobalVariable classes for the VMCore +// This file implements the Function & GlobalVariable classes for the VMCore // library. // //===----------------------------------------------------------------------===// -#include "llvm/ValueHolderImpl.h" -#include "llvm/DerivedTypes.h" -#include "llvm/SymbolTable.h" #include "llvm/Module.h" -#include "llvm/Method.h" -#include "llvm/GlobalVariable.h" -#include "llvm/BasicBlock.h" +#include "llvm/DerivedTypes.h" #include "llvm/iOther.h" +#include "Support/LeakDetector.h" +#include "SymbolTableListTraitsImpl.h" + +BasicBlock *ilist_traits::createNode() { + BasicBlock *Ret = new BasicBlock(); + // This should not be garbage monitored. + LeakDetector::removeGarbageObject(Ret); + return Ret; +} + +iplist &ilist_traits::getList(Function *F) { + return F->getBasicBlockList(); +} + +Argument *ilist_traits::createNode() { + Argument *Ret = new Argument(Type::IntTy); + // This should not be garbage monitored. + LeakDetector::removeGarbageObject(Ret); + return Ret; +} + +iplist &ilist_traits::getList(Function *F) { + return F->getArgumentList(); +} + +// Explicit instantiations of SymbolTableListTraits since some of the methods +// are not in the public header file... +template SymbolTableListTraits; +template SymbolTableListTraits; //===----------------------------------------------------------------------===// -// Method Implementation +// Argument Implementation //===----------------------------------------------------------------------===// +Argument::Argument(const Type *Ty, const std::string &Name, Function *Par) + : Value(Ty, Value::ArgumentVal, Name) { + Parent = 0; -// Instantiate Templates - This ugliness is the price we have to pay -// for having a ValueHolderImpl.h file seperate from ValueHolder.h! :( -// -template class ValueHolder; -template class ValueHolder; + // Make sure that we get added to a function + LeakDetector::addGarbageObject(this); -Method::Method(const MethodType *Ty, const string &name) - : Value(Ty, Value::MethodVal, name), SymTabValue(this), BasicBlocks(this), - ArgumentList(this, this) { - assert(Ty->isMethodType() && "Method signature must be of method type!"); - Parent = 0; + if (Par) + Par->getArgumentList().push_back(this); } -Method::~Method() { + +// Specialize setName to take care of symbol table majik +void Argument::setName(const std::string &name, SymbolTable *ST) { + Function *P; + assert((ST == 0 || (!getParent() || ST == &getParent()->getSymbolTable())) && + "Invalid symtab argument!"); + if ((P = getParent()) && hasName()) P->getSymbolTable().remove(this); + Value::setName(name); + if (P && hasName()) P->getSymbolTable().insert(this); +} + +void Argument::setParent(Function *parent) { + if (getParent()) + LeakDetector::addGarbageObject(this); + Parent = parent; + if (getParent()) + LeakDetector::removeGarbageObject(this); +} + + +//===----------------------------------------------------------------------===// +// Function Implementation +//===----------------------------------------------------------------------===// + +Function::Function(const FunctionType *Ty, bool isInternal, + const std::string &name, Module *ParentModule) + : GlobalValue(PointerType::get(Ty), Value::FunctionVal, isInternal, name) { + BasicBlocks.setItemParent(this); + BasicBlocks.setParent(this); + ArgumentList.setItemParent(this); + ArgumentList.setParent(this); + SymTab = new SymbolTable(); + + // Create the arguments vector, all arguments start out unnamed. + for (unsigned i = 0, e = Ty->getNumParams(); i != e; ++i) { + assert(Ty->getParamType(i) != Type::VoidTy && + "Cannot have void typed arguments!"); + ArgumentList.push_back(new Argument(Ty->getParamType(i))); + } + + // Make sure that we get added to a function + LeakDetector::addGarbageObject(this); + + if (ParentModule) + ParentModule->getFunctionList().push_back(this); +} + +Function::~Function() { dropAllReferences(); // After this it is safe to delete instructions. - // TODO: Should remove from the end, not the beginning of vector! - iterator BI = begin(); - while ((BI = begin()) != end()) - delete BasicBlocks.remove(BI); + BasicBlocks.clear(); // Delete all basic blocks... // Delete all of the method arguments and unlink from symbol table... - ArgumentList.delete_all(); + ArgumentList.clear(); ArgumentList.setParent(0); + delete SymTab; } // Specialize setName to take care of symbol table majik -void Method::setName(const string &name, SymbolTable *ST) { +void Function::setName(const std::string &name, SymbolTable *ST) { Module *P; - assert((ST == 0 || (!getParent() || ST == getParent()->getSymbolTable())) && + assert((ST == 0 || (!getParent() || ST == &getParent()->getSymbolTable())) && "Invalid symtab argument!"); - if ((P = getParent()) && hasName()) P->getSymbolTable()->remove(this); + if ((P = getParent()) && hasName()) P->getSymbolTable().remove(this); Value::setName(name); - if (P && getName() != "") P->getSymbolTableSure()->insert(this); + if (P && getName() != "") P->getSymbolTable().insert(this); } -void Method::setParent(Module *parent) { +void Function::setParent(Module *parent) { + if (getParent()) + LeakDetector::addGarbageObject(this); Parent = parent; + if (getParent()) + LeakDetector::removeGarbageObject(this); +} - // Relink symbol tables together... - setParentSymTab(Parent ? Parent->getSymbolTableSure() : 0); +const FunctionType *Function::getFunctionType() const { + return cast(getType()->getElementType()); } -const Type *Method::getReturnType() const { - return ((const MethodType *)getType())->getReturnType(); +const Type *Function::getReturnType() const { + return getFunctionType()->getReturnType(); } // dropAllReferences() - This function causes all the subinstructions to "let @@ -74,31 +145,42 @@ const Type *Method::getReturnType() const { // valid on an object that has "dropped all references", except operator // delete. // -void Method::dropAllReferences() { - for_each(begin(), end(), std::mem_fun(&BasicBlock::dropAllReferences)); +void Function::dropAllReferences() { + for (iterator I = begin(), E = end(); I != E; ++I) + I->dropAllReferences(); } //===----------------------------------------------------------------------===// // GlobalVariable Implementation //===----------------------------------------------------------------------===// -GlobalVariable::GlobalVariable(const Type *Ty, bool isConstant, - ConstPoolVal *Initializer = 0, - const string &Name = "") - : User(Ty, Value::GlobalVal, Name), Parent(0), Constant(isConstant) { - assert(Ty->isPointerType() && "Global Variables must be pointers!"); +GlobalVariable::GlobalVariable(const Type *Ty, bool constant, bool isIntern, + Constant *Initializer, + const std::string &Name, Module *ParentModule) + : GlobalValue(PointerType::get(Ty), Value::GlobalVariableVal, isIntern, Name), + isConstantGlobal(constant) { if (Initializer) Operands.push_back(Use((Value*)Initializer, this)); - assert(!isConstant || hasInitializer() && - "Globals Constants must have an initializer!"); + LeakDetector::addGarbageObject(this); + + if (ParentModule) + ParentModule->getGlobalList().push_back(this); +} + +void GlobalVariable::setParent(Module *parent) { + if (getParent()) + LeakDetector::addGarbageObject(this); + Parent = parent; + if (getParent()) + LeakDetector::removeGarbageObject(this); } // Specialize setName to take care of symbol table majik -void GlobalVariable::setName(const string &name, SymbolTable *ST) { +void GlobalVariable::setName(const std::string &name, SymbolTable *ST) { Module *P; - assert((ST == 0 || (!getParent() || ST == getParent()->getSymbolTable())) && + assert((ST == 0 || (!getParent() || ST == &getParent()->getSymbolTable())) && "Invalid symtab argument!"); - if ((P = getParent()) && hasName()) P->getSymbolTable()->remove(this); + if ((P = getParent()) && hasName()) P->getSymbolTable().remove(this); Value::setName(name); - if (P && getName() != "") P->getSymbolTableSure()->insert(this); + if (P && getName() != "") P->getSymbolTable().insert(this); }