Mercurial > hg > graal-compiler
view graal/com.oracle.truffle.dsl.processor/src/com/oracle/truffle/dsl/processor/template/TemplateMethod.java @ 16232:e34bb128f227
Merge.
author | Doug Simon <doug.simon@oracle.com> |
---|---|
date | Wed, 25 Jun 2014 17:34:25 +0200 |
parents | 856c2c294f84 |
children | bd28da642eea |
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.dsl.processor.template; import java.util.*; import javax.lang.model.element.*; import javax.lang.model.type.*; import com.oracle.truffle.dsl.processor.*; import com.oracle.truffle.dsl.processor.typesystem.*; import com.oracle.truffle.dsl.processor.util.*; /** * Note: this class has a natural ordering that is inconsistent with equals. */ public class TemplateMethod extends MessageContainer implements Comparable<TemplateMethod> { private String id; private final Template template; private final MethodSpec specification; private final ExecutableElement method; private final AnnotationMirror markerAnnotation; private ActualParameter returnType; private List<ActualParameter> parameters; public TemplateMethod(String id, Template template, MethodSpec specification, ExecutableElement method, AnnotationMirror markerAnnotation, ActualParameter returnType, List<ActualParameter> parameters) { this.template = template; this.specification = specification; this.method = method; this.markerAnnotation = markerAnnotation; this.returnType = returnType; this.parameters = new ArrayList<>(); for (ActualParameter param : parameters) { ActualParameter newParam = new ActualParameter(param); this.parameters.add(newParam); newParam.setMethod(this); } this.id = id; } public TemplateMethod(TemplateMethod method) { this(method.id, method.template, method.specification, method.method, method.markerAnnotation, method.returnType, method.parameters); getMessages().addAll(method.getMessages()); } public TemplateMethod(TemplateMethod method, ExecutableElement executable) { this(method.id, method.template, method.specification, executable, method.markerAnnotation, method.returnType, method.parameters); getMessages().addAll(method.getMessages()); } public void setParameters(List<ActualParameter> parameters) { this.parameters = parameters; } @Override public Element getMessageElement() { return method; } @Override public AnnotationMirror getMessageAnnotation() { return markerAnnotation; } @Override protected List<MessageContainer> findChildContainers() { return Collections.emptyList(); } public void setId(String id) { this.id = id; } public String getId() { return id; } public Template getTemplate() { return template; } public MethodSpec getSpecification() { return specification; } public ActualParameter getReturnType() { return returnType; } public void replaceParameter(String localName, ActualParameter newParameter) { if (returnType.getLocalName().equals(localName)) { returnType = newParameter; returnType.setMethod(this); } for (ListIterator<ActualParameter> iterator = parameters.listIterator(); iterator.hasNext();) { ActualParameter parameter = iterator.next(); if (parameter.getLocalName().equals(localName)) { iterator.set(newParameter); newParameter.setMethod(this); } } } public List<ActualParameter> getRequiredParameters() { List<ActualParameter> requiredParameters = new ArrayList<>(); for (ActualParameter parameter : getParameters()) { if (getSpecification().getRequired().contains(parameter.getSpecification())) { requiredParameters.add(parameter); } } return requiredParameters; } public Iterable<ActualParameter> getSignatureParameters() { return new FilteredIterable<>(getParameters(), new Predicate<ActualParameter>() { public boolean evaluate(ActualParameter value) { return value.getSpecification().isSignature(); } }); } public List<ActualParameter> getParameters() { return parameters; } public List<ActualParameter> findParameters(ParameterSpec spec) { List<ActualParameter> foundParameters = new ArrayList<>(); for (ActualParameter param : getReturnTypeAndParameters()) { if (param.getSpecification().getName().equals(spec.getName())) { foundParameters.add(param); } } return foundParameters; } public ActualParameter findParameter(String valueName) { for (ActualParameter param : getReturnTypeAndParameters()) { if (param.getLocalName().equals(valueName)) { return param; } } return null; } public List<ActualParameter> getReturnTypeAndParameters() { List<ActualParameter> allParameters = new ArrayList<>(getParameters().size() + 1); if (getReturnType() != null) { allParameters.add(getReturnType()); } allParameters.addAll(getParameters()); return Collections.unmodifiableList(allParameters); } public boolean canBeAccessedByInstanceOf(ProcessorContext context, TypeMirror type) { TypeMirror methodType = Utils.findNearestEnclosingType(getMethod()).asType(); return Utils.isAssignable(context, type, methodType) || Utils.isAssignable(context, methodType, type); } public ExecutableElement getMethod() { return method; } public String getMethodName() { if (getMethod() != null) { return getMethod().getSimpleName().toString(); } else { return "$synthetic"; } } public AnnotationMirror getMarkerAnnotation() { return markerAnnotation; } @Override public String toString() { return String.format("%s [id = %s, method = %s]", getClass().getSimpleName(), getId(), getMethod()); } public ActualParameter getPreviousParam(ActualParameter searchParam) { ActualParameter prev = null; for (ActualParameter param : getParameters()) { if (param == searchParam) { return prev; } prev = param; } return prev; } @SuppressWarnings("unused") public int getSignatureSize() { int signatureSize = 0; for (ActualParameter parameter : getSignatureParameters()) { signatureSize++; } return signatureSize; } public TypeSignature getTypeSignature() { TypeSignature signature = new TypeSignature(); signature.types.add(getReturnType().getTypeSystemType()); for (ActualParameter parameter : getSignatureParameters()) { TypeData typeData = parameter.getTypeSystemType(); if (typeData != null) { signature.types.add(typeData); } } return signature; } public ActualParameter getSignatureParameter(int searchIndex) { int index = 0; for (ActualParameter parameter : getParameters()) { if (!parameter.getSpecification().isSignature()) { continue; } if (index == searchIndex) { return parameter; } index++; } return null; } public void updateSignature(TypeSignature signature) { // TODO(CH): fails in normal usage - output ok though // assert signature.size() >= 1; int signatureIndex = 0; for (ActualParameter parameter : getReturnTypeAndParameters()) { if (!parameter.getSpecification().isSignature()) { continue; } if (signatureIndex >= signature.size()) { break; } TypeData newType = signature.get(signatureIndex++); if (!parameter.getTypeSystemType().equals(newType)) { replaceParameter(parameter.getLocalName(), new ActualParameter(parameter, newType)); } } } @Override public int compareTo(TemplateMethod o) { if (this == o) { return 0; } int compare = compareBySignature(o); if (compare == 0) { // if signature sorting failed sort by id compare = getId().compareTo(o.getId()); } if (compare == 0) { // if still no difference sort by enclosing type name TypeElement enclosingType1 = Utils.findNearestEnclosingType(getMethod()); TypeElement enclosingType2 = Utils.findNearestEnclosingType(o.getMethod()); compare = enclosingType1.getQualifiedName().toString().compareTo(enclosingType2.getQualifiedName().toString()); } return compare; } public List<ActualParameter> getParametersAfter(ActualParameter genericParameter) { boolean found = false; List<ActualParameter> foundParameters = new ArrayList<>(); for (ActualParameter param : getParameters()) { if (param.getLocalName().equals(genericParameter.getLocalName())) { found = true; } else if (found) { foundParameters.add(param); } } return foundParameters; } public int compareBySignature(TemplateMethod compareMethod) { final TypeSystemData typeSystem = getTemplate().getTypeSystem(); if (typeSystem != compareMethod.getTemplate().getTypeSystem()) { throw new IllegalStateException("Cannot compare two methods with different type systems."); } List<TypeMirror> signature1 = getSignatureTypes(this); List<TypeMirror> signature2 = getSignatureTypes(compareMethod); if (signature1.size() != signature2.size()) { return signature2.size() - signature1.size(); } int result = 0; for (int i = 1; i < signature1.size(); i++) { TypeMirror t1 = signature1.get(i); TypeMirror t2 = signature2.get(i); int typeResult = compareParameter(typeSystem, t1, t2); if (result == 0) { result = typeResult; } else if (typeResult != 0 && Math.signum(result) != Math.signum(typeResult)) { // We cannot define an order. return 0; } } if (result == 0 && signature1.size() > 0) { result = compareParameter(typeSystem, signature1.get(0), signature2.get(0)); } return result; } private static int compareParameter(TypeSystemData data, TypeMirror signature1, TypeMirror signature2) { if (Utils.typeEquals(signature1, signature2)) { return 0; } int index1 = data.findType(signature1); int index2 = data.findType(signature2); if (index1 != -1 && index2 != -1) { return index1 - index2; } if (signature1.getKind() == TypeKind.DECLARED && signature2.getKind() == TypeKind.DECLARED) { TypeElement element1 = Utils.fromTypeMirror(signature1); TypeElement element2 = Utils.fromTypeMirror(signature2); if (Utils.getDirectSuperTypes(element1).contains(element2)) { return -1; } else if (Utils.getDirectSuperTypes(element2).contains(element1)) { return 1; } } return Utils.getSimpleName(signature1).compareTo(Utils.getSimpleName(signature2)); } public static List<TypeMirror> getSignatureTypes(TemplateMethod method) { List<TypeMirror> types = new ArrayList<>(); types.add(method.getReturnType().getType()); for (ActualParameter param : method.getSignatureParameters()) { types.add(param.getType()); } return types; } public static class TypeSignature implements Iterable<TypeData>, Comparable<TypeSignature> { final List<TypeData> types; public TypeSignature() { this.types = new ArrayList<>(); } public TypeSignature(List<TypeData> signature) { this.types = signature; } @Override public int hashCode() { return types.hashCode(); } public int size() { return types.size(); } public TypeData get(int index) { return types.get(index); } public int compareTo(TypeSignature other) { if (this == other) { return 0; } else if (types.size() != other.types.size()) { return types.size() - other.types.size(); } else if (types.isEmpty()) { return 0; } for (int i = 0; i < types.size(); i++) { TypeData type1 = types.get(i); TypeData type2 = other.types.get(i); int comparison = type1.compareTo(type2); if (comparison != 0) { return comparison; } } return 0; } public TypeSignature combine(TypeSignature genericSignature, TypeSignature other) { assert types.size() == other.types.size(); assert genericSignature.types.size() == other.types.size(); if (this.equals(other)) { return this; } TypeSignature signature = new TypeSignature(); for (int i = 0; i < types.size(); i++) { TypeData type1 = types.get(i); TypeData type2 = other.types.get(i); if (type1.equals(type2)) { signature.types.add(type1); } else { signature.types.add(genericSignature.types.get(i)); } } return signature; } public boolean equalsParameters(TypeSignature other) { if (size() != other.size()) { return false; } return types.subList(1, types.size()).equals(other.types.subList(1, other.types.size())); } @Override public boolean equals(Object obj) { if (obj instanceof TypeSignature) { return ((TypeSignature) obj).types.equals(types); } return super.equals(obj); } public Iterator<TypeData> iterator() { return types.iterator(); } @Override public String toString() { return types.toString(); } public boolean hasAnyParameterMatch(TypeSignature other) { for (int i = 1; i < types.size(); i++) { TypeData type1 = types.get(i); TypeData type2 = other.types.get(i); if (type1.equals(type2)) { return true; } } return false; } public boolean isCompatibleTo(TypeSignature signature) { if (size() != signature.size()) { return false; } for (int i = 0; i < size(); i++) { TypeData o1 = get(i); TypeData o2 = signature.get(i); if (o1.equals(o2)) { continue; } else if (o2.isGeneric()) { continue; } else { return false; } } return true; } } }