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;
018    
019    import com.google.common.collect.ImmutableMap;
020    import com.intellij.psi.PsiElement;
021    import jet.runtime.typeinfo.KotlinSignature;
022    import org.jetbrains.annotations.NotNull;
023    import org.jetbrains.annotations.Nullable;
024    import org.jetbrains.annotations.ReadOnly;
025    import org.jetbrains.annotations.TestOnly;
026    import org.jetbrains.jet.lang.descriptors.*;
027    import org.jetbrains.jet.lang.descriptors.annotations.AnnotationDescriptor;
028    import org.jetbrains.jet.lang.psi.*;
029    import org.jetbrains.jet.lang.resolve.calls.TailRecursionKind;
030    import org.jetbrains.jet.lang.resolve.calls.autocasts.DataFlowInfo;
031    import org.jetbrains.jet.lang.resolve.calls.inference.ConstraintSystemCompleter;
032    import org.jetbrains.jet.lang.resolve.calls.model.ResolvedCall;
033    import org.jetbrains.jet.lang.resolve.constants.CompileTimeConstant;
034    import org.jetbrains.jet.lang.resolve.name.FqName;
035    import org.jetbrains.jet.lang.resolve.name.FqNameUnsafe;
036    import org.jetbrains.jet.lang.resolve.scopes.JetScope;
037    import org.jetbrains.jet.lang.types.DeferredType;
038    import org.jetbrains.jet.lang.types.JetType;
039    import org.jetbrains.jet.lang.types.expressions.CaptureKind;
040    import org.jetbrains.jet.util.Box;
041    import org.jetbrains.jet.util.slicedmap.*;
042    
043    import java.util.Collection;
044    import java.util.Collections;
045    
046    import static org.jetbrains.jet.util.slicedmap.RewritePolicy.DO_NOTHING;
047    
048    public interface BindingContext {
049        BindingContext EMPTY = new BindingContext() {
050            @NotNull
051            @Override
052            public Diagnostics getDiagnostics() {
053                return Diagnostics.EMPTY;
054            }
055    
056            @Override
057            public <K, V> V get(ReadOnlySlice<K, V> slice, K key) {
058                return null;
059            }
060    
061            @NotNull
062            @Override
063            public <K, V> Collection<K> getKeys(WritableSlice<K, V> slice) {
064                return Collections.emptyList();
065            }
066    
067            @NotNull
068            @TestOnly
069            @Override
070            public <K, V> ImmutableMap<K, V> getSliceContents(@NotNull ReadOnlySlice<K, V> slice) {
071                return ImmutableMap.of();
072            }
073        };
074    
075        WritableSlice<AnnotationDescriptor, JetAnnotationEntry> ANNOTATION_DESCRIPTOR_TO_PSI_ELEMENT = Slices.createSimpleSlice();
076        WritableSlice<JetAnnotationEntry, AnnotationDescriptor> ANNOTATION =
077                Slices.<JetAnnotationEntry, AnnotationDescriptor>sliceBuilder().setOpposite(ANNOTATION_DESCRIPTOR_TO_PSI_ELEMENT).build();
078    
079        WritableSlice<JetExpression, CompileTimeConstant<?>> COMPILE_TIME_VALUE = Slices.createSimpleSlice();
080    
081        WritableSlice<JetTypeReference, JetType> TYPE = Slices.createSimpleSlice();
082        WritableSlice<JetExpression, JetType> EXPRESSION_TYPE = new BasicWritableSlice<JetExpression, JetType>(DO_NOTHING);
083        WritableSlice<JetExpression, JetType> EXPECTED_EXPRESSION_TYPE = new BasicWritableSlice<JetExpression, JetType>(DO_NOTHING);
084        WritableSlice<JetExpression, DataFlowInfo> EXPRESSION_DATA_FLOW_INFO = new BasicWritableSlice<JetExpression, DataFlowInfo>(DO_NOTHING);
085        WritableSlice<JetExpression, DataFlowInfo> DATAFLOW_INFO_AFTER_CONDITION = Slices.createSimpleSlice();
086    
087        WritableSlice<JetReferenceExpression, DeclarationDescriptor> REFERENCE_TARGET =
088                new BasicWritableSlice<JetReferenceExpression, DeclarationDescriptor>(DO_NOTHING);
089    
090        @KotlinSignature("val RESOLVED_CALL: WritableSlice<Call, ResolvedCall<out CallableDescriptor>>")
091        WritableSlice<Call, ResolvedCall<?>> RESOLVED_CALL = new BasicWritableSlice<Call, ResolvedCall<?>>(DO_NOTHING);
092        WritableSlice<ResolvedCall<?>, TailRecursionKind> TAIL_RECURSION_CALL = Slices.createSimpleSlice();
093        WritableSlice<JetElement, ConstraintSystemCompleter> CONSTRAINT_SYSTEM_COMPLETER = new BasicWritableSlice<JetElement, ConstraintSystemCompleter>(DO_NOTHING);
094        WritableSlice<JetElement, Call> CALL = new BasicWritableSlice<JetElement, Call>(DO_NOTHING);
095    
096        @KotlinSignature("val AMBIGUOUS_REFERENCE_TARGET: WritableSlice<JetExpression, Collection<DeclarationDescriptor>>")
097        WritableSlice<JetExpression, Collection<? extends DeclarationDescriptor>> AMBIGUOUS_REFERENCE_TARGET =
098                new BasicWritableSlice<JetExpression, Collection<? extends DeclarationDescriptor>>(DO_NOTHING);
099    
100        WritableSlice<JetExpression, DelegatingBindingTrace> TRACE_DELTAS_CACHE = Slices.createSimpleSlice();
101    
102        WritableSlice<JetExpression, ResolvedCall<FunctionDescriptor>> LOOP_RANGE_ITERATOR_RESOLVED_CALL = Slices.createSimpleSlice();
103    
104        WritableSlice<JetExpression, ResolvedCall<FunctionDescriptor>> LOOP_RANGE_HAS_NEXT_RESOLVED_CALL = Slices.createSimpleSlice();
105        WritableSlice<JetExpression, ResolvedCall<FunctionDescriptor>> LOOP_RANGE_NEXT_RESOLVED_CALL = Slices.createSimpleSlice();
106    
107        WritableSlice<PropertyAccessorDescriptor, ResolvedCall<FunctionDescriptor>> DELEGATED_PROPERTY_RESOLVED_CALL = Slices.createSimpleSlice();
108        WritableSlice<PropertyAccessorDescriptor, Call> DELEGATED_PROPERTY_CALL = Slices.createSimpleSlice();
109    
110        WritableSlice<JetMultiDeclarationEntry, ResolvedCall<FunctionDescriptor>> COMPONENT_RESOLVED_CALL = Slices.createSimpleSlice();
111    
112        WritableSlice<JetExpression, ResolvedCall<FunctionDescriptor>> INDEXED_LVALUE_GET = Slices.createSimpleSlice();
113        WritableSlice<JetExpression, ResolvedCall<FunctionDescriptor>> INDEXED_LVALUE_SET = Slices.createSimpleSlice();
114    
115        WritableSlice<JetExpression, JetType> AUTOCAST = Slices.createSimpleSlice();
116    
117        WritableSlice<JetWhenExpression, Boolean> EXHAUSTIVE_WHEN = Slices.createSimpleSlice();
118    
119        /**
120         * A scope where type of expression has been resolved
121         */
122        WritableSlice<JetTypeReference, JetScope> TYPE_RESOLUTION_SCOPE = Slices.createSimpleSlice();
123        WritableSlice<JetExpression, JetScope> RESOLUTION_SCOPE = Slices.createSimpleSlice();
124    
125        WritableSlice<ScriptDescriptor, JetScope> SCRIPT_SCOPE = Slices.createSimpleSlice();
126    
127        /**
128         * Collected during analyze, used in IDE in auto-cast completion
129         */
130        WritableSlice<JetExpression, DataFlowInfo> NON_DEFAULT_EXPRESSION_DATA_FLOW = Slices.createSimpleSlice();
131    
132        WritableSlice<JetExpression, Boolean> VARIABLE_REASSIGNMENT = Slices.createSimpleSetSlice();
133        WritableSlice<ValueParameterDescriptor, Boolean> AUTO_CREATED_IT = Slices.createSimpleSetSlice();
134    
135        /**
136         * Has type of current expression has been already resolved
137         */
138        WritableSlice<JetExpression, Boolean> PROCESSED = Slices.createSimpleSetSlice();
139        WritableSlice<JetElement, Boolean> USED_AS_EXPRESSION = Slices.createRemovableSetSlice();
140    
141        WritableSlice<VariableDescriptor, CaptureKind> CAPTURED_IN_CLOSURE = new BasicWritableSlice<VariableDescriptor, CaptureKind>(DO_NOTHING);
142    
143        WritableSlice<CallableMemberDescriptor, Boolean> NEED_SYNTHETIC_ACCESSOR = new BasicWritableSlice<CallableMemberDescriptor, Boolean>(DO_NOTHING);
144    
145        WritableSlice<Box<DeferredType>, Boolean> DEFERRED_TYPE = Slices.createCollectiveSetSlice();
146    
147        WritableSlice<PropertyDescriptor, Boolean> BACKING_FIELD_REQUIRED = new Slices.SetSlice<PropertyDescriptor>(DO_NOTHING) {
148            @Override
149            public Boolean computeValue(
150                    SlicedMap map,
151                    PropertyDescriptor propertyDescriptor,
152                    Boolean backingFieldRequired,
153                    boolean valueNotFound
154            ) {
155                if (propertyDescriptor.getKind() != CallableMemberDescriptor.Kind.DECLARATION) {
156                    return false;
157                }
158                backingFieldRequired = valueNotFound ? false : backingFieldRequired;
159                PsiElement declarationPsiElement = DescriptorToSourceUtils.descriptorToDeclaration(propertyDescriptor);
160                if (declarationPsiElement instanceof JetParameter) {
161                    JetParameter jetParameter = (JetParameter) declarationPsiElement;
162                    return jetParameter.hasValOrVarNode() ||
163                           backingFieldRequired; // this part is unused because we do not allow access to constructor parameters in member bodies
164                }
165                if (propertyDescriptor.getModality() == Modality.ABSTRACT) return false;
166                PropertyGetterDescriptor getter = propertyDescriptor.getGetter();
167                PropertySetterDescriptor setter = propertyDescriptor.getSetter();
168                if (getter == null) {
169                    return true;
170                }
171                else if (propertyDescriptor.isVar() && setter == null) {
172                    return true;
173                }
174                else if (setter != null && !setter.hasBody() && setter.getModality() != Modality.ABSTRACT) {
175                    return true;
176                }
177                else if (!getter.hasBody() && getter.getModality() != Modality.ABSTRACT) {
178                    return true;
179                }
180                return backingFieldRequired;
181            }
182        };
183        WritableSlice<PropertyDescriptor, Boolean> IS_INITIALIZED = Slices.createSimpleSetSlice();
184    
185        WritableSlice<JetFunctionLiteralExpression, Boolean> BLOCK = new Slices.SetSlice<JetFunctionLiteralExpression>(DO_NOTHING) {
186            @Override
187            public Boolean computeValue(SlicedMap map, JetFunctionLiteralExpression expression, Boolean isBlock, boolean valueNotFound) {
188                isBlock = valueNotFound ? false : isBlock;
189                assert isBlock != null;
190                return isBlock && !expression.getFunctionLiteral().hasParameterSpecification();
191            }
192        };
193    
194        WritableSlice<PsiElement, ClassDescriptor> CLASS = Slices.<PsiElement, ClassDescriptor>sliceBuilder().build();
195        WritableSlice<PsiElement, ScriptDescriptor> SCRIPT = Slices.<PsiElement, ScriptDescriptor>sliceBuilder().build();
196        WritableSlice<JetTypeParameter, TypeParameterDescriptor> TYPE_PARAMETER =
197                Slices.<JetTypeParameter, TypeParameterDescriptor>sliceBuilder().build();
198        /**
199         * @see BindingContextUtils#recordFunctionDeclarationToDescriptor(BindingTrace, PsiElement, SimpleFunctionDescriptor)}
200         */
201        WritableSlice<PsiElement, SimpleFunctionDescriptor> FUNCTION = Slices.<PsiElement, SimpleFunctionDescriptor>sliceBuilder().build();
202        WritableSlice<PsiElement, ConstructorDescriptor> CONSTRUCTOR = Slices.<PsiElement, ConstructorDescriptor>sliceBuilder().build();
203        WritableSlice<PsiElement, VariableDescriptor> VARIABLE = Slices.<PsiElement, VariableDescriptor>sliceBuilder().build();
204        WritableSlice<JetParameter, VariableDescriptor> VALUE_PARAMETER = Slices.<JetParameter, VariableDescriptor>sliceBuilder().build();
205        WritableSlice<JetPropertyAccessor, PropertyAccessorDescriptor> PROPERTY_ACCESSOR =
206                Slices.<JetPropertyAccessor, PropertyAccessorDescriptor>sliceBuilder().build();
207        WritableSlice<PsiElement, PropertyDescriptor> PRIMARY_CONSTRUCTOR_PARAMETER =
208                Slices.<PsiElement, PropertyDescriptor>sliceBuilder().build();
209    
210        WritableSlice[] DECLARATIONS_TO_DESCRIPTORS = new WritableSlice[] {
211                CLASS, TYPE_PARAMETER, FUNCTION, CONSTRUCTOR, VARIABLE, VALUE_PARAMETER, PROPERTY_ACCESSOR,
212                PRIMARY_CONSTRUCTOR_PARAMETER
213        };
214    
215        ReadOnlySlice<PsiElement, DeclarationDescriptor> DECLARATION_TO_DESCRIPTOR = Slices.<PsiElement, DeclarationDescriptor>sliceBuilder()
216                .setFurtherLookupSlices(DECLARATIONS_TO_DESCRIPTORS).build();
217    
218        WritableSlice<JetReferenceExpression, PsiElement> LABEL_TARGET = Slices.<JetReferenceExpression, PsiElement>sliceBuilder().build();
219        @KotlinSignature("val AMBIGUOUS_LABEL_TARGET: WritableSlice<JetReferenceExpression, Collection<PsiElement>>")
220        WritableSlice<JetReferenceExpression, Collection<? extends PsiElement>> AMBIGUOUS_LABEL_TARGET =
221                Slices.<JetReferenceExpression, Collection<? extends PsiElement>>sliceBuilder().build();
222        WritableSlice<ValueParameterDescriptor, PropertyDescriptor> VALUE_PARAMETER_AS_PROPERTY =
223                Slices.<ValueParameterDescriptor, PropertyDescriptor>sliceBuilder().build();
224    
225        WritableSlice<ValueParameterDescriptor, FunctionDescriptor> DATA_CLASS_COMPONENT_FUNCTION =
226                Slices.<ValueParameterDescriptor, FunctionDescriptor>sliceBuilder().build();
227        WritableSlice<ClassDescriptor, FunctionDescriptor> DATA_CLASS_COPY_FUNCTION =
228                Slices.<ClassDescriptor, FunctionDescriptor>sliceBuilder().build();
229    
230        WritableSlice<FqNameUnsafe, ClassDescriptor> FQNAME_TO_CLASS_DESCRIPTOR = new BasicWritableSlice<FqNameUnsafe, ClassDescriptor>(DO_NOTHING, true);
231        WritableSlice<JetFile, PackageFragmentDescriptor> FILE_TO_PACKAGE_FRAGMENT = Slices.createSimpleSlice();
232        WritableSlice<FqName, Collection<JetFile>> PACKAGE_TO_FILES = Slices.createSimpleSlice();
233    
234        WritableSlice<ClassDescriptor, Boolean> INCOMPLETE_HIERARCHY = Slices.createCollectiveSetSlice();
235    
236        @SuppressWarnings("UnusedDeclaration")
237        @Deprecated // This field is needed only for the side effects of its initializer
238                Void _static_initializer = BasicWritableSlice.initSliceDebugNames(BindingContext.class);
239    
240        @NotNull
241        Diagnostics getDiagnostics();
242    
243        @Nullable
244        <K, V> V get(ReadOnlySlice<K, V> slice, K key);
245    
246        // slice.isCollective() must be true
247        @NotNull
248        @ReadOnly
249        <K, V> Collection<K> getKeys(WritableSlice<K, V> slice);
250    
251        /** This method should be used only for debug and testing */
252        @TestOnly
253        @NotNull
254        <K, V> ImmutableMap<K, V> getSliceContents(@NotNull ReadOnlySlice<K, V> slice);
255    }