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.Annotated;
025    import org.jetbrains.jet.lang.descriptors.annotations.AnnotationDescriptor;
026    import org.jetbrains.jet.lang.resolve.constants.CompileTimeConstant;
027    import org.jetbrains.jet.lang.resolve.constants.EnumValue;
028    
029    import java.util.List;
030    
031    public class InlineUtil {
032    
033        public static boolean DEFAULT_INLINE_FLAG = false;
034    
035        public static boolean DEFAULT_INLINE_FLAG_FOR_TEST = true;
036    
037        public static boolean DEFAULT_INLINE_FLAG_FOR_TOOLWINDOW = false;
038    
039        public static boolean DEFAULT_INLINE_FLAG_FOR_STUB = false; /*always false*/
040    
041        @NotNull
042        public static InlineStrategy getInlineType(@NotNull Annotated annotated) {
043            return getInlineType(annotated.getAnnotations());
044        }
045    
046        public static boolean hasNoinlineAnnotation(@NotNull CallableDescriptor valueParameterDescriptor) {
047            KotlinBuiltIns builtIns = KotlinBuiltIns.getInstance();
048            return KotlinBuiltIns.containsAnnotation(valueParameterDescriptor, builtIns.getNoinlineClassAnnotation());
049        }
050    
051        @NotNull
052        public static InlineStrategy getInlineType(@Nullable List<AnnotationDescriptor> annotations) {
053            KotlinBuiltIns builtIns = KotlinBuiltIns.getInstance();
054            ClassDescriptor annotationClass = builtIns.getInlineClassAnnotation();
055            AnnotationDescriptor annotation = getAnnotation(annotations, annotationClass);
056            if (annotation != null) {
057                ValueParameterDescriptor parameterDescriptor = annotationClass.getConstructors().iterator().next().getValueParameters().get(0);
058                CompileTimeConstant<?> argument = annotation.getValueArgument(parameterDescriptor);
059                if (argument == null) {
060                    //default parameter
061                    return InlineStrategy.AS_FUNCTION;
062                }
063                else {
064                    assert argument instanceof EnumValue : "Inline annotation parameter should be inline entry but was: " + argument + "!";
065                    ClassDescriptor value = ((EnumValue) argument).getValue();
066                    String name = value.getName().asString();
067                    return name.equals(InlineStrategy.IN_PLACE.name()) ? InlineStrategy.IN_PLACE : InlineStrategy.AS_FUNCTION;
068                }
069            }
070            else {
071                return InlineStrategy.NOT_INLINE;
072            }
073        }
074    
075        @Nullable
076        private static AnnotationDescriptor getAnnotation(@Nullable List<AnnotationDescriptor> annotations, @NotNull ClassDescriptor annotationClass) {
077            if (annotations != null) {
078                for (AnnotationDescriptor annotation : annotations) {
079                    if (annotationClass.equals(annotation.getType().getConstructor().getDeclarationDescriptor())) {
080                        return annotation;
081                    }
082                }
083            }
084            return null;
085        }
086    
087    
088    }
089