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.TestOnly;
025    import org.jetbrains.jet.lang.descriptors.*;
026    import org.jetbrains.jet.lang.descriptors.annotations.AnnotationDescriptor;
027    import org.jetbrains.jet.lang.descriptors.annotations.AnnotationDescriptorImpl;
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, AnnotationDescriptorImpl> ANNOTATION =
077                Slices.<JetAnnotationEntry, AnnotationDescriptorImpl>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<JetElement, ResolvedCall<out CallableDescriptor>>")
091        WritableSlice<JetElement, ResolvedCall<?>> RESOLVED_CALL = new BasicWritableSlice<JetElement, 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        WritableSlice<JetExpression, Call> LOOP_RANGE_ITERATOR_CALL = Slices.createSimpleSlice();
104    
105        WritableSlice<JetExpression, ResolvedCall<FunctionDescriptor>> LOOP_RANGE_HAS_NEXT_RESOLVED_CALL = Slices.createSimpleSlice();
106        WritableSlice<JetExpression, ResolvedCall<FunctionDescriptor>> LOOP_RANGE_NEXT_RESOLVED_CALL = Slices.createSimpleSlice();
107    
108        WritableSlice<PropertyAccessorDescriptor, ResolvedCall<FunctionDescriptor>> DELEGATED_PROPERTY_RESOLVED_CALL = Slices.createSimpleSlice();
109        WritableSlice<PropertyAccessorDescriptor, Call> DELEGATED_PROPERTY_CALL = Slices.createSimpleSlice();
110    
111        WritableSlice<JetMultiDeclarationEntry, ResolvedCall<FunctionDescriptor>> COMPONENT_RESOLVED_CALL = Slices.createSimpleSlice();
112    
113        WritableSlice<JetExpression, ResolvedCall<FunctionDescriptor>> INDEXED_LVALUE_GET = Slices.createSimpleSlice();
114        WritableSlice<JetExpression, ResolvedCall<FunctionDescriptor>> INDEXED_LVALUE_SET = Slices.createSimpleSlice();
115    
116        WritableSlice<JetExpression, JetType> AUTOCAST = Slices.createSimpleSlice();
117    
118        WritableSlice<JetWhenExpression, Boolean> EXHAUSTIVE_WHEN = Slices.createSimpleSlice();
119    
120        /**
121         * A scope where type of expression has been resolved
122         */
123        WritableSlice<JetTypeReference, JetScope> TYPE_RESOLUTION_SCOPE = Slices.createSimpleSlice();
124        WritableSlice<JetExpression, JetScope> RESOLUTION_SCOPE = Slices.createSimpleSlice();
125    
126        WritableSlice<ScriptDescriptor, JetScope> SCRIPT_SCOPE = Slices.createSimpleSlice();
127    
128        /**
129         * Collected during analyze, used in IDE in auto-cast completion
130         */
131        WritableSlice<JetExpression, DataFlowInfo> NON_DEFAULT_EXPRESSION_DATA_FLOW = Slices.createSimpleSlice();
132    
133        WritableSlice<JetExpression, Boolean> VARIABLE_REASSIGNMENT = Slices.createSimpleSetSlice();
134        WritableSlice<ValueParameterDescriptor, Boolean> AUTO_CREATED_IT = Slices.createSimpleSetSlice();
135        WritableSlice<JetExpression, DeclarationDescriptor> VARIABLE_ASSIGNMENT = Slices.createSimpleSlice();
136    
137        /**
138         * Has type of current expression has been already resolved
139         */
140        WritableSlice<JetExpression, Boolean> PROCESSED = Slices.createSimpleSetSlice();
141        WritableSlice<JetElement, Boolean> STATEMENT = Slices.createRemovableSetSlice();
142    
143        WritableSlice<VariableDescriptor, CaptureKind> CAPTURED_IN_CLOSURE = new BasicWritableSlice<VariableDescriptor, CaptureKind>(DO_NOTHING);
144    
145        WritableSlice<CallableMemberDescriptor, Boolean> NEED_SYNTHETIC_ACCESSOR = new BasicWritableSlice<CallableMemberDescriptor, Boolean>(DO_NOTHING);
146    
147        //    enum DeferredTypeKey {DEFERRED_TYPE_KEY}
148        //    WritableSlice<DeferredTypeKey, Collection<DeferredType>> DEFERRED_TYPES = Slices.createSimpleSlice();
149    
150        WritableSlice<Box<DeferredType>, Boolean> DEFERRED_TYPE = Slices.createCollectiveSetSlice();
151    
152        WritableSlice<PropertyDescriptor, Boolean> BACKING_FIELD_REQUIRED = new Slices.SetSlice<PropertyDescriptor>(DO_NOTHING) {
153            @Override
154            public Boolean computeValue(
155                    SlicedMap map,
156                    PropertyDescriptor propertyDescriptor,
157                    Boolean backingFieldRequired,
158                    boolean valueNotFound
159            ) {
160                if (propertyDescriptor.getKind() != CallableMemberDescriptor.Kind.DECLARATION) {
161                    return false;
162                }
163                backingFieldRequired = valueNotFound ? false : backingFieldRequired;
164                assert backingFieldRequired != null;
165                // TODO: user BindingContextAccessors
166                PsiElement declarationPsiElement = map.get(BindingContextUtils.DESCRIPTOR_TO_DECLARATION, propertyDescriptor);
167                if (declarationPsiElement instanceof JetParameter) {
168                    JetParameter jetParameter = (JetParameter) declarationPsiElement;
169                    return jetParameter.getValOrVarNode() != null ||
170                           backingFieldRequired; // this part is unused because we do not allow access to constructor parameters in member bodies
171                }
172                if (propertyDescriptor.getModality() == Modality.ABSTRACT) return false;
173                PropertyGetterDescriptor getter = propertyDescriptor.getGetter();
174                PropertySetterDescriptor setter = propertyDescriptor.getSetter();
175                if (getter == null) {
176                    return true;
177                }
178                else if (propertyDescriptor.isVar() && setter == null) {
179                    return true;
180                }
181                else if (setter != null && !setter.hasBody() && setter.getModality() != Modality.ABSTRACT) {
182                    return true;
183                }
184                else if (!getter.hasBody() && getter.getModality() != Modality.ABSTRACT) {
185                    return true;
186                }
187                return backingFieldRequired;
188            }
189        };
190        WritableSlice<PropertyDescriptor, Boolean> IS_INITIALIZED = Slices.createSimpleSetSlice();
191    
192        WritableSlice<JetFunctionLiteralExpression, Boolean> BLOCK = new Slices.SetSlice<JetFunctionLiteralExpression>(DO_NOTHING) {
193            @Override
194            public Boolean computeValue(SlicedMap map, JetFunctionLiteralExpression expression, Boolean isBlock, boolean valueNotFound) {
195                isBlock = valueNotFound ? false : isBlock;
196                assert isBlock != null;
197                return isBlock && !expression.getFunctionLiteral().hasParameterSpecification();
198            }
199        };
200    
201        WritableSlice<PsiElement, ClassDescriptor> CLASS =
202                Slices.<PsiElement, ClassDescriptor>sliceBuilder().setOpposite((WritableSlice) BindingContextUtils.DESCRIPTOR_TO_DECLARATION)
203                        .build();
204        WritableSlice<PsiElement, ScriptDescriptor> SCRIPT =
205                Slices.<PsiElement, ScriptDescriptor>sliceBuilder().setOpposite((WritableSlice) BindingContextUtils.DESCRIPTOR_TO_DECLARATION)
206                        .build();
207        WritableSlice<JetTypeParameter, TypeParameterDescriptor> TYPE_PARAMETER =
208                Slices.<JetTypeParameter, TypeParameterDescriptor>sliceBuilder()
209                        .setOpposite((WritableSlice) BindingContextUtils.DESCRIPTOR_TO_DECLARATION).build();
210        /**
211         * @see BindingContextUtils#recordFunctionDeclarationToDescriptor(BindingTrace, PsiElement, SimpleFunctionDescriptor)}
212         */
213        WritableSlice<PsiElement, SimpleFunctionDescriptor> FUNCTION = Slices.<PsiElement, SimpleFunctionDescriptor>sliceBuilder()
214                .setOpposite((WritableSlice) BindingContextUtils.DESCRIPTOR_TO_DECLARATION).build();
215        WritableSlice<PsiElement, ConstructorDescriptor> CONSTRUCTOR = Slices.<PsiElement, ConstructorDescriptor>sliceBuilder()
216                .setOpposite((WritableSlice) BindingContextUtils.DESCRIPTOR_TO_DECLARATION).build();
217        WritableSlice<PsiElement, VariableDescriptor> VARIABLE =
218                Slices.<PsiElement, VariableDescriptor>sliceBuilder().setOpposite((WritableSlice) BindingContextUtils.DESCRIPTOR_TO_DECLARATION)
219                        .build();
220        WritableSlice<JetParameter, VariableDescriptor> VALUE_PARAMETER = Slices.<JetParameter, VariableDescriptor>sliceBuilder()
221                .setOpposite((WritableSlice) BindingContextUtils.DESCRIPTOR_TO_DECLARATION).build();
222        WritableSlice<JetPropertyAccessor, PropertyAccessorDescriptor> PROPERTY_ACCESSOR =
223                Slices.<JetPropertyAccessor, PropertyAccessorDescriptor>sliceBuilder()
224                        .setOpposite((WritableSlice) BindingContextUtils.DESCRIPTOR_TO_DECLARATION).build();
225    
226        // normalize value to getOriginal(value)
227        WritableSlice<PsiElement, PropertyDescriptor> PRIMARY_CONSTRUCTOR_PARAMETER =
228                Slices.<PsiElement, PropertyDescriptor>sliceBuilder().setOpposite((WritableSlice) BindingContextUtils.DESCRIPTOR_TO_DECLARATION)
229                        .build();
230    
231        WritableSlice[] DECLARATIONS_TO_DESCRIPTORS = new WritableSlice[] {
232                CLASS, TYPE_PARAMETER, FUNCTION, CONSTRUCTOR, VARIABLE, VALUE_PARAMETER, PROPERTY_ACCESSOR,
233                PRIMARY_CONSTRUCTOR_PARAMETER
234        };
235    
236        ReadOnlySlice<PsiElement, DeclarationDescriptor> DECLARATION_TO_DESCRIPTOR = Slices.<PsiElement, DeclarationDescriptor>sliceBuilder()
237                .setFurtherLookupSlices(DECLARATIONS_TO_DESCRIPTORS).build();
238    
239        WritableSlice<JetReferenceExpression, PsiElement> LABEL_TARGET = Slices.<JetReferenceExpression, PsiElement>sliceBuilder().build();
240        @KotlinSignature("val AMBIGUOUS_LABEL_TARGET: WritableSlice<JetReferenceExpression, Collection<PsiElement>>")
241        WritableSlice<JetReferenceExpression, Collection<? extends PsiElement>> AMBIGUOUS_LABEL_TARGET =
242                Slices.<JetReferenceExpression, Collection<? extends PsiElement>>sliceBuilder().build();
243        WritableSlice<ValueParameterDescriptor, PropertyDescriptor> VALUE_PARAMETER_AS_PROPERTY =
244                Slices.<ValueParameterDescriptor, PropertyDescriptor>sliceBuilder().build();
245    
246        WritableSlice<ValueParameterDescriptor, FunctionDescriptor> DATA_CLASS_COMPONENT_FUNCTION =
247                Slices.<ValueParameterDescriptor, FunctionDescriptor>sliceBuilder().build();
248        WritableSlice<ClassDescriptor, FunctionDescriptor> DATA_CLASS_COPY_FUNCTION =
249                Slices.<ClassDescriptor, FunctionDescriptor>sliceBuilder().build();
250    
251        WritableSlice<FqNameUnsafe, ClassDescriptor> FQNAME_TO_CLASS_DESCRIPTOR = new BasicWritableSlice<FqNameUnsafe, ClassDescriptor>(DO_NOTHING, true);
252        WritableSlice<JetFile, PackageFragmentDescriptor> FILE_TO_PACKAGE_FRAGMENT = Slices.createSimpleSlice();
253        WritableSlice<FqName, Collection<JetFile>> PACKAGE_TO_FILES = Slices.createSimpleSlice();
254    
255        WritableSlice<ClassDescriptor, Boolean> INCOMPLETE_HIERARCHY = Slices.createCollectiveSetSlice();
256    
257        @SuppressWarnings("UnusedDeclaration")
258        @Deprecated // This field is needed only for the side effects of its initializer
259                Void _static_initializer = BasicWritableSlice.initSliceDebugNames(BindingContext.class);
260    
261        @NotNull
262        Diagnostics getDiagnostics();
263    
264        @Nullable
265        <K, V> V get(ReadOnlySlice<K, V> slice, K key);
266    
267        // slice.isCollective() must be true
268        @NotNull
269        <K, V> Collection<K> getKeys(WritableSlice<K, V> slice);
270    
271        /** This method should be used only for debug and testing */
272        @TestOnly
273        @NotNull
274        <K, V> ImmutableMap<K, V> getSliceContents(@NotNull ReadOnlySlice<K, V> slice);
275    }