Converted SubstitutingScope to Kotlin
This commit is contained in:
Valentin Kipyatkov
@@ -1,155 +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.resolve.scopes;
|
||||
|
||||
import org.jetbrains.annotations.NotNull;
|
||||
import org.jetbrains.annotations.Nullable;
|
||||
import org.jetbrains.jet.lang.descriptors.*;
|
||||
import org.jetbrains.jet.lang.resolve.name.Name;
|
||||
import org.jetbrains.jet.lang.types.TypeSubstitutor;
|
||||
import org.jetbrains.jet.utils.Printer;
|
||||
import org.jetbrains.jet.utils.UtilsPackage;
|
||||
|
||||
import java.util.*;
|
||||
|
||||
public class SubstitutingScope implements JetScope {
|
||||
|
||||
private final JetScope workerScope;
|
||||
private final TypeSubstitutor substitutor;
|
||||
|
||||
private Map<DeclarationDescriptor, DeclarationDescriptor> substitutedDescriptors = null;
|
||||
private Collection<DeclarationDescriptor> allDescriptors = null;
|
||||
|
||||
public SubstitutingScope(JetScope workerScope, @NotNull TypeSubstitutor substitutor) {
|
||||
this.workerScope = workerScope;
|
||||
this.substitutor = substitutor;
|
||||
}
|
||||
|
||||
@Nullable
|
||||
private <D extends DeclarationDescriptor> D substitute(@Nullable D descriptor) {
|
||||
if (descriptor == null) return null;
|
||||
if (substitutor.isEmpty()) return descriptor;
|
||||
|
||||
if (substitutedDescriptors == null) {
|
||||
substitutedDescriptors = new HashMap<DeclarationDescriptor, DeclarationDescriptor>();
|
||||
}
|
||||
|
||||
DeclarationDescriptor substituted = substitutedDescriptors.get(descriptor);
|
||||
if (substituted == null && !substitutedDescriptors.containsKey(descriptor)) {
|
||||
substituted = descriptor.substitute(substitutor);
|
||||
|
||||
//noinspection ConstantConditions
|
||||
substitutedDescriptors.put(descriptor, substituted);
|
||||
}
|
||||
|
||||
//noinspection unchecked
|
||||
return (D) substituted;
|
||||
}
|
||||
|
||||
@NotNull
|
||||
private <D extends DeclarationDescriptor> Collection<D> substitute(@NotNull Collection<D> descriptors) {
|
||||
if (substitutor.isEmpty()) return descriptors;
|
||||
if (descriptors.isEmpty()) return descriptors;
|
||||
|
||||
Set<D> result = UtilsPackage.newHashSetWithExpectedSize(descriptors.size());
|
||||
for (D descriptor : descriptors) {
|
||||
D substitute = substitute(descriptor);
|
||||
if (substitute != null) {
|
||||
result.add(substitute);
|
||||
}
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
@NotNull
|
||||
@Override
|
||||
public Collection<VariableDescriptor> getProperties(@NotNull Name name) {
|
||||
return substitute(workerScope.getProperties(name));
|
||||
}
|
||||
|
||||
@Override
|
||||
public VariableDescriptor getLocalVariable(@NotNull Name name) {
|
||||
return substitute(workerScope.getLocalVariable(name));
|
||||
}
|
||||
|
||||
@Override
|
||||
public ClassifierDescriptor getClassifier(@NotNull Name name) {
|
||||
return substitute(workerScope.getClassifier(name));
|
||||
}
|
||||
|
||||
@NotNull
|
||||
@Override
|
||||
public Collection<FunctionDescriptor> getFunctions(@NotNull Name name) {
|
||||
return substitute(workerScope.getFunctions(name));
|
||||
}
|
||||
|
||||
@Override
|
||||
public PackageViewDescriptor getPackage(@NotNull Name name) {
|
||||
return workerScope.getPackage(name);
|
||||
}
|
||||
|
||||
@NotNull
|
||||
@Override
|
||||
public List<ReceiverParameterDescriptor> getImplicitReceiversHierarchy() {
|
||||
throw new UnsupportedOperationException(); // TODO
|
||||
}
|
||||
|
||||
@NotNull
|
||||
@Override
|
||||
public DeclarationDescriptor getContainingDeclaration() {
|
||||
return workerScope.getContainingDeclaration();
|
||||
}
|
||||
|
||||
@NotNull
|
||||
@Override
|
||||
public Collection<DeclarationDescriptor> getDeclarationsByLabel(@NotNull Name labelName) {
|
||||
throw new UnsupportedOperationException(); // TODO
|
||||
}
|
||||
|
||||
@NotNull
|
||||
@Override
|
||||
public Collection<DeclarationDescriptor> getAllDescriptors() {
|
||||
if (allDescriptors == null) {
|
||||
allDescriptors = substitute(workerScope.getAllDescriptors());
|
||||
}
|
||||
return allDescriptors;
|
||||
}
|
||||
|
||||
@NotNull
|
||||
@Override
|
||||
public Collection<DeclarationDescriptor> getOwnDeclaredDescriptors() {
|
||||
return substitute(workerScope.getOwnDeclaredDescriptors());
|
||||
}
|
||||
|
||||
@Override
|
||||
public void printScopeStructure(@NotNull Printer p) {
|
||||
p.println(getClass().getSimpleName(), " {");
|
||||
p.pushIndent();
|
||||
|
||||
p.println("substitutor = ");
|
||||
p.pushIndent();
|
||||
p.println(substitutor);
|
||||
p.popIndent();
|
||||
|
||||
p.print("workerScope = ");
|
||||
workerScope.printScopeStructure(p.withholdIndentOnce());
|
||||
|
||||
p.popIndent();
|
||||
p.println("}");
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,113 @@
|
||||
/*
|
||||
* Copyright 2010-2014 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.resolve.scopes
|
||||
|
||||
import org.jetbrains.jet.lang.descriptors.*
|
||||
import org.jetbrains.jet.lang.resolve.name.Name
|
||||
import org.jetbrains.jet.lang.types.TypeSubstitutor
|
||||
import org.jetbrains.jet.utils.Printer
|
||||
import org.jetbrains.jet.utils.*
|
||||
|
||||
import java.util.*
|
||||
import kotlin.Map
|
||||
import kotlin.Collection
|
||||
import kotlin.List
|
||||
|
||||
public class SubstitutingScope(private val workerScope: JetScope, private val substitutor: TypeSubstitutor) : JetScope {
|
||||
|
||||
private var substitutedDescriptors: MutableMap<DeclarationDescriptor, DeclarationDescriptor?>? = null
|
||||
private var allDescriptors: Collection<DeclarationDescriptor>? = null
|
||||
|
||||
private fun <D : DeclarationDescriptor> substitute(descriptor: D?): D? {
|
||||
if (descriptor == null) return null
|
||||
if (substitutor.isEmpty()) return descriptor
|
||||
|
||||
if (substitutedDescriptors == null) {
|
||||
substitutedDescriptors = HashMap<DeclarationDescriptor, DeclarationDescriptor>()
|
||||
}
|
||||
|
||||
var substituted: DeclarationDescriptor? = substitutedDescriptors!!.get(descriptor)
|
||||
if (substituted == null && !substitutedDescriptors!!.containsKey(descriptor)) {
|
||||
substituted = descriptor.substitute(substitutor)
|
||||
|
||||
substitutedDescriptors!!.put(descriptor, substituted)
|
||||
}
|
||||
|
||||
[suppress("UNCHECKED_CAST")]
|
||||
return substituted as D?
|
||||
}
|
||||
|
||||
private fun <D : DeclarationDescriptor> substitute(descriptors: Collection<D>): Collection<D> {
|
||||
if (substitutor.isEmpty()) return descriptors
|
||||
if (descriptors.isEmpty()) return descriptors
|
||||
|
||||
val result = newHashSetWithExpectedSize<D>(descriptors.size())
|
||||
for (descriptor in descriptors) {
|
||||
val substitute = substitute(descriptor)
|
||||
if (substitute != null) {
|
||||
result.add(substitute)
|
||||
}
|
||||
}
|
||||
|
||||
return result
|
||||
}
|
||||
|
||||
override fun getProperties(name: Name) = substitute(workerScope.getProperties(name))
|
||||
|
||||
override fun getLocalVariable(name: Name) = substitute(workerScope.getLocalVariable(name))
|
||||
|
||||
override fun getClassifier(name: Name) = substitute(workerScope.getClassifier(name))
|
||||
|
||||
override fun getFunctions(name: Name) = substitute(workerScope.getFunctions(name))
|
||||
|
||||
override fun getPackage(name: Name) = workerScope.getPackage(name)
|
||||
|
||||
override fun getImplicitReceiversHierarchy(): List<ReceiverParameterDescriptor> {
|
||||
throw UnsupportedOperationException() // TODO
|
||||
}
|
||||
|
||||
override fun getContainingDeclaration() = workerScope.getContainingDeclaration()
|
||||
|
||||
override fun getDeclarationsByLabel(labelName: Name): Collection<DeclarationDescriptor> {
|
||||
throw UnsupportedOperationException() // TODO
|
||||
}
|
||||
|
||||
override fun getAllDescriptors(): Collection<DeclarationDescriptor> {
|
||||
if (allDescriptors == null) {
|
||||
allDescriptors = substitute(workerScope.getAllDescriptors())
|
||||
}
|
||||
return allDescriptors!!
|
||||
}
|
||||
|
||||
override fun getOwnDeclaredDescriptors() = substitute(workerScope.getOwnDeclaredDescriptors())
|
||||
|
||||
override fun printScopeStructure(p: Printer) {
|
||||
p.println(javaClass.getSimpleName(), " {")
|
||||
p.pushIndent()
|
||||
|
||||
p.println("substitutor = ")
|
||||
p.pushIndent()
|
||||
p.println(substitutor)
|
||||
p.popIndent()
|
||||
|
||||
p.print("workerScope = ")
|
||||
workerScope.printScopeStructure(p.withholdIndentOnce())
|
||||
|
||||
p.popIndent()
|
||||
p.println("}")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user