Mercurial > hg > truffle
diff truffle/com.oracle.truffle.tck/src/com/oracle/truffle/tck/ComplexNumbersB.java @ 22495:aeba89e1d8da
Add ComplexNumber sequence tests
author | Matthias Grimmer <grimmer@ssw.jku.at> |
---|---|
date | Fri, 11 Dec 2015 15:20:27 +0100 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/truffle/com.oracle.truffle.tck/src/com/oracle/truffle/tck/ComplexNumbersB.java Fri Dec 11 15:20:27 2015 +0100 @@ -0,0 +1,229 @@ +/* + * Copyright (c) 2015, 2015, 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. Oracle designates this + * particular file as subject to the "Classpath" exception as provided + * by Oracle in the LICENSE file that accompanied this code. + * + * 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.tck; + +import com.oracle.truffle.api.CallTarget; +import com.oracle.truffle.api.CompilerDirectives; +import com.oracle.truffle.api.Truffle; +import com.oracle.truffle.api.frame.VirtualFrame; +import com.oracle.truffle.api.interop.ForeignAccess; +import com.oracle.truffle.api.interop.ForeignAccess.Factory; +import com.oracle.truffle.api.interop.Message; +import com.oracle.truffle.api.interop.TruffleObject; +import com.oracle.truffle.api.nodes.Node; +import com.oracle.truffle.api.nodes.RootNode; + +public class ComplexNumbersB implements TruffleObject { + + private final double[] reals; + private final double[] imags; + + public ComplexNumbersB(double[] reals, double[] imags) { + assert reals.length == imags.length; + this.reals = reals; + this.imags = imags; + } + + public double[] getData() { + double[] data = new double[reals.length * 2]; + for (int i = 0; i < reals.length; i++) { + data[i * 2] = reals[i]; + data[i * 2 + 1] = imags[i]; + } + return data; + } + + public ForeignAccess getForeignAccess() { + return ForeignAccess.create(new ComplexNumbersBForeignAccessFactory()); + } + + private static class ComplexNumbersBForeignAccessFactory implements Factory { + + public boolean canHandle(TruffleObject obj) { + return obj instanceof ComplexNumbersB; + } + + public CallTarget accessMessage(Message tree) { + if (Message.IS_NULL.equals(tree)) { + return Truffle.getRuntime().createCallTarget(RootNode.createConstantNode(false)); + } else if (Message.IS_EXECUTABLE.equals(tree)) { + return Truffle.getRuntime().createCallTarget(RootNode.createConstantNode(false)); + } else if (Message.IS_BOXED.equals(tree)) { + return Truffle.getRuntime().createCallTarget(RootNode.createConstantNode(false)); + } else if (Message.HAS_SIZE.equals(tree)) { + return Truffle.getRuntime().createCallTarget(RootNode.createConstantNode(true)); + } else if (Message.READ.equals(tree)) { + return Truffle.getRuntime().createCallTarget(new ComplexNumbersBReadNode()); + } else if (Message.WRITE.equals(tree)) { + return Truffle.getRuntime().createCallTarget(new ComplexNumbersBWriteNode()); + } else if (Message.GET_SIZE.equals(tree)) { + return Truffle.getRuntime().createCallTarget(new ComplexNumbersBSizeNode()); + } else { + throw new IllegalArgumentException(tree.toString() + " not supported"); + } + } + } + + private static class ComplexNumbersBWriteNode extends RootNode { + protected ComplexNumbersBWriteNode() { + super(TckLanguage.class, null, null); + } + + @Child private Node readReal; + @Child private Node readImag; + + @Override + public Object execute(VirtualFrame frame) { + ComplexNumbersB complexNumbers = (ComplexNumbersB) ForeignAccess.getReceiver(frame); + Number index = (Number) ForeignAccess.getArguments(frame).get(0); + TruffleObject value = (TruffleObject) ForeignAccess.getArguments(frame).get(1); + if (readReal == null || readImag == null) { + CompilerDirectives.transferToInterpreterAndInvalidate(); + this.readReal = insert(Message.READ.createNode()); + this.readImag = insert(Message.READ.createNode()); + } + Number realPart = (Number) ForeignAccess.execute(readReal, frame, value, new Object[]{ComplexNumber.REAL_IDENTIFIER}); + Number imagPart = (Number) ForeignAccess.execute(readImag, frame, value, new Object[]{ComplexNumber.IMAGINARY_IDENTIFIER}); + + complexNumbers.reals[index.intValue()] = realPart.doubleValue(); + complexNumbers.imags[index.intValue()] = imagPart.doubleValue(); + return value; + } + } + + private static class ComplexNumbersBReadNode extends RootNode { + protected ComplexNumbersBReadNode() { + super(TckLanguage.class, null, null); + } + + @Override + public Object execute(VirtualFrame frame) { + ComplexNumbersB complexNumbers = (ComplexNumbersB) ForeignAccess.getReceiver(frame); + Number index = (Number) ForeignAccess.getArguments(frame).get(0); + return new ComplexNumberBEntry(complexNumbers, index.intValue()); + } + + } + + private static class ComplexNumbersBSizeNode extends RootNode { + protected ComplexNumbersBSizeNode() { + super(TckLanguage.class, null, null); + } + + @Override + public Object execute(VirtualFrame frame) { + ComplexNumbersB complexNumbers = (ComplexNumbersB) ForeignAccess.getReceiver(frame); + assert complexNumbers.reals.length == complexNumbers.imags.length; + return complexNumbers.reals.length; + } + + } + + private static class ComplexNumberBEntry implements TruffleObject { + + private final ComplexNumbersB numbers; + private final int index; + + public ComplexNumberBEntry(ComplexNumbersB numbers, int index) { + this.numbers = numbers; + this.index = index; + } + + public ForeignAccess getForeignAccess() { + return ForeignAccess.create(new ComplexNumberBEntryForeignAccessFactory()); + } + + private static class ComplexNumberBEntryForeignAccessFactory implements Factory { + + public boolean canHandle(TruffleObject obj) { + return obj instanceof ComplexNumberBEntry; + } + + public CallTarget accessMessage(Message tree) { + if (Message.IS_NULL.equals(tree)) { + return Truffle.getRuntime().createCallTarget(RootNode.createConstantNode(false)); + } else if (Message.IS_EXECUTABLE.equals(tree)) { + return Truffle.getRuntime().createCallTarget(RootNode.createConstantNode(false)); + } else if (Message.IS_BOXED.equals(tree)) { + return Truffle.getRuntime().createCallTarget(RootNode.createConstantNode(false)); + } else if (Message.HAS_SIZE.equals(tree)) { + return Truffle.getRuntime().createCallTarget(RootNode.createConstantNode(false)); + } else if (Message.READ.equals(tree)) { + return Truffle.getRuntime().createCallTarget(new ComplexNumbersAEntryReadNode()); + } else if (Message.WRITE.equals(tree)) { + return Truffle.getRuntime().createCallTarget(new ComplexNumbersAEntryWriteNode()); + } else { + throw new IllegalArgumentException(tree.toString() + " not supported"); + } + } + + private static class ComplexNumbersAEntryReadNode extends RootNode { + protected ComplexNumbersAEntryReadNode() { + super(TckLanguage.class, null, null); + } + + @Child private Node readReal; + @Child private Node readImag; + + @Override + public Object execute(VirtualFrame frame) { + ComplexNumberBEntry complexNumber = (ComplexNumberBEntry) ForeignAccess.getReceiver(frame); + String name = (String) ForeignAccess.getArguments(frame).get(0); + if (name.equals(ComplexNumber.IMAGINARY_IDENTIFIER)) { + return complexNumber.numbers.imags[complexNumber.index]; + } else if (name.equals(ComplexNumber.REAL_IDENTIFIER)) { + return complexNumber.numbers.reals[complexNumber.index]; + } else { + throw new IllegalArgumentException(); + } + } + } + + private static class ComplexNumbersAEntryWriteNode extends RootNode { + protected ComplexNumbersAEntryWriteNode() { + super(TckLanguage.class, null, null); + } + + @Override + public Object execute(VirtualFrame frame) { + ComplexNumberBEntry complexNumber = (ComplexNumberBEntry) ForeignAccess.getReceiver(frame); + String name = (String) ForeignAccess.getArguments(frame).get(0); + Number value = (Number) ForeignAccess.getArguments(frame).get(1); + if (name.equals(ComplexNumber.IMAGINARY_IDENTIFIER)) { + complexNumber.numbers.imags[complexNumber.index] = value.doubleValue(); + } else if (name.equals(ComplexNumber.REAL_IDENTIFIER)) { + complexNumber.numbers.reals[complexNumber.index] = value.doubleValue(); + } else { + throw new IllegalArgumentException(); + } + return value; + } + + } + } + + } + +}