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.resolve.java.scope;
018    
019    import org.jetbrains.annotations.NotNull;
020    import org.jetbrains.jet.lang.descriptors.*;
021    import org.jetbrains.jet.lang.resolve.java.JavaSemanticServices;
022    import org.jetbrains.jet.lang.resolve.java.provider.ClassPsiDeclarationProvider;
023    import org.jetbrains.jet.lang.resolve.name.LabelName;
024    import org.jetbrains.jet.lang.resolve.name.Name;
025    
026    import java.util.Collection;
027    import java.util.HashMap;
028    import java.util.Map;
029    import java.util.Set;
030    
031    public abstract class JavaClassMembersScope extends JavaBaseScope {
032        @NotNull
033        protected final ClassPsiDeclarationProvider declarationProvider;
034    
035        private Map<Name, ClassDescriptor> innerClassesMap = null;
036    
037        protected JavaClassMembersScope(
038                @NotNull ClassOrNamespaceDescriptor descriptor,
039                @NotNull ClassPsiDeclarationProvider declarationProvider,
040                @NotNull JavaSemanticServices semanticServices
041        ) {
042            super(descriptor, semanticServices, declarationProvider);
043            this.declarationProvider = declarationProvider;
044        }
045    
046        @NotNull
047        @Override
048        public Collection<DeclarationDescriptor> getDeclarationsByLabel(LabelName labelName) {
049            throw new UnsupportedOperationException(); // TODO
050        }
051    
052    
053        @NotNull
054        @Override
055        protected Set<FunctionDescriptor> computeFunctionDescriptor(@NotNull Name name) {
056            return getResolver().resolveFunctionGroup(name, declarationProvider, descriptor);
057        }
058    
059        @NotNull
060        private Map<Name, ClassDescriptor> getInnerClassesMap() {
061            if (innerClassesMap == null) {
062                Collection<ClassDescriptor> innerClasses = getInnerClasses();
063                innerClassesMap = new HashMap<Name, ClassDescriptor>();
064                for (ClassDescriptor innerClass : innerClasses) {
065                    innerClassesMap.put(innerClass.getName(), innerClass);
066                }
067            }
068            return innerClassesMap;
069        }
070    
071        @NotNull
072        @Override
073        protected Collection<ClassDescriptor> computeInnerClasses() {
074            return getResolver().resolveInnerClasses(declarationProvider);
075        }
076    
077        @Override
078        public ClassDescriptor getObjectDescriptor(@NotNull Name name) {
079            ClassDescriptor innerClass = getInnerClassesMap().get(name);
080            if (innerClass != null && innerClass.getKind().isObject()) {
081                return innerClass;
082            }
083            return null;
084        }
085    
086        @Override
087        public ClassifierDescriptor getClassifier(@NotNull Name name) {
088            ClassDescriptor innerClass = getInnerClassesMap().get(name);
089            if (innerClass == null || innerClass.getKind().isObject()) {
090                return null;
091            }
092            return innerClass;
093        }
094    }