001 /*
002 * Copyright 2010-2013 JetBrains s.r.o.
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 * http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016
017 package org.jetbrains.jet.lang.resolve.calls.inference;
018
019 import com.google.common.collect.Maps;
020 import org.jetbrains.annotations.NotNull;
021 import org.jetbrains.jet.lang.descriptors.CallableDescriptor;
022 import org.jetbrains.jet.lang.descriptors.TypeParameterDescriptor;
023 import org.jetbrains.jet.lang.resolve.calls.results.ResolutionDebugInfo;
024 import org.jetbrains.jet.lang.resolve.calls.model.ResolvedCall;
025 import org.jetbrains.jet.lang.types.JetType;
026
027 import java.util.Map;
028 import java.util.Set;
029
030 import static org.jetbrains.jet.lang.resolve.calls.results.ResolutionDebugInfo.*;
031
032 public class DebugConstraintResolutionListener implements ConstraintResolutionListener {
033
034 private final ResolutionDebugInfo.Data debugInfo;
035 private final ResolvedCall<? extends CallableDescriptor> candidateCall;
036
037 public DebugConstraintResolutionListener(@NotNull ResolvedCall<? extends CallableDescriptor> candidateCall, @NotNull ResolutionDebugInfo.Data debugInfo) {
038 this.debugInfo = debugInfo;
039 this.candidateCall = candidateCall;
040 }
041
042 @Override
043 public void constraintsForUnknown(TypeParameterDescriptor typeParameterDescriptor, BoundsOwner typeValue) {
044 if (!ResolutionDebugInfo.isResolutionDebugEnabled()) return;
045 Map<TypeParameterDescriptor, BoundsOwner> map = debugInfo.getByKey(BOUNDS_FOR_UNKNOWNS, candidateCall);
046 if (map == null) {
047 map = Maps.newLinkedHashMap();
048 debugInfo.putByKey(BOUNDS_FOR_UNKNOWNS, candidateCall, map);
049 }
050 map.put(typeParameterDescriptor, typeValue);
051 }
052
053 @Override
054 public void constraintsForKnownType(JetType type, BoundsOwner typeValue) {
055 if (!ResolutionDebugInfo.isResolutionDebugEnabled()) return;
056 Map<JetType,BoundsOwner> map = debugInfo.getByKey(BOUNDS_FOR_KNOWNS, candidateCall);
057 if (map == null) {
058 map = Maps.newLinkedHashMap();
059 debugInfo.putByKey(BOUNDS_FOR_KNOWNS, candidateCall, map);
060 }
061 map.put(type, typeValue);
062 }
063
064 @Override
065 public void done(ConstraintSystemSolution solution, Set<TypeParameterDescriptor> typeParameterDescriptors) {
066 if (!ResolutionDebugInfo.isResolutionDebugEnabled()) return;
067 debugInfo.putByKey(SOLUTION, candidateCall, solution);
068 debugInfo.putByKey(UNKNOWNS, candidateCall, typeParameterDescriptors);
069 }
070
071 @Override
072 public void log(Object... messageFragments) {
073 if (!ResolutionDebugInfo.isResolutionDebugEnabled()) return;
074 StringBuilder stringBuilder = debugInfo.getByKey(LOG, candidateCall);
075 if (stringBuilder == null) {
076 stringBuilder = new StringBuilder();
077 debugInfo.putByKey(LOG, candidateCall, stringBuilder);
078 }
079 for (Object m : messageFragments) {
080 stringBuilder.append(m);
081 }
082 stringBuilder.append("\n");
083 }
084
085 @Override
086 public void error(Object... messageFragments) {
087 if (!ResolutionDebugInfo.isResolutionDebugEnabled()) return;
088 StringBuilder stringBuilder = debugInfo.getByKey(ERRORS, candidateCall);
089 if (stringBuilder == null) {
090 stringBuilder = new StringBuilder();
091 debugInfo.putByKey(ERRORS, candidateCall, stringBuilder);
092 }
093 for (Object m : messageFragments) {
094 stringBuilder.append(m);
095 }
096 stringBuilder.append("\n");
097 }
098 }