org.jruby.ast.WhenOneArgNode Maven / Gradle / Ivy

Go to download
/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */

package org.jruby.ast;

import org.jruby.Ruby;
import org.jruby.ast.types.IEqlNode;
import org.jruby.lexer.yacc.ISourcePosition;
import org.jruby.runtime.Block;
import org.jruby.runtime.ThreadContext;
import org.jruby.runtime.builtin.IRubyObject;

/**
 *
 * @author enebo
 */
public class WhenOneArgNode extends WhenNode {
    public WhenOneArgNode(ISourcePosition position, Node expressionNode, Node bodyNode, Node nextCase) {
        super(position, expressionNode, bodyNode, nextCase);
    }

    // FIXME: Can get optimized for IEqlNode
    private IRubyObject whenNoTest(ThreadContext context, Ruby runtime, IRubyObject self, Block aBlock) {
        if (expressionNodes.interpret(runtime, context, self, aBlock).isTrue()) {
            return bodyNode.interpret(runtime, context, self, aBlock);
        }

        return null;
    }

    private IRubyObject whenSlowTest(IRubyObject test, ThreadContext context, Ruby runtime, IRubyObject self, Block aBlock) {
        IRubyObject expression = expressionNodes.interpret(runtime, context, self, aBlock);
        if (eqq.call(context, self, expression, test).isTrue()) {
                return bodyNode.interpret(runtime, context, self, aBlock);
        }

        return null;
    }

    @Override
    public IRubyObject when(IRubyObject test, ThreadContext context, Ruby runtime, IRubyObject self, Block aBlock) {
        // No actual test, so do 'when' if when expression is not nil
        if (test == null) return whenNoTest(context, runtime, self, aBlock);
        if (!(expressionNodes instanceof IEqlNode)) return whenSlowTest(test, context, runtime, self, aBlock);

        if (((IEqlNode) expressionNodes).eql(test, context, runtime, self, aBlock)) {
            return bodyNode.interpret(runtime, context, self, aBlock);
        }

        return null;
    }
}