Mercurial > hg > graal-compiler
diff graal/com.oracle.truffle.ruby.nodes/src/com/oracle/truffle/ruby/nodes/methods/AddMethodNode.java @ 13529:856c2c294f84
Merge.
author | Christian Humer <christian.humer@gmail.com> |
---|---|
date | Tue, 07 Jan 2014 18:53:04 +0100 |
parents | 0fbee3eb71f0 |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/graal/com.oracle.truffle.ruby.nodes/src/com/oracle/truffle/ruby/nodes/methods/AddMethodNode.java Tue Jan 07 18:53:04 2014 +0100 @@ -0,0 +1,72 @@ +/* + * Copyright (c) 2013 Oracle and/or its affiliates. All rights reserved. This + * code is released under a tri EPL/GPL/LGPL license. You can use it, + * redistribute it and/or modify it under the terms of the: + * + * Eclipse Public License version 1.0 + * GNU General Public License version 2 + * GNU Lesser General Public License version 2.1 + */ +package com.oracle.truffle.ruby.nodes.methods; + +import com.oracle.truffle.api.*; +import com.oracle.truffle.api.frame.*; +import com.oracle.truffle.api.nodes.*; +import com.oracle.truffle.ruby.nodes.*; +import com.oracle.truffle.ruby.runtime.*; +import com.oracle.truffle.ruby.runtime.core.*; +import com.oracle.truffle.ruby.runtime.methods.*; + +@NodeInfo(shortName = "add-method") +public class AddMethodNode extends RubyNode { + + @Child protected RubyNode receiver; + @Child protected MethodDefinitionNode method; + + public AddMethodNode(RubyContext context, SourceSection section, RubyNode receiver, MethodDefinitionNode method) { + super(context, section); + this.receiver = adoptChild(receiver); + this.method = adoptChild(method); + } + + @Override + public Object execute(VirtualFrame frame) { + final Object receiverObject = receiver.execute(frame); + + final RubyMethod methodObject = (RubyMethod) method.execute(frame); + + final FrameSlot moduleFunctionFlagSlot = frame.getFrameDescriptor().findFrameSlot(RubyModule.MODULE_FUNCTION_FLAG_FRAME_SLOT_ID); + + boolean moduleFunctionFlag; + + if (moduleFunctionFlagSlot == null) { + moduleFunctionFlag = false; + } else { + Object moduleFunctionObject; + + try { + moduleFunctionObject = frame.getObject(moduleFunctionFlagSlot); + } catch (FrameSlotTypeException e) { + throw new RuntimeException(e); + } + + if (moduleFunctionObject instanceof Boolean) { + moduleFunctionFlag = (boolean) moduleFunctionObject; + } else { + moduleFunctionFlag = false; + } + } + + final RubyModule module = (RubyModule) receiverObject; + + final RubyMethod methodWithDeclaringModule = methodObject.withDeclaringModule(module); + + module.addMethod(methodWithDeclaringModule); + + if (moduleFunctionFlag) { + module.getSingletonClass().addMethod(methodWithDeclaringModule); + } + + return NilPlaceholder.INSTANCE; + } +}