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.codegen;
018
019import org.jetbrains.annotations.NotNull;
020import org.jetbrains.annotations.Nullable;
021import org.jetbrains.jet.lang.descriptors.CallableDescriptor;
022import org.jetbrains.jet.lang.descriptors.CallableMemberDescriptor;
023import org.jetbrains.jet.lang.descriptors.FunctionDescriptor;
024import org.jetbrains.jet.lang.resolve.BindingContext;
025import org.jetbrains.jet.lang.resolve.java.JavaBindingContext;
026
027public class SamCodegenUtil {
028    @Nullable
029    public static FunctionDescriptor getOriginalIfSamAdapter(@NotNull BindingContext bindingContext, @NotNull CallableDescriptor fun) {
030        if (!(fun instanceof FunctionDescriptor)) {
031            return null;
032        }
033        FunctionDescriptor original = ((FunctionDescriptor) fun).getOriginal();
034        if (original.getKind() == CallableMemberDescriptor.Kind.SYNTHESIZED) {
035            return bindingContext.get(JavaBindingContext.SAM_ADAPTER_FUNCTION_TO_ORIGINAL, original);
036        }
037        if (original.getKind() == CallableMemberDescriptor.Kind.FAKE_OVERRIDE) {
038            for (FunctionDescriptor overridden : original.getOverriddenDescriptors()) {
039                FunctionDescriptor originalIfSamAdapter = getOriginalIfSamAdapter(bindingContext, overridden);
040                if (originalIfSamAdapter != null) {
041                    return originalIfSamAdapter;
042                }
043            }
044        }
045        return null;
046    }
047
048    private SamCodegenUtil() {
049    }
050}