java/java-impl/src/com/intellij/psi/impl/source/tree/java/MethodElement.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.JavaTokenType;
  21 import com.intellij.psi.TokenType;
  22 import com.intellij.psi.impl.PsiImplUtil;
  23 import com.intellij.psi.impl.source.Constants;
  24 import com.intellij.psi.impl.source.tree.*;
  25 import com.intellij.psi.tree.ChildRoleBase;
  26 import com.intellij.psi.tree.IElementType;
  27 import com.intellij.util.CharTable;
  28 import org.jetbrains.annotations.NotNull;
  29
  30 public class MethodElement extends CompositeElement implements Constants {
  31   private static final Logger LOG = Logger.getInstance("#com.intellij.psi.impl.source.tree.java.MethodElement");
  32
  33   public MethodElement() {
  34     super(METHOD);
  35   }
  36
  37   protected MethodElement(IElementType type) {
  38     super(type);
  39   }
  40
  41   public int getTextOffset() {
  42     return findChildByRole(ChildRole.NAME).getStartOffset();
  43   }
  44
  45   public TreeElement addInternal(TreeElement first, ASTNode last, ASTNode anchor, Boolean before) {
  46     if (first == last && first.getElementType() == ElementType.CODE_BLOCK){
  47       ASTNode semicolon = findChildByRole(ChildRole.CLOSING_SEMICOLON);
  48       if (semicolon != null){
  49         deleteChildInternal(semicolon);
  50       }
  51     }
  52     return super.addInternal(first, last, anchor, before);
  53   }
  54
  55   public void deleteChildInternal(@NotNull ASTNode child) {
  56     if (child.getElementType() == CODE_BLOCK){
  57       final ASTNode prevWS = TreeUtil.prevLeaf(child);
  58       if (prevWS != null && prevWS.getElementType() == TokenType.WHITE_SPACE) {
  59         removeChild(prevWS);
  60       }
  61       super.deleteChildInternal(child);
  62       final CharTable treeCharTab = SharedImplUtil.findCharTableByTree(this);
  63       LeafElement semicolon = Factory.createSingleLeafElement(SEMICOLON, ";", 0, 1, treeCharTab, getManager());
  64       addInternal(semicolon, semicolon, null, Boolean.TRUE);
  65     }
  66     else {
  67       super.deleteChildInternal(child);
  68     }
  69   }
  70
  71   public ASTNode findChildByRole(int role){
  72     LOG.assertTrue(ChildRole.isUnique(role));
  73     switch(role){
  74       default:
  75         return null;
  76
  77       case ChildRole.DOC_COMMENT:
  78         return PsiImplUtil.findDocComment(this);
  79
  80       case ChildRole.MODIFIER_LIST:
  81         return findChildByType(MODIFIER_LIST);
  82
  83       case ChildRole.TYPE_PARAMETER_LIST:
  84         return findChildByType(TYPE_PARAMETER_LIST);
  85
  86       case ChildRole.NAME:
  87         return findChildByType(IDENTIFIER);
  88
  89       case ChildRole.TYPE:
  90         return findChildByType(TYPE);
  91
  92       case ChildRole.METHOD_BODY:
  93         return findChildByType(CODE_BLOCK);
  94
  95       case ChildRole.PARAMETER_LIST:
  96         return findChildByType(PARAMETER_LIST);
  97
  98       case ChildRole.THROWS_LIST:
  99         return findChildByType(THROWS_LIST);
 100
 101       case ChildRole.CLOSING_SEMICOLON:
 102         return TreeUtil.findChildBackward(this, SEMICOLON);
 103     }
 104   }
 105
 106   public int getChildRole(ASTNode child) {
 107     LOG.assertTrue(child.getTreeParent() == this);
 108     IElementType i = child.getElementType();
 109     if (i == JavaTokenType.DOC_COMMENT || i == JavaDocElementType.DOC_COMMENT) {
 110       return getChildRole(child, ChildRole.DOC_COMMENT);
 111     }
 112     else if (i == C_STYLE_COMMENT || i == END_OF_LINE_COMMENT) {
 113       {
 114         return ChildRoleBase.NONE;
 115       }
 116     }
 117     else if (i == MODIFIER_LIST) {
 118       return ChildRole.MODIFIER_LIST;
 119     }
 120     else if (i == TYPE_PARAMETER_LIST) {
 121       return ChildRole.TYPE_PARAMETER_LIST;
 122     }
 123     else if (i == CODE_BLOCK) {
 124       return ChildRole.METHOD_BODY;
 125     }
 126     else if (i == PARAMETER_LIST) {
 127       return ChildRole.PARAMETER_LIST;
 128     }
 129     else if (i == THROWS_LIST) {
 130       return ChildRole.THROWS_LIST;
 131     }
 132     else if (i == TYPE) {
 133       return getChildRole(child, ChildRole.TYPE);
 134     }
 135     else if (i == IDENTIFIER) {
 136       return getChildRole(child, ChildRole.NAME);
 137     }
 138     else if (i == SEMICOLON) {
 139       return getChildRole(child, ChildRole.CLOSING_SEMICOLON);
 140     }
 141     else {
 142       return ChildRoleBase.NONE;
 143     }
 144   }
 145 }