java/java-impl/src/com/intellij/psi/impl/compiled/ClsRepositoryPsiElement.java

   1 /*
   2  * Copyright 2000-2009 JetBrains s.r.o.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package com.intellij.psi.impl.compiled;
  17
  18 import com.intellij.psi.PsiElement;
  19 import com.intellij.psi.PsiFile;
  20 import com.intellij.psi.PsiInvalidElementAccessException;
  21 import com.intellij.psi.PsiManager;
  22 import com.intellij.psi.stubs.PsiFileStub;
  23 import com.intellij.psi.stubs.StubElement;
  24 import com.intellij.util.ArrayUtil;
  25 import org.jetbrains.annotations.NotNull;
  26
  27 import java.util.List;
  28
  29 public abstract class ClsRepositoryPsiElement<T extends StubElement> extends ClsElementImpl {
  30   private final T myStub;
  31
  32   protected ClsRepositoryPsiElement(final T stub) {
  33     myStub = stub;
  34   }
  35
  36   public PsiElement getParent() {
  37     return myStub.getParentStub().getPsi();
  38   }
  39
  40   public PsiManager getManager() {
  41     final PsiFile file = getContainingFile();
  42     if (file == null) throw new PsiInvalidElementAccessException(this);
  43     return file.getManager();
  44   }
  45
  46   public PsiFile getContainingFile() {
  47     StubElement p = myStub;
  48     while (!(p instanceof PsiFileStub)) {
  49       p = p.getParentStub();
  50     }
  51     return (PsiFile)p.getPsi();
  52   }
  53
  54   public T getStub() {
  55     return myStub;
  56   }
  57
  58   @NotNull
  59   public PsiElement[] getChildren() {
  60     final List stubs = getStub().getChildrenStubs();
  61     PsiElement[] children = new PsiElement[stubs.size()];
  62     for (int i = 0; i < stubs.size(); i++) {
  63       children[i] = ((StubElement)stubs.get(i)).getPsi();
  64     }
  65     return children;
  66   }
  67
  68   public PsiElement getFirstChild() {
  69     final List<StubElement> children = getStub().getChildrenStubs();
  70     if (children.isEmpty()) return null;
  71     return children.get(0).getPsi();
  72   }
  73
  74   public PsiElement getLastChild() {
  75     final List<StubElement> children = getStub().getChildrenStubs();
  76     if (children.isEmpty()) return null;
  77     return children.get(children.size() - 1).getPsi();
  78   }
  79
  80   public PsiElement getNextSibling() {
  81     final PsiElement[] psiElements = getParent().getChildren();
  82     final int i = ArrayUtil.indexOf(psiElements, this);
  83     if (i < 0 || i >= psiElements.length - 1) {
  84       return null;
  85     }
  86     return psiElements[i + 1];
  87   }
  88
  89
  90   public PsiElement getPrevSibling() {
  91     final PsiElement[] psiElements = getParent().getChildren();
  92     final int i = ArrayUtil.indexOf(psiElements, this);
  93     if (i < 1) {
  94       return null;
  95     }
  96     return psiElements[i - 1];
  97   }
  98 }