.kotlin.kotlin-compiler.1.3.11.source-code.Slices Maven / Gradle / Ivy
/*
* Copyright 2010-2015 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.kotlin.util.slicedMap;
import com.intellij.openapi.diagnostic.Logger;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.kotlin.psi.KtElement;
import org.jetbrains.kotlin.psi.psiUtil.PsiUtilsKt;
import org.jetbrains.kotlin.resolve.constants.ConstantValueFactoryKt;
import org.jetbrains.kotlin.resolve.constants.IntegerValueTypeConstant;
import org.jetbrains.kotlin.types.TypeUtils;
import java.util.Arrays;
import java.util.List;
public class Slices {
private static final Logger LOG = Logger.getInstance(Slices.class);
public static final RewritePolicy ONLY_REWRITE_TO_EQUAL = new RewritePolicy() {
@Override
public boolean rewriteProcessingNeeded(K key) {
return true;
}
@Override
public boolean processRewrite(WritableSlice slice, K key, V oldValue, V newValue) {
if (!((oldValue == null && newValue == null) || (oldValue != null && oldValue.equals(newValue)))) {
logErrorAboutRewritingNonEqualObjects(slice, key, oldValue, newValue);
}
return true;
}
};
// Rewrite is allowed for equal objects and for signed constant values that were converted to unsigned ones
// This is needed to avoid making `CompileTimeConstant` mutable
public static final RewritePolicy COMPILE_TIME_VALUE_REWRITE_POLICY = new RewritePolicy() {
@Override
public boolean rewriteProcessingNeeded(K key) {
return true;
}
@Override
public boolean processRewrite(WritableSlice slice, K key, V oldValue, V newValue) {
if ((oldValue == null && newValue == null) || (oldValue != null && oldValue.equals(newValue))) return true;
if (oldValue instanceof IntegerValueTypeConstant && newValue instanceof IntegerValueTypeConstant) {
IntegerValueTypeConstant oldConstant = (IntegerValueTypeConstant) oldValue;
IntegerValueTypeConstant newConstant = (IntegerValueTypeConstant) newValue;
if (oldConstant.getParameters().isPure() && newConstant.getParameters().isUnsignedNumberLiteral()) {
long oldConstantValue = oldConstant.getValue(TypeUtils.NO_EXPECTED_TYPE).longValue();
Number newConstantValue = newConstant.getValue(TypeUtils.NO_EXPECTED_TYPE);
if (oldConstantValue == newConstantValue.longValue() ||
oldConstantValue == ConstantValueFactoryKt.fromUIntToLong(newConstantValue.intValue()) ||
oldConstantValue == ConstantValueFactoryKt.fromUByteToLong(newConstantValue.byteValue()) ||
oldConstantValue == ConstantValueFactoryKt.fromUShortToLong(newConstantValue.shortValue())
) {
return true;
}
}
}
logErrorAboutRewritingNonEqualObjects(slice, key, oldValue, newValue);
return true;
}
};
private Slices() {
}
public static SliceBuilder sliceBuilder() {
return new SliceBuilder<>(ONLY_REWRITE_TO_EQUAL);
}
public static WritableSlice createSimpleSlice() {
return new BasicWritableSlice<>(ONLY_REWRITE_TO_EQUAL);
}
public static WritableSlice createCollectiveSlice() {
return new BasicWritableSlice<>(ONLY_REWRITE_TO_EQUAL, true);
}
public static WritableSlice createSimpleSetSlice() {
return new SetSlice<>(RewritePolicy.DO_NOTHING);
}
public static WritableSlice createCollectiveSetSlice() {
return new SetSlice<>(RewritePolicy.DO_NOTHING, true);
}
public static class SliceBuilder {
private List> furtherLookupSlices;
private final RewritePolicy rewritePolicy;
private String debugName;
private SliceBuilder(RewritePolicy rewritePolicy) {
this.rewritePolicy = rewritePolicy;
}
public SliceBuilder setFurtherLookupSlices(ReadOnlySlice... furtherLookupSlices) {
this.furtherLookupSlices = Arrays.asList(furtherLookupSlices);
return this;
}
public SliceBuilder setDebugName(@NotNull String debugName) {
this.debugName = debugName;
return this;
}
public WritableSlice build() {
BasicWritableSlice result = doBuild();
if (debugName != null) {
result.setDebugName(debugName);
}
return result;
}
private BasicWritableSlice doBuild() {
if (furtherLookupSlices != null) {
return new BasicWritableSlice(rewritePolicy) {
@Override
public V computeValue(SlicedMap map, K key, V value, boolean valueNotFound) {
if (valueNotFound) {
for (ReadOnlySlice slice : furtherLookupSlices) {
V v = map.get(slice, key);
if (v != null) {
return v;
}
}
return null;
}
return super.computeValue(map, key, value, false);
}
};
}
return new BasicWritableSlice<>(rewritePolicy);
}
}
private static void logErrorAboutRewritingNonEqualObjects(WritableSlice slice, K key, V oldValue, V newValue) {
// NOTE: Use BindingTraceContext.TRACK_REWRITES to debug this exception
LOG.error("Rewrite at slice " + slice +
" key: " + key +
" old value: " + oldValue + '@' + System.identityHashCode(oldValue) +
" new value: " + newValue + '@' + System.identityHashCode(newValue) +
(key instanceof KtElement ? "\n" + PsiUtilsKt.getElementTextWithContext((KtElement) key) : ""));
}
}