--- /dev/null 2017-01-22 10:16:57.869617664 -0800 +++ new/src/jdk.internal.vm.compiler/share/classes/org.graalvm.compiler.replacements.verifier/src/org/graalvm/compiler/replacements/verifier/GeneratedPlugin.java 2017-02-15 17:09:01.051867289 -0800 @@ -0,0 +1,244 @@ +/* + * Copyright (c) 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. + * + * 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 org.graalvm.compiler.replacements.verifier; + +import java.io.PrintWriter; +import java.util.Set; + +import javax.annotation.processing.ProcessingEnvironment; +import javax.lang.model.element.ExecutableElement; +import javax.lang.model.element.Modifier; +import javax.lang.model.element.TypeElement; +import javax.lang.model.element.VariableElement; +import javax.lang.model.type.ArrayType; +import javax.lang.model.type.DeclaredType; +import javax.lang.model.type.TypeMirror; +import javax.lang.model.type.TypeVariable; +import javax.lang.model.type.WildcardType; + +import org.graalvm.compiler.replacements.verifier.InjectedDependencies.Dependency; +import org.graalvm.compiler.replacements.verifier.InjectedDependencies.WellKnownDependency; + +public abstract class GeneratedPlugin { + + protected final ExecutableElement intrinsicMethod; + private boolean needInjectionProvider; + + private String pluginName; + + public GeneratedPlugin(ExecutableElement intrinsicMethod) { + this.intrinsicMethod = intrinsicMethod; + this.needInjectionProvider = false; + this.pluginName = intrinsicMethod.getEnclosingElement().getSimpleName() + "_" + intrinsicMethod.getSimpleName(); + } + + public String getPluginName() { + return pluginName; + } + + public void setPluginName(String pluginName) { + this.pluginName = pluginName; + } + + public void generate(ProcessingEnvironment env, PrintWriter out) { + out.printf(" // class: %s\n", intrinsicMethod.getEnclosingElement()); + out.printf(" // method: %s\n", intrinsicMethod); + out.printf(" // generated-by: %s\n", getClass().getName()); + out.printf(" private static final class %s extends GeneratedInvocationPlugin {\n", pluginName); + out.printf("\n"); + out.printf(" @Override\n"); + out.printf(" public boolean execute(GraphBuilderContext b, ResolvedJavaMethod targetMethod, InvocationPlugin.Receiver receiver, ValueNode[] args) {\n"); + InjectedDependencies deps = createExecute(env, out); + out.printf(" }\n"); + + createPrivateMembers(out, deps); + + out.printf(" }\n"); + } + + public void register(PrintWriter out) { + out.printf(" plugins.register(new %s(", pluginName); + if (needInjectionProvider) { + out.printf("injection"); + } + out.printf("), %s.class, \"%s\"", intrinsicMethod.getEnclosingElement(), intrinsicMethod.getSimpleName()); + if (!intrinsicMethod.getModifiers().contains(Modifier.STATIC)) { + out.printf(", InvocationPlugin.Receiver.class"); + } + for (VariableElement arg : intrinsicMethod.getParameters()) { + out.printf(", %s.class", getErasedType(arg.asType())); + } + out.printf(");\n"); + } + + public abstract void extraImports(Set imports); + + protected abstract InjectedDependencies createExecute(ProcessingEnvironment env, PrintWriter out); + + private static TypeMirror resolvedJavaTypeType(ProcessingEnvironment env) { + return env.getElementUtils().getTypeElement("jdk.vm.ci.meta.ResolvedJavaType").asType(); + } + + static String getErasedType(TypeMirror type) { + switch (type.getKind()) { + case DECLARED: + DeclaredType declared = (DeclaredType) type; + TypeElement element = (TypeElement) declared.asElement(); + return element.getQualifiedName().toString(); + case TYPEVAR: + return getErasedType(((TypeVariable) type).getUpperBound()); + case WILDCARD: + return getErasedType(((WildcardType) type).getExtendsBound()); + case ARRAY: + return getErasedType(((ArrayType) type).getComponentType()) + "[]"; + default: + return type.toString(); + } + } + + static boolean hasRawtypeWarning(TypeMirror type) { + switch (type.getKind()) { + case DECLARED: + DeclaredType declared = (DeclaredType) type; + return declared.getTypeArguments().size() > 0; + case TYPEVAR: + return false; + case WILDCARD: + return false; + case ARRAY: + return hasRawtypeWarning(((ArrayType) type).getComponentType()); + default: + return false; + } + } + + static boolean hasUncheckedWarning(TypeMirror type) { + switch (type.getKind()) { + case DECLARED: + DeclaredType declared = (DeclaredType) type; + for (TypeMirror typeParam : declared.getTypeArguments()) { + if (hasUncheckedWarning(typeParam)) { + return true; + } + } + return false; + case TYPEVAR: + return true; + case WILDCARD: + return ((WildcardType) type).getExtendsBound() != null; + case ARRAY: + return hasUncheckedWarning(((ArrayType) type).getComponentType()); + default: + return false; + } + } + + private void createPrivateMembers(PrintWriter out, InjectedDependencies deps) { + if (!deps.isEmpty()) { + out.printf("\n"); + for (Dependency dep : deps) { + out.printf(" private final %s %s;\n", dep.type, dep.name); + } + + out.printf("\n"); + out.printf(" private %s(InjectionProvider injection) {\n", pluginName); + for (Dependency dep : deps) { + out.printf(" this.%s = %s;\n", dep.name, dep.inject(intrinsicMethod)); + } + out.printf(" }\n"); + + needInjectionProvider = true; + } + } + + protected static String getReturnKind(ExecutableElement method) { + switch (method.getReturnType().getKind()) { + case BOOLEAN: + case BYTE: + case SHORT: + case CHAR: + case INT: + return "Int"; + case LONG: + return "Long"; + case FLOAT: + return "Float"; + case DOUBLE: + return "Double"; + case VOID: + return "Void"; + case ARRAY: + case TYPEVAR: + case DECLARED: + return "Object"; + default: + throw new IllegalArgumentException(method.getReturnType().toString()); + } + } + + protected static void constantArgument(ProcessingEnvironment env, PrintWriter out, InjectedDependencies deps, int argIdx, TypeMirror type, int nodeIdx) { + if (hasRawtypeWarning(type)) { + out.printf(" @SuppressWarnings({\"rawtypes\"})\n"); + } + out.printf(" %s arg%d;\n", getErasedType(type), argIdx); + out.printf(" if (args[%d].isConstant()) {\n", nodeIdx); + if (type.equals(resolvedJavaTypeType(env))) { + out.printf(" arg%d = %s.asJavaType(args[%d].asConstant());\n", argIdx, deps.use(WellKnownDependency.CONSTANT_REFLECTION), nodeIdx); + } else { + switch (type.getKind()) { + case BOOLEAN: + out.printf(" arg%d = args[%d].asJavaConstant().asInt() != 0;\n", argIdx, nodeIdx); + break; + case BYTE: + out.printf(" arg%d = (byte) args[%d].asJavaConstant().asInt();\n", argIdx, nodeIdx); + break; + case CHAR: + out.printf(" arg%d = (char) args[%d].asJavaConstant().asInt();\n", argIdx, nodeIdx); + break; + case SHORT: + out.printf(" arg%d = (short) args[%d].asJavaConstant().asInt();\n", argIdx, nodeIdx); + break; + case INT: + out.printf(" arg%d = args[%d].asJavaConstant().asInt();\n", argIdx, nodeIdx); + break; + case LONG: + out.printf(" arg%d = args[%d].asJavaConstant().asLong();\n", argIdx, nodeIdx); + break; + case FLOAT: + out.printf(" arg%d = args[%d].asJavaConstant().asFloat();\n", argIdx, nodeIdx); + break; + case DOUBLE: + out.printf(" arg%d = args[%d].asJavaConstant().asDouble();\n", argIdx, nodeIdx); + break; + case DECLARED: + out.printf(" arg%d = %s.asObject(%s.class, args[%d].asJavaConstant());\n", argIdx, deps.use(WellKnownDependency.SNIPPET_REFLECTION), getErasedType(type), nodeIdx); + break; + default: + throw new IllegalArgumentException(); + } + } + out.printf(" } else {\n"); + out.printf(" return false;\n"); + out.printf(" }\n"); + } +}