Mercurial > hg > graal-compiler
view graal/com.oracle.graal.replacements.verifier/src/com/oracle/graal/replacements/verifier/FoldPluginGenerator.java @ 23141:7d5a2e297409
Support non-static @Fold methods.
author | Roland Schatz <roland.schatz@oracle.com> |
---|---|
date | Mon, 07 Dec 2015 14:06:22 +0100 |
parents | 28613ae6b05d |
children |
line wrap: on
line source
/* * 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 com.oracle.graal.replacements.verifier; import java.io.PrintWriter; import java.lang.annotation.Annotation; import java.util.List; import javax.annotation.processing.ProcessingEnvironment; import javax.lang.model.element.AnnotationMirror; import javax.lang.model.element.Element; import javax.lang.model.element.ElementKind; 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.TypeMirror; import javax.tools.Diagnostic.Kind; import com.oracle.graal.api.replacements.Fold; import com.oracle.graal.replacements.verifier.InjectedDependencies.WellKnownDependency; /** * Create graph builder plugins for {@link Fold} methods. */ public class FoldPluginGenerator extends PluginGenerator { private class FoldVerifier extends AbstractVerifier { public FoldVerifier(ProcessingEnvironment env) { super(env); } @Override public void verify(Element element, AnnotationMirror annotation) { if (element.getKind() != ElementKind.METHOD) { assert false : "Element is guaranteed to be a method."; return; } ExecutableElement intrinsicMethod = (ExecutableElement) element; if (intrinsicMethod.getModifiers().contains(Modifier.PRIVATE)) { env.getMessager().printMessage(Kind.ERROR, "@Fold method can not be private.", element); } else { FoldPluginGenerator.this.createPluginFactory(intrinsicMethod, null, null); } } @Override public Class<? extends Annotation> getAnnotationClass() { return Fold.class; } } private TypeMirror stringType() { return env.getElementUtils().getTypeElement("java.lang.String").asType(); } public FoldPluginGenerator(ProcessingEnvironment env) { super(env); } public AbstractVerifier getVerifier() { return new FoldVerifier(env); } @Override protected String getBaseName() { return "FoldFactory"; } @Override protected void createImports(PrintWriter out, ExecutableElement intrinsicMethod, ExecutableElement targetMethod) { out.printf("import jdk.vm.ci.meta.JavaConstant;\n"); out.printf("import jdk.vm.ci.meta.JavaKind;\n"); out.printf("import com.oracle.graal.nodes.ConstantNode;\n"); super.createImports(out, intrinsicMethod, targetMethod); } @Override protected InjectedDependencies createExecute(PrintWriter out, ExecutableElement intrinsicMethod, ExecutableElement constructor, TypeMirror[] signature) { InjectedDependencies deps = new InjectedDependencies(); List<? extends VariableElement> params = intrinsicMethod.getParameters(); int argCount = 0; Object receiver; if (intrinsicMethod.getModifiers().contains(Modifier.STATIC)) { receiver = intrinsicMethod.getEnclosingElement(); } else { receiver = "arg0"; TypeElement type = (TypeElement) intrinsicMethod.getEnclosingElement(); constantArgument(out, deps, argCount, type.asType(), argCount); argCount++; } int firstArg = argCount; for (VariableElement param : params) { constantArgument(out, deps, argCount, param.asType(), argCount); argCount++; } if (intrinsicMethod.getAnnotation(Deprecated.class) != null) { out.printf(" @SuppressWarnings(\"deprecation\")\n"); } out.printf(" %s result = %s.%s(", intrinsicMethod.getReturnType(), receiver, intrinsicMethod.getSimpleName()); if (argCount > firstArg) { out.printf("arg%d", firstArg); for (int i = firstArg + 1; i < argCount; i++) { out.printf(", arg%d", i); } } out.printf(");\n"); TypeMirror returnType = intrinsicMethod.getReturnType(); switch (returnType.getKind()) { case BOOLEAN: out.printf(" JavaConstant constant = JavaConstant.forInt(result ? 1 : 0);\n"); break; case BYTE: case SHORT: case CHAR: case INT: out.printf(" JavaConstant constant = JavaConstant.forInt(result);\n"); break; case LONG: out.printf(" JavaConstant constant = JavaConstant.forLong(result);\n"); break; case FLOAT: out.printf(" JavaConstant constant = JavaConstant.forFloat(result);\n"); break; case DOUBLE: out.printf(" JavaConstant constant = JavaConstant.forDouble(result);\n"); break; case ARRAY: case TYPEVAR: case DECLARED: if (returnType.equals(stringType())) { out.printf(" JavaConstant constant = %s.forString(result);\n", deps.use(WellKnownDependency.CONSTANT_REFLECTION)); } else { out.printf(" JavaConstant constant = %s.forObject(result);\n", deps.use(WellKnownDependency.SNIPPET_REFLECTION)); } break; default: throw new IllegalArgumentException(returnType.toString()); } out.printf(" ConstantNode node = ConstantNode.forConstant(constant, %s, %s);\n", deps.use(WellKnownDependency.META_ACCESS), deps.use(WellKnownDependency.STRUCTURED_GRAPH)); out.printf(" b.push(JavaKind.%s, node);\n", getReturnKind(intrinsicMethod).name()); out.printf(" return true;\n"); return deps; } }