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.JetStubElementTypes;
026    
027    public class JetDelegationSpecifier extends JetElementImplStub<KotlinPlaceHolderStub<? extends JetDelegationSpecifier>> {
028    
029        private static final JetDelegationSpecifier[] EMPTY_ARRAY = new JetDelegationSpecifier[0];
030    
031        public static ArrayFactory<JetDelegationSpecifier> ARRAY_FACTORY = new ArrayFactory<JetDelegationSpecifier>() {
032            @NotNull
033            @Override
034            public JetDelegationSpecifier[] create(int count) {
035                return count == 0 ? EMPTY_ARRAY : new JetDelegationSpecifier[count];
036            }
037        };
038    
039        public JetDelegationSpecifier(@NotNull ASTNode node) {
040            super(node);
041        }
042    
043        public JetDelegationSpecifier(
044                @NotNull KotlinPlaceHolderStub<? extends JetDelegationSpecifier> stub,
045                @NotNull IStubElementType nodeType) {
046            super(stub, nodeType);
047        }
048    
049        @Override
050        public <R, D> R accept(@NotNull JetVisitor<R, D> visitor, D data) {
051            return visitor.visitDelegationSpecifier(this, data);
052        }
053    
054        @Nullable
055        public JetTypeReference getTypeReference() {
056            return getStubOrPsiChild(JetStubElementTypes.TYPE_REFERENCE);
057        }
058    
059        @Nullable
060        public JetUserType getTypeAsUserType() {
061            JetTypeReference reference = getTypeReference();
062            if (reference != null) {
063                JetTypeElement element = reference.getTypeElement();
064                if (element instanceof JetUserType) {
065                    return ((JetUserType) element);
066                }
067            }
068            return null;
069        }
070    }