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.extapi.psi.StubBasedPsiElementBase;
020 import com.intellij.lang.ASTNode;
021 import com.intellij.lang.Language;
022 import com.intellij.psi.*;
023 import com.intellij.psi.impl.source.resolve.reference.ReferenceProvidersRegistry;
024 import com.intellij.psi.PsiElement;
025 import com.intellij.psi.PsiElementVisitor;
026 import com.intellij.psi.PsiFile;
027 import com.intellij.psi.StubBasedPsiElement;
028 import com.intellij.psi.stubs.IStubElementType;
029 import com.intellij.psi.stubs.StubElement;
030 import org.jetbrains.annotations.NotNull;
031 import org.jetbrains.jet.lang.psi.stubs.elements.JetStubElementType;
032 import org.jetbrains.jet.plugin.JetLanguage;
033
034 import java.util.Arrays;
035 import java.util.List;
036
037 public class JetElementImplStub<T extends StubElement> extends StubBasedPsiElementBase<T>
038 implements JetElement, StubBasedPsiElement<T> {
039 public JetElementImplStub(@NotNull T stub, @NotNull IStubElementType nodeType) {
040 super(stub, nodeType);
041 }
042
043 public JetElementImplStub(@NotNull ASTNode node) {
044 super(node);
045 }
046
047 @NotNull
048 @Override
049 public Language getLanguage() {
050 return JetLanguage.INSTANCE;
051 }
052
053 @Override
054 public String toString() {
055 return getElementType().toString();
056 }
057
058 @Override
059 public final void accept(@NotNull PsiElementVisitor visitor) {
060 if (visitor instanceof JetVisitor) {
061 accept((JetVisitor) visitor, null);
062 }
063 else {
064 visitor.visitElement(this);
065 }
066 }
067
068 @NotNull
069 @Override
070 public JetFile getContainingJetFile() {
071 PsiFile file = getContainingFile();
072 assert file instanceof JetFile : "JetElement not inside JetFile: " + file + " " + file.getText();
073 return (JetFile) file;
074 }
075
076 @Override
077 public <D> void acceptChildren(@NotNull JetTreeVisitor<D> visitor, D data) {
078 PsiElement child = getFirstChild();
079 while (child != null) {
080 if (child instanceof JetElement) {
081 ((JetElement) child).accept(visitor, data);
082 }
083 child = child.getNextSibling();
084 }
085 }
086
087 @Override
088 public <R, D> R accept(@NotNull JetVisitor<R, D> visitor, D data) {
089 return visitor.visitJetElement(this, data);
090 }
091
092 @Override
093 public PsiReference getReference() {
094 PsiReference[] references = getReferences();
095 if (references.length == 1) return references[0];
096 else return null;
097 }
098
099 @NotNull
100 @Override
101 public PsiReference[] getReferences() {
102 return ReferenceProvidersRegistry.getReferencesFromProviders(this, PsiReferenceService.Hints.NO_HINTS);
103 }
104
105 @NotNull
106 protected <PsiT extends JetElementImplStub<?>, StubT extends StubElement> List<PsiT> getStubOrPsiChildrenAsList(
107 @NotNull JetStubElementType<StubT, PsiT> elementType
108 ) {
109 return Arrays.asList(getStubOrPsiChildren(elementType, elementType.getArrayFactory()));
110 }
111 }