diff --git a/lib/ExecutionEngine/Interpreter/Execution.cpp b/lib/ExecutionEngine/Interpreter/Execution.cpp index 4570f1a239e..d3818063d00 100644 --- a/lib/ExecutionEngine/Interpreter/Execution.cpp +++ b/lib/ExecutionEngine/Interpreter/Execution.cpp @@ -120,8 +120,6 @@ static void SetValue(Value *V, GenericValue Val, ExecutionContext &SF) { void Interpreter::initializeExecutionEngine() { TheEE = this; - AnnotationManager::registerAnnotationFactory(FunctionInfoAID, - &FunctionInfo::Create); initializeSignalHandlers(); } @@ -899,7 +897,7 @@ void Interpreter::visitVarArgInst(VarArgInst &I) { // Dispatch and Execution Code //===----------------------------------------------------------------------===// -FunctionInfo::FunctionInfo(Function *F) : Annotation(FunctionInfoAID) { +FunctionInfo::FunctionInfo(Function *F) { // Assign slot numbers to the function arguments... for (Function::const_aiterator AI = F->abegin(), E = F->aend(); AI != E; ++AI) AI->addAnnotation(new SlotNumber(getValueSlot(AI))); @@ -959,11 +957,12 @@ void Interpreter::callFunction(Function *F, // the function. Also calculate the number of values for each type slot // active. // - FunctionInfo *FuncInfo = - (FunctionInfo*)F->getOrCreateAnnotation(FunctionInfoAID); - ECStack.push_back(ExecutionContext()); // Make a new stack frame... + FunctionInfo *&FuncInfo = FunctionInfoMap[F]; + if (!FuncInfo) FuncInfo = new FunctionInfo(F); - ExecutionContext &StackFrame = ECStack.back(); // Fill it in... + // Make a new stack frame... and fill it in. + ECStack.push_back(ExecutionContext()); + ExecutionContext &StackFrame = ECStack.back(); StackFrame.CurFunction = F; StackFrame.CurBB = F->begin(); StackFrame.CurInst = StackFrame.CurBB->begin(); diff --git a/lib/ExecutionEngine/Interpreter/ExecutionAnnotations.h b/lib/ExecutionEngine/Interpreter/ExecutionAnnotations.h index e17aa4b3552..cca1233ea3e 100644 --- a/lib/ExecutionEngine/Interpreter/ExecutionAnnotations.h +++ b/lib/ExecutionEngine/Interpreter/ExecutionAnnotations.h @@ -15,24 +15,10 @@ // information about the function, including the number of types present in the // function, and the number of values for each type. // -// This annotation object is created on demand, and attaches other annotation -// objects to the instructions in the function when it's created. -// -static AnnotationID FunctionInfoAID( - AnnotationManager::getID("Interpreter::FunctionInfo")); - -struct FunctionInfo : public Annotation { +struct FunctionInfo { FunctionInfo(Function *F); std::vector NumPlaneElements; - // Create - Factory function to allow FunctionInfo annotations to be - // created on demand. - // - static Annotation *Create(AnnotationID AID, const Annotable *O, void *) { - assert(AID == FunctionInfoAID); - return new FunctionInfo(cast((Value*)O)); - } - private: unsigned getValueSlot(const Value *V); }; diff --git a/lib/ExecutionEngine/Interpreter/Interpreter.h b/lib/ExecutionEngine/Interpreter/Interpreter.h index 26ed2dc6d15..61d391a360a 100644 --- a/lib/ExecutionEngine/Interpreter/Interpreter.h +++ b/lib/ExecutionEngine/Interpreter/Interpreter.h @@ -82,6 +82,8 @@ class Interpreter : public ExecutionEngine, public InstVisitor { // AtExitHandlers - List of functions to call when the program exits, // registered with the atexit() library function. std::vector AtExitHandlers; + + std::map FunctionInfoMap; public: Interpreter(Module *M, bool isLittleEndian, bool isLongPointer, bool TraceMode);