13514
|
1 /*
|
|
2 * Copyright (c) 2013 Oracle and/or its affiliates. All rights reserved. This
|
|
3 * code is released under a tri EPL/GPL/LGPL license. You can use it,
|
|
4 * redistribute it and/or modify it under the terms of the:
|
|
5 *
|
|
6 * Eclipse Public License version 1.0
|
|
7 * GNU General Public License version 2
|
|
8 * GNU Lesser General Public License version 2.1
|
|
9 */
|
|
10 package com.oracle.truffle.ruby.nodes.methods.arguments;
|
|
11
|
|
12 import com.oracle.truffle.api.*;
|
|
13 import com.oracle.truffle.api.frame.*;
|
|
14 import com.oracle.truffle.api.nodes.*;
|
|
15 import com.oracle.truffle.ruby.nodes.*;
|
|
16 import com.oracle.truffle.ruby.runtime.*;
|
|
17 import com.oracle.truffle.ruby.runtime.core.array.*;
|
|
18
|
|
19 /**
|
|
20 * Switches between loading arguments as normal and doing a destructure. See
|
|
21 * testBlockArgumentsDestructure and MethodTranslator.
|
|
22 */
|
|
23 @NodeInfo(shortName = "block-destructure-switch")
|
|
24 public class BlockDestructureSwitchNode extends RubyNode {
|
|
25
|
|
26 @Child protected RubyNode loadIndividualArguments;
|
|
27 @Child protected RubyNode destructureArguments;
|
|
28 @Child protected RubyNode body;
|
|
29
|
|
30 public BlockDestructureSwitchNode(RubyContext context, SourceSection sourceSection, RubyNode loadIndividualArguments, RubyNode destructureArguments, RubyNode body) {
|
|
31 super(context, sourceSection);
|
|
32 this.loadIndividualArguments = adoptChild(loadIndividualArguments);
|
|
33 this.destructureArguments = adoptChild(destructureArguments);
|
|
34 this.body = adoptChild(body);
|
|
35 }
|
|
36
|
|
37 @Override
|
|
38 public Object execute(VirtualFrame frame) {
|
|
39 final RubyArguments arguments = frame.getArguments(RubyArguments.class);
|
|
40
|
|
41 if (arguments.getArguments().length == 1 && arguments.getArguments()[0] instanceof RubyArray) {
|
|
42 destructureArguments.executeVoid(frame);
|
|
43 } else {
|
|
44 loadIndividualArguments.executeVoid(frame);
|
|
45 }
|
|
46
|
|
47 return body.execute(frame);
|
|
48 }
|
|
49
|
|
50 }
|