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
017package org.jetbrains.jet.lang.resolve.calls.tasks;
018
019import com.google.common.collect.Lists;
020import org.jetbrains.annotations.NotNull;
021import org.jetbrains.annotations.Nullable;
022import org.jetbrains.jet.lang.descriptors.CallableDescriptor;
023import org.jetbrains.jet.lang.resolve.scopes.receivers.ReceiverValue;
024
025import java.util.Collection;
026import java.util.List;
027
028import static org.jetbrains.jet.lang.resolve.scopes.receivers.ReceiverValue.NO_RECEIVER;
029
030public class ResolutionCandidate<D extends CallableDescriptor> {
031    private final D candidateDescriptor;
032    private ReceiverValue thisObject; // receiver object of a method
033    private ReceiverValue receiverArgument; // receiver of an extension function
034    private ExplicitReceiverKind explicitReceiverKind;
035    private Boolean isSafeCall;
036
037    private ResolutionCandidate(@NotNull D descriptor, @NotNull ReceiverValue thisObject, @NotNull ReceiverValue receiverArgument,
038            @NotNull ExplicitReceiverKind explicitReceiverKind, @Nullable Boolean isSafeCall) {
039        this.candidateDescriptor = descriptor;
040        this.thisObject = thisObject;
041        this.receiverArgument = receiverArgument;
042        this.explicitReceiverKind = explicitReceiverKind;
043        this.isSafeCall = isSafeCall;
044    }
045
046    /*package*/ static <D extends CallableDescriptor> ResolutionCandidate<D> create(@NotNull D descriptor) {
047        return new ResolutionCandidate<D>(descriptor, NO_RECEIVER, NO_RECEIVER, ExplicitReceiverKind.NO_EXPLICIT_RECEIVER, null);
048    }
049
050    public static <D extends CallableDescriptor> ResolutionCandidate<D> create(@NotNull D descriptor, boolean isSafeCall) {
051        return create(descriptor, NO_RECEIVER, NO_RECEIVER, ExplicitReceiverKind.NO_EXPLICIT_RECEIVER, isSafeCall);
052    }
053
054    public static <D extends CallableDescriptor> ResolutionCandidate<D> create(@NotNull D descriptor, @NotNull ReceiverValue thisObject,
055            @NotNull ReceiverValue receiverArgument, @NotNull ExplicitReceiverKind explicitReceiverKind, boolean isSafeCall) {
056        return new ResolutionCandidate<D>(descriptor, thisObject, receiverArgument, explicitReceiverKind, isSafeCall);
057    }
058
059    public void setThisObject(@NotNull ReceiverValue thisObject) {
060        this.thisObject = thisObject;
061    }
062
063    public void setReceiverArgument(@NotNull ReceiverValue receiverArgument) {
064        this.receiverArgument = receiverArgument;
065    }
066
067    public void setExplicitReceiverKind(@NotNull ExplicitReceiverKind explicitReceiverKind) {
068        this.explicitReceiverKind = explicitReceiverKind;
069    }
070
071    @NotNull
072    public D getDescriptor() {
073        return candidateDescriptor;
074    }
075
076    @NotNull
077    public ReceiverValue getThisObject() {
078        return thisObject;
079    }
080
081    @NotNull
082    public ReceiverValue getReceiverArgument() {
083        return receiverArgument;
084    }
085
086    @NotNull
087    public ExplicitReceiverKind getExplicitReceiverKind() {
088        return explicitReceiverKind;
089    }
090
091    @NotNull
092    public static <D extends CallableDescriptor> List<ResolutionCandidate<D>> convertCollection(@NotNull Collection<? extends D> descriptors, boolean isSafeCall) {
093        List<ResolutionCandidate<D>> result = Lists.newArrayList();
094        for (D descriptor : descriptors) {
095            result.add(create(descriptor, isSafeCall));
096        }
097        return result;
098    }
099
100    public void setSafeCall(boolean safeCall) {
101        assert isSafeCall == null;
102        isSafeCall = safeCall;
103    }
104
105    public boolean isSafeCall() {
106        assert isSafeCall != null;
107        return isSafeCall;
108    }
109
110    @Override
111    public String toString() {
112        return candidateDescriptor.toString();
113    }
114}