Rewrite evaluator to kotlin. Small refactoring

This commit is contained in:
Natalia Ukhorskaya
2013-11-13 20:31:47 +04:00
parent e23ef2dc98
commit dd0122a482
7 changed files with 429 additions and 511 deletions
@@ -114,7 +114,7 @@ public class TraceBasedJavaResolverCache implements JavaResolverCache {
PsiField psiField = ((JavaFieldImpl) field).getPsi();
trace.record(VARIABLE, psiField, descriptor);
if (AnnotationUtils.isPropertyAcceptableAsAnnotationParameter(descriptor)) {
if (AnnotationUtils.isPropertyCompileTimeConstant(descriptor)) {
PsiExpression initializer = psiField.getInitializer();
if (initializer instanceof PsiLiteralExpression) {
CompileTimeConstant<?> constant = JavaAnnotationArgumentResolver
@@ -1,496 +0,0 @@
/*
* Copyright 2010-2013 JetBrains s.r.o.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.jetbrains.jet.lang.evaluate;
import com.google.common.collect.Lists;
import com.intellij.psi.tree.IElementType;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.jet.lang.descriptors.*;
import org.jetbrains.jet.lang.descriptors.annotations.AnnotationDescriptorImpl;
import org.jetbrains.annotations.Nullable;
import org.jetbrains.jet.lang.descriptors.*;
import org.jetbrains.jet.lang.psi.*;
import org.jetbrains.jet.lang.resolve.*;
import org.jetbrains.jet.lang.resolve.calls.model.ResolvedCall;
import org.jetbrains.jet.lang.resolve.calls.model.ResolvedValueArgument;
import org.jetbrains.jet.lang.resolve.constants.*;
import org.jetbrains.jet.lang.resolve.constants.StringValue;
import org.jetbrains.jet.lang.resolve.name.Name;
import org.jetbrains.jet.lang.types.JetType;
import org.jetbrains.jet.lang.types.expressions.OperatorConventions;
import org.jetbrains.jet.lang.types.lang.KotlinBuiltIns;
import org.jetbrains.jet.lexer.JetTokens;
import java.util.List;
import java.util.Map;
import static org.jetbrains.jet.lang.resolve.BindingContext.COMPILE_TIME_INITIALIZER;
@SuppressWarnings("StaticMethodReferencedViaSubclass")
public class ConstantExpressionEvaluator extends JetVisitor<CompileTimeConstant<?>, Void> {
private final BindingTrace trace;
private final JetType expectedType;
private ConstantExpressionEvaluator(@NotNull BindingTrace trace, @NotNull JetType expectedType) {
this.trace = trace;
this.expectedType = expectedType;
}
@Nullable
public static CompileTimeConstant<?> evaluate(@NotNull JetExpression expression, @NotNull BindingTrace trace, @NotNull JetType expectedType) {
ConstantExpressionEvaluator evaluator = new ConstantExpressionEvaluator(trace, expectedType);
return evaluator.evaluate(expression);
}
@Nullable
public CompileTimeConstant<?> evaluate(@NotNull JetExpression expression) {
CompileTimeConstant<?> recordedCompileTimeConstant = trace.get(BindingContext.COMPILE_TIME_VALUE, expression);
if (recordedCompileTimeConstant != null) {
return recordedCompileTimeConstant;
}
CompileTimeConstant<?> compileTimeConstant = expression.accept(this, null);
if (compileTimeConstant != null) {
trace.record(BindingContext.COMPILE_TIME_VALUE, expression, compileTimeConstant);
return compileTimeConstant;
}
return null;
}
@Override
public CompileTimeConstant<?> visitConstantExpression(@NotNull JetConstantExpression expression, Void data) {
return trace.get(BindingContext.COMPILE_TIME_VALUE, expression);
}
@Override
public CompileTimeConstant<?> visitParenthesizedExpression(@NotNull JetParenthesizedExpression expression, Void data) {
JetExpression deparenthesizedExpression = getDeparenthesizedExpression(expression);
if (deparenthesizedExpression != null) {
return evaluate(deparenthesizedExpression);
}
return null;
}
@Override
public CompileTimeConstant<?> visitPrefixExpression(@NotNull JetPrefixExpression expression, Void data) {
JetExpression deparenthesizedExpression = getDeparenthesizedExpression(expression);
if (deparenthesizedExpression != null) {
return evaluate(deparenthesizedExpression);
}
return null;
}
@Nullable
private static JetExpression getDeparenthesizedExpression(JetExpression expression) {
JetExpression deparenthesizedExpr = JetPsiUtil.deparenthesize(expression);
if (deparenthesizedExpr == expression) {
return null;
}
return deparenthesizedExpr;
}
@Override
public CompileTimeConstant<?> visitStringTemplateExpression(@NotNull JetStringTemplateExpression expression, Void data) {
JetStringTemplateEntry[] stringTemplateEntries = expression.getEntries();
if (stringTemplateEntries.length == 1) {
CompileTimeConstant singleEntry = stringTemplateEntries[0].accept(this, null);
if (!(singleEntry instanceof StringValue) && singleEntry != null) {
return new StringValue(singleEntry.getValue().toString());
}
else {
return singleEntry;
}
}
else if (stringTemplateEntries.length > 1) {
CompileTimeConstant<?> first = stringTemplateEntries[0].accept(this, null);
if (first == null) {
return null;
}
CompileTimeConstant<?> second;
String tmpResult = null;
for (int i = 1; i < stringTemplateEntries.length; i++) {
second = stringTemplateEntries[i].accept(this, null);
if (second == null) {
return null;
}
Object object = EvaluatePackage.evaluateBinaryExpression(first, second, Name.identifier("plus"));
if (object instanceof String) {
tmpResult = (String) object;
}
else {
tmpResult = object.toString();
}
first = new StringValue(tmpResult);
}
return new StringValue(tmpResult);
}
return null;
}
@Override
public CompileTimeConstant<?> visitStringTemplateEntryWithExpression(@NotNull JetStringTemplateEntryWithExpression entry, Void data) {
JetExpression expression = entry.getExpression();
if (expression != null) {
return evaluate(expression);
}
return null;
}
@Override
public CompileTimeConstant<?> visitLiteralStringTemplateEntry(@NotNull JetLiteralStringTemplateEntry entry, Void data) {
return new StringValue(entry.getText());
}
@Override
public CompileTimeConstant<?> visitEscapeStringTemplateEntry(@NotNull JetEscapeStringTemplateEntry entry, Void data) {
return new StringValue(entry.getUnescapedValue());
}
@Override
public CompileTimeConstant<?> visitBinaryExpression(@NotNull JetBinaryExpression expression, Void data) {
JetExpression leftExpression = expression.getLeft();
if (leftExpression == null) {
return null;
}
IElementType operationToken = expression.getOperationToken();
if (OperatorConventions.BOOLEAN_OPERATIONS.containsKey(operationToken)) {
JetType booleanType = KotlinBuiltIns.getInstance().getBooleanType();
CompileTimeConstant<?> leftConstant = evaluate(leftExpression, trace, booleanType);
if (leftConstant == null) {
return null;
}
JetExpression rightExpression = expression.getRight();
if (rightExpression == null) {
return null;
}
CompileTimeConstant<?> rightConstant = evaluate(rightExpression, trace, booleanType);
if (rightConstant == null) {
return null;
}
Name operationName = operationToken == JetTokens.ANDAND ? Name.identifier("andand") : Name.identifier("oror");
Object result = EvaluatePackage.evaluateBinaryExpression(leftConstant, rightConstant, operationName);
return createCompileTimeConstant(result, expectedType);
}
else {
Object result = evaluateCall(expression.getOperationReference(), leftExpression);
if (OperatorConventions.COMPARISON_OPERATIONS.contains(operationToken)) {
return createCompileTimeConstantForCompareTo(result, operationToken);
}
else if (OperatorConventions.EQUALS_OPERATIONS.contains(operationToken)) {
return createCompileTimeConstantForEquals(result, operationToken);
}
else {
return createCompileTimeConstant(result, expectedType);
}
}
}
@Nullable
private Object evaluateCall(@NotNull JetExpression resolvedCallExpression, @NotNull JetExpression receiverExpression) {
ResolvedCall<?> resolvedCall = trace.getBindingContext().get(BindingContext.RESOLVED_CALL, resolvedCallExpression);
if (resolvedCall != null) {
CallableDescriptor resultingDescriptor = resolvedCall.getResultingDescriptor();
JetType receiverExpressionType = getReceiverExpressionType(resolvedCall);
if (receiverExpressionType == null) {
return null;
}
ConstantExpressionEvaluator evaluator = new ConstantExpressionEvaluator(trace, receiverExpressionType);
CompileTimeConstant<?> receiverValue = receiverExpression.accept(evaluator, null);
if (receiverValue == null) {
return null;
}
List<CompileTimeConstant<?>> arguments = Lists.newArrayList();
for (Map.Entry<ValueParameterDescriptor, ResolvedValueArgument> argumentEntry : resolvedCall.getValueArguments().entrySet()) {
arguments.addAll(resolveArguments(argumentEntry.getValue().getArguments(), argumentEntry.getKey().getType(), trace));
}
Name resultingDescriptorName = resultingDescriptor.getName();
if (arguments.isEmpty()) {
return EvaluatePackage.evaluateUnaryExpression(receiverValue, resultingDescriptorName);
}
else if (arguments.size() == 1) {
return EvaluatePackage.evaluateBinaryExpression(receiverValue, arguments.iterator().next(), resultingDescriptorName);
}
}
return null;
}
@Nullable
private static CompileTimeConstant<?> createCompileTimeConstantForEquals(
@Nullable Object result,
@NotNull IElementType operationToken
) {
if (result instanceof Boolean) {
boolean resultCode = ((Boolean) result).booleanValue();
if (operationToken == JetTokens.EQEQ) {
return resultCode ? BooleanValue.TRUE : BooleanValue.FALSE;
}
else if (operationToken == JetTokens.EXCLEQ) {
return resultCode ? BooleanValue.FALSE : BooleanValue.TRUE;
}
}
return null;
}
@Nullable
private static CompileTimeConstant<?> createCompileTimeConstantForCompareTo(
@Nullable Object result,
@NotNull IElementType operationToken
) {
if (result instanceof Integer) {
int resultCode = ((Integer) result).intValue();
if (resultCode == 0) {
if (operationToken == JetTokens.EQEQ ||
operationToken == JetTokens.LTEQ ||
operationToken == JetTokens.GTEQ) {
return BooleanValue.TRUE;
}
else {
return BooleanValue.FALSE;
}
}
if (resultCode == 1) {
if (operationToken == JetTokens.GT ||
operationToken == JetTokens.GTEQ) {
return BooleanValue.TRUE;
}
else {
return BooleanValue.FALSE;
}
}
if (resultCode == -1) {
if (operationToken == JetTokens.LT ||
operationToken == JetTokens.LTEQ) {
return BooleanValue.TRUE;
}
else {
return BooleanValue.FALSE;
}
}
}
return null;
}
@Override
public CompileTimeConstant<?> visitUnaryExpression(@NotNull JetUnaryExpression expression, Void data) {
JetExpression leftExpression = expression.getBaseExpression();
if (leftExpression == null) {
return null;
}
Object result = evaluateCall(expression.getOperationReference(), leftExpression);
return createCompileTimeConstant(result, expectedType);
}
@Nullable
public static CompileTimeConstant<?> createCompileTimeConstant(@Nullable Object value, @NotNull JetType expectedType) {
if (value == null) {
return null;
}
else if (value instanceof Integer) {
return getIntegerValue(((Integer) value).longValue(), expectedType);
}
else if (value instanceof Byte) {
return getIntegerValue(((Byte) value).longValue(), expectedType);
}
else if (value instanceof Short) {
return getIntegerValue(((Short) value).longValue(), expectedType);
}
else if (value instanceof Long) {
return new LongValue((Long) value);
}
else if (value instanceof Float) {
if (CompileTimeConstantResolver.noExpectedTypeOrError(expectedType) ||
expectedType.equals(KotlinBuiltIns.getInstance().getDoubleType())) {
return new DoubleValue(((Float) value).doubleValue());
}
else {
return new FloatValue((Float) value);
}
}
else if (value instanceof Double) {
return new DoubleValue((Double) value);
}
else if (value instanceof Boolean) {
return ((Boolean) value) ? BooleanValue.TRUE : BooleanValue.FALSE;
}
else if (value instanceof Character) {
return new CharValue((Character) value);
}
else if (value instanceof String) {
return new StringValue((String) value);
}
return null;
}
@Nullable
private static CompileTimeConstant<?> getIntegerValue(@NotNull Long value, @NotNull JetType expectedType) {
if (CompileTimeConstantResolver.noExpectedTypeOrError(expectedType)) {
if (Integer.MIN_VALUE <= value && value <= Integer.MAX_VALUE) {
return new IntValue(value.intValue());
}
return new LongValue(value);
}
KotlinBuiltIns builtIns = KotlinBuiltIns.getInstance();
if (expectedType.equals(builtIns.getIntType())) {
return new IntValue(value.intValue());
}
else if (expectedType.equals(builtIns.getLongType())) {
return new LongValue(value);
}
else if (expectedType.equals(builtIns.getShortType())) {
if (Short.MIN_VALUE <= value && value <= Short.MAX_VALUE) {
return new ShortValue(value.shortValue());
}
else if (Integer.MIN_VALUE <= value && value <= Integer.MAX_VALUE) {
return new IntValue(value.intValue());
}
return new LongValue(value);
}
else if (expectedType.equals(builtIns.getByteType())) {
if (Byte.MIN_VALUE <= value && value <= Byte.MAX_VALUE) {
return new ByteValue(value.byteValue());
}
else if (Integer.MIN_VALUE <= value && value <= Integer.MAX_VALUE) {
return new IntValue(value.intValue());
}
return new LongValue(value);
}
else if (expectedType.equals(builtIns.getCharType())) {
return new IntValue(value.intValue());
}
return null;
}
@Override
public CompileTimeConstant<?> visitSimpleNameExpression(@NotNull JetSimpleNameExpression expression, Void data) {
DeclarationDescriptor descriptor = trace.getBindingContext().get(BindingContext.REFERENCE_TARGET, expression);
if (descriptor != null && DescriptorUtils.isEnumEntry(descriptor)) {
return new EnumValue((ClassDescriptor) descriptor);
}
ResolvedCall<? extends CallableDescriptor> resolvedCall =
trace.getBindingContext().get(BindingContext.RESOLVED_CALL, expression);
if (resolvedCall != null) {
CallableDescriptor callableDescriptor = resolvedCall.getResultingDescriptor();
if (callableDescriptor instanceof PropertyDescriptor) {
PropertyDescriptor propertyDescriptor = (PropertyDescriptor) callableDescriptor;
if (AnnotationUtils.isPropertyAcceptableAsAnnotationParameter(propertyDescriptor)) {
return trace.getBindingContext().get(COMPILE_TIME_INITIALIZER, propertyDescriptor);
}
}
}
return null;
}
/* 1.toInt(); 1.plus(1); MyEnum.A */
@Override
public CompileTimeConstant<?> visitQualifiedExpression(@NotNull JetQualifiedExpression expression, Void data) {
JetExpression receiverExpression = expression.getReceiverExpression();
JetExpression selectorExpression = expression.getSelectorExpression();
if (selectorExpression instanceof JetCallExpression) {
JetExpression calleeExpression = ((JetCallExpression) selectorExpression).getCalleeExpression();
if (!(calleeExpression instanceof JetSimpleNameExpression)) {
return null;
}
if (((JetCallExpression) selectorExpression).getValueArguments().size() < 2) {
Object result = evaluateCall(calleeExpression, receiverExpression);
return createCompileTimeConstant(result, expectedType);
}
}
if (selectorExpression != null) {
return evaluate(selectorExpression);
}
return null;
}
@Nullable
private static JetType getReceiverExpressionType(ResolvedCall<? extends CallableDescriptor> resolvedCall) {
switch(resolvedCall.getExplicitReceiverKind()) {
case THIS_OBJECT: return resolvedCall.getThisObject().getType();
case RECEIVER_ARGUMENT: return resolvedCall.getReceiverArgument().getType();
case NO_EXPLICIT_RECEIVER: return null;
case BOTH_RECEIVERS: return null;
}
return null;
}
@Override
public CompileTimeConstant<?> visitCallExpression(@NotNull JetCallExpression expression, Void data) {
ResolvedCall<? extends CallableDescriptor> call =
trace.getBindingContext().get(BindingContext.RESOLVED_CALL, (expression).getCalleeExpression());
if (call != null) {
CallableDescriptor resultingDescriptor = call.getResultingDescriptor();
if (AnnotationUtils.isArrayMethodCall(call)) {
JetType type = resultingDescriptor.getValueParameters().iterator().next().getVarargElementType();
List<CompileTimeConstant<?>> arguments = Lists.newArrayList();
for (ResolvedValueArgument descriptorToArgument : call.getValueArguments().values()) {
arguments.addAll(resolveArguments(descriptorToArgument.getArguments(), type, trace));
}
return new ArrayValue(arguments, resultingDescriptor.getReturnType());
}
if (resultingDescriptor instanceof ConstructorDescriptor) {
JetType constructorReturnType = resultingDescriptor.getReturnType();
assert constructorReturnType != null : "Constructor should have return type";
if (DescriptorUtils.isAnnotationClass(constructorReturnType.getConstructor().getDeclarationDescriptor())) {
AnnotationDescriptorImpl descriptor = new AnnotationDescriptorImpl();
descriptor.setAnnotationType(constructorReturnType);
AnnotationResolver.resolveAnnotationArgument(descriptor, call, trace);
return new AnnotationValue(descriptor);
}
}
if (AnnotationUtils.isJavaClassMethodCall(call)) {
return new JavaClassValue(resultingDescriptor.getReturnType());
}
}
return null;
}
@NotNull
private static List<CompileTimeConstant<?>> resolveArguments(
@NotNull List<ValueArgument> valueArguments,
@NotNull JetType expectedType,
@NotNull BindingTrace trace
) {
List<CompileTimeConstant<?>> constants = Lists.newArrayList();
for (ValueArgument argument : valueArguments) {
JetExpression argumentExpression = argument.getArgumentExpression();
if (argumentExpression != null) {
CompileTimeConstant<?> constant = evaluate(argumentExpression, trace, expectedType);
if (constant != null) {
constants.add(constant);
}
}
}
return constants;
}
@Override
public CompileTimeConstant<?> visitJetElement(@NotNull JetElement element, Void data) {
return null;
}
}
@@ -0,0 +1,407 @@
/*
* Copyright 2010-2013 JetBrains s.r.o.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.jetbrains.jet.lang.evaluate
import com.intellij.psi.tree.IElementType
import org.jetbrains.jet.lang.descriptors.annotations.AnnotationDescriptorImpl
import org.jetbrains.jet.lang.descriptors.*
import org.jetbrains.jet.lang.psi.*
import org.jetbrains.jet.lang.resolve.*
import org.jetbrains.jet.lang.resolve.calls.model.ResolvedCall
import org.jetbrains.jet.lang.resolve.constants.*
import org.jetbrains.jet.lang.resolve.name.Name
import org.jetbrains.jet.lang.types.JetType
import org.jetbrains.jet.lang.types.expressions.OperatorConventions
import org.jetbrains.jet.lang.types.lang.KotlinBuiltIns
import org.jetbrains.jet.lexer.JetTokens
import org.jetbrains.jet.lang.resolve.BindingContext.COMPILE_TIME_INITIALIZER
import org.jetbrains.jet.lang.resolve.calls.tasks.ExplicitReceiverKind
import org.jetbrains.jet.lang.types.TypeUtils
import java.lang.Short as JShort
import java.lang.Byte as JByte
[suppress("PARAMETER_NAME_CHANGED_ON_OVERRIDE")]
public class ConstantExpressionEvaluator private (val trace: BindingTrace) : JetVisitor<CompileTimeConstant<*>, JetType>() {
public fun evaluate(expression: JetExpression, expectedType: JetType?): CompileTimeConstant<*>? {
val recordedCompileTimeConstant = trace.get(BindingContext.COMPILE_TIME_VALUE, expression)
if (recordedCompileTimeConstant != null) {
return recordedCompileTimeConstant
}
val compileTimeConstant = expression.accept(this, expectedType ?: TypeUtils.NO_EXPECTED_TYPE)
if (compileTimeConstant != null) {
trace.record(BindingContext.COMPILE_TIME_VALUE, expression, compileTimeConstant)
return compileTimeConstant
}
return null
}
override fun visitConstantExpression(expression: JetConstantExpression, expectedType: JetType?): CompileTimeConstant<*>? {
return trace.get(BindingContext.COMPILE_TIME_VALUE, expression)
}
override fun visitParenthesizedExpression(expression: JetParenthesizedExpression, expectedType: JetType?): CompileTimeConstant<*>? {
val deparenthesizedExpression = getDeparenthesizedExpression(expression)
if (deparenthesizedExpression != null) {
return evaluate(deparenthesizedExpression, expectedType)
}
return null
}
override fun visitPrefixExpression(expression: JetPrefixExpression, expectedType: JetType?): CompileTimeConstant<*>? {
val deparenthesizedExpression = getDeparenthesizedExpression(expression)
return if (deparenthesizedExpression != null) {
evaluate(deparenthesizedExpression, expectedType)
} else {
super.visitPrefixExpression(expression, expectedType)
}
}
override fun visitStringTemplateExpression(expression: JetStringTemplateExpression, expectedType: JetType?): CompileTimeConstant<*>? {
val stringTemplateEntries = expression.getEntries()
if (stringTemplateEntries.size == 1) {
return stringTemplateEntries[0].accept(this, null)
}
else if (stringTemplateEntries.size > 1) {
return stringTemplateEntries.fold(StringValue(""): CompileTimeConstant<*>?) {
f, s ->
if (f != null) {
val second = s.accept(this, null)
if (second == null) {
null
}
else {
createCompileTimeConstant(evaluateBinaryExpression(f, second, Name.identifier("plus")), expectedType)
}
}
else {
null
}
}
}
return null
}
override fun visitStringTemplateEntryWithExpression(entry: JetStringTemplateEntryWithExpression, expectedType: JetType?): CompileTimeConstant<*>? {
val expression = entry.getExpression()
if (expression != null) {
return createStringConstant(evaluate(expression, expectedType))
}
return null
}
override fun visitLiteralStringTemplateEntry(entry: JetLiteralStringTemplateEntry, expectedType: JetType?) = StringValue(entry.getText())
override fun visitEscapeStringTemplateEntry(entry: JetEscapeStringTemplateEntry, expectedType: JetType?) = StringValue(entry.getUnescapedValue())
override fun visitBinaryExpression(expression: JetBinaryExpression, expectedType: JetType?): CompileTimeConstant<*>? {
val leftExpression = expression.getLeft()
if (leftExpression == null) {
return null
}
val operationToken = expression.getOperationToken()
if (OperatorConventions.BOOLEAN_OPERATIONS.containsKey(operationToken)) {
val booleanType = KotlinBuiltIns.getInstance().getBooleanType()
val leftConstant = evaluate(leftExpression, booleanType)
if (leftConstant == null) {
return null
}
val rightExpression = expression.getRight()
if (rightExpression == null) {
return null
}
val rightConstant = evaluate(rightExpression, booleanType)
if (rightConstant == null) {
return null
}
val operationName = when(operationToken) {
JetTokens.ANDAND -> Name.identifier("&&")
JetTokens.OROR -> Name.identifier("||")
else -> throw IllegalArgumentException("Unknown boolean operation token ${operationToken}")
}
val result = evaluateBinaryExpression(leftConstant, rightConstant, operationName)
return createCompileTimeConstant(result, expectedType)
}
else {
val result = evaluateCall(expression.getOperationReference(), leftExpression)
return when(operationToken) {
in OperatorConventions.COMPARISON_OPERATIONS -> createCompileTimeConstantForCompareTo(result, operationToken!!)
in OperatorConventions.EQUALS_OPERATIONS -> createCompileTimeConstantForEquals(result, operationToken!!)
else -> createCompileTimeConstant(result, expectedType)
}
}
}
private fun evaluateCall(callExpression: JetExpression, receiverExpression: JetExpression): Any? {
val resolvedCall = trace.getBindingContext().get(BindingContext.RESOLVED_CALL, callExpression)
if (resolvedCall != null) {
val resultingDescriptor = resolvedCall.getResultingDescriptor()
// TODO getResultingDescriptor has NotNull annotation
if (resultingDescriptor == null) {
return null
}
val receiverExpressionType = getReceiverExpressionType(resolvedCall)
if (receiverExpressionType == null) {
return null
}
val receiverValue = evaluate(receiverExpression, receiverExpressionType)
if (receiverValue == null) {
return null
}
val arguments = resolvedCall.getValueArguments().entrySet().flatMap {
entry -> val (parameter, argument) = entry
resolveArguments(argument.getArguments(), parameter.getType())
}
val resultingDescriptorName = resultingDescriptor.getName()
if (arguments.isEmpty()) {
return evaluateUnaryExpression(receiverValue, resultingDescriptorName)
}
else if (arguments.size() == 1) {
return evaluateBinaryExpression(receiverValue, arguments.first!!, resultingDescriptorName)
}
}
return null
}
override fun visitUnaryExpression(expression: JetUnaryExpression, expectedType: JetType?): CompileTimeConstant<*>? {
val leftExpression = expression.getBaseExpression()
if (leftExpression == null) return null
val result = evaluateCall(expression.getOperationReference(), leftExpression)
return createCompileTimeConstant(result, expectedType)
}
override fun visitSimpleNameExpression(expression: JetSimpleNameExpression, expectedType: JetType?): CompileTimeConstant<*>? {
val enumDescriptor = trace.getBindingContext().get(BindingContext.REFERENCE_TARGET, expression);
if (enumDescriptor != null && DescriptorUtils.isEnumEntry(enumDescriptor)) {
return EnumValue(enumDescriptor as ClassDescriptor);
}
val resolvedCall = trace.getBindingContext().get(BindingContext.RESOLVED_CALL, expression)
if (resolvedCall != null) {
val callableDescriptor = resolvedCall.getResultingDescriptor()
if (callableDescriptor is PropertyDescriptor) {
if (AnnotationUtils.isPropertyCompileTimeConstant(callableDescriptor)) {
return trace.getBindingContext().get(COMPILE_TIME_INITIALIZER, callableDescriptor)
}
}
}
return null
}
/* 1.toInt(); 1.plus(1); MyEnum.A */
override fun visitQualifiedExpression(expression: JetQualifiedExpression, expectedType: JetType?): CompileTimeConstant<*>? {
val selectorExpression = expression.getSelectorExpression()
if (selectorExpression is JetCallExpression) {
val calleeExpression = selectorExpression.getCalleeExpression()
if (calleeExpression !is JetSimpleNameExpression) {
return null
}
val receiverExpression = expression.getReceiverExpression()
val result = evaluateCall(calleeExpression, receiverExpression)
return createCompileTimeConstant(result, expectedType)
}
// TODO
if (selectorExpression != null) {
return evaluate(selectorExpression, expectedType)
}
return null
}
override fun visitCallExpression(expression: JetCallExpression, expectedType: JetType?): CompileTimeConstant<*>? {
val call = trace.getBindingContext().get(BindingContext.RESOLVED_CALL, expression.getCalleeExpression())
if (call != null) {
val resultingDescriptor = call.getResultingDescriptor()
if (resultingDescriptor != null) {
if (AnnotationUtils.isArrayMethodCall(call)) {
val varargType = resultingDescriptor.getValueParameters().first?.getVarargElementType()!!
//todo flatmap
val arguments = arrayListOf<CompileTimeConstant<*>>()
for (descriptorToArgument in call.getValueArguments().values()) {
arguments.addAll(resolveArguments(descriptorToArgument.getArguments(), varargType))
}
return ArrayValue(arguments, resultingDescriptor.getReturnType()!!)
}
// Ann()
if (resultingDescriptor is ConstructorDescriptor) {
val constructorReturnType = resultingDescriptor.getReturnType()
if (DescriptorUtils.isAnnotationClass(resultingDescriptor.getContainingDeclaration())) {
val descriptor = AnnotationDescriptorImpl()
descriptor.setAnnotationType(constructorReturnType)
AnnotationResolver.resolveAnnotationArgument(descriptor, call, trace)
return AnnotationValue(descriptor)
}
}
if (AnnotationUtils.isJavaClassMethodCall(call)) {
return JavaClassValue(resultingDescriptor.getReturnType())
}
}
}
return null
}
private fun resolveArguments(valueArguments: List<ValueArgument>, expectedType: JetType): List<CompileTimeConstant<*>> {
//todo flatMap
val constants = arrayListOf<CompileTimeConstant<*>>()
for (argument in valueArguments) {
val argumentExpression = argument.getArgumentExpression()
if (argumentExpression != null) {
val constant = evaluate(argumentExpression, expectedType)
if (constant != null) {
constants.add(constant)
}
}
}
return constants
}
override fun visitJetElement(element: JetElement, expectedType: JetType?): CompileTimeConstant<*>? {
return null
}
class object {
public fun evaluate(expression: JetExpression, trace: BindingTrace, expectedType: JetType? = TypeUtils.NO_EXPECTED_TYPE): CompileTimeConstant<*>? {
val evaluator = ConstantExpressionEvaluator(trace)
return evaluator.evaluate(expression, expectedType)
}
}
}
private fun getDeparenthesizedExpression(expression: JetExpression): JetExpression? {
val deparenthesizedExpr = JetPsiUtil.deparenthesize(expression)
if (deparenthesizedExpr == expression) {
return null
}
return deparenthesizedExpr
}
private fun createCompileTimeConstantForEquals(result: Any?, operationToken: IElementType): CompileTimeConstant<*>? {
if (result is Boolean) {
return when (operationToken) {
JetTokens.EQEQ -> if (result) BooleanValue.TRUE else BooleanValue.FALSE
JetTokens.EXCLEQ -> if (result) BooleanValue.FALSE else BooleanValue.TRUE
else -> throw IllegalStateException("Unknown equals operation token: $operationToken")
}
}
return null
}
private fun createCompileTimeConstantForCompareTo(result: Any?, operationToken: IElementType): CompileTimeConstant<*>? {
if (result is Int) {
return when (result) {
-1 -> when (operationToken) {
JetTokens.LT -> BooleanValue.TRUE
JetTokens.LTEQ -> BooleanValue.TRUE
else -> BooleanValue.FALSE
}
0 -> when (operationToken) {
JetTokens.LTEQ -> BooleanValue.TRUE
JetTokens.GTEQ -> BooleanValue.TRUE
else -> BooleanValue.FALSE
}
1 -> when (operationToken) {
JetTokens.GT -> BooleanValue.TRUE
JetTokens.GTEQ -> BooleanValue.TRUE
else -> BooleanValue.FALSE
}
else -> null
}
}
return null
}
private fun createStringConstant(value: CompileTimeConstant<*>?): CompileTimeConstant<*>? {
return when (value) {
null -> null
is StringValue -> value
else -> StringValue(value.getValue().toString())
}
}
private fun createCompileTimeConstant(value: Any?, expectedType: JetType?): CompileTimeConstant<*>? {
return when(value) {
null -> null
is Byte, is Short, is Int, is Long -> getIntegerValue((value as Number).toLong(), expectedType ?: TypeUtils.NO_EXPECTED_TYPE)
is Char -> CharValue(value)
is Float -> if (CompileTimeConstantResolver.noExpectedTypeOrError(expectedType) ||
expectedType == KotlinBuiltIns.getInstance().getDoubleType()) DoubleValue(value.toDouble())
else FloatValue(value)
is Double -> DoubleValue(value)
is Boolean -> if (value) BooleanValue.TRUE else BooleanValue.FALSE
is String -> StringValue(value)
else -> null
}
}
private fun getIntegerValue(value: Long, expectedType: JetType): CompileTimeConstant<*>? {
if (CompileTimeConstantResolver.noExpectedTypeOrError(expectedType)) {
if (Integer.MIN_VALUE <= value && value <= Integer.MAX_VALUE) {
return IntValue(value.toInt())
}
return LongValue(value)
}
fun defaultIntegerValue(value: Long) = when (value) {
in Integer.MIN_VALUE..Integer.MAX_VALUE.toLong() -> IntValue(value.toInt())
else -> LongValue(value)
}
val builtIns = KotlinBuiltIns.getInstance()
return when (TypeUtils.makeNotNullable(expectedType)) {
builtIns.getIntType() -> IntValue(value.toInt())
builtIns.getLongType() -> LongValue(value)
builtIns.getShortType() -> when (value) {
in JShort.MIN_VALUE..JShort.MAX_VALUE.toLong() -> ShortValue(value.toShort())
else -> defaultIntegerValue(value)
}
builtIns.getByteType() -> when (value) {
in JByte.MIN_VALUE..JByte.MAX_VALUE.toLong() -> ByteValue(value.toByte())
else -> defaultIntegerValue(value)
}
builtIns.getCharType() -> IntValue(value.toInt())
else -> defaultIntegerValue(value)
}
}
private fun getReceiverExpressionType(resolvedCall: ResolvedCall<*>): JetType? {
return when (resolvedCall.getExplicitReceiverKind()) {
ExplicitReceiverKind.THIS_OBJECT -> resolvedCall.getThisObject().getType()
ExplicitReceiverKind.RECEIVER_ARGUMENT -> resolvedCall.getReceiverArgument().getType()
ExplicitReceiverKind.NO_EXPLICIT_RECEIVER -> null
ExplicitReceiverKind.BOTH_RECEIVERS -> null
else -> null
}
}
@@ -62,20 +62,27 @@ private val CHAR = CompileTimeType<Char>()
private val BOOLEAN = CompileTimeType<Boolean>()
private val STRING = CompileTimeType<String>()
private fun <A, B> bOp(a: CompileTimeType<A>, b: CompileTimeType<B>, functionNameAsString: String, f: (A, B) -> Any) = BinaryOperation(a, b, Name.identifier(functionNameAsString)) to f as Function2<Any?, Any?, Any>
[suppress("UNCHECKED_CAST")]
private fun <A, B> bOp(
a: CompileTimeType<A>,
b: CompileTimeType<B>,
functionNameAsString: String,
f: (A, B) -> Any
) = BinaryOperation(a, b, Name.identifier(functionNameAsString)) to f as Function2<Any?, Any?, Any>
private fun <A> uOp(a: CompileTimeType<A>, functionNameAsString: String, f: (A) -> Any) = UnaryOperation(a, Name.identifier(functionNameAsString)) to f as Function1<Any?, Any>
private data class BinaryOperation<A, B>(val f: CompileTimeType<out A>, val s: CompileTimeType<out B>, val functionName: Name)
private data class UnaryOperation<A>(val f: CompileTimeType<out A>, val functionName: Name)
private val unaryOperations = hashMapOf<UnaryOperation<*>, (Any?) -> Any>(
uOp(DOUBLE, "minus", { a -> a.minus() }),
uOp(FLOAT, "minus", { a -> a.minus() }),
uOp(LONG, "minus", { a -> a.minus() }),
uOp(INT, "minus", { a -> a.minus() }),
uOp(SHORT, "minus", { a -> a.minus() }),
uOp(BYTE, "minus", { a -> a.minus() }),
uOp(CHAR, "minus", { a -> a.minus() }),
uOp(DOUBLE, "minus", { -it }),
uOp(FLOAT, "minus", { -it }),
uOp(LONG, "minus", { -it }),
uOp(INT, "minus", { -it }),
uOp(SHORT, "minus", { -it }),
uOp(BYTE, "minus", { -it }),
uOp(CHAR, "minus", { -it }),
uOp(DOUBLE, "plus", { a -> a.plus() }),
uOp(FLOAT, "plus", { a -> a.plus() }),
uOp(LONG, "plus", { a -> a.plus() }),
@@ -24,7 +24,7 @@ import org.jetbrains.jet.lang.descriptors.annotations.Annotated;
import org.jetbrains.jet.lang.descriptors.annotations.AnnotationDescriptor;
import org.jetbrains.jet.lang.descriptors.annotations.AnnotationDescriptorImpl;
import org.jetbrains.jet.lang.diagnostics.Errors;
import org.jetbrains.jet.lang.evaluate.ConstantExpressionEvaluator;
import org.jetbrains.jet.lang.evaluate.EvaluatePackage;
import org.jetbrains.jet.lang.psi.*;
import org.jetbrains.jet.lang.resolve.calls.CallResolver;
import org.jetbrains.jet.lang.resolve.calls.autocasts.DataFlowInfo;
@@ -32,7 +32,8 @@ import org.jetbrains.jet.lang.resolve.calls.model.ResolvedCall;
import org.jetbrains.jet.lang.resolve.calls.model.ResolvedValueArgument;
import org.jetbrains.jet.lang.resolve.calls.results.OverloadResolutionResults;
import org.jetbrains.jet.lang.resolve.calls.util.CallMaker;
import org.jetbrains.jet.lang.resolve.constants.*;
import org.jetbrains.jet.lang.resolve.constants.ArrayValue;
import org.jetbrains.jet.lang.resolve.constants.CompileTimeConstant;
import org.jetbrains.jet.lang.resolve.scopes.JetScope;
import org.jetbrains.jet.lang.resolve.scopes.receivers.ReceiverValue;
import org.jetbrains.jet.lang.types.ErrorUtils;
@@ -47,7 +48,6 @@ import java.util.List;
import java.util.Map;
import static org.jetbrains.jet.lang.resolve.BindingContext.ANNOTATION_DESCRIPTOR_TO_PSI_ELEMENT;
import static org.jetbrains.jet.lang.resolve.BindingContext.COMPILE_TIME_INITIALIZER;
import static org.jetbrains.jet.lang.types.TypeUtils.NO_EXPECTED_TYPE;
public class AnnotationResolver {
@@ -257,7 +257,7 @@ public class AnnotationResolver {
@NotNull JetType expectedType,
@NotNull BindingTrace trace
) {
return ConstantExpressionEvaluator.evaluate(expression, trace, expectedType);
return EvaluatePackage.evaluate(expression, trace, expectedType);
}
@NotNull
@@ -114,7 +114,7 @@ public class AnnotationUtils {
return false;
}
public static boolean isPropertyAcceptableAsAnnotationParameter(@NotNull PropertyDescriptor descriptor) {
public static boolean isPropertyCompileTimeConstant(@NotNull PropertyDescriptor descriptor) {
if (descriptor.isVar()) {
return false;
}
@@ -555,7 +555,7 @@ public class BodyResolver {
scope, propertyDescriptor.getTypeParameters(), NO_RECEIVER_PARAMETER, trace);
JetType expectedTypeForInitializer = property.getTypeRef() != null ? propertyDescriptor.getType() : NO_EXPECTED_TYPE;
expressionTypingServices.getType(propertyDeclarationInnerScope, initializer, expectedTypeForInitializer, context.getOuterDataFlowInfo(), trace);
if (AnnotationUtils.isPropertyAcceptableAsAnnotationParameter(propertyDescriptor)) {
if (AnnotationUtils.isPropertyCompileTimeConstant(propertyDescriptor)) {
CompileTimeConstant<?> constant = AnnotationResolver.resolveExpressionToCompileTimeValue(initializer, expectedTypeForInitializer, trace);
if (constant != null) {
trace.record(BindingContext.COMPILE_TIME_INITIALIZER, propertyDescriptor, constant);