001    /*
002     * Copyright 2010-2015 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.kotlin.psi;
018    
019    import com.intellij.lang.ASTNode;
020    import com.intellij.psi.stubs.IStubElementType;
021    import com.intellij.util.ArrayFactory;
022    import org.jetbrains.annotations.NotNull;
023    import org.jetbrains.annotations.Nullable;
024    import org.jetbrains.kotlin.psi.stubs.KotlinPlaceHolderStub;
025    import org.jetbrains.kotlin.psi.stubs.elements.KtStubElementTypes;
026    
027    public class KtSuperTypeListEntry extends KtElementImplStub<KotlinPlaceHolderStub<? extends KtSuperTypeListEntry>> {
028    
029        private static final KtSuperTypeListEntry[] EMPTY_ARRAY = new KtSuperTypeListEntry[0];
030    
031        public static ArrayFactory<KtSuperTypeListEntry> ARRAY_FACTORY = new ArrayFactory<KtSuperTypeListEntry>() {
032            @NotNull
033            @Override
034            public KtSuperTypeListEntry[] create(int count) {
035                return count == 0 ? EMPTY_ARRAY : new KtSuperTypeListEntry[count];
036            }
037        };
038    
039        public KtSuperTypeListEntry(@NotNull ASTNode node) {
040            super(node);
041        }
042    
043        public KtSuperTypeListEntry(
044                @NotNull KotlinPlaceHolderStub<? extends KtSuperTypeListEntry> stub,
045                @NotNull IStubElementType nodeType) {
046            super(stub, nodeType);
047        }
048    
049        @Override
050        public <R, D> R accept(@NotNull KtVisitor<R, D> visitor, D data) {
051            return visitor.visitSuperTypeListEntry(this, data);
052        }
053    
054        @Nullable
055        public KtTypeReference getTypeReference() {
056            return getStubOrPsiChild(KtStubElementTypes.TYPE_REFERENCE);
057        }
058    
059        @Nullable
060        public KtUserType getTypeAsUserType() {
061            KtTypeReference reference = getTypeReference();
062            if (reference != null) {
063                KtTypeElement element = reference.getTypeElement();
064                if (element instanceof KtUserType) {
065                    return ((KtUserType) element);
066                }
067            }
068            return null;
069        }
070    }