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 final boolean DEFAULT_INLINE_FLAG = true;
032    
033        public static final boolean DEFAULT_INLINE_FLAG_FOR_TEST = true;
034    
035        public static final boolean DEFAULT_INLINE_FLAG_FOR_STUB = false; /*always false*/
036    
037        public static boolean optionToInlineFlag(@Nullable String option) {
038            return ("on".equalsIgnoreCase(option) || "off".equalsIgnoreCase(option)) ? "on".equalsIgnoreCase(option) : DEFAULT_INLINE_FLAG;
039        }
040    
041        public static boolean hasNoinlineAnnotation(@NotNull CallableDescriptor valueParameterDescriptor) {
042            KotlinBuiltIns builtIns = KotlinBuiltIns.getInstance();
043            return KotlinBuiltIns.containsAnnotation(valueParameterDescriptor, builtIns.getNoinlineClassAnnotation());
044        }
045    
046        @NotNull
047        public static InlineStrategy getInlineType(@Nullable Annotations annotations) {
048            KotlinBuiltIns builtIns = KotlinBuiltIns.getInstance();
049            ClassDescriptor annotationClass = builtIns.getInlineClassAnnotation();
050            AnnotationDescriptor annotation = getAnnotation(annotations, annotationClass);
051            if (annotation != null) {
052                ValueParameterDescriptor parameterDescriptor = annotationClass.getConstructors().iterator().next().getValueParameters().get(0);
053                CompileTimeConstant<?> argument = annotation.getValueArgument(parameterDescriptor);
054                if (argument == null) {
055                    //default parameter
056                    return InlineStrategy.AS_FUNCTION;
057                }
058                else {
059                    assert argument instanceof EnumValue : "Inline annotation parameter should be inline entry but was: " + argument + "!";
060                    String name = ((EnumValue) argument).getValue().getName().asString();
061                    return name.equals(InlineStrategy.IN_PLACE.name()) ? InlineStrategy.IN_PLACE : InlineStrategy.AS_FUNCTION;
062                }
063            }
064            else {
065                return InlineStrategy.NOT_INLINE;
066            }
067        }
068    
069        @Nullable
070        private static AnnotationDescriptor getAnnotation(@Nullable Annotations annotations, @NotNull ClassDescriptor annotationClass) {
071            if (annotations != null) {
072                for (AnnotationDescriptor annotation : annotations) {
073                    if (annotationClass.equals(annotation.getType().getConstructor().getDeclarationDescriptor())) {
074                        return annotation;
075                    }
076                }
077            }
078            return null;
079        }
080    
081    
082    }
083