java/java-impl/src/com/intellij/psi/impl/source/tree/java/PsiSwitchStatementImpl.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.tree.ChildRole;
  23 import com.intellij.psi.impl.source.tree.CompositePsiElement;
  24 import com.intellij.psi.tree.IElementType;
  25 import com.intellij.psi.tree.ChildRoleBase;
  26 import org.jetbrains.annotations.NotNull;
  27
  28 public class PsiSwitchStatementImpl extends CompositePsiElement implements PsiSwitchStatement, Constants {
  29   private static final Logger LOG = Logger.getInstance("#com.intellij.psi.impl.source.tree.java.PsiSwitchStatementImpl");
  30
  31   public PsiSwitchStatementImpl() {
  32     super(SWITCH_STATEMENT);
  33   }
  34
  35   public PsiExpression getExpression() {
  36     return (PsiExpression)findChildByRoleAsPsiElement(ChildRole.SWITCH_EXPRESSION);
  37   }
  38
  39   public PsiCodeBlock getBody() {
  40     return (PsiCodeBlock)findChildByRoleAsPsiElement(ChildRole.SWITCH_BODY);
  41   }
  42
  43   public PsiJavaToken getLParenth() {
  44     return (PsiJavaToken)findChildByRoleAsPsiElement(ChildRole.LPARENTH);
  45   }
  46
  47   public PsiJavaToken getRParenth() {
  48     return (PsiJavaToken)findChildByRoleAsPsiElement(ChildRole.RPARENTH);
  49   }
  50
  51   public ASTNode findChildByRole(int role) {
  52     LOG.assertTrue(ChildRole.isUnique(role));
  53     switch(role){
  54       default:
  55         return null;
  56
  57       case ChildRole.SWITCH_KEYWORD:
  58         return findChildByType(SWITCH_KEYWORD);
  59
  60       case ChildRole.LPARENTH:
  61         return findChildByType(LPARENTH);
  62
  63       case ChildRole.SWITCH_EXPRESSION:
  64         return findChildByType(EXPRESSION_BIT_SET);
  65
  66       case ChildRole.RPARENTH:
  67         return findChildByType(RPARENTH);
  68
  69       case ChildRole.SWITCH_BODY:
  70         return findChildByType(CODE_BLOCK);
  71     }
  72   }
  73
  74   public int getChildRole(ASTNode child) {
  75     LOG.assertTrue(child.getTreeParent() == this);
  76     IElementType i = child.getElementType();
  77     if (i == SWITCH_KEYWORD) {
  78       return ChildRole.SWITCH_KEYWORD;
  79     }
  80     else if (i == LPARENTH) {
  81       return ChildRole.LPARENTH;
  82     }
  83     else if (i == RPARENTH) {
  84       return ChildRole.RPARENTH;
  85     }
  86     else {
  87       if (EXPRESSION_BIT_SET.contains(child.getElementType())) {
  88         return ChildRole.SWITCH_EXPRESSION;
  89       }
  90       else if (child.getElementType() == CODE_BLOCK) {
  91         return ChildRole.SWITCH_BODY;
  92       }
  93       else {
  94         return ChildRoleBase.NONE;
  95       }
  96     }
  97   }
  98
  99   public void accept(@NotNull PsiElementVisitor visitor) {
 100     if (visitor instanceof JavaElementVisitor) {
 101       ((JavaElementVisitor)visitor).visitSwitchStatement(this);
 102     }
 103     else {
 104       visitor.visitElement(this);
 105     }
 106   }
 107
 108   public String toString() {
 109     return "PsiSwitchStatement";
 110   }
 111 }