|
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;
|
|
|
11
|
|
|
12 import com.oracle.truffle.api.*;
|
|
|
13 import com.oracle.truffle.api.frame.*;
|
|
|
14 import com.oracle.truffle.api.utilities.*;
|
|
|
15 import com.oracle.truffle.ruby.nodes.*;
|
|
|
16 import com.oracle.truffle.ruby.runtime.*;
|
|
|
17 import com.oracle.truffle.ruby.runtime.control.*;
|
|
|
18
|
|
|
19 /**
|
|
|
20 * Catch a {@code return} jump at the root of a method.
|
|
|
21 */
|
|
|
22 public class CatchReturnNode extends RubyNode {
|
|
|
23
|
|
|
24 @Child protected RubyNode body;
|
|
|
25 private final long returnID;
|
|
|
26
|
|
|
27 private final BranchProfile returnProfile = new BranchProfile();
|
|
|
28 private final BranchProfile returnToOtherMethodProfile = new BranchProfile();
|
|
|
29
|
|
|
30 public CatchReturnNode(RubyContext context, SourceSection sourceSection, RubyNode body, long returnID) {
|
|
|
31 super(context, sourceSection);
|
|
|
32 this.body = adoptChild(body);
|
|
|
33 this.returnID = returnID;
|
|
|
34 }
|
|
|
35
|
|
|
36 public CatchReturnNode(CatchReturnNode prev) {
|
|
|
37 super(prev);
|
|
|
38 returnID = prev.returnID;
|
|
|
39 }
|
|
|
40
|
|
|
41 @Override
|
|
|
42 public Object execute(VirtualFrame frame) {
|
|
|
43 try {
|
|
|
44 return body.execute(frame);
|
|
|
45 } catch (ReturnException e) {
|
|
|
46 returnProfile.enter();
|
|
|
47
|
|
|
48 if (e.getReturnID() == returnID) {
|
|
|
49 return e.getValue();
|
|
|
50 } else {
|
|
|
51 returnToOtherMethodProfile.enter();
|
|
|
52 throw e;
|
|
|
53 }
|
|
|
54 }
|
|
|
55 }
|
|
|
56
|
|
|
57 }
|
