/*
* Copyright 2003-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 jetbrains.mps.newTypesystem.context;
import gnu.trove.THashMap;
import jetbrains.mps.languageScope.LanguageScopeExecutor;
import jetbrains.mps.typesystem.inference.TypeChecker;
import jetbrains.mps.util.Computable;
import jetbrains.mps.util.Pair;
import org.jetbrains.mps.openapi.model.SNode;
import java.util.Map;
/**
* User: fyodor
* Date: 12/7/12
*/
public class CachingTypecheckingContext extends TargetTypecheckingContext {
private Map<SNode, SNode> myComputedTypes = new THashMap<SNode, SNode>(1);
public CachingTypecheckingContext(SNode node, TypeChecker typeChecker) {
super(node, typeChecker);
}
@Override
public SNode getTypeOf_resolveMode(final SNode node, TypeChecker typeChecker) {
Pair <SNode, Boolean> pair = getTypeComputed(node);
if (pair.o2) {
return pair.o1;
}
SNode resultType = LanguageScopeExecutor.execWithModelScope(node.getModel(), new Computable<SNode>() {
@Override
public SNode compute() {
return getTypechecking().computeTypesForNodeDuringResolving(node);
}
});
putTypeComputed(node, resultType);
return resultType;
}
public Pair<SNode, Boolean> getTypeComputed(SNode node) {
if (myComputedTypes != null && myComputedTypes.containsKey(node)) {
return new Pair<SNode, Boolean>(myComputedTypes.get(node), true);
} else {
return new Pair<SNode, Boolean>(null, false);
}
}
public void putTypeComputed(SNode node, SNode type) {
if (myComputedTypes != null) {
myComputedTypes.put(node, type);
}
}
}