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.autocasts;
018
019import com.google.common.collect.Lists;
020import org.jetbrains.annotations.NotNull;
021import org.jetbrains.jet.lang.resolve.BindingContext;
022import org.jetbrains.jet.lang.resolve.scopes.receivers.ReceiverValue;
023
024import java.util.List;
025
026public class AutoCastServiceImpl implements AutoCastService {
027    private final DataFlowInfo dataFlowInfo;
028    private final BindingContext bindingContext;
029
030    public AutoCastServiceImpl(DataFlowInfo dataFlowInfo, BindingContext bindingContext) {
031        this.dataFlowInfo = dataFlowInfo;
032        this.bindingContext = bindingContext;
033    }
034
035    @NotNull
036    @Override
037    public List<ReceiverValue> getVariantsForReceiver(@NotNull ReceiverValue receiverValue) {
038        List<ReceiverValue> variants = Lists.newArrayList(AutoCastUtils.getAutoCastVariants(bindingContext, dataFlowInfo, receiverValue));
039        variants.add(receiverValue);
040        return variants;
041    }
042
043    @NotNull
044    @Override
045    public DataFlowInfo getDataFlowInfo() {
046        return dataFlowInfo;
047    }
048
049    @Override
050    public boolean isNotNull(@NotNull ReceiverValue receiver) {
051        if (!receiver.getType().isNullable()) return true;
052
053        List<ReceiverValue> autoCastVariants = AutoCastUtils.getAutoCastVariants(bindingContext, dataFlowInfo, receiver);
054        for (ReceiverValue autoCastVariant : autoCastVariants) {
055            if (!autoCastVariant.getType().isNullable()) return true;
056        }
057        return false;
058    }
059}