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.descriptors.serialization.descriptors;
018    
019    import org.jetbrains.annotations.NotNull;
020    import org.jetbrains.jet.descriptors.serialization.DescriptorDeserializer;
021    import org.jetbrains.jet.descriptors.serialization.Flags;
022    import org.jetbrains.jet.descriptors.serialization.NameResolver;
023    import org.jetbrains.jet.descriptors.serialization.ProtoBuf;
024    import org.jetbrains.jet.lang.descriptors.DeclarationDescriptor;
025    import org.jetbrains.jet.lang.descriptors.SimpleFunctionDescriptor;
026    import org.jetbrains.jet.lang.descriptors.annotations.Annotations;
027    import org.jetbrains.jet.lang.descriptors.impl.FunctionDescriptorImpl;
028    import org.jetbrains.jet.lang.descriptors.impl.SimpleFunctionDescriptorImpl;
029    import org.jetbrains.jet.lang.resolve.name.Name;
030    
031    public class DeserializedSimpleFunctionDescriptor extends SimpleFunctionDescriptorImpl {
032    
033        private final ProtoBuf.Callable functionProto;
034        private final NameResolver nameResolver;
035    
036        private DeserializedSimpleFunctionDescriptor(
037                @NotNull DeclarationDescriptor containingDeclaration,
038                @NotNull Annotations annotations,
039                @NotNull Name name,
040                @NotNull Kind kind,
041                @NotNull ProtoBuf.Callable functionProto,
042                @NotNull NameResolver nameResolver
043        ) {
044            super(containingDeclaration, annotations, name, kind);
045            this.functionProto = functionProto;
046            this.nameResolver = nameResolver;
047        }
048    
049        private DeserializedSimpleFunctionDescriptor(
050                @NotNull DeclarationDescriptor containingDeclaration,
051                @NotNull SimpleFunctionDescriptor original,
052                @NotNull Annotations annotations,
053                @NotNull Name name,
054                @NotNull Kind kind,
055                @NotNull ProtoBuf.Callable functionProto,
056                @NotNull NameResolver nameResolver) {
057            super(containingDeclaration, original, annotations, name, kind);
058            this.functionProto = functionProto;
059            this.nameResolver = nameResolver;
060        }
061    
062        public DeserializedSimpleFunctionDescriptor(
063                @NotNull DeclarationDescriptor containingDeclaration,
064                @NotNull ProtoBuf.Callable functionProto,
065                @NotNull AnnotationDeserializer annotationDeserializer,
066                @NotNull NameResolver nameResolver
067        ) {
068            this(containingDeclaration,
069                 DescriptorDeserializer.getAnnotations(containingDeclaration, functionProto, functionProto.getFlags(),
070                                                       AnnotationDeserializer.AnnotatedCallableKind.FUNCTION, annotationDeserializer,
071                                                       nameResolver),
072                 nameResolver.getName(functionProto.getName()),
073                 DescriptorDeserializer.memberKind(Flags.MEMBER_KIND.get(functionProto.getFlags())),
074                 functionProto,
075                 nameResolver);
076        }
077    
078        @Override
079        protected FunctionDescriptorImpl createSubstitutedCopy(DeclarationDescriptor newOwner, boolean preserveOriginal, Kind kind) {
080            if (preserveOriginal) {
081                return new DeserializedSimpleFunctionDescriptor(
082                        newOwner,
083                        getOriginal(),
084                        getAnnotations(),
085                        getName(),
086                        kind,
087                        functionProto,
088                        nameResolver
089                );
090            }
091            else {
092                return new DeserializedSimpleFunctionDescriptor(
093                        newOwner,
094                        getAnnotations(),
095                        getName(),
096                        kind,
097                        functionProto,
098                        nameResolver
099                );
100            }
101        }
102    
103        @NotNull
104        @Override
105        public DeserializedSimpleFunctionDescriptor getOriginal() {
106            return (DeserializedSimpleFunctionDescriptor) super.getOriginal();
107        }
108    
109        public ProtoBuf.Callable getFunctionProto() {
110            return functionProto;
111        }
112    
113        public NameResolver getNameResolver() {
114            return nameResolver;
115        }
116    }