package com.oracle.truffle.tck.impl;
import com.oracle.truffle.api.CallTarget;
import com.oracle.truffle.api.CompilerDirectives;
import com.oracle.truffle.api.CompilerDirectives.TruffleBoundary;
import com.oracle.truffle.api.Truffle;
import com.oracle.truffle.api.TruffleLanguage;
import com.oracle.truffle.api.TruffleLanguage.Env;
import com.oracle.truffle.api.frame.VirtualFrame;
import com.oracle.truffle.api.interop.InteropLibrary;
import com.oracle.truffle.api.interop.TruffleObject;
import com.oracle.truffle.api.library.ExportLibrary;
import com.oracle.truffle.api.library.ExportMessage;
import com.oracle.truffle.api.library.ExportMessage.Ignore;
import com.oracle.truffle.api.nodes.RootNode;
import com.oracle.truffle.api.source.Source;
@TruffleLanguage.Registration(characterMimeTypes = "application/x-tck", id = "TCK", name = "TCK", version = "1.0")
public final class TckLanguage extends TruffleLanguage<Env> {
@Override
protected Env createContext(Env env) {
return env;
}
@Override
@SuppressWarnings("deprecation")
protected CallTarget parse(ParsingRequest request) throws Exception {
Source code = request.getSource();
final RootNode root;
final String txt = code.getCharacters().toString();
if (txt.startsWith("TCK42:")) {
int nextColon = txt.indexOf(":", 6);
String mimeType = txt.substring(6, nextColon);
Source toParse = Source.newBuilder(code.getLanguage(), txt.substring(nextColon + 1), "src.tck").mimeType(mimeType).build();
root = new MultiplyNode(this, toParse);
} else {
final double value = Double.parseDouble(txt);
root = RootNode.createConstantNode(value);
}
return Truffle.getRuntime().createCallTarget(root);
}
@ExportLibrary(InteropLibrary.class)
static final class MultiplyNode extends RootNode implements TruffleObject {
private final Source code;
MultiplyNode(TckLanguage language, Source toParse) {
super(language);
this.code = toParse;
}
@Override
@Ignore
public Object execute(VirtualFrame frame) {
Env env = lookupContextReference(TckLanguage.class).get();
Object[] arguments = frame.getArguments();
return parseAndEval(env, arguments);
}
@TruffleBoundary
private Object parseAndEval(Env env, Object[] arguments) {
if (arguments.length == 0) {
return this;
}
CallTarget call;
try {
call = env.parsePublic(code, (String) arguments[1], (String) arguments[2]);
} catch (Exception ex) {
throw new AssertionError("Cannot parse " + code, ex);
}
return call.call(6, 7);
}
@SuppressWarnings("static-method")
@ExportMessage
boolean isExecutable() {
return true;
}
@ExportMessage
Object execute(Object[] arguments) {
return execute(Truffle.getRuntime().createVirtualFrame(arguments, getFrameDescriptor()));
}
}
public static Number expectNumber(Object o) {
if (o instanceof Number) {
return (Number) o;
}
CompilerDirectives.transferToInterpreter();
throw new IllegalArgumentException(o + " not a Number");
}
public static String expectString(Object o) {
if (o instanceof String) {
return (String) o;
}
CompilerDirectives.transferToInterpreter();
throw new IllegalArgumentException(o + " not a String");
}
public static TruffleObject expectTruffleObject(Object o) {
if (o instanceof TruffleObject) {
return (TruffleObject) o;
}
CompilerDirectives.transferToInterpreter();
throw new IllegalArgumentException(o + " not a TruffleObject");
}
public static int checkBounds(int idx, int size) {
if (idx < 0 || idx >= size) {
CompilerDirectives.transferToInterpreter();
throw new IndexOutOfBoundsException("Index: " + idx + " Size: " + size);
}
return idx;
}
}