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.types.lang;
018    
019    import org.jetbrains.annotations.NotNull;
020    import org.jetbrains.annotations.Nullable;
021    import org.jetbrains.jet.lang.descriptors.CallableDescriptor;
022    import org.jetbrains.jet.lang.descriptors.ClassDescriptor;
023    import org.jetbrains.jet.lang.descriptors.ValueParameterDescriptor;
024    import org.jetbrains.jet.lang.descriptors.annotations.AnnotationDescriptor;
025    import org.jetbrains.jet.lang.descriptors.annotations.Annotations;
026    import org.jetbrains.jet.lang.resolve.constants.CompileTimeConstant;
027    import org.jetbrains.jet.lang.resolve.constants.EnumValue;
028    
029    public class InlineUtil {
030    
031        public static boolean hasNoinlineAnnotation(@NotNull CallableDescriptor valueParameterDescriptor) {
032            KotlinBuiltIns builtIns = KotlinBuiltIns.getInstance();
033            return KotlinBuiltIns.containsAnnotation(valueParameterDescriptor, builtIns.getNoinlineClassAnnotation());
034        }
035    
036        @NotNull
037        public static InlineStrategy getInlineType(@Nullable Annotations annotations) {
038            KotlinBuiltIns builtIns = KotlinBuiltIns.getInstance();
039            ClassDescriptor annotationClass = builtIns.getInlineClassAnnotation();
040            AnnotationDescriptor annotation = getAnnotation(annotations, annotationClass);
041            if (annotation != null) {
042                ValueParameterDescriptor parameterDescriptor = annotationClass.getConstructors().iterator().next().getValueParameters().get(0);
043                CompileTimeConstant<?> argument = annotation.getValueArgument(parameterDescriptor);
044                if (argument == null) {
045                    //default parameter
046                    return InlineStrategy.AS_FUNCTION;
047                }
048                else {
049                    assert argument instanceof EnumValue : "Inline annotation parameter should be inline entry but was: " + argument + "!";
050                    String name = ((EnumValue) argument).getValue().getName().asString();
051                    return name.equals(InlineStrategy.IN_PLACE.name()) ? InlineStrategy.IN_PLACE : InlineStrategy.AS_FUNCTION;
052                }
053            }
054            else {
055                return InlineStrategy.NOT_INLINE;
056            }
057        }
058    
059        @Nullable
060        private static AnnotationDescriptor getAnnotation(@Nullable Annotations annotations, @NotNull ClassDescriptor annotationClass) {
061            if (annotations != null) {
062                for (AnnotationDescriptor annotation : annotations) {
063                    if (annotationClass.equals(annotation.getType().getConstructor().getDeclarationDescriptor())) {
064                        return annotation;
065                    }
066                }
067            }
068            return null;
069        }
070    
071    
072    }
073