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.psi;
018    
019    import com.intellij.lang.ASTNode;
020    import com.intellij.util.ArrayFactory;
021    import org.jetbrains.annotations.NotNull;
022    import org.jetbrains.annotations.Nullable;
023    import org.jetbrains.jet.JetNodeTypes;
024    import org.jetbrains.jet.lang.psi.stubs.PsiJetTypeParameterStub;
025    import org.jetbrains.jet.lang.psi.stubs.elements.JetStubElementTypes;
026    import org.jetbrains.jet.lang.types.Variance;
027    import org.jetbrains.jet.lexer.JetTokens;
028    
029    public class JetTypeParameter extends JetNamedDeclarationStub<PsiJetTypeParameterStub> {
030        public static final JetTypeParameter[] EMPTY_ARRAY = new JetTypeParameter[0];
031    
032        public static final ArrayFactory<JetTypeParameter> ARRAY_FACTORY = new ArrayFactory<JetTypeParameter>() {
033            @Override
034            public JetTypeParameter[] create(int count) {
035                return count == 0 ? EMPTY_ARRAY : new JetTypeParameter[count];
036            }
037        };
038    
039        public JetTypeParameter(@NotNull ASTNode node) {
040            super(node);
041        }
042    
043        public JetTypeParameter(@NotNull PsiJetTypeParameterStub stub) {
044            super(stub, JetStubElementTypes.TYPE_PARAMETER);
045        }
046    
047        @Override
048        public <R, D> R accept(@NotNull JetVisitor<R, D> visitor, D data) {
049            return visitor.visitTypeParameter(this, data);
050        }
051    
052        @NotNull
053        public Variance getVariance() {
054            PsiJetTypeParameterStub stub = getStub();
055            if (stub != null) {
056                if (stub.isOutVariance()) return Variance.OUT_VARIANCE;
057                if (stub.isInVariance()) return Variance.IN_VARIANCE;
058                return Variance.INVARIANT;
059            }
060    
061            JetModifierList modifierList = getModifierList();
062            if (modifierList == null) return Variance.INVARIANT;
063    
064            if (modifierList.hasModifier(JetTokens.OUT_KEYWORD)) return Variance.OUT_VARIANCE;
065            if (modifierList.hasModifier(JetTokens.IN_KEYWORD)) return Variance.IN_VARIANCE;
066            return Variance.INVARIANT;
067        }
068    
069        @Nullable
070        public JetTypeReference getExtendsBound() {
071            return (JetTypeReference) findChildByType(JetNodeTypes.TYPE_REFERENCE);
072        }
073    }