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.annotations.Nullable;
021    import org.jetbrains.jet.descriptors.serialization.*;
022    import org.jetbrains.jet.lang.descriptors.ClassDescriptor;
023    import org.jetbrains.jet.lang.descriptors.DeclarationDescriptor;
024    import org.jetbrains.jet.lang.descriptors.NamespaceDescriptor;
025    import org.jetbrains.jet.lang.descriptors.ReceiverParameterDescriptor;
026    import org.jetbrains.jet.lang.resolve.DescriptorUtils;
027    import org.jetbrains.jet.lang.resolve.name.FqName;
028    import org.jetbrains.jet.lang.resolve.name.FqNameUnsafe;
029    import org.jetbrains.jet.lang.resolve.name.Name;
030    import org.jetbrains.jet.storage.StorageManager;
031    
032    import java.util.Collection;
033    
034    public class DeserializedPackageMemberScope extends DeserializedMemberScope {
035        private final DescriptorFinder descriptorFinder;
036    
037        private final FqName packageFqName;
038    
039        public DeserializedPackageMemberScope(
040                @NotNull StorageManager storageManager,
041                @NotNull NamespaceDescriptor packageDescriptor,
042                @NotNull AnnotationDeserializer annotationDeserializer,
043                @NotNull DescriptorFinder descriptorFinder,
044                @NotNull ProtoBuf.Package proto,
045                @NotNull NameResolver nameResolver
046        ) {
047            super(storageManager, packageDescriptor,
048                  DescriptorDeserializer.create(storageManager, packageDescriptor, nameResolver, descriptorFinder, annotationDeserializer),
049                  proto.getMemberList());
050            this.descriptorFinder = descriptorFinder;
051            this.packageFqName = DescriptorUtils.getFQName(packageDescriptor).toSafe();
052        }
053    
054        public DeserializedPackageMemberScope(
055                @NotNull StorageManager storageManager,
056                @NotNull NamespaceDescriptor packageDescriptor,
057                @NotNull AnnotationDeserializer annotationDeserializer,
058                @NotNull DescriptorFinder descriptorFinder,
059                @NotNull PackageData packageData
060        ) {
061            this(storageManager, packageDescriptor, annotationDeserializer, descriptorFinder, packageData.getPackageProto(),
062                 packageData.getNameResolver());
063        }
064    
065        @Nullable
066        @Override
067        protected ClassDescriptor getClassDescriptor(@NotNull Name name) {
068            return descriptorFinder.findClass(new ClassId(packageFqName, FqNameUnsafe.topLevel(name)));
069        }
070    
071        @Override
072        protected void addAllClassDescriptors(@NotNull Collection<DeclarationDescriptor> result) {
073            for (Name className : descriptorFinder.getClassNames(packageFqName)) {
074                ClassDescriptor classDescriptor = getClassDescriptor(className);
075    
076                if (classDescriptor != null) {
077                    result.add(classDescriptor);
078                }
079            }
080        }
081    
082        @Override
083        protected void addNonDeclaredDescriptors(@NotNull Collection<DeclarationDescriptor> result) {
084            // Do nothing
085        }
086    
087        @Nullable
088        @Override
089        public NamespaceDescriptor getNamespace(@NotNull Name name) {
090            return descriptorFinder.findPackage(packageFqName.child(name));
091        }
092    
093        @Nullable
094        @Override
095        protected ReceiverParameterDescriptor getImplicitReceiver() {
096            return null;
097        }
098    }