Mercurial > hg > graal-compiler
view graal/com.oracle.truffle.api.test/src/com/oracle/truffle/api/test/ReturnTypeSpecializationTest.java @ 19785:e773cc48d3e8
Truffle-DSL: make specialization types mergable with other specialization types.
author | Christian Humer <christian.humer@gmail.com> |
---|---|
date | Wed, 11 Mar 2015 11:30:20 +0100 |
parents | 94f16a759646 |
children |
line wrap: on
line source
/* * Copyright (c) 2012, 2013, Oracle and/or its affiliates. All rights reserved. * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. * * This code is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License version 2 only, as * published by the Free Software Foundation. * * This code is distributed in the hope that it will be useful, but WITHOUT * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License * version 2 for more details (a copy is included in the LICENSE file that * accompanied this code). * * You should have received a copy of the GNU General Public License version * 2 along with this work; if not, write to the Free Software Foundation, * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. * * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA * or visit www.oracle.com if you need additional information or have any * questions. */ package com.oracle.truffle.api.test; import org.junit.*; import com.oracle.truffle.api.*; import com.oracle.truffle.api.frame.*; import com.oracle.truffle.api.nodes.*; /** * <h3>Specializing Return Types</h3> * * <p> * In order to avoid boxing and/or type casts on the return value of a node, the return value the * method for executing a node can have a specific type and need not be of type * {@link java.lang.Object}. For dynamically typed languages, this return type is something that * should be speculated on. When the speculation fails and the child node cannot return the * appropriate type of value, it can use an {@link UnexpectedResultException} to still pass the * result to the caller. In such a case, the caller must rewrite itself to a more general version in * order to avoid future failures of this kind. * </p> */ public class ReturnTypeSpecializationTest { @Test public void test() { TruffleRuntime runtime = Truffle.getRuntime(); FrameDescriptor frameDescriptor = new FrameDescriptor(); FrameSlot slot = frameDescriptor.addFrameSlot("localVar", FrameSlotKind.Int); TestRootNode rootNode = new TestRootNode(frameDescriptor, new IntAssignLocal(slot, new StringTestChildNode()), new IntReadLocal(slot)); CallTarget target = runtime.createCallTarget(rootNode); Assert.assertEquals(FrameSlotKind.Int, slot.getKind()); Object result = target.call(); Assert.assertEquals("42", result); Assert.assertEquals(FrameSlotKind.Object, slot.getKind()); } class TestRootNode extends RootNode { @Child TestChildNode left; @Child TestChildNode right; public TestRootNode(FrameDescriptor descriptor, TestChildNode left, TestChildNode right) { super(null, descriptor); this.left = left; this.right = right; } @Override public Object execute(VirtualFrame frame) { left.execute(frame); return right.execute(frame); } } abstract class TestChildNode extends Node { public TestChildNode() { super(null); } abstract Object execute(VirtualFrame frame); int executeInt(VirtualFrame frame) throws UnexpectedResultException { Object result = execute(frame); if (result instanceof Integer) { return (Integer) result; } throw new UnexpectedResultException(result); } } abstract class FrameSlotNode extends TestChildNode { protected final FrameSlot slot; public FrameSlotNode(FrameSlot slot) { this.slot = slot; } } class StringTestChildNode extends TestChildNode { @Override Object execute(VirtualFrame frame) { return "42"; } } class IntAssignLocal extends FrameSlotNode { @Child private TestChildNode value; IntAssignLocal(FrameSlot slot, TestChildNode value) { super(slot); this.value = value; } @Override Object execute(VirtualFrame frame) { try { int result = value.executeInt(frame); frame.setInt(slot, result); } catch (UnexpectedResultException e) { slot.setKind(FrameSlotKind.Object); frame.setObject(slot, e.getResult()); replace(new ObjectAssignLocal(slot, value)); } return null; } } class ObjectAssignLocal extends FrameSlotNode { @Child private TestChildNode value; ObjectAssignLocal(FrameSlot slot, TestChildNode value) { super(slot); this.value = value; } @Override Object execute(VirtualFrame frame) { Object o = value.execute(frame); slot.setKind(FrameSlotKind.Object); frame.setObject(slot, o); return null; } } class IntReadLocal extends FrameSlotNode { IntReadLocal(FrameSlot slot) { super(slot); } @Override Object execute(VirtualFrame frame) { try { return frame.getInt(slot); } catch (FrameSlotTypeException e) { return replace(new ObjectReadLocal(slot)).execute(frame); } } @Override int executeInt(VirtualFrame frame) throws UnexpectedResultException { try { return frame.getInt(slot); } catch (FrameSlotTypeException e) { return replace(new ObjectReadLocal(slot)).executeInt(frame); } } } class ObjectReadLocal extends FrameSlotNode { ObjectReadLocal(FrameSlot slot) { super(slot); } @Override Object execute(VirtualFrame frame) { try { return frame.getObject(slot); } catch (FrameSlotTypeException e) { throw new IllegalStateException(e); } } } }