001    /*
002     * Copyright 2010-2016 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.kotlin.resolve.calls.context;
018    
019    import kotlin.jvm.functions.Function1;
020    import org.jetbrains.annotations.NotNull;
021    import org.jetbrains.annotations.Nullable;
022    import org.jetbrains.kotlin.descriptors.CallableDescriptor;
023    import org.jetbrains.kotlin.psi.Call;
024    import org.jetbrains.kotlin.psi.KtExpression;
025    import org.jetbrains.kotlin.resolve.BindingTrace;
026    import org.jetbrains.kotlin.resolve.StatementFilter;
027    import org.jetbrains.kotlin.resolve.calls.model.MutableDataFlowInfoForArguments;
028    import org.jetbrains.kotlin.resolve.calls.model.MutableResolvedCall;
029    import org.jetbrains.kotlin.resolve.calls.smartcasts.DataFlowInfo;
030    import org.jetbrains.kotlin.resolve.calls.tasks.TracingStrategy;
031    import org.jetbrains.kotlin.resolve.scopes.LexicalScope;
032    import org.jetbrains.kotlin.types.KotlinType;
033    
034    public final class CallCandidateResolutionContext<D extends CallableDescriptor> extends CallResolutionContext<CallCandidateResolutionContext<D>> {
035        @NotNull
036        public final MutableResolvedCall<D> candidateCall;
037        @NotNull
038        public final TracingStrategy tracing;
039        @NotNull
040        public final CandidateResolveMode candidateResolveMode;
041    
042        private CallCandidateResolutionContext(
043                @NotNull MutableResolvedCall<D> candidateCall,
044                @NotNull TracingStrategy tracing,
045                @NotNull BindingTrace trace,
046                @NotNull LexicalScope scope,
047                @NotNull Call call,
048                @NotNull KotlinType expectedType,
049                @NotNull DataFlowInfo dataFlowInfo,
050                @NotNull ContextDependency contextDependency,
051                @NotNull CheckArgumentTypesMode checkArguments,
052                @NotNull ResolutionResultsCache resolutionResultsCache,
053                @Nullable MutableDataFlowInfoForArguments dataFlowInfoForArguments,
054                @NotNull StatementFilter statementFilter,
055                @NotNull CandidateResolveMode candidateResolveMode,
056                boolean isAnnotationContext,
057                boolean isDebuggerContext,
058                boolean collectAllCandidates,
059                @NotNull CallPosition callPosition,
060                @NotNull Function1<KtExpression, KtExpression> expressionContextProvider
061        ) {
062            super(trace, scope, call, expectedType, dataFlowInfo, contextDependency, checkArguments, resolutionResultsCache,
063                  dataFlowInfoForArguments, statementFilter, isAnnotationContext, isDebuggerContext,
064                  collectAllCandidates, callPosition, expressionContextProvider);
065            this.candidateCall = candidateCall;
066            this.tracing = tracing;
067            this.candidateResolveMode = candidateResolveMode;
068        }
069    
070        public static <D extends CallableDescriptor> CallCandidateResolutionContext<D> create(
071                @NotNull MutableResolvedCall<D> candidateCall, @NotNull CallResolutionContext<?> context, @NotNull BindingTrace trace,
072                @NotNull TracingStrategy tracing, @NotNull Call call,
073                @NotNull CandidateResolveMode candidateResolveMode
074        ) {
075            return new CallCandidateResolutionContext<D>(
076                    candidateCall, tracing, trace, context.scope, call, context.expectedType,
077                    context.dataFlowInfo, context.contextDependency, context.checkArguments,
078                    context.resolutionResultsCache, context.dataFlowInfoForArguments,
079                    context.statementFilter,
080                    candidateResolveMode, context.isAnnotationContext, context.isDebuggerContext, context.collectAllCandidates,
081                    context.callPosition, context.expressionContextProvider);
082        }
083    
084        @NotNull
085        public static <D extends CallableDescriptor> CallCandidateResolutionContext<D> createForCallBeingAnalyzed(
086                @NotNull MutableResolvedCall<D> candidateCall, @NotNull BasicCallResolutionContext context, @NotNull TracingStrategy tracing
087        ) {
088            return new CallCandidateResolutionContext<D>(
089                    candidateCall, tracing, context.trace, context.scope, context.call, context.expectedType,
090                    context.dataFlowInfo, context.contextDependency, context.checkArguments, context.resolutionResultsCache,
091                    context.dataFlowInfoForArguments, context.statementFilter,
092                    CandidateResolveMode.FULLY, context.isAnnotationContext, context.isDebuggerContext, context.collectAllCandidates,
093                    context.callPosition, context.expressionContextProvider);
094        }
095    
096        @Override
097        protected CallCandidateResolutionContext<D> create(
098                @NotNull BindingTrace trace,
099                @NotNull LexicalScope scope,
100                @NotNull DataFlowInfo dataFlowInfo,
101                @NotNull KotlinType expectedType,
102                @NotNull ContextDependency contextDependency,
103                @NotNull ResolutionResultsCache resolutionResultsCache,
104                @NotNull StatementFilter statementFilter,
105                boolean collectAllCandidates,
106                @NotNull CallPosition callPosition,
107                @NotNull Function1<KtExpression, KtExpression> expressionContextProvider
108        ) {
109            return new CallCandidateResolutionContext<D>(
110                    candidateCall, tracing, trace, scope, call, expectedType, dataFlowInfo, contextDependency, checkArguments,
111                    resolutionResultsCache, dataFlowInfoForArguments, statementFilter,
112                    candidateResolveMode, isAnnotationContext, isDebuggerContext, collectAllCandidates, callPosition, expressionContextProvider);
113        }
114    }