Converted to Kotlin
This commit is contained in:
Valentin Kipyatkov
@@ -1,433 +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 com.google.common.collect.*;
|
||||
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.checker.JetTypeChecker;
|
||||
import org.jetbrains.jet.utils.Printer;
|
||||
|
||||
import java.util.*;
|
||||
|
||||
// Reads from:
|
||||
// 1. Maps
|
||||
// 2. Worker
|
||||
// 3. Imports
|
||||
|
||||
// Writes to: maps
|
||||
|
||||
public class WritableScopeImpl extends WritableScopeWithImports {
|
||||
|
||||
private final Collection<DeclarationDescriptor> allDescriptors = Lists.newArrayList();
|
||||
private final Multimap<Name, DeclarationDescriptor> declaredDescriptorsAccessibleBySimpleName = HashMultimap.create();
|
||||
private boolean allDescriptorsDone = false;
|
||||
|
||||
@NotNull
|
||||
private final DeclarationDescriptor ownerDeclarationDescriptor;
|
||||
|
||||
@Nullable
|
||||
private SetMultimap<Name, FunctionDescriptor> functionGroups;
|
||||
|
||||
@Nullable
|
||||
private Map<Name, DeclarationDescriptor> variableOrClassDescriptors;
|
||||
|
||||
@Nullable
|
||||
private SetMultimap<Name, VariableDescriptor> propertyGroups;
|
||||
|
||||
@Nullable
|
||||
private Map<Name, PackageViewDescriptor> packageAliases;
|
||||
|
||||
@Nullable
|
||||
private Map<Name, List<DeclarationDescriptor>> labelsToDescriptors;
|
||||
|
||||
@Nullable
|
||||
private ReceiverParameterDescriptor implicitReceiver;
|
||||
|
||||
public WritableScopeImpl(@NotNull JetScope scope, @NotNull DeclarationDescriptor owner,
|
||||
@NotNull RedeclarationHandler redeclarationHandler, @NotNull String debugName) {
|
||||
super(scope, redeclarationHandler, debugName);
|
||||
this.ownerDeclarationDescriptor = owner;
|
||||
}
|
||||
|
||||
@NotNull
|
||||
@Override
|
||||
public DeclarationDescriptor getContainingDeclaration() {
|
||||
return ownerDeclarationDescriptor;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void importScope(@NotNull JetScope imported) {
|
||||
checkMayWrite();
|
||||
super.importScope(imported);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void importClassifierAlias(@NotNull Name importedClassifierName, @NotNull ClassifierDescriptor classifierDescriptor) {
|
||||
checkMayWrite();
|
||||
|
||||
allDescriptors.add(classifierDescriptor);
|
||||
super.importClassifierAlias(importedClassifierName, classifierDescriptor);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void importPackageAlias(@NotNull Name aliasName, @NotNull PackageViewDescriptor packageView) {
|
||||
checkMayWrite();
|
||||
|
||||
allDescriptors.add(packageView);
|
||||
super.importPackageAlias(aliasName, packageView);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void importFunctionAlias(@NotNull Name aliasName, @NotNull FunctionDescriptor functionDescriptor) {
|
||||
checkMayWrite();
|
||||
|
||||
addFunctionDescriptor(functionDescriptor);
|
||||
super.importFunctionAlias(aliasName, functionDescriptor);
|
||||
|
||||
}
|
||||
|
||||
@Override
|
||||
public void importVariableAlias(@NotNull Name aliasName, @NotNull VariableDescriptor variableDescriptor) {
|
||||
checkMayWrite();
|
||||
|
||||
addPropertyDescriptor(variableDescriptor);
|
||||
super.importVariableAlias(aliasName, variableDescriptor);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void clearImports() {
|
||||
checkMayWrite();
|
||||
|
||||
super.clearImports();
|
||||
}
|
||||
|
||||
@NotNull
|
||||
@Override
|
||||
public Collection<DeclarationDescriptor> getAllDescriptors() {
|
||||
checkMayRead();
|
||||
|
||||
if (!allDescriptorsDone) {
|
||||
allDescriptorsDone = true;
|
||||
|
||||
// make sure no descriptors added to allDescriptors collection
|
||||
changeLockLevel(LockLevel.READING);
|
||||
|
||||
allDescriptors.addAll(getWorkerScope().getAllDescriptors());
|
||||
for (JetScope imported : getImports()) {
|
||||
allDescriptors.addAll(imported.getAllDescriptors());
|
||||
}
|
||||
}
|
||||
return allDescriptors;
|
||||
}
|
||||
|
||||
@NotNull
|
||||
private Map<Name, List<DeclarationDescriptor>> getLabelsToDescriptors() {
|
||||
if (labelsToDescriptors == null) {
|
||||
labelsToDescriptors = new HashMap<Name, List<DeclarationDescriptor>>();
|
||||
}
|
||||
return labelsToDescriptors;
|
||||
}
|
||||
|
||||
@NotNull
|
||||
@Override
|
||||
public Collection<DeclarationDescriptor> getDeclarationsByLabel(@NotNull Name labelName) {
|
||||
checkMayRead();
|
||||
|
||||
Collection<DeclarationDescriptor> superResult = super.getDeclarationsByLabel(labelName);
|
||||
Map<Name, List<DeclarationDescriptor>> labelsToDescriptors = getLabelsToDescriptors();
|
||||
List<DeclarationDescriptor> declarationDescriptors = labelsToDescriptors.get(labelName);
|
||||
if (declarationDescriptors == null) {
|
||||
return superResult;
|
||||
}
|
||||
if (superResult.isEmpty()) return declarationDescriptors;
|
||||
List<DeclarationDescriptor> result = new ArrayList<DeclarationDescriptor>(declarationDescriptors);
|
||||
result.addAll(superResult);
|
||||
return result;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void addLabeledDeclaration(@NotNull DeclarationDescriptor descriptor) {
|
||||
checkMayWrite();
|
||||
|
||||
Map<Name, List<DeclarationDescriptor>> labelsToDescriptors = getLabelsToDescriptors();
|
||||
Name name = descriptor.getName();
|
||||
List<DeclarationDescriptor> declarationDescriptors = labelsToDescriptors.get(name);
|
||||
if (declarationDescriptors == null) {
|
||||
declarationDescriptors = new ArrayList<DeclarationDescriptor>();
|
||||
labelsToDescriptors.put(name, declarationDescriptors);
|
||||
}
|
||||
declarationDescriptors.add(descriptor);
|
||||
}
|
||||
|
||||
@NotNull
|
||||
private Map<Name, DeclarationDescriptor> getVariableOrClassDescriptors() {
|
||||
if (variableOrClassDescriptors == null) {
|
||||
variableOrClassDescriptors = Maps.newHashMap();
|
||||
}
|
||||
return variableOrClassDescriptors;
|
||||
}
|
||||
|
||||
@NotNull
|
||||
private Map<Name, PackageViewDescriptor> getPackageAliases() {
|
||||
if (packageAliases == null) {
|
||||
packageAliases = Maps.newHashMap();
|
||||
}
|
||||
return packageAliases;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void addVariableDescriptor(@NotNull VariableDescriptor variableDescriptor) {
|
||||
addVariableDescriptor(variableDescriptor, false);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void addPropertyDescriptor(@NotNull VariableDescriptor propertyDescriptor) {
|
||||
addVariableDescriptor(propertyDescriptor, true);
|
||||
}
|
||||
|
||||
private void addVariableDescriptor(@NotNull VariableDescriptor variableDescriptor, boolean isProperty) {
|
||||
checkMayWrite();
|
||||
|
||||
Name name = variableDescriptor.getName();
|
||||
if (isProperty) {
|
||||
checkForPropertyRedeclaration(name, variableDescriptor);
|
||||
getPropertyGroups().put(name, variableDescriptor);
|
||||
}
|
||||
if (variableDescriptor.getExtensionReceiverParameter() == null) {
|
||||
checkForRedeclaration(name, variableDescriptor);
|
||||
// TODO : Should this always happen?
|
||||
getVariableOrClassDescriptors().put(name, variableDescriptor);
|
||||
}
|
||||
allDescriptors.add(variableDescriptor);
|
||||
addToDeclared(variableDescriptor);
|
||||
}
|
||||
|
||||
@NotNull
|
||||
@Override
|
||||
public Set<VariableDescriptor> getProperties(@NotNull Name name) {
|
||||
checkMayRead();
|
||||
|
||||
Set<VariableDescriptor> result = Sets.newLinkedHashSet(getPropertyGroups().get(name));
|
||||
|
||||
result.addAll(getWorkerScope().getProperties(name));
|
||||
|
||||
result.addAll(super.getProperties(name));
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
@Override
|
||||
public VariableDescriptor getLocalVariable(@NotNull Name name) {
|
||||
checkMayRead();
|
||||
|
||||
Map<Name, DeclarationDescriptor> variableOrClassDescriptors = getVariableOrClassDescriptors();
|
||||
DeclarationDescriptor descriptor = variableOrClassDescriptors.get(name);
|
||||
if (descriptor instanceof VariableDescriptor && !getPropertyGroups().get(name).contains(descriptor)) {
|
||||
return (VariableDescriptor) descriptor;
|
||||
}
|
||||
|
||||
VariableDescriptor variableDescriptor = getWorkerScope().getLocalVariable(name);
|
||||
if (variableDescriptor != null) {
|
||||
return variableDescriptor;
|
||||
}
|
||||
return super.getLocalVariable(name);
|
||||
}
|
||||
|
||||
@NotNull
|
||||
private SetMultimap<Name, VariableDescriptor> getPropertyGroups() {
|
||||
if (propertyGroups == null) {
|
||||
propertyGroups = LinkedHashMultimap.create();
|
||||
}
|
||||
return propertyGroups;
|
||||
}
|
||||
|
||||
@NotNull
|
||||
private SetMultimap<Name, FunctionDescriptor> getFunctionGroups() {
|
||||
if (functionGroups == null) {
|
||||
functionGroups = LinkedHashMultimap.create();
|
||||
}
|
||||
return functionGroups;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void addFunctionDescriptor(@NotNull FunctionDescriptor functionDescriptor) {
|
||||
checkMayWrite();
|
||||
|
||||
getFunctionGroups().put(functionDescriptor.getName(), functionDescriptor);
|
||||
allDescriptors.add(functionDescriptor);
|
||||
}
|
||||
|
||||
@Override
|
||||
@NotNull
|
||||
public Collection<FunctionDescriptor> getFunctions(@NotNull Name name) {
|
||||
checkMayRead();
|
||||
|
||||
Set<FunctionDescriptor> result = Sets.newLinkedHashSet(getFunctionGroups().get(name));
|
||||
|
||||
result.addAll(getWorkerScope().getFunctions(name));
|
||||
|
||||
result.addAll(super.getFunctions(name));
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void addTypeParameterDescriptor(@NotNull TypeParameterDescriptor typeParameterDescriptor) {
|
||||
checkMayWrite();
|
||||
|
||||
Name name = typeParameterDescriptor.getName();
|
||||
addClassifierAlias(name, typeParameterDescriptor);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void addClassifierDescriptor(@NotNull ClassifierDescriptor classDescriptor) {
|
||||
checkMayWrite();
|
||||
|
||||
addClassifierAlias(classDescriptor.getName(), classDescriptor);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void addClassifierAlias(@NotNull Name name, @NotNull ClassifierDescriptor classifierDescriptor) {
|
||||
checkMayWrite();
|
||||
|
||||
checkForRedeclaration(name, classifierDescriptor);
|
||||
getVariableOrClassDescriptors().put(name, classifierDescriptor);
|
||||
allDescriptors.add(classifierDescriptor);
|
||||
addToDeclared(classifierDescriptor);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void addPackageAlias(@NotNull Name name, @NotNull PackageViewDescriptor packageView) {
|
||||
checkMayWrite();
|
||||
|
||||
checkForRedeclaration(name, packageView);
|
||||
getPackageAliases().put(name, packageView);
|
||||
allDescriptors.add(packageView);
|
||||
addToDeclared(packageView);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void addFunctionAlias(@NotNull Name name, @NotNull FunctionDescriptor functionDescriptor) {
|
||||
checkMayWrite();
|
||||
|
||||
checkForRedeclaration(name, functionDescriptor);
|
||||
getFunctionGroups().put(name, functionDescriptor);
|
||||
allDescriptors.add(functionDescriptor);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void addVariableAlias(@NotNull Name name, @NotNull VariableDescriptor variableDescriptor) {
|
||||
checkMayWrite();
|
||||
|
||||
checkForRedeclaration(name, variableDescriptor);
|
||||
|
||||
getVariableOrClassDescriptors().put(name, variableDescriptor);
|
||||
getPropertyGroups().put(name, variableDescriptor);
|
||||
|
||||
allDescriptors.add(variableDescriptor);
|
||||
addToDeclared(variableDescriptor);
|
||||
}
|
||||
|
||||
private void checkForPropertyRedeclaration(@NotNull Name name, VariableDescriptor variableDescriptor) {
|
||||
Set<VariableDescriptor> properties = getPropertyGroups().get(name);
|
||||
ReceiverParameterDescriptor receiverParameter = variableDescriptor.getExtensionReceiverParameter();
|
||||
for (VariableDescriptor oldProperty : properties) {
|
||||
ReceiverParameterDescriptor receiverParameterForOldVariable = oldProperty.getExtensionReceiverParameter();
|
||||
if (((receiverParameter != null && receiverParameterForOldVariable != null) &&
|
||||
(JetTypeChecker.DEFAULT.equalTypes(receiverParameter.getType(), receiverParameterForOldVariable.getType())))) {
|
||||
getRedeclarationHandler().handleRedeclaration(oldProperty, variableDescriptor);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void checkForRedeclaration(@NotNull Name name, DeclarationDescriptor classifierDescriptor) {
|
||||
DeclarationDescriptor originalDescriptor = getVariableOrClassDescriptors().get(name);
|
||||
if (originalDescriptor != null) {
|
||||
getRedeclarationHandler().handleRedeclaration(originalDescriptor, classifierDescriptor);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public ClassifierDescriptor getClassifier(@NotNull Name name) {
|
||||
checkMayRead();
|
||||
|
||||
Map<Name, DeclarationDescriptor> variableOrClassDescriptors = getVariableOrClassDescriptors();
|
||||
DeclarationDescriptor descriptor = variableOrClassDescriptors.get(name);
|
||||
if (descriptor instanceof ClassifierDescriptor) return (ClassifierDescriptor) descriptor;
|
||||
|
||||
ClassifierDescriptor classifierDescriptor = getWorkerScope().getClassifier(name);
|
||||
if (classifierDescriptor != null) return classifierDescriptor;
|
||||
|
||||
return super.getClassifier(name);
|
||||
}
|
||||
|
||||
@Override
|
||||
public PackageViewDescriptor getPackage(@NotNull Name name) {
|
||||
checkMayRead();
|
||||
|
||||
PackageViewDescriptor aliased = getPackageAliases().get(name);
|
||||
if (aliased != null) return aliased;
|
||||
|
||||
PackageViewDescriptor packageView = getWorkerScope().getPackage(name);
|
||||
if (packageView != null) return packageView;
|
||||
return super.getPackage(name);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void setImplicitReceiver(@NotNull ReceiverParameterDescriptor implicitReceiver) {
|
||||
checkMayWrite();
|
||||
|
||||
if (this.implicitReceiver != null) {
|
||||
throw new UnsupportedOperationException("Receiver redeclared");
|
||||
}
|
||||
this.implicitReceiver = implicitReceiver;
|
||||
}
|
||||
|
||||
@Override
|
||||
protected List<ReceiverParameterDescriptor> computeImplicitReceiversHierarchy() {
|
||||
List<ReceiverParameterDescriptor> implicitReceiverHierarchy = Lists.newArrayList();
|
||||
if (implicitReceiver != null) {
|
||||
implicitReceiverHierarchy.add(implicitReceiver);
|
||||
}
|
||||
implicitReceiverHierarchy.addAll(super.computeImplicitReceiversHierarchy());
|
||||
return implicitReceiverHierarchy;
|
||||
}
|
||||
|
||||
private void addToDeclared(DeclarationDescriptor descriptor) {
|
||||
declaredDescriptorsAccessibleBySimpleName.put(descriptor.getName(), descriptor);
|
||||
}
|
||||
|
||||
@NotNull
|
||||
@Override
|
||||
public Multimap<Name, DeclarationDescriptor> getDeclaredDescriptorsAccessibleBySimpleName() {
|
||||
return declaredDescriptorsAccessibleBySimpleName;
|
||||
}
|
||||
|
||||
@NotNull
|
||||
@Override
|
||||
public Collection<DeclarationDescriptor> getOwnDeclaredDescriptors() {
|
||||
return declaredDescriptorsAccessibleBySimpleName.values();
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void printAdditionalScopeStructure(@NotNull Printer p) {
|
||||
p.println("allDescriptorsDone = ", allDescriptorsDone);
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,382 @@
|
||||
/*
|
||||
* 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 com.google.common.collect.*
|
||||
import org.jetbrains.jet.lang.descriptors.*
|
||||
import org.jetbrains.jet.lang.resolve.name.Name
|
||||
import org.jetbrains.jet.lang.types.checker.JetTypeChecker
|
||||
import org.jetbrains.jet.utils.Printer
|
||||
|
||||
import java.util.*
|
||||
import kotlin.Collection
|
||||
import kotlin.Map
|
||||
import kotlin.List
|
||||
import kotlin.Set
|
||||
|
||||
// Reads from:
|
||||
// 1. Maps
|
||||
// 2. Worker
|
||||
// 3. Imports
|
||||
|
||||
// Writes to: maps
|
||||
|
||||
public class WritableScopeImpl(scope: JetScope,
|
||||
private val ownerDeclarationDescriptor: DeclarationDescriptor,
|
||||
redeclarationHandler: RedeclarationHandler,
|
||||
debugName: String)
|
||||
: WritableScopeWithImports(scope, redeclarationHandler, debugName) {
|
||||
|
||||
private val allDescriptors = ArrayList<DeclarationDescriptor>()
|
||||
private val declaredDescriptorsAccessibleBySimpleName = HashMultimap.create<Name, DeclarationDescriptor>()
|
||||
private var allDescriptorsDone = false
|
||||
|
||||
private var functionGroups: SetMultimap<Name, FunctionDescriptor>? = null
|
||||
|
||||
private var variableOrClassDescriptors: MutableMap<Name, DeclarationDescriptor>? = null
|
||||
|
||||
private var propertyGroups: SetMultimap<Name, VariableDescriptor>? = null
|
||||
|
||||
private var packageAliases: MutableMap<Name, PackageViewDescriptor>? = null
|
||||
|
||||
private var labelsToDescriptors: MutableMap<Name, MutableList<DeclarationDescriptor>>? = null
|
||||
|
||||
private var implicitReceiver: ReceiverParameterDescriptor? = null
|
||||
|
||||
override fun getContainingDeclaration(): DeclarationDescriptor {
|
||||
return ownerDeclarationDescriptor
|
||||
}
|
||||
|
||||
override fun importScope(imported: JetScope) {
|
||||
checkMayWrite()
|
||||
super.importScope(imported)
|
||||
}
|
||||
|
||||
override fun importClassifierAlias(importedClassifierName: Name, classifierDescriptor: ClassifierDescriptor) {
|
||||
checkMayWrite()
|
||||
|
||||
allDescriptors.add(classifierDescriptor)
|
||||
super.importClassifierAlias(importedClassifierName, classifierDescriptor)
|
||||
}
|
||||
|
||||
override fun importPackageAlias(aliasName: Name, packageView: PackageViewDescriptor) {
|
||||
checkMayWrite()
|
||||
|
||||
allDescriptors.add(packageView)
|
||||
super.importPackageAlias(aliasName, packageView)
|
||||
}
|
||||
|
||||
override fun importFunctionAlias(aliasName: Name, functionDescriptor: FunctionDescriptor) {
|
||||
checkMayWrite()
|
||||
|
||||
addFunctionDescriptor(functionDescriptor)
|
||||
super.importFunctionAlias(aliasName, functionDescriptor)
|
||||
|
||||
}
|
||||
|
||||
override fun importVariableAlias(aliasName: Name, variableDescriptor: VariableDescriptor) {
|
||||
checkMayWrite()
|
||||
|
||||
addPropertyDescriptor(variableDescriptor)
|
||||
super.importVariableAlias(aliasName, variableDescriptor)
|
||||
}
|
||||
|
||||
override fun clearImports() {
|
||||
checkMayWrite()
|
||||
|
||||
super.clearImports()
|
||||
}
|
||||
|
||||
override fun getAllDescriptors(): Collection<DeclarationDescriptor> {
|
||||
checkMayRead()
|
||||
|
||||
if (!allDescriptorsDone) {
|
||||
allDescriptorsDone = true
|
||||
|
||||
// make sure no descriptors added to allDescriptors collection
|
||||
changeLockLevel(WritableScope.LockLevel.READING)
|
||||
|
||||
allDescriptors.addAll(workerScope.getAllDescriptors())
|
||||
for (imported in getImports()) {
|
||||
allDescriptors.addAll(imported.getAllDescriptors())
|
||||
}
|
||||
}
|
||||
return allDescriptors
|
||||
}
|
||||
|
||||
private fun getLabelsToDescriptors(): MutableMap<Name, MutableList<DeclarationDescriptor>> {
|
||||
if (labelsToDescriptors == null) {
|
||||
labelsToDescriptors = HashMap()
|
||||
}
|
||||
return labelsToDescriptors!!
|
||||
}
|
||||
|
||||
override fun getDeclarationsByLabel(labelName: Name): Collection<DeclarationDescriptor> {
|
||||
checkMayRead()
|
||||
|
||||
val superResult = super.getDeclarationsByLabel(labelName)
|
||||
val labelsToDescriptors = getLabelsToDescriptors()
|
||||
val declarationDescriptors = labelsToDescriptors.get(labelName)
|
||||
if (declarationDescriptors == null) {
|
||||
return superResult
|
||||
}
|
||||
if (superResult.isEmpty()) return declarationDescriptors
|
||||
val result = ArrayList(declarationDescriptors)
|
||||
result.addAll(superResult)
|
||||
return result
|
||||
}
|
||||
|
||||
override fun addLabeledDeclaration(descriptor: DeclarationDescriptor) {
|
||||
checkMayWrite()
|
||||
|
||||
val labelsToDescriptors = getLabelsToDescriptors()
|
||||
val name = descriptor.getName()
|
||||
var declarationDescriptors = labelsToDescriptors[name]
|
||||
if (declarationDescriptors == null) {
|
||||
declarationDescriptors = ArrayList()
|
||||
labelsToDescriptors.put(name, declarationDescriptors!!)
|
||||
}
|
||||
declarationDescriptors!!.add(descriptor)
|
||||
}
|
||||
|
||||
private fun getVariableOrClassDescriptors(): MutableMap<Name, DeclarationDescriptor> {
|
||||
if (variableOrClassDescriptors == null) {
|
||||
variableOrClassDescriptors = HashMap()
|
||||
}
|
||||
return variableOrClassDescriptors!!
|
||||
}
|
||||
|
||||
private fun getPackageAliases(): MutableMap<Name, PackageViewDescriptor> {
|
||||
if (packageAliases == null) {
|
||||
packageAliases = HashMap()
|
||||
}
|
||||
return packageAliases!!
|
||||
}
|
||||
|
||||
override fun addVariableDescriptor(variableDescriptor: VariableDescriptor) {
|
||||
addVariableDescriptor(variableDescriptor, false)
|
||||
}
|
||||
|
||||
override fun addPropertyDescriptor(propertyDescriptor: VariableDescriptor) {
|
||||
addVariableDescriptor(propertyDescriptor, true)
|
||||
}
|
||||
|
||||
private fun addVariableDescriptor(variableDescriptor: VariableDescriptor, isProperty: Boolean) {
|
||||
checkMayWrite()
|
||||
|
||||
val name = variableDescriptor.getName()
|
||||
if (isProperty) {
|
||||
checkForPropertyRedeclaration(name, variableDescriptor)
|
||||
getPropertyGroups().put(name, variableDescriptor)
|
||||
}
|
||||
if (variableDescriptor.getExtensionReceiverParameter() == null) {
|
||||
checkForRedeclaration(name, variableDescriptor)
|
||||
// TODO : Should this always happen?
|
||||
getVariableOrClassDescriptors().put(name, variableDescriptor)
|
||||
}
|
||||
allDescriptors.add(variableDescriptor)
|
||||
addToDeclared(variableDescriptor)
|
||||
}
|
||||
|
||||
override fun getProperties(name: Name): Set<VariableDescriptor> {
|
||||
checkMayRead()
|
||||
|
||||
val result = Sets.newLinkedHashSet(getPropertyGroups().get(name))
|
||||
|
||||
result.addAll(workerScope.getProperties(name))
|
||||
|
||||
result.addAll(super.getProperties(name))
|
||||
|
||||
return result
|
||||
}
|
||||
|
||||
override fun getLocalVariable(name: Name): VariableDescriptor? {
|
||||
checkMayRead()
|
||||
|
||||
val variableOrClassDescriptors = getVariableOrClassDescriptors()
|
||||
val descriptor = variableOrClassDescriptors.get(name)
|
||||
if (descriptor is VariableDescriptor && !getPropertyGroups().get(name).contains(descriptor)) {
|
||||
return descriptor
|
||||
}
|
||||
|
||||
val variableDescriptor = workerScope.getLocalVariable(name)
|
||||
if (variableDescriptor != null) {
|
||||
return variableDescriptor
|
||||
}
|
||||
return super.getLocalVariable(name)
|
||||
}
|
||||
|
||||
private fun getPropertyGroups(): SetMultimap<Name, VariableDescriptor> {
|
||||
if (propertyGroups == null) {
|
||||
propertyGroups = LinkedHashMultimap.create()
|
||||
}
|
||||
return propertyGroups!!
|
||||
}
|
||||
|
||||
private fun getFunctionGroups(): SetMultimap<Name, FunctionDescriptor> {
|
||||
if (functionGroups == null) {
|
||||
functionGroups = LinkedHashMultimap.create()
|
||||
}
|
||||
return functionGroups!!
|
||||
}
|
||||
|
||||
override fun addFunctionDescriptor(functionDescriptor: FunctionDescriptor) {
|
||||
checkMayWrite()
|
||||
|
||||
getFunctionGroups().put(functionDescriptor.getName(), functionDescriptor)
|
||||
allDescriptors.add(functionDescriptor)
|
||||
}
|
||||
|
||||
override fun getFunctions(name: Name): Collection<FunctionDescriptor> {
|
||||
checkMayRead()
|
||||
|
||||
val result = Sets.newLinkedHashSet(getFunctionGroups().get(name))
|
||||
|
||||
result.addAll(workerScope.getFunctions(name))
|
||||
|
||||
result.addAll(super.getFunctions(name))
|
||||
|
||||
return result
|
||||
}
|
||||
|
||||
override fun addTypeParameterDescriptor(typeParameterDescriptor: TypeParameterDescriptor) {
|
||||
checkMayWrite()
|
||||
|
||||
val name = typeParameterDescriptor.getName()
|
||||
addClassifierAlias(name, typeParameterDescriptor)
|
||||
}
|
||||
|
||||
override fun addClassifierDescriptor(classDescriptor: ClassifierDescriptor) {
|
||||
checkMayWrite()
|
||||
|
||||
addClassifierAlias(classDescriptor.getName(), classDescriptor)
|
||||
}
|
||||
|
||||
override fun addClassifierAlias(name: Name, classifierDescriptor: ClassifierDescriptor) {
|
||||
checkMayWrite()
|
||||
|
||||
checkForRedeclaration(name, classifierDescriptor)
|
||||
getVariableOrClassDescriptors().put(name, classifierDescriptor)
|
||||
allDescriptors.add(classifierDescriptor)
|
||||
addToDeclared(classifierDescriptor)
|
||||
}
|
||||
|
||||
override fun addPackageAlias(name: Name, packageView: PackageViewDescriptor) {
|
||||
checkMayWrite()
|
||||
|
||||
checkForRedeclaration(name, packageView)
|
||||
getPackageAliases().put(name, packageView)
|
||||
allDescriptors.add(packageView)
|
||||
addToDeclared(packageView)
|
||||
}
|
||||
|
||||
override fun addFunctionAlias(name: Name, functionDescriptor: FunctionDescriptor) {
|
||||
checkMayWrite()
|
||||
|
||||
checkForRedeclaration(name, functionDescriptor)
|
||||
getFunctionGroups().put(name, functionDescriptor)
|
||||
allDescriptors.add(functionDescriptor)
|
||||
}
|
||||
|
||||
override fun addVariableAlias(name: Name, variableDescriptor: VariableDescriptor) {
|
||||
checkMayWrite()
|
||||
|
||||
checkForRedeclaration(name, variableDescriptor)
|
||||
|
||||
getVariableOrClassDescriptors().put(name, variableDescriptor)
|
||||
getPropertyGroups().put(name, variableDescriptor)
|
||||
|
||||
allDescriptors.add(variableDescriptor)
|
||||
addToDeclared(variableDescriptor)
|
||||
}
|
||||
|
||||
private fun checkForPropertyRedeclaration(name: Name, variableDescriptor: VariableDescriptor) {
|
||||
val properties = getPropertyGroups().get(name)
|
||||
val receiverParameter = variableDescriptor.getExtensionReceiverParameter()
|
||||
for (oldProperty in properties) {
|
||||
val receiverParameterForOldVariable = oldProperty.getExtensionReceiverParameter()
|
||||
if (((receiverParameter != null && receiverParameterForOldVariable != null) && (JetTypeChecker.DEFAULT.equalTypes(receiverParameter.getType(), receiverParameterForOldVariable.getType())))) {
|
||||
redeclarationHandler.handleRedeclaration(oldProperty, variableDescriptor)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private fun checkForRedeclaration(name: Name, classifierDescriptor: DeclarationDescriptor) {
|
||||
val originalDescriptor = getVariableOrClassDescriptors().get(name)
|
||||
if (originalDescriptor != null) {
|
||||
redeclarationHandler.handleRedeclaration(originalDescriptor, classifierDescriptor)
|
||||
}
|
||||
}
|
||||
|
||||
override fun getClassifier(name: Name): ClassifierDescriptor? {
|
||||
checkMayRead()
|
||||
|
||||
val variableOrClassDescriptors = getVariableOrClassDescriptors()
|
||||
val descriptor = variableOrClassDescriptors.get(name)
|
||||
if (descriptor is ClassifierDescriptor) return descriptor
|
||||
|
||||
val classifierDescriptor = workerScope.getClassifier(name)
|
||||
if (classifierDescriptor != null) return classifierDescriptor
|
||||
|
||||
return super.getClassifier(name)
|
||||
}
|
||||
|
||||
override fun getPackage(name: Name): PackageViewDescriptor? {
|
||||
checkMayRead()
|
||||
|
||||
val aliased = getPackageAliases().get(name)
|
||||
if (aliased != null) return aliased
|
||||
|
||||
val packageView = workerScope.getPackage(name)
|
||||
if (packageView != null) return packageView
|
||||
return super.getPackage(name)
|
||||
}
|
||||
|
||||
override fun setImplicitReceiver(implicitReceiver: ReceiverParameterDescriptor) {
|
||||
checkMayWrite()
|
||||
|
||||
if (this.implicitReceiver != null) {
|
||||
throw UnsupportedOperationException("Receiver redeclared")
|
||||
}
|
||||
this.implicitReceiver = implicitReceiver
|
||||
}
|
||||
|
||||
override fun computeImplicitReceiversHierarchy(): List<ReceiverParameterDescriptor> {
|
||||
val implicitReceiverHierarchy = ArrayList<ReceiverParameterDescriptor>()
|
||||
if (implicitReceiver != null) {
|
||||
implicitReceiverHierarchy.add(implicitReceiver)
|
||||
}
|
||||
implicitReceiverHierarchy.addAll(super.computeImplicitReceiversHierarchy())
|
||||
return implicitReceiverHierarchy
|
||||
}
|
||||
|
||||
private fun addToDeclared(descriptor: DeclarationDescriptor) {
|
||||
declaredDescriptorsAccessibleBySimpleName.put(descriptor.getName(), descriptor)
|
||||
}
|
||||
|
||||
override fun getDeclaredDescriptorsAccessibleBySimpleName(): Multimap<Name, DeclarationDescriptor> {
|
||||
return declaredDescriptorsAccessibleBySimpleName
|
||||
}
|
||||
|
||||
override fun getOwnDeclaredDescriptors(): Collection<DeclarationDescriptor> {
|
||||
return declaredDescriptorsAccessibleBySimpleName.values()
|
||||
}
|
||||
|
||||
override fun printAdditionalScopeStructure(p: Printer) {
|
||||
p.println("allDescriptorsDone = ", allDescriptorsDone)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user