001    /*
002     * Copyright 2010-2014 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.lazy.descriptors;
018    
019    import com.google.common.collect.Sets;
020    import org.jetbrains.annotations.NotNull;
021    import org.jetbrains.jet.lang.descriptors.impl.AbstractLazyTypeParameterDescriptor;
022    import org.jetbrains.jet.lang.psi.*;
023    import org.jetbrains.jet.lang.resolve.BindingContext;
024    import org.jetbrains.jet.lang.resolve.DescriptorResolver;
025    import org.jetbrains.jet.lang.resolve.lazy.ForceResolveUtil;
026    import org.jetbrains.jet.lang.resolve.lazy.LazyEntity;
027    import org.jetbrains.jet.lang.resolve.lazy.ResolveSession;
028    import org.jetbrains.jet.lang.resolve.lazy.ResolveSessionUtils;
029    import org.jetbrains.jet.lang.types.JetType;
030    import org.jetbrains.jet.lang.types.lang.KotlinBuiltIns;
031    import org.jetbrains.jet.lexer.JetTokens;
032    
033    import java.util.Set;
034    
035    public class LazyTypeParameterDescriptor extends AbstractLazyTypeParameterDescriptor implements LazyEntity {
036        private final ResolveSession resolveSession;
037    
038        private final JetTypeParameter jetTypeParameter;
039    
040        public LazyTypeParameterDescriptor(
041                @NotNull ResolveSession resolveSession,
042                @NotNull LazyClassDescriptor containingDeclaration,
043                @NotNull JetTypeParameter jetTypeParameter,
044                int index) {
045            super(
046                    resolveSession.getStorageManager(),
047                    containingDeclaration,
048                    ResolveSessionUtils.safeNameForLazyResolve(jetTypeParameter.getNameAsName()),
049                    jetTypeParameter.getVariance(),
050                    jetTypeParameter.hasModifier(JetTokens.REIFIED_KEYWORD),
051                    index
052            );
053            this.resolveSession = resolveSession;
054            this.jetTypeParameter = jetTypeParameter;
055    
056            this.resolveSession.getTrace().record(BindingContext.TYPE_PARAMETER, jetTypeParameter, this);
057        }
058    
059        @NotNull
060        @Override
061        protected Set<JetType> resolveUpperBounds() {
062            Set<JetType> upperBounds = Sets.newLinkedHashSet();
063    
064            JetTypeParameter jetTypeParameter = this.jetTypeParameter;
065    
066            resolveUpperBoundsFromWhereClause(upperBounds);
067    
068            JetTypeReference extendsBound = jetTypeParameter.getExtendsBound();
069            if (extendsBound != null) {
070                upperBounds.add(resolveBoundType(extendsBound));
071            }
072    
073            if (upperBounds.isEmpty()) {
074                upperBounds.add(KotlinBuiltIns.getInstance().getDefaultBound());
075            }
076    
077            return upperBounds;
078        }
079    
080        private void resolveUpperBoundsFromWhereClause(Set<JetType> upperBounds) {
081            JetClassOrObject classOrObject = JetStubbedPsiUtil.getPsiOrStubParent(jetTypeParameter, JetClassOrObject.class, true);
082            if (classOrObject instanceof JetClass) {
083                JetClass jetClass = (JetClass) classOrObject;
084                for (JetTypeConstraint jetTypeConstraint : jetClass.getTypeConstraints()) {
085                    DescriptorResolver.reportUnsupportedClassObjectConstraint(resolveSession.getTrace(), jetTypeConstraint);
086    
087                    JetSimpleNameExpression constrainedParameterName = jetTypeConstraint.getSubjectTypeParameterName();
088                    if (constrainedParameterName != null) {
089                        if (getName().equals(constrainedParameterName.getReferencedNameAsName())) {
090                            resolveSession.getTrace().record(BindingContext.REFERENCE_TARGET, constrainedParameterName, this);
091    
092                            JetTypeReference boundTypeReference = jetTypeConstraint.getBoundTypeReference();
093                            if (boundTypeReference != null) {
094                                JetType boundType = resolveBoundType(boundTypeReference);
095                                if (!jetTypeConstraint.isClassObjectConstraint()) {
096                                    upperBounds.add(boundType);
097                                }
098                            }
099                        }
100                    }
101                }
102            }
103    
104        }
105    
106        private JetType resolveBoundType(@NotNull JetTypeReference boundTypeReference) {
107            return resolveSession.getTypeResolver()
108                        .resolveType(getContainingDeclaration().getScopeForClassHeaderResolution(), boundTypeReference,
109                                     resolveSession.getTrace(), false);
110        }
111    
112        @NotNull
113        @Override
114        public LazyClassDescriptor getContainingDeclaration() {
115            return (LazyClassDescriptor) super.getContainingDeclaration();
116        }
117    
118        @Override
119        public void forceResolveAllContents() {
120            ForceResolveUtil.forceResolveAllContents(getAnnotations());
121            ForceResolveUtil.forceResolveAllContents(getClassObjectType());
122            getContainingDeclaration();
123            getDefaultType();
124            getIndex();
125            ForceResolveUtil.forceResolveAllContents(getLowerBounds());
126            getOriginal();
127            ForceResolveUtil.forceResolveAllContents(getTypeConstructor());
128            ForceResolveUtil.forceResolveAllContents(getUpperBounds());
129            getUpperBoundsAsType();
130            getVariance();
131        }
132    }