Mercurial > hg > graal-compiler
diff graal/com.oracle.truffle.ruby.nodes/src/com/oracle/truffle/ruby/nodes/methods/CatchReturnNode.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/CatchReturnNode.java Tue Jan 07 18:53:04 2014 +0100 @@ -0,0 +1,57 @@ +/* + * 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.utilities.*; +import com.oracle.truffle.ruby.nodes.*; +import com.oracle.truffle.ruby.runtime.*; +import com.oracle.truffle.ruby.runtime.control.*; + +/** + * Catch a {@code return} jump at the root of a method. + */ +public class CatchReturnNode extends RubyNode { + + @Child protected RubyNode body; + private final long returnID; + + private final BranchProfile returnProfile = new BranchProfile(); + private final BranchProfile returnToOtherMethodProfile = new BranchProfile(); + + public CatchReturnNode(RubyContext context, SourceSection sourceSection, RubyNode body, long returnID) { + super(context, sourceSection); + this.body = adoptChild(body); + this.returnID = returnID; + } + + public CatchReturnNode(CatchReturnNode prev) { + super(prev); + returnID = prev.returnID; + } + + @Override + public Object execute(VirtualFrame frame) { + try { + return body.execute(frame); + } catch (ReturnException e) { + returnProfile.enter(); + + if (e.getReturnID() == returnID) { + return e.getValue(); + } else { + returnToOtherMethodProfile.enter(); + throw e; + } + } + } + +}