Mercurial > hg > truffle
view graal/com.oracle.truffle.api.test/src/com/oracle/truffle/api/test/FrameSlotTypeSpecializationTest.java @ 7657:a103e28ce09a
Small clean up.
author | Thomas Wuerthinger <thomas.wuerthinger@oracle.com> |
---|---|
date | Fri, 01 Feb 2013 19:46:52 +0100 |
parents | 5e3d1a68664e |
children | 07f8d136a05e |
line wrap: on
line source
/* * Copyright (c) 2012, 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 Frame Slot Types</h3> * * <p> * Dynamically typed languages can speculate on the type of a frame slot and only fall back at run * time to a more generic type if necessary. The new type of a frame slot can be set using the * {@link FrameSlot#setType(Class)} method. It is the responsibility of the language implementor to * update the content of currently active frames (using {@link Frame#updateToLatestVersion()}). * Also, nodes that depend a specific type of a frame slot must be replaced. Such node can register * a listener that implements {@link FrameSlotTypeListener} using * {@link FrameSlot#registerOneShotTypeListener(FrameSlotTypeListener)}. The event of a type change * on the frame slot will fire only once for the next upcoming change. * </p> * * <p> * The next part of the Truffle API introduction is at * {@link com.oracle.truffle.api.test.ReturnTypeSpecializationTest}. * </p> */ public class FrameSlotTypeSpecializationTest { @Test public void test() { TruffleRuntime runtime = Truffle.getRuntime(); FrameDescriptor frameDescriptor = new FrameDescriptor(); FrameSlot slot = frameDescriptor.addFrameSlot("localVar", Integer.class); TestRootNode rootNode = new TestRootNode(new IntAssignLocal(slot, new StringTestChildNode()), new IntReadLocal(slot)); CallTarget target = runtime.createCallTarget(rootNode, frameDescriptor); Assert.assertEquals(Integer.class, slot.getType()); Object result = target.call(); Assert.assertEquals("42", result); Assert.assertEquals(Object.class, slot.getType()); } class TestRootNode extends RootNode { @Child TestChildNode left; @Child TestChildNode right; public TestRootNode(TestChildNode left, TestChildNode right) { this.left = adoptChild(left); this.right = adoptChild(right); } @Override public Object execute(VirtualFrame frame) { left.execute(frame); return right.execute(frame); } } abstract class TestChildNode extends Node { abstract Object execute(VirtualFrame frame); } 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 implements FrameSlotTypeListener { @Child private TestChildNode value; IntAssignLocal(FrameSlot slot, TestChildNode value) { super(slot); this.value = adoptChild(value); slot.registerOneShotTypeListener(this); } @Override Object execute(VirtualFrame frame) { Object o = value.execute(frame); if (o instanceof Integer) { frame.setInt(slot, (Integer) o); } else { slot.setType(Object.class); frame.updateToLatestVersion(); frame.setObject(slot, o); } return null; } @Override public void typeChanged(FrameSlot changedSlot, Class<?> oldType) { if (changedSlot.getType() == Object.class) { this.replace(new ObjectAssignLocal(changedSlot, value)); } } } class ObjectAssignLocal extends FrameSlotNode { @Child private TestChildNode value; ObjectAssignLocal(FrameSlot slot, TestChildNode value) { super(slot); this.value = adoptChild(value); } @Override Object execute(VirtualFrame frame) { Object o = value.execute(frame); frame.setObject(slot, o); return null; } } class IntReadLocal extends FrameSlotNode implements FrameSlotTypeListener { IntReadLocal(FrameSlot slot) { super(slot); slot.registerOneShotTypeListener(this); } @Override Object execute(VirtualFrame frame) { return frame.getInt(slot); } @Override public void typeChanged(FrameSlot changedSlot, Class<?> oldType) { if (changedSlot.getType() == Object.class) { this.replace(new ObjectReadLocal(changedSlot)); } } } class ObjectReadLocal extends FrameSlotNode { ObjectReadLocal(FrameSlot slot) { super(slot); } @Override Object execute(VirtualFrame frame) { return frame.getObject(slot); } } }