java/java-impl/src/com/intellij/psi/impl/source/tree/java/PsiBreakStatementImpl.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.source.tree.java;
  17
  18 import com.intellij.lang.ASTNode;
  19 import com.intellij.openapi.diagnostic.Logger;
  20 import com.intellij.psi.*;
  21 import com.intellij.psi.impl.source.Constants;
  22 import com.intellij.psi.impl.source.PsiLabelReference;
  23 import com.intellij.psi.impl.source.SourceTreeToPsiMap;
  24 import com.intellij.psi.impl.source.tree.ChildRole;
  25 import com.intellij.psi.impl.source.tree.CompositeElement;
  26 import com.intellij.psi.impl.source.tree.CompositePsiElement;
  27 import com.intellij.psi.impl.source.tree.TreeUtil;
  28 import com.intellij.psi.tree.IElementType;
  29 import com.intellij.psi.tree.ChildRoleBase;
  30 import org.jetbrains.annotations.NotNull;
  31
  32 public class PsiBreakStatementImpl extends CompositePsiElement implements PsiBreakStatement, Constants {
  33   private static final Logger LOG = Logger.getInstance("#com.intellij.psi.impl.source.tree.java.PsiBreakStatementImpl");
  34
  35   public PsiBreakStatementImpl() {
  36     super(BREAK_STATEMENT);
  37   }
  38
  39   public PsiIdentifier getLabelIdentifier() {
  40     return (PsiIdentifier)findChildByRoleAsPsiElement(ChildRole.LABEL);
  41   }
  42
  43   public PsiStatement findExitedStatement() {
  44     PsiIdentifier label = getLabelIdentifier();
  45     if (label == null){
  46       for(ASTNode parent = getTreeParent(); parent != null; parent = parent.getTreeParent()){
  47         IElementType i = parent.getElementType();
  48         if (i == FOR_STATEMENT || i == WHILE_STATEMENT || i == DO_WHILE_STATEMENT || i == SWITCH_STATEMENT || i == FOREACH_STATEMENT) {
  49           return (PsiStatement)SourceTreeToPsiMap.treeElementToPsi(parent);
  50         }
  51         else if (i == METHOD || i == CLASS_INITIALIZER) {
  52           return null; // do not pass through anonymous/local class
  53         }
  54       }
  55     }
  56     else{
  57       String labelName = label.getText();
  58       for(CompositeElement parent = getTreeParent(); parent != null; parent = parent.getTreeParent()){
  59         if (parent.getElementType() == LABELED_STATEMENT){
  60           ASTNode statementLabel = parent.findChildByRole(ChildRole.LABEL_NAME);
  61           if (statementLabel.getText().equals(labelName)){
  62             return ((PsiLabeledStatement)SourceTreeToPsiMap.treeElementToPsi(parent)).getStatement();
  63           }
  64         }
  65
  66         if (parent.getElementType() == METHOD || parent.getElementType() == CLASS_INITIALIZER) return null; // do not pass through anonymous/local class
  67       }
  68     }
  69     return null;
  70   }
  71
  72   public ASTNode findChildByRole(int role) {
  73     LOG.assertTrue(ChildRole.isUnique(role));
  74     switch(role){
  75       default:
  76         return null;
  77
  78       case ChildRole.BREAK_KEYWORD:
  79         return findChildByType(BREAK_KEYWORD);
  80
  81       case ChildRole.LABEL:
  82         return findChildByType(IDENTIFIER);
  83
  84       case ChildRole.CLOSING_SEMICOLON:
  85         return TreeUtil.findChildBackward(this, SEMICOLON);
  86     }
  87   }
  88
  89   public int getChildRole(ASTNode child) {
  90     LOG.assertTrue(child.getTreeParent() == this);
  91     IElementType i = child.getElementType();
  92     if (i == BREAK_KEYWORD) {
  93       return ChildRole.BREAK_KEYWORD;
  94     }
  95     else if (i == IDENTIFIER) {
  96       return ChildRole.LABEL;
  97     }
  98     else if (i == SEMICOLON) {
  99       return ChildRole.CLOSING_SEMICOLON;
 100     }
 101     else {
 102       return ChildRoleBase.NONE;
 103     }
 104   }
 105
 106   public void accept(@NotNull PsiElementVisitor visitor) {
 107     if (visitor instanceof JavaElementVisitor) {
 108       ((JavaElementVisitor)visitor).visitBreakStatement(this);
 109     }
 110     else {
 111       visitor.visitElement(this);
 112     }
 113   }
 114
 115   public PsiReference getReference() {
 116     final PsiReference[] references = getReferences();
 117     if (references != null && references.length > 0)
 118       return references[0];
 119     return null;
 120   }
 121
 122   @NotNull
 123   public PsiReference[] getReferences() {
 124     if (getLabelIdentifier() == null)
 125       return PsiReference.EMPTY_ARRAY;
 126     return new PsiReference[]{new PsiLabelReference(this, getLabelIdentifier())};
 127   }
 128
 129   public String toString() {
 130     return "PsiBreakStatement";
 131   }
 132 }